# HG changeset patch # User greg # Date 1449329195 18000 # Node ID ca481c46294056d34f5eb55d576947cbb8aaaf24 # Parent 0d67a350bb1d3b167784256a46e193084420cd19 Uploaded diff -r 0d67a350bb1d -r ca481c462940 icqsol_add_surface_field_from_expression.py --- a/icqsol_add_surface_field_from_expression.py Sat Dec 05 10:26:26 2015 -0500 +++ b/icqsol_add_surface_field_from_expression.py Sat Dec 05 10:26:35 2015 -0500 @@ -25,7 +25,7 @@ tmp_dir = icqsol_utils.get_temp_dir() # Instantiate a ShapeManager for loading the input. -if input_format == 'vtk': +if input_format == icqsol_utils.VTK: shape_mgr = ShapeManager(file_format=input_format, vtk_dataset_type=args.input_dataset_type) else: shape_mgr = ShapeManager(file_format=input_format) @@ -50,7 +50,7 @@ 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=icqsol_utils.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)