TiO:dZddlZddlmZddlmZddlmZddlm Z  dd l m Z n##e $rZ dd lmZee Z YdZ [ ndZ [ wwxYwdd Zdd Zd ZddZedZedZedZejeeeZddZdZdZddZdS)z6Basic morphology operations that create new encodings.N)util)log_timeencoding)ops)ndimage)ExceptionWrapperct|tjr|}n4t|tjr|j}nt d||rt|||S)N-encoding must be np.ndarray or Encoding, got ) isinstancenpndarrayencEncodingdense ValueError _assert_rank)rrankrs R/DATA/AppData/hermes/venv/lib/python3.11/site-packages/trimesh/voxel/morphology.py_densersp(BJ''W Hcl + +WUUUVVV "UD!!! Lct|tjr|}n4t|tjr|j}nt d|t|d|S)Nr )rrrrrsparse_indicesr_assert_sparse_rank)rrrs r_sparse_indicesrsj(BJ''W! Hcl + +W!0UUUVVV*** rc|t|j|kr#td|t|jdS)NzExpected rank %d, got shape %s)lenshaperstrvaluers rrr+s< 5;494U[AQAQRRR rct|jdkrtd|j|-|jd|krtd||jddSdS)Nrz)sparse_indices must be rank 2, got shape z*sparse_indices.shape[1] must be %d, got %d)r r!rr#s rrr0su 5;1TU[TTUUU  ;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 7 7H 7 7 77rc htjtjt |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;s6  W4VH15M5M5MXXQWXX Y YYrc htjtjt |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=s6  W3F8!4L4L4LWWPVWW X XXrcHt|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 F5"'&444: F F FE N5 ! !E%eyAAAGnUG,,QrT1R42-=>G  W % %%r)N)r3)__doc__numpyrr constantsrrrr scipyr BaseExceptionE exceptionsr rrrrr*r-r2FunctionRegistryr7r9r;r=rJrrrUs<< """"------q!!GGGGGG "        SSS   V V  V NN N&  ( $  "    8888.ZZZYYY&&&&&&s'AAA