def __init__(self, module_manager):
SimpleVTKClassModuleBase.__init__(
self, module_manager,
vtk.vtkMINCImageWriter(), 'Writing vtkMINCImage.',
('vtkMINCImage',), (),
replaceDoc=True,
inputFunctions=None, outputFunctions=None)
def getWriter(self,polyData):
writer = None;
if self.destFormat == "vti":
writer = vtk.vtkXMLImageDataWriter();
writer.SetFileName(self.dest);
elif self.destFormat in ("mha","mhd","raw"):
writer = vtk.vtkMetaImageWriter();
writer.SetFilePrefix(self.dest);
writer.SetFileName(self.dest);
elif self.destFormat == "mnc":
writer = vtk.vtkMINCImageWriter();
writer.SetFileName(self.dest);
writer.StrictValidationOff();
elif self.destFormat == "vtk":
writer = vtk.vtkUnstructuredGridWriter();
writer.SetFileName(self.dest);
writer.SetInputData(polyData);
writer.Update();
return writer;
zres = spacing[2]
if zres > 1.5:
bt = 0.50
else:
bt = 0.70
# Optimized parameters
stripper.SetRMin(8.0)
stripper.SetRMax(10.0)
stripper.SetD1(7.0)
stripper.SetD2(3.0)
stripper.SetBT(bt)
stripper.Update()
# Get the basename
name = string.split(filename, "/")[-1]
basename = string.split(name, ".")[0]
# Write the mesh
writer = vtk.vtkPolyDataWriter()
writer.SetFileName(basename + "_mesh.vtk")
writer.SetInput( stripper.GetBrainMesh() )
writer.Write()
# Write the segemented image
writer2 = vtk.vtkMINCImageWriter()
writer2.SetInput( stripper.GetOutput() )
writer2.SetFileName(basename + "_seg.mnc" )
writer2.Write()
VTK_DATA_ROOT = vtkGetDataRoot()
# Image pipeline
reader = vtk.vtkMINCImageReader()
reader.SetFileName(VTK_DATA_ROOT + "/Data/t3_grid_0.mnc")
reader.RescaleRealValuesOn()
attributes = vtk.vtkMINCImageAttributes()
image = reader
# The current directory must be writeable.
#
try:
channel = open("minc1.mnc", "wb")
channel.close()
minc1 = vtk.vtkMINCImageWriter()
minc1.SetInputConnection(reader.GetOutputPort())
minc1.SetFileName("minc1.mnc")
attributes.ShallowCopy(reader.GetImageAttributes())
attributes.SetAttributeValueAsString(
"patient", "full_name", "DOE^JOHN DAVID")
minc2 = vtk.vtkMINCImageWriter()
minc2.SetImageAttributes(attributes)
minc2.SetInputConnection(reader.GetOutputPort())
minc2.SetFileName("minc2.mnc")
minc3 = vtk.vtkMINCImageWriter()
minc3.SetImageAttributes(attributes)
minc3.AddInputConnection(reader.GetOutputPort())
zres = spacing[2]
if zres > 1.5:
bt = 0.50
else:
bt = 0.70
# Optimized parameters
stripper.SetRMin(8.0)
stripper.SetRMax(10.0)
stripper.SetD1(7.0)
stripper.SetD2(3.0)
stripper.SetBT(bt)
stripper.Update()
# Get the basename
name = string.split(filename, "/")[-1]
basename = string.split(name, ".")[0]
# Write the mesh
writer = vtk.vtkPolyDataWriter()
writer.SetFileName(basename + "_mesh.vtk")
writer.SetInput(stripper.GetBrainMesh())
writer.Write()
# Write the segemented image
writer2 = vtk.vtkMINCImageWriter()
writer2.SetInput(stripper.GetOutput())
writer2.SetFileName(basename + "_seg.mnc")
writer2.Write()
VTK_DATA_ROOT = vtkGetDataRoot()
# Image pipeline
reader = vtk.vtkMINCImageReader()
reader.SetFileName(VTK_DATA_ROOT + "/Data/t3_grid_0.mnc")
reader.RescaleRealValuesOn()
attributes = vtk.vtkMINCImageAttributes()
image = reader
# The current directory must be writeable.
#
try:
channel = open("minc1.mnc", "wb")
channel.close()
minc1 = vtk.vtkMINCImageWriter()
minc1.SetInputConnection(reader.GetOutputPort())
minc1.SetFileName("minc1.mnc")
attributes.ShallowCopy(reader.GetImageAttributes())
attributes.SetAttributeValueAsString("patient", "full_name",
"DOE^JOHN DAVID")
minc2 = vtk.vtkMINCImageWriter()
minc2.SetImageAttributes(attributes)
minc2.SetInputConnection(reader.GetOutputPort())
minc2.SetFileName("minc2.mnc")
minc3 = vtk.vtkMINCImageWriter()
minc3.SetImageAttributes(attributes)
minc3.AddInputConnection(reader.GetOutputPort())
def test_get_mnc(self):
'''mnc filetype returns None'''
expected = type(vtk.vtkMINCImageWriter())
self.assertEqual(type(get_vtk_writer('test.mnC')), expected)
