h,l`dZddlmZddlmZmZmZmZddlm Z m Z m Z ddl m Z ddlmZddlmZmZmZmZmZmZer ddlmZdd lmZd Zd Zd e z Zd e z ZdZGdde Z!GddZ"e"Z#GddZ$Gdde$Z%Gdde$Z&Gdde&Z'Gdde$Z(ddZ)y) aDimension lines as composite entities build up by DXF primitives. This add-on exist just for an easy transition from `dxfwrite` to `ezdxf`. Classes ------- - LinearDimension - AngularDimension - ArcDimension - RadiusDimension - DimStyle This code was written long before I had any understanding of the DIMENSION entity and therefore, the classes have completely different implementations and styling features than the dimensions based on the DIMENSION entity . .. warning:: Try to not use these classes beside porting `dxfwrite` code to `ezdxf` and even for that case the usage of the regular DIMENSION entity is to prefer because this module will not get much maintenance and may be removed in the future. ) annotations)Any TYPE_CHECKINGIterableOptional)radiansdegreespi)abstractmethod)TextEntityAlignment)Vec3Vec2distancelerpConstructionRayUVec)Drawing)GenericLayoutType皙?.f@gi@?c<eZdZdZgdZdfd ZddZddZxZS) DimStylezBDimStyle parameter struct, a dumb object just to store values ))tick DIMTICK_ARCH) tickfactorr)tick2xF)scalegY@)roundvalr) roundhalfF) textcolor)height?)prefix)suffixr')stylezOpenSansCondensed-Light)layer DIMENSIONS) dimlinecolorr#) dimlineext333333?) textabove皙?) dimextlineT)dimextlinecolor) dimextlinegapr.c lt|tj||d<|j |y)Nname)super__init__rdefault_valuesupdate)selfr6kwargs __class__s 5lib/python3.12/site-packages/ezdxf/addons/dimlines.pyr8zDimStyle.__init__\s+ 001V  Fc ||SN)r;attrs r> __getattr__zDimStyle.__getattr__bs Dzr?c|||<yrArB)r;rCvalues r> __setattr__zDimStyle.__setattr__es T r?)r6str)rCrHreturnr)rCrHrFrrINone) __name__ __module__ __qualname____doc__r9r8rDrG __classcell__r=s@r>rr/s'NR r?rc:eZdZdZddZddZddZed dZy) DimStylesz DimStyle container c i|_td|_|jdtt dddddd |jd t d ddddd |jd td ddddd y)NDefault angle.deg°rDIMTICK_RADIUSTF)rr(r rrr-r1z angle.gradgonz angle.radrad)_stylesrdefaultnew ANGLE_DEGrH ANGLE_GRAD ANGLE_RADr;s r>r8zDimStyles.__init__os,.  *   t9!    !    !  r?cN|jj||jS)z0 Get DimStyle() object by name. )r\getr])r;r6s r>rdz DimStyles.gets||dll33r?c <t|fi|}||j|<|S)z' Create a new dimstyle )rr\)r;r6r<r)s r>r^z DimStyles.news&((" T r?cFtjjdd}ddd}ddd}|jj d}|j dd||j d d ||jj d }|j dd||j d d ||jj d}|j dd||jgd||jj d}|jgd|y)zy Insert necessary definitions into drawing: ticks: DIMTICK_ARCH, DIMTICK_DOT, DIMTICK_ARROW BYBLOCKcolorr*r#r)rXr%)rXg࿩startend dxfattribs)皙ɿro)r0r0 DIMTICK_DOTrrg?)centerradiusrn DIMTICK_ARROW)rqr.rr.g)rnrW)rqru)g?rXrvN) dimstylesr]r2blocksr^add_line add_circle add_solid)drawingdimcolorcolor4color7blocks r>setupzDimStyles.setups&&&66 y1y1"">2 [khO \{vN""=1 Z[XN svF""?3 Z\hO ;O""#34  <  r?NrIrJ)r6rHrIr)r|z 'Drawing') rKrLrMrNr8rdr^ staticmethodrrBr?r>rRrRis+ ! F4 ( ( r?rRc:eZdZdZddZddZd dZed dZy) _DimensionBasez3 Abstract base class for dimension lines. cTtj||_||_||_yrA)rwrddimstyler*r )r;rr*r s r>r8z_DimensionBase.__init__s ! h/    r?c|dk(r.|j |jS|jjS|dk(r.|j |jS|jjS|j|S)zw Get dimension line properties by `property_name` with the possibility to override several properties. r*r )r*rr )r; property_names r>propz_DimensionBase.propsu G #!%!74:: PT]]=P=P P j (==,  ]]++  ==/ /r?cdt|jdzdz}||z}t|vr&|jd|jt|jd|z|jdzS)z, Format the dimension text. z%.r f0r&r()rHrDIMENSIONS_FLOATINGPOINTrstrip)r;dimvalue dimtextfmtdimtexts r>format_dimtextz_DimensionBase.format_dimtextsnC * 566< x' #w . NN3  NN3 4yy"W,tyy/BBBr?cyrArBr;layouts r>renderz_DimensionBase.renders r?N)rrHr*rHr int)rrHrIr)rfloatrIrH)r'GenericLayoutType') rKrLrMrNr8rrr rrBr?r>rrs+ ! 0 C  r?rceZdZdZ d dfd ZddZddZddZddZddZ e ddZ e dd Z dd Z edd Zdd Zedd ZddZddZddZddZd dZddZxZS)!LinearDimensionz Simple straight dimension line with two or more measure points, build with basic DXF entities. This is NOT a dxf dimension entity. And This is a 2D element, so all z-values will be ignored! ct||||||_t||_dg|j z|_t||_d|_ y)a LinearDimension Constructor. Args: pos: location as (x, y) tuple of dimension line, line goes through this point measure_points: list of points as (x, y) tuples to dimension (two or more) angle: angle (in degree) of dimension line dimstyle: dimstyle name, 'Default' - style is the default value layer: dimension line layer, override the default value of dimstyle roundval: count of decimal places r'N) r7r8anglelistmeasure_points section_count text_overrider dimlineposr)r;posrrrr*r r=s r>r8zLinearDimension.__init__sQ* 5(3 ">2 TD$6$66s) r?c"||j|<y)zg Set and override the text of the dimension text for the given dimension line section. N)r)r;sectiontexts r>set_textzLinearDimension.set_texts'+7#r?cz|jDcgc] }t|c}|_t|jt |j }|jDcgc]}|j ||c}|_|j|j|_ |jycc}wcc}w)z` Calc setup values and determines the point order of the dimension line points. )rN) rr rrrr_get_point_on_dimlinedimline_points_indices_of_sorted_points point_order_build_vectors)r;point dimlinerays r>_setupzLinearDimension._setup$s9=8K8KL8KutE{8KL$T__GDJJ_get_dimline_pointz"LinearDimension._get_dimline_point1s ""4#3#3E#:;;r?cL|j||j|dzfS)zU Get start and end point on the dimension line of dimension section. r)r;rs r>_get_section_pointsz#LinearDimension._get_section_points7s0&&w/1H1H aK2   r?cF|jd|jdfS)zE Get the first and the last point of dimension line. rrrbs r>_get_dimline_boundsz#LinearDimension._get_dimline_bounds?s%&&q)4+B+B2+FFFr?c2t|jdz S)zcount of dimline sectionsrlenrrbs r>rzLinearDimension.section_countEs4&&'!++r?c,t|jS)zcount of dimline pointsrrbs r> point_countzLinearDimension.point_countJs4&&''r?c|j|j||jdr|j||j ||j |y3build dimension line object with basic dxf entitiesr1N)r _draw_dimliner_draw_extension_lines _draw_text _draw_ticksrs r>rzLinearDimension.renderOsM  6" 99\ "  & &v .   r?ct|Dcgc] \}}||f }}}|j|Dcgc]\}}| c}}Scc}}wcc}}w)z7get indices of points, for points sorted by x, y values) enumeratesort)pointsidxrindexed_pointss r>rz)LinearDimension._indices_of_sorted_pointsXsV:C69JK9J:35#,9JK&45n sn55L5s A A c|j\}}t|t|z j|_|jj |_y)z9build unit vectors, parallel and normal to dimension lineN)rr normalizeparallel_vector orthogonal normal_vector)r;point1point2s r>rzLinearDimension._build_vectors_sI113 $V tF| ;FFH!11<<>r?cB|j|j|S)z=get the measure target point projection on the dimension line) intersectr)rdimrays r>rz%LinearDimension._get_point_on_dimlinees  1 1% 899r?c |j\}}|jd}|dkDr$||j|zz }||j|zz}|jd|jdd}|j|||y)zbuild dimension line entityr-rr,r*rhrkN)rrrry)r;r start_point end_pointr-attribss r>rzLinearDimension._draw_dimlinejs!%!9!9!; YYY|, >%)=)= )JKK!T%9%9J%FGIYY~.YYw'    r?cV|jd}|jd|jdd}t|j|jD]R\}}t ||t |t kDs$||z j}|||zz }|j|||Ty)"build the extension lines entitiesr4r,r*rhrkN) rziprrrmaxDIMENSIONS_MIN_DISTANCErry)r;rr4r dimline_point target_pointdirection_vectors r>rz%LinearDimension._draw_extension_lines}s /2 YY~.YYw'  ,/   !4!4, 'M< |4s68%1=$@#K#K#M +/?-/OP '$& , r?c||jd|jd|jd|j|jdd}t|jD]V}|j |}|j |}|j ||j|tjXy) z%build the dimension value text entityr$r"r*r))r$rir*rotationr)rrnalignN) rrranger_get_dimvalue_text_get_text_insert_pointadd_text set_placementr MIDDLE_CENTER)r;rrr dimvalue_text insert_points r>rzLinearDimension._draw_textsii)YY{+YYw' YYw'  T//0G 33Grz"LinearDimension._get_dimvalue_texts]%%g. x=O11':FF+dii.@@""8,,r?c|j|\}}|jddz |jdz}t|||j|zzS)z)get the dimension value text insert pointr$@r/)rrrr)r;rrrdists r>rz&LinearDimension._get_text_insert_pointsT11':yy"S(499[+AAFF#t'9'9D'@AAr?cjdjdjddd d fd }jdrLtdjdz D] }||d tdjD] }||d  y tjD] }||d y ) z@insert the dimension line ticks, (markers on the dimension line)rr*xscaleyscaler*Fcj|rdndzd<jj|jdy)z*build the insert-entity for the tick blockrrXrrinsertr6rnN)r add_blockrefrr)rrotaterrr;s r>add_tickz-LinearDimension._draw_ticks..add_ticksJ"&**S"IGJ    ..u5YYv&"  r?rrrTN)F)rrrboolrIrJ)rrr)r;rrrrs`` @r>rzLinearDimension._draw_tickssii -ii -YYw'     99X q$"2"2Q"67&8q$"2"23%4t//0&1r?)rXrTNN) rrrIterable[UVec]rrrrHr* Optional[str]r Optional[int])rrrrHrIrJr)rrrIr)rrrItuple[Vec3, Vec3])rIr)rIrrrrIrJ)rrrIz list[int])rrrrrIr )rrrIrH)rrrIr )rKrLrMrNr8rrrrrpropertyrrrrrrrrrrrrrrOrPs@r>rrs!#"& '      8+ <  G ,,((!66 ? :: &,&-B 'r?rceZdZdZeZeZeZ d dfd Z ddZ ddZ ddZ ddZddZddZdd Zdd Zdd Zdd ZxZS)AngularDimensionz Draw an angle dimensioning line at dimline pos from start to end, dimension text is the angle build of the three points start-center-end. ct||||t||_t||_t||_t||_y)a2 AngularDimension constructor. Args: pos: location as (x, y) tuple of dimension line, line goes through this point center: center point as (x, y) tuple of angle start: line from center to start is the first side of the angle end: line from center to end is the second side of the angle dimstyle: dimstyle name, 'Default' - style is the default value layer: dimension line layer, override the default value of dimstyle roundval: count of decimal places N)r7r8r rrrrlrm) r;rrrrlrmrr*r r=s r>r8zAngularDimension.__init__sB. 5(3s)6l %[ 9r?ct|j|j|_t|j|j|_|j|jz j |_|j|jz j |_ |jj|_ |jj|_ y)zsetup calculation valuesN) rrrr pos_radiusrlrsr start_vectorrm end_vectorr start_angle end_anglerbs r>rzAngularDimension._setups"4;;@t{{DJJ7 !ZZ$++5@@B88dkk1<<>,,22..r?c|j|j||jdr|j||j ||j |yr)r_draw_dimension_linerr_draw_dimension_textrrs r>rzAngularDimension.rendersQ  !!&) 99\ "  & &v . !!&)  r?c |j|j|jt|jt|j |j d|j ddy)z0draw the dimension line from start- to endangle.r*r,r*ri)rsrrrrrnN)add_arcr rrr rrrrs r>rz%AngularDimension._draw_dimension_line sX??;; 0 01dnn-7+>2  r?c |j|jfD]U}|j|j||j ||j d|j ddWy)rr*r2rrkN)r r ry_get_extline_start_get_extline_endr)r;rvectors r>rz&AngularDimension._draw_extension_linessf(($//:F OO--f5))&1!YYw/!YY'89  ;r?cD|j||jdzzSNr4)rrrr;rs r>rz#AngularDimension._get_extline_start%s{{ftyy'AABBr?c:|j||jzzSrArrr rs r>rz!AngularDimension._get_extline_end({{ft677r?c|jdt|j|jzdz tdz z |jd|jd|jdd}|j |j |j|jtj y) Nr$rr*r)r"r$rr*r)rirr) rr rrr r _get_dimtextrrr r)r;rrs r>rz%AngularDimension._draw_dimension_text+sii)!!DNN2a7"s(BYYw'YYw'YY{+  ""$   -  ' ' )%33  r?c|j|jzdz j}|j|j dz|j ddz z}|j ||zzS)Nrr/r$)r r rr rrr)r; midvectorlengths r>rz'AngularDimension._get_text_insert_point=sg''$//9S@KKM OOdii 4 4tyy7JS7P P {{i&011r?c|jd|jd|jdd}|jdf|j|jdffD]s\}}|j||jzz}|j t dz z}t||rt ndz|d<|j||jd | uy) Nrr*rFrrrXrrr) rr r rrr rr r r)r;rrrmirrorrrs r>rzAngularDimension._draw_ticksDsii -ii -YYw'     & __dii1 2 NFF ;;&4??*BCL||b3h.H")(Fb*L"MGJ    #YYv&"   r?cz|j|jz |jdz}|j|SNr)rrrr)r;rs r>r#zAngularDimension._get_dimtextWs7$"2"22dii6HH""5))r?)rUNN)rrrrrrlrrmrrrHr*rr rrrrr rIr rIr rIrH)rKrLrMrNr_DEGr`GRADraRADr8rrrrrrrrrr#rOrPs@r>rrs C D C$#"&       :/!   C8 $2&*r?rcpeZdZdZ d dfd Zd fd Zd dZd dZd dZxZ S) ArcDimensionz Arc is defined by start- and endpoint on arc and the center point, or by three points lying on the arc if acr3points is True. Measured length goes from start- to endpoint. The dimension line goes through the dimlinepos. c >t ||||||||||_y)a Args: pos: location as (x, y) tuple of dimension line, line goes through this point center: center point of arc start: start point of arc end: end point of arc arc3points: if **True** arc is defined by three points on the arc (center, start, end) dimstyle: dimstyle name, 'Default' - style is the default value layer: dimension line layer, override the default value of dimstyle roundval: count of decimal places N)r7r8 arc3points) r;rrrrlrmr4rr*r r=s r>r8zArcDimension.__init__fs%. feS(E8L$r?ct||jr1t|j|j |j |_yyrA)r7rr4center_of_3points_arcrrrlrm)r;r=s r>rzArcDimension._setups8  ??/ TZZDK r?c^|j||j|jdzzzSr)rrrsrrs r>rzArcDimension._get_extline_starts-{{ dkkDIIo$>> ?  r?c:|j||jzzSrArrs r>rzArcDimension._get_extline_endrr?c|j|jz |jz|jdz}|j |Sr*)rrrsrr)r; arc_lengths r>r#zArcDimension._get_dimtextsI ^^d.. .kk ii  !  "":..r?)FrTNN)rrrrrrlrrmrr4rrrHr*rr rrr+r-) rKrLrMrNr8rrrr#rOrPs@r>r2r2_s}!!#"&% %% %  %  %%% %4 8/r?r2c|eZdZdZ d d fd Zd dZd dZd dZd dZddZ ddZ d d Z xZ S)RadialDimensionz Draw a radius dimension line from `target` in direction of `center` with length drawing units. RadiusDimension has a special tick!! ct||||t||_t||_t ||_y)at Args: center: center point of radius target: target point of radius length: length of radius arrow (drawing length) dimstyle: dimstyle name, 'Default' - style is the default value layer: dimension line layer, override the default value of dimstyle roundval: count of decimal places N)r7r8r rrtargetrr&)r;rrr>r&rr*r r=s r>r8zRadialDimension.__init__s7& 5(36l 6l Fm r?c|j|jz j|_t |j|j|_yrA)r>rrr target_vectorrrsrbs r>rzRadialDimension._setups9 KK$++ % )+ t{{DKK8 r?c|j|j||j||j|y)rN)rrrrrs r>rzRadialDimension.renders5  !!&) !!&)  r?c|j|j|j|jz zz}|j ||j |j d|j ddy)Nr,r*rhrk)rrr@rsr&ryr>r)r;rrs r>rz$RadialDimension._draw_dimension_linesekk   $++ "; <   >27+  r?c L|j|j|jd|jj|jd|jd|jddj |j tjy)Nr$r*r)r"r"rr) rr#rr@ angle_degr_get_insert_pointr MIDDLE_RIGHTrs r>rz$RadialDimension._draw_dimension_texts""$))H- ..887+7+;/    -  " " $,?,L,L  r?cr|j|j|j|jdzzz S)Nr/)r>r@r&rrbs r>rEz!RadialDimension._get_insert_points3{{   $++ +0F"F G  r?c\|j|j|jdzSr*)rrsrrbs r>r#zRadialDimension._get_dimtexts%""4;;71C#CDDr?c |j|jd|jjdz|j d|j d|j ddy)NrWrr*)rrrr*r)rr>r@rDrrs r>rzRadialDimension._draw_tickss\;;! ..883>))L1))L17+   r?)rrTNN) rrrr>rr&rrrHr*rr rrrr,r-) rKrLrMrNr8rrrrrEr#rrOrPs@r>r<r<sy!#"&$$$ $  $  $ $09 !     E  r?r<ct||}t||}t||}t||}|j|}|j|}|j|S)z Calc center point of 3 point arc. ConstructionCircle is defined by 3 points on the circle: point1, point2 and point3. )rrrr) rrpoint3ray1ray2 midpoint1 midpoint2 center_ray1 center_ray2s r>r6r6sa 66 *D 66 *DVV$IVV$I//),K//),K   --r?N)rrrrrLrrIr)*rN __future__rtypingrrrrmathrr r abcr ezdxf.enumsr ezdxf.mathr rrrrrezdxf.documentr ezdxf.eztypesrrrr_r`radictrrRrwrrrr2r<r6rBr?r>r\s2#99%%+HH&/ BJ RZ  7t7t` ` F K ) ) XS'nS'lK*~K*\6/#6/rX nX v .r?