1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86
|
#!/usr/bin/env python
## Program: VMTK
## Module: $RCSfile: vmtkimagereslice.py,v $
## Language: Python
## Date: $Date: 2006/07/17 09:53:14 $
## Version: $Revision: 1.8 $
## Copyright (c) Luca Antiga, David Steinman. All rights reserved.
## See LICENCE file for details.
## This software is distributed WITHOUT ANY WARRANTY; without even
## the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
## PURPOSE. See the above copyright notices for more information.
import vtk
import sys
import pypes
vmtkimagecast = 'vmtkImageCast'
class vmtkImageCast(pypes.pypeScript):
def __init__(self):
pypes.pypeScript.__init__(self)
self.Image = None
self.OutputType = 'float'
self.ShiftScale = 1
self.WindowLevel = [0.0,0.0]
self.SetScriptName('vmtkimagecast')
self.SetScriptDoc('cast an image to a specified type')
self.SetInputMembers([
['Image','i','vtkImageData',1,'','the input image','vmtkimagereader'],
['OutputType','type','str',1,'["float","double","uchar","short"]','the output image type'],
['ShiftScale','shiftscale','int',1,'','shift scale values to fit windowlevel'],
['WindowLevel','windowlevel','float',2,'','window and level for value mapping']
])
self.SetOutputMembers([
['Image','o','vtkImageData',1,'','the output image','vmtkimagewriter']
])
def Execute(self):
if self.Image == None:
self.PrintError('Error: No input image.')
if self.OutputType == 'uchar' and self.ShiftScale:
shiftScale = vtk.vtkImageShiftScale()
shiftScale.SetInput(self.Image)
if self.WindowLevel[0] == 0.0:
scalarRange = self.Image.GetScalarRange()
shiftScale.SetShift(-scalarRange[0])
shiftScale.SetScale(255.0/(scalarRange[1]-scalarRange[0]))
else:
shiftScale.SetShift(-(self.WindowLevel[1]-self.WindowLevel[0]/2.0))
shiftScale.SetScale(255.0/self.WindowLevel[0])
shiftScale.SetOutputScalarTypeToUnsignedChar()
shiftScale.ClampOverflowOn()
shiftScale.Update()
self.Image = shiftScale.GetOutput()
else:
cast = vtk.vtkImageCast()
cast.SetInput(self.Image)
if self.OutputType == 'float':
cast.SetOutputScalarTypeToFloat()
elif self.OutputType == 'double':
cast.SetOutputScalarTypeToDouble()
elif self.OutputType == 'uchar':
cast.SetOutputScalarTypeToUnsignedChar()
elif self.OutputType == 'short':
cast.SetOutputScalarTypeToShort()
cast.Update()
self.Image = cast.GetOutput()
if __name__=='__main__':
main = pypes.pypeMain()
main.Arguments = sys.argv
main.Execute()
|