o i@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  rDcCt|jt}|d||S)z8 Whether the hash value denotes an empty entry. ==)rr3r4EMPTY icmp_unsigned)rrr=emptyr#r#r$ is_hash_emptySrJcCrE)z9 Whether the hash value denotes a deleted entry. rFrr3r4DELETEDrHrrr=Zdeletedr#r#r$is_hash_deletedZrKrOcCrE)z9 Whether the hash value denotes an active entry. ksz!check_all_set..z All arguments must be Sets, got csg|] }dj|jkqS)rdtype)rSsrAr#r$rTnsz'All Sets must be of the same type, got )allrrXr#rXr$ check_all_setjs rZSetLoopindexentrydo_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__cC|jjSNremaskrhr#r#r$rmz_SetPayload.maskcC ||j_dSrkrlrhr8r#r#r$rms cCrjrkreusedrnr#r#r$rsroz_SetPayload.usedcCrprkrrrqr#r#r$rs cCrjrkrefillrnr#r#r$rvroz_SetPayload.fillcCrprkrurqr#r#r$rvrtcCrjrkrefingerrnr#r#r$rxroz_SetPayload.fingercCrprkrwrqr#r#r$rxrtcCrjrkredirtyrnr#r#r$rzroz_SetPayload.dirtycCrprkryrqr#r#r$rzrtcC|jS)z> A pointer to the start of the entries array. )rfrnr#r#r$raz_SetPayload.entriescCr{)zC A pointer to the start of the NRT-allocated area. )rgrnr#r#r$r r|z_SetPayload.ptrcCs2t|j|j|}|jj|jt|j|d}|S)z) Get entry number *idx*. r) rgeprcrfrbrrr&rd)rhidxZ 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|} rQ|dtd}ddd} f dd }tt |t#| }|||| }||}|| Wd n1swY||A| }||| }|| }| || }|||}|| || |Wd n1swY,r"| }}d |||}|| |Wd n 1s2wY |Wd n 1sMwY|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 s2 |}|j}d|* |jf}| Wdn1s0wYWdn1s?wYt| Wdn1s]wYrt| } d|||} |WddS1swYdSdS)zO Check entry *i* against the value being searched for. rFN) rr/if_thenrHkeybranchrJrOloadr6store)ir^Z entry_hasheqj Zbb_foundZ bb_not_foundrrZeqfn for_insertZ free_indexZfree_index_sentinelr=itemrhr#r$ check_entrys*     "z(_SetPayload._lookup..check_entryNrFfound)#rbrcr4rmrdrVr-r.operatorrr0r1rr3ralloca_once_valueand_append_basic_block for_range LINEAR_PROBESraddrr goto_blocklshrmulr6rHposition_at_endphiZIntTypeZ add_incomingtrue_bit false_bit)rhrr=rintp_trmrVZtyctxr:r;onefiveZperturbr]bb_bodybb_endrrprrr#rr$_lookupsx                               z_SetPayload._lookupNc cs|j}|j}|tj}t|d}||j|}t j |||d?}| |j }t |||j} || t|j ||jd} | VWdn1sLwYWddSWddS1sdwYdS)zG Iterate over the payload's entries. Yield a SetLoop. rstartr\N)rbrcget_value_typerintprr3rrmrrrr]r2r/rr[r_) rhrrrrrsizeZ range_loopr^is_usedloopr#r#r$_iterate7s$    "z_SetPayload._iteratec cs|j}|j}|tj}t|d}t|d}|j}|d}|d}t ||j } | || |0|| } |||| |} || | || } t||| j} || ||Wdn1sjwY|||| } | |_ || VdS)z Yield a random entry from the payload. Caller must ensure the set isn't empty, otherwise the function won't end. rrZnext_entry_bodyZnext_entry_endN)rbrcrrrrr3rmrrrrxrrrrrrrr2r/cbranchr) rhrrrzerorrmrrr]rr^rr#r#r$ _next_entryKs.           z_SetPayload._next_entryFrk)__name__ __module__ __qualname__ripropertyrmsetterrsrvrxrzrar rr contextlibcontextmanagerrrr#r#r#r$r`us@             v r`c@seZdZddZeddZeddZeddZed d Zed d Z e j d d Z ddZ ddZ dUddZ dVddZdVddZdUddZdUddZdUddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0ZdWd2d3Zd4d5Zd6d7ZedXd9d:Z edXd;d<Z!ed=d>Z"ed?d@Z#dAdBZ$dCdDZ%dEdFZ&dGdHZ'dWdIdJZ(dKdLZ)dMdNZ*dOdPZ+dQdRZ,dSdTZ-d8S)Y SetInstancecCs2||_||_||_t|||_|||||_dSrk)rbrcrdr( _entrysize make_helper_set)rhrrrset_valr#r#r$rivs  zSetInstance.__init__cCrjrk)rdrVrnr#r#r$rV}rozSetInstance.dtypecCs.|j}|j}|jj||j}t|||j|S)z/ The _SetPayload for this set. )rbrcnrt meminfo_datameminfor`rd)rhrrr r#r#r$r"szSetInstance.payloadcC |jSrk)r _getvaluernr#r#r$r8 zSetInstance.valuecCrjrk)rrrnr#r#r$rrozSetInstance.meminfocCrjrkrparentrnr#r#r$rrozSetInstance.parentcCrprkrrqr#r#r$rrtcCrj)z< Return the number of elements in the size. )r"rsrnr#r#r$get_sizeszSetInstance.get_sizecCs$|jjr|r tjntj|j_dSdSrk)rdZ reflectedrrrr"rzrhvalr#r#r$ set_dirtyszSetInstance.set_dirtyTc Cs|j}|j}|j}||_||||_|j} t| jd} | | | } |_|j t |||dd| |j | |_ Wdn1sEwY|rQ| | |ddS)NrTZlikely)rbrcr/ incref_valuerrsrr3r4rrrJrvupsizer) rhr"r^rr= do_resizerrold_hashrsrr#r#r$ _add_entrys"  zSetInstance._add_entryc Cs|j}|j}|j||dd\}} ||} || `|| } | j} || _|r-|||| _|j } t | j d}| | |} |_ |jt||| dd| |j||_Wdn1sawY|rm|| |dWddS1s}wYdS)NTrrr)rbrcrnot_rrr/rrrsrr3r4rrJrvrr)rhr"rr=r do_increfrrrr not_foundr^rrsrr#r#r$_add_keys.      "zSetInstance._add_keycCsft|jjt|_|r||j|j}t|jd}|j ||}|_|r,| || ddS)NrT) rr3r/r4rM decref_valuerrsrcsubdownsizer)rhr"r^r do_decrefrsrr#r#r$ _remove_entrys  zSetInstance._remove_entryc Csd|j}|j}|||\}}||||} ||| |Wd|S1s+wY|Srk)rbrcrrrr) rhr"rr=rrrrrr^r#r#r$ _remove_keys   zSetInstance._remove_keycCs8|j}|j}|j}t|||jj|}|||||dSrk)rbrcr"r?rdrVr)rhrrrrr"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)rbrcr"_pyapi_get_hash_valuer)rhpyapirrrrr"r=r#r#r$ add_pyapis zSetInstance.add_pyapic Cs|jjg}tj}dd}|g}tj|g|R} ||| |\} } |j| dd|| Wd| S1s;wY| S)z=Python API compatible version of `get_hash_value()`. cSt|Srk)rD)rr#r#r$wrapper z2SetInstance._pyapi_get_hash_value..wrapperFrN) rdrVrrr signatureZ call_jit_coderretZget_null_object) rhrrrrargtypesZrestyrrAr;Zis_errorretvalr#r#r$rs  z!SetInstance._pyapi_get_hash_valuecCs8|j}|j}|j}t|||jj|}|||\}}|Srk)rbrcr"r?rdrVr)rhrrrr"r=rrr#r#r$containss zSetInstance.containscCs6|j}|j}|j}t|||jj|}||||}|Srk)rbrcr"r?rdrVr)rhrrrr"r=rr#r#r$discard s zSetInstance.discardcCs||j}|j}||jj}t||}|j}|}| |j ||j ||ddWdn1s4wY| |S)NF)r) rbrcrrdrVrZ alloca_oncer"rrrrr)rhrrZltyrr"r^r#r#r$pop)s   zSetInstance.popcCs<|j}|j}|tj}t|t}||| ddS)NT) rbrcrrrrr3MINSIZE_replace_payloadr)rhrrrminsizer#r#r$clear8s    zSetInstance.clearc Cs|j}|j}|j}|j}|j}t||||jd}|d||}|j|dd\}} |/| |} |j | | dd|j |tdWdn1sOwYWdn1s^wY| [||||} || } |j | | dd|j |tdWdn1swY|j} |} | j}|j| |j|jddWdn1swYWdn1swYWd|SWd|S1swY|S)z, Return a copy of this set. NrFTrF)zcannot copy setr)rbrcr"rsrvr4rdrHif_else _copy_payloadrr call_convreturn_user_exc MemoryErrorchoose_alloc_size_allocate_payloadrr^rrr/)rhrrr"rsrvotherZno_deleted_entriesZ if_no_deletedZ if_deletedoknentries other_payloadrr^r#r#r$copyAsT       zSetInstance.copyc Cs|j}|j}|j}|j}|5}|j}||j|j\}} || ||j ||ddWdn1s9wYWdn1sHwY| |j dS)z9 In-place intersection with *other* set. FrN) rbrcr"rr^rrr/rrrrrs) rhrrrr"rrr^r_r#r#r$ intersecths zSetInstance.intersectcCsl|j}|j}|j}|j}|}|j}|j||j|jddWdn1s)wY||j dS)z7 In-place difference with *other* set. FrN) rbrcr"rr^rrr/rrs)rhrrrr"rrr^r#r#r$ differencezs zSetInstance.differencec Cs|j}|j}|j}|m}|jj}|jj}|j}|j||dd\} } || } | | >\} } | |j || ddWdn1sEwY| | || ||Wdn1s_wYWdn1snwYWdn1s}wY| |jj dS)zA In-place symmetric difference with *other* set. TrFrN)rbrcr"rr^rr/rrrrrrrs)rhrrrrrrr=r"rrr^Z if_commonZ if_not_commonr#r#r$symmetric_differences*  z SetInstance.symmetric_differenceFc CsV|j}|j}|j}|j}|rdnd}t|tj}||||j|jy\} } | | tj |Wdn1s}|j}||j|j\}}|tj| Wdn1s-wYWddSWddS1sEwYdSrk) rr^rrr/rrrrr_)ZsmallerZlargerrr^rrrrr#r$checks   "z%SetInstance.isdisjoint..check>) rbrcr"rrrrrHrsr)rhrrr"rrr otherwiser#rr$ isdisjoints&    zSetInstance.isdisjointc CsJ|j}|j}|j}|j}t|tj}||d|j|jy\}}|L| 8} | j } | | j | j \} } ||| |tj|| Wdn1sUwYWdn1sdwYWdn1sswY||tj|Wdn1swYWdn1swY||S)NrF)rbrcr"rrrrrHrsrr^rrr/rrrrr_r) rhrrrr"rrZ if_same_sizerrr^rrr#r#r$equalss6   zSetInstance.equalsNc Csf|tj}|durt|t}nt|trt||}||||}||||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) rrrrr3rrQintrr) clsrrrnitemsrrrhrr#r#r$ allocate_exs    zSetInstance.allocate_excCs^|||||\}}|j||dd|j|tdWd|S1s(wY|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 setN)r rrrrr)rrrrr rrhr#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)rrrrrrhr#r#r$ from_meminfos zSetInstance.from_meminfoc Cs|j}t|d}t|t}|||}t||}|d} |d} || | | A| |} | d| |} |j | dd || Wdn1sSwY|| |} | | ||| Wdn1sswY|| | |S)zT Choose a suitable number of entries for the given number of items. r calcsize.body calcsize.end>=FrN)r4rr3rshlrrrrrrrHrrr)rrrr rrr min_entriesZsize_prrrZis_large_enoughZ next_sizer#r#r$r$s(              zSetInstance.choose_alloc_sizec CsX|j}|j}|j}t|d}t|d}|j}|||}||j|} | d|| } |j | ddot || } | d} | d} || || '|| }|||}||| | d||}||| | Wdn1sxwY|| || }tr||d || |||||d WddS1swYdS) zs When adding to the set, ensure it is properly sized for the given number of used entries. rrFrrrNzKupsize to %zd items: current size = %zd, min entries = %zd, new size = %zd zcannot grow set)rbrcr4rr3r"rrrmrHrrrrrrrrrr DEBUG_ALLOCSprintf_resize)rhr 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 | |}| | ||Wd n1swY||| ||Wd n1swY|||| }tr||d || | ||||d Wd d S1swYd S)zw When removing from the set, ensure it is properly sized for the given number of used entries. rrrrrPFrrrrNzMdownsize to %zd items: current size = %zd, min entries = %zd, new size = %zd zcannot shrink set)rbrcr4rr3rr"rr6rHrrmrrrrrrrrrrrrrr)rhr rrrrrrr"rmax_sizerrrrrrrr#r#r$rqsR                   "zSetInstance.downsizec Cs|j}|j}|}|j|dd}|j||dd|j|t|fWdn1s-wY|j}| }|j } |j || j | j dddWdn1sSwY||jdS)zw Resize the payload to the given number of entries. CAUTION: *nentries* must be a power of 2! TreallocFrN)rr)rbrcrrrrrrr"rr^rrr/ _free_payloadr ) rhr"rerrmsgrrZ old_payloadrrr^r#r#r$rs"  zSetInstance._resizecCs|j}|j}|j}|j}||jWdn1swY||jj|j |dd}|j | |dd|j |tdWddS1sPwYdS)z Replace the payload with a new empty payload with the given number of entries. CAUTION: *nentries* must be a power of 2! NTrFr)zcannot reallocate set)rbrcr"rr^rrr!r rrrrrr)rhrrrrr^rr#r#r$rs  "zSetInstance._replace_payloadc Csd|j}|j}t|tj}|tj}t |d}t |d}| t |j } | | } |j} | | 8} t||t || t || \} } |j| dd|tj|Wdn1s_wY|j||dd|r|jj}|jj||| d}t||}n|||j}|j|| ||tj}t||}|j|ddn\}}||tj|Wdn1swY|@|s||j_|tj |j_!|j"}t#||j$| d||_%||_&||_'|(||}||_)t*r|+|d | |j$|Wdn 1swYWdn 1swYWdn 1s(wY||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! rrFrNT)r.allocated %zd bytes for set at %p: mask = %zd ),rbrcrrrrrrrr3rrrdr'rZmuladd_with_overflowrrrrrrrZmeminfo_varsize_alloc_uncheckedis_null _imp_dtormodule"meminfo_new_varsize_dtor_uncheckedr voidptr_trr r rr"Zmemsetr rsrvrxrrmrr)rhrr rrrrrrr! payload_size entry_size allocsizeZovfrr alloc_okdtorif_errorif_okr"Znew_maskr#r#r$rsx         $zSetInstance._allocate_payloadcCs|jj|j|j|dS)z9 Free an allocated old payload at *ptr*. N)rbrZmeminfo_varsize_freercr)rhr 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|Wdn1swY|O||j_|j }|j!|_!|j"|_"||_#| |_t$||j%|j%| | |&}|'|j(j)Wdn1swYt*r|+|d| |j,| Wdn1swYWdn1swYWdn 1s wY||S)z7 Raw-copy the given payload into self. rrTrFNr$)-rbrcrrrrrrrr3rrrdr'rrmrrrrr&r'rr(rr)r%rrrrrr"rsrvrxZ raw_memcpyrarrr^rrrr )rhZ src_payloadrrrrrrr!r*r+rmrr,r.rr-r/r0r"rr#r#r$r(sj         "zSetInstance._copy_payloadc Cstj}|tj}tt|||g}d|jj }tj |||d}|j rad|_ t |}t|||j|jd} | } | j} |j||jj | jWdn1sXwY||S)z Define the dtor for set z .dtor.set.)nameZ linkonce_odrrN)rr)rrZsize_tr FunctionTypeZVoidTyperdrVZget_or_insert_functionZis_declarationlinkageZ IRBuilderrr`rArr^rdecrefrZret_void) rhrr'Z llvoidptrZllsize_tr:fnamer<rr"rr^r#r#r$r&is$  zSetInstance._imp_dtorcC|jj|j|jj|dS)z Incref an element value N)rbrrrcrdrVrr#r#r$rzSetInstance.incref_valuecCr6)z Decref an element value N)rbrr4rcrdrVrr#r#r$rr7zSetInstance.decref_value)T)TTrrk).rrrrirrVr"r8rrrrrrrrrrrrrrrrrrrrrrr classmethodr r rrrrrrrr!rr&rrr#r#r#r$rtsh               '     .< CA rc@sZeZdZddZeddZeddZeddZed d Z e j d d Z d d Z dS)SetIterInstancecCsL||_||_||_|||||_|jj||j}t|||jj ||_ dSrk) rbrcrdr_iterrrrr` containerre)rhrr iter_typeZiter_valr r#r#r$ris zSetIterInstance.__init__cCsJt|||j|}||||d}|tjd}t|||j_|j |j_ |Sr@) rr; get_constantrrrrr:r]r)rrrr<rZset_instrhr]r#r#r$from_sets  zSetIterInstance.from_setcCrrk)r:rrnr#r#r$r8rzSetIterInstance.valuecCrjrk)r:rrnr#r#r$rrozSetIterInstance.meminfocCs|j|jjSrk)rcrr:r]rnr#r#r$r]szSetIterInstance.indexcCs|j||jjdSrk)rcrr:r]rqr#r#r$r]scCs|j}|j}t|jd}||j|d#}|j}|| |j |j |j||_| WddS1s=wYdS)Nrr)r]rerr3r4Z set_exhaustedrr^Z set_validZyield_rrcrr_)rhresultr]r"rrr^r#r#r$iternexts  "zSetIterInstance.iternextN) rrrrir8r>rr8rr]rr@r#r#r#r$r9s      r9c Cst|}t||||}|dkrJt||}t||}|tj|}t ||} | t ||d| j } | | Wdn1sEwYt||||jS)zD Build a set of the given type, containing the given items. rN)lenrr rZ pack_arrayrr=rrrrr}r]rr r8) rrritemsr instarrayZ array_ptrcountrrr#r#r$ build_sets   rFcCs$|j}t|||}t||||jSrk) return_typerr r r8)rrr;rArrCr#r#r$set_empty_constructorsrHc Cs|j}|j\}|\}t||||}t||||}t||||} || j|j ||j | jWdn1s;wYt ||||jSrk) rGrArrr r rr8rr4rVr ) rrr;rAr items_typerBnrCrr#r#r$set_constructors rKcCs t|||jd|d}|Sr@)rrArrrr;rArCr#r#r$set_lensrMcCs&t|||jd|d}||dSNrr)rrArrLr#r#r$in_setsrOZgetitercCs(t|||j|d}t|||j|jSr@)r9r>rGr r8rLr#r#r$ getiter_setsrPr@cCs&t|||jd|d}||dSr@)r9rAr@)rrr;rAr?rCr#r#r$iternext_listiter srQzset.addcC2t|||jd|d}|d}|||SrN)rrArget_dummy_valuerrr;rArCrr#r#r$set_adds rUcCt||}dd}||fS)NcSrRrN)rrArrSrTr#r#r$ set_discard"s z!_set_discard..set_discardrnone)r9rWrr;rWr#r#r$ _set_discards rZrcCddS)NcS t||Srk)rZrWrr#r#r$. z ol_set_discard..r#r]r#r#r$ol_set_discard,ror`cCs||}dd}||fS)NcSstt|||jd|d}|jj}|jt||dd|j|t dWd| S1s1wY| S)NrFr)zset.pop(): empty set) rrAr"rsrrr%rrKeyErrorr)rrr;rArCrsr#r#r$set_pop5s  z_set_pop..set_poprU)r9rWr;rbr#r#r$_set_pop1s  rcrcCr[)NcSrrk)rcrWr#r#r$r^Czol_set_pop..r#rdr#r#r$ ol_set_popArorfcCrV)NcSs|t|||jd|d}|d}||}|j||dd|j|tdWd|S1s5wY|S)NrrFr)zset.remove(): key not in set) rrArrrrrrarS)rrr;rArCrrr#r#r$ set_removeJs   z_set_remove..set_removerX)r9rWrr;rgr#r#r$ _set_removeFs  rhremovecCs|j|kr ddSdS)NcSr\rk)rhr]r#r#r$r^Zr_zol_set_remove..rUr]r#r#r$ ol_set_removeWs rjcCst|}dd}||fS)NcSs(t|||jd|d}||Sr@)rrArrSrLr#r#r$ set_clearcsz_set_clear..set_clearrX)r9rWr;rkr#r#r$ _set_clear_s rlrcCr[)NcSrrk)rlrdr#r#r$r^mrezol_set_clear..r#rdr#r#r$ ol_set_clearkrormcCs||}dd}||fS)NcSs2t|||jd|d}|}t|||j|jSr@)rrArr rGr8rrr;rArCrr#r#r$set_copytsz_set_copy..set_copyr#)r9rWr;ror#r#r$ _set_copypsrprcCr[)NcSrrk)rprdr#r#r$r^~rezol_set_copy..r#rdr#r#r$ ol_set_copy|rorqcCBt|||jd|d}t|||jd|d}|||SrN)rrArrSrnr#r#r$set_difference_updates rscCt||}|tfSrk)rrYrsr9abr;r#r#r$_set_difference_update rxdifference_updatecCt||ddS)NcSr\rk)rxrvrwr#r#r$r^r_z,set_difference_update_impl..rZr|r#r#r$set_difference_update_impl r~cCrrrN)rrArrSrnr#r#r$set_intersection_update rcCrtrk)rrYrrur#r#r$_set_intersection_updateryrintersection_updatecCr{)NcSr\rk)rr|r#r#r$r^r_z.set_intersection_update_impl..r}r|r#r#r$set_intersection_update_implrrcCrrrN)rrArrSrnr#r#r$set_symmetric_difference_updaterrcCrtrk)rrYrrur#r#r$ _set_symmetric_difference_updateryrsymmetric_difference_updatecCr{)NcSr\rk)rr|r#r#r$r^r_z6set_symmetric_difference_update_impl..r}r|r#r#r$$set_symmetric_difference_update_implrrz set.updatec Cst|||jd|d}|jd}|d}t||||}|dur-||jj|}||t||||"} ||| j |j |j } || |j ||j | j Wdn1sYwY|duri| |jj|SrN)rrArrr"rsrr castr8rVrr4rrS) rrr;rArCrIrBrJrrZcastedr#r#r$ set_updates    rcs(tfddt|fdd}dS)Ncs|||}fdd}||fS)Ncs:|j|jdks J||||t|||jd|dSr@)rGrAr )rrr;rArCr#r$codegensz>gen_operator_impl.._set_operator_intr..codegenr#r9rvrwr;rrr#r$_set_operator_intrs  z-gen_operator_impl.._set_operator_intrcst||fddS)Ncs ||Srkr#r|rr#r$r^r_z=gen_operator_impl.._ol_set_operator..r}r|rr#r$_ol_set_operators  z+gen_operator_impl.._ol_set_operator)rr)oprCrr#)rrCr$gen_operator_implsrrcCt||dd}|S)NcSs|}|||Srk)rrzrvrwrWr#r#r$difference_impls z,impl_set_difference..difference_implr})rvrwrr#r#r$impl_set_differences r intersectioncCr)NcSs<t|t|kr|}|||S|}|||Srk)rArrrr#r#r$intersection_impl  z+set_intersection..intersection_implr})rvrwrr#r#r$set_intersection  rrcCr)NcS<t|t|kr|}|||S|}|||Srk)rArrrr#r#r$symmetric_difference_implrz;set_symmetric_difference..symmetric_difference_implr})rvrwrr#r#r$set_symmetric_differencerrunioncCr)NcSrrk)rArupdaterr#r#r$ union_impl(rzset_union..union_implr})rvrwrr#r#r$ set_union#rrcCrV)NcS:t|||jd|d}t|||jd|d}||SrN)rrArrnr#r#r$r; z _set_isdisjoint..codegenrbooleanrr#r#r$_set_isdisjoint7 rrcCr{)NcSr\rk)rr|r#r#r$r^Hr_z set_isdisjoint..r}r|r#r#r$set_isdisjointD rcCrV)NcSrrNrrArrnr#r#r$rOrz_set_issubset..codegenrrr#r#r$ _set_issubsetKrrrcCr{)NcSr\rk)rr|r#r#r$r^\r_zset_issubset..r}r|r#r#r$ set_issubsetWs r issupersetcCr)NcSs ||Srk)rr|r#r#r$ superset_impld z%set_issuperset..superset_implr})rvrwrr#r#r$set_issuperset_s rcCrV)NcSrrN)rrArrnr#r#r$rmrz_set_eq..codegenrrr#r#r$_set_eqirrcCr{)NcSr\rk)rr|r#r#r$r^yr_zset_eq..r}r|r#r#r$set_equrrcCr)NcSs ||k Srkr#r|r#r#r$ne_implrzset_ne..ne_implr})rvrwrr#r#r$set_ne{ rcCrV)NcSs>t|||jd|d}t|||jd|d}|j|ddS)NrrT)rrrnr#r#r$rsz_set_lt..codegenrrr#r#r$_set_ltrrcCr{)NcSr\rk)rr|r#r#r$r^r_zset_lt..r}r|r#r#r$set_ltrrcCr)NcSs||kSrkr#r|r#r#r$gt_implrzset_gt..gt_implr})rvrwrr#r#r$set_gtrrcCs^t|||jd|d}t|||jd|d}||jtj}||jtj}|d||S)NrrrF)rrAZptrtointrrrZ icmp_signed)rrr;rArvrwmambr#r#r$set_iss rcCs|j|jksJ|SrkrU)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(rGrMr5rrrr?rDrJrOr2rZ namedtupler[objectr`rr9rFsetrHZ IterableTyperKrArRrMrAnyrOrPZSetIterZBORROWEDrQrUrZr`rcrfrhrjrlrmrprqrsrxr~rrrrrrrriandiorisubixorZop_Zop_implrrrrxorror_rrrrlergerrrrnerrltrgtris_rrr#r#r#r$s  ,      "6