From a1927f84c46ef50e9b3ff92463955a92047bbbdc Mon Sep 17 00:00:00 2001 From: Philip Eisenlohr Date: Fri, 22 Apr 2016 13:09:23 -0500 Subject: [PATCH] renamed "coordinates" option to "position". added explicit choice to write out as legacy format. --- processing/post/vtk_pointcloud.py | 30 +++++++++++++++++++++--------- 1 file changed, 21 insertions(+), 9 deletions(-) diff --git a/processing/post/vtk_pointcloud.py b/processing/post/vtk_pointcloud.py index f35911135..0ffa52fa2 100755 --- a/processing/post/vtk_pointcloud.py +++ b/processing/post/vtk_pointcloud.py @@ -18,12 +18,19 @@ Produce a VTK point cloud dataset based on coordinates given in an ASCIItable. """, version = scriptID) -parser.add_option('-c', '--coordinates', +parser.add_option('-p', + '--pos', '--position', dest = 'pos', type = 'string', metavar = 'string', help = 'coordinate label [%default]') +parser.add_option('-l', + '--legacy', + dest = 'legacy', + action = 'store_true', + help = 'force legacy VTK output') -parser.set_defaults(pos = 'pos' +parser.set_defaults(pos = 'pos', + legacy = False, ) (options, filenames) = parser.parse_args() @@ -81,21 +88,26 @@ for name in filenames: # ------------------------------------------ output result --------------------------------------- - if name: + if options.legacy: + writer = vtk.vtkDataSetWriter() + writer.SetHeader('# powered by '+scriptID) + else: writer = vtk.vtkXMLPolyDataWriter() - (directory,filename) = os.path.split(name) writer.SetDataModeToBinary() writer.SetCompressorTypeToZLib() - writer.SetFileName(os.path.join(directory,os.path.splitext(filename)[0]\ - +'.'+writer.GetDefaultFileExtension())) + + if name: + writer.SetFileName(os.path.join(os.path.split(name)[0], + os.path.splitext(os.path.split(name)[1])[0] + + '.' + ('vtk' if options.legacy else writer.GetDefaultFileExtension()))) else: - writer = vtk.vtkDataSetWriter() writer.WriteToOutputStringOn() - writer.SetHeader('# powered by '+scriptID) if vtk.VTK_MAJOR_VERSION <= 5: writer.SetInput(Polydata) else: writer.SetInputData(Polydata) + writer.Write() - if name is None: sys.stdout.write(writer.GetOutputString()[0:writer.GetOutputStringLength()]) + + if name is None: sys.stdout.write(writer.GetOutputString()) #[0:writer.GetOutputStringLength()] table.close()