def writeMeshFile(triangles, filename, binary=True, verbose=False):
"""Write mesh file.
The output format is determined by file name extension. Files can be written
in binary (default) and ASCII format."""
outformat = path.splitext(options.outfilename)[1].strip('.')
# set writer based on filename extension
if outformat=='stl':
write = vtk.vtkSTLWriter()
elif outformat=='vtk':
write = vtk.vtkPolyDataWriter()
elif outformat=='obj':
write = vtk.vtkMNIObjectWriter()
elif outformat=='tag':
write = vtk.vtkMNITagPointWriter()
else:
raise ValueError('cannot write outpur format' + outformat)
write.SetInputConnection(triangles.GetOutputPort())
if outformat!='tag':
if binary:
if verbose: print 'setting ouptut to binary'
write.SetFileTypeToBinary()
else:
if verbose: print 'setting ouptut to ascii'
write.SetFileTypeToASCII()
write.SetFileName(filename)
err = write.Write()
if err != 1:
raise IOError('failed to write')
if verbose:
print "wrote", filename
pass
def writeMeshFile(triangles, filename, binary=True, verbose=False):
"""Write mesh file.
The output format is determined by file name extension. Files can be written
in binary (default) and ASCII format."""
outformat = path.splitext(options.outfilename)[1].strip('.')
# set writer based on filename extension
if outformat == 'stl':
write = vtk.vtkSTLWriter()
elif outformat == 'vtk':
write = vtk.vtkPolyDataWriter()
elif outformat == 'obj':
write = vtk.vtkMNIObjectWriter()
elif outformat == 'tag':
write = vtk.vtkMNITagPointWriter()
else:
raise ValueError('cannot write outpur format' + outformat)
write.SetInputConnection(triangles.GetOutputPort())
if outformat != 'tag':
if binary:
if verbose: print 'setting ouptut to binary'
write.SetFileTypeToBinary()
else:
if verbose: print 'setting ouptut to ascii'
write.SetFileTypeToASCII()
write.SetFileName(filename)
err = write.Write()
if err != 1:
raise IOError('failed to write')
if verbose:
print "wrote", filename
pass
weights = vtk.vtkDoubleArray()
weights.InsertNextValue(1.0)
weights.InsertNextValue(1.1)
weights.InsertNextValue(1.2)
weights.InsertNextValue(1.3)
weights.InsertNextValue(1.4)
weights.InsertNextValue(1.5)
weights.InsertNextValue(1.6)
weights.InsertNextValue(1.7)
weights.InsertNextValue(1.8)
weights.InsertNextValue(1.9)
weights.InsertNextValue(0.9)
weights.InsertNextValue(0.8)
weights.InsertNextValue(0.7)
fname = "" + str(dir) + "/mni-tagtest.tag"
writer = vtk.vtkMNITagPointWriter()
writer.SetFileName("" + str(fname) + "")
writer.SetInputConnection(sphere1.GetOutputPort())
writer.SetInputConnection(1, xformFilter.GetOutputPort())
writer.SetLabelText(labels)
writer.SetWeights(weights)
writer.SetComments("Volume 1: sphere points\nVolume 2: transformed points")
writer.Write()
reader = vtk.vtkMNITagPointReader()
reader.CanReadFile("" + str(fname) + "")
reader.SetFileName("" + str(fname) + "")
textProp = vtk.vtkTextProperty()
textProp.SetFontSize(12)
textProp.SetColor(1.0, 1.0, 0.5)
labelHier = vtk.vtkPointSetToLabelHierarchy()
labelHier.SetInputConnection(reader.GetOutputPort())
weights = vtk.vtkDoubleArray()
weights.InsertNextValue(1.0)
weights.InsertNextValue(1.1)
weights.InsertNextValue(1.2)
weights.InsertNextValue(1.3)
weights.InsertNextValue(1.4)
weights.InsertNextValue(1.5)
weights.InsertNextValue(1.6)
weights.InsertNextValue(1.7)
weights.InsertNextValue(1.8)
weights.InsertNextValue(1.9)
weights.InsertNextValue(0.9)
weights.InsertNextValue(0.8)
weights.InsertNextValue(0.7)
writer = vtk.vtkMNITagPointWriter()
writer.SetFileName(fname)
writer.SetInputConnection(sphere1.GetOutputPort())
writer.SetInputConnection(1, xformFilter.GetOutputPort())
writer.SetLabelText(labels)
writer.SetWeights(weights)
writer.SetComments("Volume 1: sphere points\nVolume 2: transformed points")
writer.Write()
reader = vtk.vtkMNITagPointReader()
reader.CanReadFile(fname)
reader.SetFileName(fname)
textProp = vtk.vtkTextProperty()
textProp.SetFontSize(12)
textProp.SetColor(1.0, 1.0, 0.5)