+jɺdZ ddlZn#e$r edwxYweejddZddlZddl Z ddl Z ddl m Z ddl mZddlmZdd l mZddlZd ZGd d ZGd deZd=dZd>dZdZdZGddeZeeGddejZGddeZeeGddeZ e Z!GddeZ"ee"GddeZ#Gd d!eZ$Gd"d#Z%Gd$d%eZ&Gd&d'e&Z'Gd(d)eZ(Gd*d+e(Z)Gd,d-e(Z*Gd.d/e(Z+Gd0d1e(Z,Gd2d3e,Z-Gd4d5e-Z.Gd6d7e-Z/Gd8d9e(Z0Gd:d;e(Z1d<Z2dS)?aVThis module provides classes that allow Numpy-type access to VTK datasets and arrays. This is best described with some examples. To normalize a VTK array: from vtkmodules.vtkImagingCore vtkRTAnalyticSource import vtkmodules.numpy_interface.dataset_adapter as dsa import vtkmodules.numpy_interface.algorithms as algs rt = vtkRTAnalyticSource() rt.Update() image = dsa.WrapDataObject(rt.GetOutput()) rtdata = image.PointData['RTData'] rtmin = algs.min(rtdata) rtmax = algs.max(rtdata) rtnorm = (rtdata - rtmin) / (rtmax - rtmin) image.PointData.append(rtnorm, 'RTData - normalized') print image.GetPointData().GetArray('RTData - normalized').GetRange() To calculate gradient: grad= algs.gradient(rtnorm) To access subsets: >>> grad[0:10] VTKArray([[ 0.10729134, 0.03763443, 0.03136338], [ 0.02754352, 0.03886006, 0.032589 ], [ 0.02248248, 0.04127144, 0.03500038], [ 0.02678365, 0.04357527, 0.03730421], [ 0.01765099, 0.04571581, 0.03944477], [ 0.02344007, 0.04763837, 0.04136734], [ 0.01089381, 0.04929155, 0.04302051], [ 0.01769151, 0.05062952, 0.04435848], [ 0.002764 , 0.05161414, 0.04534309], [ 0.01010841, 0.05221677, 0.04594573]]) >>> grad[:, 0] VTKArray([ 0.10729134, 0.02754352, 0.02248248, ..., -0.02748174, -0.02410045, 0.05509736]) All of this functionality is also supported for composite datasets even though their data arrays may be spread across multiple datasets. We have implemented a VTKCompositeDataArray class that handles many Numpy style operators and is supported by all algorithms in the algorithms module. This module also provides an API to access composite datasets. For example: from vtkmodules.vtkCommonDataModel import vtkMultiBlockDataSet mb = vtkMultiBlockDataSet() mb.SetBlock(0, image.VTKObject) mb.SetBlock(1e, image.VTKObject) cds = dsa.WrapDataObject(mb) for block in cds: print block Note that this module implements only the wrappers for datasets and arrays. The classes implement many useful operators. However, to make best use of these classes, take a look at the algorithms module. NzWThis module depends on the numpy module. Please makesure that it is installed properly..) buffer_shared) numpy_support) vtkDataObject)vtkWeakReferencec||g}t|tjrt|tjrt|j}t|j}||ks(|dks"|dks|jd|jdkr|S||kr|}|}d}n|}|}d}||kr)tj|||||<|dz}||k)|S)aeReturns a list with the two arguments, any of them may be processed. If the arguments are numpy.ndarrays, append 1s to the shape of the array with the smallest number of dimensions until the arrays have the same number of dimensions. Does nothing if the arguments are not ndarrays or the arrays have the same number of dimensions. r) isinstancenumpyndarraylenshape expand_dims)a1a2llen1len2d maxLengthis d/DATA/AppData/hermes/venv/lib/python3.11/site-packages/vtkmodules/numpy_interface/dataset_adapter.pyreshape_append_onesrRs RA2u}%%*R*G*G28}}28}} DLLDAII HQK28A; & &HTkkAIAAAIA9}}$QqT1--AaDAA9}} HcJeZdZdZejZejZejZejZdS)ArrayAssociationz+Easy access to vtkDataObject.AttributeTypesN) __name__ __module__ __qualname____doc__rPOINTCELLFIELDROWrrrros455  E  D  E  CCCrrceZdZdZdZdZdS)VTKObjectWrapperzSuperclass for classes that wrap VTK objects with Python objects. This class holds a reference to the wrapped VTK object. It also forwards unresolved methods to the underlying object by overloading __get__attr.c||_dSN) VTKObjectself vtkobjects r__init__zVTKObjectWrapper.__init__{s "rc,t|j|S)z2Forwards unknown attribute requests to VTK object.)getattrr+r-names r __getattr__zVTKObjectWrapper.__getattr__~st~t,,,rN)rrr r!r/r4r&rrr(r(vs<###-----rr(ctj|}|j}t|dkr?|ddkr3||dddfddd}t |||S)zAGiven a vtkDataArray and a dataset owning it, returns a VTKArray.rr r)arraydataset)r vtk_to_numpyrrreshape transposeVTKArray)r8r9narrayrs rvtkDataArrayToVTKArrayr?sy  ' . .F LE 5zzQ58q==q1a 011;;Aq!DD F% 9 9 99r numpy_arrayc\tj||}|||S)zGiven a numpy array or a VTKArray and a name, returns a vtkDataArray. The resulting vtkDataArray will store a reference to the numpy array: the numpy array is released only when the vtkDataArray is destroyed. array_type)r numpy_to_vtkSetName)r8r3rCvtkarrays rnumpyTovtkDataArrayrGs2)%JGGGH T Orc|dS|jjr|Stj|}|jj}|j}t|dkr5|d|z dkr&|d|z dkr|dddS|S)Nr7r rr) flags contiguousr asarraydtypeitemsizestridesrr<)r8sizerNs r_make_tensor_array_contiguousrPs }t { M% E ; DmG 7||qWQZ_11gajo6J6Jq!Q''' Lrcfd}|S)z0For compatibility between python 2 and python 3.ct|}|dd|dd|j|j|S)N__dict__ __weakref__)varscopypopr __bases__)clsbodymcss r decoratorz_metaclass..decoratorsXCyy~~ T""" %%%s3<555rr&)r[r\s` r _metaclassr]s$66666 rceZdZdZdS)VTKArrayMetaClasscfdfdfdfdfd}|d|d|d|d |d |d |d |d |dd|ddddddddt|||S)aWe overwrite numerical/comparison operators because we might need to reshape one of the arrays to perform the operation without broadcast errors. For instance: An array G of shape (n,3) resulted from computing the gradient on a scalar array S of shape (n,) cannot be added together without reshaping. G + expand_dims(S,1) works, G + S gives an error: ValueError: operands could not be broadcast together with shapes (n,3) (n,) This metaclass overwrites operators such that it computes this reshape operation automatically by appending 1s to the dimensions of the array with fewer dimensions. c,fd}|_|<dS)XCreate an attribute named attr_name that calls _numeric_op(self, other, op).c<t||Sr*)r= _numeric_opr-other attr_names rclosurezBVTKArrayMetaClass.__new__..add_numeric_op..closures++D%CCCrNrrgrhattrs` radd_numeric_opz1VTKArrayMetaClass.__new__..add_numeric_ops7 D D D D D(G %DOOOrc$d|zdSz;Adds '__[op_name]__' attribute that uses operator.[op_name]z__%s__Nr&op_namerls radd_default_numeric_opz9VTKArrayMetaClass.__new__..add_default_numeric_ops N8G+ , , , , ,rc,fd}|_|<dS)`Create an attribute named attr_name that calls _reverse_numeric_op(self, other, op).c<t||Sr*)r=_reverse_numeric_opres rrhzJVTKArrayMetaClass.__new__..add_reverse_numeric_op..closures33D%KKKrNrirjs` radd_reverse_numeric_opz9VTKArrayMetaClass.__new__..add_reverse_numeric_ops7 L L L L L(G %DOOOrc$d|zdSz.add_default_reverse_numeric_ops " "9W#4 5 5 5 5 5rc4||dSzDCall both add_default_numeric_op and add_default_reverse_numeric_op.Nr&rprqrzs radd_default_numeric_opsz:VTKArrayMetaClass.__new__..add_default_numeric_ops+ " "7 + + + * *7 3 3 3 3 3raddsubmultruedivfloordivmodpowlshiftrshiftandxororltleeqnegegt)type__new__ r[r3parentrkr~rqrzrlrvs ` @@@@rrzVTKArrayMetaClass.__new__s" & & & & & - - - - - & & & & & 6 6 6 6 6 4 4 4 4 4 4 &&&&&&&&& *** +++&&&&&&))))))u&&&tt$$$t$$$t$$$t$$$t$$$t$$$||Cvt444rNrrr rr&rrr_r_s(A5A5A5A5A5rr_cpeZdZdZdZdZd dZdZdZd d Z e d Z e j d Z dS)r=zThis is a sub-class of numpy ndarray that stores a reference to a vtk array as well as the owning dataset. The numpy array and vtk array should point to the same memory location.ct||}ttj||d|dSYUsed to implement numpy-style numerical operations such as __add__, __mul__, etc.rr rr1r r r-rfrgrs rrdzVTKArray._numeric_op9 e , ,0wu}i001qt<<sf  !w( !4,, 3 0E 0 2 2 M  g/ 0 0 0 0 0 DMMMr)NN)NF) rrr r!rdrurrr4rpropertyrsetterr&rrr=r=s === ===    555"    HHHH  X  ^ ! !^ ! ! !rr=ceZdZdZdS)VTKNoneArrayMetaClassc~fdfdfdfd}|d|d|d|d|d |d |d |d |d dtjdtj|ddtjdtjddddddt|||S)@Simplify the implementation of the numeric/logical sequence API.c,fd}||_||<dS)rbc<t||Sr*) VTKNoneArray_opr-rfops rrhz?VTKNoneArrayMetaClass.__new__.._add_op..closureTs#''eR888rNrirgrrhrks ` r_add_opz.VTKNoneArrayMetaClass.__new__.._add_opQs2 9 9 9 9 9(G %DOOOrcLd|ztt|dSrxr1operatorrprs r_add_default_reverse_opz>VTKNoneArrayMetaClass.__new__.._add_default_reverse_opYs* GIg%wx'A'A B B B B BrcLd|ztt|dSrnrrs r_add_default_opz6VTKNoneArrayMetaClass.__new__.._add_default_op]s* GHW$gh&@&@ A A A A Arc4||dSr|r&)rprrs r_add_default_opsz7VTKNoneArrayMetaClass.__new__.._add_default_opsas* OG $ $ $ # #G , , , , ,rrrrrrrrrr__and____rand__r__or____ror__rrrrrrrand_or_rr)r[r3rrkrrrrs ` @@@rrzVTKNoneArrayMetaClass.__new__Os & & & & & C C C C C B B B B B - - - - - - ###$$$"""""" 8=))) HM***(,''' 8<(((||Cvt444rNrr&rrrrNs#,5,5,5,5,5rrc$eZdZdZdZdZdZdS)raVTKNoneArray is used to represent a "void" array. An instance of this class (NoneArray) is returned instead of None when an array that doesn't exist in a DataSetAttributes is requested. All operations on the NoneArray return NoneArray. The main reason for this is to support operations in parallel where one of the processes may be working on an empty dataset. In such cases, the process is still expected to evaluate a whole expression because some of the functions may perform bulk MPI communication. None cannot be used in these instances because it cannot properly override operators such as __add__, __sub__ etc. This is the main raison d'etre for VTKNoneArray.ctSr* NoneArray)r-indexs r __getitem__zVTKNoneArray.__getitem__srctS)rrrs rrzVTKNoneArray._ops rctS)z'Implements numpy array's astype method.r)r-rLs rastypezVTKNoneArray.astypesrN)rrr r!rrrr&rrrr}sK    rrceZdZdZdS)VTKCompositeDataArrayMetaClasscfdfdfdfdfd}|d|d|d|d |d |d |d |d |ddtjdtj|ddtjdtjddddddt|||S)rc,fd}||_||<dS)rbc<t||Sr*)VTKCompositeDataArrayrdrs rrhzOVTKCompositeDataArrayMetaClass.__new__..add_numeric_op..closures,88ubIIIrNrirs ` rrlz>VTKCompositeDataArrayMetaClass.__new__..add_numeric_ops7 J J J J J(G %DOOOrc,fd}||_||<dS)rsc<t||Sr*)rrurs rrhzWVTKCompositeDataArrayMetaClass.__new__..add_reverse_numeric_op..closures,@@ubQQQrNrirs ` rrvzFVTKCompositeDataArrayMetaClass.__new__..add_reverse_numeric_ops7 R R R R R(G %DOOOrcLd|ztt|dSrxrrys rrzzNVTKCompositeDataArrayMetaClass.__new__..add_default_reverse_numeric_ops+ " "9W#4gh6P6P Q Q Q Q QrcLd|ztt|dSrnrros rrqzFVTKCompositeDataArrayMetaClass.__new__..add_default_numeric_ops* N8G+WXw-G-G H H H H Hrc4||dSr|r&r}s rr~zGVTKCompositeDataArrayMetaClass.__new__..add_default_numeric_opsrrrrrrrrrrrrrrrrrrrrrrrrs ` @@@@rrz&VTKCompositeDataArrayMetaClass.__new__s & & & & & & & & & & R R R R R I I I I I 4 4 4 4 4 4 &&&&&&&&& *** +++&&&&&&))))))y(-000z8=999&&&x...y(,777t$$$t$$$t$$$t$$$t$$$t$$$||Cvt444rNrr&rrrrs#4545454545rrc~eZdZdZgdddfdZdZdZeeZdZ ee Z dZ dZ d Z d Zd ZdS) radThis class manages a set of arrays of the same name contained within a composite dataset. Its main purpose is to provide a Numpy-type interface to composite data arrays which are naturally nothing but a collection of vtkDataArrays. A VTKCompositeDataArray makes such a collection appear as a single Numpy array and support all array operations that this module and the associated algorithm module support. Note that this is not a subclass of a Numpy array and as such cannot be passed to native Numpy functions. Instead VTK modules should be used to process composite arrays. Nc||_||_||_d}|dkr>|jD]6}t|dr$|dkr|j} |jr||jkrd}n7|r||_nt j|_d|_dS)aConstruct a composite array given a container of arrays, a dataset, name and association. It is sufficient to define a container of arrays to define a composite array. It is also possible to initialize an array by defining the dataset, name and array association. In that case, the underlying arrays will be created lazily when they are needed. It is recommended to use the latter method when initializing from an existing composite dataset.TNrF)_ArraysrNamerrrr$ Initialized)r-arraysr9r3 associationvalidAssociationr8s rr/zVTKCompositeDataArray.__init__s    $    5-00"d**&+&7 *{e>O/O/O+0(  6*D  /5D  rc|jrdSd|_|j|jdSg|_|jD]?}|j||j|j@dS)NT)rrrrappend GetAttributesr)r-dss r__init_from_compositez+VTKCompositeDataArray.__init_from_composites    F < 49#4 F , O OB L   0 01A B B49 M N N N N O Orc|tjd}|jD]} ||jz }#t $rYwxYw|S)z,Returns the number of elements in the array.r)+_VTKCompositeDataArray__init_from_compositer int64rrOr)r-rOas rGetSizezVTKCompositeDataArray.GetSizesi ""$$${1~~  A !     s > A  A c8||jS)z~Returns the internal container of VTKArrays. If necessary, this will populate the array list from a composite dataset.)rrrs r GetArrayszVTKCompositeDataArray.GetArrayss ""$$$|rc|g}t|tkrmt|j|jD]Q\}}|t ur)|||7|t RnV|jD]N}|t ur)|||4|t Ot||j S)aCOverwritten to refer indexing to underlying VTKArrays. For the most part, this will behave like Numpy. Note that indexing is done per array - arrays are never treated as forming a bigger array. If the index is another composite array, a one-to-one mapping between arrays is assumed. r9) rrrziprArraysrrrr)r-rresridxs rrz!VTKCompositeDataArray.__getitem__s ""$$$ ;;/ / /dlEL99 * *3I%%JJq}}S112222JJy))))  * \ * *I%%JJq}}U334444JJy))))$S$,????rcz|g}t|tkrt|j|jD]m\}}|t urE|t ur+>+B33AJJrr!A$qt}}----JJy))))  *\ * *I%%+Au55AJJrr!A$!~~....JJy))))$ 43CEEE Ercz|g}t|tkrt|j|jD]m\}}|t urE|tur>JJy))))JJqxx////$ 4+K+K+M+M$M$M788 8>**3// ~66s;;H  &x>> , rcg}|j}t|D]E}|j|}|r||F|Sz*Returns the names of the arrays as a list.r+rranger GetNamer)r-kysnarraysrr3s rkeyszDataSetAttributes.keyssp.2244w ! !A>22155==??D ! 4    rcg}|j}t|D]E}|j|}|r||F|S)zReturns the arrays as a list.r#)r-valsr'rrs rvalueszDataSetAttributes.valuessi.2244w  A//22Ayy{{  A rc |j|dS#t$r#|j|jYdSwxYw)z"A wrapper for vtkDataSet.PassData.N)r+PassDatar)r-rfs rr-zDataSetAttributes.PassDatas_ 5 N # #E * * * * * 5 5 5 N # #EO 4 4 4 4 4 4 5s)A  A c|turdS|jtjkr|j}n|jtjkr|j}nv|jtjkr7|j dkr|j }n*t|tj sd}n |jd}t|tj rtj|dkr_t|tj r|jnt#|}tj||}|||}n |jd|krd}|jD]}||z} tj||f|j}n#tjjj$r}t.jdt.jt5t#|dzt.jt5|Yd}~dSd}~wwxYw||dd<|}|j} t9| dkr|jj} |jd| z dkr|jd| z dks4|jd| z dkr7|jd| z dkr#|jj s|!ddd}|jj stj"|}t9| dkr+|#| d| d| dz}tI|} |j%| _%n#tL$rYnwxYwtO| |} |j%(| dS) aAppends narray to the dataset attributes. If narray is a scalar, create an array with this scalar for each element. If narray is an array with a size not matching the array association (e.g. size should be equal to GetNumberOfPoints() for PointData), copy the input narray for each element. This is intended to ease initialization, typically using same 3d vector for each element. In any case, be careful about memory explosion.Nrr )rLa]Fail to copy input array for each dataset element: array is too big to be duplicated. Input should either be small enough to be duplicated for each element, or shape[0] should match number of element. Example of correct usage: to add a point PointData array, it is common to have array.shape[0] == 3 or array.shape[0] == dataset.GetNumberOfPoints()  r7r))rrrr"rGetNumberOfPointsr#GetNumberOfCellsr%GetNumberOfColumnsGetNumberOfRowsr r r rndimrLremptyfillcore _exceptions_ArrayMemoryErrorsysstderrwritestrflattenrrMrNrIrJr<ascontiguousarrayr;r=r+rrGAddArray) r-r>r3 arrLengthrLtmparray componentsrnpErrrrOrVarrs rrzDataSetAttributes.appends Y   F  /5 5 5 6688II  !1!6 6 6 5577II  !1!5 5 5l--//!33 4466IIfem44 , "LO &%-00 EJv4F4F!4K4K$.vu}$E$EWFLL4PV<flSSS:);      "Y    T%[[!1!1D!8999   U,,, !..**HQQQKF  u::?? <(Dq!$&!++q0A$0F!0K0K"4'1,,1B41G11L1L\,2M **1a33|& 5,V44F u::??^^E!HeAhuQx.?@@F  $#-DNN###tt#!$-- $$$$$s+F22I BII N NN) rrr r!r/rrr(r+r-r __classcell__)rs@rrrns&& '''''"""   555S%S%S%S%S%S%S%rrc<eZdZdZdZdZdZdZdZdZ dZ d S) CompositeDataSetAttributesanThis is a python friendly wrapper for vtkDataSetAttributes for composite datasets. Since composite datasets themselves don't have attribute data, but the attribute data is associated with the leaf nodes in the composite dataset, this class simulates a DataSetAttributes interface by taking a union of DataSetAttributes associated with all leaf nodes.cf||_||_g|_i|_|dSr*)rr ArrayNamesr1_CompositeDataSetAttributes__determine_arraynames)r-r9rs rr/z#CompositeDataSetAttributes.__init__s; & ##%%%%%rct}g}|jD]a}||j}|D]0}||vr*||||1b||_dSr*)setrrrr(rrrJ)r- array_set array_listr9dsa array_names r__determine_arraynamesz1CompositeDataSetAttributes.__determine_arraynamessEE  | 2 2G''(899C!hhjj 2 2 Y..MM*---%%j111 2%rc|jSr")rJrs rr(zCompositeDataSetAttributes.keyss rc,||S)z2Implements the [] operator. Accepts an array name.rrs rrz&CompositeDataSetAttributes.__getitem__"rrc|turdSd}t|tsZ|jD]2}||j||d}3|r|j|dSdSt|j|j D]7\}}|0||j||d}8|r8|j|tj ||j |<dSdS)z8Appends a new array to the composite dataset attributes.NFT) rr rrrrrrJrrweakrefref)r-r>r3addedrr8s rrz!CompositeDataSetAttributes.append&s1 Y   F&"788 8l    !12299&$GGG -&&t,,,,, - - !v}== ! ! E$$$T%566==eTJJJ E 8&&t,,,$+K$7$7 D!!! 8 8rc|}||jvrtS||jvs|j|9t|j||j}t j||j|<n|j|}|S)z*Given a name, returns a VTKCompositeArray.N)r9r3r)rJrrrrrrVrW)r-r  arraynamer8s rrz#CompositeDataSetAttributes.GetArray>s DO + +  DK ' '+A4;y+A+C+C+K),yHXZZZE%,[%7%7DK " "*DK *,,E rc\t|j|jD]\}}tjtjtjfD]g}||rP||r;||||hdS)z(Emulate PassData for composite datasets.N) rrrr"r#r% HasAttributesrr-)r-rfthisthatassocs rr-z#CompositeDataSetAttributes.PassDataKsT\5=99 R RID*02B2GIYI]^ R R%%e,,R1C1CE1J1JR&&u--66t7I7I%7P7PQQQ R R RrN) rrr r!r/rKr(rrrr-r&rrrHrHsBB & & & % % %"""8880   RRRRRrrHc0eZdZdZdZdZdZdZdZdS)CompositeDataIteratorzWrapper for a vtkCompositeDataIterator class to satisfy the python iterator protocol. This iterator iterates over non-empty leaf nodes. To iterate over empty or non-leaf nodes, use the vtkCompositeDataIterator directly. c||_|jr5|jd|jdSdSr*) NewIteratorIterator UnRegister GoToFirstItemr-cdss rr/zCompositeDataIterator.__init__YsW)) = * M $ $T * * * M ' ' ) ) ) ) ) * *rc|Sr*r&rs r__iter__zCompositeDataIterator.__iter___s rc|jst|jrt|j}|jt |Sr*)rd StopIterationIsDoneWithTraversalGetCurrentDataObject GoToNextItemr)r-retVals r__next__zCompositeDataIterator.__next__bsc}   = , , . .  3355 ""$$$f%%%rc*|Sr*rqrs rnextzCompositeDataIterator.nextl}}rc,t|j|S)z5Returns attributes from the vtkCompositeDataIterator.)r1rdr2s rr4z!CompositeDataIterator.__getattr__ost}d+++rN) rrr r!r/rjrqrtr4r&rrraraRsi  *** &&&,,,,,rrac$eZdZdZdZdZdZdS)MultiCompositeDataIteratoradIterator that can be used to iterate over multiple composite datasets together. This iterator works only with arrays that were copied from an original using CopyStructured. The most common use case is to use CopyStructure, then iterate over input and output together while creating output datasets from corresponding input datasets.cVt||d||_dS)Nr)rar/Datasetsrgs rr/z#MultiCompositeDataIterator.__init__{s'&&tSV444 rc|jst|jrtg}|t |jt |jdkrL|jddD]<}|t ||j=|j |S)Nr ) rdrlrmrrrnrrz GetDataSetro)r-rpcds rrqz#MultiCompositeDataIterator.__next__s}   = , , . .   nT]%G%G%I%IJJKKK t}   ! !mABB' L L nR]]4=-I-IJJKKKK ""$$$ rc*|Sr*rsrs rrtzMultiCompositeDataIterator.nextrurN)rrr r!r/rqrtr&rrrxrxssK   rrxc@eZdZdZdZdZdZeedddZdS) DataObjectz`A wrapper for vtkDataObject that makes it easier to access FielData arrays as VTKArrays c|tjkr(t|j||St|j|||S)zVReturns the attributes specified by the type as a DataSetAttributes instance.)rr$rr+ GetFieldDatarr-rs rrzDataObject.GetAttributessU #) ) )$T^%@%@%B%BD$OO O !=!=d!C!CT4PPPrc"|tjkSz6Returns if current object support this attributes type)rr$rs rr\zDataObject.HasAttributess'---rcft|j|tjSz7Returns the field data as a DataSetAttributes instance.)rr+rrr$rs rrzDataObject.GetFieldDatas& !!>FVF\]]]rNz6This property returns the field data of a data object.) rrr r!rr\rr FieldDatar&rrrrscQQQ...^^^tT3kllIIIrrc:eZdZdZdZdZeedddZdS)TablezYA wrapper for vtkTable that makes it easier to access RowData array as VTKArrays c@|tjS)z5Returns the row data as a DataSetAttributes instance.)rrr%rs r GetRowDatazTable.GetRowDatas!!"2"6777rcX|tjkpt||Sr)rr%rr\rs rr\zTable.HasAttributess&'++Sz/G/Gd/S/SSrNz0This property returns the row data of the table.)rrr r!rr\rRowDatar&rrrrsQ888TTThz4/abbGGGrrc:eZdZdZdZdZeedddZdS) HyperTreeGridzdA wrapper for vtkHyperTreeGrid that makes it easier to access CellData arrays as VTKArrays. c@|tjS)z4Returns the cell data as DataSetAttributes instance.rrr#rs r GetCellDatazHyperTreeGrid.GetCellData!!"2"7888rcX|tjkpt||Sr)rr#rr\rs rr\zHyperTreeGrid.HasAttributess&',,T 0H0Ht0T0TTrNz:This property returns the cell data of the hypertree grid.)rrr r!rr\rCellDatar&rrrrsQ999UUUx T41mnnHHHrrceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zee d d dZee d d dZee d d dZeed d dZd S)CompositeDataSetzA wrapper for vtkCompositeData and subclasses that makes it easier to access Point/Cell/Field data as VTKCompositeDataArrays. It also provides a Python type iterator.ctt||d|_d|_d|_d|_dSr*)rr/ _PointData _CellData _FieldData_Pointsr,s rr/zCompositeDataSet.__init__s8D),,, rc t|S)z/Creates an iterator for the contained datasets.)rars rrjzCompositeDataSet.__iter__s$T***rc^d}|D]}|||z }t|S)zReturns the total number of cells or points depending on the value of assoc which can be ArrayAssociation.POINT or ArrayAssociation.CELL.r)GetNumberOfElementsr)r-r_resultr9s rrz$CompositeDataSet.GetNumberOfElementss> 9 9G g11%88 8FF6{{rc@|tjS)zReturns the total number of points of all datasets in the composite dataset. Note that this traverses the whole composite dataset every time and should not be called repeatedly for large composite datasets.)rrr"rs rr0z"CompositeDataSet.GetNumberOfPointss ''(8(>???rc@|tjS)zReturns the total number of cells of all datasets in the composite dataset. Note that this traverses the whole composite dataset every time and should not be called repeatedly for large composite datasets.)rrr#rs rr1z!CompositeDataSet.GetNumberOfCellss ''(8(=>>>rc"t||S)z^Returns the attributes specified by the type as a CompositeDataSetAttributes instance.)rHrs rrzCompositeDataSet.GetAttributess*$555rc@|D]}||sdSdS)zUReturns true if every leaves of current composite object support this attributes typeFT)r\)r-rr9s rr\zCompositeDataSet.HasAttributess:  G((.. uu trc|j|8|tj}t j||_|Sz7Returns the point data as a DataSetAttributes instance.)rrrr"rVrW)r-pdatas r GetPointDatazCompositeDataSet.GetPointDataQ ? "doo&7&7&?&&'7'=>>E%k%00DO   rc|j|8|tj}t j||_|Sz6Returns the cell data as a DataSetAttributes instance.)rrrr#rVrW)r-cdatas rrzCompositeDataSet.GetCellDatasQ > !T^^%5%5%=&&'7'<==E$[//DN~~rc|j|8|tj}t j||_|Sr)rrrr$rVrW)r-fdatas rrzCompositeDataSet.GetFieldDatarrc|j|g}|D]O} |j}n#t$rd}YnwxYw||t:||Pt |dkst d|Drt}nt||}tj ||_|S)z7Returns the points as a VTKCompositeDataArray instance.Nrc g|] }|tu Sr&r).0rs r z.CompositeDataSet.GetPoints..s$A$A$AQ)^$A$A$Arr) rPointsrrrrallrrVrW)r-ptsr_ptscptss r GetPointszCompositeDataSet.GetPoints s < 4<<>>#9C % % 9DD%   DDD <JJy))))JJt$$$$3xx1}}$A$AS$A$A$A B B} ,S$???";t,,DL||~~s + ::N4This property returns the point data of the dataset.1This property returns the cell data of a dataset.z2This property returns the field data of a dataset.z0This property returns the points of the dataset.)rrr r!r/rjrr0r1rr\rrrrr PointDatarrrr&rrrrs((+++@@@???666 !!!   !!!*tT3ijjIx T41deeHtT3ghhI Xit-_ ` `FFFrrc\eZdZdZdZdZdZeedddZeedddZ dS)rz[This is a python friendly wrapper of a vtkDataSet that defines a few useful properties.c@|tjSr)rrr"rs rrzDataSet.GetPointData)s!!"2"8999rc@|tjSrrrs rrzDataSet.GetCellData-rrcx|tjkp*|tjkpt||Sr)rr"r#rr\rs rr\zDataSet.HasAttributes1s7'--v9I9N1NvR\RjRjkoquRvRvvrNrr) rrr r!rrr\rrrr&rrrr%sv  :::999wwwtT3ijjIx T41deeHHHrrc:eZdZdZdZdZeeeddZdS)PointSetz\This is a python friendly wrapper of a vtkPointSet that defines a few useful properties.c|jsdSt|j|}tj|_|S)zcReturns the points as a VTKArray instance. Returns None if the dataset has implicit points.N)r+rr?GetDatarr"r)r-r8s rrzPointSet.GetPoints;s]~'')) 4& N $ $ & & . . 0 0$88,2 rcddlm}t||r|}n.t|}|}|||j|dS)z:Given a VTKArray instance, sets the points of the dataset.r) vtkPointsN) vtkCommonCorerr rGSetDatar+ SetPoints)r-rrps rrzPointSet.SetPointsEsp------ c9 % % AA%c**C A IIcNNN   #####rNz7This property returns the point coordinates of dataset.)rrr r!rrrrr&rrrr8sN   $ $ $XiD2k l lFFFrrc4eZdZdZdZeedddZdS)PolyDataz\This is a python friendly wrapper of a vtkPolyData that defines a few useful properties.c|jsdSt|j|S)z)Returns the polys as a VTKArray instance.N)r+GetPolysr?rrs r GetPolygonszPolyData.GetPolygonsVN~&&(( 4% N # # % % - - / /77 7rNz3This property returns the connectivity of polygons.)rrr r!rrPolygonsr&rrrrRs?  777x T41fggHHHrrc~eZdZdZdZdZdZdZeedddZ eedddZ eeddd Z dS) UnstructuredGridzdThis is a python friendly wrapper of a vtkUnstructuredGrid that defines a few useful properties.c|jsdSt|j|S)z.Returns the cell types as a VTKArray instance.N)r+GetCellTypesArrayr?rs r GetCellTypeszUnstructuredGrid.GetCellTypescsB~//11 4% N , , . .66 6rc|jsdSt|j|S)z2Returns the cell locations as a VTKArray instance.N)r+GetCellLocationsArrayr?rs rGetCellLocationsz!UnstructuredGrid.GetCellLocationsjsB~3355 4% N 0 0 2 2D:: :rc|jsdSt|j|S)z)Returns the cells as a VTKArray instance.N)r+GetCellsr?rrs rrzUnstructuredGrid.GetCellsqrrcddlm}ddlm}t |}t ||}t ||}|}||||j|||dS)zlGiven cellTypes, cellLocations, cells as VTKArrays, populates the unstructured grid data structures.r) VTK_ID_TYPE) vtkCellArrayrBN)util.vtkConstantsrvtkCommonDataModelrrGSetCellsGetNumberOfTuplesr+)r- cellTypes cellLocationscellsrrcas rrzUnstructuredGrid.SetCellsxs 433333555555' 22 +MkRRR #EkBBB \^^ I//115999  ="=====rNz)This property returns the types of cells.z-This property returns the locations of cells.z0This property returns the connectivity of cells.) rrr r!rrrrr CellTypes CellLocationsCellsr&rrrr_s  666:::777 > > >tT3^__IH-tT;jkkM HXtT+] ^ ^EEErrcVeZdZdZdZdZeedddZeedddZdS)GraphzYThis is a python friendly wrapper of a vtkGraph that defines a few useful properties.c@|tjS)z8Returns the vertex data as a DataSetAttributes instance.)rrVERTEXrs r GetVertexDatazGraph.GetVertexDatas!!"2"9:::rc@|tjS)z6Returns the edge data as a DataSetAttributes instance.)rrEDGErs r GetEdgeDatazGraph.GetEdgeDatarrNz3This property returns the vertex data of the graph.z1This property returns the edge data of the graph.) rrr r!rrr VertexDataEdgeDatar&rrrrsd  ;;;999-t5jkkJx T41deeHHHrrcVeZdZdZdZdZeedddZeedddZdS)Moleculez\This is a python friendly wrapper of a vtkMolecule that defines a few useful properties.c*|S)z6Returns the atom data as a DataSetAttributes instance.)rrs r GetAtomDatazMolecule.GetAtomDatas!!###rc*|S)z6Returns the bond data as a DataSetAttributes instance.)rrs r GetBondDatazMolecule.GetBondDatas!!!rNz4This property returns the atom data of the molecule.z4This property returns the bond data of the molecule.) rrr r!rrrAtomDataBondDatar&rrrrsd  $$$"""x T41ghhHx T41ghhHHHrrc|drt|S|drt|S|drt|S|drt |S|drt |S|drt |S|drt|S|drt |S|d rt|S|d rt|Sd S) z4Returns a Numpy friendly wrapper of a vtkDataObject. vtkPolyDatavtkUnstructuredGrid vtkPointSet vtkDataSetvtkCompositeDataSetvtkTable vtkMoleculevtkGraphvtkHyperTreeGridrN) IsArrrrrrrrr)rs rrrsU vvm|| % & &###   ||    r{{ % & & ###    Ryy   ||   Ryy " # #R    "~~rr*)r@N)3r!r ImportError RuntimeErrorr __version__splitr itertoolsrr:rrutilrrrrrVrrobjectr(r?rGrPr]rr_r r=rrrrrrrHrarxrrrrrrrrrrrr&rrrs>>~&LLLL&&& ,% & &&& c%+11#66q9:: )))))) ......,,,,,,   : - - - - -v - - - : : : :   B5B5B5B5B5B5B5B5H  \!\!\!\!\!u}\!\!\!|-5-5-5-5-5D-5-5-5^  !""6#"2 LNN 5555555555T555555n  *++ZIZIZIZIZIFZIZI,+ZIzN%N%N%N%N%(N%N%N%bQRQRQRQRQRQRQRQRf,,,,,F,,,B!6:mmmmm!mmm, c c c c cJ c c c o o o o oJ o o oaaaaaaaaaazaaaaaaFfffffjfff&mmmmmwmmm4 h h h h hx h h h'_'_'_'_'_x'_'_'_R f f f f fJ f f f i i i i iz i i is