# test the force load function
itk.force_load()
fileName = sys.argv[1]
PixelType = itk.UC
dim = 2
ImageType = itk.Image[PixelType, dim]
ReaderType = itk.ImageFileReader[ImageType]
reader = ReaderType.New(FileName=fileName)
# test snake_case keyword arguments
reader = ReaderType.New(file_name=fileName)
# test echo
itk.echo(reader)
itk.echo(reader, sys.stdout)
# test class_
assert itk.class_(reader) == ReaderType
assert itk.class_("dummy") == str
# test template
assert itk.template(ReaderType) == (itk.ImageFileReader, (ImageType,))
assert itk.template(reader) == (itk.ImageFileReader, (ImageType,))
try:
itk.template(str)
raise Exception("unknown class should send an exception")
except KeyError:
pass
imageType = itk.Image[pixelType, dim]
reader = itk.ImageFileReader[imageType].New(FileName = imageFileName)
reader.Update()
if beVerbose:
print 'image "%s" read' % (imageFileName)
return reader.GetOutput()
def writeImage(outFileName, input, beVerbose = True):
"""
Write input to disk.
@param outFileName Name of the output file
@param input Reference to the object providing the input to store
@param beVerbose Flag determining wheterh a message is printed
after the input has been written (default: True)
"""
#
# set up and execute writer
writer = itk.ImageFileWriter[input].New(Input = input,
FileName = outFileName)
writer.Update()
if beVerbose:
print 'file "%s" written' % (outFileName)
if __name__ == '__main__':
#
# test / demo
img = readImage(os.path.join('CThead', 'CThead_l3.mhd'))
itk.echo(img)
writeImage('/tmp/out.mhd', img)
itk.force_load()
filename = sys.argv[1]
mesh_filename = sys.argv[2]
PixelType = itk.UC
dim = 2
ImageType = itk.Image[PixelType, dim]
ReaderType = itk.ImageFileReader[ImageType]
reader = ReaderType.New(FileName=filename)
# test snake_case keyword arguments
reader = ReaderType.New(file_name=filename)
# test echo
itk.echo(reader)
itk.echo(reader, sys.stdout)
# test class_
assert itk.class_(reader) == ReaderType
assert itk.class_("dummy") == str
# test template
assert itk.template(ReaderType) == (itk.ImageFileReader, (ImageType,))
assert itk.template(reader) == (itk.ImageFileReader, (ImageType,))
try:
itk.template(str)
raise Exception("unknown class should send an exception")
except KeyError:
pass
node.SetValue(seedValue)
node.SetIndex(seedPosition)
seeds.Initialize()
seeds.InsertElement(0, node)
fastMarching.SetTrialPoints(seeds)
fastMarching.SetSpeedConstant(1.0)
reader.Update()
fastMarching.SetOutputSize(
reader.GetOutput().GetBufferedRegion().GetSize())
writer.Update()
itk.echo(thresholdSegmentation)
InternalWriterType = itk.ImageFileWriter[InternalImageType]
mapWriter = InternalWriterType.New()
mapWriter.SetInput(fastMarching.GetOutput())
mapWriter.SetFileName("fastMarchingImage.mha")
mapWriter.Update()
speedWriter = InternalWriterType.New()
speedWriter.SetInput(thresholdSegmentation.GetSpeedImage())
speedWriter.SetFileName("speedTermImage.mha")
speedWriter.Update()
seedValue = -initialDistance
node.SetValue(seedValue)
node.SetIndex(seedPosition)
seeds.Initialize()
seeds.InsertElement(0, node)
fastMarching.SetTrialPoints(seeds)
fastMarching.SetSpeedConstant(1.0)
reader.Update()
fastMarching.SetOutputSize(reader.GetOutput().GetBufferedRegion().GetSize())
writer.Update()
itk.echo(thresholdSegmentation)
InternalWriterType = itk.ImageFileWriter[InternalImageType]
mapWriter = InternalWriterType.New()
mapWriter.SetInput(fastMarching.GetOutput())
mapWriter.SetFileName("fastMarchingImage.mha")
mapWriter.Update()
speedWriter = InternalWriterType.New()
speedWriter.SetInput(thresholdSegmentation.GetSpeedImage())
speedWriter.SetFileName("speedTermImage.mha")
speedWriter.Update()
