tjPddlmZddlmZddlZddlmZddlm Z ddl m Z m Z ddl mZdd lmZmZmZd Zd Zd Zd Zd6dZdZdZdZdZdZd6dZd6dZdZdZ dZ!dZ"dZ#dZ$d6dZ%dZ&dZ'dZ(d6d Z)d!Z*d"Z+d#Z,d6d$Z-d%Z.d&Z/d6d'Z0d6d(Z1d)Z2d*Z3d+Z4d,Z5d6d-Z6d6d.Z7d/Z8d0Z9d6d1Z:d2Z;d6d3Zy)7)absolute_import)dataset_adapterN) numpy_support) vtkImageData)vtkCellDataToPointDatavtkPolyDataNormals)vtkCellDerivatives)vtkCellSizeFiltervtkCellQualityvtkMatrixMathFilterc4 |s tdt|jdk(r |j|jddf}|jd}|dk(r&|dk7r!tdt |jz|dk(r&|dk7r!tdt |jz|j j s|j}tj|}|dk(r|jdn|jd|j}|jd|j|jtj j"|j$k(r td tj j&|j$k(rr|jd|j)k7r td |dk(r!|j+j-|n|j+j/|ntj j0|j$k(rn|jd|j3k7r td |j}|jd|j|j|dk(r |j5j-|n|j5j/|t7}|j9||j;|dk(rJ|j+j-|j=j+j?nI|j+j/|j=j+jA|j9|tj j&|j$k(rWt7}|jC|jE|j;|j=j+Stj j0|j$k(r.|j;|j=j5Std ) Nz,Need a dataset to compute _cell_derivatives.rrscalarsz+This function expects scalars. Input shape vectorsz+This function expects vectors. Input shape zIUnknown data association. Data should be associated with points or cells.zEThe number of points does not match the number of tuples in the arrayz>The number of does not match the number of tuples in the array)# RuntimeErrorlenshapereshapestrflags contiguouscopyr numpy_to_vtkSetName NewInstance UnRegister CopyStructure VTKObjectdsaArrayAssociationFIELD AssociationPOINTGetNumberOfPoints GetPointData SetScalars SetVectorsCELLGetNumberOfCells GetCellDatar SetInputDataUpdate GetOutput GetScalars GetVectorsSetInputConnection GetOutputPort) narraydatasetattribute_typefilterncompvarraydsds2c2ps [/DATA/.local/lib/python3.12/site-packages/vtkmodules/numpy_interface/internal_algorithms.py_cell_derivativesr= s  H II 6<<A Q34v LLOE"uz *,/ ,=>??"uz *,/ ,=>?? << " "fkkmV  ' ' /F"FNN9$= >>) $    BMM$W&&' !!V%7%77 e ff !!V%7%77 ,,q/W668 8de e I %r'8'C'CF'Koo**62   " "f&8&8 8 ,,q/W557 7]^ ^  "s ~~d **+ I %(9(D(DV(Loo**62 # %s  zz| I % //  & &s}}'C'C'E'P'P'R S //  & &s}}'C'C'E'P'P'R S  !!V%7%77$ %s f2245 zz|mmo**,,   " "f&8&8 8 }} ,,..e ffc|s td|j}|jd|j|jt }|j |d|k(r|jnd|k(r|jnd|k(r|jnd|k(r|jnd|k(r|jnd|k(r|jnd|k(r|jnid |k(r|jnSd |k(r|jn=d |k(r|j!n'd |k(r|j#ntd |zdz|j%|j'j)j+d}t-j.||}t,j0j2|_|S)Nz'Need a dataset to compute _cell_qualityareaaspect aspect_gamma conditiondiagonaljacobian max_angleshearskew min_anglevolumezUnknown cell quality [z]. CellQuality)rrrrrr r,SetQualityMeasureToAreaSetQualityMeasureToAspectRatioSetQualityMeasureToAspectGammaSetQualityMeasureToConditionSetQualityMeasureToDiagonalSetQualityMeasureToJacobianSetQualityMeasureToMaxAngleSetQualityMeasureToShearSetQualityMeasureToSkewSetQualityMeasureToMinAngleSetQualityMeasureToVolumer-r.r+GetArrayr vtkDataArrayToVTKArrayr!r)r#)r4qualityr9r6r8anss r< _cell_qualityr[Xs <(QRR    BMM$W&&'  F  7 "V%C%C%E 7 "V%J%J%L 7 "V%J%J%L 7 "V%H%H%J 7 "V%G%G%I 7 "V%G%G%I 7 "V%G%G%I 7 "V%D%D%F 7 "V%C%C%E 7 "V%G%G%I 7 "V%E%E%G6w>tCD D MMO     + + - 6 6} EF $ $VW 5C**//CO Jr>c|dvrtd|zdzdz|jdk7r't|dzdzt|jz|jd|jd k7r't|d zdzt|jz|jj s|j }|jd }|jd|jd z}|j||}t}|j|ddtj|}|jd |jj|t}|d k(r|j!nA|dk(r|j#n+|dk(r|j%n|dk(r|j'|j)||j+|j-jj/|}t1j2|}|j4|_|j6|_|S)N) DeterminantInverse Eigenvalue EigenvectorzUnknown quality measure []z> Supported are [Determinant, Inverse, Eigenvalue, Eigenvector]rz/ only works for an array of matrices(3D array).z Input shape rz) requires an array of 2D square matrices.rtensorsr]r^r_r`)rndimrrrrrrr SetDimensionsrrrr& SetTensorsr SetOperationToDeterminantSetOperationToInverseSetOperationToEigenvalueSetOperationToEigenvectorr,r-r.rWr rXr#DataSet)r3 operationnrowsncolsr9r8r6rZs r<_matrix_math_filterro{sOO 5i?CZ[\\{{a )$UU*+-0->?@@ aFLLO + )$OO*+-0->?@@ << " "fkkmV LLOE LLOfll1o -E ^^E5 )F BUAq!  ' ' /F NN9OO  ( "F m #v'G'G'I i v'C'C'E l "v'F'F'H m #v'G'G'I  MMO     , , . 7 7 BF $ $V ,C((CO..CK Jr>c,tj|S)zCReturns the absolute values of an array of scalars/vectors/tensors.)numpyabsr3s r<rrrr 99V r>c|tjurtjStjtj||}|Sz=Returns the min value of an array of scalars/vectors/tensors.)r NoneArrayrqallarrayr3axisrZs r<rxrxs6  ]] ))EKK' .C Jr>ct|dS)z0Returns the surface area of each cell in a mesh.r@r[r4s r<r@r@ & ))r>ct|dS)z0Returns the aspect ratio of each cell in a mesh.rAr}r~s r<rArAs ( ++r>ct|dS)z6Returns the aspect ratio gamma of each cell in a mesh.rBr}r~s r<rBrBs . 11r>ct|dS)z4Returns the condition number of each cell in a mesh.rCr}r~s r<rCrC + ..r>c|tjus|tjurtjS|j|jk7s|j|jk7r:t dt |jzdzt |jz|jdk7rI|jdk7r:t dt |jzdzt |jz|jdk(rL|jddk7r:t dt |jzdzt |jz|jdk(rL|jddk7r:t d t |jzdzt |jzt j||S) zJReturn the cross product for two 3D vectors from two arrays of 3D vectors.z?Both operands must have same dimension and shape. Input shapes  and rrbzHCross only works for 3D vectors or an array of 3D vectors. Input shapes rrz.Cross only works for 3D vectors. Input shapes z9Cross only works for an array of 3D vectors.Input shapes )r rwrdrrrrqcross)xys r<rrsCMMQ#--/ ]]vv177agg- +-0\:c|s |j}|s td|jdk7s|jddk7r!tdt |jzt }|j t||d|}|j}|jdtj||}|j|_ |S)z+Returns the curl of an array of 3D vectors.zNeed a dataset to compute curl.rbrrz8Curl only works with an array of 3D vectors.Input shape r vorticity) rkrrdrrr SetVectorModeToComputeVorticityr=r0rr rXr#r3r4cdresretValrZs r<curlrs v~~W <(IJJ {{a6<<?a/ )+.v||+<=>>  B&&( FGY ;C ^^ F NN; $ $VW 5C((CO Jr>c|s |j}|s td|jdk7s|jddk7r!tdt |jzt ||}|j |jddd}tjtjj|jddd|j} |j|_ |S#t$rY|SwxYw) z1Returns the divergence of an array of 3D vectors.z$Need a dataset to compute divergencerbrrz?Divergence only works with an array of 3D vectors. Input shape raxis1axis2r~)rkrrdrrgradientrr VTKArrayrqaddreducerDr#AttributeError)r3r4gas r< divergencers v~~W <(NOO {{a6<<?a/ *,/ ,=>?? !A !''!*a#A II  QZZaqZ91 =qyy RA   H 4 H sC.. C;:C;ct|dS):Returns the determinant of an array of 2D square matrices.r]rorss r<detr  v} 55r>ct|S)r)rrss r< determinantr v;r>ct|dS)z6Returns the diagonal length of each cell in a dataset.rDr}r~s r<rDrD * --r>c|tjus|tjurtjS|jd|jdk7r:tdt |jzdzt |jz||z}tj t jj|d}t|dr|j|_ |j|jk(r|j|_ |S)zOReturns the dot product of two scalars/vectors of two array of scalars/vectors.rzDDot product only works with vectors of same dimension. Input shapes rr#) r rwrrrrrqrrhasattrr#rk)a1a2mvas r<dotrs S]]bCMM1 ]] xx{bhhqk! *,/M:ct|dS)z9Returns the eigenvalue of an array of 2D square matrices.r_rrss r< eigenvaluer's v| 44r>ct|dS)z:Returns the eigenvector of an array of 2D square matrices.r`rrss r< eigenvectorr+rr>c||s |j}|s td |jd}|dk7r&|dk7r!tdt |jzt }|dk(rd}nd}t ||||}|dk(r|j}n|j} |jr#|jd|jzn|jdtj||}|j|_|S#t$rd}YwxYw#t$r|jdYVwxYw) z4Returns the gradient of an array of scalars/vectors.z"Need a dataset to compute gradientrrzUGradient only works with scalars (1 component) and vectors (3 component) Input shape rrz gradient of r)rkrr IndexErrorrr r=r0 GetTensorsGetNamerrr rXr#)r3r4r7rr5rrrZs r<rr/s. fnnG ,'KLLll1oe zeqj *,/ ,=>??  B zYN%> FG^R @C zS^^-FNN$67 >> fnn^fnn>N-NO~~j)  $ $VW 5C((CO J5 e$ 6FNN:66s$D AD DDD;:D;ct|dS)z3Returns the inverse an array of 2D square matrices.r^rrss r<invrRs vy 11r>ct|S)z6Returns the inverse of an array of 2D square matrices.)rrss r<inverserVrr>ct|dS)z7Returns the jacobian of an array of 2D square matrices.rEr}r~s r<rErEZrr>cf|s |j}|s tdt||}t|S)z,Returns the jacobian of an array of scalars.z#Need a dataset to compute laplacian)rkrrr)r3r4rZs r< laplacianr^s1 v~~W <(MNN 67 #C c?r>c,tj|SzEReturns the natural logarithm of an array of scalars/vectors/tensors.)rqlogrss r<lnrertr>ct|Sr)rrss r<rris f:r>c,tj|S)zEReturns the base 10 logarithm of an array of scalars/vectors/tensors.)rqlog10rss r<rrms ;;v r>cv|tjurtjStj||}|S)zAReturns the maximum value of an array of scalars/vectors/tensors.)r rwrqmaxrzs r<rrq-  ]] ))FD !C Jr>ct|dS)z4Returns the maximum angle of each cell in a dataset.rFr}r~s r<rFrFyrr>c@tjt||S)z5Returns the magnigude of an array of scalars/vectors.)rqsqrtrrs r<magr}s ::c!Qi  r>c|tjurtjStjtj||}|Sz>Returns the mean value of an array of scalars/vectors/tensors.)r rwrqmeanryrzs r<rrs6  ]] **U[[($ /C Jr>cv|tjurtjStj||}|Srv)r rwrqminrzs r<rrrr>ct|dS)z4Returns the minimum angle of each cell in a dataset.rIr}r~s r<rIrIrr>cZ|t|j|jddfz S)z=Returns the normalized values of an array of scalars/vectors.rr)rrrrs r<normrs& SV^^QWWQZO , ,,r>ct|dS)z,Returns the shear of each cell in a dataset.rGr}r~s r<rGrGs ' **r>ct|dS)z+Returns the skew of each cell in a dataset.rHr}r~s r<rHrHrr>c|s |j}|s tdd|jk7sd|jdk7r!tdt |jzt }|j t||d|}|j}|jdtj||}|j|_ |S)z-Returns the strain of an array of 3D vectors.z Need a dataset to compute strainrbrrz9strain only works with an array of 3D vectorsInput shape rstrain) rkrrdrrr SetTensorModeToComputeStrainr=rrr rXr#rs r<rrs v~~W <(JKKFKK1 Q/ )+.v||+<=>>  B##% FGY ;C ^^ F NN8 $ $VW 5C((CO Jr>cr|tjurtjStj||Srv)r rwrqsumr3r{s r<rr)  ]] 99VT ""r>c|s td|j}|jd|j|jt }|j ||j|j|jd|j|j|j|j|j|j|j!j#j%}t'j(||}t&j*j,|_|S)z5Returns the surface normal of each cell in a dataset.z(Need a dataset to compute surface_normalN)rrrrrr r,ComputeCellNormalsOnComputePointNormalsOffSetFeatureAngle SplittingOffConsistencyOffAutoOrientNormalsOffFlipNormalsOffNonManifoldTraversalOffr-r.r+ GetNormalsr rXr!r)r#r4r9r6r8rZs r<surface_normalrs  <(RSS    BMM$W&&'  !F  ! !!# 3   !  ""$ MMO     + + - 8 8 :F $ $VW 5C**//CO Jr>c`d}d}|jdkDrd}d}tj|||S)z4Returns the trace of an array of 2D square matrices.rrrbr)rdrqtrace)r3ax1ax2s r<rrs4 C C {{Q s s ;;vS 44r>cr|tjurtjStj||Sr)r rwrqvarrs r<rrrr>c4|s td|j}|jd|j|jt }|j ||j|j|j|j|jjjd}|jdtj ||}tj"j$|_|S)z-Returns the volume of each cell in a dataset.z Need a dataset to compute volumeNVolumerK)rrrrrr r,ComputeVertexCountOffComputeLengthOffComputeAreaOffr-r.r+rWrr rXr!r)r#rs r<rJrJs <(JKK    BMM$W&&'  F    "   MMO     + + - 6 6x @F NN=! $ $VW 5C**//CO Jr>ct||S)z5Returns the vorticity/curl of an array of 3D vectors.)r)r3r4s r<rrs   r>c|s td|j}|jd|j|jt }|j ||j|j|jd|j|j|j|j|j|j|j!j#j%}t'j(||}t&j*j,|_|S)z5Returns the vertex normal of each point in a dataset.z'Need a dataset to compute vertex_normalNr)rrrrrr r,ComputeCellNormalsOffComputePointNormalsOnrrrrrrr-r.r&rr rXr!r$r#rs r< vertex_normalr s  <(QRR    BMM$W&&'  !F    "   " 3   !  ""$ MMO     , , . 9 9 ;F $ $VW 5C**00CO Jr>c|tjus$|tjus|tjurtjSt|jdk7s2t|jdk7s|#t|jdk7r t d|t j |j}t j|||gjjtj} |j|_ |j|_ |S#t$rYwxYw#t$rY|SwxYw)NrzCan only merge 1D arrays)r rwrr ValueErrorrqzerosvstack transposeviewrrkrr#)axayazvs r< make_vectorr*s S]]bCMM1R3==5H ]] 288}S]a/BNs288}XYGY344 z [[ " b"b\",,.33CLLAA JJ   H 4 4 H s$D&D5& D21D25 EE)N)? __future__rrr rqvtkmodules.utilrvtkmodules.vtkCommonDataModelrvtkmodules.vtkFiltersCorerr vtkmodules.vtkFiltersGeneralr vtkmodules.vtkFiltersVerdictr r r r=r[rorrrxr@rArBrCrrrrrrDrrrrrrrErrrrrrFrrrrIrrGrHrrrrrrJrrrr>r<r s&$ )6P;__Lg\!F/d*,2/.2 (6. 56!F2./!/-+*2# <5# 6!< r>