mjzZUddlmZddlmZmZmZmZmZmZm 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<GddZ0Gdde0Z1 d dZ2edZ3Gdde0Z4d!dZ5GddeZ6y)") 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 Quadrilateralc8eZdZeddZddZ d ddZy) AbstractTracecyNselfs ?/DATA/.local/lib/python3.12/site-packages/ezdxf/render/trace.pyfaceszAbstractTrace.faces2s c  dd}g}g}|jD]-\}}}}|j||f|j||f/t||}|jtt|||S)Nc3Kt|syt|}t|}||D]}|j|r||}ywr1)leniternextisclose)vertices _verticesprevr%s r5mergez$AbstractTrace.polygon..merge;sGx=XI ?DJ#||F+ L!D$s =A  A )r>r,returnzIterable[UVec])r6extendlistreversed) r4rAforward_contourbackward_contourup1down1down2up2contours r5polygonzAbstractTrace.polygon:s "')')&*jjl "Cs  " "E5> 2  # #S#J /'3u_-.xU+;%< =>?r7Nc#<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 .vtxrRvtx2vtx3N)ezdxf.entities.factoryrO TypeErrordictr6rangeentitydbadd)r4dxftype dxfattribsdocrOfaceientitys r5virtual_entitieszAbstractTrace.virtual_entitiesQs" / 6 6.wiq9: :**+ JJLD1X(,Q S9%("%)!W 6"%)!W 6"*c2F   (L!sBBrBzIterable[Face])rBr,rQNNrbzOptional[Drawing]rBzIterable[Quadrilateral])__name__ __module__ __qualname__r r6rMrfr2r7r5r/r/1s5  0JN!5F! !r7r/cXeZdZdZd dZdZdZed dZ d d dZ d dZ y)r"ztLinear 2D banded lines like polylines with start- and end width. Accepts 3D input, but z-axis is ignored. c g|_d|_yNg-q=) _stationsabs_tolr3s r5__init__zLinearTrace.__init__|s.0 r7c,t|jSr1r:rpr3s r5__len__zLinearTrace.__len__4>>""r7c |j|Sr1rpr4items r5 __getitem__zLinearTrace.__getitem__~~d##r7c,t|jS)z%`True` if at least one station exist.)boolrpr3s r5 is_startedzLinearTrace.is_startedsDNN##r7Nc  ||}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)rq) rrpr~r%r=rqpopappendr$float)r4pointr&r'stationss r5 add_stationzLinearTrace.add_stationsp"  #IU >> >hrl1199%9V LLN eU;-?yAQRSr7c # 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. rXNcJfd}\}}}}||||||fS)z0Create offset rays from segment offset vertices.c|j|r8jdzjz j}t||St||S)N)r=r%angler)v1v2rsegmentrs r5rayz3LinearTrace.faces..offset_rays..raysX::b>!)008GaK3H3O3OOe+2u55*2r22r7r2)rrleft1left2right1right2segmentsrs` r5 offset_raysz&LinearTrace.faces..offset_rayss6  3,4G+< (E5&&ue$c&&&99 9r7ct|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 r5rz$LinearTrace.faces..intersectsj 44T^^DEEx3tww#7(#B ~~d++$  sA'' A54A5rrr)rintrBz'tuple[ConstructionRay, ConstructionRay])rrrrrrrBr)rpr:r%r=r]r_normal_offset_points)!r4countrr is_closedstation start_vertexr&r' end_vertex offset_ray1 offset_ray2prev_offset_ray1prev_offset_ray2last_up1last_up2 last_down1 last_down2rdrHrKrIrJlast_offset_ray1last_offset_ray2vtx0vtx1next_offset_ray1next_offset_ray2rWrYrrs! @@r5r6zLinearTrace.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 rBNone)rBr~r1)rrr&rr'zOptional[float]rBrrg) rjrkrl__doc__rrrur{propertyrrr6r2r7r5r"r"usa #$$$ MQTT(-T:IT T6o)r7r"c||z j}|jd}|j|dz }|j|dz }||z||z||z ||z fS)NTrX) normalize orthogonal)startendr&r' dir_vectorortho offset_start offset_ends r5rrst+((*J  ! !$ 'E??;?3LQ/J  j  j  r7)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|_yr1rxr3s r5rrzCurvedTrace.__init__4s -/r7c,t|jSr1rtr3s r5ruzCurvedTrace.__len__7rvr7c |j|Sr1rxrys r5r{zCurvedTrace.__getitem__:r|r7cF|}|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 r5 from_splinezCurvedTrace.from_spline=s$e 1  KK6<< /*-   qA  & KE(R+ & UJ*%006F   U VU ; + r7c|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: rSgv@rVr)radius ValueErrormaxrceil angle_spanrcenterrr>anglesrrr) rarcr&r'rrrrrrs r5from_arczCurvedTrace.from_arcYs. ::?/ |1=> >e DIIcnnu4x?@!DqHcjj! LLE* + KK Y 6 LE5   uv~u =  r7ctj|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 rXN) _NULLVEC2r=rrprr()r4rrrs r5rzCurvedTrace._append|sL   V $F%%eai0F l56>56>JKr7c#Kt|j}|dkryd}d}|jD]\}}||}|} ||||f|}|}yw)zBYields all faces as 4-tuples of :class:`~ezdxf.math.Vec2` objects.rXNrt)r4rrrrWrYs r5r6zCurvedTrace.facesseDNN# 19 ..JD$|dD( (DD)sAA Nr) rrr&rr'rrrrBr#@) rrr&rr'rrrrBr#)rrrrrrrBrrg) rjrkrlrrrrur{ classmethodrrrr6r2r7r5r#r#*s0#$     6           DL r7r#c,|tzry|tzryy)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) vertex_flagss r5#should_include_vertex_for_renderingrs#"..44 r7cxeZdZdZddZdZdZddZddZddZ ddZ dd Z d dd Z d Z eddd Zy )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! c g|_d|_yro)_tracesrqr3s r5rrzTraceBuilder.__init__s,.  r7c,t|jSr1)r:rr3s r5ruzTraceBuilder.__len__s4<<  r7c |j|Sr1)rrys r5r{zTraceBuilder.__getitem__s||D!!r7c:|jj|y)zAppend a new trace.N)rrr4traces r5rzTraceBuilder.appends E"r7c#bK|jD]}|jEd{y7w)zaYields all faces as 4-tuples of :class:`~ezdxf.math.Vec2` objects in :ref:`OCS`. N)rr6rs r5r6zTraceBuilder.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'ywr1rxy.0v elevations r5 z)TraceBuilder.faces_wcs..s#)RT$qssACC*CT+.N)r6tuple points_to_wcs)r4ocsrrcs ` r5 faces_wcszTraceBuilder.faces_wcss5JJLD)))RT)RRS S!s.s#!U_$qssACC";_rN)rrrrM)r4rrrs ` r5 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)rrf)r4r`rarbrs r5rfzTraceBuilder.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.rXNrr)rr: isinstancer"rrr)r4tracesfirstlastrr&r's r5closezTraceBuilder.close s v;?  fQi -*VBZ2Ufjjm4E VBZ0D16-{I   Y?273V -r7c |j}|dk(rptd|}|jj}g}|jD]/\}}}} } t ||} |r|}|} |j | || | f1|j} ni|dk(rUtd|}|js td|j} |jj} |jj}|jDcgc])}t|jjddr|+}}g}|D]}t |jj } |jj#dr|jj$}n| }|jj#d r|jj&} n|} |jj(} |j | || | fntd || r1|ddj+|d ds|j |d|}d }d }d }d }t-}|D]\}}} } |d k7rt/|||\}}}}|dkDrt1||t3j4|t3j4|d}|j6j+|r|}|}n|}|}|j t8j;||||d }| d k7rC|j<r.|j?||| |j |t-}| }|}| }|}|j?||| |j<r|j || rtA|dkDr|jC|Scc}w)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 requiredflagsrr&r'zInvalid DXF type rgNT)is_counter_clockwiser)"r`rdxf const_widthlwpointsrrclosedis_2d_polyliner[rdefault_start_widthdefault_end_widthr>rgetlocationhasattrr&r'bulger=r"rrrdegrees start_pointr#rrrr:r )rpolylinerr`rpointsrrr&r'rrrrrr%filtered_verticesr store_bulgestore_start_widthstore_end_width store_point linear_tracerr start_angle end_anglerrswews r5 from_polylinezTraceBuilder.from_polylinesn""$ l "L(3H",,22KF7?7H7H31k9e1:#.K +I xiGH8I__F  "J1H** 677''F"*,,"B"B  ( > > &.%6%6!%666vzz~~gq7QR%6 ! 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 S!s5.Nr)rr/rBrrg)rrrrrBzIterable[Sequence[Vec3]])rBzIterable[Polygon]rhrir)r rrrrBr!)rjrkrlrrrrur{rr6rrrrfr rr,r2r7r5r!r!sk !"#%T"JNH5FH H0 @mmr7r!) rrrrr&rr'rrBr+)rrrBr~)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!r2r7r5r9s#(" V   &UU 8?,RS .*@A dD$./i/d^# !;< y<AAH`)-`)F    ). ;@    L q-qh>I8Ir7