U ,d@sdZddlZddlZddlZddlZddlmZddlmZm Z m Z ddl m Z m Z mZmZmZmZmZmZmZddlmZddlmZddlmZdd lmZdd lmZd d Zd dZ dZ!dZ"dZ#dZ$dZ%dZ&ddZ'eddZ(ddZ)ddZ*ddZ+e,dd Z-Gd!d"d"e.Z/Gd#d$d$e.Z0Gd%d&d&e.Z1d'd(Z2e e3d)d*Z4e e3ej5d+d,Z6e e7ej8d-d.Z9e ej:ej8ej;d/d0Zeej?d5d6Z@e d7ej8ej;d8d9ZAe d:ej8ej;d;d<ZBe d=ej8d>d?ZCe d@ej8ej;dAdBZDe dCej8dDdEZEe dFej8dGdHZFe dIej8ej5dJdKZGe dLej8ej8dMdNZHe dOej8ej8dPdQZIe dRej8ej5dSdTZJejKeHfejLeJfejMeGfejNeIffD]&\ZOZPe eOej8ej8ePfdUdVZQqe ejRej8ej8e dWej8ej8dXdYZSe ejTej8ej8e dZej8ej8d[d\ZUe ejVej8ej8e d]ej8ej8d^d_ZWe ejXej8ej8e d`ej8ej8dadbZYe dcej8ej8dddeZZe ej[ej8ej8e dfej8ej8dgdhZ\e ej]ej8ej8e diej8ej8djdkZ^e ej_ej8ej8dldeZZe ej`ej8ej8dmdnZae ejbej8ej8dodpZce ejdej8ej8dqdrZee ejfej8ej8dsdtZge ej8ej8dudvZhdS)wz& Support for native homogeneous sets. N)ir)typestypingcgutils) lower_builtin lower_cast iternext_implimpl_ret_borrowedimpl_ret_new_refimpl_ret_untrackedfor_itercall_lenRefType)cached_property) quicksort)slicing) intrinsic)NumbaValueErrorcCs4t|}||}|||}|j|||dS)zx Given a set value and type, get its payload structure (as a reference, so that mutations are seen by all). ref)r SetPayload get_data_typeZ as_pointerbitcastmake_data_helper)contextbuilderset_typeptr payload_typeZptrtypayloadr 8/tmp/pip-unpacked-wheel-eu7e0c37/numba/cpython/setobj.pyget_payload_structs  r"cCs|t|}||S)z7 Return the entry size for the given set type. )rrSetEntryget_abi_sizeof)rrZlltyr r r!get_entry_size#sr%iFc Cs`|j}|t}|||fi}|||}|||f}t|||} t|jt } | | || S)z. Compute the hash of the given value. ) typing_contextresolve_value_typehash get_call_type get_function is_hash_usedrConstanttypeFALLBACKselect) rrtypvalue typingctxfntysigfnhZis_okfallbackr r r!get_hash_value;s    r<cs.fdd}|t}||fi}||fS)Ncst|||dSNr)r<)rrr4argsr5r r!implLsz'_get_hash_value_intrinsic..impl)r+r,r-)r6r5r@r7r8r r?r!_get_hash_value_intrinsicJs  rAcCst|jt}|d||S)z8 Whether the hash value denotes an empty entry. ==)rr0r1EMPTY icmp_unsigned)rrr:emptyr r r! is_hash_emptySsrFcCst|jt}|d||S)z9 Whether the hash value denotes a deleted entry. rBrr0r1DELETEDrDrrr:Zdeletedr r r!is_hash_deletedZsrJcCst|jt}|d||S)z9 Whether the hash value denotes an active entry.  A pointer to the start of the entries array. )rWr^r r r!rRsz_SetPayload.entriescCs|jS)zC A pointer to the start of the NRT-allocated area. )rXr^r r r!rsz_SetPayload.ptrcCs2t|j|j|}|jj|jt|j|d}|S)z) Get entry number *idx*. r) rgeprTrWrSrrr#rU)rYidxZ entry_ptrrOr 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. rBN) rjr,if_thenrDkeybranchrFrJloadr3store)irOZ entry_hasheqj Zbb_foundZ bb_not_foundrrZeqfn for_insertZ free_indexZfree_index_sentinelr:itemrYr r! check_entrys    z(_SetPayload._lookup..check_entryrBfound)#rSrTr1r]rUdtyper*r+operatorrsr-r.rr0ralloca_once_valueand_append_basic_block for_range LINEAR_PROBESrpaddrqro goto_blocklshrmulr3rDposition_at_endphiZIntTypeZ add_incomingtrue_bit false_bit)rYrwr:rvintp_tr]rzZtyctxr7r8oneZfiveZperturbrNbb_bodybb_endrxrrprtryr rur!_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. rkstartrMN)rSrTget_value_typerintprr0rr]rrrjrNr/r,rmrLrP) rYrrrrrsizeZ range_looprOis_usedloopr r r!_iterate/s    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. rrkZnext_entry_bodyZnext_entry_endN)rSrTrrrrr0r]r~rr|rerorrpr}rrqrjr/r,cbranchr) rYrrrzerorr]rrrNrrrOrr r r! _next_entryCs*            z_SetPayload._next_entry)F)N)__name__ __module__ __qualname__rZpropertyr]setterrarcrergrRrrjr contextlibcontextmanagerrrr r r r!rQms>              v rQc@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|||_|||||_dSr[)rSrTrUr% _entrysize make_helper_set)rYrrrset_valr r r!rZns  zSetInstance.__init__cCs|jjSr[)rUrzr^r r r!rzuszSetInstance.dtypecCs.|j}|j}|jj||j}t|||j|S)z/ The _SetPayload for this set. )rSrTnrt meminfo_datameminforQrU)rYrrrr r r!ryszSetInstance.payloadcCs |jSr[)r _getvaluer^r r r!r5szSetInstance.valuecCs|jjSr[)rrr^r r r!rszSetInstance.meminfocCs|jjSr[rparentr^r r r!rszSetInstance.parentcCs ||j_dSr[rr_r r r!rscCs|jjS)z< Return the number of elements in the size. )rrar^r r r!get_sizeszSetInstance.get_sizecCs |jjr|rtjntj|j_dSr[)rUZ reflectedrrrrrgrYvalr r r! set_dirtyszSetInstance.set_dirtyTc Cs|j}|j}|j}||_||||_|j} t| jd} | | | } |_|j t |||dd| |j | |_ W5QRX|r| | |ddS)NrkTZlikely)rSrTr, incref_valuernrarr0r1rrmrFrcupsizer) rYrrOrwr: do_resizerrold_hashrarr r r! _add_entrys   zSetInstance._add_entryc Cs|j}|j}|j||dd\}} ||} || || } | j} || _|rZ|||| _|j } t | j d}| | |} |_ |jt||| dd| |j||_W5QRX|r|| |dW5QRXdS)NTrvrkr)rSrTrnot_rmrjr,rrnrarr0r1rrFrcrr)rYrrwr:r do_increfrrryrr not_foundrOrrarr r r!_add_keys*     zSetInstance._add_keycCsft|jjt|_|r"||j|j}t|jd}|j ||}|_|rX| || ddS)NrkT) rr0r,r1rH decref_valuernrarTsubdownsizer)rYrrOr do_decrefrarr r r! _remove_entrys  zSetInstance._remove_entryc CsN|j}|j}|||\}}||||} ||| |W5QRX|Sr[)rSrTrrmrjr) rYrrwr:rrrryrrrOr r r! _remove_keys  zSetInstance._remove_keycCs8|j}|j}|j}t|||jj|}|||||dSr[)rSrTrr<rUrzr)rYrwrrrrr:r r r!rs zSetInstance.addcCs6|j}|j}|j}|||||}|||||dS)z`A version of .add for use inside functions following Python calling convention. N)rSrTr_pyapi_get_hash_valuer)rYpyapirwrrrrr: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|Sr[)rA)rr r r!wrappersz2SetInstance._pyapi_get_hash_value..wrapperFr) rUrzrrr signatureZ call_jit_codermretZget_null_object) rYrrrrwargtypesZrestyrr>r8Zis_errorretvalr r r!rs z!SetInstance._pyapi_get_hash_valuecCs8|j}|j}|j}t|||jj|}|||\}}|Sr[)rSrTrr<rUrzr)rYrwrrrr:ryrrr r r!containss zSetInstance.containscCs6|j}|j}|j}t|||jj|}||||}|Sr[)rSrTrr<rUrzr)rYrwrrrr:ryr r r!discards zSetInstance.discardc Csh|j}|j}||jj}t||}|j}|$}| |j ||j ||ddW5QRX| |S)NF)r) rSrTrrUrzrZ alloca_oncerrrqrnrrp)rYrrZltyrnrrOr r r!pop!s  zSetInstance.popcCs<|j}|j}|tj}t|t}||| ddS)NT) rSrTrrrrr0MINSIZE_replace_payloadr)rYrrrminsizer r r!clear0s    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. NrBTrF)zcannot copy setr)rSrTrrarcr1rUrDif_else _copy_payloadrmr call_convreturn_user_exc MemoryErrorchoose_alloc_size_allocate_payloadrrOrrnr,)rYrrrrarcotherZno_deleted_entriesZ if_no_deletedZ if_deletedoknentries other_payloadrrOr r r!copy9s:     $zSetInstance.copyc Cs|j}|j}|j}|j}|L}|j}||j|j\}} || ||j ||ddW5QRXW5QRX| |j dS)z9 In-place intersection with *other* set. FrN) rSrTrrrOrrnr,rmrrrra) rYrrrrrrrOry_r r r! intersect`s $zSetInstance.intersectc CsX|j}|j}|j}|j}|"}|j}|j||j|jddW5QRX||j dS)z7 In-place difference with *other* set. FrN) rSrTrrrOrrnr,rra)rYrrrrrrrOr r r! differencers  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)rSrTrrrOrnr,rrjrrrrra)rYrrrrrrnr:rryrrrOZ 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)NrK<=)rSrTrrr|rrrDrarqrrrOrrnr,rmrrPrp)rYrstrictrrrrcmp_opresZ if_smaller if_largerrrOryrr r r!issubsets(  0zSetInstance.issubsetc s|j}|j|j}|j}ttjfdd}d|j|j>\}}||||W5QRX||||W5QRXW5QRX S)Nc s^|L}|j}||j|j\}}|tj| W5QRXW5QRXdSr[) rrOrrnr,rmrqrrrP)ZsmallerZlargerrrOryrrrr r!checks   z%SetInstance.isdisjoint..check>) rSrTrrr|rrrDrarp)rYrrrrrr 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)NrB)rSrTrrr|rrrDrarrOrrnr,rmrrqrrPrp) rYrrrrrrZ if_same_sizerrrOryrr 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) rrrrr0r isinstanceintrr) clsrrrnitemsrrrYrr 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)rrmrrrr)rrrrrrrYr 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_nullrpyobjectrrincrefr5)rrrrrrYr 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. rk calcsize.body calcsize.end>=Fr)r1rr0rshlrr|r~rorrprDrmrqr)rrrrrrr min_entriesZsize_prrrZis_large_enoughZ next_sizer r r!rs$            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. rkrFrrrzKupsize to %zd items: current size = %zd, min entries = %zd, new size = %zd zcannot grow setN)rSrTr1rr0rrrr]rDrmrr|r~rorrprqrr DEBUG_ALLOCSprintf_resize)rYrrrrrtworrr need_resize new_size_prrnew_size is_too_smallr r r!r;s>             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. rkrrrrKFrrrrzMdownsize to %zd items: current size = %zd, min entries = %zd, new size = %zd zcannot shrink setN)rSrTr1rr0rrrr3rDrr]r}rmrr|r~rorrprrqrrrr)rYrrrrrrrrrmax_sizerrrrrrrr r r!risT                  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)rSrTrrmrrrrrrrOrrnr, _free_payloadr) rYrrerrmsgrrZ old_payloadrrrOr 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! TrFr)zcannot reallocate setN)rSrTrrrOrrnrrrrmrrrr)rYrrrrrOrr 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! rrkFrT)r.allocated %zd bytes for set at %p: mask = %zd ),rSrTrr|rrrrrr0rrrUr$rZmuladd_with_overflowrmrqrrprrrZmeminfo_varsize_alloc_uncheckedis_null _imp_dtormodule"meminfo_new_varsize_dtor_uncheckedr voidptr_trrrrrmemsetrrarcrerr]rr)rYrrrrrrrrr payload_size entry_size allocsizeZovfrralloc_okdtorif_errorif_okrZnew_maskr r r!rsp         "zSetInstance._allocate_payloadcCs|jj|j|j|dS)z9 Free an allocated old payload at *ptr*. N)rSrZmeminfo_varsize_freerTr)rYrr r r!rszSetInstance._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. rrkTrFr )-rSrTrr|rrrrrr0rrrUr$rr]rrrmrpr rrrrrr rrqrrrrrarcreZ raw_memcpyrRrrrOrnrrr)rYZ src_payloadrrrrrrrrrr]rrrrrrrrrr 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 FunctionTypeZVoidTyperUrzZget_or_insert_functionZis_declarationlinkageZ IRBuilderr~rQr>rrOrdecrefrnZret_void) rYrrZ llvoidptrZllsize_tr7fnamer9rrrrOr r r!r as"   zSetInstance._imp_dtorcCs|jj|j|jj|dS)z Incref an element value N)rSrrrTrUrzrr r r!r}szSetInstance.incref_valuecCs|jj|j|jj|dS)z Decref an element value N)rSrrrTrUrzrr r r!rszSetInstance.decref_value)T)TT)TT)T)T)T)F)N)N)F).rrrrZrrzrr5rrrrrrrrrrrrrrrrrrrrrrr classmethodrrrrrrrrrrrr rrr r r r!rlsf              '     .< CArc@sZeZdZddZeddZeddZeddZed d Z e j d d Z d d Z dS)SetIterInstancecCsL||_||_||_|||||_|jj||j}t|||jj ||_ dSr[) rSrTrUr_iterrrrrQ containerrV)rYrr iter_typeZiter_valrr r r!rZs zSetIterInstance.__init__cCsJt|||j|}||||d}|tjd}t|||j_|j |j_ |Sr=) rr! get_constantrrrr|r rNr)rrrr"rZset_instrYrNr r r!from_sets  zSetIterInstance.from_setcCs |jSr[)r rr^r r r!r5szSetIterInstance.valuecCs|jjSr[)r rr^r r r!rszSetIterInstance.meminfocCs|j|jjSr[)rTrpr rNr^r r r!rNszSetIterInstance.indexcCs|j||jjdSr[)rTrqr rNr_r r r!rNsc Csr|j}|j}t|jd}||j|d:}|j}|| |j |j |j||_| W5QRXdS)Nrkr)rNrVrr0r1Z set_exhaustedrrOZ set_validZyield_rnrTrrP)rYresultrNrrrrOr r r!iternexts zSetIterInstance.iternextN) rrrrZrr$rr5rrNrr&r r r r!rs     rc 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_arrayr|r#rrrrprhrNrr r5) rrritemsrinstarrayZ array_ptrcountrrwr r r! build_sets  r,cCs$|j}t|||}t||||jSr[) return_typerrr r5)rrr8r>rr)r r r!set_empty_constructorsr.c Cs||j}|j\}|\}t||||}t||||}t||||&} || j|j ||j | jW5QRXt ||||jSr[) r-r>r rrr rr5rrrzr ) rrr8r>r items_typer(nr)rr r r!set_constructors r1cCs t|||jd|d}|Sr=)rr>rrrr8r>r)r r r!set_lensr3cCs&t|||jd|d}||dSNrrk)rr>rr2r r r!in_setsr5ZgetitercCs(t|||j|d}t|||j|jSr=)rr$r-r r5r2r r r! getiter_setsr6r&cCs&t|||jd|d}||dSr=)rr>r&)rrr8r>r%r)r r r!iternext_listitersr7zset.addcCs2t|||jd|d}|d}|||Sr4)rr>rget_dummy_valuerrr8r>r)rwr r r!set_add s r:z set.discardcCs2t|||jd|d}|d}|||Sr4)rr>rr8r9r r r! set_discards r;zset.popc CsZt|||jd|d}|jj}|jt||dd|j|t dW5QRX| S)NrFr)zset.pop(): empty set) rr>rrarmrr rrKeyErrorr)rrr8r>r)rar r r!set_pops r=z set.removec Csbt|||jd|d}|d}||}|j||dd|j|tdW5QRX|S)NrrkFr)zset.remove(): key not in set) rr>rrmrrrr<r8)rrr8r>r)rwryr r r! set_remove's  r>z set.clearcCs(t|||jd|d}||Sr=)rr>rr8r2r r r! set_clear5sr?zset.copycCs2t|||jd|d}|}t|||j|jSr=)rr>rr r-r5rrr8r>r)rr r r!set_copy;srAzset.difference_updatecCsBt|||jd|d}t|||jd|d}|||Sr4)rr>rr8r@r r r!set_difference_updateAs rBzset.intersection_updatecCsBt|||jd|d}t|||jd|d}|||Sr4)rr>rr8r@r r r!set_intersection_updateJs rCzset.symmetric_difference_updatecCsBt|||jd|d}t|||jd|d}|||Sr4)rr>rr8r@r r r!set_symmetric_difference_updateSs rDz set.updatec Cst|||jd|d}|jd}|d}t||||}|dk rZ||jj|}||t||||:} ||| j |j |j } || |j ||j | j W5QRX|dk r| |jj|Sr4)rr>r rrrarr castr5rzrrrr8) rrr8r>r)r/r(r0rrZcastedr r r! set_update\s   rFcCs:|j|jdkst|||||t|||jd|dSr=)r-r>AssertionErrorr )rrr8r>op_implr r r! set_inplacesrIzset.differencecCsdd}|||||S)NcSs|}|||Sr[)rdifference_updateabsr r r!difference_impls z'set_difference..difference_implZcompile_internal)rrr8r>rOr r r!set_differencesrQzset.intersectioncCsdd}|||||S)NcSs@t|t|kr&|}|||S|}|||SdSr[)r'rintersection_updaterKr r r!intersection_impls  z+set_intersection..intersection_implrP)rrr8r>rSr r r!set_intersections rTzset.symmetric_differencecCsdd}|||||S)NcSs@t|t|kr&|}|||S|}|||SdSr[)r'rsymmetric_difference_updaterKr r r!symmetric_difference_impls  z;set_symmetric_difference..symmetric_difference_implrP)rrr8r>rVr r r!set_symmetric_differences  rWz set.unioncCsdd}|||||S)NcSs@t|t|kr&|}|||S|}|||SdSr[)r'rupdaterKr r r! union_impls  zset_union..union_implrP)rrr8r>rYr r r! set_unions rZzset.isdisjointcCs:t|||jd|d}t|||jd|d}||Sr4)rr>rr@r r r!set_isdisjointsr[z set.issubsetcCs:t|||jd|d}t|||jd|d}||Sr4rr>rr@r r r! set_issubsetsr]zset.issupersetcCsdd}|||||S)NcSs ||Sr[)rrLrMr r r! superset_implsz%set_issuperset..superset_implrP)rrr8r>r_r r r!set_issupersetsr`cCs:t|||jd|d}t|||jd|d}||Sr4)rr>rr@r r r!r[scCsdd}|||||S)NcSs ||k Sr[r r^r r r!ne_implszset_ne..ne_implrP)rrr8r>rar r r!set_nesrbcCs>t|||jd|d}t|||jd|d}|j|ddS)NrrkT)rr\r@r r r!set_ltsrccCsdd}|||||S)NcSs||kSr[r r^r r r!gt_implszset_gt..gt_implrP)rrr8r>rdr r r!set_gtsrecCs^t|||jd|d}t|||jd|d}||jtj}||jtj}|d||S)NrrkrB)rr>ZptrtointrrrZ icmp_signed)rrr8r>rLrMmambr r r!set_iss rhcCs|j|jkst|Sr[)rzrG)rrZfromtyZtotyrr r r! set_to_setsri)i__doc__ collectionsrmathr{ZllvmliterZ numba.corerrrZnumba.core.imputilsrrrr r r r r rZnumba.core.utilsrZ numba.miscrZ numba.cpythonrZnumba.extendingrZnumba.core.errorsrr"r%rCrHr2rrrr<rArFrJr/ namedtuplerLobjectrQrrr,setr.Z IterableTyper1r'Setr3rAnyr5r6ZSetIterZBORROWEDr7r:r;r=r>r?rArBrCrDrFiandiorisubixorZop_rHrIrrQr}rTxorrWor_rZr[ler]ger`rsnerbltrcgtreis_rhrir r r r!s ,         "6