hQE\ddlmZddlmZmZmZddlZddlZddlm Z m Z m Z m Z m Z mZmZmZmZmZmZerddlmZdZdZdd ej2d z d f dd Zdd ej2d z ej2dz d f ddZGddZGddZGddZy)) annotations) TYPE_CHECKINGIterableOptionalN) Vec3Vec2UVecMatrix44perlinBezier4Pglobal_bspline_interpolationBSplineopen_uniform_bsplineclosed_uniform_bspline EulerSpiral) BaseLayoutc<|dz tj|zz SN@)random) max_values 3lib/python3.12/site-packages/ezdxf/render/curves.pyrndrs s?V]]_y8 88cltj|j|j}|dz ||zz Sr)r snoise2xy)rwalkerrs r rnd_perlinr!s-vxx*A s?Q] **rdg?c#0 K||z fd}tdd}|}t|D]f}||zdk(r ||z}||z j}|t||z} |t jz} |tj | | z}|hyw)u]Returns a random 2D path as iterable of :class:`~ezdxf.math.Vec2` objects. Args: steps: count of vertices to generate max_step_size: max step size max_heading: limit heading angle change per step to ± max_heading/2 in radians retarget: specifies steps before changing global walking target cBtttfSN)rrmax_srnext_global_targetz*random_2d_path..next_global_target5sSYD *++rrN)rrangeangler!r from_angle) steps max_step_size max_headingretargetr*rtargetir,headinglengthr)s @rrandom_2d_pathr6"s" 5 D,!QZF  !F 5\ x<1 022F&''*[&990$//'6:: sBBrg @c#K||zfd}t}|}t|D]}||zdk(r ||z}||z j} |tjz} | t ||z} tj | | } t ||} |t j| j| z }|yw)uReturns a random 3D path as iterable of :class:`~ezdxf.math.Vec3` objects. Args: steps: count of vertices to generate max_step_size: max step size max_heading: limit heading angle change per step to ± max_heading/2, rotation about the z-axis in radians max_pitch: limit pitch angle change per step to ± max_pitch/2, rotation about the x-axis in radians retarget: specifies steps before changing global walking target cVttttfSr')rrr(srr*z*random_3d_path..next_global_targetZs!SYD 3t9566rrN) rr+r,rr!r-r x_rotate transform)r.r/r0 max_pitchr1r*rr2r3r,r5 heading_angle next_step pitch_angler)s @rrandom_3d_pathr?Ds( 5 D7VF  !F 5\ x<1 022F&''0 ; ?? OOM6:  F3 (##K0::9EE sCCcpeZdZdZGddZd dZd dZ d d dZddZ d dd Z y)Beziera>Render a bezier curve as 2D/3D :class:`~ezdxf.entities.Polyline`. The :class:`Bezier` class is implemented with multiple segments, each segment is an optimized 4 point bezier curve, the 4 control points of the curve are: the start point (1) and the end point (4), point (2) is start point + start vector and point (3) is end point + end vector. Each segment has its own approximation count. .. seealso:: The new :mod:`ezdxf.path` package provides many advanced construction tools based on the :class:`~ezdxf.path.Path` class. c0eZdZ ddZddZy)Bezier.Segmentct||_t||_t||_t||_||_yr')rstartend start_tangent end_tangentsegments)selfrErFrGrHrIs r__init__zBezier.Segment.__init__|s@eDJCyDH!%"D  $K0D $DMrc|j|j|jz|j|jz|jg}t |}|j |j Sr')rErGrFrHr approximaterI)rJcontrol_pointsbeziers rrMzBezier.Segment.approximates\  T///4+++ N n-F%%dmm4 4rN) rEr rFr rGr rHr rIint)returnIterable[Vec3])__name__ __module__ __qualname__rKrMrrSegmentrC{s< % % %  %   %   %  5rrWcg|_yr')points)rJs rrKzBezier.__init__s  rcT|jjt|d|dfy)zSet start point and start tangent. Args: point: start point tangent: start tangent as vector, example: (5, 0, 0) means a horizontal tangent with a length of 5 drawing units N)rYappendr)rJpointtangents rrEz Bezier.starts# DKw=>rNct|}|| }n t|}|jjt|||t|fy)aAppend a control point with two control tangents. Args: point: control point tangent1: first tangent as vector "left" of the control point tangent2: second tangent as vector "right" of the control point, if omitted `tangent2` = `-tangent1` segments: count of line segments for the polyline approximation, count of line segments from the previous control point to the appended control point. N)rrYr[rP)rJr\tangent1tangent2rIs rr[z Bezier.appendsF&>   yHH~H DK8S]KLrc#Kt|jdkDrct|jdd|jddD]9\}}|d}|d}|d}|d}|d}tj |||||;yt dw)Nrr#zTwo or more points needed!)lenrYziprArW ValueError)rJ from_pointto_point start_pointrG end_pointrHcounts r_build_bezier_segmentszBezier._build_bezier_segmentss t{{ a (+DKK,z Bezier.render..s/1!A$s dxfattribsN)rmextendrManyadd_polyline3dadd_polyline2d)rJlayoutforce3drtrYsegments rrenderz Bezier.rendersg 224G MM'--/ 05 c///  ! !&Z ! @  ! !&Z ! @r)rQNone)r\r r]r rQr})Nr$)r\r r_r r`zOptional[UVec]rIrP)rQzIterable[Segment])FN)ryrrzboolrQr}) rSrTrU__doc__rWrKrEr[rmr|rVrrrArAks 556 ?$( MMM! M  M4 ;" AAA  ArrAceZdZdZ d d dZdddZ d ddZeZ d ddZ d ddZ d ddZ d dd Z d dd Z d dd Z y)Splinea-This class can be used to render B-splines into DXF R12 files as approximated :class:`~ezdxf.entities.Polyline` entities. The advantage of this class over the :class:`R12Spline` class is, that this is a real 3D curve, which means that the B-spline vertices do have to be located in a flat plane, and no :ref:`UCS` class is needed to place the curve in 3D space. .. seealso:: The newer :class:`~ezdxf.math.BSpline` class provides the advanced vertex interpolation method :meth:`~ezdxf.math.BSpline.flattening`. Nc`|g}tj||_t||_y)z Args: points: spline definition points segments: count of line segments for approximation, vertex count is `segments` + 1 N)rlistrYrPrI)rJrYrIs rrKzSpline.__init__s) >F"&))F"3 H  rcDt|jdz |z|_y)adCalculate overall segment count, where segments is the sub-segment count, `segments` = 4, means 4 line segments between two definition points e.g. 4 definition points and 4 segments = 12 overall segments, useful for fit point rendering. Args: segments: sub-segments count between two definition points rbN)rerYrI)rJrIs r subdividezSpline.subdividesT[[)A-9 rct|j||}t|j|j}t d|Dr|j ||y|j||y)aRender a B-spline as 2D/3D :class:`~ezdxf.entities.Polyline`, where the definition points are fit points. - 2D spline vertices uses: :meth:`~ezdxf.layouts.BaseLayout.add_polyline2d` - 3D spline vertices uses: :meth:`~ezdxf.layouts.BaseLayout.add_polyline3d` Args: layout: :class:`~ezdxf.layouts.BaseLayout` object degree: degree of B-spline (order = `degree` + 1) method: "uniform", "distance"/"chord", "centripetal"/"sqrt_chord" or "arc" calculation method for parameter t dxfattribs: DXF attributes for :class:`~ezdxf.entities.Polyline` )degreemethodc3:K|]}|jdk7yw)gN)z)rpvertexs rrrz.Spline.render_as_fit_points..(s6X6vxx3XsrsN)r rYrrMrIrvrwrx)rJryrrrtsplineverticess rrender_as_fit_pointszSpline.render_as_fit_pointssh*. KKv **4==9: 6X6 6  ! !(z ! B  ! !(z ! Brct|j|dz}|jt|j |j |y)aXRender an open uniform B-spline as 3D :class:`~ezdxf.entities.Polyline`. Definition points are control points. Args: layout: :class:`~ezdxf.layouts.BaseLayout` object degree: degree of B-spline (order = `degree` + 1) dxfattribs: DXF attributes for :class:`~ezdxf.entities.Polyline` rborderrsNrrYrwrrMrIrJryrrtrs rrender_open_bsplinezSpline.render_open_bspline/sDFQJ7 ##DMM2 3   rct|j|dz}|jt|j |j |y)aRRender a uniform B-spline as 3D :class:`~ezdxf.entities.Polyline`. Definition points are control points. Args: layout: :class:`~ezdxf.layouts.BaseLayout` object degree: degree of B-spline (order = `degree` + 1) dxfattribs: DXF attributes for :class:`~ezdxf.entities.Polyline` rbrrsN)rrYrwrrMrIrs rrender_uniform_bsplinezSpline.render_uniform_bspline@sD&dkk!D ##DMM2 3   rct|j|dz}|jt|j |j |y)aYRender a closed uniform B-spline as 3D :class:`~ezdxf.entities.Polyline`. Definition points are control points. Args: layout: :class:`~ezdxf.layouts.BaseLayout` object degree: degree of B-spline (order = `degree` + 1) dxfattribs: DXF attributes for :class:`~ezdxf.entities.Polyline` rbrrsNrrYrwrrMrIrs rrender_closed_bsplinezSpline.render_closed_bsplineQsD( 6A:F ##DMM2 3   rct|j|dz|}|jt|j |j |y)aRender a rational open uniform BSpline as 3D :class:`~ezdxf.entities.Polyline`. Definition points are control points. Args: layout: :class:`~ezdxf.layouts.BaseLayout` object weights: list of weights, requires a weight value (float) for each definition point. degree: degree of B-spline (order = `degree` + 1) dxfattribs: DXF attributes for :class:`~ezdxf.entities.Polyline` rbrweightsrsNrrJryrrrtrs rrender_open_rbsplinezSpline.render_open_rbsplinebsF$FQJH ##DMM2 3   rct|j|dz|}|jt|j |j |y)aRender a rational uniform B-spline as 3D :class:`~ezdxf.entities.Polyline`. Definition points are control points. Args: layout: :class:`~ezdxf.layouts.BaseLayout` object weights: list of weights, requires a weight value (float) for each definition point. degree: degree of B-spline (order = `degree` + 1) dxfattribs: DXF attributes for :class:`~ezdxf.entities.Polyline` rbrrsNrrs rrender_uniform_rbsplinezSpline.render_uniform_rbsplineyK$( KKvz7   ##DMM2 3   rct|j|dz|}|jt|j |j |y)aRender a rational B-spline as 3D :class:`~ezdxf.entities.Polyline`. Definition points are control points. Args: layout: :class:`~ezdxf.layouts.BaseLayout` object weights: list of weights, requires a weight value (float) for each definition point. degree: degree of B-spline (order = `degree` + 1) dxfattribs: DXF attributes for :class:`~ezdxf.entities.Polyline` rbrrsNrrs rrender_closed_rbsplinezSpline.render_closed_rbsplinerr)Nr")rYzOptional[Iterable[UVec]]rIrP))rIrPrQr})rdchordN) ryrrrPrstrrtzOptional[dict]rQr})rdN)ryrrrPrQr})ryrrzIterable[float]rrPrQr})rSrTrUrrKrrr|rrrrrrrVrrrrsw HK &. &AD & :%) CCC C # C  C<"F?C   *-  $?C   *-  $?C   *-  *   !     6   !     :   !     rrc`eZdZdZdddZ d d dZ d d dZy) rzRender an `euler spiral `_ as a 3D :class:`~ezdxf.entities.Polyline` or a :class:`~ezdxf.entities.Spline` entity. This is a parametric curve, which always starts at the origin (0, 0). c6tt||_y)zC Args: curvature: Radius of curvature N) _EulerSpiralfloatspiral)rJ curvatures rrKzEulerSpiral.__init__s #5#34 rNc|jj||}||j|}|jt ||S)a Render curve as :class:`~ezdxf.entities.Polyline`. Args: layout: :class:`~ezdxf.layouts.BaseLayout` object length: length measured along the spiral curve from its initial position segments: count of line segments to use, vertex count is `segments` + 1 matrix: transformation matrix as :class:`~ezdxf.math.Matrix44` dxfattribs: DXF attributes for :class:`~ezdxf.entities.Polyline` Returns: :class:`~ezdxf.entities.Polyline` rs)rrMtransform_verticesrwr)rJryr5rImatrixrtrYs rrender_polylinezEulerSpiral.render_polylinesJ*((:  ..v6F$$T&\j$IIrc|jj|||}|j}||j|}|j ||j |j |S)a Render curve as :class:`~ezdxf.entities.Spline`. Args: layout: :class:`~ezdxf.layouts.BaseLayout` object length: length measured along the spiral curve from its initial position fit_points: count of spline fit points to use degree: degree of B-spline matrix: transformation matrix as :class:`~ezdxf.math.Matrix44` dxfattribs: DXF attributes for :class:`~ezdxf.entities.Spline` Returns: :class:`~ezdxf.entities.Spline` )r)rNrknotsrt)rbsplinerNradd_open_splinerr) rJryr5 fit_pointsrrrtrrYs r render_splinezEulerSpiral.render_splinesn0$$VZ$G&&  ..v6F%%!==,,.! &  r)rb)rr)rbr"NN)ryrr5rrIrPrOptional[Matrix44])rb rdNN) ryrr5rrrPrrPrr)rSrTrUrrKrrrVrrrrs5%) JJJ J # J:%)! ! !  !  ! # ! rr) r.rPr/rr0rr1rPrQzIterable[Vec2]) r.rPr/rr0rr;rr1rPrQrR) __future__rtypingrrrrmath ezdxf.mathrrr r r r r rrrrr ezdxf.layoutsrrr!pir6r?rArrVrrrs #44     (9+ 1     F3ww} $ $$$ $  $  $NwAwAtD D NL L r