vjcddlmZmZddlZddlmZddlmZmZm 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:y)9)abstractmethodABCN)Vector) ShapeProtocolShape1DProtocol FaceProtocol geom_LUT_EDGE geom_LUT_FACE) pyparsing_commonLiteralWordnumsOptionalCombineone_ofGroupinfix_notationopAssoc)reduce)IterableListSequenceTypeVarcastShape)boundcDeZdZdZdeedeefdZdZdZ dZ dZ y ) Selectorzd Filters a list of objects. Filters must provide a single method that filters objects. objectListreturnct|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 ?/DATA/.local/lib/python3.12/site-packages/cadquery/selectors.pyfilterzSelector.filter$sJct||SN) AndSelectorr%others r&__and__zSelector.__and__04''r(ct||Sr*) SumSelectorr,s r&__add__zSelector.__add__3r/r(ct||Sr*)SubtractSelectorr,s r&__sub__zSelector.__sub__6se,,r(ct|Sr*)InverseSelector)r%s r&__neg__zSelector.__neg__9s t$$r(N) __name__ __module__ __qualname____doc__rrrr'r.r2r5r8r(r&rrs6  %  T%[  ((-%r(rc(eZdZdZdZdeefdZy)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||_yr*)pnt)r%rAs r&__init__zNearestToPointSelector.__init__Os r(r c*fd}t||gS)Ncv|jjtjjSr*)CentersubrrALength)tShaper%s r&distz+NearestToPointSelector.filter..distSs)==?&&vtxx'89@@ @r(key)min)r%r rIs` r&r'zNearestToPointSelector.filterRs AJD)**r(Nr9r:r;r<rBrrr'r=r(r&r?r?=s"+%+r(r?c*eZdZdZddZdeefdZy) 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))) cFt||_t||_||_yr*)rp0p1test_boundingbox)r%point0point1 boundingboxs r&rBzBoxSelector.__init__hs &/&/ +r(r c$ g}|jj\ |jj\  fd}|D]}|jr|j }|t |j |j|jsQ|t |j|j|js|j|||js|j||S)Nc|jk|jkz xr@|jk|jkz xr|jk|jkz Sr*)xyz)px0x1y0y1z0z1s r& isInsideBoxz'BoxSelector.filter..isInsideBoxss\##(qssRx(.ccBh1338,.ccBh1338, r()rQtoTuplerRrS BoundingBoxrxminyminzminxmaxymaxzmaxappendrE) r%r resultrcobbr]r^r_r`rarbs @@@@@@r&r'zBoxSelector.filtermsWW__& BWW__& B  A$$]]_vbggrww@Ak277BGGRWW5GMM!$qxxz*MM!$ r(N)FrMr=r(r&rOrOYs , %r(rOcNeZdZdZd dedefdZdedefdZde e de e fd Z y ) BaseDirSelectorzV A selector that handles selection on the basis of a single direction vector. vector tolerancec ||_||_yr*) directionrs)r%rrrss r&rBzBaseDirSelector.__init__s"r(vecr!cy)zMTest a specified vector. Subclasses override to provide other implementationsTr=r%rvs r&testzBaseDirSelector.testsr(r cvg}|D]}|jdk(r3|jdk(r tt|j d}nF|jdk(r2|jdk(rtt |j }n|j|s|j||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 tangentAtryrl)r%r rrn test_vectors r&r'zBaseDirSelector.filters A{{}&1::<7+B#<3<>**3/$''A+=>OOr(Nrr=r(r&rrs&PP4Pr(rc8eZdZdZdefdZdeedeefdZ y) 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") typeStringc.|j|_yr*)upperr)r%rs r&rBzTypeSelector.__init__s$**,r(r r!cvg}|D]1}|j|jk(s!|j|3|Sr*)rrrl)r%r rrns r&r'zTypeSelector.filter s5 Azz|t. r(N) r9r:r;r<strrBrrrr'r=r(r&rrs."-3-%T%[r(rc~eZdZdZd dededefdZdee de e fdZ e d e defd Z dee de e e fd Zy ) _NthSelectorzj An abstract class that provides the methods to select the Nth object/objects of an ordered list. n directionMaxrsc.||_||_||_yr*)rrrs)r%rrrss r&rBz_NthSelector.__init__s("r( objectlistr!c t|dk(r td|j|}|js|j  ||j }|S#t $r%t d|j dt|wxYw)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 NO OLL,       DFF#C   ,TVVH4KCPYNK[\  s A.Bobjct)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(cXg}|D]'} |j|}|j||f)|jdgg}|dd}|D]I\}}t ||z |j kr|dj|6|j|g|}K|S#t$rYwxYw)zS Clusters the elements of objectlist if they are within tolerance. c |dS)Nrr=)rYs r&z&_NthSelector.cluster..Ksqtr(rJr)rKrrlsortrrs)r%r key_and_objrrKrstarts r&rz_NthSelector.cluster<s C hhsm   Sz * ^,D Aq!#HC3;4>>1" $$S)  #' $   sB B)(B)NTr)r9r:r;r<intrrrBrrrr'rrKrr=r(r&rrs~###T#U# %T%[,"u"""(5/d4;6Gr(rc eZdZdZdedefdZy)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!cx|jdvrtt|jSt d)Nr}Wirez%Can not get a radius from this object)rrrradiusrrs r&rKzRadiusNthSelector.keybs3 ==?. .-446 6DE Er(Nr9r:r;r<rrrKr=r(r&rrWsFuFFr(rc JeZdZdZ d dedededeffd Zde defd 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. rrrrrsc6t||||||_yr*)superrBru)r%rrrrrs __class__s r&rBzCenterNthSelector.__init__qs L)4r(rr!cT|jj|jSr*)rEdotrurs r&rKzCenterNthSelector.key{szz|//r(r) r9r:r;r<rrrrrBrrK __classcell__rs@r&rrisN"!         0u00r(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") rrrrsc,t|d|||y)Nr)rrrrrs)rrB)r%rrrrsrs r&rBz DirectionMinMaxSelector.__init__s  li  r(r) r9r:r;r<rrrrBrrs@r&rrs1,MS  ,0 DI  r(rc JeZdZdZ d dedededefdZde e de e fd Z y ) DirectionNthSelectorz Filters for objects parallel (or normal) to the specified direction then returns the Nth one. Applicability: Linear Edges Planar Faces rrrrrscbtj|||tj||||yr*)rrBr)r%rrrrrss r&rBzDirectionNthSelector.__init__s* $$T69=dA|Y?r(rr!c^tj||}tj||}|Sr*)rr'r)r%rs r&r'zDirectionNthSelector.filters,(//jA !((z: r(Nr) r9r:r;r<rrrrrBrrrr'r=r(r&rrsZ"! @@ @ @  @%T%[r(rc eZdZdZdedefdZy)LengthNthSelectorzh Select the object(s) with the Nth length Applicability: All Edge and Wire objects rr!c|jdvrtt|jSt dt |j )Nrz5LengthNthSelector supports only Edges and Wires, not )rrrrGrtyper9rs r&rKzLengthNthSelector.keysI ==?. .-446 6GS HZHZG[\ r(Nrr=r(r&rrsur(rc eZdZdZdedefdZy)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!cd|jdvr|jS|jdk(r; ddlm}m}t |j t||jStdt|j#t$r}td|dd}~wwxYw)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 888:  ]]_ & ?4--d4o>CCEFF UVZ[^V_VhVhUij    8<>><<[]index)|#rrfrontbackrrtopbottomonly_dirtype_opcq_typedir_opdir center_nth_opother_op named_view)r rrrrrrr valuesr set_parse_actionr upcase_tokensrsuppress)pointplusminnumberintegerfloatnlbracketrbracketcommarrrrucqtyper direction_opr ix_number lsqbracket rsqbracketrrrs r& _makeGrammarr1KsE CLEclWS\)G $ZFhw'&01G Wx0@(@AA BFs|Hs|H CLE 6#;&4u*>*@&AADF $ $%5%C%C DFclG3*%LD$<(Mhsmd4j01I&&(J&&(J 7+ +j 8E*+HKLJ * 9 y 1 1 3  !Ie$4 4x F H  )Ie,< .s ,aR0r()rrrr) raxes namedViewsoperatorMinMaxrrroperator parseResults_chooseSelector mySelector)r%r;s r&rBz$_SimpleStringSyntaxSelector.__init__s1a1a1aAq/Aq/Aq/   Q1ot,Aq!_e,Aq!_e,Q1ot,1aOT*aA.    #0)$   )..|<r(c  d|vr|j|}t|Sd|vrt|jSd|vrw|j|}|j|j }d|vr=t |tdj|jj|St||Sd|vrx|j|}|j|j}d|vr=t|tdj|jj|St|d|Sd|vr0|j|}|j|j|S|j |j"}t|S) z< Sets up the underlying filters accordingly rrrrrrr) _getVectorrrrr9rrrjoinrasListrrrr:rr8r)r%prrvminmaxargss r&r<z+_SimpleStringSyntaxSelector._chooseSelectors`  //"%C$S) ) "_ + + ^//"%C((3F"}+RWWRXX__%678&/sF;;  "//"%C(()9)9:F"}(c"''"((//:K2L.MvVV(b&99 2 //"%C-4==-c2 2??2==1D*D1 1r(cd|vrR|j}tt|jt|jt|j S|j |jS)zA Translate parsed vector string into a CQ Vector r)rrrrYrZr[r7r)r%rCrvs r&r@z&_SimpleStringSyntaxSelector._getVectorsO 2 --C%,cee eCEElC C99R]]+ +r(r c8|jj|S)z~ selects minimum, maximum, positive or negative values relative to a direction ``[+|-|<|>|] `` r=r'r$s r&r'z"_SimpleStringSyntaxSelector.filters %%j11r(N) r9r:r;r<rBr<r@rrr'r=r(r&r3r3s( $=L%2N,2%2r(r3c Rtd}td}tddg}td}d}|j|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) andorexcexceptnotct|Sr*)r3)ress r& atom_callbackz-_makeExpressionGrammar..atom_callbacks *3//r(cT|jdddd}tt|SNrr)rBrr+rPitemss r& and_callbackz,_makeExpressionGrammar..and_callback' Q!$k5))r(cT|jdddd}tt|SrS)rBrr1rTs r& or_callbackz+_makeExpressionGrammar..or_callback rWr(cT|jdddd}tt|SrS)rBrr4rTs r& exc_callbackz,_makeExpressionGrammar..exc_callbacks( Q!$&..r(cD|jdd}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 DMEuh'(H U^F0 -(* * / &  Q l 3 Aw||[ 1 q',, 5 Q | 4   D Kr(c(eZdZdZdZdeefdZy)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 cp||_tj|d}|jd|_y)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__\s7-*77RV7W &--/2r(r c8|jj|S)z` Filter give object list through th already constructed complex selector object rHr$s r&r'zStringSyntaxSelector.filterds%%j11r(NrMr=r(r&rhrh)s0d32%2r(rh);abcrrr occ_impl.geomrocc_impl.shape_protocolsrrr r r pyparsingr r rrrrrrrr functoolsrtypingrrrrrrobjectrr?rOrqrrrrrrrrrrrrr+r1r4r7r1_grammarr3rfrlrhr=r(r&rxs# !   ::}-%v%@+X+8.(.b#h#LA/A.==.PP08:A8SAHF F$0 0, / >.0A2 "AlAH"X"&0.0+.+0~0 Nh N<~ >b2(b2J1h-X6?28?2r(