jf^ddlZddlZddlmZmZddlmZddlZddl m Z m Z m Z m Z ddlmZmZddlmZmZddlmZmZmZmZmZdd l mZd d lmZmZd d lmZm Z m!Z!d Z"dZ#de#dZ$ejJdZ&de&jNd<d&dZ(dedefdZ)d'dZ*de+deejXfdZ-d(dedefdZ.d&dZ/d)dZ0dZ1d Z2d!Z3d"Z4d#e(iZ5 dd$l6m7Z7 dd%l;m?? U6 GF 4  !(S:5E)F!GF: V 8$++c,?&?@AC!iik d6*--;?F:&t,Z8* M6MM      4 #d34   M  8 II'$ 7 M  8sHF $F;c}}Scc}}w#t $ricYSwxYw)Nr )r= startswith_ns_urllstripr5r:r6)r1kvs rH_attrib_metadatarQmss  &1u( HHSM71: %&    s#A AAA A A.-A.cPt}|}t|D]Q}d|jvr-|jt |jdddd|j }|Qnt |dk(rtSt |dk(r|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 rHr2r2yswHG 9  '.. (    5gnn[6Q RSWUWSW X Y##% ?   8} X! {~~h''rSc |jjdDcgc]Z}t|dkDrJ|jjdDcgc]!}t|dkDs|j#c}\}}}g}|D]}t|dk(rt|dk7r t d|\}}t j |jddjDcgc] }t|c}}|dk(r0|jtj||d d ddf<|d k(rB|jt j|jd jgd f|dk(rLt j|d} t|dk(r|dd } nd } |jt!| | H|dk(r:tj} | d dd dfxx|zcc<|j| t#j$d|t j |t j&Scc}wcc}}wcc}w)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 )r(rz"should always have two components!, translater!Nmatrix)rr)rrr rotaterr )thetapointscalezunknown SVG transform: dtype)lowersplitrXstripr8nparrayreplacefloatr0r7copyvstackreshapeTdegreesr r r<float64) rSij componentsr[linekeyargsvaluesanglerhmats rHrVrVs ."((--A q6A:GGIOOC0?0qCFQJ0?- H t9>  Y!^AB B TT\\#s-C-I-I-KL-K58-KLM +  OOINN, -"(HRL!Q  H_ OOBIIv~~f'='?'?&KL M H_JJvay)E6{aqr  OOMUC D G^.."C BQBK6 !K OOC II/u5 6EH 88HBJJ //U @Ms#4II +I =I+I Ir$r%c(+,d+,(fd}+,(fd}+,(fd}+,(fd}Gdd}||||d}tt} tt} td ,|D]\} } | jd d } t| d k(rt j d =t | jtdz(tjtt| }t|d k(rdddd}tj|Dcgc]'}|jt|jd )c}t}tj|dd}t j"rGt%j&tj(|tj*t|sJg}|D]}||}t||d j}|dvr|j-||@t|dkDr|dk(rtj|Dcgc]}|j.j0|j.j2|j4j0|j4j2|j6j0|j6j2|j8j0|j8j2|j:g c}tj<}d}tj>|ddddfd jAdkrB|ddddf|ddddf}}t%j&|tjB|dd }|r d|d _"|j-|d |jG||jG|tI| }|D]}t|j}||vs|||\}} |jJjM|| (j-|| (j-tO| | ,(xxt| z cc<|D]\}!} } t | jtdz(|!dk(rZtQtS| dtS| dgtS| d }"tUtj*d!,(zd"}#n|!d#k(rtj| d$| d%gtj<}$tj| d&| d'gtj<\}%}&tj|$|$|%d fz|$|%|&fz|$d |&fz|$gtj<}"tWtj*t|",(z(}#n|!d)k(rtjX| d*j[j]d+d,d,tj<-j_d.}"tWtj*t|",(z(}#nh|!d/k(rtjX| d*j[j]d+d,d,tj<-j_d.}"|"d |"d0k(jar%tj*t|",(z}'n/tj*t|"dz,(z}'|'d |'d0<tW|'(}#n|!d1k(rstj| d2| d3| d4| d5gtj<j_d6}"tWtj*t|",(z(}#nt j d7|!d8F| (j-|#| (j-tO|"| ,(xxt|"z cc<t| d k(rggd9S| jcD( cic]!\}(} |(tjd| | |(d9#})}(} t|)dkDs|d:k(rd;|)i}*|*Stgti|)jk}*|*Scc}wcc}wcc} }(w).complex_to_floats6xx6:6a!&&!&&)6:"**MM:sA c}|t|jz}ttj|||jfS)Npoints)rXrrroarange)multistartendcountsrFs rH load_multiz%_svg_path_convert..load_multis@t c%,,''BIIeS12ELLAAr^c|j|jd|jg}tt j dzt |dd}||fS)Ng?rclosedFrr)rrhrrrorgetattr)svg_arcrarcrrrFs rHload_arcz#_svg_path_convert..load_arcs]!7=='--2Dgkk"RS99Q<&,.7He4   F{r^c|j|j|jg}tt j dz|fS)Nrr)rcontrolrrror) svg_quadraticrrrrFs rHload_quadraticz)_svg_path_convert..load_quadraticsL!  -"7"79J9J K RYYq\F4L896AAr^c|j|j|j|jg}t t j dz|fS)N)rcontrol1control2rrror) svg_cubicrrrrFs rH load_cubicz%_svg_path_convert..load_cubic sP! __i00)2D2Dimm T biilVD\12F::r^ceZdZdZy)$_svg_path_convert..MultiLinecjtjrtd|DsJ|Dcgc]}|j}}|j |dj t j|Dcgc]}|j|jgc}t j|_ ycc}wcc}w)Nc3JK|]}t|jdvyw)rCloseN)type__name__).0Ls rH z@_svg_path_convert..MultiLine.__init__..s!P%Q47++/@@%s!#r!rj) r strictallrr0rrorprrrxr)selflinesrrrys rH__init__z-_svg_path_convert..MultiLine.__init__szzP%PPPP',-u!agguF- MM%)-- (((f#EfQVVQVV$4f#ERZZXDK .$Fs B+/B0N)r __module__ __qualname__rr^rH MultiLiners Yr^r)rr CubicBezierQuadraticBeziercy)Nrrrr^rHz#_svg_path_convert..(sr^r#rzempty path string!rFr r)rrrrjF)min_len only_nonzerorrNr)axisgMbP?T)r1circlecxcyrrrrectxywidthheightrpolylinerrbrc)seprk)r!rpolygonr!r|x1y1x2y2)rrzunsupported SVG shape: ``)verticesentitiesScener*)6rlistgetrXr r<r:r5rorp parse_pathrrintrblocksr rr allclosehstackrr0rrrrcenterradiusrotationrxptpmeanrollrextendrQr'updaterrrrrrr-rnrqrurr=rtnextr/r)-r$r%r&rrrrrloadersrrr1rer@raw kinds_lookuprykindsrparsedbchunkr\avertsrrr entity_meta svg_entity type_nameerPkindrentityoriginwhindexrFgeomskwargsrrs- ` @@rHr9r9sYNB B; Y Y!) G4 H4 H  #Fjjb) { q II* + vzz#,/0hhtJ{345 s8q= !"Aa8 MA,**47+;+;Q?MUXYF ::==6!2BIIc#h4GH HHAFE3qt9o..G++ i./Q!5 0"' "'AGGLLGGLLEEJJEEJJHHMMHHMMHHMMHHMMJJ "' **$66%12,Q/446=!&q"1"uuQ!V}3E"]]5"''#qq2IJF&*E!HOMM%(+MM%( e$]`'f5 !JZ(11IG#)wy)*51 !!+.%%a(%%&6q&&ABt A& !g B!'ffvzz#,/0 8 "vd|$eF4L&9:E&+?rzzRDAqXX1a&&Aq6/6QF?FSjjF3v;!7&,!FGF Z ]]x &&(00c:2::gg 3v;!7&,!FGF Y ]]x &&(00c:2::gg  q VBZ',,. #f+.= #f+/2VD\A!!Hb 'F V^XXvd|VD\6$<HPRPZPZgfo 3v;!7&,!FGF II0a8 9 f%.vv>?t F # k!'n 8}B// ~~' 'GD! 299Q*+ McN. ` s ,a1 B._cross_2ds'tad{QqTAaD[((r^c  |j}t|dd}|j|j|j}}}|j r!j |d|z |d|d|zS|\}}}t|tjkD} t ||z ||z dkD} j |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) rrrrspanrformatrropi)rrinfoCrr vertex_start vertex_mid vertex_end large_flag sweep_flagrrtemp_arc temp_circles rHrz!_entities_to_str..svg_arcs #**%(%dKkk4;; e1 ::%%!qAaDAq%I I/7, j*'  j</l1J Kc Q AA!}!}  r^c|j}t|dk(ryt|dz zzj|jdS)za Use an entities discrete representation to export a curve as a polyline rrr r!)discreterXrru)rrr temp_line temp_moves rH svg_discretez&_entities_to_str..svg_discretesX ??6* x=A E Y#h-!*;<=EE   b !  r^rrF) rsrrqrrlayer __class__rrr'r0)rrrFdigits only_layers temp_digitsrrpairsrr@rGrrrrrrs @@@@@@rH_entities_to_strrs'"~ ]]_Fs6{m1%K [ gdK &&t[9I&&t[9IGOO kH)W))V)  6   E  "v||;'F     $ $ -!&/K'v.K(  DL dK() Lr^c dt|ji}tj|drg}i}|jj D]^\}} tj| ds| j ||<|jt| j| j|||`t|dkDrTt||d<nEtj|dr$t|j|j||}n td|rd jd |DStj d } g} |D]&\} } | j#d | d t%| d(d|vrt'|d}n|j(j+dz } t|j |d<dj| |j2dd|j2dd|j(d|j(d|t%|d}| j4di|S#t,$rt/j0ddYwxYw)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)rrrFrrrr+)rrrrz'drawing must be Scene or Path2D object!rcc3&K|] }|d yw)r Nr)rrPs rHrzexport_svg..ss,e!esztemplates/base.svgz stroke_widthg@r'zfailed to encodeTr( r )elementsmin_xmin_yrrr attribsr)rrr is_instance_namedr*r=r'rrrrrX_encoder8joinr get_stringr0_format_attribrrextentsmaxr6r r<boundsr)drawing return_pathrrrr%r geom_metarFgeom template_svgr"rGr@r subss rH export_svgr4:s:*W ../G gw/ !**002JD$))$9"mmIdO LL !]]!]]! +   3 y>A ,39+=G' (   2 %%%%#  BCCxx,e,,,''(<=LH"k)K=>$3G2HKL# VN34 **,u4 5%g&6&67  IIh'"1%"1%#//!$$!'* D <   & && 5 $t45sH"" IIc|jDcic]\}}|t|}}}djd|jDScc}}w)z Format attribs into the trimesh namespace. Parameters ----------- attrib : dict Bag of keys and values. r!c3|K|]4\}}t|dkDr!|t|dkDrtd|d|d6yw)rN:z="")rX_ns_name)rrOrPs rHrz!_format_attrib..sHDAq q6A:!-CFQJ *AaS1#Qs:<)r=r'r()r1rOrPrEs rHr*r*sS&,\\^ 4^TQ1gaj=^C 4 99IIK  5sAc ~t|trd|vr|Stjt |j Dcic]\}}|j dr||c}}djd}dtj|z}tjrt|t||Scc}}w)z Wangle things into a string. Parameters ----------- stuff : dict, str Thing to pack Returns ------------ encoded : str Packaged into url-safe b64 string r8r])rbr7) separatorsutf-8base64,) isinstancestrbase64urlsafe_b64encoderr=rLencoder decode_textr r _deep_samer:)stuffrOrPpackrDs rHr'r's%#U"2  # ##kkm Emda1<<3DQTm E!  &/  D ))$/ /F zz5'&/* M Fs B9B9ct|tjr"t|ttjfs;Jt|trt|tsJt|t |sJt|tt fr||k(sJyt|tttjfrtj||sJyt|tr:t|t|k(sJt||D]\}}t||yt|tsJt|jt|jk(sJ|jD]}t||||y)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)r>rondarrayrr?rbytesrrrrrXziprDdictsetkeys)originalotherrrrOs rHrDrDsD&(BJJ'%$ !3444 Hc "%%%%(DK000(S%L)5    Huc2::6 7{{8U+++ Hd #8}E ***%(DAq q! ) h %% % x}} 3uzz|#4 44 4 ]]_8A;a)r^c|ytj|}|jdrItjt j |ddjdjdS|S)z Decode a base64 bag of stuff. Parameters ------------ bag : str Starts with `base64,` Returns ------------- loaded : dict Loaded bag of stuff Nr=r<) r rCrLjsonloadsr@urlsafe_b64decoderBdecode)rEtexts rHr:r:sh {   C D y!zz  $ $T!"X__W%= > E Eg N   Kr^svg)r)r,)NNN) )N)FNN)=r@rR collectionsrrrsrnumpyrorrrr r constantsr r transformationsr rtypedrrrrrrrrrrrrrr9rMr5eyer7flagsrIrQr2r?rxrVr9rr4r*r'rDr: _svg_loaderssvg.pathrr6EExceptionWrapperlxmlr,rr^rHres^ *55!>==+((  - 7)2 BFF1I $  Qh W  (@@0S@0WRZZ-@@0FsXsxslePY'x"<3*l0{# %# %-,,Q/J$L %  (J ' ' *EL  s0CDD"C>>DD, D''D,