'jzZUddlmZddlmZmZmZmZmZmZm Z ddl m Z ddl m Z ddlmZddlZddlZddlmZmZddlmZmZmZmZmZmZmZmZmZerdd l m!Z!dd l"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(gd Z)ed d Z*eddZ+eeeeefZ,de-d<eeZ.de-d<edZ/de-d<GddZ0Gdde0Z1d*d Z2ed!Z3Gd"d#e0Z4d+d'Z5Gd(d)eZ6dS),) annotations) TYPE_CHECKINGIterableTupleUnioncastSequenceOptional) TypeAlias)abstractmethod) namedtupleNVTX_EXTRA_VERTEX_CREATEDVTX_SPLINE_FRAME_CONTROL_POINT) Vec2Vec3UVecBSplineConstructionRayOCSParallelRaysError bulge_to_arcConstructionArc)Drawing) DXFGraphicSolidTraceFace3d LWPolylinePolyline) TraceBuilder LinearTrace CurvedTrace LinearStation)vertex start_width end_width CurveStation)vertex0vertex1r FacePolygon)rrr Quadrilateralc:eZdZed dZd dZ ddd ZdS) AbstractTracereturnIterable[Face]cdSNselfs L/DATA/AppData/hermes/venv/lib/python3.11/site-packages/ezdxf/render/trace.pyfaceszAbstractTrace.faces2s r,c Jdd}g}g}|D]5\}}}}|||f|||f6t||}|tt|||S)Nverticesr,r0Iterable[UVec]c3Kt|sdSt|}t|}|V|D]}||s|V|}dSr3)leniternextisclose)r; _verticesprevr%s r7mergez$AbstractTrace.polygon..merge;stx== XI ??DJJJ# " "||F++" LLL!D " "r9)r;r,r0r<)r8extendlistreversed) r6rDforward_contourbackward_contourup1down1down2up2contours r7polygonzAbstractTrace.polygon:s " " " "')')&*jjll 0 0 "Cs  " "E5> 2 2 2  # #S#J / / / /uu_--..xUU+;%<%< = =>>???r9TRACENdocOptional[Drawing]Iterable[Quadrilateral]c#fKddlm}|dvrtd|dt|pi}|D]k}t dD]}|||d|<|dkr|d |d <|d |d <||||}|r|j||Vld S)a Yields faces as SOLID, TRACE or 3DFACE entities with DXF attributes given in `dxfattribs`. If a document is given, the doc attribute of the new entities will be set and the new entities will be automatically added to the entity database of that document. Args: dxftype: DXF type as string, "SOLID", "TRACE" or "3DFACE" dxfattribs: DXF attributes for SOLID, TRACE or 3DFACE entities doc: associated document r)new>SOLIDrP3DFACEzInvalid dxftype .vtxrWvtx2vtx3N)ezdxf.entities.factoryrU TypeErrordictr8rangeentitydbadd)r6dxftype dxfattribsrQrUfaceientitys r7virtual_entitieszAbstractTrace.virtual_entitiesQs" /..... 6 6 69w999:: :**++ JJLL  D1XX 0 0(,Q 999%%(""%)!W 6"%)!W 6"S*c22F )   (((LLLL  r9r0r1)r0r,rPNNrQrRr0rS)__name__ __module__ __qualname__r r8rOrjr4r9r7r/r/1sf   ^ 0JN!!!!!!!r9r/cReZdZdZddZdZdZeddZ dddZ ddZ d S)r"ztLinear 2D banded lines like polylines with start- and end width. Accepts 3D input, but z-axis is ignored. r0Nonec"g|_d|_dSNg-q=) _stationsabs_tolr5s r7__init__zLinearTrace.__init__|s.0 r9c*t|jSr3r>rur5s r7__len__zLinearTrace.__len__4>"""r9c|j|Sr3rur6items r7 __getitem__zLinearTrace.__getitem__~d##r9boolc*t|jS)z%`True` if at least one station exist.)rrur5s r7 is_startedzLinearTrace.is_startedsDN###r9Npointrr&floatr'Optional[float]c J||}t|}|j}t|r;|dj||jr||t|t|t|dS)aaAdd a trace station (like a vertex) at location `point`, `start_width` is the width of the next segment starting at this station, `end_width` is the end width of the next segment. Adding the last location again, replaces the actual last location e.g. adding lines (a, b), (b, c), creates only 3 stations (a, b, c), this is very important to connect to/from splines. Args: point: 2D location (vertex), z-axis of 3D vertices is ignored. start_width: start width of next segment end_width: end width of next segment N)rv) rrurr%rArvpopappendr$r)r6rr&r'stationss r7 add_stationzLinearTrace.add_stations"  #IU > >> hrl199%9VV  LLNNN eU;-?-?yAQAQRRSSSSSr9r1c #x K|j t }|dkrdSd fd }dd } dj dj}gt |dz D]C} |\}}} |dzj} t || ||D|d\} } d} d} d\}}}}t tD]}|\}}}}|dkrH|r=|tdz \}}||| |}||| |}|}|}n|}|}| } | } n|| | |}|| | |}|tdz kr4||dz\}}||| |}||| |}| } | } |} |} n|}|}||||fVdS)a2Yields all faces as 4-tuples of :class:`~ezdxf.math.Vec2` objects. First and last miter is 90 degrees if the path is not closed, otherwise the intersection of first and last segment is taken into account, a closed path has to have explicit the same last and first vertex. r]Nsegmentintr0'tuple[ConstructionRay, ConstructionRay]c\fd}\}}}}||||||fS)z0Create offset rays from segment offset vertices.c||r3jdzjz j}t||St||S)N)rAr%angler)v1v2rrrs r7rayz3LinearTrace.faces..offset_rays..rays]::b>>3!)08GaK3H3OO+2u555*2r222r9r4)rrleft1left2right1right2segmentsrs` r7 offset_raysz&LinearTrace.faces..offset_rayssZ  3 3 3 3 3 3,4G+< (E5&&3ue$$cc&&&9&99 9r9ray1rray2defaultrct|j|j}|dks ttj|z dkr|S ||S#t $r|cYSwxYw)z7Intersect two rays but take parallel rays into account.g*5{?)abs direction angle_betweenmathpi intersectr)rrrrs r7rz$LinearTrace.faces..intersects 44T^DDEEEx3tw#7#7(#B#B ~~d+++$    sA++ A:9A:rrr)rrr0r)rrrrrrr0r)rur>r%rArbr_normal_offset_points)!r6countrr is_closedstation start_vertexr&r' end_vertex offset_ray1 offset_ray2prev_offset_ray1prev_offset_ray2last_up1last_up2 last_down1 last_down2rhrJrMrKrLlast_offset_ray1last_offset_ray2vtx0vtx1next_offset_ray1next_offset_ray2r\r^rrs! @@r7r8zLinearTrace.facess>H  199 F : : : : : : :$    QK&..x|/BCC UQY''  G3;G3D 0L+y!'A+.5J OO%lJ YWW    $/;q>> [6>b\2(J s8}}%%& )& )A%-a[ "CeUAvv !:ES]]UVEV9W9W6$&6$9%5{CHHD$9%5{EJJD $H!%JJD D#. #.  !y!1;DD y!1;FF3x==1$$$5@[Q5G5G2 "2 y!1;FF y!1;DD#. #. . . "dD( ( ( ( (M& )& )r9r0rr)r0rr3)rrr&rr'rr0rrrk) rnrorp__doc__rwrzrpropertyrrr8r4r9r7r"r"us ###$$$$$$X$ MQTTTTT6o)o)o)o)o)o)r9r"startrendr&rr'r0c||z }|d}||dz }||dz }||z||z||z ||z fS)NTr]) normalize orthogonal)rrr&r' dir_vectorortho offset_start offset_ends r7rrs+((**J  ! !$ ' 'E??;?33LQ//J  j  j  r9)rrcjeZdZdZddZdZdZeddZe dddZ ddZ d dZ dS)!r#z2D banded curves like arcs or splines with start- and end width. Represents always only one curved entity and all miter of curve segments are perpendicular to curve tangents. Accepts 3D input, but z-axis is ignored. r0rrcg|_dSr3r}r5s r7rwzCurvedTrace.__init__4s -/r9c*t|jSr3ryr5s r7rzzCurvedTrace.__len__7r{r9c|j|Sr3r}r~s r7rzCurvedTrace.__getitem__:rr9splinerr&rr'rrcj|}|dz}tjd|j|}t||dtj|||D]N\\}} } t | d} |t || | O|S)z Create curved trace from a B-spline. Args: spline: :class:`~ezdxf.math.BSpline` object start_width: start width end_width: end width segments: count of segments for approximation rr)nT)nplinspacemax_tzip derivativesrr_append) clsrr&r'r curve_tracertr derivativewidthnormals r7 from_splinezCurvedTrace.from_spline=s$cee 1  K6< / /*-   qA  & & KE(R(R+ +  < < & UJ*%%0066F   U VU ; ; ; ;r9@arcrc|jdkrtd|jd|}ttj|jdz |zddz}t |j}t| | |tj |||D]\}} | |||z |  |S)a Create curved trace from an arc. Args: arc: :class:`~ezdxf.math.ConstructionArc` object start_width: start width end_width: end width segments: count of segments for full circle (360 degree) approximation, partial arcs have proportional less segments, but at least 3 Raises: ValueError: if arc.radius <= 0 rzInvalid radius: rXgv@r[r)radius ValueErrormaxrceil angle_spanrcenterrr;anglesrrr) rrr&r'rrrrrrs r7from_arczCurvedTrace.from_arcYs. :??= ===>> >cee DIcnu4x?@@!DDqHcj!! LLE** + + K Y 6 6   > >LE5   uv~u = = = =r9rrrrct|rt}n||dz }|jt ||z||z dS)z Add a curve trace station (like a vertex) at location `point`. Args: point: 2D curve location (vertex), z-axis of 3D vertices is ignored. normal: curve normal width: width of station r]N) _NULLVEC2rArrurr()r6rrrs r7rzCurvedTrace._append|sf   V $ $ 1FF%%eai00F l56>56>JJKKKKKr9r1c#Kt|j}|dkrdSd}d}|jD]\}}||}|} ||||fV|}|}dS)zBYields all faces as 4-tuples of :class:`~ezdxf.math.Vec2` objects.r]Nry)r6rrrr\r^s r7r8zCurvedTrace.facessDN## 199 F.  JD$|dD( ( ( (DDD  r9Nr) rrr&rr'rrrr0r#r) rrr&rr'rrrr0r#)rrrrrrr0rrrk) rnrorprrwrzr classmethodrrrr8r4r9r7r#r#*s0000###$$$[6     [ DLLLL r9r# vertex_flagsrrc6|tzrdS|tzrdSdS)aaDetermine if a vertex should be included in rendering based on its flags. According to the DXF specification: - Flag 0: Straight-segment vertex (include) - Flag 1: Extra curve-fit vertex (exclude - auto-generated) - Flag 2: Curve-fit tangent vertex (include - carries tangent info) - Flag 8: Spline-vertex (fit-point) for 2D spline (include) - Flag 16: Spline-frame control point (exclude - internal calculation only) Args: vertex_flags: The vertex flags value (group code 70) Returns: True if the vertex should be included in rendering, False otherwise FTr)rs r7#should_include_vertex_for_renderingrs3"..u44u 4r9czeZdZdZd"dZdZdZd#d Zd$d Zd%dZ d&dZ d%dZ d'd(dZ dZ ed)d*d!ZdS)+r!aSequence of 2D banded lines like polylines with start- and end width or curves with start- and end width. .. note:: Accepts 3D input, but z-axis is ignored. The :class:`TraceBuilder` is a 2D only object and uses only the :ref:`OCS` coordinates! r0rrc"g|_d|_dSrt)_tracesrvr5s r7rwzTraceBuilder.__init__s,.  r9c*t|jSr3)r>rr5s r7rzzTraceBuilder.__len__s4<   r9c|j|Sr3)rr~s r7rzTraceBuilder.__getitem__s|D!!r9tracer/c:|j|dS)zAppend a new trace.N)rrr6rs r7rzTraceBuilder.appends E"""""r9r1c#RK|jD]}|Ed{VdS)zaYields all faces as 4-tuples of :class:`~ezdxf.math.Vec2` objects in :ref:`OCS`. N)rr8rs r7r8zTraceBuilder.facessF\ % %E{{}} $ $ $ $ $ $ $ $ % %r9ocsr elevationrIterable[Sequence[Vec3]]c#K|D]2}t|fd|DV3dS)zaYields all faces as 4-tuples of :class:`~ezdxf.math.Vec3` objects in :ref:`WCS`. c3NK|]}t|j|jV dSr3rxy.0vrs r7 z)TraceBuilder.faces_wcs..s3)R)R$qsAC*C*C)R)R)R)R)R)Rr9N)r8tuple points_to_wcs)r6rrrgs ` r7 faces_wcszTraceBuilder.faces_wcsshJJLL T TD)))R)R)R)RT)R)R)RRRSS S S S S T Tr9Iterable[Polygon]c#JK|jD]}|VdS)z{Yields for each sub-trace a single polygon as sequence of :class:`~ezdxf.math.Vec2` objects in :ref:`OCS`. N)rrOrs r7polygonszTraceBuilder.polygonss:\ " "E--// ! ! ! ! " "r9c#K|jD]D}t|fd|DVEdS)z{Yields for each sub-trace a single polygon as sequence of :class:`~ezdxf.math.Vec3` objects in :ref:`WCS`. c3NK|]}t|j|jV dSr3rrs r7rz,TraceBuilder.polygons_wcs..s3!U!U$qsAC";";!U!U!U!U!U!Ur9N)rrrrO)r6rrrs ` r7 polygons_wcszTraceBuilder.polygons_wcsss\  E!!!U!U!U!UU]]__!U!U!UUU      r9rPNrQrRrSc#XK|jD]}||||Ed{V dS)arYields faces as SOLID, TRACE or 3DFACE entities with DXF attributes given in `dxfattribs`. If a document is given, the doc attribute of the new entities will be set and the new entities will be automatically added to the entity database of that document. .. note:: The :class:`TraceBuilder` is a 2D only object and uses only the :ref:`OCS` coordinates! Args: dxftype: DXF type as string, "SOLID", "TRACE" or "3DFACE" dxfattribs: DXF attributes for SOLID, TRACE or 3DFACE entities doc: associated document N)rrj)r6rerfrQrs r7rjzTraceBuilder.virtual_entitiessT*\ H HE--gz3GG G G G G G G G G H Hr9cx|j}t|dkrdSt|dtr~t|dtret t|d}t t|d}|D]!\}}}||||dSdSdS)zEClose multi traces by merging first and last trace, if linear traces.r]Nrr)rr> isinstancer"rrr)r6tracesfirstlastrr&r's r7closezTraceBuilder.close s v;;?? F fQi - - @*VBZ2U2U @fjjmm44E VBZ00D16 @ @-{I   Y????  @ @ @ @ @ @r9rpolylinerrrc |}|dkrftd|}|jj}g}|jD]7\}}}} } t ||} |r|}|} || || | f8|j} n|dkrtd|}|jstd|j } |jj } |jj }d|j D}g}|D]}t |jj} |jdr |jj}n| }|jdr |jj} n|} |jj} || || | fntd || rH|d d |d d s||d |}d }d }d }d }t)}|D]\}}} } |d krt+|||\}}}}|d krt-||t/j|t/j|d}|j|r|}|}n|}|}|t4||||d }| d krK|jr:|||| ||t)}| }|}| }|}|||| !|jr||| r't=|dkr||S)a Create a complete trace from a LWPOLYLINE or a 2D POLYLINE entity, the trace consist of multiple sub-traces if :term:`bulge` values are present. Uses only the :ref:`OCS` coordinates! Args: polyline: :class:`~ezdxf.entities.LWPolyline` or 2D :class:`~ezdxf.entities.Polyline` segments: count of segments for bulge approximation, given count is for a full circle, partial arcs have proportional less segments, but at least 3 LWPOLYLINErPOLYLINEr z2D POLYLINE requiredcbg|],}t|jdd*|-S)flagsr)rdxfget)rr%s r7 z.TraceBuilder.from_polyline..=sF!!!!6vz~~gq7Q7QRR!!!!r9r&r'zInvalid DXF type rrgNT)is_counter_clockwiser) rerr const_widthlwpointsrrclosedis_2d_polyliner`rdefault_start_widthdefault_end_widthr;locationhasattrr&r'bulgerAr"rrrdegrees start_pointr#rrrr>r)rrrrerpointsrrr&r'r'r%r!r#r$filtered_verticesr%r store_bulgestore_start_widthstore_end_width store_point linear_tracerr start_angle end_anglerrswews r7 from_polylinezTraceBuilder.from_polylines""$$ l " "L(33H",2KF7?7H I I31k9e1::,#.K +I xiGHHHH_FF  " "J11H* 8 6777'F"*,"B  ( > !!%-%6!!!  F+ I I 344:%%m446"(*"8KK"5K:%%k222 & 4II 1I ( xiGHHHH I999:: :  %&)A,..vbz!}== % MM&) $ $ $ #' "}} 4:! D! D 0E; 5c!!9E ::6 YA::) [11 Y//-1 C..u55/,.,.LL!5!5c2r8!L!LMMM! ||*1 ,,UKKKKLL...#.==L# $/!"+#   $ $UK C C C C  " ' LL & & &  c%jj1nn KKMMM r9r)rr/r0rrrk)rrrrr0r)r0rrlrmr)rrrrr0r!)rnrorprrwrzrrr8rr r rjrrr5r4r9r7r!r!s  !!!"""####%%%%TTTT""""JNHHHHH0 @ @ @mmmm[mmmr9r!) rrrrr&rr'rr0r+)rrr0r)7 __future__rtypingrrrrrr r typing_extensionsr abcr collectionsr rnumpyrezdxf.lldxf.constrr ezdxf.mathrrrrrrrrrezdxf.documentrezdxf.entitiesrrrrrr __all__r$r(r+__annotations__r,r-r/r"rrr#rr!r4r9r7rBs#""""""('''''"""""" VVVVVVVV                      V&&&&&&UUUUUUUUUUUUUUUU 8 8 8 ?,RSS z.*@AA dD$./////d^#### !;< <<<<AAAAAAAAH`)`)`)`)`)-`)`)`)F     DLL qqqqq-qqqh>IIIII8IIIIIr9