:vj35ddlZddlmZddlmZddlmZmZdZ dZ e Z dded eezd edzfd Z d Z dd ZddZej fdZdZddZddZddZy)N)util)log) ArrayLikeNumberc<|jtjtjdDddidtj|fd}|d}t dt D]}tj|||} |S) Nc3FK|]}tj|yw)N)nparange).0ss b/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/trimesh/voxel/ops.py z$fill_orthographic.. s2EqbiilEs!indexingijaxiscd|f}|j}||<tj||d}|j}d|<tj||d}tj||k\||kS)N.T)rkeepdimsr)copyr minmax logical_and)rbase_local_indices local_indicesminsmaxsemptyindicesshapes r fill_axisz$fill_orthographic..fill_axiss$S$Y/*//1 $T{ evvm$>*//1 ! evvm$>~~ $ & $ &  r)r!r stackmeshgrid logical_notrangelenr)denser"filledrrr r!s @@@rfill_orthographicr,s KKEhh 2E2BTBG NN5 !E  A FaU$ $8% Mr#c tj|tj}tj|ds t d|j dz}tj|||ft}tj|d}d|t|j<t|D]c}d}td|dz D]}g}tj|||dd f|||ddf}tjtj|ddz}t!|} | d zdkDxr| d kD}| d krytd| | d zz d D]} d||||| || dzf<|std|dz D]} g}tj||dd | f||dd| f}tjtj|ddz}t!|} | d krjtd| | d zz d D]} d|||| || dz| f<ftj"tj|} | dz} | S) z Given a sparse surface voxelization, fill in between columns. Parameters -------------- sparse_indices: (n, 3) int, location of filled cells Returns -------------- filled: (m, 3) int, location of filled cells dtyperzincorrect shaper1r$FrNr)r asanyarrayint64ris_shape ValueErrorrzerosbooladdtupleTr(equalwherer'r) column_stack) sparse_indices max_valuegrid voxels_sparsei check_dir2jidxeqcr kr+s r fill_baserJ#sG]]>BN == 1*++""$q(I 88Y 95t 1-M#$D}  9  q)a-(AC$q!SbSy/41ab>:B((2>>"-.q1A5CCAa%1.QJ1u1a!a%i+23Q3q6CAJ../,) q)a-(AC$q#2#qy/412q>:B((2>>"-.q1A5CCA1u1a!a%i+23QAQU+Q./,)%<__RXXd^ ,F aKF Mr#matrixpitch thresholdcfddlm}ddlm}|t j ||kD}nt j |t }t j|}d}t j||dd }t|d r |j}n |j}t j |}|jdk(r|fd z}||d | }t|dk(rtjd|\} } d} nt|dk(r|\} } } } t j  ||z} ||   } | S)a& Convert an (n, m, p) matrix into a mesh, using marching_cubes. Parameters ----------- matrix : (n, m, p) bool Occupancy array pitch : float or length-3 tuple of floats, optional Voxel spacing in each dimension threshold : float or None, optional If specified, converts the input into a boolean matrix by considering values above `threshold` as True Returns ---------- mesh : trimesh.Trimesh Mesh generated by meshing voxels using the marching cubes algorithm in skimage r)measurerTrimeshNr.r$constant) pad_widthmodeconstant_valuesmarching_cubes_lewinerr1g?)volumelevelspacingz0using old marching cubes, may not be watertight!r2)verticesfacesvertex_normals)skimagerObaserQr r3r8r'padhasattrrVmarching_cubessizer)rwarningsubtract)rKrLrMrOrQ rev_matrixrSfuncmeshedrZr[normals_valsmeshs rmatrix_to_marching_cubesrkcs(2 v&2vT2'JIy QJ w01--%% MM% E zzQ1  F 6{a FG % V *0'%%{{8Y%67H HE' JD Kr#ctj|tj}tj|ds t d|j ddz}tjtj|t}tjtj|dd|ddg}||zjd}d ||<|j|}|S) a Take a sparse (n,3) list of integer indexes of filled cells, turn it into a dense (m,o,p) matrix. Parameters ----------- sparse : (n, 3) int Index of filled cells Returns ------------ dense : (m, o, p) bool Matrix of filled cells r.r0zsparse must be (n,3)!rrr$NrT) r r3r4rr5r6rr7prodr8arraysumreshape)sparser!rK multiplierindexr*s rsparse_to_matrixrts ]]6 2F == )011 JJAJ  "E XXbggenD 1F27759-uQx;?  IIc'l #S\ xx%077< E<01 )8+l:HHQAeAhi7QR GGCF$/ K F# ;;! D\((WA(>F ;;! F s7| ; ---3I>K AQy\{ KD KRsL#cddl}tj|jd|j d|jd|j dg}|jddz }t tj |ddz}|j||z jtjt|t|}|j||z jtjt|t|}|||} tj| j|z} | S)a Find common rows between two arrays very quickly using 3D boolean sparse matrices. Parameters ----------- a: (n, d) int, coordinates in space b: (m, d) int, coordinates in space operation: numpy operation function, ie: np.logical_and np.logical_or Returns ----------- coords: (q, d) int, coordinates in space rNrr$rr.)datar!)rqr rnrrr:ptpCOOr;rr)r8r>coords) ar operationrqextremar{rbsp_asp_bappliedrs rboolean_sparser<s&hh1 quu!u}aeeemQUUPQU]STG [[a[ 1 $F a(1, -D ::q6znn2773q6+Fd: SD ::q6znn2773q6+Fd: SDd#G __W^^ ,v 5F Mr#c|j}t|}g}g}tt|D]}tt|tt|dz|z}t j ||}t j |\}|d} |d} |j| | g|jt| | |t|t j|tfS)Nr$rrr) r!r)r(r:r anynonzeroappendslicernr) rr!ndimspaddingslicesdimrr+r pad_left pad_rights r strip_arrayrcs JJE JEG FSZ U3Z 5sQw)>#??4(ZZ' 1:BK ),- eHi01! f # 6 66r#c@tj|}|jdddk7r tdtj|tj }||t |z}|4tj|}|jdk7r td||z }|S)ak Convert indices of an (n,m,p) matrix into a set of voxel center points. Parameters ---------- indices: (q, 3) int, index of voxel matrix (n,m,p) pitch: float, what pitch was the voxel matrix computed with origin: (3,) float, what is the origin of the voxel matrix Returns ---------- points: (q, 3) float, list of points r$Nr1zshape of indices must be (q, 3)r.shape of origin must be (3,))r r3r!r6rnrvrw)r rLr{rzs rindices_to_pointsrssmmG$G}}QRD :;; XXgRZZ 0F %, v& <<4 ;< <& Mr#crtjtj|}t|||}|S)aS Convert an (n,m,p) matrix into a set of points for each voxel center. Parameters ----------- matrix: (n,m,p) bool, voxel matrix pitch: float, what pitch was the voxel matrix computed with origin: (3,) float, what is the origin of the voxel matrix Returns ---------- points: (q, 3) list of points )r rLr{)r r>rr)rKrLr{r rzs rmatrix_to_pointsrs.oobjj01G weF KF Mr#ctj|tj}|j|jddfk7r t d|4tj |}|jdk7r t d||z}|||z}tj |tj}t |}tj|jt}|S)a Convert center points of an (n,m,p) matrix into its indices. Parameters ---------- points : (q, 3) float Center points of voxel matrix (n,m,p) pitch : float What pitch was the voxel matrix computed with origin : (3,) float What is the origin of the voxel matrix Returns ---------- indices : (q, 3) int List of indices r.rr1zshape of points must be (q, 3)rr) r rnrvr!r6r3rwrxryr)rzrLr{r s rpoints_to_indicesrs$XXfBJJ /F || Q++9:: v& <<4 ;< <& % ]]6 4F %LEhhv%%c*G Nr#)?N)r)rNF)NN)numpyr rr constantsrtypedrrr,rJfill_voxelizationrkrtr|rrrrrrrr#rrs%6:z !$#H H I H}HV>DJZ$&>>$N 7 <&"r#