vjf|ddlmZddlmZmZmZmZmZmZm 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@ZAdZBGddeCZDy))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/cddlm}m}m}m}m}m}|dj}|dj}||||dztdjd}|||dzjd } tjd } |d |d z|d z|dzjd} |||| 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 >/DATA/.local/lib/python3.12/site-packages/cadquery/assembly.py_define_grammarrT5s %%'I3<((*L  VY_%z4 v  vy3 ' 8 8 ?C 11*=H GG,,ww/??'*BUU' <#% & ' 9|+i7(B C Dc<t|tr t|S|S)z/ Convert string to Material if needed. ) isinstancestrr)materials rS_ensure_materialrZWs",Hc!:8H HHrUc(eZdZUdZeed<eed<eeed<ee ed<e ee fed<e ed<eded<e ded <e edfed <e eed <eeefed <eeefed <eeefed<ee ee fed< dEde deedeedeedee dee ee ff dZdFdZe dGdddeedeedeedeee efdef dZe dHde deedeedeedeee efdee ee fdefdZdZdeddfdZdedeeeeffdZdedeeeffdZe dIdededede def dZedIdedede defd Ze dId!ed"ed#ed$edede defd%Ze dId!ed"edede def d&Zdd'd(ZdJd)edefd*Ze dKd+ed,ee d-e!d.e"d/e"def d0Z# dKd+ed,ee d-e!d.e"d/e"def d1Z$e%d+edefd2Z&e%dId+ed3ee'defd4Z(e)de efd5Z*de+eedffd6Z,gfd7Z- dLdeedeedeede+eeeeeeffd8Z.de/fd9Z0d:Z1 dLd;edeedeedZ3de efd?Z4dede5fd@Z6dededeffdAZ7dBZ8dCZ9dDZ:y)MAssemblyzQNested assembly of Workplane and Shape objects defining their relative positions.locr@colorrYmetadataobjparentchildrenobjects constraints_subshape_names_subshape_colors_subshape_layers _solve_resultNcj||_|r|n t|_|r|ntt |_|r|nd|_|r|nd|_|r|ni|_d|_ g|_ g|_ |j |i|_ d|_ t|_t|_t|_y)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@38: Dc$&k #U $,$ $,"    4( !%x & &rUreturncH|j|j|j|j|j|j |j }t|j|_t|j|_ t|j|_ |jD]r}|j}||_ |jj|||j|j<|jj!|jt|S)z1 Make a deep copy of an assembly ) __class__r`r]r@r^rYr_r,rfrergrb_copyraappendrcupdate)rlrvchch_copys rSrqzAssembly._copys ^^ HHdhh 4::t}}dmm %T%:%:;#D$8$89$T%:%:;--BhhjGGN KK  w ''.BJJw|| $ JJ  goo .   rUcy)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( rUcy)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 xt|trh|jdr|dn |j}||jvrt d|d|j }|jdr|dn |j|_|jdr|dn |j|_|jdr|dn |j|_t|jdr|dn |j|_ |jdr|dn |j|_ ||_ |jj||jj|j!|Sd|vrt|d|d<|j"|fi|}||_ |j%||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%76&>SXXDt||# /v5PQiikG+1::e+<&-#''GK-3ZZ-?6&>SXXGL/5zz'/BF7O GM/&,jj&<z"#,, G '-jj&<z"#,,  "GN MM  ) LL   0 0 2 3 V#%5fZ6H%Iz"!4>>#00DDK HHTN rUcf||jvrtd|d|j|}|jr%|jjj ||j|=|j j D]}||jvs|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|| #4TF:QRS SLL&        % % , ,Y 7 LL  )11388:O$,,.LL1 ;   rUqctj|d}|j}|j|j}t |t r(|jr|j|j}n;t |t tfrt j|}n td|jr't||j|j}n|}|j}|t |tr|fSdfS)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.JJll4 $$ c9 %%))..+C i/ 0+//#&CQR R >>3'#u223ENNCCCggiJsE2S<<<t}|j|}|}||jvrq||urmg}|j|urP|j |j t t|j}|j}|j|urPtd|}||fS)z Calculate relative location of an object in a subassembly. Returns the relative positions as well as the name of the top assembly. c ||zSNrx)l1l2s rSz"Assembly._subloc..vsrBwrU) rrcrbrarrr]r r\r@r)rlr@rtr`name_outlocss rS_subloczAssembly._subloccsZll4  dmm #4DjjD( CGG$8SZZ088jjD( .5BH~rUq1q2kindparamcyrrx)rlrrrrs rS constrainzAssembly.constrainz rUcyrrx)rlrrrs rSrzAssembly.constrains rUid1s1id2s2cyrrx)rlrrrrrrs rSrzAssembly.constrains rUcyrrx)rlrrrrs rSrzAssembly.constrainrrU)rct|dk(r|\}}|j|\}}nt|dk(r.t|dtr|\}}}|j|\}}nt|dk(r/|\}}}|j|\}}|j|\}} n{t|dk(r0|\}}}}|j|\}}|j|\}} n=t|dk(r |\}}}} }n&t|dk(r |\}}}} }}nt d|t|tr'|j |\} } t | f|f| f||} n\t|tr>|j |\} } |j \} }t | |f| f| | f||} nt d||jj| |S) z* Define a new constraint. rzIncompatible arguments: zUnknown constraint: ) lenrr-rr}r Constraintr rdrr)rlrargsrrrrrrrloc1id1_topcloc2id2_tops rSrzAssembly.constrains t9>HBkk"oGC Y!^ DG5H I"OBekk"oGC Y!^LBDkk"oGCkk"oGC Y!^"& BD%kk"oGCkk"oGC Y!^%) "CS"d Y!^,0 )CS"dE7v>? ? t0 1 LL-MD'G:utgtUCA 3 4 LL-MD' LL-MD'GW-Bx$tUSA3D6:; ; " rU verbositycfid}g}|jD]\}|jD]K}|vr ||<|dz }|jdk(s||jk(s0||vs8|j |M^|s|jDcgc]+}t |jt r|jd-}}|jDcgc]+}t |jtr|jd-}}|D]}||vs|j |n|s|j dD cgc]} |j| j} } g} |jD]J}tfd|jD} |j} | D]}| j | |fL| s tdtdkr td|jjj}t!| | ||}|j#|\}|_t'}|j(r1t+|D]"\}} | |jk(s|j,}nt+|D],\}} | |jk7s||z|j| _.|Scc}wcc}wcc} w) z( Solve the constraints. rrFixedc3(K|] }| ywrrx).0r`entss rS z!Assembly.solve..s7YcS Ysz At least one constraint requiredrz,At least two entities need to be constrained)lockedscale)rdrcrr@rrr-rr r]tupletoPODsr}r toCompound BoundingBoxDiagonalLengthrsolverhrr`zipinverse)rlrirrr@ unary_objectsbinary_objectsbnrrdixspodspodrsolverlocs_new loc_root_invloc_newrs @rSrzAssembly.solves  !!A t#!"DJFAFFg%):AA MM$t*-"")))Aqvv':; ! ) )))Aqvv';< ! )  $M)MM$q'*$  MM! -12T Q##T2 !!A7QYY77C88:D""C:. "?@ @ t9q=KL L!--/>>!$ F%P(.||I'>$$$  z 88!(D1  >#*??L2 h-JGQDII~&2W&< Q#. { 3s0J$0J)- J.path exportTypemode toleranceangularTolerancec 0|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.saves+0t{{ *dI/? CI  rUc x|ttvrtd|d|B|jddj }|dvrt t |}n td|dk(rt|||fi||S|d k(rt|||S|d k(rt||d |S|d k(rt|||||S|dk(s|dk(rt||d|||S|dk(rt|||S|dk(rDd}d|vrt|jd}|jj|||||Std|)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 67 7 ? 6 !Z5%8 tT44D E 7 " d # 5  L& #FJJw$78 OO  ' 'i9I< X / |<= =rUc(|j|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.importSteplsxxx00rUrc |B|jddj}|dvrtt|}n t d|}|dk(rt |||S|dk(rt |||S|dk(r t|||S)z( Load step, xbf or xml. rrr.rr/r0r1)rrr ImportLiteralsr} _importStepr)r()rrrrrs rSrz Assembly.loadws   3#))+A**!.!4  !TUUu   d #  5 dD ! 5 dD ! rUc g}t|jtr|jg}|St|jtr8|jj Dcgc]}t|ts|}}|Scc}w)z9 List of Shape objects in the .obj field )rWr`rrvals)rlrtels rSshapeszAssembly.shapessk  dhh &((B ) ,#xx}}H*R2G"BH Is "B8Bc#K|jD]}|jD]}||j|fyw)zA Yield (name, child) pairs in a bottom-up manner N)rbtraverser@)rlrurs rSrzAssembly.traverses< --Bkkm$ yy$s<>ci}|jD]1}|j|j||jgz3||tj ||jgz<|S)za Generate a dict with all ancestors with keys indicating parent-child relations. )parents)rbrsr~r@rJjoin)rlrrtrus rSr~zAssembly._flattens^ --B IIbkk'TYYK*?k@ A 6::??7dii[0 12 rUc#K|r|d|jn |j}|r||jzn |j}|jr |jn|}|jr`t |jt r |jn3t jd|jjD|||f|jD]}|j|||Ed{y7w)zQ Assembly iterator yielding shapes, names, locations and colors. r6c3BK|]}t|ts|ywr)rWr)rss rSrz$Assembly.__iter__..sU*ajE.B?sN) r@r]r^r`rWrr makeCompoundrrb__iter__)rlr]r@r^rus rSrzAssembly.__iter__s)-$q $$)) #cDHHn"jj e 88(59$((x?T?TU88==?U@S%  --B{{3e4 4 4 4sC(C4*C2+C4c|j}|jd|jDtj|j |j S)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|]}|jywr)r)rchilds rSrz&Assembly.toCompound..sE}eu'')}s)rextendrbrrlocater])rlrs rSrzAssembly.toCompoundsC  Et}}EF$$V,33DHH==rUc:ddlm}||jS)z3 Jupyter 3D representation support r)display)occ_impl.jupyter_toolsr_repr_javascript_)rlrs rSr zAssembly._repr_javascript_s 4t}..00rUrlayercTtfd|jDr|}nRtdd}|jD]&}tfd|jDs"|}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. c36K|]}t|ywrrrr`rs rSrz'Assembly.addSubshape..s9[cz!S![zlCurrent node does not contain any Shapes, searching in subnodes. In the future this will result in an error.Fc36K|]}t|ywrr rs rSrz'Assembly.addSubshape..s?Ycz!S)YrTz6 is not a subshape of the current node or its children)anyrrrbr}rerfrg)rlrr@r^r rfoundrus ` rS addSubshapezAssembly.addSubshapes$ 9T[[9 9D ~ Emm?RYY??D E $  cOP &*D  # ',D ! !! $ ',D ! !! $ rUc||jvr|j|S||jjvr7|jj|}t|dk(r|dSt |St )z/ [] based access to children. rr)rcreinvrrKeyErrorrlr@rts rS __getitem__zAssembly.__getitem__sk 4<< <<% % T))-- -%%))$/BGqL2a5 :hrl :rUctt|jj|jj jS)z0 IPython autocompletion helper. )listrrcrrerrls rS_ipython_key_completions_z"Assembly._ipython_key_completions_s8 E$,,++-t/C/C/G/G/L/L/NOPPrUcR||jvxs||jjvSr)rcrer)rlr@s rS __contains__zAssembly.__contains__%s't||#Gtt/C/C/G/G'GGrUc||jvr|j|S||jjvr7|jj|}t|dk(r|dSt |St |d|)z- . based access to children. rrz is not an attribute of )rcrerrrAttributeErrorrs rS __getattr__zAssembly.__getattr__)s} 4<< <<% % T))-- -%%))$/BGqL2a5 :hrl :v%=dVDEErUct|jtd|jDzt|jjj zS)z6 Modified __dir__ for autocompletion. c34K|]}|jywr)r@)rrus rSrz#Assembly.__dir__..=s3]r277]s)r__dict__rbrerrrs rS__dir__zAssembly.__dir__6sN  3T]]33 44''++0023 4 rUc|jS)z: Explicit getstate needed due to getattr. r$rs rS __getstate__zAssembly.__getstate__As }}rUc||_y)z: Explicit setstate needed due to getattr. Nr')rlds rS __setstate__zAssembly.__setstate__Hs  rU)NNNNNN)rnr\)NNNN)NNNNNr)r)Ndefault皙?r-)NNN);__name__ __module__ __qualname____doc__r__annotations__rXrrrrrAssemblyObjectsrrr,rrmrqr rrryrr rrrrintrr+rr&floatrr classmethodr'rrpropertyrr rr~rrrr rrrrrr!r%r(r+rxrUrSr\r\^s$[ M I E?x  38n  Z  : #z/ ""j!!E3J''UE\**UCZ((DcN++ $"&"!%'+-10) 0)h 0)sm 0)  0) 8$ 0)4S>*0)d.#'"!%37   h  sm    53/0     *#'"!%37-1  h  sm    53/0  4S>*    0)V!3!:!F&=&=c8E?&: ;&=PCE(C-$8.CG   &4 =@    C ~ c T                      FJ  ! )7 @C   &*(TTsT4Tl[04'0"%   ^, %         <04'0"% 88^,8% 8  8  8 8t1c1d11.)Ad. U    (5j#9:   "  #'"!% 5 h 5sm5 5 %sHhuo=> ? 5, >H >1#!%# - -sm- - } -  -^  j%.?(@ Q49QHHH F Fj%.?(@ 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-r3rrrJrTrrZobjectr\rxrUrSrGs   ,BB#.YX?11y$./-.MN  >  Io vo rU