g$xddlmZddlmZddlmZddlmZddlmZddlm Z Gdde Z Gd d e Z y ) )vtkInformation) vtkDataObject) vtkAlgorithm)vtkDemandDrivenPipeline) vtkStreamingDemandDrivenPipeline)vtkPythonAlgorithmcXeZdZdZ ddZdZdZdZdZdZ dZ d Z d Z d Z d Zy ) VTKAlgorithmaThis is a superclass which can be derived to implement Python classes that work with vtkPythonAlgorithm. It implements Initialize(), ProcessRequest(), FillInputPortInformation() and FillOutputPortInformation(). Initialize() sets the input and output ports based on data members. ProcessRequest() calls RequestXXX() methods to implement various pipeline passes. FillInputPortInformation() and FillOutputPortInformation() set the input and output types based on data members. c<||_||_||_||_y)zSets up default NumberOfInputPorts, NumberOfOutputPorts, InputType and OutputType that are used by various initialization methods.N)NumberOfInputPortsNumberOfOutputPorts InputType OutputTypeself nInputPorts inputType nOutputPorts outputTypes &&t'?'?@rcl||j|jtjSzsConvenience method that returns an input data object given a vector of information objects and two indices.GetInformationObjectGetr DATA_OBJECTrinInfoijs r GetInputDatazVTKAlgorithm.GetInputData*-ay--a044]5N5N5PQQrcf|j|jtjSzgConvenience method that returns an output data object given an information object and an index.r!routInfor's r GetOutputDatazVTKAlgorithm.GetOutputData0)++A.22=3L3L3NOOrcyzOverwritten by subclass to manage data object creation. There is not need to overwrite this class if the output can be created based on the OutputType data member.rrrequestr&r.s rRequestDataObjectzVTKAlgorithm.RequestDataObject5rcyzLOverwritten by subclass to provide meta-data to downstream pipeline.r3r4r5s rRequestInformationzVTKAlgorithm.RequestInformation;rcyzROverwritten by subclass to modify data request going to upstream pipeline.r3r4r5s rRequestUpdateExtentz VTKAlgorithm.RequestUpdateExtent@r<rctdz1Overwritten by subclass to execute the algorithm.zRequestData must be implementedNotImplementedErrorr5s r RequestDatazVTKAlgorithm.RequestDataE!"CDDrc|jtjr|j||||S|jtjr|j ||||S|jt jr|j||||S|jtjr|j||||Syz)Splits a request to RequestXXX() methods.r3 HasrREQUEST_DATA_OBJECTr7REQUEST_INFORMATIONr;rREQUEST_UPDATE_EXTENTr? REQUEST_DATArDr5s rProcessRequestzVTKAlgorithm.ProcessRequestIs ;;.BBD E))'7FGL L [[0DDF G**7GVWM M [[9OOQ R++GWfgN N [[0==? @##GWfgF Frc`|jtj|jyz*Sets the required input type to InputType.r3SetrINPUT_REQUIRED_DATA_TYPErrrportinfos rFillInputPortInformationz%VTKAlgorithm.FillInputPortInformationV  668$..Irc`|jtj|jyz+Sets the default output type to OutputType.r3rRrDATA_TYPE_NAMErrTs rFillOutputPortInformationz&VTKAlgorithm.FillOutputPortInformation[  --/ArNr3 vtkDataSetr3 vtkPolyData)__name__ __module__ __qualname____doc__rrr)r/r7r;r?rDrNrWr]r4rrr r sJ 1=2? %AR P    E  rr cheZdZdZGddeZ ddZdZdZdZ dZ d Z d Z d Z d Zd Zy)VTKPythonAlgorithmBaseaiThis is a superclass which can be derived to implement Python classes that act as VTK algorithms in a VTK pipeline. It implements ProcessRequest(), FillInputPortInformation() and FillOutputPortInformation(). ProcessRequest() calls RequestXXX() methods to implement various pipeline passes. FillInputPortInformation() and FillOutputPortInformation() set the input and output types based on data members. Common use is something like this: class HDF5Source(VTKPythonAlgorithmBase): def __init__(self): VTKPythonAlgorithmBase.__init__(self, nInputPorts=0, nOutputPorts=1, outputType='vtkImageData') def RequestInformation(self, request, inInfo, outInfo): f = h5py.File("foo.h5", 'r') dims = f['RTData'].shape[::-1] info = outInfo.GetInformationObject(0) info.Set(vtkmodules.vtkCommonExecutionModel.vtkStreamingDemandDrivenPipeline.WHOLE_EXTENT(), (0, dims[0]-1, 0, dims[1]-1, 0, dims[2]-1), 6) return 1 def RequestData(self, request, inInfo, outInfo): f = h5py.File("foo.h5", 'r') data = f['RTData'][:] output = dsa.WrapDataObject(vtkmodules.vtkCommonDataModel.vtkImageData.GetData(outInfo)) output.SetDimensions(data.shape) output.PointData.append(data.flatten(), 'RTData') output.PointData.SetActiveScalars('RTData') return 1 alg = HDF5Source() cf = vtkmodules.vtkFiltersCore.vtkContourFilter() cf.SetInputConnection(alg.GetOutputPort()) cf.Update() c(eZdZdZdZdZdZdZy)(VTKPythonAlgorithmBase.InternalAlgorithmzInternal class. Do not use.cyNr4rs rrz3VTKPythonAlgorithmBase.InternalAlgorithm.Initializes rc&|j||Srk)rWrTs rrWzAVTKPythonAlgorithmBase.InternalAlgorithm.FillInputPortInformations33D$? ?rc&|j||Srk)r]rTs rr]zBVTKPythonAlgorithmBase.InternalAlgorithm.FillOutputPortInformations44T4@ @rc(|j|||Srk)rNr5s rrNz7VTKPythonAlgorithmBase.InternalAlgorithm.ProcessRequests))'67C CrN)rbrcrdrerrWr]rNr4rrInternalAlgorithmris%  @ A Drroc|jtj|j||j |||_||_y)zSets up default NumberOfInputPorts, NumberOfOutputPorts, InputType and OutputType that are used by various methods. Make sure to call this method from any subclass' __init__N)SetPythonObjectrgrorrrrrs rrzVTKPythonAlgorithmBase.__init__sF 3EEGH "";/ ##L1"$rcl||j|jtjSr r!r%s rr)z#VTKPythonAlgorithmBase.GetInputDatar*rcf|j|jtjSr,r!r-s rr/z$VTKPythonAlgorithmBase.GetOutputDatar0rc`|jtj|jyrPrQrrUrVs rrWz/VTKPythonAlgorithmBase.FillInputPortInformationrXrc`|jtj|jyrZr[rus rr]z0VTKPythonAlgorithmBase.FillOutputPortInformationr^rc|jtjr|j|||S|jtjr|j |||S|jt jr|j|||S|jtjr|j|||SyrGrHrr6r&r.s rrNz%VTKPythonAlgorithmBase.ProcessRequests ;;.BBD E))'67C C [[0DDF G**7FGD D [[9OOQ R++GVWE E [[0==? @##GVW= =rcyr2r4rxs rr7z(VTKPythonAlgorithmBase.RequestDataObjectr8rcyr:r4rxs rr;z)VTKPythonAlgorithmBase.RequestInformationr<rcyr>r4rxs rr?z*VTKPythonAlgorithmBase.RequestUpdateExtentr<rctdrArBrxs rrDz"VTKPythonAlgorithmBase.RequestDatarErNr_)rbrcrdreobjectrorr)r/rWr]rNr7r;r?rDr4rrrgrg`sS)V DF D1=2? %R P      ErrgN) vtkmodules.vtkCommonCorervtkmodules.vtkCommonDataModelr"vtkmodules.vtkCommonExecutionModelrrrvtkmodules.vtkFiltersPythonrr}r rgr4rrrs737;FO:V6Vp|E/|Er