:vjO"dZddlZddlmZddlmZddlmZddlm Z  dd l m Z dd Zdd Zd ZddZedZedZedZej.eeeZddZdZdZddZy#e $rZ dd lmZee Z YdZ [ cdZ [ wwxYw)z6Basic morphology operations that create new encodings.N)util)log_timeencoding)ops)ndimage)ExceptionWrapperct|tjr|}n5t|tjr |j }nt d||r t|||S)N-encoding must be np.ndarray or Encoding, got ) isinstancenpndarrayencEncodingdense ValueError _assert_rank)rrankrs i/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/trimesh/voxel/morphology.py_densersQ(BJJ' Hcll +H UVV UD! Lct|tjr|}n5t|tjr |j }nt d|t|d|S)Nr )rrrrrsparse_indicesr_assert_sparse_rank)rrrs r_sparse_indicesrsQ(BJJ'! Hcll +!00H UVV* rctt|j|k7r td|t|jy)NzExpected rank %d, got shape %s)lenshaperstrvaluers rrr+s1 5;;494U[[AQRR rct|jdk7rtd|j|-|jd|k7rtd||jdyy)Nrz)sparse_indices must be rank 2, got shape z*sparse_indices.shape[1] must be %d, got %d)r r!rr#s rrr0sg 5;;1DU[[OTUU  ;;r?d " filled_encoding` and should not modify encoding. Parameters -------------- encoding: Encoding object (left unchanged). method: method present in `fillers`. **kwargs: additional kwargs passed to the specified implementation. Returns -------------- A new filled Encoding object. r)fillers)rmethodr1s rfillr9zs( 6 7H 7 77rc jtjtjt |dfi|S)z Encoding wrapper around scipy.ndimage.morphology.binary_dilation. https://docs.scipy.org/doc/scipy-0.15.1/reference/generated/scipy.ndimage.morphology.binary_dilation.html#scipy.ndimage.morphology.binary_dilation rr()rr,r binary_dilationrr0s rr;r;s-   W44VH15MXQWX YYrc jtjtjt |dfi|S)z Encoding wrapper around scipy.ndimage.morphology.binary_closing. https://docs.scipy.org/doc/scipy-0.15.1/reference/generated/scipy.ndimage.morphology.binary_closing.html#scipy.ndimage.morphology.binary_closing rr()rr,r binary_closingrr0s rr=r=s-   W33F8!4LWPVW XXrcDt|d}tj|tjdtd}tj |}t j||}tj||dd dd dd f}tj|S) a\ Get elements on the surface of encoding. A surface element is any one in encoding that is adjacent to an empty voxel. Parameters -------------- encoding: Encoding or dense rank-3 array structure: adjacency structure. If None, square connectivity is used. Returns -------------- new surface Encoding. rr()rr)dtypeconstant)mode) structurerr&) rrpadonesint logical_notr r; logical_andrr,)rrBremptydilatedsurfaces rrJrJs 8! $E FF5"''&4: FE NN5 !E%%eyAGnnUG,QrT1R42-=>G   W %%r)N)r3)__doc__numpyrr constantsrrrr scipyr BaseExceptionE exceptionsr rrrrr*r-r2FunctionRegistryr7r9r;r=rJrrrUs< "  S   V  V N N&  ( $   "  8.ZY&o"-q!G "sA11B6B  B