Ti9ddlZddlmZddlmZddlmZ ddlZ n(#e $r Z ddlm Z e j e Z YdZ [ ndZ [ wwxYwdZdd Zdd Zd Zejd fd edefdZdZddZ ddZddZdS)N)util)log)NDArray) exceptionsct|gd}|8tj|tj|ddd |dddf<|S)a Given the origin and normal of a plane find the transform that will move that plane to be coplanar with the XY plane. Parameters ---------- origin : (3,) float Point that lies on the plane normal : (3,) float Vector that points along normal of plane Returns --------- transform: (4,4) float Transformation matrix to move points onto XY plane )rrrNr) align_vectorsnpdotappend)originnormal transforms J/DATA/AppData/hermes/venv/lib/python3.11/site-packages/trimesh/geometry.pyplane_transformrsY"fiii00I F9bi.B.BCCBQBGG "1"a% Fctj|tj}tj|tj}|jdks |jdkrt dtj|dd}tj|dd}tj|dkr|dddfxxdzcc<tj|dkr|dddfxxdzcc<tj d }| |j |dd dd f<|rbtj |d|d}tj tj |dd }|d kr|tjz }||fS|S) a Find the rotation matrix that transforms one 3D vector to another. Parameters ------------ a : (3,) float Unit vector b : (3,) float Unit vector return_angle : bool Return the angle between vectors or not Returns ------------- matrix : (4, 4) float Homogeneous transform to rotate from `a` to `b` angle : float If `return_angle` angle in radians between `a` and `b` dtype)r zvectors must be (3,)!rrNrr ?gh㈵)r arrayfloat64shape ValueErrorlinalgsvdreshapedeteyer Tarccosclippi)ab return_angleaubumatrixr angles rr r 's, "*%%%A "*%%%Aw$!'T//0111 qyy)) * *1 -B qyy)) * *1 -B y}}R1 111b5 T  y}}R1 111b5 T VAYYFVVBD\\F2A2rr6NfRUBqE"" "'#tS1122 ;; RUNEu} Mrc*tj|tj}|ddgdfd}|rPtjtjt |djd}||fS|S)a Given a list of faces (n,3), return a list of edges (n*3,2) Parameters ----------- faces : (n, 3) int Vertex indices representing faces Returns ----------- edges : (n*3, 2) int Vertex indices representing edges N)rrrr1r)rr1)r rr)r asanyarrayint64r"tilearangelenr%)faces return_indexedges face_indexs rfaces_to_edgesr;\s M% * *E !!!'''' ( 0 0 9 9E!WRYs5zz22F;;=EEbII j  Lrctj|tj}t|dkrtjgSt j|dr|d}n$t j|dstdt j |dddf|dddf}tj |d d }tj tj |}|S) z Find the angles between pairs of unit vectors. Parameters ---------- pairs : (n, 2, 3) float Unit vector pairs Returns ---------- angles : (n,) float Angles between vectors in radians rrr1r )rr1r )rr1r=zpairs must be (n,2,(2|3))!Nrrr) r r2rr6rris_shaper"r diagonal_dotr'absr&)pairsdotsangless r vector_anglerDvs M%rz 2 2 2E 5zzQx|| uf % %7 j)) ]5/ 2 275666  U111a4[%1+ 6 6D 74s # #D VBIdOO $ $F MrTuse_fanreturncbtdkrtjd|S tj|tjdkr&jddkr|StjdkrRjddkrAtjddgdfddgd ff|Sn#t$rYnwxYwtjd Dtj}tjfd tj |dkdDtj}tjfd tj |dkdDtj}|r+fd tj |dkdD}ng}t|dkr(t|dkr||St|dkrtj |}t|dkr.tj|ddgdf|ddgd ff}tj |||g|S)a Given an array of quad faces return them as triangle faces, also handles pure triangles and mixed triangles and quads. Parameters ----------- quads: (n, 4) int Vertex indices of quad faces. dtype Data type requested for the return use_fan Triangulate holes larger than quads with fans, which may be wrong if the holes are non-convex Returns ----------- faces : (m, 3) int Vertex indices of triangular faces. rrr1rr rN)rrr1)r1r rc,g|]}t|S)r6).0is r z%triangulate_quads..s...1A...rc g|] }| SrIrIrJrKquadss rrLz%triangulate_quads..sBBBE!HBBBrc g|] }| SrIrIrNs rrLz%triangulate_quads..sCCC!U1XCCCrc g|] }| SrIrIrNs rrLz%triangulate_quads..s===Qa===r) r6r zerosrrastypevstackrr3nonzerortriangle_fans_to_faces vstack_empty)rOrrElengthstriquadpolys` rtriangulate_quadsr\s* 5zzQx'''' e,,, u{  q U[^q%8%8<<&& & u{  q U[^q%8%89eAAAyyyL15III3FGHHOOPUVV V      h.....bh???G (BBBBbjA&>&>q&ABBB"( S S SC 8CCCCrz'Q,'?'?'BCCC28 T T TD===="*Wq["9"9!"<=== 4yyA~~#d))q..zz%    4yy1}}*400 4yy1}}y$qqq)))|,d111iii<.@ABB  c4. / / 6 6u = ==sAC+A)C++ C87C8c tj||}nR#t$rEt jdtj|dd}YnwxYwt||ksJ||ksJtj dtj |dd}tj ||dddfz}|||zdddfk tj  } tjt|t" }||d}|||dk<n#t($rt jd d tj|jtj } |} t/|D]F} tj| | kdd zddd| || || || z<G| ||dk<YnwxYw|S) a Find vertex face indices from the faces array of vertices Parameters ----------- vertex_count : int The number of vertices faces refer to faces : (n, 3) int List of vertex indices faces_sparse : scipy.sparse.COO Sparse matrix Returns ----------- vertex_faces : (vertex_count, ) int Face indices for every vertex Array padded with -1 in each row for all vertices with fewer face indices than the max number of face indices. ) minlengthzcasting failed, falling back!T) return_countsrrNrrzJvertex_faces falling back to slow loop! mesh probably has degenerate facesexc_infor )r bincountflatten TypeErrorrwarninguniquer6maxr cumsumr5rSr3scipysparseidentityintr rU BaseExceptionrRsizerangewhere) vertex_countr7 faces_sparsecountsstartspackpaddedrk sorted_facessortflatvs rvertex_face_indicesr{sg,CU]]__ EEE CCC 34445==??$???B C v;;, & & & & 99;; % % % %Yq")F++CRC0 1 1F 9VZZ\\ " "VAAAtG_ 4DD1199"(CC CF#<((U3(??#''1199;;A> +v{  # # #  3    x "(333}}|$$ Z ZA79x 7J7J17MQR7RTXTXVXTX6YDVAY22 3 3"v{ # Ms$(+A A:9A:A)F**B8I%$I%c fd}fd} |}n0#t$r#tjdd|}YnwxYwtj|}|S)a Find vertex normals from the mean of the faces that contain that vertex. Parameters ----------- vertex_count : int The number of vertices faces refer to faces : (n, 3) int List of vertex indices face_normals : (n, 3) float Normal vector for each face Returns ----------- vertex_normals : (vertex_count, 3) float Normals for every vertex Vertices unreferenced by faces will be zero. c\t}n}|}|SN) index_sparser )r.summed face_normalsr7rjrqs r summed_sparsez*mean_vertex_normals..summed_sparse/s6 >!,66FFFL)) rctjdf}tD]\}}||xx|z cc<|SNr )r rRzip)rfacerrr7rqs r summed_loopz(mean_vertex_normals..summed_loop:sS<+,,|44 # #LD& 4LLLF "LLLL r*unable to use sparse matrix, falling back!Tr`)rmrrerunitize) rqr7rrjkwargsrrrvertex_normalss ```` rmean_vertex_normalsrs*          @4PPPP \&))N s *A  A cVfd}fd}dzddk}||||sC tj|S#t$rt jdd YnwxYwtj|S) aV Compute vertex normals from the faces that contain that vertex. The contribution of a face's normal to a vertex normal is the ratio of the corner-angle in which the vertex is, with respect to the sum of all corner-angles surrounding the vertex. Grit Thuerrner & Charles A. Wuethrich (1998) Computing Vertex Normals from Polygonal Facets, Journal of Graphics Tools, 3:1, 43-46 Parameters ----------- vertex_count : int The number of vertices faces refer to faces : (n, 3) int List of vertex indices face_normals : (n, 3) float Normal vector for each face face_angles : (n, 3) float Angles at each vertex in the face Returns ----------- vertex_normals : (vertex_count, 3) float Normals for every vertex Vertices unreferenced by faces will be zero. cvt}|S)N)data)rravelr )r. face_anglesrr7rqs rrz.weighted_vertex_normals..summed_sparsems7 lE 8I8I8K8KLLLzz,'''rc(tjdftj}tjD]Z}tj|k\}}||f}tj||z |||<[|Sr)r rRrr5rpr sum) r vertex_idx face_idxs inface_idxssurrounding_anglesrrr7rqs rrz,weighted_vertex_normals..summed_loopus<+RZ88)L11  J&(Xez.A%B%B "I{!,Y -C!D !#"%7%;%;%=%==|I?V""F:   rr1r)axisg?rTr`)rrrrmrre)rqr7rruse_looprrface_oks```` rweighted_vertex_normalsrNs >((((((((        Q###++c1G 'NE(Lg&K U U< 00 0 U U U KDt T T T T T T U < & &&s A)) B  B ctj|}t|}|d}tjtjt |dd|jdfd}|t |f}|)tjt |t}n/t |t |krtd|| |}tj |||ff||jS)a Return a sparse matrix for which vertices are contained in which faces. A data vector can be passed which is then used instead of booleans Parameters ------------ columns : int Number of columns, usually number of vertices indices : (m, d) int Usually mesh.faces Returns --------- sparse: scipy.sparse.coo_matrix of shape (columns, len(faces)) dtype is boolean Examples ---------- In [1]: sparse = faces_sparse(len(mesh.vertices), mesh.faces) In [2]: sparse.shape Out[2]: (12, 20) In [3]: mesh.faces.shape Out[3]: (20, 3) In [4]: mesh.vertices.shape Out[4]: (12, 3) In [5]: dense = sparse.toarray().astype(int) In [6]: dense Out[6]: array([[1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 0, 1, 1, 1, 0, 0, 0, 0], [0, 0, 1, 1, 0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 1, 1, 0, 0, 1, 0, 0], [0, 0, 1, 0, 1, 0, 0, 1, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 1, 1, 0, 1, 0, 0, 0, 1], [1, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1, 1, 0, 0], [0, 0, 0, 0, 0, 1, 0, 0, 1, 0, 0, 0, 0, 0, 1, 0, 1, 1, 0, 0], [0, 1, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 1, 1], [0, 0, 0, 0, 0, 0, 1, 0, 0, 1, 0, 0, 0, 0, 0, 1, 0, 0, 1, 1]]) In [7]: dense.sum(axis=0) Out[7]: array([3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3]) rrrNrzdata incorrect length)rr)r r2rlr"r4r5r6ronesboolrrSrirj coo_matrixr)columnsindicesrrrowcolrs rrrsdmG$$G'llG //"  C ' #g,,''001gmA6F2G   gbkkc'll #E |ws3xxt,,, Tc#hh  0111 {{5!! < " "D3*#5U$* " U UUr)Fr~)NN)numpyr r constantsrtypedr scipy.sparserirmErExceptionWrapperrr r;rDr3rr\r{rrrrIrrrs++++ (J ' * *EEEEEE +.2222j4@$&8TA>A>dA>gA>A>A>A>H===@1111j>CB'B'B'B'JEVEVEVEVEVEVsA=A