Ji}UdZddlZddlZddlmZmZddlmZm Z ddl m Z ddl m Z mZmZmZmZddlmZmZmZmZmZmZddlmZdd lmZdd lmZdd lm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'dd l(m)Z)dd l*m+Z+ddl,m-Z-ddl.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;med?e'dzd@e1e;zdzdeEdede4e8zfdAZ^y)Ca$ Generic Operations name: operations_generic.py by: Gumyr date: March 21th 2023 desc: This python module contains operations (functions) that work on combinations of Curves, Sketches and Parts. 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. N)radianstan)cast TypeAlias)Iterable)Builder LocationList WorkplaneListflatten_sequencevalidate_inputs)KeepKindModeSide TransitionGeomType) BuildLine) BuildPart) BuildSketch)AxisLocationMatrixPlaneRotation RotationLikeVector VectorLike)BasePartObject)BaseSketchObject)BaseLineObject)CompoundCurveEdgeFaceGroupByPartShape ShapeListShellSketchSolidVertexWire isclose_b build123dAddTypeTobjectsrotationcleanmodereturnc * tjd}| tdt|trt|t s t |}n|g}|Dcgc]B}t|t r|jdnt|tr |jn|D}}t|d|t|tr|tddd}n$t|tr t|}n td|Dcgc]}|j|}}|Dcgc]}t|ts|}}|Dcgc]}t|t s|} }|Dcgc]}t|t"s|} }|Dcgc]}t|t$s|} }t'd|D]} |j)| j+t| j)| j+t | j)| j+t"| j)| j+t$| D]!} |j)| j-#t/js td |Dcgc]6}t/jj0D]}|j|8}}}|j2||}t5jj6D]q}g}t/jj0D])}| D]"}|j9|j|$+|j2|d |i|j)|s| Dcgc]6}t/jj0D]}|j|8}}}|j:|||d |j)|t |St|t<rt|t>t@fr|nd }g}|D]j}|j)t/jjBDcgc]1}|jEtFjH|j|3c}l|j:|d |it |St|tJrt|t>t@fr|nd }g}|D]j}|j)t/jjBDcgc]1}|jEtFjH|j|3c}l|j:|d |it |Std|jLjNdcc}wcc}wcc}wcc}wcc}wcc}wcc}}wcc}}wcc}wcc}w)a<Generic Object: Add Object to Part or Sketch Add an object to a builder. BuildPart: Edges and Wires are added to pending_edges. Compounds of Face are added to pending_faces. Solids or Compounds of Solid are combined into the part. BuildSketch: Edges and Wires are added to pending_edges. Compounds of Face are added to sketch. BuildLine: Edges and Wires are added to line. Args: objects (Edge | Wire | Face | Solid | Compound or Iterable of): objects to add rotation (float | RotationLike, optional): rotation angle for sketch, rotation about each axis for part. Defaults to None. clean (bool, optional): Remove extraneous internal structure. Defaults to True. mode (Mode, optional): combine mode. Defaults to Mode.ADD. Nz'Add must have an active builder contextFfullyaddrzInvalid rotation valuec"t|tSN) isinstancer!)os m/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/build123d/operations_generic.pyzadd..s Ax)@z$There is no active Locations context face_planer3r4r4zBuilder z is unsupported)(r _get_context RuntimeErrorr<rr!listunwrap_objr rrtuple ValueErrormovedr#r-r$r+filterextendget_typeedgesr locations_add_to_pendingr workplanesappend_add_to_contextrintfloatlocal_locationsrotaterZr __class____name__)r1r2r3r4context object_listobj object_iter new_edges new_wires new_faces new_solidscompoundnew_wireedgelocation located_edges new_objects workplanefaces_per_workplanefacesolidlocated_solidsrotation_angles r>r9r9Us"2&2248GDEE'8$Z-J7m i   C#x( JJUJ #'W53 ? GUK0'9%  1a(H % (*H56 66ABkssyy*k B$/IKS:c43HSK I$/IKS:c43HSK I$/IKS:c43HSK I%0K[cJsE4Jc[ K@+NH   X..t4 5   X..t4 5   X..t4 5   h//6 7 O "H   X^^- ."((*EF F" !(557AA JJx A !!  /# '335@@I"$ (557AA%D'..tzz(/CD&B $G # #%8 OY O   2 3 A$ #(557AA KK !A "#  u4H>*: K  7 G[ )%/3,%GS C   %1$=$=$?$O$O$OJJtvv~6<rqrqsR&22>BG ?g1gll6J78 8||n &w/ G^[9 [ 1[cCHHM[ 12 C#v&##%DTXXZZ(TXXZZ(TXXZZ(TXXZZ(TXXZZ( H   T$"3"34Q1VAY4Q"RS T   #G # #Y :T :hy)1122K c6 " ! NN   txxzz488::txxzz:;   848 %-- ..C 25Rs "L%LChamferFilletTypelengthlength2angle referencec  tjd}|r |r td|r|tt |z}|r|s |s td||n|}|||||j tdt |}t|d|| |j n|dj tdjdk(rttrtjnt|Dcgc]}t|tc}s tdj!||t#||}|!|j%|t&j( tt+|gjSjd k(r tt,rt/jnt|Dcgc]}t|t0c}s td g} j3D][} | j5D cgc] } | |vs|  } } | r$| j7| j9||| |K| j7| ]t/t+| j} |!|j%| t&j( | Sjd k(rtt:rt=jnj?dt|Dcgc]}t|t0c}s td j@stCtEfd|}j9||||}|!|j%|t&j( |Stdcc}wcc}wcc} wcc}w)aaGeneric Operation: chamfer Applies to 2 and 3 dimensional objects. Chamfer the given sequence of edges or vertices. Args: objects (Edge | Vertex or Iterable of): edges or vertices to chamfer length (float): chamfer size length2 (float, optional): asymmetric chamfer size. Defaults to None. angle (float, optional): chamfer angle in degrees. Defaults to None. reference (Edge | Face): identifies the side where length is measured. Edge(s) must be part of the face. Vertex/Vertices must be part of edge Raises: ValueError: no objects provided ValueError: objects must be Edges ValueError: objects must be Vertices ValueError: Only one of length2 or angle should be provided ValueError: reference can only be used in conjunction with length2 or angle chamferz/Only one of length2 or angle should be providedz?reference can only be used in conjunction with length2 or angleNo objects providedrzNothing to chamferz%3D chamfer operation takes only Edgesr4rsz(2D chamfer operation takes only Vertices'1D fillet operation takes only Verticesctt|jjdz jdxs<tt|jjdz jd SNrrCrr.rto_tuple position_atrrtargets r>r?zchamfer..zm!#QZZ\2V5G5G5JJRR%#QZZ\2V5G5G5JJRR r@Invalid object dimension)#rrDrJrrrHr r topo_parentrur<rr&r{rtr#rrFrTrREPLACEr!rr*r,facesrrS chamfer_2dr r-wires is_closedr(rL)r1rrrrr\r]r^new_partrbrlrvertices_in_face new_sketchrers @r>rrsQ8&229=G5JKK 3wu~..'U M   fWGGOG/GLL4H.//"7+KGY 4Q++ ~-.. {{a)3FN)Kfnn%QW[A[cJsD)[ABDE E>>&'4 3DiP    # #H4<< # @HhZ(0011 {{a'19I&JF6>> "PV {C{JsF+{CDGH H LLND+/==?O?aa;>N? O  OOFG5EyQ  &#HY/778    # #JT\\ # B {{a&.1  "  {C{JsF+{CDFG G#   K$$VWk9M    # #H4<< # @ / 00yBD P&Ds7M<?N NN+N radiusc  tjd}|||||j tdt |}t |d|| |j n|dj td jdk(rt trt jn t|Dcgc]}t|tc}s td j|t|}|!|j!|t"j$tt'|gjS jdk(rt t(rt+ jn t|Dcgc]}t|t,c}s td g} j/D]b}|j1Dcgc]}|t|vs|} }| r"|j3|j5|| R|j3|dt+t'|j} |!|j!| t"j$| S jd k(rt t6rt9 jn j;d t|Dcgc]}t|t,c}s td  j<st?tA fd |} j5||} |!|j!| t"j$| Std cc}wcc}wcc}wcc}w)a%Generic Operation: fillet Applies to 2 and 3 dimensional objects. Fillet the given sequence of edges or vertices. Note that vertices on either end of an open line will be automatically skipped. Args: objects (Edge | Vertex or Iterable of): edges or vertices to fillet radius (float): fillet size - must be less than 1/2 local width Raises: ValueError: no objects provided ValueError: objects must be Edges ValueError: objects must be Vertices ValueError: nothing to fillet filletrrzNothing to filletrz$3D fillet operation takes only Edgesrrsz'2D fillet operation takes only Verticesrrctt|jjdz jdxs<tt|jjdz jd Srrrs r>r?zfillet..rr@r)!rrDrHrJr r rrur<rr&r{rtr#rrFrTrrr!rr*r,rrrS fillet_2dr r-rrr(rL) r1rr\r]r^rrbrlrrrrers @r>rrs*&228> "PV {C{JsF+{CDFG G LLND+/==?U?aa4 CT>T? U  8H!IJ  & # HY/778    # #JT\\ # B {{a&.1  "  {C{JsF+{CDFG G#   K##FK8    # #H4<< # @ / 00wBD V"Ds8M>MM !M 1M MirrorTypeaboutc>tjd}t|trt|ts t |}n|g}|)|||j td|j g}n t|}t|d||Dcgc]&}tj|j|(}}||j|d|it |}t|Dcgc]}|jdk(c}rt!|j"St|Dcgc]}|jdk(c}rt%|j"St|Dcgc]}|jdk(c}rt'|j"S|Scc}wcc}wcc}wcc}w)aGeneric Operation: mirror Applies to 1, 2, and 3 dimensional objects. Mirror a sequence of objects over the given plane. Args: objects (Edge | Face | Compound or Iterable of): objects to mirror about (Plane, optional): reference plane. Defaults to "XZ". mode (Mode, optional): combination mode. Defaults to Mode.ADD. Raises: ValueError: missing objects mirrorrrr4rrsr)rrDr<rr!rFrHrJr r copy_moduledeepcopyrrTrtrur&r{r*r") r1rr4r\r]r=mirroredmirrored_compoundr^s r>rrsv&&228k(rbttj@| d d z| d d ztj@| d d z| d d zgj'||||g}nt| j'||||g}|!|Dcgc]}|j)|}}ng}t |tr|g}g}| D]Z}|r#|Dcgc]}||jCvs|}}ng}|j|jE|||jG\||z|z}||jH|d|it|}tK| D cgc]} | jLdk(c} rtO|jPStK| D cgc]} | jLdk(c} rtS|jPStK| D cgc]} | jLd k(c} rtU|jPS|S#t,$rY^wxYwcc}wcc}wcc}wcc} wcc} wcc} w)aGeneric Operation: offset Applies to 1, 2, and 3 dimensional objects. Offset the given sequence of Edges, Faces, Compound of Faces, or Solids. The kind parameter controls the shape of the transitions. For Solid objects, the openings parameter allows selected faces to be open, like a hollow box with no lid. Args: objects (Edge | Face | Solid | Compound or Iterable of): objects to offset amount (float): positive values external, negative internal openings (list[Face], optional), sequence of faces to open in part. Defaults to None. kind (Kind, optional): transition shape. Defaults to Kind.ARC. side (Side, optional): side to place offset. Defaults to Side.BOTH. closed (bool, optional): if Side!=BOTH, close the LEFT or RIGHT offset. Defaults to True. min_edge_length (float, optional): repair degenerate edges generated by offset by eliminating edges of minimum length in offset wire. Defaults to None. mode (Mode, optional): combination mode. Defaults to Mode.REPLACE. Raises: ValueError: missing objects ValueError: Invalid object type offsetrrzUnsupported type z for )rgMbP?Tr7rrrCg??)rrrr4rrs)+rrDrHrJr r r<r!rMrNr#r$r+rSr-rO TypeErrortype outer_wire offset_2dfix_degenerate_edges inner_wires Exception normal_atrcutrGr(len geom_typerLINE make_liner offset_3dfixrTrtrur&r{r*r")r1rrrrrrr4r\r]rOrsolidsr^rbrlrr inner_wire offset_wirenew_facew inner_faces subtractionrarcrmr=openings_in_this_solidrioffset_compounds r>rr/sH&228>#3 3 ; ;e C yH ,78Kq47KK8&(,, 4K+x0  t!<=K3  -  -   X &=> u:?uQx11X]]BuQx#~uQx#~FuQx#~uQx#~F )FD)H IU %%f4d6%RI  &JST)Q//@)IT (D!:J 19%PAQ%++-=Oa "%P%' " OO2FO F J J L i'*4K848{+O [ 1[cCHHM[ 12O++,, [ 1[cCHHM[ 12o--.. [ 1[cCHHM[ 12_,,-- y  92U&Q 2 1 1s=&AT42U U UU UUU4 UU ProjectTyperjrc  tjd}t|tr t d|s | t d|sx|vt|t rf|j |jz}g|_g|_t|jdkDr|jd}g|_ n|jd}n t|}|Dcgc]*}t|trt|jn|,}}t|j!dj"}|Dcgc]}t|ttfs|} }| D cgc] } t| } } |Dcgc]}t|t$s|} }|Dcgc]}t|t&t(fs|} }|| t dt|t*r1|j,d}|t.j0k7rP| s| rLt d t|t2r1|j,d}|t.j0k7r| s| r t d t5t6|}d }|At|t r1|t.j0k7r | r t d | |j8}d }nt%j:d|zd|z|}| t dt=|dg}| | zD]}|j?|j?z jA}|jC|jDdkr|jF |z}n|jF|z}|jI||}|st|t2r/|jK|Dcgc]}|jM|c}t|t*r|jK||jO|dtQ}| D]} |jR| z jA}|jC|jDdkrtU| |jF|z }ntU| |jF|z}|jW|}t|tTr tYd||jO|jM|||jZ|d|i|r|}|St|}t]|Dcgc]}|j^dk(c}rta|jb}|St]|Dcgc]}|j^d k(c}rte|jb}|Scc}wcc}wcc} wcc}wcc}wcc}wcc}wcc}w)aGeneric Operation: project Applies to 0, 1, and 2 dimensional objects. Project the given objects or points onto a BuildLine or BuildSketch workplane in the direction of the normal of that workplane. When projecting onto a sketch a Face(s) are generated while Edges are generated for BuildLine. Will only use the first if BuildSketch has multiple active workplanes. In algebra mode a workplane must be provided and the output is either a Face, Curve, Sketch, Compound, or ShapeList[Vector]. Note that only if mode is not Mode.PRIVATE only Faces can be projected into BuildSketch and Edge/Wires into BuildLine. Args: objects (Edge | Face | Wire | VectorLike | Vertex or Iterable of): objects or points to project workplane (Plane, optional): screen workplane mode (Mode, optional): combination mode. Defaults to Mode.ADD. Raises: ValueError: project doesn't accept group_by ValueError: Either a workplane must be provided or a builder must be active ValueError: Points and faces can only be projected in PRIVATE mode ValueError: Edges, wires and points can only be projected in PRIVATE mode RuntimeError: BuildPart doesn't have a project operation projectz2project doesn't accept group_by, did you miss [n]?zNo object to projectr)childrenF)optimalz?Either a workplane must be provided or a builder must be activez6Points and faces can only be projected in PRIVATE modez=Edges, wires and points can only be projected in PRIVATE moderz,Points can only be projected in PRIVATE moder)planez'A target object could not be determinedz.working_plane and projection_axis are parallelr4rs)3rrDr<r%rJr pending_edges pending_facesrpending_face_planesexit_workplanesr rr,rr!rqdiagonalr$r#r-rrRrPRIVATErrrpart make_rectr center normalizedfrom_local_coordsrYz_dirproject_to_shaperMto_local_coordsrSr(originr intersectrErTrtrur*r{r")r1rjrr4r\r]r= shape_list object_size vct_vrt_listpnt point_list face_list line_list working_planeprojection_flipprojected_shapesr^ obj_to_screenprojection_direction projectionpprojected_points pnt_to_targetprojection_axis intersectionresults r>rrsB&229=G'7#MNN w/00 w*z'9/M++g.C.CC " " w** +a /33A6I*,G '//2I&w/ HSGR!Av!6 A={J/<KI?'GKq:a$+FKIG ?Q  gy )**1-It||#j L -**1-It||#j S *MOz'9= 4<< JKL L >\\FK[ V ~BCCGY'9$3::<7CCE  * *= 9 ; ;a ?$1$7$7#7/#I #0#6#6#H ))&2FG ';/ ''?IJz!]2215zJGY/ '' 3 '' 1 6%"+4+&--3??A  * *= 9 ; ;a ?#C)<)<)NOOO"3 (;(;o(MNO$..? lD )OP P  #  # #M$A$A,$O P!1==! M *+ 5#A 5 6FNN+F M;7;C#((a-;7 86>>*F Mu O6?GdK667sB /T2$T7T7 T<"U8UUU,U U>Urbyc,tjd}|)|||j td|jg}n t |}t |d|t |ttfr t|}nt |trgt|dk(rYtd|DrGt|}t|jdddgd|jddgdd|j dggdg}n tdg}|D]} | | j"} | J| j%t't} t |ttfr!| j)j+| } n | j-j+| } |j/| ||j0|d|it3|} t|D cgc]} | j4dk(c} rt7| j8} nwt|D cgc]} | j4d k(c} rt;| j8} n;t|D cgc]} | j4d k(c} rt=| j8} | j?d Scc} wcc} wcc} w) aGeneric Operation: scale Applies to 1, 2, and 3 dimensional objects. Scale a sequence of objects. Note that when scaling non-uniformly across the three axes, the type of the underlying object may change to bspline from line, circle, etc. Args: objects (Edge | Face | Compound | Solid or Iterable of): objects to scale by (float | tuple[float, float, float]): scale factor mode (Mode, optional): combination mode. Defaults to Mode.REPLACE. Raises: ValueError: missing objects scalerrrc3HK|]}t|ttfywr;)r<rUrV).0ss r> zscale..{s8R 1sEl+Rs "rC)rCrCrCrz,by must be a float or a three tuple of floatr4rsrFr7) rrDrHrJr r r<rUrVrIrrtrrrwrxrYrglocatedrrlocatetransform_geometryrSrTr!rur&r{r*r"rG)r1rr4r\r]factor by_vector scale_matrixrir^current_location obj_at_origin new_objectscale_compounds r>rrVsY*&227;G ?g1gll6J78 8||n &w/ GWk2"sEl#r2 GqL 8R8 82J c3,ikk3,c9;;,$   GHHK ; <<+++ HVX$67 b3, '&,,V4;;r r s,&227;G ?g1gll6J78 8||n &w/ GWk2#%K 499 ))It4KC))It,CV}G'8,""7+$""7+ % 848k*N [ 1[cCHHM[ 12N**++ [ 1[cCHHM[ 12n,,-- [ 1[cCHHM[ 12^++,,  2 1 1sF0F5<F: SweepTypeFsectionspath multisection is_frenet transitionnormalbinormalc ntjd} t|ttt frg|n|g} | D cgc]} | |  } } t | d| |L| "t| ttfr | js tdt| j} g| _ nHt|tr t|} n+t|tst|jn|} | sD| 7t| tr'| jr| j} g| _g| _n tdg}g}| D]\} t| t tt"fr |j%| j>|j%| j'^g}|r|| t)|}nt|t"r t|g}n|}|r8|Dcgc]}|j+}}t-j.|| d||g}n(|Dcgc]}t-j0|| d|||}}g}|r=| D]8} t3j0| | |}|j%|j':| | j4||z||dn<|r:|Dcgc]}|j7}}|Dcgc]}|j7}}|rt9t;|j<St?t;|j<Scc} w#t$r} td| d} ~ wwxYwcc}wcc}wcc}wcc}w) a{Generic Operation: sweep Sweep pending 1D or 2D objects along path. Args: sections (Compound | Edge | Wire | Face | Solid): cross sections to sweep into object path (Curve | Edge | Wire, optional): path to follow. Defaults to context pending_edges. multisection (bool, optional): sweep multiple on path. Defaults to False. is_frenet (bool, optional): use frenet algorithm. Defaults to False. transition (Transition, optional): discontinuity handling option. Defaults to Transition.TRANSFORMED. normal (VectorLike, optional): fixed normal. Defaults to None. binormal (Edge | Wire, optional): guide rotation along path. Defaults to None. clean (bool, optional): Remove extraneous internal structure. Defaults to True. mode (Mode, optional): combination. Defaults to Mode.ADD. sweepNzpath must be providedzUnable to build path from edgeszNo sections providedT)sectionr make_solidrr4rrB) rrDr<rFrIrLr rrrrJr-rrOrrr"r#rMrrrr+ sweep_multirr)rTr3r&r!r{r*)rrrrrrrr3r4r\ section_listsec path_wireerr edge_listrrc binormal_moderlrbsweptrms r>rrs8&227;G"(T5&,AB (  $0CrCs:"$ON**.   252"+))*='*=*=*?@   ; 'D[4'%/(:WDD5 -1 y! x( (y!l"T)y! y!  y!  y!z/3:/ Xe_ $t +:/ :/ d]:/z $f}9,8 !$( z1 *;!< <z1 z1T\z1 4< z1 d{T! z1  d] z1zd1 *;!< <d1 d1 d]Ud1Nt d*X5=FM IM+9=88. (:. . 5. . . V^dX% .bt e+h6 I6)9=)-$(M (:. . 5M MT$Z$&M  M  M  MT\M M V^dX%M`t+f4v= Y=,;?"-1 P 8K0 04 7Pt|P H t #d *P  P  V^h6!22 Ph/3-.L. Xe_ $t +L. eUE)**L. L. V^dX% L.^d{T)E1 91(7;&+hh 8 ), ,t 38t|e#8 8  8v $-4u< 9<(8<8<'33 $#'v/(9--4v/ $,  . 5v/v/ v/  v/   v/TkD v/ v/ v/ F]v/r@