+j$ddlmZddlmZddlmZddlmZddlmZddlm Z Gdde Z Gd d e Z d S) )vtkInformation) vtkDataObject) vtkAlgorithm)vtkDemandDrivenPipeline) vtkStreamingDemandDrivenPipeline)vtkPythonAlgorithmcZeZdZdZ ddZdZdZdZd Zd Z d Z d Z d Z dZ dZdS) 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.  vtkDataSet vtkPolyDatac>||_||_||_||_dS)zSets up default NumberOfInputPorts, NumberOfOutputPorts, InputType and OutputType that are used by various initialization methods.N)NumberOfInputPortsNumberOfOutputPorts InputType OutputTypeself nInputPorts inputType nOutputPorts outputTypes V/DATA/AppData/hermes/venv/lib/python3.11/site-packages/vtkmodules/util/vtkAlgorithm.py__init__zVTKAlgorithm.__init__s& #.#/ "$cn||j||jdS)zeSets up number of input and output ports based on NumberOfInputPorts and NumberOfOutputPorts.N)SetNumberOfInputPortsrSetNumberOfOutputPortsrrvtkselfs r InitializezVTKAlgorithm.Initialize#s8 %%d&=>>>&&t'?@@@@@rc|||tjSzsConvenience method that returns an input data object given a vector of information objects and two indices.GetInformationObjectGetr DATA_OBJECTrinInfoijs r GetInputDatazVTKAlgorithm.GetInputData*3ay--a0044]5N5P5PQQQrct||tjSzgConvenience method that returns an output data object given an information object and an index.r$routInfor*s r GetOutputDatazVTKAlgorithm.GetOutputData0/++A..22=3L3N3NOOOrcdSzOverwritten 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.r rr requestr)r1s rRequestDataObjectzVTKAlgorithm.RequestDataObject5 qrcdSzLOverwritten by subclass to provide meta-data to downstream pipeline.r r6r7s rRequestInformationzVTKAlgorithm.RequestInformation; qrcdSzROverwritten by subclass to modify data request going to upstream pipeline.r r6r7s rRequestUpdateExtentz VTKAlgorithm.RequestUpdateExtent@r>rc tdz1Overwritten by subclass to execute the algorithm.zRequestData must be implementedNotImplementedErrorr7s r RequestDatazVTKAlgorithm.RequestDataE!"CDDDrc|tjr|||||S|tjr|||||S|t jr|||||S|tj r| ||||SdSz)Splits a request to RequestXXX() methods.r HasrREQUEST_DATA_OBJECTr9REQUEST_INFORMATIONr=rREQUEST_UPDATE_EXTENTrA REQUEST_DATArFr7s rProcessRequestzVTKAlgorithm.ProcessRequestIs ;;.BDD E E G))'7FGLL L [[0DFF G G G**7GVWMM M [[9OQQ R R G++GWfgNN N [[0=?? @ @ G##GWfgFF Fqrc^|tj|jdSz*Sets the required input type to InputType.r SetrINPUT_REQUIRED_DATA_TYPErrr portinfos rFillInputPortInformationz%VTKAlgorithm.FillInputPortInformationV& 688$.IIIqrc^|tj|jdSz+Sets the default output type to OutputType.r rTrDATA_TYPE_NAMErrVs rFillOutputPortInformationz&VTKAlgorithm.FillOutputPortInformation[& -//AAAqrNr r r r )__name__ __module__ __qualname____doc__rr!r,r2r9r=rArFrPrYr_r6rrr r s  1=2? % % % %AAARRR PPP    EEE    rr cpeZdZdZGddeZ ddZdZd Zd Z d Z d Z d Z dZ dZdZdS)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*eZdZdZdZdZdZdZdS)(VTKPythonAlgorithmBase.InternalAlgorithmzInternal class. Do not use.cdSNr6rs rr!z3VTKPythonAlgorithmBase.InternalAlgorithm.Initializes Drc.|||Srk)rYrVs rrYzAVTKPythonAlgorithmBase.InternalAlgorithm.FillInputPortInformations33D$?? ?rc.|||Srk)r_rVs rr_zBVTKPythonAlgorithmBase.InternalAlgorithm.FillOutputPortInformations44T4@@ @rc0||||Srk)rPr7s rrPz7VTKPythonAlgorithmBase.InternalAlgorithm.ProcessRequests))'67CC CrN)rbrcrdrer!rYr_rPr6rrInternalAlgorithmrisa%%    @ @ @ A A A D D D D Drror r r c|t||||||_||_dS)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__s_ 3EEGGHHH "";/// ##L111"$rc|||tjSr#r$r(s rr,z#VTKPythonAlgorithmBase.GetInputDatar-rct||tjSr/r$r0s rr2z$VTKPythonAlgorithmBase.GetOutputDatar3rc^|tj|jdSrRrSrrWrXs rrYz/VTKPythonAlgorithmBase.FillInputPortInformationrZrc^|tj|jdSr\r]rus rr_z0VTKPythonAlgorithmBase.FillOutputPortInformationr`rc|tjr||||S|tjr||||S|t jr||||S|tj r| |||SdSrIrJrr8r)r1s rrPz%VTKPythonAlgorithmBase.ProcessRequests ;;.BDD E E >))'67CC C [[0DFF G G >**7FGDD D [[9OQQ R R >++GVWEE E [[0=?? @ @ >##GVW== =qrcdSr5r6rxs rr9z(VTKPythonAlgorithmBase.RequestDataObjectr:rcdSr<r6rxs rr=z)VTKPythonAlgorithmBase.RequestInformationr>rcdSr@r6rxs rrAz*VTKPythonAlgorithmBase.RequestUpdateExtentr>rc tdrCrDrxs rrFz"VTKPythonAlgorithmBase.RequestDatarGrNra)rbrcrdreobjectrorr,r2rYr_rPr9r=rArFr6rrrgrg`s))V D D D D DF D D D1=2? % % % %RRR PPP        EEEEErrgN) vtkmodules.vtkCommonCorervtkmodules.vtkCommonDataModelr"vtkmodules.vtkCommonExecutionModelrrrvtkmodules.vtkFiltersPythonrr}r rgr6rrrs333333777777;;;;;;FFFFFFOOOOOO::::::VVVVV6VVVp|E|E|E|E|E/|E|E|E|E|Er