o
?}hA @ s d dl mZ ddlT d dlZd dlZd dlmZmZ d dlm Z ddl
m Z ddl m
Z
mZmZmZ d dlZd dlZd dlZd dlZed Zd
d ZG d d
d
ZG dd deZG dd deZG dd deZ G dd deZ G dd deZ G dd de Z G dd deZ dS ) ) annotations )*N)ListUnion)minidom)SVGPath)SVGCommandLine
SVGCommandArcSVGCommandBezierSVGCommandClosez([-+]?[0-9]*\.?[0-9]+(?:[eE][-+]?[0-9]+)?c C s t ttt| S N)listmapfloatFLOAT_REfindallargs r :/data/cref/OmniSVG-kaiyuan/deepsvg/svglib/svg_primitive.py extract_args r c @ sl e Zd ZdZd ddZd d
Zed d
dZedfddZ d ddZ
dd Z dd Z dd Z
d dd ZdS )! SVGPrimitivez[
Reference: https://developer.mozilla.org/en-US/docs/Web/SVG/Tutorial/Basic_Shapes
blackFN.3 ?c C s" || _ || _|| _|| _|| _d S r
)color dasharray stroke_widthopacityfill)selfr r! r r r r r r __init__ s
zSVGPrimitive.__init__c C s` | j rd| j d| j dnd| j d| j d| j d}| jd ur.| j s.|d| j d7 }|S )Nzfill="z" fill-opacity=""zfill="none" stroke="z" stroke-width="z" stroke-opacity="z stroke-dasharray=")r! r r r r )r" fill_attrr r r _get_fill_attr" s :zSVGPrimitive._get_fill_attrxminidom.Elementc C t r
NotImplementedError)clsr' r r r from_xml( s zSVGPrimitive.from_xml c O s& ddl m} || g|dj|i |S )Nr )SVG)viewbox)svgr/ draw)r" r0 r kwargsr/ r r r r2 , s zSVGPrimitive.drawTc C s g S r
r )r" with_points with_handles with_bboxescolor_firstlast
with_movesr r r _get_viz_elements0 z SVGPrimitive._get_viz_elementsc C r) r
r* r" r r r to_path3 r: zSVGPrimitive.to_pathc C r) r
r* r; r r r copy6 r: zSVGPrimitive.copyc C r) r
r* r; r r r bbox9 r: zSVGPrimitive.bboxc C s
|| _ | S r
r! )r" r! r r r fill_< s zSVGPrimitive.fill_)r FNr r r' r( FFFTT)T)__name__
__module__ __qualname____doc__r# r& classmethodr- Bboxr2 r9 r< r= r> r@ r r r r r s
r c D e Zd Zd fdd Zdd Zd d
Zedd
dZdd Z Z S )
SVGEllipsecenterPointradiusRadiusc " t j|i | || _|| _d S r
)superr# rK rM )r" rK rM r r3 __class__r r r# B
zSVGEllipse.__init__c C d| j d| j dS )Nz
SVGEllipse(c= r=)rK rM r; r r r __repr__H zSVGEllipse.__repr__c O < | }d| d| jj d| jj d| jj d| jj d S )Nz r& rK r' yrM r" r r3 r% r r r to_strK 4zSVGEllipse.to_strr' r( c C sf | d p
|ddk }tt|dt|d}tt|dt|d}t|||dS )Nr! nonecxcyrxryr? ) hasAttribute getAttributerL r rN rJ _r' r! rK rM r r r r- O s zSVGEllipse.from_xmlc
C s | j | j | j | j }}| j | j | j | j }}t|| jtdtdtd|t|| jtdtdtd|t|| jtdtdtd|t|| jtdtdtd|g}t|ddj| j dS )N r Tclosedr? )
rK rM xprojyprojr
AngleFlagr to_groupr! r" p0p1p2p3commandsr r r r< W s "" zSVGEllipse.to_path)rK rL rM rN rA
rC rD rE r# rX ra rG r- r<
__classcell__r r rQ r rJ A rJ c s: e Zd Z fddZdd Zdd Zed d d
Z ZS ) SVGCirclec t j|i | d S r
rP r# r" r r3 rQ r r r# d rY zSVGCircle.__init__c C rT )Nz SVGCircle(c=rU rV rW r; r r r rX g rY zSVGCircle.__repr__c O s2 | }d| d| jj d| jj d| jj d S )Nz c S g | ]}| qS r ra .0pr r r
z&SVGPolyline.to_str..r& formatjoinr r` r r r ra zSVGPolyline.to_strr' r( c sz | d p
|ddk }t|d t d dks'J dt d fdd t d d d D }| ||d
S ) Nr! rc r r z3Expected even number of arguments for SVGPolyline: z givenc s& g | ]\}}t | d | d qS )r r )rL )r ir' r r r r & z(SVGPolyline.from_xml..r? )rh ri r len enumerate)r, r' r! r r r r r- s
$ zSVGPolyline.from_xmlc C sJ dd t | jd d | jdd D }| jjdk}t||dj| jdS )Nc S s g | ] \}}t ||qS r )r )r rv rw r r r r s z'SVGPolyline.to_path..r
SVGPolygonrm r? )zipr rR rC r rs r! )r" ry is_closedr r r r< s ( zSVGPolyline.to_path)r r rA rz r r rQ r r s r c s, e Zd Z fddZdd Zdd Z ZS )r c r~ r
r r rQ r r r# rY zSVGPolygon.__init__c C r )NzSVGPolygon(points=rV r r; r r r rX r zSVGPolygon.__repr__c O r )Nzr c S r r r r r r r r r z%SVGPolygon.to_str..r r` r r r ra r zSVGPolygon.to_str)rC rD rE r# rX ra r{ r r rQ r r s r c s e Zd Zdfdg fdd
Zedd Zedd Zd
d Zd d
Zdd Z edd Z
edd Z dhddZ diddZ
d d Zd d Zdjd"d#Zd$d% Zd&d' Z dkd(d)Zdld+d,Zd-d. Zd/d0 Zdmd3d4Zd5d6 Zdnd8d9Zd:d; Zdd? Z d@dA Z dBdC Z dDdE Z dFdG Z dHdI Z!dodLdMZ"dNdO Z#dpdPdQZ$dRdS Z%dTdU Z&dVdW Z'dXdY Z(dZd[ Z)d\d] Z*dqd_d`Z+drdbdcZ,ddde Z- Z.S )s SVGPathGroupN svg_paths
List[SVGPath]c s2 t j|i | || _|d u rtd}|| _d S )Nrl )rP r# r rL origin)r" r r r r3 rQ r r r# s
zSVGPathGroup.__init__c C s | j S r
r r; r r r paths s zSVGPathGroup.pathsc C s | j sd S | j d S Nr r r; r r r path s
zSVGPathGroup.pathc C s
| j | S r
r )r" idxr r r __getitem__
zSVGPathGroup.__getitem__c C s
t | jS r
)r r r; r r r __len__ r zSVGPathGroup.__len__c C t dd | jD S )Nc S s g | ]}t |qS r )r r r r r r r r z*SVGPathGroup.total_len..)sumr r; r r r total_len r zSVGPathGroup.total_lenc C s | j d jS r )r r r; r r r r s zSVGPathGroup.start_posc C s | j d }|jr |jS |jS Nr )r rn r r )r" last_pathr r r r s
zSVGPathGroup.end_posr rL c C s$ || _ | jr || jd _ | d S r )r r recompute_origins)r" r r r r
set_origin s zSVGPathGroup.set_originr r c C s | j | d S r
)r append)r" r r r r r s zSVGPathGroup.appendc C s0 t dd | jD | j | j| j| j| j| jS )Nc S r r )r= r svg_pathr r r r r z%SVGPathGroup.copy..) r r r r= r r! r r r r; r r r r= s zSVGPathGroup.copyc C s d ddd | jD S )NzSVGPathGroup({})z, c s | ]}| V qd S r
)rX r r r r z(SVGPathGroup.__repr__..)r r r r; r r r rX s zSVGPathGroup.__repr__FTc
C s> g }| j D ]}||||||| q|r || |S r
)r extendr9 r
_get_bbox_viz)r" r4 r5 r6 r7 r8 viz_elementsr r r r r9 s
z SVGPathGroup._get_viz_elementsc C s( | j dkrdn| j }| j|d}|S )Nr red)r )r r> to_rectangle)r" r r> r r r r s zSVGPathGroup._get_bbox_vizc C s | S r
r r; r r r r< % r: zSVGPathGroup.to_pathc O s | d u st | dr| jd u st | dr| jd u rdS | jr'tdd | jD r)dS z | }|r2dnd}d||| jjd d
d | jD W S tt fyS Y dS w ) Nr r zGc s s | ]}|d u V qd S r
r r r r r r 5 r z&SVGPathGroup.to_str..zmarker-start="url(#arrow)" z'r c s r r
r r r r r r ? r )
hasattrr r anyr& r fillingr AttributeError TypeError)r" with_markersr r3 r% Z marker_attrr r r ra / s 0 zSVGPathGroup.to_strr c s t j fdd| jD ddS )Nc s g | ]}|j d qS )PAD_VAL) to_tensorr r r r r F s z*SVGPathGroup.to_tensor..r )dim)torchcatr )r" r r r r r E r zSVGPathGroup.to_tensorc O s$ | j D ] }t|||i | q| S r
)r getattr)r" methodr r3 r r r r _apply_to_pathsH s
z SVGPathGroup._apply_to_pathsc C | d|S )N translater )r" vecr r r r M zSVGPathGroup.translateanglerq c C r )Nrotater )r" r r r r r P r zSVGPathGroup.rotatec C r )Nscaler )r" factorr r r r S r zSVGPathGroup.scale c C r )N numericalizer )r" nr r r r V r zSVGPathGroup.numericalizec C s | ddS )N
set_closedFr r; r r r drop_zY r zSVGPathGroup.drop_zc C s& | j }| jD ]
}| |_ |j}q| S r
)r r r= r )r" r r r r r r \ s
z SVGPathGroup.recompute_originsc C | d | | S )Nreorderr r r; r r r r c s
zSVGPathGroup.reorderc C s dd | j D | _ | S )Nc S s g | ]}|j r|qS r )
path_commandsr r r r r i s z-SVGPathGroup.filter_empty..r r; r r r filter_emptyh s zSVGPathGroup.filter_emptyc C s8 t | jdd d| _| jd s| d | | S )Nc S s | j d d d S r )r tolist)r' r r r m r z+SVGPathGroup.canonicalize..)keyr reverse)sortedr is_clockwiser r r; r r r canonicalizel s
zSVGPathGroup.canonicalizec C r )Nr r r; r r r r t
zSVGPathGroup.reversec C s | d | S )Nduplicate_extremitiesr r; r r r r z s
z"SVGPathGroup.duplicate_extremitiesc C r )Nreverse_non_closedr r; r r r r ~ r z SVGPathGroup.reverse_non_closed皙? `f@c C s | j d||||d | | S )Nsimplify) toleranceepsilonangle_threshold force_smoothr )r" r r r r r r r r s
zSVGPathGroup.simplifyc s fdd j D S )Nc
s. g | ]}t |g j j j j j jqS r )r r r r! r r r r r; r r r s
z,SVGPathGroup.split_paths..r r; r r; r split_paths s
zSVGPathGroup.split_pathsc C s | j d|||dS )Nsplit)r max_dist
include_linesr )r" r r r r r r r zSVGPathGroup.splitc C
| dS )N
simplify_arcsr r; r r r r r zSVGPathGroup.simplify_arcsc C r )Nfilter_consecutivesr r; r r r r r z SVGPathGroup.filter_consecutivesc C r )Nfilter_duplicatesr r; r r r r r z SVGPathGroup.filter_duplicatesc C r )Nc S r r )r> r r r r r r z%SVGPathGroup.bbox..)
union_bboxr r; r r r r> r zSVGPathGroup.bboxc C s t jdd | jD S )Nc S r r
to_shapelyr r r r r r z+SVGPathGroup.to_shapely..)shapelyops unary_unionr r; r r r r s zSVGPathGroup.to_shapelyc
s | j r{| dd D }|D ]h}| j| jsqd|fg}|rzt }t }|D ]:\}}| j| |dk |D ]&}||vrc|| j| | j| kd d } | | | | |f q=q*
dd |D fdd|D }|s"q| S )Nc S s g | ]
\}}|d kr|qS r r )r r dr r r r s z0SVGPathGroup.compute_filling..r r r c S s g | ]\}}|qS r r r r r r r r r r c s& g | ]\}} |d kr||fqS r ) in_degreer Gr r r r ) r!
overlap_graphr r rn set set_filling neighborsr addremove_nodes_from)
r"
root_nodesrootcurrentvisitedr r r n2d2r r r compute_filling s0
(
z SVGPathGroup.compute_filling?c C s t }dd | jD }t|D ]7\}}|| | j| jrGt|D ]#\}}||krF| j| jrF||j|j } | |krF|j||| d q#q|ret |}
t j
||
dd t |d} t j ||
| d |S )Nc S r r r
r r r r r r z.SVGPathGroup.overlap_graph..)weightT) with_labelsr# ) edge_labels)
nxDiGraphr r add_nodern intersectionareaadd_edge
spring_layout
draw_networkxget_edge_attributesdraw_networkx_edge_labels) r" thresholdr2 r shapesr path1jpath2overlapr labelsr r r r s"
zSVGPathGroup.overlap_graphotherc C s | | S r
)r> r5 )r" r7 r r r bbox_overlap r zSVGPathGroup.bbox_overlapc C s t dd | jD S )Nc S r r ) to_pointsr r r r r r z*SVGPathGroup.to_points..)np concatenater r; r r r r9 rY zSVGPathGroup.to_points)NN)r r )r rL )r r rB )Fr )r rq )r )r r r F)NNT)r" F)r7 r )/rC rD rE r# propertyr r r r r r r r r r= rX r9 r r< ra r r r r r r r r r r r r r r r r r r r r r> r r! r r8 r9 r{ r r rQ r r s`
r )!
__future__r geomr retypingr r xml.domr r r svg_commandr r
r r r Z shapely.opsZshapely.geometrynetworkxr& compiler r r rJ r} r r r r r r r r r s,
+"%