JiddZddlmZddlZddlmZmZmZmZm Z m Z ddl m Z ddl mZddlmZmZmZddlmZmZmZdd lmZdd lmZmZmZmZmZmZm Z dd l!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)Gd d e&Z*Gdde*Z+Gdde*Z,Gdde*Z-Gdde*Z.Gdde*Z/Gdde*Z0Gdde*Z1Gdde*Z2Gdde*Z3Gd d!e*Z4Gd"d#e*Z5Gd$d%e*Z6Gd&d'e*Z7y)(a Sketch Objects name: objects_sketch.py by: Gumyr date: March 22nd 2023 desc: This python module contains objects (classes) that create 2D Sketches. license: Copyright 2022 Gumyr Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. ) annotationsN)cosdegreespiradianssintan)cast)Iterable) LocationListflatten_sequencevalidate_inputs)Align FontStyleMode) BuildSketch)AxisLocationRotationVector VectorLiketo_align_offset TOLERANCE)CompoundEdgeFace ShapeListSketchWiretuplifytopo_explore_common_vertexcheZdZdZej gZddejf dfd Z xZ S)BaseSketchObjectaBaseSketchObject Base class for all BuildSketch objects Args: face (Face): face to create rotation (float, optional): angles to rotate objects. Defaults to 0. align (Union[Align, tuple[Align, Align]], optional): align min, center, or max of object. Defaults to None. mode (Mode, optional): combination mode. Defaults to Mode.ADD. rNc4|Ct|d}|jt|jj |t j |d}|+|jtdd|j}nl||_ ||_ |jtdd|}td|jD}t|t r|j|d|it|At#|j$y)NF)logrc3K|]8}tjjD]}|j|:ywN)r _get_contextlocal_locationsmoved).0facelocations i/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/build123d/objects_sketch.py z,BaseSketchObject.__init__..bs="'D , 9 9 ; K KH 8$ K%'s>Amode)r mover bounding_boxrrr)r+rfacesrotationr1r isinstance_add_to_contextsuper__init__rwrapped)selfobjr5alignr1context new_faces __class__s r/r9zBaseSketchObject.__init__Ms  E1%E HHXc..0@@GH I&1&>&>t&O ? (1a":;AACI%DMDI))HQ845C!"IIK"I ';/'''>> ),445)r<zCompound | Facer5floatr="Align | tuple[Align, Align] | Noner1r __name__ __module__ __qualname____doc__r_tag _applies_torADDr9 __classcell__r@s@r/r#r#>sW ##$K 48XX 6 662 6  66rAr#ceZdZdZej gZejejfe jf dfd Z xZ S)CircleaJSketch Object: Circle Add circle(s) to the sketch. Args: radius (float): circle size align (Union[Align, tuple[Align, Align]], optional): align min, center, or max of object. Defaults to (Align.CENTER, Align.CENTER). mode (Mode, optional): combination mode. Defaults to Mode.ADD. ctj|}t||||_t |d|_t tj|}t|)|d|j |y)Nr%r) rr)rradiusr r=rr make_circler8r9)r;rQr=r1r>r-r@s r/r9zCircle.__init__{s` '2&>&>t&D& UA& D$$V,- q$**d3rA)rQrBr=rCr1r rErFrGrHrrIrJrCENTERrrKr9rLrMs@r/rOrOmsW ##$K 6;\\5<<4PXX 4 42 4 4 4rArOceZdZdZej gZdejejfe jf dfd Z xZ S)EllipseaSketch Object: Ellipse Add ellipse(s) to sketch. Args: x_radius (float): horizontal radius y_radius (float): vertical radius rotation (float, optional): angles to rotate objects. Defaults to 0. align (Union[Align, tuple[Align, Align]], optional): align min, center, or max of object. Defaults to (Align.CENTER, Align.CENTER). mode (Mode, optional): combination mode. Defaults to Mode.ADD. rctj|}t||||_||_t |d|_ttj||}t|-|||j |yNr%) rr)rx_radiusy_radiusr r=rr make_ellipser8r9) r;rYrZr5r=r1r>r-r@s r/r9zEllipse.__init__si'2&>&>t&D&    UA& D%%h9: xT:rA) rYrBrZrBr5rBr=rCr1rrSrMs@r/rVrVsn ##$K 5:\\5<<4PXX ;;; ; 2 ;  ;;rArVceZdZdZej gZdejejfe jd dfdZ xZ S)Polygona:Sketch Object: Polygon Add polygon(s) defined by given sequence of points to sketch. Note that the order of the points define the normal of the Face that is created in Algebra mode, where counter clockwise order creates Faces with their normal being up while a clockwise order will have a normal that is down. In Builder mode, all Faces added to the sketch are up. Args: pts (Union[VectorLike, Iterable[VectorLike]]): sequence of points defining the vertices of the polygon rotation (float, optional): angles to rotate objects. Defaults to 0. align (Union[Align, tuple[Align, Align]], optional): align min, center, or max of object. Defaults to (Align.CENTER, Align.CENTER). mode (Mode, optional): combination mode. Defaults to Mode.ADD. rr5r=r1c6tj|}t||t|}||_t |d|_|Dcgc] }t|}}ttj|} t |1| ||j |ycc}wrX) rr)rr ptsr r=rrr make_polygonr8r9) r;r5r=r1r`r> flattened_ptsppoly_ptsr-r@s r/r9zPolygon.__init__s'2&>&>t&D&(#.  UA& '*+s!F1Is+D%%h/0 xT:,sB)r`z!VectorLike | Iterable[VectorLike]r5rBr=rCr1rrSrMs@r/r]r]sd$##$K 5:\\5<<4PXX ;/;;2 ;  ;;rAr]ceZdZdZej gZdejejfe jf dfd Z xZ S) RectangleaSketch Object: Rectangle Add rectangle(s) to sketch. Args: width (float): horizontal size height (float): vertical size rotation (float, optional): angles to rotate objects. Defaults to 0. align (Union[Align, tuple[Align, Align]], optional): align min, center, or max of object. Defaults to (Align.CENTER, Align.CENTER). mode (Mode, optional): combination mode. Defaults to Mode.ADD. rctj|}t||||_||_t |d|_tj||}t|)|||j |yrX) rr)rwidthrectangle_heightr r=r make_rectr8r9 r;rhheightr5r=r1r>r-r@s r/r9zRectangle.__init__sc'2&>&>t&D& &UA& ~~eV, xT:rA rhrBrlrBr5rBr=rCr1rrSrMs@r/rfrfsn ##$K 5:\\5<<4PXX ;;; ; 2 ;  ;;rArfceZdZdZej gZdejejfe jf dfd Z xZ S)RectangleRoundedaSketch Object: RectangleRounded Add rectangle(s) with filleted corners to sketch. Args: width (float): horizontal size height (float): vertical size radius (float): fillet radius rotation (float, optional): angles to rotate objects. Defaults to 0. align (Union[Align, tuple[Align, Align]], optional): align min, center, or max of object. Defaults to (Align.CENTER, Align.CENTER). mode (Mode, optional): combination mode. Defaults to Mode.ADD. rcZtj|}t|||d|zks|d|zkr td||_||_||_t|d|_tj||}|j||j}t |9||||y)Nr%z#width and height must be > 2*radius)rr)r ValueErrorrhrirQr r=rrj fillet_2dverticesr8r9) r;rhrlrQr5r=r1r>r-r@s r/r9zRectangleRounded.__init__ s'2&>&>t&D& AJ &AJ"6BC C & UA& ~~eV,~~fdmmo6 x5rA) rhrBrlrBrQrBr5rBr=rCr1rrSrMs@r/rorosx ##$K5:\\5<<4PXX666 6  6 2 666rAroceZdZdZej gZddejejfe jf dfd Z xZ S)RegularPolygonaSketch Object: Regular Polygon Add regular polygon(s) to sketch. Args: radius (float): distance from origin to vertices (major), or optionally from the origin to side (minor) with major_radius = False side_count (int): number of polygon sides major_radius (bool): If True the radius is the major radius, else the radius is the minor radius (also known as inscribed radius). Defaults to True. rotation (float, optional): angles to rotate objects. Defaults to 0. align (Union[Align, tuple[Align, Align]], optional): align min, center, or max of object. Defaults to (Align.CENTER, Align.CENTER). mode (Mode, optional): combination mode. Defaults to Mode.ADD. Trctj|}t|||dkrtd||r|}n|t t |z z }||_|t t |z z|_||_||_ tt|dzD cgc]X} t|t | dzt z|z t|zz|t| dzt z|z t|zzZc} } | jt j"| jt j$g} | ddj"| ddj$g} | ddj"| ddj$g} t'| | |d}| Dcgc]}||z }}t)t+j,|}t.|a|dd| ycc} wcc}w) Nz3RegularPolygon must have at least three sides, not r%r)rr)centerr^)rr)rrqrrrQapothem side_countr=rrangerrrsort_byrXYrrrrar8r9)r;rQr| major_radiusr5r=r1r>radir` pts_sortedminsmaxs align_offsetpointpts_aor-r@s r/r9zRegularPolygon.__init__6s'2&>&>t&D& >Ej\R  C3rJ//C  !C O%   %  zA~.   /A #a!ebj:58IIJJ#a!ebj:58IIJJ/   kk$&&)3;;tvv+>? 1 a ""JqM!$4$6$671 b!##Z]2%6%8%89&tT5H 478C5%,&C8D%%f-. DA% 9s AG G) rQrBr|intrboolr5rBr=ztuple[Align, Align]r1rrSrMs@r/ruru"s"##$K "&+llELL%AXX2B2B2B 2B  2B # 2B2B2BrArucfeZdZdZej gZdejf dfd Z xZ S)SlotArcaMSketch Object: Arc Slot Add slot(s) following an arc to sketch. Args: arc (Union[Edge, Wire]): center line of slot height (float): diameter of end circles rotation (float, optional): angles to rotate objects. Defaults to 0. mode (Mode, optional): combination mode. Defaults to Mode.ADD. rc:tj|}t||||_||_t |t r|n t |g}t|j|dz jtj|}t|5||d|yrX)rr)rarc slot_heightr6rr offset_2drotaterZr8r9)r;rrlr5r1r>r-r@s r/r9zSlotArc.__init__ys'2&>&>t&D&!T*ccU CMM&1*-.55dffhG xt4rA)rz Edge | WirerlrBr5rBr1rrDrMs@r/rrksT ##$K XX 5 55 5  55rArcjeZdZdZej gZdejf dfd Z xZ S)SlotCenterPointaSketch Object: Center Point Slot Add a slot(s) defined by the center of the slot and the center of one of the circular arcs at the end. The other end will be generated to create a symmetric slot. Args: center (VectorLike): slot center point point (VectorLike): slot center of arc point height (float): diameter of end circles rotation (float, optional): angles to rotate objects. Defaults to 0. mode (Mode, optional): combination mode. Defaults to Mode.ADD. rc tj|}t||t|}t|}||_||_||_||z } | jdz|kr tdttjtj||tj||| z gdj|dz } t |A| |d|y)Nr%zVSlots must have width > height. Got: {height=} width={half_line.length * 2} (computed)r)rr)rr slot_centerrrlengthrqrrcombiner make_linerr8r9) r;rzrrlr5r1r>center_vpoint_v half_liner-r@s r/r9zSlotCenterPoint.__init__s'2&>&>t&D&&>-# !h&   a 6 )I   LLNN7H5NN8X -AB     6A:&   xt4rA) rzrrrrlrBr5rBr1rrDrMs@r/rrs^ ##$KXX !5!5!5 !5  !5  !5!5rArcfeZdZdZej gZdejf dfd Z xZ S)SlotCenterToCenteraSketch Object: Center to Center points Slot Add slot(s) defined by the distance between the center of the two end arcs. Args: center_separation (float): distance between two arc centers height (float): diameter of end circles rotation (float, optional): angles to rotate objects. Defaults to 0. mode (Mode, optional): combination mode. Defaults to Mode.ADD. rc |dkrtd|tj|}t||||_||_t ttjt| dz ddttjtt|dz ddgj|dz }t|5||d|y)Nrz7Requires center_separation > 0. Got: center_separation=r%)rqrr)rcenter_separationrrrrrrrr8r9)r;rrlr5r1r>r-r@s r/r9zSlotCenterToCenter.__init__s  !J8I7KL '2&>&>t&D&!2! NN6+<* height. Got: width=z , height=r%r)r1)rqrr)rrhrrrrrrrr rOr-r8r9rks r/r9zSlotOverall.__init__s  F?A5(*VIN '2&>&>t&D& ! F?vufqj6A:.Eq!'LfhWvx VaZ8OQRTU1VW )FQJ' DfUQYT:??ABD x5rArmrSrMs@r/rrsn ##$K 5:\\5<<4PXX 666 6 2 6  66rArceZdZdZej gZddeje je jfddde jf dfd Z xZS)TextaSketch Object: Text Add text(s) to the sketch. Args: txt (str): text to be rendered font_size (float): size of the font in model units font (str, optional): font name. Defaults to "Arial". font_path (str, optional): system path to font library. Defaults to None. font_style (Font_Style, optional): style. Defaults to Font_Style.REGULAR. align (Union[Align, tuple[Align, Align]], optional): align min, center, or max of object. Defaults to (Align.CENTER, Align.CENTER). path (Union[Edge, Wire], optional): path for text to follow. Defaults to None. position_on_path (float, optional): the relative location on path to position the text, values must be between 0.0 and 1.0. Defaults to 0.0. rotation (float, optional): angles to rotate objects. Defaults to 0. mode (Mode, optional): combination mode. Defaults to Mode.ADD. ArialNgc 2tj|} t| |||_||_||_||_||_||_||_ ||_ | |_ | |_ tj||||||||} t |A| | d| y)N)txt font_sizefont font_path font_styler=position_on_path text_path)rr)rrrrrrr=rrr5r1r make_textr8r9)r;rrrrrr=pathrr5r1r> text_stringr@s r/r9z Text.__init__4s'2&>&>t&D&" "$  0   ((!-   hd;rA)rstrrrBrrrz str | Nonerrr=rCrzEdge | Wire | NonerrBr5rBr1r)rErFrGrHrrIrJrREGULARrrTrrKr9rLrMs@r/rrs(##$K  $ ) 1 15:\\5<<4P#'"%XX%< %<%< %<  %<  %<2%<!%< %<%<%<%reduction_leftreduction_righttop_width_lefttop_width_rightbot_width_leftbot_width_rightr`r-r@s r/r9zTrapezoid.__init__rs'2&>&>t&D&2B?HX &. 0UA& !B&AFS9Q5R,R "R'AVc'BR:S6T-T !)!) A  n ,N n ,N Q   .O  .O _ , 1GH H _ , 1DE E 6>/F7Q;78 6/F7Q;78 6/6A:67 6>/6A:67 3q6D%%c*+ xT:rA)rhrBrlrBrrBr float | Noner5rBr=rCr1rrSrMs@r/rr\s&##$K*.5:\\5<<4PXX8;8;8; 8; ' 8;  8;28;8;8;rArc eZdZdZej gZddddddddejd dfdZ xZ S)Trianglea1Sketch Object: Triangle Add any triangle to the sketch by specifying the length of any side and any two other side lengths or interior angles. Note that the interior angles are opposite the side with the same designation (i.e. side 'a' is opposite angle 'A'). Args: a (float, optional): side 'a' length. Defaults to None. b (float, optional): side 'b' length. Defaults to None. c (float, optional): side 'c' length. Defaults to None. A (float, optional): interior angle 'A' in degrees. Defaults to None. B (float, optional): interior angle 'B' in degrees. Defaults to None. C (float, optional): interior angle 'C' in degrees. Defaults to None. rotation (float, optional): angles to rotate objects. Defaults to 0. align (Union[Align, tuple[Align, Align]], optional): align min, center, or max of object. Defaults to None. mode (Mode, optional): combination mode. Defaults to Mode.ADD. Raises: ValueError: One length and two other values were not provided Nr) abcABCr=r5r1c tj} t| |||fD cgc]} | duc} jddk(s*||||||fD cgc]} | duc} jddk7r t dd|||fD\}}}t j ||||||\} } }}__| _ t| _ t|_ t|_ ttj t#ddt#dt#| dj%t&j(jg}t+d|j-Dt#ddddz }|j/t1| |dn t3|d}t4m|||| j9j;fd d_j9j;fd d_j9j;fd d_ tCj>j@_"tCj<j@_#tCj<j>_$ycc} wcc} w) NTrwz0One length and two other values must be providedc3:K|]}| t|ndywr()r)r,angles r/r0z$Triangle.__init__..sXiUU%675>D@isrc32K|]}t|ywr()r)r,vs r/r0z$Triangle.__init__..s8$7q$7sr%)r<r5r=r1cBt|jz tkSr()absrr)ears r/z#Triangle.__init__..ss188b=7II7UrAcdt|jz tkxr|jfvSr()rrredge_a)rbrr;s r/rz#Triangle.__init__..s*c!((R-(94O4;;-9OOrAc8|jjfvSr()redge_b)rr;s r/rz#Triangle.__init__..sa T[[99rA)%rr)rcountrqtrianglesolversolverrrrrrrrrrarrrrsumrsr2rr r8r9edges filter_byrredge_cr!vertex_Avertex_Bvertex_C)r;rrrrrrr=r5r1r>rcrArBrCrtrianglecenter_of_geometry alignmentrrr@s` @@r/r9zTriangle.__init__s'2&>&>t&D& !1ay )y!AIy ) / / 5 : !1aAq1? 1!AI1? %+?OP PXqRSUViX1a!/!5!5aAq!Q!GBBB   1vb!}fRm.B.B466466.RS   8H$5$5$78&Aq/ JQ N   h 2234!MDwua/@  X PTUjjl,,-UV   jjl,, O    jjl,, 9    3 KK  3 KK  3 KK  W *? s J; K)rrrrrrrrrrrrr=rCr5rBr1rrDrMs@r/rrs,##$K 48XX=  =   =  =  =  =  = 2= = = = rAr)8rH __future__rrmathrrrrrr typingr collections.abcr build123d.build_commonr r rbuild123d.build_enumsrrrbuild123d.build_sketchrbuild123d.geometryrrrrrrrbuild123d.topologyrrrrrrr r!r#rOrVr]rfrorurrrrrrrrAr/rs8#44$RR88.   ,6v,6^4 4< ; ;F%;%;P ; ;F&6'&6RFB%FBR55@25&25j)5))5X.6".6b<< <<~N; N;bV V rA