g#ndZddlmZddlmZmZmZddlZejZ e dk(r ejZ ne dk(r ejZ ejZ e dk(rejZejZne dk(rejZej"ZdZd Zd Zd Zdd Zdd ZdZy)aThis module adds support to easily import and export NumPy (http://numpy.scipy.org) arrays into/out of VTK arrays. The code is loosely based on TVTK (https://svn.enthought.com/enthought/wiki/TVTK). This code depends on an addition to the VTK data arrays made by Berk Geveci to make it support Python's buffer protocol (on Feb. 15, 2008). The main functionality of this module is provided by the two functions: numpy_to_vtk, vtk_to_numpy. Caveats: -------- - Bit arrays in general do not have a numpy equivalent and are not supported. Char arrays are also not easy to handle and might not work as you expect. Patches welcome. - You need to make sure you hold a reference to a Numpy array you want to import into VTK. If not you'll get a segfault (in the best case). The same holds in reverse when you convert a VTK array to a numpy array -- don't delete the VTK array. Created by Prabhu Ramachandran in Feb. 2008. ) vtkConstants) vtkDataArrayvtkIdTypeArray vtkLongArrayNctjtjtjtj tj tjtjtjtjtjtjtjtjtjtj tj"tj$tj&tj(tj*tj,tj&tj.tj*i }|j1D]<\}}||k(s/tj2||s|tj4|k(s:|cSt7dt9|z)z+Returns a VTK typecode given a numpy array.z)Could not find a suitable VTK type for %s)numpyuint8rVTK_UNSIGNED_CHARuint16VTK_UNSIGNED_SHORTuint32VTK_UNSIGNED_INTuint64VTK_UNSIGNED_LONG_LONGint8VTK_SIGNED_CHARint16 VTK_SHORTint32VTK_INTint64 VTK_LONG_LONGfloat32 VTK_FLOATfloat64 VTK_DOUBLE complex64 complex128items issubdtypedtype TypeErrorstr)numpy_array_type_np_vtkkeyvtk_types =lib/python3.12/site-packages/vtkmodules/util/numpy_support.pyget_vtk_array_typer,1s%{{<99 \<< \:: \@@ <77 L22 L00 L66 l44 l55 6 6  !8!8 :G! X s "   ,c 2 u{{3/ /O ) 3s;K7LM OOcRtjtjtjtj tj tj tjtjtjtjtjtjtjtjtjtjtj t"tj$tj&tj(t*tj,tj.tj0t2tj4tj6tj8tj:i}|S)z7Returns the VTK array type to numpy array type mapping.)rVTK_BITr r VTK_CHARrrr rrrrrrrrVTK_LONGLONG_TYPE_CODErrVTK_UNSIGNED_LONGULONG_TYPE_CODErr VTK_ID_TYPE ID_TYPE_CODErrrr)_vtk_nps r+get_vtk_to_numpy_typemapr8Hs##EKK%%ejj,,UZZ..u{{&&u{{// $$U[[--ell%%n**5;;..33ELL((&&u}}'' 7G Nr-ct|S)z6Returns a numpy array typecode given a VTK array type.)r8)vtk_array_types r+get_numpy_array_typer;\s # %n 55r-c,tj|S)zkInternal function used to create a VTK data array from another VTK array given the VTK array type. )rCreateDataArray) vtk_arr_types r+create_vtk_arrayr?as  ' ' 55r-c2tj|}|jjstj|}|j }|jjsJdt |dksJdtj|jtjtjrJd|r|}nt|j}t|} |d}t |dk(r|jdn|j|d|j|dt|}tj|j|s"|jtj|k(rtj |} n&tj |j#|} d}|j%| t | d|r%|j'} | j)|| }|S||_|S#d}Y$xYw)a=Converts a real numpy Array to a VTK array object. This function only works for real arrays. Complex arrays are NOT handled. It also works for multi-component arrays. However, only 1, and 2 dimensional arrays are supported. This function is very efficient, so large arrays should not be a problem. If the second argument is set to 1, the array is deep-copied from from numpy. This is not as efficient as the default behavior (shallow copy) and uses more memory but detaches the two arrays such that the numpy array can be released. WARNING: You must maintain a reference to the passed numpy array, if the numpy data is gc'd and VTK will point to garbage which will in the best case give you a segfault. Parameters: num_array a 1D or 2D, real numpy array. z%Only contiguous arrays are supported.z5Only arrays of dimensionality 2 or lower are allowed!zComplex numpy arrays cannot be converted to vtk arrays.Use real() or imag() to get a component of the array before passing it to vtk.rrr)r asarrayflags contiguousascontiguousarrayshapelenr#r$complextyper,r?SetNumberOfComponentsSetNumberOfTuplesr;ravelastype SetVoidArray NewInstanceDeepCopy_numpy_reference) num_arraydeep array_typezrG vtk_typecode result_arraytestVar arr_dtypez_flatcopys r+ numpy_to_vtkr]hs2  i A 77    # #A & GGE 77  FFF  u:>C BC>W)=)B)BC! !! C ! )!''2 #L1L(  5zQ**1-**584""58,%\2I +ww%++i((QQ&&y1fc&k15 '') l#  )* % As -HHc2tj}|j}|dk(r*|tjk7rAt dt |z|tjk7rt dt |zt||tjS)Nrz.Expecting a numpy.int32 array, got %s instead.z.Expecting a numpy.int64 array, got %s instead.) rGetDataTypeSizer$r r ValueErrorr&rr]rr5)rSrTisizer$s r+numpy_to_vtkIdTypeArrayrbs   , , .E OOE z EKK  =U LN N EKK  =U LN N  4)A)A BBr-c|j}|tjvs Jd|z|tjk7sJd|j |j f}t|} tj||}|ddk(r|df} ||_ |S#t$r$|ddk(rtj||}nYDwxYw#t$r$|ddk(rtj||}Y|SwxYw)aLConverts a VTK data array to a numpy array. Given a subclass of vtkDataArray, this function returns an appropriate numpy array containing the same data -- it actually points to the same data. WARNING: This does not work for bit arrays. Parameters vtk_array The VTK data array to be converted. zUnsupported array type %szBit arrays are not supported.)r$rr) GetDataTyper8keysrr/GetNumberOfTuplesGetNumberOfComponentsr;r frombufferr`emptyrG) vtk_arraytyprGr$results r+ vtk_to_numpyrms0    !C *,113 3+ &s *+ 3 ,&& &G(GG &  ' ' )  + + - .E ! %E !!)59 Qx1}q   M+   8q=[[e4F     8q=KKU36 M s$;B*!C**CC(DD)rNrB)__doc__rvtkmodules.vtkCommonCorerrrr r_VTK_ID_TYPE_SIZErr6rVTK_LONG_TYPE_SIZEr2rr4rr,r8r;r?r]rbrmr-r+rts8OO "#335q;;L;;L!^335[[NllO1[[NllOO.(6 6Pd C0r-