o Td@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  rHcCt|jt}|d||S)z8 Whether the hash value denotes an empty entry. ==)rr6r7EMPTY icmp_unsigned)rrr@emptyr%r%r& is_hash_emptySrNcCrI)z9 Whether the hash value denotes a deleted entry. rJrr6r7DELETEDrLrrr@deletedr%r%r&is_hash_deletedZrOrTcCrI)z9 Whether the hash value denotes an active entry. ksz!check_all_set..z All arguments must be Sets, got csg|] }dj|jkqS)rdtype)rXsrEr%r&rYnsz'All Sets must be of the same type, got )allrr]r%r]r& check_all_setjs r_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__cC|jjSNrjmaskrnr%r%r&rsz_SetPayload.maskcC ||j_dSrqrrrnr;r%r%r&rss cCrprqrjusedrtr%r%r&ryruz_SetPayload.usedcCrvrqrxrwr%r%r&ry cCrprqrjfillrtr%r%r&r|ruz_SetPayload.fillcCrvrqr{rwr%r%r&r|rzcCrprqrjfingerrtr%r%r&r~ruz_SetPayload.fingercCrvrqr}rwr%r%r&r~rzcCrprqrjdirtyrtr%r%r&rruz_SetPayload.dirtycCrvrqrrwr%r%r&rrzcC|jS)z> A pointer to the start of the entries array. )rlrtr%r%r&rfz_SetPayload.entriescCr)zC A pointer to the start of the NRT-allocated area. )rmrtr%r%r&r!rz_SetPayload.ptrcCs2t|j|j|}|jj|jt|j|d}|S)z) Get entry number *idx*. r) rgeprhrlrgrrr(ri)rnidx entry_ptrrcr%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. rJN) rr2if_thenrLkeybranchrNrTloadr9store)irc entry_hasheqj bb_found bb_not_foundrreqfn for_insert free_indexfree_index_sentinelr@itemrnr%r& check_entrys*     "z(_SetPayload._lookup..check_entryNrJfound)#rgrhr7rsrir[r0r1operatorrr3r4rr6ralloca_once_valueand_append_basic_block for_range LINEAR_PROBESraddrr goto_blocklshrmulr9rLposition_at_endphiIntType add_incomingtrue_bit false_bit)rnrr@rintp_trsr[tyctxr=r>onefiveperturbrbbb_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. rstartraN)rgrhget_value_typerintprr6rrsrrrrbr5r2rr`rd) rnrrrrrsize range_looprcis_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. rrnext_entry_bodynext_entry_endN)rgrhrrrrr6rsrrrr~rrrrrrrr5r2cbranchr) rnrrrzerorrsrrrbrrcrr%r%r& _next_entryKs.           z_SetPayload._next_entryFrq)__name__ __module__ __qualname__ropropertyrssetterryr|r~rrfr!rr contextlibcontextmanagerrrr%r%r%r&reus@             v rec@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|||_|||||_dSrq)rgrhrir+ _entrysize make_helper_set)rnrrr set_valr%r%r&rovs  zSetInstance.__init__cCrprq)rir[rtr%r%r&r[}ruzSetInstance.dtypecCs.|j}|j}|jj||j}t|||j|S)z/ The _SetPayload for this set. )rgrhnrt meminfo_datameminforeri)rnrrr!r%r%r&r$szSetInstance.payloadcC |jSrq)r _getvaluertr%r%r&r; zSetInstance.valuecCrprq)rrrtr%r%r&rruzSetInstance.meminfocCrprqrparentrtr%r%r&rruzSetInstance.parentcCrvrqrrwr%r%r&rrzcCrp)z< Return the number of elements in the size. )r$ryrtr%r%r&get_sizeszSetInstance.get_sizecCs$|jjr|r tjntj|j_dSdSrq)ri reflectedrrrr$rrnvalr%r%r& set_dirtyszSetInstance.set_dirtyTc Cs|j}|j}|j}||_||||_|j} t| jd} | | | } |_|j t |||dd| |j | |_ Wdn1sEwY|rQ| | |ddS)NrTlikely)rgrhr2 incref_valuerryrr6r7rrrNr|upsizer) rnr$rcrr@ do_resizerrold_hashryrr%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)rgrhrnot_rrr2rrryrr6r7rrNr|rr)rnr$rr@r do_increfrrrr not_foundrcrryrr%r%r&_add_keys.      "zSetInstance._add_keycCsft|jjt|_|r||j|j}t|jd}|j ||}|_|r,| || ddS)NrT) rr6r2r7rQ decref_valuerryrhsubdownsizer)rnr$rcr do_decrefryrr%r%r& _remove_entrys  zSetInstance._remove_entryc Csd|j}|j}|||\}}||||} ||| |Wd|S1s+wY|Srq)rgrhrrrr) rnr$rr@rrrrrrcr%r%r& _remove_keys   zSetInstance._remove_keycCs8|j}|j}|j}t|||jj|}|||||dSrq)rgrhr$rCrir[r)rnrrrrr$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)rgrhr$_pyapi_get_hash_valuer)rnpyapirrrrr$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|Srq)rH)rr%r%r&wrapper z2SetInstance._pyapi_get_hash_value..wrapperFrN) rir[rrr signature call_jit_coderretget_null_object) rnrrrrargtypesrestyrrEr>is_errorretvalr%r%r&rs  z!SetInstance._pyapi_get_hash_valuecCs8|j}|j}|j}t|||jj|}|||\}}|Srq)rgrhr$rCrir[r)rnrrrr$r@rrr%r%r&containss zSetInstance.containscCs6|j}|j}|j}t|||jj|}||||}|Srq)rgrhr$rCrir[r)rnrrrr$r@rr%r%r&discard s zSetInstance.discardcCs||j}|j}||jj}t||}|j}|}| |j ||j ||ddWdn1s4wY| |S)NF)r) rgrhrrir[r alloca_oncer$rrrrr)rnrrltyrr$rcr%r%r&pop)s   zSetInstance.popcCs<|j}|j}|tj}t|t}||| ddS)NT) rgrhrrrrr6MINSIZE_replace_payloadr)rnrrrminsizer%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. NrJTrF)zcannot copy setr)rgrhr$ryr|r7rirLif_else _copy_payloadrr call_convreturn_user_exc MemoryErrorchoose_alloc_size_allocate_payloadrrcrrr2)rnrrr$ryr|otherno_deleted_entries if_no_deleted if_deletedoknentries other_payloadrrcr%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) rgrhr$rrcrrr2rrrrry) rnrrrr$rrrcr_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) rgrhr$rrcrrr2rry)rnrrrr$rrrcr%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)rgrhr$rrcrr2rrrrrrry)rnrrrrrrr@r$rrrc if_common 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-wYWddSWddS1sEwYdSrq) rrcrrr2rrrrrd)smallerlargerrrcrrrrr%r&checks   "z%SetInstance.isdisjoint..check>) rgrhr$rrrrrLryr)rnrrr$rr&r! otherwiser%r%r& isdisjoints&    zSetInstance.isdisjointc CsJ|j}|j}|j}|j}t|tj}||d|j|jy\}}|L| 8} | j } | | j | j \} } ||| |tj|| Wdn1sUwYWdn1sdwYWdn1sswY||tj|Wdn1swYWdn1swY||S)NrJ)rgrhr$rrrrrLryrrcrrr2rrrrrdr) rnrrrr$rr if_same_sizer(rrcrrr%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) rrrrr6rrVintr r ) clsrrr nitemsrrrnrr%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/rrr r r )r-rrr r.rrnr%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 rrnr%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)r7rr6rshlrrrrrrrLrrr)r-rrr.rrr min_entriessize_prrris_large_enough 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. rr7Frr5r6NzKupsize to %zd items: current size = %zd, min entries = %zd, new size = %zd zcannot grow set)rgrhr7rr6r$r8rrsrLrrrrrrrrrr DEBUG_ALLOCSprintf_resize)rnr.rrrrtwor$r9r 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. rr=r7rrUFrr5r6r'NzMdownsize to %zd items: current size = %zd, min entries = %zd, new size = %zd zcannot shrink set)rgrhr7rr6rr$r8r9rLrrsrrrrrrrrrrrr>r?r@)rnr.rrrrrArr$r9max_sizerrBrCrrrDrEr%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)rgrhr rrr r r r$rrcrrr2 _free_payloadr!) rnr$rerrmsgrr old_payloadrrrcr%r%r&r@s"  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! NTrGFr)zcannot reallocate set)rgrhr$rrcrrrIr!r rrr r r )rnrrrrrcrr%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 ),rgrhrrrrrrrr6rrrir)rmuladd_with_overflowrrrrrrrmeminfo_varsize_alloc_uncheckedis_null _imp_dtormodule"meminfo_new_varsize_dtor_uncheckedr voidptr_trr1r2rr$memsetr!ryr|r~rrsr>r?)rnrrHrrrrrrr" payload_size entry_size allocsizeovfrr!alloc_okdtorif_errorif_okr$new_maskr%r%r&r sx         $zSetInstance._allocate_payloadcCs|jj|j|j|dS)z9 Free an allocated old payload at *ptr*. N)rgrmeminfo_varsize_freerhr)rnr!r%r%r&rI"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. rrTrFNrM)-rgrhrrrrrrrr6rrrir)rrsrrrrrQrRrrSrrTrPrrrrrr$ryr|r~ raw_memcpyrfrrrcrr>r?r!)rn src_payloadrrrrrrr"rVrWrsrrXr[rrZr\r]r$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.)name linkonce_odrrN)rrTrrsize_tr FunctionTypeVoidTyperir[get_or_insert_functionis_declarationlinkage IRBuilderrrerErrcrdecrefrret_void) rnrrR llvoidptrllsize_tr=fnamer?rr$rrcr%r%r&rQis$  zSetInstance._imp_dtorcC|jj|j|jj|dS)z Incref an element value N)rgrr3rhrir[rr%r%r&rzSetInstance.incref_valuecCrp)z Decref an element value N)rgrrkrhrir[rr%r%r&rrqzSetInstance.decref_value)T)TTrrq).rrrrorr[r$r;rrrrrrrrrrrrrrrrrrrrr"r)r+ classmethodr/r0r4r rrr@rr rIrrQrrr%r%r%r&rtsh               '     .< CA rc@sZeZdZddZeddZeddZeddZed d Z e j d d Z d d Z dS)SetIterInstancecCsL||_||_||_|||||_|jj||j}t|||jj ||_ dSrq) rgrhrir_iterrrrre containerrj)rnrr iter_typeiter_valr!r%r%r&ros zSetIterInstance.__init__cCsJt|||j|}||||d}|tjd}t|||j_|j |j_ |SrD) rru get_constantrrrrrtrbr)r-rrrvrset_instrnrbr%r%r&from_sets  zSetIterInstance.from_setcCrrq)rtrrtr%r%r&r;rzSetIterInstance.valuecCrprq)rtrrtr%r%r&rruzSetIterInstance.meminfocCs|j|jjSrq)rhrrtrbrtr%r%r&rbszSetIterInstance.indexcCs|j||jjdSrq)rhrrtrbrwr%r%r&rbscCs|j}|j}t|jd}||j|d#}|j}|| |j |j |j||_| WddS1s=wYdS)Nrr)rbrjrr6r7 set_exhaustedrrc set_validyield_rrhrrd)rnresultrbr$rrrcr%r%r&iternexts  "zSetIterInstance.iternextN) rrrrorrrzrr;rrbrrr%r%r%r&rss      rsc Cst|}t||||}|dkrJt||}t||}|tj|}t ||} | t ||d| j } | | Wdn1sEwYt||||jS)zD Build a set of the given type, containing the given items. rN)lenrr0r pack_arrayrrxrrrrrrbrr r;) rrr itemsr.instarray array_ptrcountrrr%r%r& build_sets   rcCs$|j}t|||}t||||jSrq) return_typerr0r r;)rrr>rEr rr%r%r&set_empty_constructorsrc Cs|j}|j\}|\}t||||}t||||}t||||} || j|j ||j | jWdn1s;wYt ||||jSrq) rrErrr0r rr;rrkr[r ) rrr>rEr items_typernrrr%r%r&set_constructors rcCs t|||jd|d}|SrD)rrErrrr>rErr%r%r&set_lensrcCs&t|||jd|d}||dSNrr)rrErrr%r%r&in_setsrgetitercCs(t|||j|d}t|||j|jSrD)rsrzrr r;rr%r%r& getiter_setsrrcCs&t|||jd|d}||dSrD)rsrEr)rrr>rEr~rr%r%r&iternext_listiter srzset.addcC2t|||jd|d}|d}|||Sr)rrErget_dummy_valuerrr>rErrr%r%r&set_adds rcCt||}dd}||fS)NcSrr)rrErrrr%r%r& set_discard"s z!_set_discard..set_discardrnone)r<r\rr>rr%r%r& _set_discards rrcCddS)NcS t||Srq)rr\rr%r%r&. z ol_set_discard..r%rr%r%r&ol_set_discard,rurcCs||}dd}||fS)NcSstt|||jd|d}|jj}|jt||dd|j|t dWd| S1s1wY| S)NrFr)zset.pop(): empty set) rrEr$ryrrrPr r KeyErrorr)rrr>rErryr%r%r&set_pop5s  z_set_pop..set_poprZ)r<r\r>rr%r%r&_set_pop1s  rrcCr)NcSrrq)rr\r%r%r&rCzol_set_pop..r%rr%r%r& ol_set_popArurcCr)NcSs|t|||jd|d}|d}||}|j||dd|j|tdWd|S1s5wY|S)NrrFr)zset.remove(): key not in set) rrErrrr r rr)rrr>rErrrr%r%r& set_removeJs   z_set_remove..set_remover)r<r\rr>rr%r%r& _set_removeFs  rremovecCs|j|kr ddSdS)NcSrrq)rrr%r%r&rZrzol_set_remove..rZrr%r%r& ol_set_removeWs rcCst|}dd}||fS)NcSs(t|||jd|d}||SrD)rrErrrr%r%r& set_clearcsz_set_clear..set_clearr)r<r\r>rr%r%r& _set_clear_s rrcCr)NcSrrq)rrr%r%r&rmrzol_set_clear..r%rr%r%r& ol_set_clearkrurcCs||}dd}||fS)NcSs2t|||jd|d}|}t|||j|jSrD)rrErr rr;rrr>rErrr%r%r&set_copytsz_set_copy..set_copyr%)r<r\r>rr%r%r& _set_copypsrrcCr)NcSrrq)rrr%r%r&r~rzol_set_copy..r%rr%r%r& ol_set_copy|rurcCBt|||jd|d}t|||jd|d}|||Sr)rrErrrr%r%r&set_difference_updates rcCt||}|tfSrq)rrrr<abr>r%r%r&_set_difference_update rdifference_updatecCt||ddS)NcSrrq)rrrr%r%r&rrz,set_difference_update_impl..r_rr%r%r&set_difference_update_impl rcCrr)rrErrrr%r%r&set_intersection_update rcCrrq)rrrrr%r%r&_set_intersection_updaterrintersection_updatecCr)NcSrrq)rrr%r%r&rrz.set_intersection_update_impl..rrr%r%r&set_intersection_update_implrrcCrr)rrErrrr%r%r&set_symmetric_difference_updaterrcCrrq)rrrrr%r%r& _set_symmetric_difference_updaterrsymmetric_difference_updatecCr)NcSrrq)rrr%r%r&rrz6set_symmetric_difference_update_impl..rrr%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|Sr)rrErrr$ryrr castr;r[rrkrr) rrr>rErrrrrDrcastedr%r%r& set_updates    rcs(tfddt|fdd}dS)Ncs|||}fdd}||fS)Ncs:|j|jdks J||||t|||jd|dSrD)rrEr )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 ||Srqr%rrr%r&rrz=gen_operator_impl.._ol_set_operator..rrrr%r&_ol_set_operators  z+gen_operator_impl.._ol_set_operator)rr)oprGrr%)rrGr&gen_operator_implsrrcCt||dd}|S)NcSs|}|||Srq)rrrrr\r%r%r&difference_impls z,impl_set_difference..difference_implr)rrrr%r%r&impl_set_differences r intersectioncCr)NcSs<t|t|kr|}|||S|}|||Srq)rrrrr%r%r&intersection_impl  z+set_intersection..intersection_implr)rrrr%r%r&set_intersection  rrcCr)NcS<t|t|kr|}|||S|}|||Srq)rrrrr%r%r&symmetric_difference_implrz;set_symmetric_difference..symmetric_difference_implr)rrrr%r%r&set_symmetric_differencerrunioncCr)NcSrrq)rrupdaterr%r%r& union_impl(rzset_union..union_implr)rrrr%r%r& set_union#rrcCr)NcS:t|||jd|d}t|||jd|d}||Sr)rrEr)rr%r%r&r; z _set_isdisjoint..codegenrbooleanrr%r%r&_set_isdisjoint7 rr)cCr)NcSrrq)rrr%r%r&rHrz set_isdisjoint..rrr%r%r&set_isdisjointD rcCr)NcSrrrrEr"rr%r%r&rOrz_set_issubset..codegenrrr%r%r& _set_issubsetKrrr"cCr)NcSrrq)rrr%r%r&r\rzset_issubset..rrr%r%r& set_issubsetWs r issupersetcCr)NcSs ||Srq)r"rr%r%r& superset_impld z%set_issuperset..superset_implr)rrrr%r%r&set_issuperset_s rcCr)NcSrr)rrEr+rr%r%r&rmrz_set_eq..codegenrrr%r%r&_set_eqirrcCr)NcSrrq)rrr%r%r&ryrzset_eq..rrr%r%r&set_equrrcCr)NcSs ||k Srqr%rr%r%r&ne_implrzset_ne..ne_implr)rrrr%r%r&set_ne{ rcCr)NcSs>t|||jd|d}t|||jd|d}|j|ddS)NrrT)rrrr%r%r&rsz_set_lt..codegenrrr%r%r&_set_ltrrcCr)NcSrrq)rrr%r%r&rrzset_lt..rrr%r%r&set_ltrrcCr)NcSs||kSrqr%rr%r%r&gt_implrzset_gt..gt_implr)rrrr%r%r&set_gtrrcCs^t|||jd|d}t|||jd|d}||jtj}||jtj}|d||S)NrrrJ)rrEptrtointrrr icmp_signed)rrr>rErrmambr%r%r&set_iss rcCs|j|jksJ|SrqrZ)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+rKrQr8rrr>rCrHrNrTr5r_ namedtupler`objectrerrsrsetr IterableTyperrrWrrAnyrrSetIterBORROWEDrrrrrrrrrrrrrrrrrrrrrrriandiorisubixorop_op_implrrrrxorror_rrrrlergerrrrnerrltrgtris_rrr%r%r%r&s  ,      "6