,jcZddlmZmZddlZddlmZddlmZmZm Z m Z m Z ddl m Z mZmZmZmZmZmZmZmZmZddlmZddlmZmZmZmZmZed e ZGd d e Z!Gd de!Z"Gdde!Z#Gdde!Z$Gdde$Z%Gdde$Z&Gdde$Z'Gdde!Z(Gdde!eZ)Gdde)Z*Gdd e)Z+Gd!d"e+Z,Gd#d$e%e+Z-Gd%d&e)Z.Gd'd(e)Z/Gd)d*e!Z0Gd+d,e0Z1Gd-d.e0Z2Gd/d0e0Z3Gd1d2e!Z4d3Z5e5Z6Gd4d5e!Z7d6Z8e8e6Z9Gd7d8e!Z:dS)9)abstractmethodABCN)Vector) ShapeProtocolShape1DProtocol FaceProtocol geom_LUT_EDGE geom_LUT_FACE) pyparsing_commonLiteralWordnumsOptionalCombineone_ofGroupinfix_notationopAssoc)reduce)IterableListSequenceTypeVarcastShape)boundcReZdZdZdeedeefdZdZdZ dZ dZ d S) Selectorzd Filters a list of objects. Filters must provide a single method that filters objects. objectListreturnc t|S)z Filter the provided list. The default implementation returns the original list unfiltered. :param objectList: list to filter :type objectList: list of OCCT primitives :return: filtered list )listselfr s L/DATA/AppData/hermes/venv/lib/python3.11/site-packages/cadquery/selectors.pyfilterzSelector.filter$sJc"t||SN) AndSelectorr%others r&__and__zSelector.__and__04'''r(c"t||Sr*) SumSelectorr,s r&__add__zSelector.__add__3r/r(c"t||Sr*)SubtractSelectorr,s r&__sub__zSelector.__sub__6se,,,r(c t|Sr*)InverseSelector)r%s r&__neg__zSelector.__neg__9st$$$r(N) __name__ __module__ __qualname____doc__rrrr'r.r2r5r8r(r&rrs  %  T%[     ((((((---%%%%%r(rc0eZdZdZdZdeefdZdS)NearestToPointSelectora Selects object nearest the provided point. If the object is a vertex or point, the distance is used. For other kinds of shapes, the center of mass is used to to compute which is closest. Applicability: All Types of Shapes Example:: CQ(aCube).vertices(NearestToPointSelector((0, 1, 0))) returns the vertex of the unit cube closest to the point x=0,y=1,z=0 c||_dSr*)pnt)r%rAs r&__init__zNearestToPointSelector.__init__Os r(r c2fd}t||gS)Nct|tjjSr*)CentersubrrALength)tShaper%s r&distz+NearestToPointSelector.filter..distSs)==??&&vtx'899@ @r(key)min)r%r rIs` r&r'zNearestToPointSelector.filterRs9 A A A A AJD)))**r(Nr9r:r;r<rBrrr'r=r(r&r?r?=sL"+%++++++r(r?c2eZdZdZddZdeefdZdS) BoxSelectora^ Selects objects inside the 3D box defined by 2 points. If `boundingbox` is True only the objects that have their bounding box inside the given box is selected. Otherwise only center point of the object is tested. Applicability: all types of shapes Example:: CQ(aCube).edges(BoxSelector((0, 1, 0), (1, 2, 1))) FcLt||_t||_||_dSr*)rp0p1test_boundingbox)r%point0point1 boundingboxs r&rBzBoxSelector.__init__hs&&/&/ +r(r c g}|j\ |j\  fd}|D]}|jr||}|t |j|j|jr>|t |j |j |j r| ||| r| ||S)Nc|jk|jkz o-|jk|jkz o|jk|jkz Sr*)xyz)px0x1y0y1z0z1s r& isInsideBoxz'BoxSelector.filter..isInsideBoxssP#(qsRx(.cBh138,.cBh138, r()rQtoTuplerRrS BoundingBoxrxminyminzminxmaxymaxzmaxappendrE) r%r resultrcobbr]r^r_r`rarbs @@@@@@r&r'zBoxSelector.filterms+W__&& BW__&& B           % %A$ %]]__;vbgrw@@AA%kk27BGRW55GG%MM!$$$;qxxzz**%MM!$$$ r(N)FrMr=r(r&rOrOYsQ  ,,,, %r(rOc\eZdZdZd dedefdZdedefdZd e e de e fd Z d S) BaseDirSelectorzV A selector that handles selection on the basis of a single direction vector. -C6?vector tolerancec"||_||_dSr*) directionrt)r%rsrts r&rBzBaseDirSelector.__init__s"r(vecr!cdS)zMTest a specified vector. Subclasses override to provide other implementationsTr=r%rws r&testzBaseDirSelector.teststr(r cg}|D]}|dkrA|dkr)tt|d}nY|dkr@|dkr(tt |}n||r|||S)z There are lots of kinds of filters, but for planes they are always based on the normal of the plane, and for edges on the tangent vector along the edge FacePLANENEdgeLINE) ShapeTypegeomTyperr normalAtr tangentAtrzrl)r%r rrn test_vectors r&r'zBaseDirSelector.filters   A{{}}&&1::<<7+B+B#<33<**3//$'A+=>>OOr(Nrr=r(r&rrsH&PP4PPPPPPr(rcFeZdZdZdefdZdeedeefdZ dS) TypeSelectora Selects objects having the prescribed geometry type. Applicability: Faces: PLANE, CYLINDER, CONE, SPHERE, TORUS, BEZIER, BSPLINE, REVOLUTION, EXTRUSION, OFFSET, OTHER Edges: LINE, CIRCLE, ELLIPSE, HYPERBOLA, PARABOLA, BEZIER, BSPLINE, OFFSET, OTHER You can use the string selector syntax. For example this:: CQ(aCube).faces(TypeSelector("PLANE")) will select 6 faces, and is equivalent to:: CQ(aCube).faces("%PLANE") typeStringc8||_dSr*)upperr)r%rs r&rBzTypeSelector.__init__s$**,,r(r r!cxg}|D]4}||jkr||5|Sr*)rrrl)r%r rrns r&r'zTypeSelector.filter s@   Azz||t.. r(N) r9r:r;r<strrBrrrr'r=r(r&rrsc"-3----%T%[r(rceZdZdZddededefdZdee d e e fd Z e d e d efd Z dee d e e e fd ZdS) _NthSelectorzj An abstract class that provides the methods to select the Nth object/objects of an ordered list. Trrn directionMaxrtc0||_||_||_dSr*)rrrt)r%rrrts r&rBz_NthSelector.__init__s("r( objectlistr!c 2t|dkrtd||}|js| ||j}n5#t $r(t d|jdt|wxYw|S)z| Return the nth object in the objectlist sorted by self.key and clustered if within self.tolerance. rz/Can not return the Nth element of an empty listzAttempted to access index z of a list with length )len ValueErrorclusterrreverser IndexError)r%r clusteredouts r&r'z_NthSelector.filters z??a  NOO OLL,,         DF#CC   \TV\\CPYNN\\    s  A""2Bobjct)z Return the key for ordering. Can raise a ValueError if obj can not be used to create a key, which will result in obj being dropped by the clustering method. NotImplementedErrorr%rs r&rKz_NthSelector.key3s "!r(cg}|D]@} ||}n#t$rY%wxYw|||fA|dgg}|dd}|D]T\}}t ||z |jkr|d|<||g|}U|S)zS Clusters the elements of objectlist if they are within tolerance. c|dS)Nrr=)rYs r&z&_NthSelector.cluster..Ks qtr(rJr)rKrrlsortrrt)r%r key_and_objrrKrstarts r&rz_NthSelector.cluster<s  + +C hhsmm       Sz * * * *^^,,,D Aq!#  HC3;4>11" $$S))))  #'''s  ++NTrr)r9r:r;r<intrrrBrrrr'rrKrr=r(r&rrs####T#U#### %T%[,"u""""^"(5/d4;6Gr(rc"eZdZdZdedefdZdS)RadiusNthSelectorz Select the object with the Nth radius. Applicability: All Edge and Wires. Will ignore any shape that can not be represented as a circle or an arc of a circle. rr!c|dvr'tt|St d)Nr~Wirez%Can not get a radius from this object)rrrradiusrrs r&rKzRadiusNthSelector.keybsA ==??. . .--4466 6DEE Er(Nr9r:r;r<rrrKr=r(r&rrWsHFuFFFFFFFr(rc JeZdZdZ d dedededeffd Zd e d efd Z xZ S) CenterNthSelectorz Sorts objects into a list with order determined by the distance of their center projected onto the specified direction. Applicability: All Shapes. Trrrsrrrtc\t|||||_dSr*)superrBrv)r%rsrrrt __class__s r&rBzCenterNthSelector.__init__qs, L)444r(rr!cZ||jSr*)rEdotrvrs r&rKzCenterNthSelector.key{s zz||///r(r) r9r:r;r<rrrrrBrrK __classcell__rs@r&rris"!               0u000000000r(rc4eZdZdZ ddededeffd ZxZS) DirectionMinMaxSelectoraP Selects objects closest or farthest in the specified direction. Applicability: All object types. for a vertex, its point is used. for all other kinds of objects, the center of mass of the object is used. You can use the string shortcuts >(X|Y|Z) or <(X|Y|Z) if you want to select based on a cardinal direction. For example this:: CQ(aCube).faces(DirectionMinMaxSelector((0, 0, 1), True)) Means to select the face having the center of mass farthest in the positive z direction, and is the same as:: CQ(aCube).faces(">Z") TrrrsrrtcRtd|||dS)Nr)rrsrrt)rrB)r%rsrrtrs r&rBz DirectionMinMaxSelector.__init__s9 li      r(r) r9r:r;r<rrrrBrrs@r&rrsk,MS   ,0 DI          r(rc XeZdZdZ d dedededefdZd e e d e e fd Z d S)DirectionNthSelectorz Filters for objects parallel (or normal) to the specified direction then returns the Nth one. Applicability: Linear Edges Planar Faces Trrrsrrrtcxt|||t||||dSr*)rrBr)r%rsrrrts r&rBzDirectionNthSelector.__init__s< $$T69===dA|Y?????r(rr!crt||}t||}|Sr*)rr'r)r%rs r&r'zDirectionNthSelector.filters2(//jAA !((z:: r(Nr) r9r:r;r<rrrrrBrrrr'r=r(r&rrs"! @@@ @ @  @@@@%T%[r(rc"eZdZdZdedefdZdS)LengthNthSelectorzh Select the object(s) with the Nth length Applicability: All Edge and Wire objects rr!c|dvr'tt|St dt |j)Nrz5LengthNthSelector supports only Edges and Wires, not )rrrrGrtyper9rs r&rKzLengthNthSelector.keysX ==??. . .--4466 6\S HZ\\ r(Nrr=r(r&rrs@ur(rc"eZdZdZdedefdZdS)AreaNthSelectora Selects the object(s) with Nth area Applicability: - Faces, Shells, Solids - Shape.Area() is used to compute area - closed planar Wires - a temporary face is created to compute area Will ignore non-planar or non-closed wires. Among other things can be used to select one of the nested coplanar wires or faces. For example to create a fillet on a shank:: result = ( cq.Workplane("XY") .circle(5) .extrude(2) .circle(2) .extrude(10) .faces(">Z[-2]") .wires(AreaNthSelector(0)) .fillet(2) ) Or to create a lip on a case seam:: result = ( cq.Workplane("XY") .rect(20, 20) .extrude(10) .edges("|Z or Z") .shell(2) .faces(">Z") .wires(AreaNthSelector(-1)) .toPending() .workplane() .offset2D(-1) .extrude(1) .faces(">Z[-2]") .wires(AreaNthSelector(0)) .toPending() .workplane() .cutBlind(2) ) rr!c|dvr|S|dkrp ddlm}m}t |t||S#t$r}td|dd}~wwxYwtdt|j )N)r|ShellSolidrr)r|rz"Can not compute area of the Wire: z4. AreaNthSelector supports only closed planar Wires.zCAreaNthSelector supports only Wires, Faces, Shells and Solids, not ) rAreacadquery.occ_impl.shapesr|rr makeFromWiresr Exceptionrrr9)r%rr|rexs r&rKzAreaNthSelector.keys ==??8 8 888::  ]]__ & & ????????4--d4oo>>CCEEFFF    qqqq  jVZ[^V_V_Vhjj sA B B0B++B0Nrr=r(r&rrsA//bur(rc6eZdZdZdZdeefdZdZdS)BinarySelectorz Base class for selectors that operates with two other selectors. Subclass must implement the :filterResults(): method. c"||_||_dSr*)leftright)r%rrs r&rBzBinarySelector.__init__s  r(r c||j||j|Sr*) filterResultsrr'rr$s r&r'zBinarySelector.filters?!! I  Z ( ($**;*;J*G*G   r(ctr*rr%r_leftr_rights r&rzBinarySelector.filterResultss!!r(N) r9r:r;r<rBrrr'rr=r(r&rr s[  %    """""r(rceZdZdZdZdS)r+zT Intersection selector. Returns objects that is selected by both selectors. cZtt|t|zSr*r#setrs r&rzAndSelector.filterResults$s!CKK#g,,.///r(Nr9r:r;r<rr=r(r&r+r+s-00000r(r+ceZdZdZdZdS)r1zC Union selector. Returns the sum of two selectors results. c@tt||zSr*rrs r&rzSumSelector.filterResults.sC())***r(Nrr=r(r&r1r1)s-+++++r(r1ceZdZdZdZdS)r4za Difference selector. Subtract results of a selector from another selectors results. cZtt|t|z Sr*rrs r&rzSubtractSelector.filterResults9s!CKK#g,,.///r(Nrr=r(r&r4r43s- 00000r(r4c0eZdZdZdZdeefdZdS)r7z Inverts the selection of given selector. In other words, selects all objects that is not selected by given selector. c||_dSr*)selector)r%rs r&rBzInverseSelector.__init__Cs   r(r cjtt|j|Sr*)r4rrr'r$s r&r'zInverseSelector.filterFs& DM::AA*MMMr(NrMr=r(r&r7r7=sS !!!N%NNNNNNr(r7c td}tdtdz}tt}tt ||z}t|t |t |zz}td}td}td}t||dz|z|dz|z|d z|zd }t gd } | d |dz} t t tjt tjzd} | tj } td} t ddg} t ddg}tt dttz}td}td}||dz|z}t gd}t gd}| d| d| dzz| d| dzt |zz|d | dzt |zz|d!| dzz|d"zS)#zC Define the simple string selector grammar using PyParsing .+-(),rYrZr[F)adjacent)XYZXYXZYZ simple_dir vector_dirT)caseless%><>><<[]index)|#rrfrontbackrrtopbottomonly_dirtype_opcq_typedir_opdir center_nth_opother_op named_view)r rrrrrrr valuesr set_parse_actionr upcase_tokensrsuppress)pointplusminnumberintegerfloatnlbracketrbracketcommarsrrvcqtyper direction_opr ix_number lsqbracket rsqbracketrrrs r& _makeGrammarr1Ks CLLEcllWS\\)G $ZZFhw''&011G Wx0@0@(@AAA B BFs||Hs||H CLLE 66#;;&4u*@*@&A&AADF $ $%5%C D DFcllG3*%%LD$<((Mhsmmd4jj011I&&((J&&((J 7++ +j 8E***++HKKKLLJ  * 79  y 1 1 1 3 < ! !IIe$4$4 4x F H = ) )IIe,<,< .s,aR00r()rrrr) raxes namedViewsoperatorMinMaxrrroperator parseResults_chooseSelector mySelector)r%r;s r&rBz$_SimpleStringSyntaxSelector.__init__s;1a1a1aAq//Aq//Aq//    Q1oot,Aq!__e,Aq!__e,Q1oot,1aOOT*aA.      #00)$    )..|<<r(c >d|vr$||}t|Sd|vrt|jSd|vr||}|j|j}d|vrHt |td|j |St||Sd|vr||}|j|j }d|vrHt|td|j |St|d|Sd|vr0||}|j|j|S|j|j}t|S) z< Sets up the underlying filters accordingly rrrrrrr) _getVectorrrrr9rrrjoinrasListrrrr:rr8r)r%prrwminmaxargss r&r<z+_SimpleStringSyntaxSelector._chooseSelectors   //"%%C$S)) ) "__ ++ + ^^//"%%C(3F"}}+RWWRX__%6%67788&/sF;;;  " "//"%%C()9:F"}}(c"''"(//:K:K2L2L.M.MvVVV(b&999 2  //"%%C-4=-c22 2?2=1D*D1 1r(cd|vrN|j}tt|jt|jt|jS|j|jS)zA Translate parsed vector string into a CQ Vector r)rrrrYrZr[r7r)r%rCrws r&r@z&_SimpleStringSyntaxSelector._getVectorsO 2  -C%,,ce eCEllCC C9R]+ +r(r c6|j|S)z~ selects minimum, maximum, positive or negative values relative to a direction ``[+|-|<|>|] `` r=r'r$s r&r'z"_SimpleStringSyntaxSelector.filters %%j111r(N) r9r:r;r<rBr<r@rrr'r=r(r&r3r3sl $=$=$=L%2%2%2N,,,2%222222r(r3c btd}td}tddg}td}d}||d}d}d }d } t||d tj|f|d tj|f|d tj|f|d tj| fg} | S) zx Define the complex string selector grammar using PyParsing (which supports logical operations and nesting) andorexcexceptnotc t|Sr*)r3)ress r& atom_callbackz-_makeExpressionGrammar..atom_callbacks*3///r(cr|dddd}tt|SNrr)rBrr+rPitemss r& and_callbackz,_makeExpressionGrammar..and_callback. Q!$k5)))r(cr|dddd}tt|SrS)rBrr1rTs r& or_callbackz+_makeExpressionGrammar..or_callback rWr(cr|dddd}tt|SrS)rBrr4rTs r& exc_callbackz,_makeExpressionGrammar..exc_callbacks/ Q!$&...r(c`|dd}t|S)Nrr)rBr7)rPrs r& not_callbackz,_makeExpressionGrammar..not_callbacks& Q"u%%%r(rr)r rr!rrLEFTRIGHT) atomand_opor_opdelta_opnot_oprQrVrYr[r]exprs r&_makeExpressionGrammarrfsU^^F DMMEuh'((H U^^F000 -(((*** *** /// &&&  Q l 3 Aw|[ 1 q', 5 Q | 4    D Kr(c0eZdZdZdZdeefdZdS)StringSyntaxSelectoray Filter lists objects using a simple string syntax. All of the filters available in the string syntax are also available ( usually with more functionality ) through the creation of full-fledged selector objects. see :py:class:`Selector` and its subclasses Filtering works differently depending on the type of object list being filtered. :param selectorString: A two-part selector string, [selector][axis] :return: objects that match the specified selector ***Modifiers*** are ``('|','+','-','<','>','%')`` :\|: parallel to ( same as :py:class:`ParallelDirSelector` ). Can return multiple objects. :#: perpendicular to (same as :py:class:`PerpendicularDirSelector` ) :+: positive direction (same as :py:class:`DirectionSelector` ) :-: negative direction (same as :py:class:`DirectionSelector` ) :>: maximize (same as :py:class:`DirectionMinMaxSelector` with directionMax=True) :<: minimize (same as :py:class:`DirectionMinMaxSelector` with directionMax=False ) :%: curve/surface type (same as :py:class:`TypeSelector`) ***axisStrings*** are: ``X,Y,Z,XY,YZ,XZ`` or ``(x,y,z)`` which defines an arbitrary direction It is possible to combine simple selectors together using logical operations. The following operations are supported :and: Logical AND, e.g. >X and >Y :or: Logical OR, e.g. \|X or \|Y :not: Logical NOT, e.g. not #XY :exc(ept): Set difference (equivalent to AND NOT): \|X exc >Z Finally, it is also possible to use even more complex expressions with nesting and arbitrary number of terms, e.g. (not >X[0] and #XY) or >XY[0] Selectors are a complex topic: see :ref:`selector_reference` for more information c||_t|d}|d|_dS)zl Feed the input string through the parser and construct an relevant complex selector object T) parse_allrN)selectorString_expression_grammar parse_stringrBr=)r%rk parse_results r&rBzStringSyntaxSelector.__init__\sA-*77RV7WW &--//2r(r c6|j|S)z` Filter give object list through th already constructed complex selector object rHr$s r&r'zStringSyntaxSelector.filterds%%j111r(NrMr=r(r&rhrh)sM00d3332%222222r(rh);abcrrr occ_impl.geomrocc_impl.shape_protocolsrrr r r pyparsingr r rrrrrrrr functoolsrtypingrrrrrrobjectrr?rOrqrrrrrrrrrrrrr+r1r4r7r1_grammarr3rfrlrhr=r(r&rxss######## !!!!!!                        ::::::::::::::}---%%%%%v%%%@+++++X+++8.....(...b#####h###LAAAAA/AAA.========.PPPPPPPP08:AAAAA8SAAAHFFFFF FFF$00000 000,     /   >.0A2 "AAAAAlAAAH"""""X"""&00000.000+++++.+++00000~000 N N N N Nh N N N<<<~ <>>b2b2b2b2b2(b2b2b2J111h-,X66?2?2?2?2?28?2?2?2?2?2r(