Ti!0<dZddlZddlmZmZddlmZddlm Z m Z ddlm Z ddl m Z mZ ddlZn##e$rZdd lmZeeZYdZ[ndZ[wwxYw dd lmZn##e$rZdd lmZeeZYdZ[ndZ[wwxYwd Zdd efdZddZddZddefdZddZdS)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||jrdStj|j}d}|jt j}tj |D]}| |}tt| }tj||D]}t j|}||}t!|} t#t j| dd} t'| dkrf| | d} | dd| ddkr%|dz }||dddd||d<|dkr||_t)jd|t'|jd zdS) 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 H/DATA/AppData/hermes/venv/lib/python3.11/site-packages/trimesh/repair.py fix_windingr4!s ! !455IG JOOBJ ' ' , , . .E-i88@@   z * *T!''))__%%a// @ @I ++I#D"4((E Q!7!7!7qIIIG7||q  gaj)I|A)A,q/111 &+IaL&9$$B$&?il#% @({{ I#Wc$*oo.ABBBBBF multibodyc(|s(|jr|jdkr|dSt|j}t jt|jt}|j }|j }|j d}|D]~}t|dkrtj|| d\}} |r| sKtj||||dj} | dkrd||<|rgd |jvr*|j} | |xxd zcc<nd} t j|j||j|<| | |_dSdSdS) 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"3 KKMMM"$"5 6 6F 8C OO4 0 0 0D .C  E##G,,J$$  z??Q   "0J1G1O1OPW1X1XYY( X  *  OU:%6T     C<<#D  xxzz ( T[ ( (',,..G DMMMT !MMMMG9TZ%566 4   'D    ( (  r5cFt|t||dS)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+,$),,,,,,r5cntj|j}dt|D}t j|}|H|jdkr=t j|}|j dks|j dksgd}||j j |<|S)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 c$g|] \}}|dk |Sr).0kvs r3 z broken_faces..s! G G GDAqQar5Nr)r;rX)rrr^) rrrdictdegreeitemsrarraysizeshapevisual face_colors)r(color adjacencybrokens r3 broken_facesrjs" !455I G GD!1!1!3!344::<< G G GF Xf  F V[A-- t##u{d':':$$$E*/ ' Mr5use_fanct|jdkrdS|jrdSt|jd}t|dkrdS|j|}t jt j|}t||}t|dkrdSt|}t|}t|}tj ||dd } tj|| || <|||jS) a Fill boundary holes in-place using fans, which may result in bad answers if the holes are non convex! Face colors and attributes will be padded with default values so shapes match. Parameters ---------- mesh : trimesh.Trimesh Mesh will be repaired in-place. use_fan If passed, holes larger than quads will be triangulated using fans which are only valid for non-convex holes. rFTrr)rkr)rrr)r&rr?r edges_sortedr0r cycle_basisrrrr risinrErGrI extend_faces) r(rkboundary_groupsboundaryholes new_faces new_edges hashable_new hashable_old needs_reverses r3 fill_holesrysG  4:u t!!2!DDDO ?auz/*H N2+H55 6 6E"%999I 9~~uy))I !++L **LGL,77??HHLLRSLTTM "y=)ABBIm i    r5cZ|&tjt|j}dt ||dD}|j|j}|j}|j}|ratj fd|D}ttj |ffdt|D}n d|D}ttt||D]\} } | ddddf} | d || d } t| d krgtj| } ||| }t%j| dd \}}|s|d }tj||}|d krtj| || <tj |}|r||fS|S)aC Create a fan stitch over the boundary of the specified faces. If the boundary is non-convex a triangle fan is going to be extremely wonky. Parameters ----------- mesh : trimesh.Trimesh Mesh to create fan stitch on. faces : (n,) int Face indexes to stitch with triangle fans. insert_vertices : bool Allow stitching to insert new vertices? Returns ---------- fan : (m, 3) int New triangles referencing mesh.vertices. vertices : (p, 3) float Inserted vertices (only returned `if insert_vertices`) Ncg|]=}t|jdk|jd|jdk6|j>S)rrr)r&points)rZes r3r]zstitch..3sM qx==1  ! !.Cs.GGG1hqk..A.66GGGr5c g|][\}}tjtjt|dz t|zz|dd|ddf\S)rr8Nrr column_stackonesr&int)rZircounts r3r]zstitch..Jsu   1 ORWSVVaZs;;;uqyI1SbS6STUVUWUWSXY Z Z   r5c g|][}tjtjt|dz t|dz|dd|ddf\S)rr8rrrrr)rZrs r3r]zstitch..Qsl    ORWSVVaZs;;;adBAadGQqQStWU V V   r5 rrg|=)rrr)raranger&rr rr> edges_faceedges_sorted_treerbvstack enumerateziprangerr%query_ball_point concatenaterrRrGdotrrI)r(rinsert_verticesr|rRr tree_edge centroidsfanrtr}query edge_indexoriginalcheckvalidsignrrs @@r3stitchrsM, } #dj//** tU++J7F}HGJ&I HGGGGGGGHH H 9h 233    !&))        E#c((OOS))""1 crc122gJOO   A **1*66 u::?? ^E** :j12!(!BQB%99 uyy{{  avh&& 99;;??Yq\\CF )C..CI~ Jr5)F)N)NF)__doc__numpyrrr constantsrgeometryrrr groupingr r networkxr BaseExceptionE exceptionsr path.exchange.miscr r4rCrSrUrjryrrYr5r3rs77777777''''''////////-,,,,,  !  BBBBBB (1111111(((,,,,,,$$Q''MMMMMM( 6C6C6CrB(B(4B(B(B(B(J----6:==d====@dddddds,1AA  AAA<!A77A<