j!0*dZddlZddlmZmZddlmZddlm Z m Z ddlm Z ddl m Z mZ ddlZ dd lmZd Zdd efd ZddZddZddefdZddZy#e$rZdd lmZeeZYdZ[=dZ[wwxYw#e$rZdd lmZeeZYdZ[VdZ[wwxYw)zL repair.py ------------- Fill holes and fix winding and normals of meshes. N)graph triangles)log)faces_to_edgestriangulate_quads)connected_components) group_rows hashable_rows)ExceptionWrapper) faces_to_pathc"|jrytj|j}d}|jj t jj}tj|D]}|j|}tt|j}tj||D]}t j|}||}t!|} t#t j$| dd} t'| dk(rY| | d} | dd| ddk(ss|dz }||dddd||d<|dkDr||_t)j*d|t'|jd zy) z Traverse and change mesh faces in-place to make sure winding is correct with edges on adjacent faces in opposite directions. Parameters ------------- mesh : Trimesh Source geometry to alter in-place. Nrraxis require_countzflipped %d/%d edges)is_winding_consistentnx from_edgelistface_adjacencyfacesviewnpndarraycopyr subgraphnextiternodes bfs_edgesravelrr sortlenrdebug) mesh graph_allflippedr componentsgstart face_pairpairedgesoverlap edge_pairs ;/DATA/.local/lib/python3.12/site-packages/trimesh/repair.py fix_windingr4!se !!  !4!45IG JJOOBJJ ' , , .E--i8   z *T!'')_%a/I +I#D"4(E Q!7qIG7|q gaj)I|A)A,q/11 &+IaL&9$B$&?il#%09:{ II#Wc$**o.AB multibodyc@|s,|jr|jdkr|jyt|j}t j t|jt}|j}|j}|jjd}|D]s}t|dkrtj||jd\}} |r| sAtj||||dj} | dksod||<u|j!rrd |j"vr(|j$j'} | |xxd zcc<nd} t j(|j||j|<| | |_yyy) z Check to see if a mesh has normals pointing "out." Parameters ------------- mesh : trimesh.Trimesh Mesh to fix in-place. multibody : bool If True will try to fix normals on every body gNdtype)r)rrT)crosses skip_inertia face_normalsg) is_watertightvolumeinvertr rrzerosr&rboolrtriangles_crossr0reshapermass_propertiesany_cacher>rfliplr) r(r6groupsfliptricross face_edges face_indexis_tightis_woundr@normalss r3 fix_inversionrSZs}    $++"3 KKM"$"5"5 6F 88C O4 0D ..C  E##G,J  z?Q  "00J1G1O1OPW1XY(X **  OU:%6T &  C<#D ', xxz T[[ ('',,.G DMT !MG99TZZ%56 4   'D  r5c4t|t||y)a Fix the winding and direction of a mesh face and face normals in-place. Really only meaningful on watertight meshes but will orient all faces and winding in a uniform way for non-watertight face patches as well. Parameters ------------- mesh : trimesh.Trimesh Mesh to fix normals on multibody : bool if True try to correct normals direction on every body rather than just one Notes -------------- mesh.faces : will flip columns on inverted faces )r6N)r4rS)r(r6s r3 fix_normalsrUs,$),r5ctj|j}t|j j Dcgc] \}}|dk7s |}}}t j|}|_|jdk7rPt j|}|jdk(s|jdk(sgd}||jj|<|Scc}}w)al Return the index of faces in the mesh which break the watertight status of the mesh. Parameters -------------- mesh : trimesh.Trimesh Mesh to check broken faces on color: (4,) uint8 or None Will set broken faces to this color if not None Returns --------------- broken : (n, ) int Indexes of mesh.faces rr)r;)r)rrrW) rrrdictdegreeitemsrarraysizeshapevisual face_colors)r(color adjacencykvbrokens r3 broken_facesres"  !4!45I !1!1!34::< G edges_faceedges_sorted_treer[meanvstack enumerate column_stackonesintziprangerr%query_ball_point concatenaterrRrGdotrI)r(rinsert_verticeser{r|rRr} tree_edgep centroidscountifantquery edge_indexoriginalcheckvalidsigns r3stitchrs, } #djj/*tU+J77A qxx=1 ! !< 7 }}HGJ&&IHHG1hqk..A.6GH H 99h 23 "&) )1 OORWWSVaZs;uqyI1Sb6STUVUWSXY Z)    OORWWSVaZs;adBAaGQqQSWU V  E#c(OS)1 crc12gJOO  A **1*6 u:? ^^E* :j12!((!BQ%9 uyy{ avvh& 99;?YYq\CF;*> ))C.CI~ JQ H  sA K0 K5 AK:;AL)F)N)NF)__doc__numpyrrr constantsrgeometryrrr groupingr r networkxr BaseExceptionE exceptionsr path.exchange.miscr r4rCrSrUrertrr5r3rs7'/(16CrB(4B(J-6:=d=@dK- ! B (,$Q'M(s.AA5A2A--A25B:B  B