U |e@sdZddlZddlZddlZddlZddlmZddlmZddl m Z m Z m Z ddl mZmZmZmZmZmZmZmZmZddlmZddlmZdd lmZmZdd lmZm Z m!Z!d d Z"d dZ#dZ$dZ%dZ&dZ'dZ(dZ)ddZ*e!ddZ+ddZ,ddZ-ddZ.dd Z/e0d!d"Z1Gd#d$d$e2Z3Gd%d&d&e2Z4Gd'd(d(e2Z5d)d*Z6ee7d+d,Z8ee7e j9d-d.Z:ee;e j<d/d0Z=eej>e jd?d@ZGe!dAdBZHe e j.impl)r1r2r3)r<r;rGr=r>r%rFr&_get_hash_value_intrinsicJs  rHcCst|jt}|d||S)z8 Whether the hash value denotes an empty entry. ==)rr6r7EMPTY icmp_unsigned)rrr@emptyr%r%r& is_hash_emptySsrMcCst|jt}|d||S)z9 Whether the hash value denotes a deleted entry. rIrr6r7DELETEDrKrrr@deletedr%r%r&is_hash_deletedZsrRcCst|jt}|d||S)z9 Whether the hash value denotes an active entry. ksz!check_all_set..z All arguments must be Sets, got csg|]}dj|jkqS)rdtype)rVsrEr%r&rWnsz'All Sets must be of the same type, got )allrr[r%r[r& check_all_setjsr]SetLoopindexentrydo_breakc@seZdZddZeddZejddZeddZejddZed d Zejd d Zed d Z e jdd Z eddZ e jddZ eddZ eddZ ddZ d ddZejd!ddZejddZdS)" _SetPayloadcCs<t||||}||_||_||_||_|d|_||_dS)Nentries)r'_context_builder_ty_payload_get_ptr_by_name_entries_ptr)selfrrr r!r$r%r%r&__init__ws z_SetPayload.__init__cCs|jjSNrhmaskrlr%r%r&rpsz_SetPayload.maskcCs ||j_dSrnrorlr;r%r%r&rpscCs|jjSrnrhusedrqr%r%r&rtsz_SetPayload.usedcCs ||j_dSrnrsrrr%r%r&rtscCs|jjSrnrhfillrqr%r%r&rvsz_SetPayload.fillcCs ||j_dSrnrurrr%r%r&rvscCs|jjSrnrhfingerrqr%r%r&rxsz_SetPayload.fingercCs ||j_dSrnrwrrr%r%r&rxscCs|jjSrnrhdirtyrqr%r%r&rzsz_SetPayload.dirtycCs ||j_dSrnryrrr%r%r&rzscCs|jS)z> A pointer to the start of the entries array. )rjrqr%r%r&rdsz_SetPayload.entriescCs|jS)zC A pointer to the start of the NRT-allocated area. )rkrqr%r%r&r!sz_SetPayload.ptrcCs2t|j|j|}|jj|jt|j|d}|S)z) Get entry number *idx*. r) rgeprfrjrerrr(rg)rlidxZ entry_ptrrar%r%r& get_entrys   z_SetPayload.get_entryFc s j jj} j} jj}j}|tj }| |||fi}  || t |d} t |d} t} t|} r|dtd}ddd} f dd }tt |t<| }|||| }||}|| W5QRX||x| }||| }|| }| || }|||}|| || |W5QRXNr| }}d |||}|| |W5QRX|W5QRX|t dd }| tj!| tj"|| fS) ag Lookup the *item* with the given hash values in the entries. Return a (found, entry index) tuple: - If found is true, points to the entry containing the item. - If found is false, points to the empty entry that the item can be written to (only if *for_insert* is true) r,z lookup.bodyz lookup.foundzlookup.not_foundz lookup.endc s |}|j}d|6 |jf}|W5QRXW5QRXt|W5QRXr؈t|4} d|||} |W5QRXdS)zO Check entry *i* against the value being searched for. rIN) r}r2if_thenrKkeybranchrMrRloadr9store)iraZ entry_hasheqj Zbb_foundZ bb_not_foundrrZeqfn for_insertZ free_indexZfree_index_sentinelr@itemrlr%r& check_entrys    z(_SetPayload._lookup..check_entryrIfound)#rerfr7rprgrYr0r1operatorrr3r4rr6ralloca_once_valueand_append_basic_block for_range LINEAR_PROBESraddrr goto_blocklshrmulr9rKposition_at_endphiIntType add_incomingtrue_bit false_bit)rlrr@rintp_trprYtyctxr=r>oneZfiveZperturbr`bb_bodybb_endrrprrr%rr&_lookupsr                              z_SetPayload._lookupNc cs|j}|j}|tj}t|d}||j|}t j |||dN}| |j }t |||j} || t|j ||jd} | VW5QRXW5QRXdS)zG Iterate over the payload's entries. Yield a SetLoop. r~startr_N)rerfget_value_typerintprr6rrprrr}r`r5r2rr^rb) rlrrrrrsizeZ range_looprais_usedloopr%r%r&_iterate7s    z_SetPayload._iteratec cs|j}|j}|tj}t|d}t|d}|j}|d}|d}t ||j } | || |V|| } |||| |} || | || } t||| j} || ||W5QRX|||| } | |_ || VdS)z Yield a random entry from the payload. Caller must ensure the set isn't empty, otherwise the function won't end. rr~Znext_entry_bodyZnext_entry_endN)rerfrrrrr6rprrrrxrrrrrrr}r5r2cbranchr) rlrrrzerorrprrr`rrarr%r%r& _next_entryKs*            z_SetPayload._next_entry)F)N)__name__ __module__ __qualname__rmpropertyrpsetterrtrvrxrzrdr!r}r contextlibcontextmanagerrrr%r%r%r&rcus>              v rcc@seZdZddZeddZeddZeddZed d Zed d Z e j d d Z ddZ ddZ dUddZ dVddZdWddZdXddZdYddZdZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd[d2d3Zd4d5Zd6d7Zed\d9d:Z ed]d;d<Z!ed=d>Z"ed?d@Z#dAdBZ$dCdDZ%dEdFZ&dGdHZ'd^dIdJZ(dKdLZ)dMdNZ*dOdPZ+dQdRZ,dSdTZ-d8S)_ SetInstancecCs2||_||_||_t|||_|||||_dSrn)rerfrgr+ _entrysize make_helper_set)rlrrr set_valr%r%r&rmvs  zSetInstance.__init__cCs|jjSrn)rgrYrqr%r%r&rY}szSetInstance.dtypecCs.|j}|j}|jj||j}t|||j|S)z/ The _SetPayload for this set. )rerfnrt meminfo_datameminforcrg)rlrrr!r%r%r&r$szSetInstance.payloadcCs |jSrn)r _getvaluerqr%r%r&r;szSetInstance.valuecCs|jjSrn)rrrqr%r%r&rszSetInstance.meminfocCs|jjSrnrparentrqr%r%r&rszSetInstance.parentcCs ||j_dSrnrrrr%r%r&rscCs|jjS)z< Return the number of elements in the size. )r$rtrqr%r%r&get_sizeszSetInstance.get_sizecCs |jjr|rtjntj|j_dSrn)rg reflectedrrrr$rzrlvalr%r%r& set_dirtyszSetInstance.set_dirtyTc Cs|j}|j}|j}||_||||_|j} t| jd} | | | } |_|j t |||dd| |j | |_ W5QRX|r| | |ddS)Nr~Tlikely)rerfr2 incref_valuerrtrr6r7rrrMrvupsizer) rlr$rarr@ do_resizerrold_hashrtrr%r%r& _add_entrys   zSetInstance._add_entryc Cs|j}|j}|j||dd\}} ||} || || } | j} || _|rZ|||| _|j } t | j d}| | |} |_ |jt||| dd| |j||_W5QRX|r|| |dW5QRXdS)NTrr~r)rerfrnot_rr}r2rrrtrr6r7rrMrvrr)rlr$rr@r do_increfrrrr not_foundrarrtrr%r%r&_add_keys*     zSetInstance._add_keycCsft|jjt|_|r"||j|j}t|jd}|j ||}|_|rX| || ddS)Nr~T) rr6r2r7rO decref_valuerrtrfsubdownsizer)rlr$rar do_decrefrtrr%r%r& _remove_entrys  zSetInstance._remove_entryc CsN|j}|j}|||\}}||||} ||| |W5QRX|Srn)rerfrrr}r) rlr$rr@rrrrrrar%r%r& _remove_keys  zSetInstance._remove_keycCs8|j}|j}|j}t|||jj|}|||||dSrn)rerfr$rCrgrYr)rlrrrrr$r@r%r%r&rs zSetInstance.addcCs6|j}|j}|j}|||||}|||||dS)z`A version of .add for use inside functions following Python calling convention. N)rerfr$_pyapi_get_hash_valuer)rlpyapirrrrr$r@r%r%r& add_pyapis zSetInstance.add_pyapic Csl|jjg}tj}dd}|g}tj|f|} ||| |\} } |j| dd|| W5QRX| S)z=Python API compatible version of `get_hash_value()`. cSst|Srn)rH)rr%r%r&wrapper sz2SetInstance._pyapi_get_hash_value..wrapperFr) rgrYrrr signature call_jit_coderretget_null_object) rlrrrrargtypesrestyrrEr>is_errorretvalr%r%r&rs z!SetInstance._pyapi_get_hash_valuecCs8|j}|j}|j}t|||jj|}|||\}}|Srn)rerfr$rCrgrYr)rlrrrr$r@rrr%r%r&containss zSetInstance.containscCs6|j}|j}|j}t|||jj|}||||}|Srn)rerfr$rCrgrYr)rlrrrr$r@rr%r%r&discard s zSetInstance.discardc Csh|j}|j}||jj}t||}|j}|$}| |j ||j ||ddW5QRX| |S)NF)r) rerfrrgrYr alloca_oncer$rrrrr)rlrrltyrr$rar%r%r&pop)s  zSetInstance.popcCs<|j}|j}|tj}t|t}||| ddS)NT) rerfrrrrr6MINSIZE_replace_payloadr)rlrrrminsizer%r%r&clear8s    zSetInstance.clearc Cs:|j}|j}|j}|j}|j}t||||jd}|d||}|j|dd\}} |@| |} |j | | dd|j |tdW5QRXW5QRX| ||||} || } |j | | dd|j |tdW5QRX|j} |"} | j}|j| |j|jddW5QRXW5QRXW5QRX|S)z, Return a copy of this set. NrITrF)zcannot copy setr)rerfr$rtrvr7rgrKif_else _copy_payloadrr call_convreturn_user_exc MemoryErrorchoose_alloc_size_allocate_payloadrrarrr2)rlrrr$rtrvotherZno_deleted_entriesZ if_no_deletedZ if_deletedoknentries other_payloadrrar%r%r&copyAs:     $zSetInstance.copyc Cs|j}|j}|j}|j}|L}|j}||j|j\}} || ||j ||ddW5QRXW5QRX| |j dS)z9 In-place intersection with *other* set. FrN) rerfr$rrarrr2rrrrrt) rlrrrr$rrrar_r%r%r& intersecths $zSetInstance.intersectc CsX|j}|j}|j}|j}|"}|j}|j||j|jddW5QRX||j dS)z7 In-place difference with *other* set. FrN) rerfr$rrarrr2rrt)rlrrrr$rrrar%r%r& differencezs  zSetInstance.differencec Cs|j}|j}|j}|}|jj}|jj}|j}|j||dd\} } || } | | J\} } | |j || ddW5QRX| | || ||W5QRXW5QRXW5QRX| |jj dS)zA In-place symmetric difference with *other* set. TrFrN)rerfr$rrarr2rr}rrrrrt)rlrrrrrrr@r$rrraZ if_commonZ if_not_commonr%r%r&symmetric_differences  .z SetInstance.symmetric_differenceFc Cs|j}|j}|j}|j}|r dnd}t|tj}||||j|j\} } | | tj |W5QRX| f| R} | j } | | j| j\} }||| | tj || W5QRXW5QRXW5QRXW5QRX||S)NrS<=)rerfr$rrrrrKrtrrrrarrr2rrrbr)rlrstrictrrr$rcmp_opresZ if_smaller if_largerrrarrr%r%r&issubsets(  0zSetInstance.issubsetc s|j}|j|j}|j}ttjfdd}d|j|j>\}}||||W5QRX||||W5QRXW5QRX S)Nc s^|L}|j}||j|j\}}|tj| W5QRXW5QRXdSrn) rrarrr2rrrrrb)ZsmallerZlargerrrarrrrr%r&checks   z%SetInstance.isdisjoint..check>) rerfr$rrrrrKrtr)rlrrr$rrr otherwiser%rr& isdisjoints zSetInstance.isdisjointc Cs|j}|j}|j}|j}t|tj}||d|j|j\}}|f| R} | j } | | j | j \} } ||| |tj|| W5QRXW5QRXW5QRX||tj|W5QRXW5QRX||S)NrI)rerfr$rrrrrKrtrrarrr2rrrrrbr) rlrrrr$rrZ if_same_sizerrrarrr%r%r&equalss& &"zSetInstance.equalsNc Csf|tj}|dkr"t|t}n$t|tr8t||}||||}||||d}| |}||fS)z Allocate a SetInstance with its storage. Return a (ok, instance) tuple where *ok* is a LLVM boolean and *instance* is a SetInstance object (the object's contents are only valid when *ok* is true). N) rrrrr6rrTintrr) clsrrr nitemsrrrlrr%r%r& allocate_exs    zSetInstance.allocate_exc CsH|||||\}}|j||dd|j|tdW5QRX|S)z Allocate a SetInstance with its storage. Same as allocate_ex(), but return an initialized *instance*. If allocation failed, control is transferred to the caller using the target's current call convention. Fr)zcannot allocate set)r rrrrr)r rrr r rrlr%r%r&allocates  zSetInstance.allocatecCs<||||d}||j_|tj|j_|j|||j|S)z Allocate a new set instance pointing to an existing payload (a meminfo pointer). Note the parent field has to be filled by the caller. N) rrget_constant_nullrpyobjectrrincrefr;)r rrr rrlr%r%r& from_meminfos zSetInstance.from_meminfoc Cs|j}t|d}t|t}|||}t||}|d} |d} || | | d| |} | d| |} |j | dd|| W5QRX|| |} | | ||| W5QRX|| | |S)zT Choose a suitable number of entries for the given number of items. r~ calcsize.body calcsize.end>=Fr)r7rr6rshlrrrrrrrKrrr)r rrr rrr min_entriesZsize_prrrZis_large_enoughZ next_sizer%r%r&r$s$            zSetInstance.choose_alloc_sizec Cs0|j}|j}|j}t|d}t|d}|j}|||}||j|} | d|| } |j | ddt || } | d} | d} || || D|| }|||}||| | d||}||| | W5QRX|| || }tr||d|| |||||d W5QRXd S) zs When adding to the set, ensure it is properly sized for the given number of used entries. r~rFrrrzKupsize to %zd items: current size = %zd, min entries = %zd, new size = %zd zcannot grow setN)rerfr7rr6r$rrrprKrrrrrrrrrr DEBUG_ALLOCSprintf_resize)rlr rrrrtwor$rr need_resize new_size_prrnew_size is_too_smallr%r%r&rCs>             zSetInstance.upsizec Cs|j}|j}|j}t|d}t|d}t|t}|j}|||} || d| || |} || |} | |j |} | | d| | | d|| } |j | ddt|| } |d}|d }||||`|| }|||}| d | |}| |||W5QRX||| ||W5QRX|||| }trr||d || | ||||d W5QRXd S)zw When removing from the set, ensure it is properly sized for the given number of used entries. r~rrrrSFrrrrzMdownsize to %zd items: current size = %zd, min entries = %zd, new size = %zd zcannot shrink setN)rerfr7rr6rr$rr9rKrrprrrrrrrrrrrrrr)rlr rrrrrrr$rmax_sizerrrrrrr r%r%r&rqsT                  zSetInstance.downsizec Cs|j}|j}|}|j|dd}|j||dd|j|t|fW5QRX|j}| $}|j } |j || j | j dddW5QRX||jdS)zw Resize the payload to the given number of entries. CAUTION: *nentries* must be a power of 2! TreallocFr)rrN)rerfrrrrrrr$rrarrr2 _free_payloadr!) rlr$rerrmsgrrZ old_payloadrrrar%r%r&rs   zSetInstance._resizec Cs|j}|j}|j}|j}||jW5QRX||jj|j |dd}|j | |dd|j |tdW5QRXdS)z Replace the payload with a new empty payload with the given number of entries. CAUTION: *nentries* must be a power of 2! Tr"Fr)zcannot reallocate setN)rerfr$rrarrr$r!rrrrrr)rlrrrrrarr%r%r&rs  zSetInstance._replace_payloadc Cs|j}|j}t|tj}|tj}t |d}t |d}| t |j } | | } |j} | | 8} t||t || t || \} } |j| dd|tj|W5QRX|j||dd&|r|jj}|jj||| d}t||}n4|||j}|j|| ||tj}t||}|j|dd\}}||tj|W5QRX|z|s||j_|tj |j_!|j"}t#||j$| d||_%||_&||_'|(||}||_)t*r|+|d| |j$|W5QRXW5QRXW5QRX||S) z Allocate and initialize payload for the given number of entries. If *realloc* is True, the existing meminfo is reused. CAUTION: *nentries* must be a power of 2! rr~FrT)r.allocated %zd bytes for set at %p: mask = %zd ),rerfrrrrrrrr6rrrgr)rmuladd_with_overflowrrrrrrrZmeminfo_varsize_alloc_uncheckedis_null _imp_dtormodule"meminfo_new_varsize_dtor_uncheckedr voidptr_trrrrr$memsetr!rtrvrxrrprr)rlrr#rrrrrrr" payload_size entry_size allocsizeovfrr!alloc_okdtorif_errorif_okr$new_maskr%r%r&rsp         "zSetInstance._allocate_payloadcCs|jj|j|j|dS)z9 Free an allocated old payload at *ptr*. N)rerZmeminfo_varsize_freerfr)rlr!r%r%r&r$"szSetInstance._free_payloadc Cs|j}|j}t|tj}|tj}t |d}t |d}| t |j }| |} |j} | | 8} |j} ||| } |t || |t || | } |j||dd|||j}|j|| ||tj}t||}|j|dd\}}||tj|W5QRX|||j_|j }|j!|_!|j"|_"||_#| |_t$||j%|j%| | |&}|'|j(j)W5QRXt*r|+|d| |j,| W5QRXW5QRXW5QRX||S)z7 Raw-copy the given payload into self. rr~TrFr')-rerfrrrrrrrr6rrrgr)rrprrrrr*r+rr,rr-r)rrrrrr$rtrvrx raw_memcpyrdrrrarrrr!)rlZ src_payloadrrrrrrr"r/r0rprr1r4rr3r5r6r$rr%r%r&r(sd         "zSetInstance._copy_payloadc Cstj}|tj}tt|||g}d|jj }tj |||d}|j rd|_ t |}t|||j|jd} | "} | j} |j||jj | jW5QRX||S)z Define the dtor for set z .dtor.set.)name linkonce_odrr)rr-rrsize_tr FunctionTypeVoidTypergrYget_or_insert_functionis_declarationlinkage IRBuilderrrcrErrardecrefrret_void) rlrr+ llvoidptrZllsize_tr=fnamer?rr$rrar%r%r&r*is"   zSetInstance._imp_dtorcCs|jj|j|jj|dS)z Incref an element value N)rerrrfrgrYrr%r%r&rszSetInstance.incref_valuecCs|jj|j|jj|dS)z Decref an element value N)rerrBrfrgrYrr%r%r&rszSetInstance.decref_value)T)TT)TT)T)T)T)F)N)N)F).rrrrmrrYr$r;rrrrrrrrrrrrrrrrrrrrrrr  classmethodr rrrrrrrrr$rr*rrr%r%r%r&rtsf              '     .< CArc@sZeZdZddZeddZeddZeddZed d Z e j d d Z d d Z dS)SetIterInstancecCsL||_||_||_|||||_|jj||j}t|||jj ||_ dSrn) rerfrgr_iterrrrrc containerrh)rlrr iter_typeiter_valr!r%r%r&rms zSetIterInstance.__init__cCsJt|||j|}||||d}|tjd}t|||j_|j |j_ |SrD) rrI get_constantrrrrrHr`r)r rrrJrZset_instrlr`r%r%r&from_sets  zSetIterInstance.from_setcCs |jSrn)rHrrqr%r%r&r;szSetIterInstance.valuecCs|jjSrn)rHrrqr%r%r&rszSetIterInstance.meminfocCs|j|jjSrn)rfrrHr`rqr%r%r&r`szSetIterInstance.indexcCs|j||jjdSrn)rfrrHr`rrr%r%r&r`sc Csr|j}|j}t|jd}||j|d:}|j}|| |j |j |j||_| W5QRXdS)Nr~r)r`rhrr6r7 set_exhaustedrra set_validyield_rrfrrb)rlresultr`r$rrrar%r%r&iternexts zSetIterInstance.iternextN) rrrrmrFrMrr;rr`rrRr%r%r%r&rGs     rGc Cst|}t||||}|dkrt||}t||}|tj|}t ||(} | t ||d| j } | | W5QRXt||||jS)zD Build a set of the given type, containing the given items. r)lenrrr pack_arrayrrLrrrrr{r`rr r;) rrr itemsr instarray array_ptrcountrrr%r%r& build_sets  rZcCs$|j}t|||}t||||jSrn) return_typerrr r;)rrr>rEr rVr%r%r&set_empty_constructorsr\c Cs||j}|j\}|\}t||||}t||||}t||||&} || j|j ||j | jW5QRXt ||||jSrn) r[rErrrr rr;rrBrYr ) rrr>rEr items_typerUnrVrr%r%r&set_constructors r_cCs t|||jd|d}|SrD)rrErrrr>rErVr%r%r&set_lensracCs&t|||jd|d}||dSNrr~)rrErr`r%r%r&in_setsrcgetitercCs(t|||j|d}t|||j|jSrD)rGrMr[r r;r`r%r%r& getiter_setsrerRcCs&t|||jd|d}||dSrD)rGrErR)rrr>rErQrVr%r%r&iternext_listiter srfzset.addcCs2t|||jd|d}|d}|||Srb)rrErget_dummy_valuerrr>rErVrr%r%r&set_adds ricCst||}dd}||fS)NcSs2t|||jd|d}|d}|||Srb)rrErrgrhr%r%r& set_discard"s z!_set_discard..set_discardrnone)r<rZrr>rjr%r%r& _set_discards rmrcCsddS)NcSs t||Srn)rmrZrr%r%r&.z ol_set_discard..r%rnr%r%r&ol_set_discard,srqcCs||}dd}||fS)Nc SsZt|||jd|d}|jj}|jt||dd|j|t dW5QRX| S)NrFr)zset.pop(): empty set) rrEr$rtrrr)rrKeyErrorr)rrr>rErVrtr%r%r&set_pop5s z_set_pop..set_poprX)r<rZr>rsr%r%r&_set_pop1s  rtrcCsddS)NcSst|Srn)rtrZr%r%r&roCrpzol_set_pop..r%rur%r%r& ol_set_popAsrvcCst||}dd}||fS)Nc Ssbt|||jd|d}|d}||}|j||dd|j|tdW5QRX|S)Nrr~Fr)zset.remove(): key not in set) rrErrrrrrrrg)rrr>rErVrrr%r%r& set_removeJs  z_set_remove..set_removerk)r<rZrr>rwr%r%r& _set_removeFs  rxremovecCs|j|krddSdS)NcSs t||Srn)rxrnr%r%r&roZrpzol_set_remove..rXrnr%r%r& ol_set_removeWs rzcCst|}dd}||fS)NcSs(t|||jd|d}||SrD)rrErrgr`r%r%r& set_clearcsz_set_clear..set_clearrk)r<rZr>r{r%r%r& _set_clear_s r|rcCsddS)NcSst|Srn)r|rur%r%r&romrpzol_set_clear..r%rur%r%r& ol_set_clearksr}cCs||}dd}||fS)NcSs2t|||jd|d}|}t|||j|jSrD)rrErr r[r;rrr>rErVrr%r%r&set_copytsz_set_copy..set_copyr%)r<rZr>rr%r%r& _set_copypsrrcCsddS)NcSst|Srn)rrur%r%r&ro~rpzol_set_copy..r%rur%r%r& ol_set_copy|srcCsBt|||jd|d}t|||jd|d}|||Srb)rrErrgr~r%r%r&set_difference_updates rcCst||}|tfSrn)rrlrr<abr>r%r%r&_set_difference_updates rdifference_updatecCst||ddS)NcSs t||Srn)rrrr%r%r&rorpz,set_difference_update_impl..r]rr%r%r&set_difference_update_impls rcCsBt|||jd|d}t|||jd|d}|||Srb)rrErrgr~r%r%r&set_intersection_updates rcCst||}|tfSrn)rrlrrr%r%r&_set_intersection_updates rintersection_updatecCst||ddS)NcSs t||Srn)rrr%r%r&rorpz.set_intersection_update_impl..rrr%r%r&set_intersection_update_impls rcCsBt|||jd|d}t|||jd|d}|||Srb)rrErrgr~r%r%r&set_symmetric_difference_updates rcCst||}|tfSrn)rrlrrr%r%r& _set_symmetric_difference_updates rsymmetric_difference_updatecCst||ddS)NcSs t||Srn)rrr%r%r&rorpz6set_symmetric_difference_update_impl..rrr%r%r&$set_symmetric_difference_update_impls rz set.updatec Cst|||jd|d}|jd}|d}t||||}|dk rZ||jj|}||t||||:} ||| j |j |j } || |j ||j | j W5QRX|dk r| |jj|Srb)rrErrr$rtrr castr;rYrrBrrg) rrr>rErVr]rUr^rrcastedr%r%r& set_updates   rcs(tfddt|fdd}dS)Ncs|||}fdd}||fS)Ncs:|j|jdkst||||t|||jd|dSrD)r[rEAssertionErrorr )rrr>rErGr%r&codegensz>gen_operator_impl.._set_operator_intr..codegenr%r<rrr>rrr%r&_set_operator_intrs  z-gen_operator_impl.._set_operator_intrcst||fddS)Ncs ||Srnr%rrr%r&rorpz=gen_operator_impl.._ol_set_operator..rrrr%r&_ol_set_operators z+gen_operator_impl.._ol_set_operator)rr)oprGrr%)rrGr&gen_operator_implsrrcCst||dd}|S)NcSs|}|||Srn)rrrrrZr%r%r&difference_impls z,impl_set_difference..difference_implr)rrrr%r%r&impl_set_differences r intersectioncCst||dd}|S)NcSs@t|t|kr&|}|||S|}|||SdSrn)rSrrrr%r%r&intersection_impls  z+set_intersection..intersection_implr)rrrr%r%r&set_intersections  rrcCst||dd}|S)NcSs@t|t|kr&|}|||S|}|||SdSrn)rSrrrr%r%r&symmetric_difference_impls  z;set_symmetric_difference..symmetric_difference_implr)rrrr%r%r&set_symmetric_differences  runioncCst||dd}|S)NcSs@t|t|kr&|}|||S|}|||SdSrn)rSrupdaterr%r%r& union_impl(s  zset_union..union_implr)rrrr%r%r& set_union#s  rcCst||}dd}||fS)NcSs:t|||jd|d}t|||jd|d}||Srb)rrErr~r%r%r&r;sz _set_isdisjoint..codegenrbooleanrr%r%r&_set_isdisjoint7s rrcCst||ddS)NcSs t||Srn)rrr%r%r&roHrpz set_isdisjoint..rrr%r%r&set_isdisjointDs rcCst||}dd}||fS)NcSs:t|||jd|d}t|||jd|d}||SrbrrErr~r%r%r&rOsz_set_issubset..codegenrrr%r%r& _set_issubsetKs rrcCst||ddS)NcSs t||Srn)rrr%r%r&ro\rpzset_issubset..rrr%r%r& set_issubsetWs r issupersetcCst||dd}|S)NcSs ||Srn)rrr%r%r& superset_impldsz%set_issuperset..superset_implr)rrrr%r%r&set_issuperset_s rcCst||}dd}||fS)NcSs:t|||jd|d}t|||jd|d}||Srb)rrEr r~r%r%r&rmsz_set_eq..codegenrrr%r%r&_set_eqis rcCst||ddS)NcSs t||Srn)rrr%r%r&royrpzset_eq..rrr%r%r&set_equs rcCst||dd}|S)NcSs ||k Srnr%rr%r%r&ne_implszset_ne..ne_implr)rrrr%r%r&set_ne{s rcCst||}dd}||fS)NcSs>t|||jd|d}t|||jd|d}|j|ddS)Nrr~T)rrr~r%r%r&rsz_set_lt..codegenrrr%r%r&_set_lts rcCst||ddS)NcSs t||Srn)rrr%r%r&rorpzset_lt..rrr%r%r&set_lts rcCst||dd}|S)NcSs||kSrnr%rr%r%r&gt_implszset_gt..gt_implr)rrrr%r%r&set_gts rcCs^t|||jd|d}t|||jd|d}||jtj}||jtj}|d||S)Nrr~rI)rrEptrtointrrr icmp_signed)rrr>rErrmambr%r%r&set_iss rcCs|j|jkst|Srn)rYr)rrfromtytotyrr%r%r& set_to_setsr)}__doc__ collectionsrmathr functoolsrllvmliter numba.corerrrnumba.core.imputilsrrr r r r r rr numba.miscr numba.cpythonrnumba.core.errorsrrnumba.core.extendingrrrr'r+rJrOr8rrrrCrHrMrRr5r] namedtupler^objectrcrrGrZsetr\ IterableTyper_rSrUrarAnyrcreSetIterBORROWEDrfrirmrqrtrvrxrzr|r}rrrrrrrrrrrrriandiorisubixorZop_Zop_implrrrrxorror_rrrrlergerrrrnerrltrgtris_rrr%r%r%r&s  ,       "6