|
0
|
1 import os
|
|
|
2 import sys
|
|
|
3 import tempfile
|
|
|
4
|
|
7
|
5 from icqsol.shapes.icqShapeManager import ShapeManager
|
|
|
6 from icqsol.bem.icqLaplaceMatrices import LaplaceMatrices
|
|
|
7
|
|
6
|
8 PLY = 'ply'
|
|
4
|
9 POLYDATA = 'POLYDATA'
|
|
6
|
10 VTK = 'vtk'
|
|
4
|
11
|
|
0
|
12
|
|
4
|
13 def asbool(val):
|
|
|
14 return str(val).lower() in ['yes', 'true']
|
|
0
|
15
|
|
4
|
16
|
|
|
17 def get_format_and_type(galaxy_ext):
|
|
0
|
18 # Define the output file format and type.
|
|
|
19 format = None
|
|
|
20 datatype = None
|
|
4
|
21 if galaxy_ext in ['vtkascii', 'vtkbinary']:
|
|
6
|
22 format = VTK
|
|
4
|
23 elif galaxy_ext in ['plyascii', 'plybinary']:
|
|
6
|
24 format = PLY
|
|
4
|
25 if galaxy_ext in ['vtkascii', 'plyascii']:
|
|
0
|
26 datatype = 'ascii'
|
|
4
|
27 elif galaxy_ext in ['vtkbinary', 'plybinary']:
|
|
0
|
28 datatype = 'binary'
|
|
|
29 return format, datatype
|
|
|
30
|
|
4
|
31
|
|
|
32 def get_input_file_path(tmp_dir, input_file, format):
|
|
0
|
33 """
|
|
|
34 iCqSol uses file extensions (e.g., .ply, .vtk) when reading and
|
|
|
35 writing files, so the Galaxy dataset naming convention of
|
|
|
36 setting all file extensions as .dat must be handled.
|
|
|
37 """
|
|
4
|
38 file_path = get_temporary_file_path(tmp_dir, format)
|
|
0
|
39 # Remove the file so we can create a symlink.
|
|
4
|
40 os.remove(file_path)
|
|
|
41 os.symlink(input_file, file_path)
|
|
0
|
42 return file_path
|
|
|
43
|
|
4
|
44
|
|
7
|
45 def get_laplace_solver(shape_data, max_edge_length=float('inf')):
|
|
|
46 return LaplaceMatrices(shape_data, max_edge_length=max_edge_length)
|
|
|
47
|
|
|
48
|
|
9
|
49 def get_shape_manager(format=None, dataset_type=None):
|
|
7
|
50 # Instantiate a ShapeManager.
|
|
9
|
51 if format is None and dataset_type is None:
|
|
|
52 return ShapeManager()
|
|
7
|
53 if format == VTK:
|
|
|
54 return ShapeManager(file_format=format, vtk_dataset_type=dataset_type)
|
|
|
55 else:
|
|
|
56 return ShapeManager(file_format=format)
|
|
|
57
|
|
|
58
|
|
4
|
59 def get_temp_dir(prefix='tmp-vtk-', dir=None):
|
|
0
|
60 """
|
|
|
61 Return a temporary directory.
|
|
|
62 """
|
|
4
|
63 return tempfile.mkdtemp(prefix=prefix, dir=dir)
|
|
0
|
64
|
|
4
|
65
|
|
|
66 def get_tempfilename(dir=None, suffix=None):
|
|
0
|
67 """
|
|
|
68 Return a temporary file name.
|
|
|
69 """
|
|
6
|
70 if suffix is None:
|
|
|
71 s = None
|
|
|
72 elif suffix.startswith('.'):
|
|
|
73 s = suffix
|
|
|
74 else:
|
|
|
75 s = '.%s' % suffix
|
|
|
76 fd, name = tempfile.mkstemp(suffix=s, dir=dir)
|
|
4
|
77 os.close(fd)
|
|
0
|
78 return name
|
|
|
79
|
|
4
|
80
|
|
|
81 def get_temporary_file_path(tmp_dir, file_extension):
|
|
0
|
82 """
|
|
|
83 Return the path to a temporary file with a valid VTK format
|
|
|
84 file extension.
|
|
|
85 """
|
|
4
|
86 return get_tempfilename(tmp_dir, file_extension)
|
|
|
87
|
|
0
|
88
|
|
4
|
89 def stop_err(msg):
|
|
|
90 sys.stderr.write("%s\n" % msg)
|
|
0
|
91 sys.exit()
|
