def __init__(self, module_manager):
SimpleVTKClassModuleBase.__init__(
self, module_manager,
vtk.vtkImageCorrelation(), 'Processing.',
('vtkImageData', 'vtkImageData'), ('vtkImageData',),
replaceDoc=True,
inputFunctions=None, outputFunctions=None)
def __init__(self, module_manager):
SimpleVTKClassModuleBase.__init__(self,
module_manager,
vtk.vtkImageCorrelation(),
'Processing.',
('vtkImageData', 'vtkImageData'),
('vtkImageData', ),
replaceDoc=True,
inputFunctions=None,
outputFunctions=None)
def correlationWith(self, vol2, dim=2):
"""Find the correlation between two volumetric data sets.
Keyword `dim` determines whether the correlation will be 3D, 2D or 1D.
The default is a 2D Correlation.
The output size will match the size of the first input.
The second input is considered the correlation kernel.
"""
imc = vtk.vtkImageCorrelation()
imc.SetInput1Data(self._data)
imc.SetInput2Data(vol2._data)
imc.SetDimensionality(dim)
imc.Update()
return Volume(imc.GetOutput())
from vtk.util.misc import vtkGetDataRoot VTK_DATA_ROOT = vtkGetDataRoot() # Show the constant kernel. Smooth an impulse function. s1 = vtk.vtkImageCanvasSource2D() s1.SetScalarTypeToFloat() s1.SetExtent(0, 255, 0, 255, 0, 0) s1.SetDrawColor(0) s1.FillBox(0, 255, 0, 255) s1.SetDrawColor(2.0) s1.FillTriangle(10, 100, 190, 150, 40, 250) s1.Update() s2 = vtk.vtkImageCanvasSource2D() s2.SetScalarTypeToFloat() s2.SetExtent(0, 31, 0, 31, 0, 0) s2.SetDrawColor(0.0) s2.FillBox(0, 31, 0, 31) s2.SetDrawColor(2.0) s2.FillTriangle(10, 1, 25, 10, 1, 5) s2.Update() convolve = vtk.vtkImageCorrelation() convolve.SetDimensionality(2) convolve.SetInput1Data(s1.GetOutput()) convolve.SetInput2Data(s2.GetOutput()) viewer = vtk.vtkImageViewer() viewer.SetInputConnection(convolve.GetOutputPort()) viewer.SetColorWindow(256) viewer.SetColorLevel(127.5) viewer.Render() # --- end of script --
from vtk.util.misc import vtkGetDataRoot VTK_DATA_ROOT = vtkGetDataRoot() # Show the constant kernel. Smooth an impulse function. s1 = vtk.vtkImageCanvasSource2D() s1.SetScalarTypeToFloat() s1.SetExtent(0,255,0,255,0,0) s1.SetDrawColor(0) s1.FillBox(0,255,0,255) s1.SetDrawColor(2.0) s1.FillTriangle(10,100,190,150,40,250) s1.Update() s2 = vtk.vtkImageCanvasSource2D() s2.SetScalarTypeToFloat() s2.SetExtent(0,31,0,31,0,0) s2.SetDrawColor(0.0) s2.FillBox(0,31,0,31) s2.SetDrawColor(2.0) s2.FillTriangle(10,1,25,10,1,5) s2.Update() convolve = vtk.vtkImageCorrelation() convolve.SetDimensionality(2) convolve.SetInput1Data(s1.GetOutput()) convolve.SetInput2Data(s2.GetOutput()) viewer = vtk.vtkImageViewer() viewer.SetInputConnection(convolve.GetOutputPort()) viewer.SetColorWindow(256) viewer.SetColorLevel(127.5) viewer.Render() # --- end of script --
