,jfddlmZddlmZmZmZmZmZmZm Z m Z m Z m Z ddl mZmZddlmZddlmZddlmZddlmZdd lmZmZmZmZdd lmZdd lm Z m!Z!dd l"m#Z#m$Z$m%Z&m'Z'm(Z(dd l)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/ddl0m1Z2m3Z3m4Z4ddl5m6Z7ddl8m9Z9m:Z:m;Z;eeedfZdZ?dZ@e@ZAdZBGddeCZDdS))reduce) UnionOptionalListDictAnyoverloadTupleIteratorcastget_args)LiteralSelf)uuid1)warn)chain) Workplane)ShapeCompound isSubshapecompound)Location)ColorMaterial)ConstraintKindConstraintSolverConstraintSpecUnaryConstraintKindBinaryConstraintKind)exportAssembly exportCAF exportVTKJS exportVRML exportGLTFSTEPExportModeLiterals) importStep importXbf importXml)_expression_grammar) deprecateBiDict instance_ofNSTEPXMLXBF)r/r0r1GLTFVTKJSVRMLSTL/c8ddlm}m}m}m}m}m}|d}|d}||||dztd d}|||dz d } tj d } |d |d z|d z|dz d} |||| zz||| z|z| zzS)Nr)rWordralphas alphanums DelimitedList@?_T)combinenametagselectorsolidsfacesedgesvertices selector_kind) pyparsingrr8rr9r:r;suppress PATH_DELIMset_results_name_selector_grammar) rr8rr9r:r; Separator TagSeparatorNameTagSelector SelectorTypes K/DATA/AppData/hermes/venv/lib/python3.11/site-packages/cadquery/assembly.py_define_grammarrT5s %%''I73<<((**L = VY_%%z4   v  $vy3 ' ' 8 8 ? ?C 1*==H GGG,,,www/?/??''*BUBUU'' (<#% & & ' (9|+i7(B C C DcNt|trt|n|S)z/ Convert string to Material if needed. ) isinstancestrr)materials rS_ensure_materialrZWs&",Hc!:!: H8H   HrUceZdZUdZeed<eed<eeed<ee ed<e ee fed<e ed<eded<e ded <e edfed <e eed <eeefed <eeefed <eeefed<ee ee fed< dHde deedeedeedee dee ee ff dZdIdZe dJdddeedeedeedeee efdef dZe dKde deedeedeedeee efdee ee fdefdZdZdeddfdZdedeeeeffdZdedeeeffdZe dLdededede def dZedLdedede defd Ze dLd!ed"ed#ed$edede defd%Ze dLd!ed"edede def d&Zdd'd(ZdMd*edefd+Ze dNd.ed/ee d0e!d1e"d2e"def d3Z# dNd.ed/ee d0e!d1e"d2e"def d4Z$e%d.edefd5Z&e%dLd.ed6ee'defd7Z(e)de efd8Z*de+eedffd9Z,gfd:Z- dOdeedeedeede+eeeeeeffd;Z.de/fd<Z0d=Z1 dOd>edeedeed?eeddf d@Z2dededeffdAZ3de efdBZ4dede5fdCZ6dededeffdDZ7dEZ8dFZ9dGZ:dS)PAssemblyzQNested assembly of Workplane and Shape objects defining their relative positions.locr@colorrYmetadataobjparentchildrenobjects constraints_subshape_names_subshape_colors_subshape_layers _solve_resultNc||_|r|n t|_|r|ntt |_|r|nd|_|r|nd|_|r|ni|_d|_ g|_ g|_ |j|i|_ d|_ t|_t|_t|_dS)aK construct an assembly :param obj: root object of the assembly (default: None) :param loc: location of the root object (default: None, interpreted as identity transformation) :param name: unique name of the root object (default: None, resulting in an UUID being generated) :param color: color of the added object (default: None) :param material: material (for visual and/or physical properties) of the added object (default: None) :param metadata: a store for user-defined metadata (default: None) :return: An Assembly object. To create an empty assembly use:: assy = Assembly(None) To create one constraint a root object:: b = Workplane().box(1, 1, 1) assy = Assembly(b, Location(Vector(0, 0, 1)), name="root") N)r`rr]rXuuidr@r^rYr_rarbrdrcrhr,rerfrgselfr`r]r@r^rYr_s rS__init__zAssembly.__init__us@-338:: 1DDc$&&kk #-UU $,6$ $,4"    4( !%xx & &rUreturnc||j|j|j|j|j|j}t|j|_t|j |_ t|j |_ |j D]e}| }||_ |j |||j|j<|j|jf|S)z1 Make a deep copy of an assembly ) __class__r`r]r@r^rYr_r,rfrergrb_copyraappendrcupdate)rlrvchch_copys rSrqzAssembly._copys ^^ Hdh 4:t}dm  %T%:;;#D$899$T%:;;- / /BhhjjGGN K  w ' ' ''.BJw| $ J  go . . . . rUcdS)a Add a subassembly to the current assembly. :param obj: subassembly to be added :param loc: location of the root object (default: None, resulting in the location stored in the subassembly being used) :param name: unique name of the root object (default: None, resulting in the name stored in the subassembly being used) :param color: color of the added object (default: None, resulting in the color stored in the subassembly being used) N)rlr`r]r@r^rYs rSaddz Assembly.adds ( rUcdS)a~ Add a subassembly to the current assembly with explicit location and name. :param obj: object to be added as a subassembly :param loc: location of the root object (default: None, interpreted as identity transformation) :param name: unique name of the root object (default: None, resulting in an UUID being generated) :param color: color of the added object (default: None) :param material: material (for visual and/or physical properties) of the added object (default: None) :param metadata: a store for user-defined metadata (default: None) Nrxrks rSryz Assembly.adds . rUc t|tr||dr|dn|j}||jvrt d|d|}|dr|dn|j|_|dr|dn|j|_|dr|dn|j|_t|dr|dn|j |_ |dr|dn|j |_ ||_ |j ||j|nFd|vrt|d|d<|j|fi|}||_ |||S)z< Add a subassembly to the current assembly. r@zUnique name is required. z is already in the assemblyr]r^rYr_)rWr\getr@rc ValueErrorrqr]r^rZrYr_rarbrrrs_flattenrpry)rlargkwargsr@subassyassys rSryz Assembly.adds c8 $ $" &,ZZ%7%7E6&>>SXDt|## QQQQiikkG+1::e+<+<I&--#'GK-3ZZ-?-?M6&>>SXGL/5zz'/B/BQF7OO GM/&,jj&<&<Nz""#,  G '-jj&<&<Nz""#,  "GN M  ) ) ) L   0 0 2 2 3 3 3 3V##%5fZ6H%I%Iz"!4>#0000DDK HHTNNN rUc6||jvrtd|d|j|}|jr|jj||j|=|D]}||jvr|j|=d|_|S)aQ Remove a part/subassembly from the current assembly. :param name: Name of the part/subassembly to be removed :return: The modified assembly *NOTE* This method can cause problems with deeply nested assemblies and does not remove constraints associated with the removed part/subassembly. zNo object with name 'z' found in the assemblyN)rcr}rarbremover~keys)rlr@ to_removedescendant_names rSrzAssembly.removes t| # #RTRRRSS SL&    8   % , ,Y 7 7 7 L  )113388:: 2 2O$,..L1  rUqc*t|d}|j}|j|j}t |t r"|jr||j}nMt |t tfr"t |}ntd|j r$t||j|j }n|}|}|t |tr|ndfS)a3 Execute a selector query on the assembly. The query is expected to be in the following format: name[?tag][@kind@args] valid example include: obj_name @ faces @ >Z obj_name?tag1@faces@>Z obj_name ? tag obj_name Tz2Workplane or Shape required to define a constraintN)_grammar parse_stringr@rcr`rWrrA _getTaggedrryr}rBgetattrrGval)rlrqueryr@r`tmpresrs rS_queryzAssembly._query;s&%%a..Jl4 $ c9 % % S%) S..++CC i/ 0 0 S++//#&&CCQRR R > 3'#u233ENCCCCCggiiJsE22z"Assembly._subloc..vs rBwrU) rrcrbrarrr]r r\r@r)rlr@rtr`name_outlocss rS_subloczAssembly._subloccsZZl4  dm # #4DjD(( CG$$$8SZ008jD(( ..55BH~rUq1q2kindparamcdSrrx)rlrrrrs rS constrainzAssembly.constrainz  rUcdSrrx)rlrrrs rSrzAssembly.constrains rUid1s1id2s2cdSrrx)rlrrrrrrs rSrzAssembly.constrains  rUcdSrrx)rlrrrrs rSrzAssembly.constrainrrU)rcbt|dkr|\}}||\}}n-t|dkr:t|dtr|\}}}||\}}nt|dkr7|\}}}||\}}||\}} nt|dkr8|\}}}}||\}}||\}} nKt|dkr |\}}}} }n/t|dkr |\}}}} }}nt d|t|tr/||\} } t | f|f| f||} nqt|trJ||\} } ||\} }t | |f|| f| | f||} nt d||j | |S) z* Define a new constraint. rzIncompatible arguments: zUnknown constraint: ) lenrr-rr}r Constraintr rdrr)rlrargsrrrrrrrloc1id1_topcloc2id2_tops rSrzAssembly.constrains0 t99>>HBkk"ooGC YY!^^ DG5H I I^"OBekk"ooGC YY!^^LBDkk"ooGCkk"ooGC YY!^^"& BD%kk"ooGCkk"ooGC YY!^^%) "CS"dd YY!^^,0 )CS"dEE>>>?? ? t0 1 1 < LL--MD'G:utgtUCCAA 3 4 4 < LL--MD' LL--MD'GW-Bx$tUSSAA:D::;; ; """ rUr verbosityclid}g}jD]U}|jD]K}|vr ||<|dz }|jdks |jkr%||vr||LV|sHdjD}djD}|D]#}||vr||n$|s|dfdD} g} jD]R}t fd|jD} |} | D]} | | | fS| stdtd krtd  j }t| | || }| |\}_t}jr*t#|D]\}}|jkr |j}nt#|D]%\}}|jkr||zj|_&S) z( Solve the constraints. rrFixedc\g|])}t|jt|jd*Sr)r-rrrc.0rs rS z"Assembly.solve..sAqv':;; ! rUc\g|])}t|jt|jd*Sr)r-rr rcrs rSrz"Assembly.solve..sAqv';<< ! rUc4g|]}j|jSrx)rcr])rnrls rSrz"Assembly.solve..s"222 Q#222rUc3(K|] }|V dSrrx)rr`entss rS z!Assembly.solve..s'77cS 777777rUz At least one constraint requiredrz,At least two entities need to be constrained)lockedscale)rdrcrr@rrtupletoPODsr}r toCompound BoundingBoxDiagonalLengthrsolverhrr`zipinverser])rlrirrr@ unary_objectsbinary_objectsbrrdixspodspodrsolverlocs_new loc_root_invloc_newrrs` @rSrzAssembly.solves  ! . .A  . .t##!"DJFAFg%%):):AA A MM$t*--- . )M )N $  M))MM$q'***E*   MM!   2222T222 ! / /A7777QY77777C88::D / /""C:.... / A?@@ @ t99q==KLL L!!--//>!$ F%PPP(.||I'>'>$$$  zz 8 !(D11    >>#*?LE" h-- = =JGQDI~~&2W&< Q# rUdefault皙?path exportTypemode toleranceangularTolerancec &|j|||||fi|S) Save assembly to a file. :param path: Path and filename for writing. :param exportType: export format (default: None, results in format being inferred form the path) :param mode: STEP only - See :meth:`~cadquery.occ_impl.exporters.assembly.exportAssembly`. :param tolerance: the deflection tolerance, in model units. Only used for glTF, VRML. Default 0.1. :param angularTolerance: the angular tolerance, in radians. Only used for glTF, VRML. Default 0.1. :param \**kwargs: Additional keyword arguments. Only used for STEP, glTF and STL. See :meth:`~cadquery.occ_impl.exporters.assembly.exportAssembly`. :param ascii: STL only - Sets whether or not STL export should be text or binary :type ascii: bool )export)rlrrrrrrs rSsavez Assembly.saves30t{ *dI/?  CI   rUc |ttvrtd|d|V|dd}|dvrt t |}ntd|dkrt|||fi|n|d krt||n|d krt||d n|d krt||||n|dks|dkrt||d||n|dkrt||nk|dkrSd}d|vr"t| d}|||||ntd||S)rzUnknown assembly export mode z for STEPN.)r/r0r1r4r3r2GLBr51Unknown extension, specify export type explicitlyr/r0r1T)binaryr4r2rr3r5FasciizUnknown format: )r r&r}splitupperr ExportLiteralsr!r"r$r%r#boolr|r exportStl) rlrrrrrrt export_asciis rSrzAssembly.export2s0 x 677 7 7LTLLLMM M   3#))++AQQQ!.!44  !TUUU    4t 6 6v 6 6 6 6 5 dD ! ! ! ! 5 dD . . . . . 6 ! ! tT9.> ? ? ? ? 6 ! !Z5%8%8 tT44D E E E E 7 " " d # # # # 5  L&  #FJJw$7$788 OO   ' 'i9I< X X X X< <<== = rUc0||dS)z Reads an assembly from a STEP file. :param path: Path and filename for reading. :return: An Assembly object. r/) importType)load)clsrs rSr'zAssembly.importSteplsxxx000rUrcR|V|dd}|dvrtt|}nt d|}|dkrt ||n-|dkrt ||n|dkrt|||S) z( Load step, xbf or xml. Nrrr.rr/r0r1)rrr ImportLiteralsr} _importStepr)r()rrrrrs rSrz Assembly.loadws   3#))++A***!.!44  !TUUUsuu    d # # # # 5 dD ! ! ! ! 5 dD ! ! ! rUcg}t|jtr |jg}n=t|jtr#d|jD}|S)z9 List of Shape objects in the .obj field c<g|]}t|t|SrxrWr)rels rSrz#Assembly.shapes..s'HHH*R2G2GH"HHHrU)rWr`rrvals)rlrts rSshapeszAssembly.shapess`  dh & & I(BB ) , , IHHtx}}HHHB rUc#jK|jD]}|D]}|V|j|fVdS)zA Yield (name, child) pairs in a bottom-up manner N)rbtraverser@)rlrurs rSrzAssembly.traverses[ -  Bkkmm   y$rUci}|jD]4}||||jgz5||t||jgz<|S)za Generate a dict with all ancestors with keys indicating parent-child relations. )parents)rbrsr~r@rJjoin)rlrrtrus rSr~zAssembly._flattensl - B BB IIbkk'TYK*?k@@ A A A A59:??7di[0 1 12 rUc#K|r |d|jn|j}|r ||jzn|j}|jr|jn|}|jr\t |jt r|jn4t jd|jD|||fV|j D]}| |||Ed{V dS)zQ Assembly iterator yielding shapes, names, locations and colors. r6c3DK|]}t|t|VdSrr)rss rSrz$Assembly.__iter__..sNUUjE.B.BUUUUUUUrUN) r@r]r^r`rWrr makeCompoundrrb__iter__)rlr]r@r^rus rSr zAssembly.__iter__s)-;$$$$$$$) #1cDHnn"j3 e 8 (599$((x?TUU8==??UUU@@S%    - 5 5B{{3e44 4 4 4 4 4 4 4 4 5 5rUc|j}|d|jDtj||jS)z Returns a Compound made from this Assembly (including all children) with the current Locations applied. Usually this method would only be used after solving. c3>K|]}|VdSr)r)rchilds rSrz&Assembly.toCompound..s.EEeu''))EEEEEErU)rextendrbrr locater])rlrs rSrzAssembly.toCompoundsO  EEt}EEEFFF$V,,33DH===rUcHddlm}||S)z3 Jupyter 3D representation support r)display)occ_impl.jupyter_toolsr_repr_javascript_)rlrs rSrzAssembly._repr_javascript_s0 433333wt}}..000rUrlayerc@tfd|jDr|}nUtdd}|jD](}tfd|jDr|}d}n)|st d|r ||j<|r ||j<|r ||j<|S)aR Handles name, color and layer metadata for subshapes. :param s: The subshape to add metadata to. :param name: The name to assign to the subshape. :param color: The color to assign to the subshape. :param layer: The layer to assign to the subshape. :return: The modified assembly. c38K|]}t|VdSrrrr`rs rSrz'Assembly.addSubshape..s-99cz!S!!999999rUzlCurrent node does not contain any Shapes, searching in subnodes. In the future this will result in an error.Fc38K|]}t|VdSrrrs rSrz'Assembly.addSubshape..s-??cz!S))??????rUTz6 is not a subshape of the current node or its children)anyrrrbr}rerfrg)rlrr@r^rrfoundrus ` rS addSubshapezAssembly.addSubshapes $ 9999T[999 9 9 DD ~   Em  ????RY?????D EE   PPP  +&*D  #  -',D !! $  -',D !! $ rUc||jvr |j|S||jjvr<|jj|}t|dkr|dnt |St )z/ [] based access to children. rr)rcreinvrrKeyErrorrlr@rts rS __getitem__zAssembly.__getitem__sf 4<  <% % T)- - -%)$/BGGqLL2a55hrll :rUctt|j|jjS)z0 IPython autocompletion helper. )listrrcrrerrls rS_ipython_key_completions_z"Assembly._ipython_key_completions_s< E$,++--t/C/G/L/L/N/NOOPPPrUc0||jvp ||jjvSr)rcrer)rlr@s rS __contains__zAssembly.__contains__%st|#Gtt/C/G'GGrUc||jvr |j|S||jjvr<|jj|}t|dkr|dnt |St |d|)z- . based access to children. rrz is not an attribute of )rcrerrrAttributeErrorr s rS __getattr__zAssembly.__getattr__)s} 4<  <% % T)- - -%)$/BGGqLL2a55hrll :DDdDDEEErUct|jtd|jDzt|jjzS)z6 Modified __dir__ for autocompletion. c3$K|] }|jV dSr)r@)rrus rSrz#Assembly.__dir__..=s$33r27333333rU)r#__dict__rbrerrr$s rS__dir__zAssembly.__dir__6sY   33T]33333 44'+002233 4 rUc|jS)z: Explicit getstate needed due to getattr. r-r$s rS __getstate__zAssembly.__getstate__As }rUc||_dS)z: Explicit setstate needed due to getattr. Nr0)rlds rS __setstate__zAssembly.__setstate__Hs  rU)NNNNNN)rnr\)NNNN)NNNNNrr)Nrrr)NNN);__name__ __module__ __qualname____doc__r__annotations__rXrrrrrAssemblyObjectsrrr,rrmrqr rrryrr rrrrintrr+rr&floatrr classmethodr'rrpropertyrr rr~r rrrrr!r%rr'r*r.r1r4rxrUrSr\r\^s[[ MMM III E?x    38n  Z    : #z/ """"j!!!!E3J''''UE\****UCZ((((DcN++++ $"&"!%'+-10)0) 0)h 0)sm 0)  0) 8$ 0)4S>*0)0)0)0)d.#'"!%37    h  sm    53/0      X *#'"!%37-1   h  sm    53/0  4S>*     X 0)))V!3!:!!!!F&=&=c8E?&: ;&=&=&=&=PCE(C-$8.CG    &4 =@    X   C ~ c T   X                      X  FJ   ! )7 @C    X &*(((((TTTsT4TTTTlY[[04'0"%    ^, %          [ <04'0"% 888^,8% 8  8  8 8888t1c1d111[1.)Ad[. U   X   (5j#9:      "     #'"!% 55 h 5sm5 5 %sHhuo=> ? 5555, >H > > > >111#!%# -- -sm- - } -  ----^  j%.?(@    Q49QQQQHHHHHH F Fj%.?(@ F F F F    rUr\)E functoolsrtypingrrrrrr r r r r typing_extensionsrrrjrwarningsr itertoolsrcqrocc_impl.shapesrrrr occ_impl.geomrocc_impl.assemblyrrocc_impl.solverrrrrrr occ_impl.exporters.assemblyr!r"r#r$r%r&occ_impl.importers.assemblyr'rr(r) selectorsr*rLutilsr+r,r-r:rrrJrTrrZobjectr\rxrUrSrNs                        ,+++++++BBBBBBBBBBBB######........YXXXXXXXXX??????1111111111y$./-.MN  > ?  IIIo o o o o vo o o o o rU