a Df@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. kz!check_all_set..z All arguments must be Sets, got csg|]}dj|jkqS)rdtype)rQsrAr#r$rRnrSz'All Sets must be of the same type, got )allrrWr#rWr$ check_all_setjsrYSetLoopindexentrydo_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|jjSNrdmaskrgr#r#r$rksz_SetPayload.maskcCs ||j_dSrirjrgr8r#r#r$rkscCs|jjSrirdusedrlr#r#r$rosz_SetPayload.usedcCs ||j_dSrirnrmr#r#r$roscCs|jjSrirdfillrlr#r#r$rqsz_SetPayload.fillcCs ||j_dSrirprmr#r#r$rqscCs|jjSrirdfingerrlr#r#r$rssz_SetPayload.fingercCs ||j_dSrirrrmr#r#r$rsscCs|jjSrirddirtyrlr#r#r$rusz_SetPayload.dirtycCs ||j_dSrirtrmr#r#r$ruscCs|jS)z> A pointer to the start of the entries array. )rerlr#r#r$r`sz_SetPayload.entriescCs|jS)zC A pointer to the start of the NRT-allocated area. )rfrlr#r#r$r sz_SetPayload.ptrcCs2t|j|j|}|jj|jt|j|d}|S)z) Get entry number *idx*. r) rgeprbrerarrr&rc)rgidxZ 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 |tF| }|||| }||}|| Wd n1sL0Y||| }||| }|| }| || }|||}|| || |Wd n1s0YXrH| }}d |||}|| |Wd n1sh0Y|Wd n1s0Y|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 s0 |}|j}d|T |jf}|Wdn1s`0YWdn1s~0Yt|Wdn1s0Yr,t|>} d|||} |Wdn1s"0YdS)zO Check entry *i* against the value being searched for. rEN) rxr/if_thenrGkeybranchrIrMloadr6store)ir]Z entry_hasheqj Zbb_foundZ bb_not_foundrrZeqfn for_insertZ free_indexZfree_index_sentinelr=itemrgr#r$ check_entrys  F( z(_SetPayload._lookup..check_entryNrEfound)#rarbr4rkrcrUr-r.operatorrr0r1rr3ralloca_once_valueand_append_basic_block for_range LINEAR_PROBESr~addrr} goto_blocklshrmulr6rGposition_at_endphiZIntTypeZ add_incomingtrue_bit false_bit)rgrr=rintp_trkrUZtyctxr:r;oneZfiveZperturbr\bb_bodybb_endrrprrr#rr$_lookupsp                 ,       *    * * z_SetPayload._lookupNc cs|j}|j}|tj}t|d}||j|}t j |||dl}| |j }t |||j} || (t|j ||jd} | VWdn1s0YWdn1s0YdS)zG Iterate over the payload's entries. Yield a SetLoop. rystartr[N)rarbget_value_typerintprr3rrkrrrxr\r2r/r{rZr^) rgrrrrrsizeZ range_loopr]is_usedloopr#r#r$_iterate7s    z_SetPayload._iteratec cs|j}|j}|tj}t|d}t|d}|j}|d}|d}t ||j } | || |`|| } |||| |} || | || } t||| j} || ||Wdn1s0Y|||| } | |_ || VdS)z Yield a random entry from the payload. Caller must ensure the set isn't empty, otherwise the function won't end. rryZnext_entry_bodyZnext_entry_endN)rarbrrrrr3rkrrrrsr}rr~rrrrxr2r/cbranchr) rgrrrzerorrkrrr\rr]rr#r#r$ _next_entryKs*          ,  z_SetPayload._next_entry)F)N)__name__ __module__ __qualname__rhpropertyrksetterrorqrsrur`r rxr 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|||_|||||_dSri)rarbrcr( _entrysize make_helper_set)rgrrrset_valr#r#r$rhvs  zSetInstance.__init__cCs|jjSri)rcrUrlr#r#r$rU}szSetInstance.dtypecCs.|j}|j}|jj||j}t|||j|S)z/ The _SetPayload for this set. )rarbnrt meminfo_datameminfor_rc)rgrrr r#r#r$r"szSetInstance.payloadcCs |jSri)r _getvaluerlr#r#r$r8szSetInstance.valuecCs|jjSri)rrrlr#r#r$rszSetInstance.meminfocCs|jjSrirparentrlr#r#r$rszSetInstance.parentcCs ||j_dSrirrmr#r#r$rscCs|jjS)z< Return the number of elements in the size. )r"rorlr#r#r$get_sizeszSetInstance.get_sizecCs |jjr|rtjntj|j_dSri)rcZ reflectedrrrr"rurgvalr#r#r$ set_dirtyszSetInstance.set_dirtyTc Cs|j}|j}|j}||_||||_|j} t| jd} | | | } |_|j t |||dd | |j | |_ Wdn1s0Y|r| | |ddS)NryTZlikely)rarbr/ incref_valuer|rorr3r4rr{rIrqupsizer) rgr"r]rr= do_resizerrold_hashrorr#r#r$ _add_entrys  . zSetInstance._add_entryc Cs|j}|j}|j||dd\}} ||} || || } | j} || _|rZ|||| _|j } t | j d}| | |} |_ |jt||| dd | |j||_Wdn1s0Y|r|| |dWdn1s0YdS)NTrryr)rarbrnot_r{rxr/rr|rorr3r4rrIrqrr)rgr"rr=r do_increfrrrr not_foundr]rrorr#r#r$_add_keys*    . zSetInstance._add_keycCsft|jjt|_|r"||j|j}t|jd}|j ||}|_|rX| || ddS)NryT) rr3r/r4rK decref_valuer|rorbsubdownsizer)rgr"r]r do_decrefrorr#r#r$ _remove_entrys  zSetInstance._remove_entryc Csb|j}|j}|||\}}||(||} ||| |Wdn1sT0Y|Sri)rarbrr{rxr) rgr"rr=rrrrrr]r#r#r$ _remove_keys  ,zSetInstance._remove_keycCs8|j}|j}|j}t|||jj|}|||||dSri)rarbr"r?rcrUr)rgrrrrr"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)rarbr"_pyapi_get_hash_valuer)rgpyapirrrrr"r=r#r#r$ add_pyapis zSetInstance.add_pyapic Cs|jjg}tj}dd}|g}tj|g|R} ||| |\} } |j| dd|| Wdn1st0Y| S)z=Python API compatible version of `get_hash_value()`. cSst|Sri)rD)rr#r#r$wrapper sz2SetInstance._pyapi_get_hash_value..wrapperFrN) rcrUrrr signatureZ call_jit_coder{retZget_null_object) rgrrrrargtypesZrestyrrAr;Zis_errorretvalr#r#r$rs ,z!SetInstance._pyapi_get_hash_valuecCs8|j}|j}|j}t|||jj|}|||\}}|Sri)rarbr"r?rcrUr)rgrrrr"r=rrr#r#r$containss zSetInstance.containscCs6|j}|j}|j}t|||jj|}||||}|Sri)rarbr"r?rcrUr)rgrrrr"r=rr#r#r$discard s zSetInstance.discardcCs||j}|j}||jj}t||}|j}|.}| |j ||j ||ddWdn1sh0Y| |S)NF)r) rarbrrcrUrZ alloca_oncer"rrr|rr~)rgrrZltyr|r"r]r#r#r$pop)s  .zSetInstance.popcCs<|j}|j}|tj}t|t}||| ddS)NT) rarbrrrrr3MINSIZE_replace_payloadr)rgrrrminsizer#r#r$clear8s    zSetInstance.clearc Cs|j}|j}|j}|j}|j}t||||jd}|d||}|j|ddV\}} |^| |} |j | | dd |j |tdWdn1s0YWdn1s0Y| ||||} || } |j | | dd |j |tdWdn1s"0Y|j} |,} | j}|j| |j|jddWdn1sn0YWdn1s0YWdn1s0Y|S)z, Return a copy of this set. NrETrF)zcannot copy setr)rarbr"rorqr4rcrGif_else _copy_payloadr{r call_convreturn_user_exc MemoryErrorchoose_alloc_size_allocate_payloadrr]rr|r/)rgrrr"rorqotherZno_deleted_entriesZ if_no_deletedZ if_deletedoknentries other_payloadrr]r#r#r$copyAs:  @  $ fzSetInstance.copyc Cs|j}|j}|j}|j}|j}|j}||j|j\}} || | |j ||ddWdn1sr0YWdn1s0Y| |j dS)z9 In-place intersection with *other* set. FrN) rarbr"rr]rr|r/r{rrrro) rgrrrr"rrr]r_r#r#r$ intersecths LzSetInstance.intersectcCsl|j}|j}|j}|j}|,}|j}|j||j|jddWdn1sR0Y||j dS)z7 In-place difference with *other* set. FrN) rarbr"rr]rr|r/rro)rgrrrr"rrr]r#r#r$ differencezs 4zSetInstance.differencec Cs|j}|j}|j}|}|jj}|jj}|j}|j||dd\} } || } | | |\} } | |j || ddWdn1s0Y| | || ||Wdn1s0YWdn1s0YWdn1s0Y| |jj dS)zA In-place symmetric difference with *other* set. TrFrN)rarbr"rr]r|r/rrxrrrrro)rgrrrrrr|r=r"rrr]Z if_commonZ if_not_commonr#r#r$symmetric_differences  .jz SetInstance.symmetric_differenceFc Cs\|j}|j}|j}|j}|r dnd}t|tj}||||j|j\} } | | tj |Wdn1sx0Y| | p} | j } | | j| j\} }||| &| tj || Wdn1s0YWdn1s0YWdn1s(0YWdn1sH0Y||S)NrN<=)rarbr"rrrrrGrorrrr]rr|r/r{rr^r~)rgrstrictrrr"rcmp_opresZ if_smaller if_largerrr]rrr#r#r$issubsets( , zSetInstance.issubsetc s|j}|j|j}|j}ttjfdd}d|j|jp\}}||||Wdn1sv0Y||||Wdn1s0YWdn1s0Y S)Nc s|j}|j}||j|j\}}|&tj| Wdn1sZ0YWdn1sx0YdSri) rr]rr|r/r{rrrr^)ZsmallerZlargerrr]rrrrr#r$checks   z%SetInstance.isdisjoint..check>) rarbr"rrrrrGror~)rgrrr"rrr otherwiser#rr$ isdisjoints (FzSetInstance.isdisjointc CsN|j}|j}|j}|j}t|tj}||d|j|j\}}|| p} | j } | | j | j \} } ||| &|tj|| Wdn1s0YWdn1s0YWdn1s0Y||tj|Wdn1s0YWdn1s:0Y||S)NrE)rarbr"rrrrrGrorr]rr|r/r{rrrr^r~) rgrrrr"rrZ if_same_sizerrr]rrr#r#r$equalss& bNzSetInstance.equalsNc Csf|tj}|dur"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) clsrrrnitemsrrrgrr#r#r$ allocate_exs    zSetInstance.allocate_excCs\|||||\}}|j||dd |j|tdWdn1sN0Y|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)rr{rrrr)rrrrrrrgr#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)rrrrrrgr#r#r$ from_meminfos zSetInstance.from_meminfoc Cs|j}t|d}t|t}|||}t||}|d} |d} || | | | |} | d| |} |j | dd|| Wdn1s0Y|| |} | | ||| Wdn1s0Y|| | |S)zT Choose a suitable number of entries for the given number of items. ry calcsize.body calcsize.end>=FrN)r4rr3rshlrrrr}rr~rGr{rr)rrrrrrr min_entriesZsize_prrrZis_large_enoughZ next_sizer#r#r$r$s$         (  ( zSetInstance.choose_alloc_sizec CsZ|j}|j}|j}t|d}t|d}|j}|||}||j|} | d|| } |j | ddt || } | d} | d} || || N|| }|||}||| | d||}||| | Wdn1s0Y|| || }tr(||d || |||||d Wdn1sL0YdS) zs When adding to the set, ensure it is properly sized for the given number of used entries. ryrFrrrNzKupsize to %zd items: current size = %zd, min entries = %zd, new size = %zd zcannot grow set)rarbr4rr3r"rrrkrGr{rrrr}rr~rrr DEBUG_ALLOCSprintf_resize)rgrrrrrtwor"rr need_resize new_size_prrnew_size is_too_smallr#r#r$rCs8           ,  zSetInstance.upsizec Cs|j}|j}|j}t|d}t|d}t|t}|j}|||} || d| || |} || |} | |j |} | | d| | | d|| } |j | ddt|| } |d}|d }|||||| }|||}| d | |}| |||Wd n1s20Y||| ||Wd n1sh0Y|||| }tr||d || | ||||d Wd n1s0Yd S)zw When removing from the set, ensure it is properly sized for the given number of used entries. ryrrrrNFrrrrNzMdownsize to %zd items: current size = %zd, min entries = %zd, new size = %zd zcannot shrink set)rarbr4rr3rr"rr6rGrrkrr{rrrr}rr~rrrr r r )rgrrrrrr rr"rmax_sizerr rrrrrr#r#r$rqsL               * *  zSetInstance.downsizec Cs|j}|j}|}|j|dd}|j||dd"|j|t|fWdn1sZ0Y|j}| .}|j } |j || j | j dddWdn1s0Y||jdS)zw Resize the payload to the given number of entries. CAUTION: *nentries* must be a power of 2! TreallocFrN)rr)rarbrr{rrrrr"rr]rr|r/ _free_payloadr ) rgr"rerrmsgrrZ old_payloadrrr]r#r#r$r s " $zSetInstance._resizecCs|j}|j}|j"}|j}||jWdn1s>0Y||jj|j |dd}|j | |dd |j |tdWdn1s0YdS)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)rarbr"rr]rr|rr rr{rrrr)rgrrrrr]rr#r#r$rs * zSetInstance._replace_payloadc Csn|j}|j}t|tj}|tj}t |d}t |d}| t |j } | | } |j} | | 8} t||t || t || \} } |j| dd|tj|Wdn1s0Y|j||ddt|r|jj}|jj||| d}t||}n4|||j}|j|| ||tj}t||}|j|dd\}}||tj|Wdn1s0Y||s||j_|tj |j_!|j"}t#||j$| d||_%||_&||_'|(||}||_)t*r|+|d | |j$|Wdn1s0YWdn1s:0YWdn1sZ0Y||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! rryFrNT)r.allocated %zd bytes for set at %p: mask = %zd ),rarbrrrrrrrr3rrrcr'rZmuladd_with_overflowr{rrr~rrrZmeminfo_varsize_alloc_uncheckedis_null _imp_dtormodule"meminfo_new_varsize_dtor_uncheckedr voidptr_trrrrr"memsetr rorqrsrrkr r )rgrrrrrrrrr! payload_size entry_size allocsizeZovfrr alloc_okdtorif_errorif_okr"Znew_maskr#r#r$rsh      , . dzSetInstance._allocate_payloadcCs|jj|j|j|dS)z9 Free an allocated old payload at *ptr*. N)rarZmeminfo_varsize_freerbr)rgr r#r#r$r"szSetInstance._free_payloadc Cs6|j}|j}t|tj}|tj}t |d}t |d}| t |j }| |} |j} | | 8} |j} ||| } |t || |t || | } |j||ddf|||j}|j|| ||tj}t||}|j|dd\}}||tj|Wdn1s*0Y|||j_|j }|j!|_!|j"|_"||_#| |_t$||j%|j%| | |&}|'|j(j)Wdn1s0Yt*r|+|d| |j,| Wdn1s0YWdn1s0YWdn1s"0Y||S)z7 Raw-copy the given payload into self. rryTrFNr)-rarbrrrrrrrr3rrrcr'rrkrrr{r~rrrrrrrrrrrrr"rorqrsZ raw_memcpyr`rrr]r|r r r )rgZ src_payloadrrrrrrr!rrrkrr r"rr!r#r$r"rr#r#r$r(sZ      .  .dzSetInstance._copy_payloadc Cstj}|tj}tt|||g}d|jj }tj |||d}|j rd|_ t |}t|||j|jd} | ,} | j} |j||jj | jWdn1s0Y||S)z Define the dtor for set z .dtor.set.)nameZ linkonce_odrrN)rrrrZsize_tr FunctionTypeZVoidTypercrUZget_or_insert_functionZis_declarationlinkageZ IRBuilderrr_rArr]rdecrefr|Zret_void) rgrrZ llvoidptrZllsize_tr:fnamer<rr"rr]r#r#r$ris"  4zSetInstance._imp_dtorcCs|jj|j|jj|dS)z Incref an element value N)rarrrbrcrUrr#r#r$rszSetInstance.incref_valuecCs|jj|j|jj|dS)z Decref an element value N)rarr(rbrcrUrr#r#r$rszSetInstance.decref_value)T)TT)TT)T)T)T)F)N)N)F).rrrrhrrUr"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 ||_ dSri) rarbrcr_iterrrrr_ containerrd)rgrr iter_typeZiter_valr r#r#r$rhs zSetIterInstance.__init__cCsJt|||j|}||||d}|tjd}t|||j_|j |j_ |Sr@) rr- get_constantrrrrr,r\r)rrrr.rZset_instrgr\r#r#r$from_sets  zSetIterInstance.from_setcCs |jSri)r,rrlr#r#r$r8szSetIterInstance.valuecCs|jjSri)r,rrlr#r#r$rszSetIterInstance.meminfocCs|j|jjSri)rbr~r,r\rlr#r#r$r\szSetIterInstance.indexcCs|j||jjdSri)rbrr,r\rmr#r#r$r\scCs|j}|j}t|jd}||j|dD}|j}|| |j |j |j||_| Wdn1sx0YdS)Nryr)r\rdrr3r4Z set_exhaustedrr]Z set_validZyield_r|rbrr^)rgresultr\r"rrr]r#r#r$iternexts zSetIterInstance.iternextN) rrrrhr*r0rr8rr\rr2r#r#r#r$r+s     r+c Cst|}t||||}|dkrt||}t||}|tj|}t ||2} | t ||d| j } | | Wdn1s0Yt||||jS)zD Build a set of the given type, containing the given items. rN)lenrrrZ pack_arrayrr/rrrr~rvr\rr r8) rrritemsrinstarrayZ array_ptrcountrrr#r#r$ build_sets  (r8cCs$|j}t|||}t||||jSri) return_typerrr r8)rrr;rArr5r#r#r$set_empty_constructorsr:c Cs|j}|j\}|\}t||||}t||||}t||||0} || j|j ||j | jWdn1sv0Yt ||||jSri) r9rArrrr rr8rr(rUr ) rrr;rAr items_typer4nr5rr#r#r$set_constructors 2r=cCs t|||jd|d}|Sr@)rrArrrr;rAr5r#r#r$set_lensr?cCs&t|||jd|d}||dSNrry)rrArr>r#r#r$in_setsrAZgetitercCs(t|||j|d}t|||j|jSr@)r+r0r9r r8r>r#r#r$ getiter_setsrBr2cCs&t|||jd|d}||dSr@)r+rAr2)rrr;rAr1r5r#r#r$iternext_listiter srCzset.addcCs2t|||jd|d}|d}|||Sr@)rrArget_dummy_valuerrr;rAr5rr#r#r$set_adds rFcCst||}dd}||fS)NcSs2t|||jd|d}|d}|||Sr@)rrArrDrEr#r#r$ set_discard"s z!_set_discard..set_discardrnone)r9rVrr;rGr#r#r$ _set_discards rJrcCsddS)NcSs t||Sri)rJrVrr#r#r$.rSz ol_set_discard..r#rKr#r#r$ol_set_discard,srMcCs||}dd}||fS)NcSsnt|||jd|d}|jj}|jt||dd |j|t dWdn1s\0Y| S)NrFr)zset.pop(): empty set) rrAr"ror{rrrrKeyErrorr)rrr;rAr5ror#r#r$set_pop5s "z_set_pop..set_poprT)r9rVr;rOr#r#r$_set_pop1s  rPrcCsddS)NcSst|Sri)rPrVr#r#r$rLCrSzol_set_pop..r#rQr#r#r$ ol_set_popAsrRcCst||}dd}||fS)NcSsvt|||jd|d}|d}||}|j||dd |j|tdWdn1sd0Y|S)NrryFr)zset.remove(): key not in set) rrArr{rrrrNrD)rrr;rAr5rrr#r#r$ set_removeJs  "z_set_remove..set_removerH)r9rVrr;rSr#r#r$ _set_removeFs  rTremovecCs|j|krddSdS)NcSs t||Sri)rTrKr#r#r$rLZrSzol_set_remove..rTrKr#r#r$ ol_set_removeWs rVcCst|}dd}||fS)NcSs(t|||jd|d}||Sr@)rrArrDr>r#r#r$ set_clearcsz_set_clear..set_clearrH)r9rVr;rWr#r#r$ _set_clear_s rXrcCsddS)NcSst|Sri)rXrQr#r#r$rLmrSzol_set_clear..r#rQr#r#r$ ol_set_clearksrYcCs||}dd}||fS)NcSs2t|||jd|d}|}t|||j|jSr@)rrArr r9r8rrr;rAr5rr#r#r$set_copytsz_set_copy..set_copyr#)r9rVr;r[r#r#r$ _set_copypsr\rcCsddS)NcSst|Sri)r\rQr#r#r$rL~rSzol_set_copy..r#rQr#r#r$ ol_set_copy|sr]cCsBt|||jd|d}t|||jd|d}|||Sr@)rrArrDrZr#r#r$set_difference_updates r^cCst||}|tfSri)rrIr^r9abr;r#r#r$_set_difference_updates rbdifference_updatecCst||ddS)NcSs t||Sri)rbr`rar#r#r$rLrSz,set_difference_update_impl..rYrdr#r#r$set_difference_update_impls rfcCsBt|||jd|d}t|||jd|d}|||Sr@)rrArrDrZr#r#r$set_intersection_updates rgcCst||}|tfSri)rrIrgr_r#r#r$_set_intersection_updates rhintersection_updatecCst||ddS)NcSs t||Sri)rhrdr#r#r$rLrSz.set_intersection_update_impl..rerdr#r#r$set_intersection_update_impls rjcCsBt|||jd|d}t|||jd|d}|||Sr@)rrArrDrZr#r#r$set_symmetric_difference_updates rkcCst||}|tfSri)rrIrkr_r#r#r$ _set_symmetric_difference_updates rlsymmetric_difference_updatecCst||ddS)NcSs t||Sri)rlrdr#r#r$rLrSz6set_symmetric_difference_update_impl..rerdr#r#r$$set_symmetric_difference_update_impls rnz set.updatec Cst|||jd|d}|jd}|d}t||||}|durZ||jj|}||t||||D} ||| j |j |j } || |j ||j | j Wdn1s0Y|dur| |jj|Sr@)rrArrr"rorr castr8rUrr(rrD) rrr;rAr5r;r4r<rrZcastedr#r#r$ set_updates   2rpcs(tfddt|fdd}dS)Ncs|||}fdd}||fS)Ncs:|j|jdksJ||||t|||jd|dSr@)r9rAr )rrr;rArCr#r$codegensz>gen_operator_impl.._set_operator_intr..codegenr#r9r`rar;rrrqr#r$_set_operator_intrs  z-gen_operator_impl.._set_operator_intrcst||fddS)Ncs ||Srir#rdrtr#r$rLrSz=gen_operator_impl.._ol_set_operator..rerdrur#r$_ol_set_operators z+gen_operator_impl.._ol_set_operator)rr)oprCrvr#)rtrCr$gen_operator_implsrxrcCst||dd}|S)NcSs|}|||Sri)rrcr`rarVr#r#r$difference_impls z,impl_set_difference..difference_implre)r`rarzr#r#r$impl_set_differences r{ intersectioncCst||dd}|S)NcSs@t|t|kr&|}|||S|}|||SdSri)r3rriryr#r#r$intersection_impls  z+set_intersection..intersection_implre)r`rar}r#r#r$set_intersections  r~rcCst||dd}|S)NcSs@t|t|kr&|}|||S|}|||SdSri)r3rrmryr#r#r$symmetric_difference_impls  z;set_symmetric_difference..symmetric_difference_implre)r`rarr#r#r$set_symmetric_differences  runioncCst||dd}|S)NcSs@t|t|kr&|}|||S|}|||SdSri)r3rupdateryr#r#r$ union_impl(s  zset_union..union_implre)r`rarr#r#r$ set_union#s  rcCst||}dd}||fS)NcSs:t|||jd|d}t|||jd|d}||Sr@)rrArrZr#r#r$rr;sz _set_isdisjoint..codegenrbooleanrsr#r#r$_set_isdisjoint7s rrcCst||ddS)NcSs t||Sri)rrdr#r#r$rLHrSz set_isdisjoint..rerdr#r#r$set_isdisjointDs rcCst||}dd}||fS)NcSs:t|||jd|d}t|||jd|d}||Sr@rrArrZr#r#r$rrOsz_set_issubset..codegenrrsr#r#r$ _set_issubsetKs rrcCst||ddS)NcSs t||Sri)rrdr#r#r$rL\rSzset_issubset..rerdr#r#r$ set_issubsetWs r issupersetcCst||dd}|S)NcSs ||Sri)rrdr#r#r$ superset_impldsz%set_issuperset..superset_implre)r`rarr#r#r$set_issuperset_s rcCst||}dd}||fS)NcSs:t|||jd|d}t|||jd|d}||Sr@)rrArrZr#r#r$rrmsz_set_eq..codegenrrsr#r#r$_set_eqis rcCst||ddS)NcSs t||Sri)rrdr#r#r$rLyrSzset_eq..rerdr#r#r$set_equs rcCst||dd}|S)NcSs ||k Srir#rdr#r#r$ne_implszset_ne..ne_implre)r`rarr#r#r$set_ne{s rcCst||}dd}||fS)NcSs>t|||jd|d}t|||jd|d}|j|ddS)NrryT)rrrZr#r#r$rrsz_set_lt..codegenrrsr#r#r$_set_lts rcCst||ddS)NcSs t||Sri)rrdr#r#r$rLrSzset_lt..rerdr#r#r$set_lts rcCst||dd}|S)NcSs||kSrir#rdr#r#r$gt_implszset_gt..gt_implre)r`rarr#r#r$set_gts rcCs^t|||jd|d}t|||jd|d}||jtj}||jtj}|d||S)NrryrE)rrAZptrtointrrrZ icmp_signed)rrr;rAr`ramambr#r#r$set_iss rcCs|j|jksJ|SrirT)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(rFrKr5rrr r?rDrIrMr2rY namedtuplerZobjectr_rr+r8setr:Z IterableTyper=r3rPr?rAnyrArBZSetIterZBORROWEDrCrFrJrMrPrRrTrVrXrYr\r]r^rbrfrgrhrjrkrlrnrprxiandiorisubixorZop_Zop_implrr{rr~xorror_rrrrlergerrrrnerrltrgtris_rrr#r#r#r$s  ,       "6