diff icqsol_refine_shape.py @ 2:da0ef88d5af2 draft

Uploaded
author greg
date Fri, 04 Dec 2015 09:24:26 -0500
parents a38daf377d55
children 918771601157
line wrap: on
line diff
--- a/icqsol_refine_shape.py	Fri Dec 04 09:24:17 2015 -0500
+++ b/icqsol_refine_shape.py	Fri Dec 04 09:24:26 2015 -0500
@@ -7,23 +7,23 @@
 
 # Parse Command Line.
 parser = argparse.ArgumentParser()
-parser.add_argument( '--input', dest='input', help='Shape dataset selected from history' )
-parser.add_argument( '--input_file_format_and_type', dest='input_file_format_and_type', help='Input file format and type' )
-parser.add_argument( '--input_dataset_type', dest='input_dataset_type', help='Input dataset_type' )
-parser.add_argument( '--max_edge_length', dest='max_edge_length', type=float, default='0', help='Maximum edge length' )
-parser.add_argument( '--output', dest='output', help='Output dataset' )
-parser.add_argument( '--output_vtk_type', dest='output_vtk_type', help='Output VTK type' )
+parser.add_argument('--input', dest='input', help='Shape dataset selected from history')
+parser.add_argument('--input_file_format_and_type', dest='input_file_format_and_type', help='Input file format and type')
+parser.add_argument('--input_dataset_type', dest='input_dataset_type', help='Input dataset_type')
+parser.add_argument('--max_edge_length', dest='max_edge_length', type=float, default='0', help='Maximum edge length')
+parser.add_argument('--output', dest='output', help='Output dataset')
+parser.add_argument('--output_vtk_type', dest='output_vtk_type', help='Output VTK type')
 
 args = parser.parse_args()
 
-input_format, input_file_type = icqsol_utils.get_format_and_type( args.input_file_format_and_type )
+input_format, input_file_type = icqsol_utils.get_format_and_type(args.input_file_format_and_type)
 tmp_dir = icqsol_utils.get_temp_dir()
 
 # Instantiate a ShapeManager for loading the input.
 if input_format == 'vtk':
-    shape_mgr = ShapeManager( file_format=input_format, vtk_dataset_type=args.input_dataset_type )
+    shape_mgr = ShapeManager(file_format=input_format, vtk_dataset_type=args.input_dataset_type)
 else:
-    shape_mgr = ShapeManager( file_format=input_format )
+    shape_mgr = ShapeManager(file_format=input_format)
 
 # Get the vtk polydata from the input dataset.
 vtk_poly_data = shape_mgr.loadAsVtkPolyData(args.input)
@@ -31,13 +31,13 @@
 # Refine the shape if requested.
 vtk_poly_data = shape_mgr.refineVtkPolyData(vtk_poly_data, max_edge_length=args.max_edge_length)
 
-# Define the output file format and type (the outpur_format can only be 'vtk').
-output_format, output_file_type = icqsol_utils.get_format_and_type( args.output_vtk_type )
-tmp_output_path = icqsol_utils.get_temporary_file_path( tmp_dir, output_format )
+# Define the output file format and type (the output_format can only be 'vtk').
+output_format, output_file_type = icqsol_utils.get_format_and_type(args.output_vtk_type)
+tmp_output_path = icqsol_utils.get_temporary_file_path(tmp_dir, output_format)
 
 # Make sure the ShapeManager's writer is vtk.
-shape_mgr.setWriter( file_format='vtk', vtk_dataset_type=icqsol_utils.POLYDATA )
+shape_mgr.setWriter(file_format='vtk', vtk_dataset_type=icqsol_utils.POLYDATA)
 
 # Save the output.
-shape_mgr.saveVtkPolyData( vtk_poly_data=vtk_poly_data, file_name=tmp_output_path, file_type=output_file_type)
-shutil.move( tmp_output_path, args.output )
+shape_mgr.saveVtkPolyData(vtk_poly_data=vtk_poly_data, file_name=tmp_output_path, file_type=output_file_type)
+shutil.move(tmp_output_path, args.output)