hZPUddlmZddlmZmZmZmZddlmZm Z ddl Z ddl Z ddl mZddlmZmZmZmZmZmZmZmZmZddlmZmZmZmZmZmZm Z m!Z! ddl"m#Z#gdZ%e&ejNZ(e&ejRZ*e&ejVZ,e&ejZZ.Gd d e/Z0Gd d e0Z1e jdZ3d e4d<e jjZ6d e4d<e jnge6Z8e jnge3Z9Gdde jtZ;Gdde;Z<Gdde;Z=d&dZ>dZ?dZ@dZAdZBde@feAeAfeBeBeBfe?fgZCddd&dZDd'd ZEd(d!ZFGd"d#e jtZGGd$d%eGZHy#e$$rdZ#Y+wxYw))) annotations)IterableOptionalIteratorSequence)Self TypeAliasN) Matrix44UVecVec2Vec3has_clockwise_orientationBezier3PBezier4P BoundingBox2d BoundingBox)PathCommand PathElementLineToMoveToCurve3ToCurve4Tonesting) np_support) NumpyPath2d NumpyPoints2d NumpyPoints3dNumpyShapesExceptionEmptyShapeErrorto_qpainter_pathto_matplotlib_path single_paths orient_pathsc eZdZy)rN__name__ __module__ __qualname__.lib/python3.12/site-packages/ezdxf/npshapes.pyrr9r+rc eZdZy)r Nr&r*r+r,r r =r-r+r r CommandNumpyTypeVertexNumpyTypedtypec~eZdZUdZeZded<d dZejddZ ddZ ddZ ddZ dd Zdd Zdd Zy ) NumpyShape2dzThis is an optimization to store many 2D paths and polylines in a compact way without sacrificing basic functions like transformation and bounding box calculation. npt.NDArray[VertexNumpyType] _verticesc|j}t|dkDr4t|jdt|j dfSt dzBReturns the extents of the bounding box as tuple (extmin, extmax).rzempty shape has no extends)r6lenr minmaxr selfvs r,extentszNumpyShape2d.extentsNE NN q6A:a>4a>1 1!">? ?r+cyNr*r=s r,clonezNumpyShape2d.cloneV r+c|jSrBr6rCs r, np_verticeszNumpyShape2d.np_verticesZ ~~r+c^|j}t|dk(ry|j|dy)-Transforms the vertices of the shape inplace.rNr6r9transform_array_inplacer=mr>s r,transform_inplacezNumpyShape2d.transform_inplace]) NN q6Q;  !!!Q'r+cR|jDcgc] }t|c}Scc}w)ziReturns the shape vertices as list of :class:`Vec2` e.g. [Vec2(1, 2), Vec2(3, 4), ...] r6r r<s r,verticeszNumpyShape2d.verticesds$"&0AQ000$cR|jDcgc] }t|c}Scc}w)z[Returns the shape vertices as list of 2-tuples e.g. [(1, 2), (3, 4), ...] )r6tupler<s r, to_tupleszNumpyShape2d.to_tuplesjs$#'..1.Qa.111rVc6|jjS)zXReturns the shape vertices as list of lists e.g. [[1, 2], [3, 4], ...] )r6tolistrCs r,to_listzNumpyShape2d.to_listps~~$$&&r+c4t|jSz)Returns the bounding box of all vertices.)rr?rCs r,bboxzNumpyShape2d.bboxvsT\\^,,r+N)returnztuple[Vec2, Vec2]r`rr`r5rPr r`Noner`z list[Vec2])r`zlist[tuple[float, float]])r`zlist[list[float]])r`r)r'r(r)__doc__ EMPTY_SHAPEr6__annotations__r?abcabstractmethodrDrHrQrUrYr\r_r*r+r,r4r4GsR/:I+9@   (1 2 ' -r+r4c,eZdZdZddZddZeZddZy) rz5Represents an array of 2D points stored as a ndarray.c|rEtj|Dcgc]}|j|jfc}t|_yycc}wNr1)nparrayxyr0r6r=pointsr>s r,__init__zNumpyPoints2d.__init__~s= XX%+,V!##qssV,ODN ,sA cf|jd}|jj|_|SrB __class__r6copyr=rDs r,rDzNumpyPoints2d.clone)t$..--/ r+c,t|jSrBr9r6rCs r,__len__zNumpyPoints2d.__len__4>>""r+N)rszOptional[Iterable[Vec2 | Vec3]]r`rdrar`intr'r(r)rfrtrD__copy__r}r*r+r,rr{?  H#r+rc0eZdZUdZded<ddZddZeddZeddZ ddZ dd Z e Z d d Z d!d Zd"d Ze d# d$d Zed%dZed%dZed%dZed%dZd&dZd%dZddZddZddZd'd(dZd)dZed*dZy)+rabRepresents a 2D path, the path control vertices and commands are stored as ndarray. This class cannot build paths from scratch and is therefore not a drop-in replacement for the :class:`ezdxf.path.Path` class. Operations like transform and reverse are done inplace to utilize the `numpy` capabilities. This behavior is different from the :class:`ezdxf.path.Path` class!!! Construct new paths by the :class:`Path` class and convert them to :class:`NumpyPath2d` instances:: path = Path((0, 0)) path.line_to((50, 70)) ... path2d = NumpyPath2d(path) znpt.NDArray[CommandNumpyType] _commandsc|t|_t|_y|j Dcgc]}|j |j f}}t|dk(r t|jg}tj|t|_tj|jt|_ycc}w#t$rg}YawxYw)Nrr1)rgr6 NO_COMMANDSrcontrol_verticesrprqr9r start IndexErrorrnror0 command_codesr/)r=pathr>rUs r,rtzNumpyPath2d.__init__s <(DN(DN (,(=(=(?@(?1QSS!##J(?@ x=A   ,-(/B$"4"4"6>NOA  sCC CCc,t|jSrB)r9rrCs r,r}zNumpyPath2d.__len__r~r+c2t|jdS)z>Returns the start point as :class:`~ezdxf.math.Vec2` instance.rr r6rCs r,rzNumpyPath2d.startsDNN1%&&r+c2t|jdS)z>Ck!Xe_--  %x 2HUODD  %UQY'%(519:M  #Xe_-- "s BB8 B8c|jDcgc] }t|}}|jDcgc] }t|}}t j ||Scc}wcc}w)z0Returns a new :class:`ezdxf.path.Path` instance.)r6r rrrfrom_vertices_and_commands)r=r>rUcrs r,to_pathzNumpyPath2d.to_pathsU%)^^4^DG^4(,71GAJ7..xBB57s AA c|d}t|}t|dk(r|S|r+|dj|ds|j|d|Dcgc]}|j|j f}}t j|t|_ t jt|dz tt|_ |Scc}w)Nrrr1r) fill_valuer2)rr9iscloseappendrprqrnror0r6fullrr/r)clsrUclosenew_pathr>rss r, from_verticeszNumpyPath2d.from_verticesst9> x=A O !,,Xb\: OOHQK (&./h133*h/XXfODWW K!O ;K  0sC c&t|jvS)zoReturns ``True`` if the path is a :term:`Multi-Path` object that contains multiple sub-paths. )rrrCs r, has_sub_pathszNumpyPath2d.has_sub_pathss dnn,,r+c~t|jdkDr%|jj|jSy)z>Returns ``True`` if the start point is close to the end point.rF)r9r6rrrrCs r, is_closedzNumpyPath2d.is_closeds1 t~~  "::%%dhh/ /r+c&t|jvS)z3Returns ``True`` if the path has any line segments.)rrrCs r, has_lineszNumpyPath2d.has_linessdnn,,r+cNt|jvxst|jvS)z4Returns ``True`` if the path has any curve segments.)rrrrCs r, has_curveszNumpyPath2d.has_curves s!.Q-4>>2QQr+c z d fd }jtdk(rgStvrgSgjd dddD]P}|tk(rdz n:|t k(rdz n+|t k(rdz n|tk(r|dz  dzdz Rdtk7r|S)zYield all sub-paths as :term:`Single-Path` objects. It's safe to call :meth:`sub_paths` on any path-type: :term:`Single-Path`, :term:`Multi-Path` and :term:`Empty-Path`. cxjd}dz|_|_j|y)Nr)rwr6rr) s cmd_indexcmd_start_indexrr= sub_pathsrU vtx_indexvtx_start_indexs r,append_sub_pathz.NumpyPath2d.sub_paths..append_sub_paths?nnT*A"?Y]CAK"?9=AK   Q r+rrrLrr)r`rd)rr9rr6rrr) r=rrrrrrrUrrs ` @@@@@@@r,rzNumpyPath2d.sub_pathss  >> x=A I h &6M " >>  Ck!Q  %Q  %Q  #!Q "+"+a- NI B<; &  r+c|jr tdtt|j Stj|j S)zReturns ``True`` if 2D path has clockwise orientation. Raises: TypeError: can't detect orientation of a :term:`Multi-Path` object z/can't detect orientation of a multi-path object)r TypeErrorrrrUr6rCs r,rz%NumpyPath2d.has_clockwise_orientation?sE   MN N  ,T]]_= =77G Gr+c|j}t|js|S|dtk(rhtj|ddj |_tj|j dddfdj |_|Stj|j |_tj|j dj |_|S)z!Reverse path orientation inplace.rN.raxis)rr9rrnfliprxr6)r=rs r,reversezNumpyPath2d.reverseMs>>4>>"K B<; &  WWXcr]388:DNWWT^^CRCH%=AFKKMDN  WWX.335DNWWT^^!<AACDN r+cF|js|j|S)zApply clockwise orientation inplace. Raises: TypeError: can't detect orientation of a :term:`Multi-Path` object rrrCs r, clockwisezNumpyPath2d.clockwise^s--/ LLN r+cF|jr|j|S)zApply counter-clockwise orientation inplace. Raises: TypeError: can't detect orientation of a :term:`Multi-Path` object rrCs r,counter_clockwisezNumpyPath2d.counter_clockwiseis  ) ) + LLN r+c #~Kt|jsy|j}|d}|d}|jD]}|tk(s |tk(r||}|dz }|n|t k(rW|||dz\}}|dz }t jt|||fj||} t| | Ed{np|tk(rY|||dz\} } }|dz }t jt|| | |fj||} t| | Ed{ntd||}y7z7w)z4Flatten path to vertices as :class:`Vec2` instances.NrrrLrzInvalid command: )r9rrUrrrr generater flatteningnextrr ValueError) r=distancesegmentsrUrrr end_locationctrlptsctrl1ctrl2s r,rzNumpyPath2d.flatteningusR4>>" ==?  >>Ck!SK%7'  "" %%-eeai%@"l mmeT<89DDXxXS  %-5eeai-H*ul mmeUE<@ALL ( S  #4SE!:;; E3"s%BD9?A!D= D;!D=;D=ct|syt|js |d}|dd}n|}|jg}|jg}|j}|D]}t|jdk(r|j |j sK|j tjtft|j |jn|j |jdd|j}|j |jtj|d|_tj||_y)zExtend an existing path by appending additional paths. The paths are connected by MOVE_TO commands if the end- and start point of sequential paths are not coincident (multi-path). Nrrr1r) r9rr6rrrrrnrorr/ concatenate)r=pathsfirstrUrr next_paths r,extendzNumpyPath2d.extends 5z 4>>"!HE!"IEE&+oo%6&+oo%6III9&&'1,;;y/+?O PQ 3 34 3 3AB 78--C OOI// 0q91r+cn|s tdS|dj}|j|dd|S)zReturns a new path of concatenated paths. The paths are connected by MOVE_TO commands if the end- and start point of sequential paths are not coincident (multi-path). Nrr)rrDr)rrs r,rzNumpyPath2d.concatenates9t$ $a  U12Y r+N)rzOptional[Path]r`rdr)r`r rera)r`z list[int])r`zIterator[PathElement])r`r)F)rUzIterable[Vec2 | Vec3]rboolr`r)r`r)r`z list[Self]))rfloatrrr`zIterator[Vec2])rSequence[NumpyPath2d]r`rd)rrr`r)r'r(r)rfrhrtr}propertyrrrrDrrrr classmethodrrrrrrrrrrrr staticmethodrr*r+r,rrs"-, P#''((1 H$&C Convert the given `paths` into a single :class:`QPainterPath`.r) QPainterPathQPointFone or more paths requiredrrLr)ezdxf.addons.xqtrrrr9rrUrprqmoveTorrlineTorquadTorcubicTor) rrrqpathrr>rsrrs r,r!r!s86 KE 5zQ566 NE-1]]_=_'!##qss#_= VAY%%'Ck! VE]+  % VE]F519,=>  % fUmVEAI->uqy@QR  # VE]+ ( $ L#>s $ErrLrr)rF) detect_holesc ddlm}t|}t|dk(r t d|r t |}g}g}|D]c}|j |j|j t|jD]}|jt|etj|} |||S#t$r4}t dt|dt|dt|d}~wwxYw)aConvert the given `paths` into a single :class:`matplotlib.path.Path`. Matplotlib requires counter-clockwise oriented outside paths and clockwise oriented holes. Set the `detect_holes` argument to ``True`` if this path orientation is not yet satisfied. r)rrzmatplotlib.path.Path(z, z): N)matplotlib.pathrrr9rr$rrH MPL_MOVETOrr MPL_CODESrnr Exceptionstr) rrrrUcodesrrrses r,r"r"s% KE 5zQ566U#!#HE((*+ Z %%'C LL3 (( ^^H %FYFE"" Y0V RE |3sSTvhWXXYs:C D /C;;Dc`g}|D]&}|j}|s|j|(|SrB)rr)r single_paths_prs r,r#r#s5')M KKM    + r+ct|}t|dkr|Stj|}tj|\}}|D]}|j |D]}|j ||zS)zoReturns a new list of paths, with outer paths oriented counter-clockwise and holes oriented clockwise. rL)r#r9rmake_polygon_structurewinding_deconstructionrr)rrpolygons outer_pathsholesrs r,r$r$s}$0#6I 9~ --i8H!77AK     r+cneZdZUdZeZded<d dZejd dZ d dZ ddZ ddZ dd Zy ) NumpyShape3dzThis is an optimization to store many 3D paths and polylines in a compact way without sacrificing basic functions like transformation and bounding box calculation. r5r6c|j}t|dkDr4t|jdt|j dfSt dr8)r6r9r r:r;r r<s r,r?zNumpyShape3d.extents;r@r+cyrBr*rCs r,rDzNumpyShape3d.cloneCrEr+c|jSrBrGrCs r,rHzNumpyShape3d.np_verticesGrIr+c^|j}t|dk(ry|j|dy)rKrNrrMrOs r,rQzNumpyShape3d.transform_inplaceJrRr+cR|jDcgc] }t|c}Scc}w)z4Returns the shape vertices as list of :class:`Vec3`.)r6r r<s r,rUzNumpyShape3d.verticesQs"!%0AQ000rVc4t|jSr^)rr?rCs r,r_zNumpyShape3d.bboxUs4<<>**r+N)r`ztuple[Vec3, Vec3]rarbrc)r`z list[Vec3])r`r)r'r(r)rfrgr6rhr?rirjrDrHrQrUr_r*r+r,rr4sH/:I+9@   (1+r+rc,eZdZdZddZddZeZddZy) rz5Represents an array of 3D points stored as a ndarray.c|rBtj|Dcgc]}t|jc}t|_yycc}wrm)rnror xyzr0r6rrs r,rtzNumpyPoints3d.__init__]s9 XX&,-faf-_DN -sAcf|jd}|jj|_|SrBrvrys r,rDzNumpyPoints3d.clonecrzr+c,t|jSrBr|rCs r,r}zNumpyPoints3d.__len__jr~r+N)rszOptional[Iterable[UVec]]r`rdrarrr*r+r,rrZrr+r)rIterable[NumpyPath2d])rrr`list[NumpyPath2d])rrr`r)I __future__rtypingrrrrtyping_extensionsrr rinumpyrn numpy.typingnpt ezdxf.mathr r r r rrrrr ezdxf.pathrrrrrrrr ezdxf.accr ImportError__all__rMOVE_TOrLINE_TOr CURVE3_TOr CURVE4_TOrrrr int8r/rhfloat64r0rorgrABCr4rrr!r MPL_LINETO MPL_CURVE3 MPL_CURVE4rr"r#r$rrr*r+r,r%s#99-       $ '//" '//" G%%& G%%&  9  * !gg)%ZZ'bhhr1 bhhr!12 1-3771-h#L#(y,yx <     MZ(M   FKY>*#+377#+L#L#oJsE<<FF