TifJddlZddlZddlmZmZddlmZddlZddl m Z m Z m Z m Z ddlmZmZddlmZmZddlmZmZmZmZmZdd l mZd d lmZmZd d lmZm Z m!Z!d Z"dZ#de#dZ$ej%dZ&de&j'd<d'dZ(dedefdZ)d(dZ*de+deej,fdZ-d)dedefdZ.d'dZ/d*dZ0d Z1d!Z2d"Z3d#Z4d$e(iZ5 dd%l6m7Z7n'#e8$rZ9e j:e9Z7e7e5d$<YdZ9[9ndZ9[9wwxYw dd&l;m??? U6 G G GF 4  !(S:5E)F!G!GF:       444 #d3333334 V 8$+c,?&?@AAC!iikk @ @ d6*---;?F:&t,Z8 @    D 8 8 8 I'$ 7 7 7 7 7 7 8 MsCC** C76C7='E%% F1FFAG33 H!?H! H!r1returncd d|DS#t$ricYSwxYw)Nci|]O\}}|ddt'|tt |PSr N) startswith_ns_urllstripr5r:.0kvs rH z$_attrib_metadata..psZ   1u(( HHSMM71::   )r=r6r1s rH_attrib_metadatarXmsV        s  // ct}|}t|D]Y}d|jvr6|t |jdddd|}|nZt |dkrtSt |dkr|dStj |S)z Find a transformation matrix for an XML element. Parameters -------------- e : lxml.etree.Element Element to search upwards from. max_depth : int Maximum depth to search for transforms. transformNr!rr ) rranger1 extendlefttransform_to_matrices getparentlenr7r multi_dot)rB max_depthmatricescurrent_s rHr2r2yswwHG 9   '. ( (    5gn[6Q R RSWSWUWSW X Y Y Y##%% ? E  8}} X!  {~h'''rVr[c`d|dD}g}|D]}t|dkrt|dkrtd|\}}t jd|ddD}|d kr<|t ||d d ddf<|d krE|t j | d j gdf|dkr`t j |d}t|dkr |dd }nd }|t||u|dkrFt } | d dd dfxx|zcc<|| tjd|t j|tjS)a Convert an SVG transform string to an array of matrices. i.e. "rotate(-10 50 100) translate(-36 45.5) skewX(40) scale(1 0.5)" Parameters ----------- transform : str Contains transformation information in SVG form Returns ----------- matrices : (n, 3, 3) float Multiple transformation matrices from input transform string cg|]F}t|dkd|dDGS)rc\g|])}t|dk|*Sr)r`strip)rRjs rH z4transform_to_matrices...s+???qCFFQJJJJJrV()r`rjsplitrRis rHrlz)transform_to_matrices..sR q66A:: @?AGGIIOOC00??? ::rV)rrz"should always have two components!c,g|]}t|S)floatros rHrlz)transform_to_matrices..sLLL588LLLrV, translater!Nmatrix)rr)rrr rotaterr )thetapointscalezunknown SVG transform: dtype)lowerrnr`r8nparrayreplacer0r7copyvstackreshapeTdegreesr r r<float64) r[ componentsrclinekeyargsvaluesangler{mats rHr^r^s(*""((--J H"7"7 t99>>  YY!^^ABB B TLLT\\#s-C-C-I-I-K-KLLLMM +   OOINN,, - - -"(HRL!Q   H__ OOBIv~~f'='='?&KLL M M M M H__Jvay))E6{{aqrr  OOMUCCC D D D D G^^..""C BQBKKK6 !KKK OOC I555 6 6 6 6 8HBJ / / //rVr$r%c 6&'()*d&'*fd}&'*fd}&'*fd}&'*fd}Gdd}||||d}tt(tt} td '|D]s\} } | d d } t| d krt jd Dt | tdz*tj tt| } t| d krdddd)tj )fd| Dt}tj |dd}tjrHt!jtj|tjt| sJg}|D]i}| |}t)| |d j}|dvr|||Nt|dkr|dkrtj d|Dtj}d}tj|ddddfd dkrF|ddddf|ddddf}}t!j|tj|dd }|r*d|d _||d =||T||kt;| }|D]}t)|j}||vr|||\}}|j|(*|| *tA|| '*xxt|z cc<u|D]*\}} } t | tdz*|dkrytCtE| d tE| d!gtE| d"}tGtjd#'*zd$}n|d%krtj | d&| d'gtj} tj | d(| d)gtj\}!}"tj | | |!d fz| |!|"fz| d |"fz| gtj}tItjt|'*z*}nF|d+krtj%| d,&'d-d.d.tj/(d0}tItjt|'*z*}n|d1krtj%| d,&'d-d.d.tj/(d0}|d |d2k)r+tjt|'*z}#n8tjt|dz'*z}#|#d |#d2<tI|#*}n|d3krtj | d4| d5| d6| d7gtj(d8}tItjt|'*z*}nt jd9|d:ɉ(*|| *tA|| '*xxt|z cc<,t| d krggd;S(fd<| *D}$t|$dks|d=krd>|$i}%n.tWtY|$-}%|%S)?a Convert an SVG path string into a Path2D object Parameters ------------- paths: list of tuples Containing (path string, (3, 3) matrix, metadata) Returns ------------- drawing : dict Kwargs for Path2D constructor cVtjd|DtjS)Nc*g|]}|j|jgSrsrealimagros rHrlz?_svg_path_convert..complex_to_float..s!:::a!&!&):::rVr})rrr)rs rHcomplex_to_floatz+_svg_path_convert..complex_to_floats(x::6:::"*MMMMrVc}|t|jz}ttj|||jfS)Npoints)r`rrrarange)multistartendcountsrFs rH load_multiz%_svg_path_convert..load_multisFt c%,'''BIeS11222ELAArVc|j|d|jg}tt jdzt |dd}||fS)Ng?rclosedFrr)rr{rrrrgetattr)svg_arcrarcrrrFs rHload_arcz#_svg_path_convert..load_arcso!!7='--2D2Dgk"RSS9Q<<&,.7He44     F{rVc|j|j|jg}tt jdz|fS)Nrr)rcontrolrrrr) svg_quadraticrrrrFs rHload_quadraticz)_svg_path_convert..load_quadraticsQ!!  -"79J K  RYq\\F4L89996AArVc|j|j|j|jg}t t jdz|fS)N)rcontrol1control2rrrr) svg_cubicrrrrFs rH load_cubicz%_svg_path_convert..load_cubic sQ!! _i0)2Dim T  billVD\122F::rVceZdZdZdS)$_svg_path_convert..MultiLinec tjrtd|DsJd|D}||djt jd|Dt j|_dS)Nc3BK|]}t|jdvVdS)rCloseN)type__name__rRLs rH z@_svg_path_convert..MultiLine.__init__..s1PPQ477+/@@PPPPPPrVcg|] }|j Srs)rrs rHrlzA_svg_path_convert..MultiLine.__init__..s---!ag---rVr!c*g|]}|j|jgSrsrros rHrlzA_svg_path_convert..MultiLine.__init__..s!#E#E#EQVQV$4#E#E#ErVr}) r strictallr0rrrrr)selflinesrs rH__init__z-_svg_path_convert..MultiLine.__init__sz QPP%PPPPPPPP--u---F MM%)- ( ( ((#E#Ef#E#E#ERZXXXDKKKrVN)r __module__ __qualname__rrsrVrH MultiLiners( Y Y Y Y YrVr)rr CubicBezierQuadraticBeziercdS)NrrsrsrVrHz#_svg_path_convert..(srVr#rzempty path string!rFr r)rrrc`g|]*}t|jd+Sri)getrr)rRrp kinds_lookups rHrlz%_svg_path_convert..>s2MMMA,**477+;Q??MMMrVr}F)min_len only_nonzerorrc g|]b}|jj|jj|jj|jj|jj|jj|jj|jj|jg cSrs)rrrrcenterradiusrotation)rRas rHrlz%_svg_path_convert..Usi   GLGLEJEJHMHMHMHMJ    rVNr)axisgMbP?TrWcirclecxcyrrrrectxywidthheightrpolylinerrurv)sepr~)r!rpolygonr!rx1y1x2y2)rrzunsupported SVG shape: ``verticesentitiescRi|]#\}}|tj||d$S)r)rr)rRrFrTrs rHrUz%_svg_path_convert..sC    D! 29Q<?rzRRRDAqX1a&&Aq6/6QF?FSjF3v;;!7!7&,!FGGGFF Z  ]x &&((00c::2:gg 3v;;!7!7&,!FGGGFF Y  ]x &&((00c::2:gg  q VBZ',,.. % #f++..= #f++/22VD\A!!Hb '''FF V^^Xvd|VD\6$<HPRPZgfoo 3v;;!7!7&,!FGGGFF I8888 9 9 9 f%%%.vv>>???t F #  8}}B///    ~~''   E 5zzA~~'))e$d5<<>>**++ MrVc@ |d}| dt|d}dd|dd|dd|d d|d td td tfd  fd} fd}g}|D]g} | | j|vr| jjdkr || } n || } t| j } ||| d<| | | fh|S)a Convert the entities of a path to path strings. Parameters ------------ entities : (n,) list Entity objects vertices : (m, 2) float Vertices entities reference name : any Trimesh namespace name to assign to entity digits : int Number of digits to format exports into only_layers : set Only export these layers if passed N z0.fzKM {x:DI},{y:DI}a{r:DI},{r:DI},0,1,0,{d:DI},0a{r:DI},{r:DI},0,1,0,-{d:DI},0ZDIz M{:DI},{:DI}z L{:DI},{:DI}z6M{SX:DI} {SY:DI}A{R},{R} 0 {L:d},{S:d} {EX:DI},{EY:DI}rrrJcH|d|dz|d|dzz S)zD Numpy 2.0 depreciated cross products of 2D arrays. rr rs)rrs rH _cross_2dz#_entities_to_str.._cross_2ds'tad{QqTAaD[((rVc  |j}t|dd}|j|j|j}}}|jr+|d|z |d|d|zS|\}}}t|tj k} t ||z ||z dk} |d|d| | |d|d| S) z arc string: (rx ry x-axis-rotation large-arc-flag sweep-flag x y)+ large-arc-flag: greater than 180 degrees sweep flag: direction (cw/ccw) FT) return_normal return_anglerr g@)rrrr#g)SXSYrSEXEYR) rrrrspanrformatrrpi)rrinfoCrr vertex_start vertex_mid vertex_end large_flag sweep_flagrrtemp_arc temp_circles rHrz!_entities_to_str..svg_arcs #*%(%dKKKk4; e1 : J%%!qAaDAq%II I/7, j*''  Ij</l1J K Kc Q  AA!}!}   rVc|}t|dkrdSt|dz zzj|dS)za Use an entities discrete representation to export a curve as a polyline rrr r!)discreter`rr)rr'r temp_line temp_moves rH svg_discretez&_entities_to_str..svg_discretesd ??6** x==A  2E Y#h--!*;<=E   b ! !  rVrrF) rrrrrlayer __class__rrr'r0)rrrFdigits only_layers temp_digitsrr*pairsrr@rGrrr$r%r(r)s @@@@@@rH_entities_to_strr1s"~ ]]__F&s6{{%%%K [ gdK  &&t[99I&&t[99IGOO kH)W))V))))         6         E * *  "v|;'F'F    $ - -!'&//KK',v..K((  DL dK()))) LrVc dt|ji}tj|drg}i}|jD]V\}} tj| ds| j||<|t| j | j |||Wt|dkrt||d<nBtj|drt|j |j ||}ntd|rd d |DStjd } g} |D].\} } | d | d t%| d/d|vrt'|d}n|jdz } t|j|d<n&#t,$rt/jddYnwxYwd| |jdd|jdd|jd|jd|t%|d}| jdi|S)a Export a Path2D object into an SVG file. Parameters ----------- drawing : Path2D Source geometry return_path : bool If True return only path string not wrapped in XML only_layers : None or set If passed only export the specified layers digits : None or int Number of digits for floating point values Returns ----------- as_svg : str XML formatted SVG, or path string r"rPath2D)rrrFr-r.rr+)rrr-r.z'drawing must be Scene or Path2D object!rvc3&K|] }|dV dSrMrs)rRrTs rHrzexport_svg..ss&,,!,,,,,,rVztemplates/base.svgz stroke_widthg@r'zfailed to encodeTr( r )elementsmin_xmin_yrrr5attribsrs)rrr is_instance_namedr*r=r'rr1rrr`_encoder8joinr get_stringr0_format_attribrtextentsmaxr6r r<boundsr)drawing return_pathr.r-r r:r0 geom_metarFgeom template_svgr7rGr@r5subss rH export_svgrI:s*W ./G gw//D !*0022  JD$)$99 "mIdO LL !]!]! +      y>>A  ,39+=+=G' (  2 2 D %%#    BCCC-xx,,e,,,,,,'(<==LH"MMkKKKK>$3G3GKKKLLLLVN344 **,,u4 5%g&677  555 $t4444445 IIh''"1%"1%#/!$$!'**  D <  & & & &&s'F?? G"!G"cd|D}dd|DS)z Format attribs into the trimesh namespace. Parameters ----------- attrib : dict Bag of keys and values. c4i|]\}}|t|Srs)r<rQs rHrUz"_format_attrib..s$ 4 4 4TQ1gajj 4 4 4rVr6c3K|]?\}}t|dk|t|dk-td|d|dV@dS)rN:z="")r`_ns_namerQs rHrz!_format_attrib..si Aq q66A::!-CFFQJJ a1,6JJJrV)r=r=)r1rEs rHr?r?sY 5 4V\\^^ 4 4 4C 99IIKK  rVc`t|trd|vr|Stjt d|Ddd}dtj|z}tj rt|t||S)z Wangle things into a string. Parameters ----------- stuff : dict, str Thing to pack Returns ------------ encoded : str Packaged into url-safe b64 string rNcDi|]\}}|d||S)re)rNrQs rHrUz_encode..s/ E E Eda1<<3D3D EQ E E ErV)rurM) separatorsutf-8base64,) isinstancestrbase64urlsafe_b64encoderr=encoder decode_textr r _deep_samer:)stuffpackrDs rHr<r<s%#U"2"2  # E Eekkmm E E E!    &//   D )$// /F z+5'&//*** MrVct|tjr$t|ttjfsJnLt|trt|tsJnt|t |sJt|tt fr ||ksJdSt|tttjfrtj ||sJdSt|trJt|t|ksJt||D]\}}t||dSt|tsJt|t|ksJ|D]}t||||dS)aR Do a recursive comparison of two items to check our encoding scheme in unit tests. Parameters ----------- original : str, bytes, list, dict Original item other : str, bytes, list, dict Item that should be identical Raises ------------ AssertionError If items are not the same. N)rUrndarrayrrVrbytesrtrrr`zipr[dictsetkeys)originalotherrrrSs rHr[r[s&(BJ''1%$ !3444444 Hc " "1%%%%%%%(DKK00000(S%L))5     Huc2:6 7 7 {8U+++++ Hd # #8}}E ****%((  DAq q!     h % %%% % x}}  3uzz||#4#4 4 4 4 4 ]]__**8A;a))))**rVc|dStj|}|drTtjt j|ddddS|S)z Decode a base64 bag of stuff. Parameters ------------ bag : str Starts with `base64,` Returns ------------- loaded : dict Loaded bag of stuff NrTrS) r rZrNjsonloadsrWurlsafe_b64decoderYdecode)rEtexts rHr:r:s} {  C D y!! z  $T!""X__W%=%= > > E Eg N N    KrVsvg)r)r,)NNN)rY)N)FNN)=rWri collectionsrrrrnumpyrrrrr r constantsr r transformationsr rtypedrrrrrrrrrrrrrrOrOr5eyer7flagsrIrXr2rVrr^r9r1rIr?r<r[r: _svg_loaderssvg.pathrr6EExceptionWrapperlxmlr,rsrVrHr{sF ********555555555555!!!!!!!!>>>>>>>>==============++++++++((((((((((  -7 BF1II $  QQQQh W     ((((@@0S@0WRZ-@@0@0@0@0FssXsxssssleeeePY'Y'Y'Y'x"<3*3*3*l0{# %#######%%%-,Q//J$L %      (J ' * *EL  s0C C1C,,C15C==D"DD"