Mercurial > repos > greg > icqsol_refine_shape
changeset 4:5107ac19d47a draft
Uploaded
author | greg |
---|---|
date | Fri, 04 Dec 2015 09:24:43 -0500 |
parents | 5265b8f55dd6 |
children | b2ded1c154a8 |
files | icqsol_utils.py |
diffstat | 1 files changed, 28 insertions(+), 21 deletions(-) [+] |
line wrap: on
line diff
--- a/icqsol_utils.py Fri Dec 04 09:24:35 2015 -0500 +++ b/icqsol_utils.py Fri Dec 04 09:24:43 2015 -0500 @@ -2,58 +2,65 @@ import sys import tempfile -POLYDATA='POLYDATA' +POLYDATA = 'POLYDATA' + -def asbool( val ): - return str( val ).lower() in [ 'yes', 'true' ] +def asbool(val): + return str(val).lower() in ['yes', 'true'] -def get_format_and_type( galaxy_ext ): + +def get_format_and_type(galaxy_ext): # Define the output file format and type. format = None datatype = None - if galaxy_ext in [ 'vtkascii', 'vtkbinary' ]: + if galaxy_ext in ['vtkascii', 'vtkbinary']: format = 'vtk' - elif galaxy_ext in [ 'plyascii', 'plybinary' ]: + elif galaxy_ext in ['plyascii', 'plybinary']: format = 'ply' - if galaxy_ext in [ 'vtkascii', 'plyascii' ]: + if galaxy_ext in ['vtkascii', 'plyascii']: datatype = 'ascii' - elif galaxy_ext in [ 'vtkbinary', 'plybinary' ]: + elif galaxy_ext in ['vtkbinary', 'plybinary']: datatype = 'binary' return format, datatype -def get_input_file_path( tmp_dir, input_file, format ): + +def get_input_file_path(tmp_dir, input_file, format): """ iCqSol uses file extensions (e.g., .ply, .vtk) when reading and writing files, so the Galaxy dataset naming convention of setting all file extensions as .dat must be handled. """ - file_path = get_temporary_file_path( tmp_dir, format ) + file_path = get_temporary_file_path(tmp_dir, format) # Remove the file so we can create a symlink. - os.remove( file_path ) - os.symlink( input_file, file_path ) + os.remove(file_path) + os.symlink(input_file, file_path) return file_path -def get_temp_dir( prefix='tmp-vtk-', dir=None ): + +def get_temp_dir(prefix='tmp-vtk-', dir=None): """ Return a temporary directory. """ - return tempfile.mkdtemp( prefix=prefix, dir=dir ) + return tempfile.mkdtemp(prefix=prefix, dir=dir) -def get_tempfilename( dir=None, suffix=None ): + +def get_tempfilename(dir=None, suffix=None): """ Return a temporary file name. """ - fd, name = tempfile.mkstemp( suffix=suffix, dir=dir ) - os.close( fd ) + fd, name = tempfile.mkstemp(suffix=suffix, dir=dir) + os.close(fd) return name -def get_temporary_file_path( tmp_dir, file_extension ): + +def get_temporary_file_path(tmp_dir, file_extension): """ Return the path to a temporary file with a valid VTK format file extension. """ - return get_tempfilename( tmp_dir, file_extension ) + return get_tempfilename(tmp_dir, file_extension) + -def stop_err( msg ): - sys.stderr.write( "%s\n" % msg ) +def stop_err(msg): + sys.stderr.write("%s\n" % msg) sys.exit()