U hd@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)r.r/r0)r9r8rCr:r;r#rBr$_get_hash_value_intrinsicJs  rDcCst|jt}|d||S)z8 Whether the hash value denotes an empty entry. ==)rr3r4EMPTY icmp_unsigned)rrr=emptyr#r#r$ is_hash_emptySsrIcCst|jt}|d||S)z9 Whether the hash value denotes a deleted entry. rErr3r4DELETEDrGrrr=Zdeletedr#r#r$is_hash_deletedZsrMcCst|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)rQsrAr#r$rRnsz'All Sets must be of the same type, got )allrrVr#rVr$ check_all_setjsrXSetLoopindexentrydo_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_payloadZ_get_ptr_by_name_entries_ptr)selfrrrr r"r#r#r$__init__ws z_SetPayload.__init__cCs|jjSNrcmaskrfr#r#r$rjsz_SetPayload.maskcCs ||j_dSrhrirfr8r#r#r$rjscCs|jjSrhrcusedrkr#r#r$rnsz_SetPayload.usedcCs ||j_dSrhrmrlr#r#r$rnscCs|jjSrhrcfillrkr#r#r$rpsz_SetPayload.fillcCs ||j_dSrhrorlr#r#r$rpscCs|jjSrhrcfingerrkr#r#r$rrsz_SetPayload.fingercCs ||j_dSrhrqrlr#r#r$rrscCs|jjSrhrcdirtyrkr#r#r$rtsz_SetPayload.dirtycCs ||j_dSrhrsrlr#r#r$rtscCs|jS)z> A pointer to the start of the entries array. )rdrkr#r#r$r_sz_SetPayload.entriescCs|jS)zC A pointer to the start of the NRT-allocated area. )rerkr#r#r$r sz_SetPayload.ptrcCs2t|j|j|}|jj|jt|j|d}|S)z) Get entry number *idx*. r) rgeprardr`rrr&rb)rfidxZ entry_ptrr\r#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. rEN) rwr/if_thenrGkeybranchrIrMloadr6store)ir\Z entry_hasheqj Zbb_foundZ bb_not_foundrrZeqfn for_insertZ free_indexZfree_index_sentinelr=itemrfr#r$ check_entrys    z(_SetPayload._lookup..check_entryrEfound)#r`rar4rjrbrTr-r.operatorrr0r1rr3ralloca_once_valueand_append_basic_block for_range LINEAR_PROBESr}addr~r| goto_blocklshrmulr6rGposition_at_endphiZIntTypeZ add_incomingtrue_bit false_bit)rfrr=rintp_trjrTZtyctxr:r;onefiveZperturbr[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. rxstartrZN)r`raget_value_typerintprr3rrjrrrwr[r2r/rzrYr]) rfrrrrrsizeZ range_loopr\is_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. rrxZnext_entry_bodyZnext_entry_endN)r`rarrrrr3rjrrrrrr|rr}rrr~rwr2r/cbranchr) rfrrrzerorrjrrr[rr\rr#r#r$ _next_entryKs*            z_SetPayload._next_entry)F)N)__name__ __module__ __qualname__rgpropertyrjsetterrnrprrrtr_r rwr contextlibcontextmanagerrrr#r#r#r$r^us>              v r^c@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|||_|||||_dSrh)r`rarbr( _entrysize make_helper_set)rfrrrset_valr#r#r$rgvs  zSetInstance.__init__cCs|jjSrh)rbrTrkr#r#r$rT}szSetInstance.dtypecCs.|j}|j}|jj||j}t|||j|S)z/ The _SetPayload for this set. )r`ranrt meminfo_datameminfor^rb)rfrrr r#r#r$r"szSetInstance.payloadcCs |jSrh)r _getvaluerkr#r#r$r8szSetInstance.valuecCs|jjSrh)rrrkr#r#r$rszSetInstance.meminfocCs|jjSrhrparentrkr#r#r$rszSetInstance.parentcCs ||j_dSrhrrlr#r#r$rscCs|jjS)z< Return the number of elements in the size. )r"rnrkr#r#r$get_sizeszSetInstance.get_sizecCs |jjr|rtjntj|j_dSrh)rbZ reflectedrrrr"rtrfvalr#r#r$ set_dirtyszSetInstance.set_dirtyTc Cs|j}|j}|j}||_||||_|j} t| jd} | | | } |_|j t |||dd| |j | |_ W5QRX|r| | |ddS)NrxTZlikely)r`rar/ incref_valuer{rnrr3r4rrzrIrpupsizer) rfr"r\rr= do_resizerrold_hashrnrr#r#r$ _add_entrys   zSetInstance._add_entryc Cs|j}|j}|j||dd\}} ||} || || } | j} || _|rZ|||| _|j } t | j d}| | |} |_ |jt||| dd| |j||_W5QRX|r|| |dW5QRXdS)NTrrxr)r`rarnot_rzrwr/rr{rnrr3r4rrIrprr)rfr"rr=r do_increfrrrr not_foundr\rrnrr#r#r$_add_keys*     zSetInstance._add_keycCsft|jjt|_|r"||j|j}t|jd}|j ||}|_|rX| || ddS)NrxT) rr3r/r4rK decref_valuer{rnrasubdownsizer)rfr"r\r do_decrefrnrr#r#r$ _remove_entrys  zSetInstance._remove_entryc CsN|j}|j}|||\}}||||} ||| |W5QRX|Srh)r`rarrzrwr) rfr"rr=rrrrrr\r#r#r$ _remove_keys  zSetInstance._remove_keycCs8|j}|j}|j}t|||jj|}|||||dSrh)r`rar"r?rbrTr)rfrrrrr"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)r`rar"_pyapi_get_hash_valuer)rfpyapirrrrr"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|Srh)rD)rr#r#r$wrapper sz2SetInstance._pyapi_get_hash_value..wrapperFr) rbrTrrr signatureZ call_jit_coderzretZget_null_object) rfrrrrZargtypesZrestyrrAr;Zis_errorretvalr#r#r$rs z!SetInstance._pyapi_get_hash_valuecCs8|j}|j}|j}t|||jj|}|||\}}|Srh)r`rar"r?rbrTr)rfrrrr"r=rrr#r#r$containss zSetInstance.containscCs6|j}|j}|j}t|||jj|}||||}|Srh)r`rar"r?rbrTr)rfrrrr"r=rr#r#r$discard s zSetInstance.discardc Csh|j}|j}||jj}t||}|j}|$}| |j ||j ||ddW5QRX| |S)NF)r) r`rarrbrTrZ alloca_oncer"rr~r{rr})rfrrZltyr{r"r\r#r#r$pop)s  zSetInstance.popcCs<|j}|j}|tj}t|t}||| ddS)NT) r`rarrrrr3MINSIZE_replace_payloadr)rfrrrminsizer#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. NrETrF)zcannot copy setr)r`rar"rnrpr4rbrGif_else _copy_payloadrzr call_convreturn_user_exc MemoryErrorchoose_alloc_size_allocate_payloadrr\rr{r/)rfrrr"rnrpotherZno_deleted_entriesZ if_no_deletedZ if_deletedoknentries other_payloadrr\r#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) r`rar"rr\rr{r/rzrrrrn) rfrrrr"rrr\r_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) r`rar"rr\rr{r/rrn)rfrrrr"rrr\r#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)r`rar"rr\r{r/rrwrrrrrn)rfrrrrrr{r=r"rrr\Z 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)NrN<=)r`rar"rrrrrGrnr~rrr\rr{r/rzrr]r})rfrstrictrrr"rcmp_opresZ if_smaller if_largerrr\rrr#r#r$issubsets(  0zSetInstance.issubsetc s|j}|j|j}|j}ttjfdd}d|j|j>\}}||||W5QRX||||W5QRXW5QRX S)Nc s^|L}|j}||j|j\}}|tj| W5QRXW5QRXdSrh) rr\rr{r/rzr~rrr])ZsmallerZlargerrr\rrrrr#r$checks   z%SetInstance.isdisjoint..check>) r`rar"rrrrrGrnr})rfrrr"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)NrE)r`rar"rrrrrGrnrr\rr{r/rzrr~rr]r}) rfrrrr"rrZ if_same_sizerrr\rrr#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) rrrrr3rrOintrr) clsrrrnitemsrrrfrr#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)rrzrrrr)rrrrrrrfr#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_nullrpyobjectrrincrefr8)rrrrrrfr#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. rx calcsize.body calcsize.end>=Fr)r4rr3rshlrrrr|rr}rGrzr~r)rrrrrrr 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. rxrFrrrzKupsize to %zd items: current size = %zd, min entries = %zd, new size = %zd zcannot grow setN)r`rar4rr3r"rrrjrGrzrrrr|rr}r~rr DEBUG_ALLOCSprintf_resize)rfrrrrrtwor"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. rxrrrrNFrrrrzMdownsize to %zd items: current size = %zd, min entries = %zd, new size = %zd zcannot shrink setN)r`rar4rr3rr"rr6rGrrjrrzrrrr|rr}rr~rrr r )rfrrrrrr rr"rmax_sizerr r rrrrr#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)r`rarrzrrrrr"rr\rr{r/ _free_payloadr ) rfr"rerrmsgrrZ old_payloadrrr\r#r#r$r s   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! TrFr)zcannot reallocate setN)r`rar"rr\rr{rr rrzrrrr)rfrrrrr\rr#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! rrxFrT)r.allocated %zd bytes for set at %p: mask = %zd ),r`rarrrrrrrr3rrrbr'rZmuladd_with_overflowrzr~rr}rrrZmeminfo_varsize_alloc_uncheckedis_null _imp_dtormodule"meminfo_new_varsize_dtor_uncheckedr voidptr_trrrrr"Zmemsetr rnrprrrrjrr )rfrrrrrrrrr! payload_size entry_size allocsizeZovfrr alloc_okdtorif_errorif_okr"Znew_maskr#r#r$rsp         "zSetInstance._allocate_payloadcCs|jj|j|j|dS)z9 Free an allocated old payload at *ptr*. N)r`rZmeminfo_varsize_freerar)rfr 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. rrxTrFr)-r`rarrrrrrrr3rrrbr'rrjrrrzr}rrrrrrrrr~rrrr"rnrprrZ raw_memcpyr_rrr\r{rr r )rfZ src_payloadrrrrrrr!rrrjrrr rrr!r"r"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.)nameZ linkonce_odrr)rrrrZsize_tr FunctionTypeZVoidTyperbrTZget_or_insert_functionZis_declarationlinkageZ IRBuilderrr^rArr\rdecrefr{Zret_void) rfrrZ llvoidptrZllsize_tr:fnamer<rr"rr\r#r#r$ris"   zSetInstance._imp_dtorcCs|jj|j|jj|dS)z Incref an element value N)r`rrrarbrTrr#r#r$rszSetInstance.incref_valuecCs|jj|j|jj|dS)z Decref an element value N)r`rr&rarbrTrr#r#r$rszSetInstance.decref_value)T)TT)TT)T)T)T)F)N)N)F).rrrrgrrTr"r8rrrrrrrrrrrrrrrrrrrrrrr classmethodrrrrrrr rrrrrrrr#r#r#r$rtsf              '     .< CArc@sZeZdZddZeddZeddZeddZed d Z e j d d Z d d Z dS)SetIterInstancecCsL||_||_||_|||||_|jj||j}t|||jj ||_ dSrh) r`rarbr_iterrrrr^ containerrc)rfrr iter_typeZiter_valr r#r#r$rgs zSetIterInstance.__init__cCsJt|||j|}||||d}|tjd}t|||j_|j |j_ |Sr@) rr+ get_constantrrrrr*r[r)rrrr,rZset_instrfr[r#r#r$from_sets  zSetIterInstance.from_setcCs |jSrh)r*rrkr#r#r$r8szSetIterInstance.valuecCs|jjSrh)r*rrkr#r#r$rszSetIterInstance.meminfocCs|j|jjSrh)rar}r*r[rkr#r#r$r[szSetIterInstance.indexcCs|j||jjdSrh)rar~r*r[rlr#r#r$r[sc Csr|j}|j}t|jd}||j|d:}|j}|| |j |j |j||_| W5QRXdS)Nrxr)r[rcrr3r4Z set_exhaustedrr\Z set_validZyield_r{rarr])rfresultr[r"rrr\r#r#r$iternexts zSetIterInstance.iternextN) rrrrgr(r.rr8rr[rr0r#r#r#r$r)s     r)c Cst|}t||||}|dkrt||}t||}|tj|}t ||(} | t ||d| j } | | W5QRXt||||jS)zD Build a set of the given type, containing the given items. r)lenrrrZ pack_arrayrr-rrrr}rur[rr r8) rrritemsrinstarrayZ array_ptrcountrrr#r#r$ build_sets  r6cCs$|j}t|||}t||||jSrh) return_typerrr r8)rrr;rArr3r#r#r$set_empty_constructorsr8c Cs||j}|j\}|\}t||||}t||||}t||||&} || j|j ||j | jW5QRXt ||||jSrh) r7rArrrr rr8rr&rTr ) rrr;rAr items_typer2nr3rr#r#r$set_constructors r;cCs t|||jd|d}|Sr@)rrArrrr;rAr3r#r#r$set_lensr=cCs&t|||jd|d}||dSNrrx)rrArr<r#r#r$in_setsr?ZgetitercCs(t|||j|d}t|||j|jSr@)r)r.r7r r8r<r#r#r$ getiter_setsr@r0cCs&t|||jd|d}||dSr@)r)rAr0)rrr;rAr/r3r#r#r$iternext_listiter srAzset.addcCs2t|||jd|d}|d}|||Sr>)rrArget_dummy_valuerrr;rAr3rr#r#r$set_adds rDcCst||}dd}||fS)NcSs2t|||jd|d}|d}|||Sr>)rrArrBrCr#r#r$ set_discard"s z!_set_discard..set_discardrnone)r9rUrr;rEr#r#r$ _set_discards rHrcCsddS)NcSs t||Srh)rHrUrr#r#r$.z ol_set_discard..r#rIr#r#r$ol_set_discard,srLcCs||}dd}||fS)Nc SsZt|||jd|d}|jj}|jt||dd|j|t dW5QRX| S)NrFr)zset.pop(): empty set) rrAr"rnrzrrrrKeyErrorr)rrr;rAr3rnr#r#r$set_pop5s z_set_pop..set_poprS)r9rUr;rNr#r#r$_set_pop1s  rOrcCsddS)NcSst|Srh)rOrUr#r#r$rJCrKzol_set_pop..r#rPr#r#r$ ol_set_popAsrQcCst||}dd}||fS)Nc Ssbt|||jd|d}|d}||}|j||dd|j|tdW5QRX|S)NrrxFr)zset.remove(): key not in set) rrArrzrrrrMrB)rrr;rAr3rrr#r#r$ set_removeJs  z_set_remove..set_removerF)r9rUrr;rRr#r#r$ _set_removeFs  rSremovecCs|j|krddSdS)NcSs t||Srh)rSrIr#r#r$rJZrKzol_set_remove..rSrIr#r#r$ ol_set_removeWs rUcCst|}dd}||fS)NcSs(t|||jd|d}||Sr@)rrArrBr<r#r#r$ set_clearcsz_set_clear..set_clearrF)r9rUr;rVr#r#r$ _set_clear_s rWrcCsddS)NcSst|Srh)rWrPr#r#r$rJmrKzol_set_clear..r#rPr#r#r$ ol_set_clearksrXcCs||}dd}||fS)NcSs2t|||jd|d}|}t|||j|jSr@)rrArr r7r8rrr;rAr3rr#r#r$set_copytsz_set_copy..set_copyr#)r9rUr;rZr#r#r$ _set_copypsr[rcCsddS)NcSst|Srh)r[rPr#r#r$rJ~rKzol_set_copy..r#rPr#r#r$ ol_set_copy|sr\cCsBt|||jd|d}t|||jd|d}|||Sr>)rrArrBrYr#r#r$set_difference_updates r]cCst||}|tfSrh)rrGr]r9abr;r#r#r$_set_difference_updates radifference_updatecCst||ddS)NcSs t||Srh)rar_r`r#r#r$rJrKz,set_difference_update_impl..rXrcr#r#r$set_difference_update_impls recCsBt|||jd|d}t|||jd|d}|||Sr>)rrArrBrYr#r#r$set_intersection_updates rfcCst||}|tfSrh)rrGrfr^r#r#r$_set_intersection_updates rgintersection_updatecCst||ddS)NcSs t||Srh)rgrcr#r#r$rJrKz.set_intersection_update_impl..rdrcr#r#r$set_intersection_update_impls ricCsBt|||jd|d}t|||jd|d}|||Sr>)rrArrBrYr#r#r$set_symmetric_difference_updates rjcCst||}|tfSrh)rrGrjr^r#r#r$ _set_symmetric_difference_updates rksymmetric_difference_updatecCst||ddS)NcSs t||Srh)rkrcr#r#r$rJrKz6set_symmetric_difference_update_impl..rdrcr#r#r$$set_symmetric_difference_update_impls rmz set.updatec Cst|||jd|d}|jd}|d}t||||}|dk rZ||jj|}||t||||:} ||| j |j |j } || |j ||j | j W5QRX|dk r| |jj|Sr>)rrArrr"rnrr castr8rTrr&rrB) rrr;rAr3r9r2r:rrZcastedr#r#r$ set_updates   rocs(tfddt|fdd}dS)Ncs|||}fdd}||fS)Ncs:|j|jdkst||||t|||jd|dSr@)r7rAAssertionErrorr )rrr;rArCr#r$codegensz>gen_operator_impl.._set_operator_intr..codegenr#r9r_r`r;rrrqr#r$_set_operator_intrs  z-gen_operator_impl.._set_operator_intrcst||fddS)Ncs ||Srhr#rcrtr#r$rJrKz=gen_operator_impl.._ol_set_operator..rdrcrur#r$_ol_set_operators z+gen_operator_impl.._ol_set_operator)rr)oprCrvr#)rtrCr$gen_operator_implsrxrcCst||dd}|S)NcSs|}|||Srh)rrbr_r`rUr#r#r$difference_impls z,impl_set_difference..difference_implrd)r_r`rzr#r#r$impl_set_differences r{ intersectioncCst||dd}|S)NcSs@t|t|kr&|}|||S|}|||SdSrh)r1rrhryr#r#r$intersection_impls  z+set_intersection..intersection_implrd)r_r`r}r#r#r$set_intersections  r~rcCst||dd}|S)NcSs@t|t|kr&|}|||S|}|||SdSrh)r1rrlryr#r#r$symmetric_difference_impls  z;set_symmetric_difference..symmetric_difference_implrd)r_r`rr#r#r$set_symmetric_differences  runioncCst||dd}|S)NcSs@t|t|kr&|}|||S|}|||SdSrh)r1rupdateryr#r#r$ union_impl(s  zset_union..union_implrd)r_r`rr#r#r$ set_union#s  rcCst||}dd}||fS)NcSs:t|||jd|d}t|||jd|d}||Sr>)rrArrYr#r#r$rr;sz _set_isdisjoint..codegenrbooleanrsr#r#r$_set_isdisjoint7s rrcCst||ddS)NcSs t||Srh)rrcr#r#r$rJHrKz set_isdisjoint..rdrcr#r#r$set_isdisjointDs rcCst||}dd}||fS)NcSs:t|||jd|d}t|||jd|d}||Sr>rrArrYr#r#r$rrOsz_set_issubset..codegenrrsr#r#r$ _set_issubsetKs rrcCst||ddS)NcSs t||Srh)rrcr#r#r$rJ\rKzset_issubset..rdrcr#r#r$ set_issubsetWs r issupersetcCst||dd}|S)NcSs ||Srh)rrcr#r#r$ superset_impldsz%set_issuperset..superset_implrd)r_r`rr#r#r$set_issuperset_s rcCst||}dd}||fS)NcSs:t|||jd|d}t|||jd|d}||Sr>)rrArrYr#r#r$rrmsz_set_eq..codegenrrsr#r#r$_set_eqis rcCst||ddS)NcSs t||Srh)rrcr#r#r$rJyrKzset_eq..rdrcr#r#r$set_equs rcCst||dd}|S)NcSs ||k Srhr#rcr#r#r$ne_implszset_ne..ne_implrd)r_r`rr#r#r$set_ne{s rcCst||}dd}||fS)NcSs>t|||jd|d}t|||jd|d}|j|ddS)NrrxT)rrrYr#r#r$rrsz_set_lt..codegenrrsr#r#r$_set_lts rcCst||ddS)NcSs t||Srh)rrcr#r#r$rJrKzset_lt..rdrcr#r#r$set_lts rcCst||dd}|S)NcSs||kSrhr#rcr#r#r$gt_implszset_gt..gt_implrd)r_r`rr#r#r$set_gts rcCs^t|||jd|d}t|||jd|d}||jtj}||jtj}|d||S)NrrxrE)rrAZptrtointrrrZ icmp_signed)rrr;rAr_r`mambr#r#r$set_iss rcCs|j|jkst|Srh)rTrp)rrZfromtyZtotyrr#r#r$ set_to_setsr)}__doc__ collectionsrmathr functoolsrZllvmliterZ numba.corerrrZnumba.core.imputilsrrr r r r r rrZ numba.miscrZ numba.cpythonrZnumba.core.errorsrrZnumba.core.extendingrrrr%r(rFrKr5rrrr?rDrIrMr2rX namedtuplerYobjectr^rr)r6setr8Z IterableTyper;r1rPr=rAnyr?r@ZSetIterZBORROWEDrArDrHrLrOrQrSrUrWrXr[r\r]rarerfrgrirjrkrmrorxiandiorisubixorZop_Zop_implrr{rr~xorror_rrrrlergerrrrnerrltrgtris_rrr#r#r#r$s  ,       "6