'jYC*ddlmZddlmZmZmZddlZddlZddlm Z m Z m Z m Z m Z mZmZmZmZmZmZerddlmZdZdZdd ejd z d fd dZdd ejdz ejdz d fd!dZGddZGddZGddZdS)") annotations) TYPE_CHECKINGIterableOptionalN) Vec3Vec2UVecMatrix44perlinBezier4Pglobal_bspline_interpolationBSplineopen_uniform_bsplineclosed_uniform_bspline EulerSpiral) BaseLayoutc:|dz tj|zz SN@)random) max_values M/DATA/AppData/hermes/venv/lib/python3.11/site-packages/ezdxf/render/curves.pyrndrs s?V]__y8 88cVtj|j|j}|dz ||zz Sr)r snoise2xy)rwalkerrs r rnd_perlinr!s*vx**A s?Q] **rdg?stepsint max_step_sizefloat max_headingretargetreturnIterable[Vec2]c#B K||z fd}tdd}|}t|D]g}||zdkr ||z}||z j}|t||z} |t jz} |tj| | z}|VhdS)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 cZtttfSN)rrmax_srnext_global_targetz*random_2d_path..next_global_target5s"SYYD *+++rrN)rrangeangler!r from_angle) r%r'r)r*r2rtargetir4headinglengthr1s @rrandom_2d_pathr:"s" 5 D,,,,,!QZZF   ! !F 5\\ x<1  00222F&'*[&9990$/'6::: rrg @ max_pitchIterable[Vec3]c#K||zfd}t}|}t|D]}||zdkr ||z}||z j} |tjz} | t ||z} tj| | } t ||} |t j| | z }|VdS)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/)rrr0srr2z*random_3d_path..next_global_targetZs*SYYD 3t995666rrN) rr3r4rr!r5r x_rotate transform)r%r'r)r;r*r2rr6r7r4r9 heading_angle next_step pitch_angler1s @rrandom_3d_pathrDDs( 5 D77777VVF   ! !F 5\\   x<1  00222F&'0 ; ? ?? OM6::  F33 (#K00::9EEE   rc`eZdZdZGddZddZdd Z dddZddZ d d!dZ d S)"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. ceZdZd dZdd Zd S)Bezier.Segmentstartr end start_tangent end_tangentsegmentsr&ct||_t||_t||_t||_||_dSr/)rrIrJrKrLrM)selfrIrJrKrLrMs r__init__zBezier.Segment.__init__|sPeDJCyyDH!%""D  $K00D $DMMMrr+r<c|j|j|jz|j|jz|jg}t |}||jSr/)rIrKrJrLr approximaterM)rOcontrol_pointsbeziers rrRzBezier.Segment.approximatesR  T//4++ N n--F%%dm44 4rN) rIr rJr rKr rLr rMr&)r+r<)__name__ __module__ __qualname__rPrRrrSegmentrH{s< % % % %  5 5 5 5 5 5rrYr+Nonecg|_dSr/)points)rOs rrPzBezier.__init__s  rpointr tangentc\|jt|d|dfdS)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)r\appendr)rOr]r^s rrIz Bezier.starts/ DKKw=>>>>>rNr$tangent1tangent2Optional[UVec]rMr&ct|}|| }nt|}|jt|||t|fdS)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)rr\r`r&)rOr]rarbrMs rr`z Bezier.appendsZ&>>   yHHH~~H DKK8S]]KLLLLLrIterable[Segment]c#HKt|jdkrzt|jdd|jddD]M\}}|d}|d}|d}|d}|d}t|||||VNdSt d)Nrr#zTwo or more points needed!)lenr\ziprFrY ValueError)rO from_pointto_point start_pointrK end_pointrLcounts r_build_bezier_segmentszBezier._build_bezier_segmentss t{  a  (+DK,z Bezier.render..s&//1!A$//////r dxfattribsN)rrextendrRanyadd_polyline3dadd_polyline2d)rOrsrtr|r\segments rrenderz Bezier.renders 2244 1 1G MM'--// 0 0 0 0  Ac/////// A  ! !&Z ! @ @ @ @ @  ! !&Z ! @ @ @ @ @r)r+rZ)r]r r^r r+rZ)Nr$)r]r rar rbrcrMr&)r+re)FN)rsrrtrur+rZ) rUrVrW__doc__rYrPrIr`rrrrXrrrFrFks  555555556 ????$( MMMMM4 ; ; ; ;" AAAAAAArrFceZdZdZ dd dZd!d"d Z d#d$dZeZ d%d&dZ d%d&dZ d%d&dZ d%d'dZ d%d'dZ d%d'dZ dS)(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`. Nr"r\Optional[Iterable[UVec]]rMr&ch|g}tj||_t||_dS)z Args: points: spline definition points segments: count of line segments for approximation, vertex count is `segments` + 1 N)rlistr\r&rM)rOr\rMs rrPzSpline.__init__s0 >F"&)F"3"3 H  rr+rZcDt|jdz |z|_dS)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 rgN)rjr\rM)rOrMs r subdividezSpline.subdivides#T[))A-9 rrichordrsrdegreemethodstrr|Optional[dict]ct|j||}t||j}t d|Dr|||dS|||dS)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` )rrc3,K|]}|jdkVdS)gN)z)rxvertexs rrzz.Spline.render_as_fit_points..(s(666vx3666666rr{N)r r\rrRrMr~rr)rOrsrrr|splineverticess rrender_as_fit_pointszSpline.render_as_fit_pointss*. Kv   **4=99:: 66X666 6 6 C  ! !(z ! B B B B B  ! !(z ! B B B B Brct|j|dz}|t||j|dS)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` rgorderr{Nrr\rrrRrMrOrsrr|rs rrender_open_bsplinezSpline.render_open_bspline/s`FQJ777 ##DM22 3 3       rct|j|dz}|t||j|dS)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` rgrr{N)rr\rrrRrMrs rrender_uniform_bsplinezSpline.render_uniform_bspline@s`&dk!DDD ##DM22 3 3       rct|j|dz}|t||j|dS)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` rgrr{Nrr\rrrRrMrs rrender_closed_bsplinezSpline.render_closed_bsplineQs`( 6A:FFF ##DM22 3 3       rweightsIterable[float]ct|j|dz|}|t||j|dS)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` rgrrr{NrrOrsrrr|rs rrender_open_rbsplinezSpline.render_open_rbsplinebsb$FQJHHH ##DM22 3 3       rct|j|dz|}|t||j|dS)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` rgrr{Nrrs rrender_uniform_rbsplinezSpline.render_uniform_rbsplineyk$( Kvz7     ##DM22 3 3       rct|j|dz|}|t||j|dS)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` rgrr{Nrrs rrender_closed_rbsplinezSpline.render_closed_rbsplinerr)Nr")r\rrMr&)r)rMr&r+rZ)rirN) rsrrr&rrr|rr+rZ)riN)rsrrr&r+rZ)rsrrrrr&r+rZ)rUrVrWrrPrrrrrrrrrrXrrrrsG  HK & & & & & : : : : :%) CCCCC<"F?C     $?C     $?C     *      6      :        rrcBeZdZdZdddZ dddZ dddZdS)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). rg curvaturer(cHtt||_dS)zC Args: curvature: Radius of curvature N) _EulerSpiralr(spiral)rOrs rrPzEulerSpiral.__init__s #5#3#344 rr"Nrsrr9rMr&matrixOptional[Matrix44]c|j||}|||}|t ||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` Nr{)rrRtransform_verticesrr)rOrsr9rMrr|r\s rrender_polylinezEulerSpiral.render_polylinesS*((::  ..v66F$$T&\\j$IIIr ri fit_pointsrc|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` )rN)rSrknotsr|)rbsplinerSradd_open_splinerr) rOrsr9rrrr|rr\s r render_splinezEulerSpiral.render_splinest0$$VZ$GG&  ..v66F%%!=,,..! &   r)rg)rr()rgr"NN)rsrr9r(rMr&rr)rgrriNN) rsrr9r(rr&rr&rr)rUrVrWrrPrrrXrrrrs55555%) JJJJJ:%)! ! ! ! ! ! ! rr) r%r&r'r(r)r(r*r&r+r,) r%r&r'r(r)r(r;r(r*r&r+r<) __future__rtypingrrrrmath ezdxf.mathrrr r r r r rrrrr ezdxf.layoutsrrr!pir:rDrFrrXrrrs#"""""4444444444                           )((((((999+++ 1 F3w} $$$$$NwAwAwAwAwAwAwAwAtD D D D D D D D NL L L L L L L L L L r