Ti?8ddlZddlZddlmZmZmZddlmZm Z m Z m Z m Z ddlm Z  ddlZn##e$rZddlmZeeZYdZ[ndZ[wwxYwdZdd Zd Zdd ZGd dZ ddede e de e dedede ejf dZdZdS)N) constantsgroupingutil) ArrayLikeIntegerNDArrayNumberOptional)is_ccw)ExceptionWrapperctj}g}t|D]>\}}|jr||"||j|?|tj|fS)a Given a set of entity objects generate a networkx.Graph that represents their vertex nodes. Parameters -------------- entities : list Objects with 'closed' and 'nodes' attributes Returns ------------- graph : networkx.Graph Graph where node indexes represent vertices closed : (n,) int Indexes of entities which are 'closed' ) entity_index) nxGraph enumerateclosedappendadd_edges_fromnodesnparray)entitiesgraphrindexentitys P/DATA/AppData/hermes/venv/lib/python3.11/site-packages/trimesh/path/traversal.py vertex_graphrs" HJJE F"8,,CC v = C MM%   E B B B B "(6"" ""c4fd}||jddkrd}n6t|tjd}|dzdz }tjtjgtjtdzD]f}tjtjd|zt}|} |j | d} | gtj dd|tdkr*|dkr"d} |  Stjd} t| dd| dd} | D]d\|jj\} }| 8 |  | etjS) a Convert a path of vertex indices to a path of entity indices. Parameters ---------- vertex_path : (n,) int Ordered list of vertex indices representing a path graph : nx.Graph Vertex connectivity entities : (m,) list Entity objects vertices : (p, dimension) float Vertex points in space Returns ---------- entity_path : (q,) int Entity indices which make up vertex_path cD|d|dkrdS|d|dkrdS|d|dkrdS|d|dkrdStjdgdjjd S) a Given two edges, figure out if the first needs to be reversed to keep the progression forward. [1,0] [1,2] -1 1 [1,0] [2,1] -1 -1 [0,1] [1,2] 1 1 [0,1] [2,1] 1 -1 Parameters ------------ a : (2,) int b : (2,) int Returns ------------ a_direction : int b_direction : int rr r )r$r$)r r )r r$ )zedges not connected!zvertex path %szentity path: %szentity[a]: %s,z entity[b]: %s)NN)rlogdebugjoinpoints)abeaebr entity_path vertex_paths redge_directionz-vertex_to_entity_path..edge_directionDs( Q41Q4<<5 qTQqT\\6 qTQqT\\4 qTQqT\\5 M    # #    :r Nr rrdtyperr$) direction)shaper rr asanyarrayint64arangelenmod get_edge_datarunique_orderedreversezip end_pointsr)r/rrverticesr0 ccw_direction ccw_checkivertex_path_pos vertex_indexrr round_tripdadbr,r-r.s` ` @@@rvertex_to_entity_pathrH/s>*---------^8>!,11 8BIk;q>$J$JKLL "Q!+ - 28<< RL 2 . . . RL 2 . . .(;''K r c 2t|\}}tj|d}tj|}|D];}t|dkr|t||||<|S)a, Paths are lists of entity indices. We first generate vertex paths using graph cycle algorithms, and then convert them to entity paths. This will also change the ordering of entity.points in place so a path may be traversed without having to reverse the entity. Parameters ------------- entities : (n,) entity objects Entity objects vertices : (m, dimension) float Vertex points in space Returns ------------- entity_paths : sequence of (n,) int Ordered traversals of entities r#r) rrreshapetolistrcycles cycle_basisr8rrH)rr?rr entity_paths vertex_pathsr/s r closed_pathsrPs,!**ME6:fg..5577L9((//L$[[ {  a   1+uhPXYYZZZZ r ?cptj|}t|}|dkrtd|dkr6tj||d||}ng}t |D]^\}}||||}||dz kr||A||dd_tj|}|jddkr,t|stj |ddd}|S)a4 Turn a list of entity indices into a path of connected points. Parameters ----------- entities : (j,) entity objects Objects like 'Line', 'Arc', etc. vertices: (n, dimension) float Vertex points in space. path : (m,) int Indexes of entities scale : float Overall scale of drawing used for Number tolerances in certain cases Returns ----------- discrete : (p, dimension) float Connected points in space that lie on the path and can be connected with line segments. rzCannot discretize empty path!r )scaleNr$r) rr5r8 ValueErrordiscreterrvstackr4r ascontiguousarray) rr?pathrSpath_lenrUrB entity_idcurrents rdiscretize_pathr\s9.}X&&H4yyH1}}89991}}=$q'!2!;!;HE!;!R!RSS%dOO . .LAyy)22852IIGX\""(((( ----9X&&~aAfX&6&6'277 Or cneZdZdefdZ d dededeejfdZ de deejfd Z d S) PathSampler)ctj||_tj|jd|_t j|j|_|jtj j k}|j |_ |j |xx|j| dzcc<|j|_tj|j|_dS)Nraxisr#)rr_pointsdiff_vectorsrrow_norm_normsrtol_pathzerocopy _unit_vecrJsumlengthcumsum _cum_norm)selfr)nonzeros r__init__zPathSample.__init__sx''  1555 mDM22 + 2 77++-- w4;w#7#?#?#H#HHkoo'' 4;//r F distancesinclude_originalreturncltj|j|}tj|dt |jdz }tjd|j|}||z }|j|}|j|}|||dzz}|rtj tj|jd|} | rPtj|tj | d} tj || |j| d}|S)a\ Return points at the distances along the path requested. Parameters ---------- distances Distances along the path to sample at. include_original Include the original vertices even if they are not specified in `distance`. Useful as this will return a result with identical area and length, however indexes of `distance` will not correspond with result. Returns -------- samples : (n, dimension) Samples requested. `n==len(distances)` if not `include_original` rr r#gr`) r searchsortedrnclipr8rjrrbrJisinanyrpinsert) rorrrs positionsoffsets projectionr3origin resampled uninsertedrs rsamplezPathSample.sample s0ODNI>> GIq#dn*=*=*ABB )At~..y9( N9- i(i**<*s*?4>8<<DN8a<88 I&, , , ^x!|^4II\ RYq\\H%<=AFFFNNrRRF f F|H-6H 4<(Q,#?"JKKI M"')!444 5 5 9 9 ; ;h F   $%%%%r N)F) __name__ __module__ __qualname__rrqboolr rfloat64rr rr rr^r^s0y0000$>C00"06:0  0000dGBJ,?r r^TFr)countstep step_roundrsrtctj|tj}||td||tdt |}|@|r>||jkr |ddgSt tj|j|z }|tjd|j|}n|tj d|j|}| ||}tj j rctj|ddg|ddgz }|dtjjksJ||dtjjksJ|S) a Given a path along (n,d) points, resample them such that the distance traversed along the path is constant in between each of the resampled points. Note that this can produce clipping at corners, as the original vertices are NOT guaranteed to be in the new, resampled path. ONLY ONE of count or step can be specified Result can be uniformly distributed (np.linspace) by specifying count Result can have a specific distance (np.arange) by specifying step Parameters ---------- points: (n, d) float Points in space count : int, Number of points to sample evenly (aka np.linspace) step : float Distance each step should take along the path (aka np.arange) step_round Alter `step` to the nearest integer division of overall length. include_original Include the exact original points in the output. Returns ---------- resampled : (j,d) float Points on the path r1Nz#Only step OR count can be specifiedz&Either step or count must be specifiedrr$)rsr )rrrrTr^rlintceillinspacer7rrtolstrictrrergr) r)rrrrssamplersamplesrchecks r resample_pathr`s`JXfBJ / / /F  0>??? DLABBB  G J 7> ! !1b'? "BGGNT12233 +a77 )Aw~t44w9IJJI}7 faWo 1b'0BBCCQx),22222  8i066666 r c $t|}g}|j}|j|j}|j}t |jD]\}}t||}| |g} g} |D]c} || } | tj t| t| z| | dtj|j} || d<tjt| dz g}|j5| |tj|j| tj|j| |dj| ||fd|D|d|djdddn #1swxYwYtj|S)a, Split a Path2D into multiple Path2D objects where each one has exactly one root curve. Parameters -------------- path : trimesh.path.Path2D Input geometry Returns ------------- split : list of trimesh.path.Path2D Original geometry as separate paths split_2Dr )rr?metadatar$c g|] }| Srr).0crUs r zsplit..s @ @ @! @ @ @r )pathspolygons_closedrUrootN)typerrUrenclosure_directedrrlistkeysrrr7r8extendrideepcopyrr_cacherr?updateid_set)rXPath2Dsplitrrr root_indexr connected new_paths new_entitiesrrrnew_rootrUs @rrrsZ $ZZF E JE}H*O0%di00,&,& D+D1668899    ' 'E%LE   RYs5zz22S5F5FF G G G    & & & &=//)8S^^a/011[ & & LL!]4=+FGG!]4=99%    "I  # #&'6y'A @ @ @ @i @ @ @$     "I  # # % % %) & & & & & & & & & & & & & & &, 8E??s9B,G11G5 8G5 )N)rQ)NNTF)rinumpyrrrrtypedrrr r r r networkxr BaseExceptionE exceptionsrrrHrPr\r^rrrrrr rrs ((((((((((AAAAAAAAAAAAAA.-----  !  BBBBBB ###8kkkk\$$$N3333lccccccccP $!" @@ @ G @ 6 @ @  @  RZ @@@@FJJJJJs-A AA