pjG ddlZddlZddlZddlZddlZej jddZejeejd<ddl m Z ddl mZddZdZdd Zdd Zd Zd d ddddddddd ZdZddZy)NVTK_MODULE_NAME vtkmodules vtk_module) vtk_to_numpy)vtkDataSetSurfaceFilterc|yd}|jdr|}n;t}|j||j|j }|j ydii}t |j }|r||dd<t |j}|r||dd<t |j} | r| |dd<t |j} | r| |dd<t |j} | r| |dd<|ud} |jj|} |jj|}|rt|d | } | rt| d | } | r|jd | i|r9t!|j|d }t#|r|jd |i|r9t!|j|d }t#|r|jd|i|S)zNExpect any dataset and extract its surface into a dash_vtk.Mesh state propertyN vtkPolyDatameshpointsvertslinespolysstripsCellData)locationname PointDatafield pointArrays cellArrays)IsAr SetInputDataUpdate GetOutput GetPoints mesh_arrayGetVertsGetLinesGetPolys GetStrips GetPointDataGetArray GetCellData data_arrayupdate field_datalen)dataset field_to_keep point_arrays cell_arrayspolydataextract_skin_filterstater r r rrrp_arrayc_array point_data cell_datas S/DATA/.local/lib/python3.12/site-packages/trame_vtk/modules/vtk/serializers/mesh.pyr r sH{{=!57((1""$&002#RLE **, -F "(f h x((* +E !&f g x((* +E !&f g x((* +E !&f g **, -F "(f h '')22=A&&(11-@ w-PE w=QE  LL'5) * 5 5 7{S z? LL-4 5x335{JO y> LL, 2 3 LcL|r"tt|jSyN)b64_encode_numpyrGetData)arrays r3rrVs   U]]_ =>> r4c|rl|jd}|j}t|}tt |j }|r|n|j t|||||dSy)N)rvaluesnumberOfComponentstyper dataRange)GetRangeGetNumberOfComponentsr to_js_typestrdtypeGetNamer7)r$rr data_rangenb_compr<js_typess r3r$r$[ss((, 224j)c&,,/0 Dj&8&8&:&v.") #   r4c|g}|D]4}|j|}t|||}|s$|j|6|Sr6)r"r$append)r&namesrfieldsrr9js_arrays r3r&r&ksE F##D)eXt4  MM( #  Mr4c||jdsyd|j|j|jdi}t |dr[|j }g}t dD]3}t dD]#}|j|j||%5||dd<|jj}t|d }|r||d <|S) zKExpect a vtkImageData and extract its setting for the dash_vtk.Volume stateN vtkImageDataimage) dimensionsspacingoriginGetDirectionMatrix directionr)rr) r GetDimensions GetSpacing GetOriginhasattrrTrangerJ GetElementr! GetScalarsr$)r(r.matrixjs_matjiscalarsrs r3volumercvsgkk.9 !//1))+'')  Ew,-++-qA1X f//156'-g{#""$//1G w 5E g Lr4 Int8Array Uint8Array Int16Array Uint16Array Int32Array Uint32Array Float32Array Float64Array) int8uint8int16uint16int32uint32int64uint64float32float64ct|dk(r|jS|j}|jdk(r t |S|jdk(rt |t j S|jdvrt j|}t j|}|dk}t|| }|r|t jt jkrt |t jS|t jt jjkrt |t jS|t jt jjkrt |t jS|jS|t jt j jkrt |t j S|t jt jjkrt |t jS|t jt jjkrt |t jS|jS)Nrfb)ura)r'tolistrDkind np_encodenprmamaxaminmaxiinforlrnrprorq)objrD max_value min_valuesigned test_values r3r7r7s 3x1}zz| IIE zzS~ s bhh'' z !GGCL GGCL QYJ/ BHHRWW-- bgg..BHHRXX.222 bhh//BHHRXX.222 bhh// ::<BHHRXX.222 bhh//BHHRYY/333 bii00BHHRYY/333 bii00 ::<r4c|rx|j|jd}tjt |j dt |jt|jdStjt |jdj dt |jt|jdS)NC)orderzutf-8)bvalsrDshape) astyperavelbase64 b64encode memoryviewdecoderCrDlistr)r9np_typen_arrays r3r|r|s,,w'--C-8%%j&9:AA'J'%++&  !!*U[[s[-C"DELLWUU[[!ekk" r4)NNN)rN)rr6)rnumpyr}os importlibsysenvirongetvtk_module_name import_modulemodulesvtk_module.util.numpy_supportrvtk_module.vtkFiltersGeometryrr rr$r&rcrBr7r|r4r3rs   **..!2LA3I33OD L6ADN?  B       !H r4