hAWUddlmZddlmZmZmZmZmZmZm Z ddl m Z ddl m Z ddlmZddlZddlZddlmZmZmZmZmZmZmZmZmZerddlmZdd lm Z m!Z!m"Z"m#Z#m$Z$m%Z%gd Z&ed d Z'ed dZ(eeeeefZ)de*d<eeZ+de*d<edZ,de*d<GddZ-Gdde-Z. ddZ/edZ0Gdde-Z1GddeZ2y)) annotations) TYPE_CHECKINGIterableTupleUnioncastSequenceOptional) TypeAlias)abstractmethod) namedtupleN) Vec2Vec3UVecBSplineConstructionRayOCSParallelRaysError bulge_to_arcConstructionArc)Drawing) DXFGraphicSolidTraceFace3d LWPolylinePolyline) TraceBuilder LinearTrace CurvedTrace LinearStation)vertex start_width end_width CurveStation)vertex0vertex1r FacePolygon)rrr Quadrilateralc8eZdZeddZddZ d ddZy) AbstractTracecyNselfs 2lib/python3.12/site-packages/ezdxf/render/trace.pyfaceszAbstractTrace.faces1s c  dd}g}g}|jD]-\}}}}|j||f|j||f/t||}|jtt|||S)Nc3Kt|syt|}t|}||D]}|j|r||}ywr.)leniternextisclose)vertices _verticesprevr"s r2mergez$AbstractTrace.polygon..merge:sGx=XI ?DJ#||F+ L!D$s =A  A )r;r)returnzIterable[UVec])r3extendlistreversed) r1r>forward_contourbackward_contourup1down1down2up2contours r2polygonzAbstractTrace.polygon9s "')')&*jjl "Cs  " "E5> 2  # #S#J /'3u_-.xU+;%< =>?r4Nc#<Kddlm}|dvrtd|dt|xsi}|j D]]}t dD] }|||d|<|dk7r|d |d <|d |d <||||}|r|j j||_y w)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>SOLIDTRACE3DFACEzInvalid dxftype .vtxrOvtx2vtx3N)ezdxf.entities.factoryrL TypeErrordictr3rangeentitydbadd)r1dxftype dxfattribsdocrLfaceientitys r2virtual_entitieszAbstractTrace.virtual_entitiesPs" / 6 6.wiq9: :**+ JJLD1X(,Q S9%("%)!W 6"%)!W 6"*c2F   (L!sBBr?zIterable[Face])r?r)rNNNr_zOptional[Drawing]r?zIterable[Quadrilateral])__name__ __module__ __qualname__r r3rJrcr/r4r2r,r,0s5  0JN!5F! !r4r,cXeZdZdZd dZdZdZed dZ d d dZ d dZ y)rztLinear 2D banded lines like polylines with start- and end width. Accepts 3D input, but z-axis is ignored. c g|_d|_yNg-q=) _stationsabs_tolr0s r2__init__zLinearTrace.__init__{s.0 r4c,t|jSr.r7rmr0s r2__len__zLinearTrace.__len__4>>""r4c |j|Sr.rmr1items r2 __getitem__zLinearTrace.__getitem__~~d##r4c,t|jS)z%`True` if at least one station exist.)boolrmr0s r2 is_startedzLinearTrace.is_startedsDNN##r4Nc  ||}t|}|j}t|r:|djj ||j r|j |jt|t|t|y)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)rn) rrmr{r"r:rnpopappendr!float)r1pointr#r$stationss r2 add_stationzLinearTrace.add_stationsp"  #IU >> >hrl1199%9V LLN eU;-?yAQRSr4c # K|j t }|dkry d fd } d d} djj dj}gt |dz D]:} |\}}} |dzj} j t || ||<|d\} } d} d} d\}}}}t tD]}|\}}}}|dk(r;|r0|tdz \}}||| |}||| |}|}|}n|}|}| } | } n|| | |}|| | |}|tdz kr+||dz\}}||| |}||| |}| } | } |} |} n|}|}||||fyw) 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. rUNcJfd}\}}}}||||||fS)z0Create offset rays from segment offset vertices.c|j|r8jdzjz j}t||St||S)N)r:r"angler)v1v2rsegmentrs r2rayz3LinearTrace.faces..offset_rays..raysX::b>!)008GaK3H3O3OOe+2u55*2r22r4r/)rrleft1left2right1right2segmentsrs` r2 offset_raysz&LinearTrace.faces..offset_rayss6  3,4G+< (E5&&ue$c&&&99 9r4ct|jj|j}|dksttj|z dkr|S |j |S#t $r|cYSwxYw)z7Intersect two rays but take parallel rays into account.g*5{?)abs direction angle_betweenmathpi intersectr)ray1ray2defaultrs r2rz$LinearTrace.faces..intersectsj 44T^^DEEx3tww#7(#B ~~d++$  sA'' A54A5rr~r)rintr?z'tuple[ConstructionRay, ConstructionRay])rrrrrrr?r)rmr7r"r:rZr_normal_offset_points)!r1countrr is_closedstation start_vertexr#r$ end_vertex offset_ray1 offset_ray2prev_offset_ray1prev_offset_ray2last_up1last_up2 last_down1 last_down2rarErHrFrGlast_offset_ray1last_offset_ray2vtx0vtx1next_offset_ray1next_offset_ray2rTrVrrs! @@r2r3zLinearTrace.facess5>>H  19  : : 4 :$ ! )8 CG   QK&&..x|/B/BC UQY'G3;G3D 0L+y!'A+.55J OO%lJ YW  ($/q> [6>b\2(J s8}%A%-a[ "CeUAv:ES]UVEV9W6$&6$%5{CHD$%5{EJD $H!%JD D#. #. !!1;D !1;F3x=1$$5@Q5G2 "2 !1;F !1;D#. #. . . "dD( (M&sFF r?None)r?r{r.)rrr#rr$zOptional[float]r?rrd) rgrhri__doc__rorrrxpropertyr|rr3r/r4r2rrtsa #$$$ MQTT(-T:IT T6o)r4rc||z j}|jd}|j|dz }|j|dz }||z||z||z ||z fS)NTrU) normalize orthogonal)startendr#r$ dir_vectorortho offset_start offset_ends r2rrst+((*J  ! !$ 'E??;?3LQ/J  j  j  r4)rrceZdZdZd dZdZdZe d dZe d d dZ ddZ ddZ y )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. cg|_yr.rur0s r2rozCurvedTrace.__init__3s -/r4c,t|jSr.rqr0s r2rrzCurvedTrace.__len__6rsr4c |j|Sr.rurvs r2rxzCurvedTrace.__getitem__9ryr4cF|}|dz}tjd|j|}t|j |dtj|||D]>\\}} } t | j d} |jt || | @|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) clsspliner#r$r curve_tracertr derivativewidthnormals r2 from_splinezCurvedTrace.from_spline<s$e 1  KK6<< /*-   qA  & KE(R+ & UJ*%006F   U VU ; + r4c|jdkrtd|jd|}ttj|j dz |zddz}t |j}t|j|j|tj|||D]\}} |j|||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: rPgv@rSr)radius ValueErrormaxrceil angle_spanrcenterrr;anglesrrr) rarcr#r$rrrrrrs r2from_arczCurvedTrace.from_arcXs. ::?/ |1=> >e DIIcnnu4x?@!DqHcjj! LLE* + KK Y 6 LE5   uv~u =  r4ctj|rt}n|j|dz }|jj t ||z||z y)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 rUN) _NULLVEC2r:rrmrr%)r1rrrs r2rzCurvedTrace._append{sL   V $F%%eai0F l56>56>JKr4c#Kt|j}|dkryd}d}|jD]\}}||}|} ||||f|}|}yw)zBYields all faces as 4-tuples of :class:`~ezdxf.math.Vec2` objects.rUNrq)r1rrrrTrVs r2r3zCurvedTrace.facesseDNN# 19 ..JD$|dD( (DD)sAA Nr) rrr#rr$rrrr?r @) rrr#rr$rrrr?r )rrrrrrr?rrd) rgrhrirrorrrx classmethodrrrr3r/r4r2r r )s0#$     6           DL r4r cxeZdZdZddZdZdZddZddZddZ ddZ dd Z d dd Z d Z eddd Zy )raSequence 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! c g|_d|_yrl)_tracesrnr0s r2rozTraceBuilder.__init__s,.  r4c,t|jSr.)r7rr0s r2rrzTraceBuilder.__len__s4<<  r4c |j|Sr.)rrvs r2rxzTraceBuilder.__getitem__s||D!!r4c:|jj|y)zAppend a new trace.N)rrr1traces r2rzTraceBuilder.appends E"r4c#bK|jD]}|jEd{y7w)zaYields all faces as 4-tuples of :class:`~ezdxf.math.Vec2` objects in :ref:`OCS`. N)rr3rs r2r3zTraceBuilder.facess(\\E{{} $ $" $s #/-/c#K|jD]'}t|jfd|D)yw)zaYields all faces as 4-tuples of :class:`~ezdxf.math.Vec3` objects in :ref:`WCS`. c3`K|]%}t|j|j'ywr.rxy.0v elevations r2 z)TraceBuilder.faces_wcs..s#)RT$qssACC*CT+.N)r3tuple points_to_wcs)r1ocsrr`s ` r2 faces_wcszTraceBuilder.faces_wcss5JJLD)))RT)RRS S!s.s#!U_$qssACC";_rN)rrrrJ)r1rrrs ` r2 polygons_wcszTraceBuilder.polygons_wcss>\\E!!!UU]]_!UU "sAA Nc#hK|jD]}|j|||Ed{y7w)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)rrc)r1r]r^r_rs r2rczTraceBuilder.virtual_entitiess1*\\E--gz3G G G" Gs &202c*|j}t|dkryt|dtret|dtrQt t|j d}t t|d}|D]\}}}|j |||yyy)zEClose multi traces by merging first and last trace, if linear traces.rUNrr~)rr7 isinstancerrrr)r1tracesfirstlastrr#r$s r2closezTraceBuilder.closes v;?  fQi -*VBZ2Ufjjm4E VBZ0D16-{I   Y?273V -r4c |j}|dk(rptd|}|jj}g}|jD]/\}}}} } t ||} |r|}|} |j | || | f1|j} n5|dk(r!td|}|js td|j} |jj} |jj}g}|jD]}t |jj} |jjdr|jj }n| }|jjdr|jj"} n|} |jj$} |j | || | fntd|| r1|d d j'|d d s|j |d |}d }d }d }d }t)}|D]\}}} } |d k7rt+|||\}}}}|d kDrt-||t/j0|t/j0|d }|j2j'|r|}|}n|}|}|j t4j7||||d }| d k7rC|j8r.|j;||| |j |t)}| }|}| }|}|j;||| |j8r|j || rt=|dkDr|j?|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 LWPOLYLINErPOLYLINErz2D POLYLINE requiredr#r$zInvalid DXF type rr~gNT)is_counter_clockwiser) r]rdxf const_widthlwpointsrrclosedis_2d_polylinerXrdefault_start_widthdefault_end_widthr;locationhasattrr#r$bulger:rrrrdegrees start_pointr rr|rr7r)rpolylinerr]r pointsrrr#r$rrrrrr"r store_bulgestore_start_widthstore_end_width store_point linear_tracerr start_angle end_anglerrswews r2 from_polylinezTraceBuilder.from_polylines-""$ l "L(3H",,22KF7?7H7H31k9e1:#.K +I xiGH8I__F  "J1H** 677''F"*,,"B"B  ( > > F"++ 3 34::%%m4"(**"8"8K"5K::%%k2 & 4 4I 1I (( xiGH,/y9: : &)A,..vbz!}= MM&) $ #' "} 4: 0E; 5c!9E :6 YA:) [1 Y/-1 C..u5,.,.LL!5!5c2r8!LM! |** ,,UKKLL.#.=L# $/!"+#   $ $UK CC5;D  " " LL & c%j1n KKM r4r)rr,r?rrd)rrrrr?zIterable[Sequence[Vec3]])r?zIterable[Polygon]rerfr)rrrrr?r)rgrhrirrorrrxrr3rrrrcrrr#r/r4r2rrsk !"#%T"JNH5FH H0 @ggr4r) rrrrr#rr$rr?r()3 __future__rtypingrrrrrr r typing_extensionsr abcr collectionsr rnumpyr ezdxf.mathrrrrrrrrrezdxf.documentrezdxf.entitiesrrrrrr__all__r!r%r(__annotations__r)r*r,rrrr rr/r4r2r/s#("    &UU 8?,RS .*@A dD$./i/d^# !;< y<AAH`)-`)F    ). ;@    L q-qhC8Cr4