1@jMl dZddlZddlZddlZddlZddlmZddlmZddl m Z m Z m Z ddl mZmZmZmZmZmZmZmZmZddlmZddlmZdd lmZmZdd lmZm Z m!Z!d Z"d Z#d Z$dZ%dZ&dZ'dZ(dZ)dZ*e!dZ+dZ,dZ-dZ.dZ/ej0ddZ1Gdde2Z3Gdde2Z4Gdd e2Z5d!Z6ee7d"Z8ee7e j9d#Z:ee;e j<d$Z=eej>e j<e j?d%Z@ed&e j<d'ZAed(e jBeejCd)ZDed*e j<e j?d+ZEe!d,ZFe e j<d-d.ZGe!d/ZHe e j<d0d1ZIe!d2ZJe e j<d3d4ZKe!d5ZLe e j<d6d7ZMe!d8ZNe e j<d9d:ZOd;ZPe!d<ZQe e j<d=d>ZRd?ZSe!d@ZTe e j<dAdBZUdCZVe!dDZWe e j<dEdFZXedGe j<e j9dHZYdIZZej[eSfej\eYfej]ePfej^eVffD]\Z_Z`eZe_e`eejae e j<dJdKZbeejce e j<dLdMZdeejee e j<dNdOZfeejge e j<dPdQZhe!dRZie e j<dSdTZje!dUZkeejle e j<dVdWZmeejne e j<dXdYZoe!dZZpeejqd[Zreejsd\Zte!d]Zueejvd^Zweejxd_Zyeejze j<e j<d`Z{ee j<e j<daZ|dS)bz& Support for native homogeneous sets. N)cached_property)ir)typestypingcgutils) lower_builtin lower_cast iternext_implimpl_ret_borrowedimpl_ret_new_refimpl_ret_untrackedfor_itercall_lenRefType) quicksort)slicing)NumbaValueError TypingError)overloadoverload_method intrinsicctj|}||}|||}||||S)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_type as_pointerbitcastmake_data_helper)contextbuilderset_typeptr payload_typeptrtypayloads EC:\Projects\Historical\venv\Lib\site-packages\numba/cpython/setobj.pyget_payload_structr(sa #H--L  ! !, / / : : < >%H - --r)czfd}|t}||fi}||fS)Nc4t|||dSNr)rG)r r!r>argsr?s r'implz'_get_hash_value_intrinsic..implLsgwtAw???r))r5r6r7)r@r?rLrArBs ` r'_get_hash_value_intrinsicrMJsW@@@@@  ' ' - -D   Y" 5 5C 9r)cntj|jt}|d||S)z8 Whether the hash value denotes an empty entry. ==)rr:r;EMPTY icmp_unsigned)r r!rDemptys r' is_hash_emptyrSSs/ K & &E  q% 0 00r)cntj|jt}|d||S)z9 Whether the hash value denotes a deleted entry. rOrr:r;DELETEDrQr r!rDdeleteds r'is_hash_deletedrYZs/k!&'**G  q' 2 22r)cntj|jt}|d||S)z9 Whether the hash value denotes an active entry. s r' z!check_all_set..ks$;;;s 3 **;;;r)z All arguments must be Sets, got c<g|]}dj|jkS)rdtype)rasrKs r'rbz!check_all_set..ns&777QQ (777r)z'All Sets must be of the same type, got )allr)rKs`r' check_all_setrhjs ;;d;;; < <ECTCCDDD 7777$777 8 8LJDJJKKKLLr)SetLoopindexentrydo_breakceZdZdZedZejdZedZejdZedZejdZedZ e jd Z ed Z e jd Z ed Z ed Z dZ ddZejddZejdZdS) _SetPayloadct||||}||_||_||_||_|d|_||_dS)Nentries)r(_context_builder_ty_payload_get_ptr_by_name_entries_ptr)selfr r!r"r#r&s r'__init__z_SetPayload.__init__wsQ$WgxEE   00;;  r)c|jjSNrumaskrys r'r~z_SetPayload.mask }!!r)c||j_dSr|r}ryr?s r'r~z_SetPayload.masks# r)c|jjSr|ruusedrs r'rz_SetPayload.usedrr)c||j_dSr|rrs r'rz_SetPayload.used" r)c|jjSr|rufillrs r'rz_SetPayload.fillrr)c||j_dSr|rrs r'rz_SetPayload.fillrr)c|jjSr|rufingerrs r'rz_SetPayload.fingers }##r)c||j_dSr|rrs r'rz_SetPayload.fingers$ r)c|jjSr|rudirtyrs r'rz_SetPayload.dirtys }""r)c||j_dSr|rrs r'rz_SetPayload.dirtys# r)c|jS)z> A pointer to the start of the entries array. )rwrs r'rqz_SetPayload.entriess }r)c|jS)zC A pointer to the start of the NRT-allocated area. )rxrs r'r#z_SetPayload.ptrs yr)ctj|j|j|}|j|jt j|j|}|S)z) Get entry number *idx*. r) rgeprsrwrrrrr+rt)ryidx entry_ptrrls r' get_entryz_SetPayload.get_entrysTK t}cBB  ..t}/4~dh/G/G3</>> r)Fc  jjj}j}jj}j}|tj }| |||fi}  || tj |d} tj |d} tj} tj|} r*|dtjd}ddd}f d}tjtj |t&5| }|||| }||}|| d d d n #1swxYwY||5| }||| }|| }| || }|||}|| || |d d d n #1swxYwY5rl| }}d |||}|| |d d d n #1swxYwY5|d d d n #1swxYwY|tjdd }| 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@ |}|j}d |5 |jf}|5dddn #1swxYwYdddn #1swxYwYt |5dddn #1swxYwY rt|5 } d| ||} | ddddS#1swxYwYdSdS)zO Check entry *i* against the value being searched for. rON) rr6if_thenrQkeybranchrSrYloadr=store)irl entry_hasheqjbb_found bb_not_foundr!r eqfn for_insert free_indexfree_index_sentinelrDitemrys r' check_entryz(_SetPayload._lookup..check_entrysNN1%%EJ!6!6tQ !K!KLL - -T'D%)#455__R((--NN8,,,---------------  - - - - - - - - - - - - - - -w!L!LMM - -|,,, - - - - - - - - - - - - - - - 1___Wgz%R%RSS11 Z00Aw'<'>MMBO..O25O2P00P47P4Nc#K|j}|j}|tj}t j|d}||j|}tj |||5}| |j }t|||j} || 5t!|j ||j} | Vdddn #1swxYwYddddS#1swxYwYdS)zG Iterate over the payload's entries. Yield a SetLoop. rstartrjN)rrrsget_value_typerintprr:rr~rrrrkr9r6rrirm) ryrr r!rrsize range_looprlis_usedloops r'_iteratez_SetPayload._iterate7s --'' 33k&!$${{49c**  wE : : : jNN:#344E"7GUZ@@G))  Z%5U(2(;===                                  s77AD=!C* D*C. .D1C. 2DD Dc#K|j}|j}|tj}t j|d}t j|d}|j}|d}|d}tj ||j } | || |5|| } |||| |} || | || } t'||| j} || ||dddn #1swxYwY|||| } | |_ || 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)rrrsrrrrr:r~rrrrrrrrrrrr9r6cbranchr) ryr r!rzerorr~rrrkrrlrs r' _next_entryz_SetPayload._next_entryKs --'' 33{61%%k&!$$y ,,->??++,<==)'4;??w    ( ( 6 6 U##A T7;;q##6#677A MM!U # # #NN1%%E"7GUZ@@G OOGVW 5 5 5 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 ''' LL   nnQsBE33E7:E7Fr|)__name__ __module__ __qualname__rzpropertyr~setterrrrrrqr#rr contextlibcontextmanagerrrr^r)r'rorous""X" [##[#""X" [##[#""X" [##[#$$X$ ]%%]%##X# \$$\$X X t*t*t*t*l&% % % % % r)roceZdZdZedZedZedZedZedZ e j dZ dZ d Z d-d Z d.d Zd.d Zd-dZd-dZd-dZdZdZdZdZdZdZdZdZdZd/dZdZdZed0dZ ed0d Z!ed!Z"ed"Z#d#Z$d$Z%d%Z&d&Z'd/d'Z(d(Z)d)Z*d*Z+d+Z,d,Z-dS)1 SetInstancec||_||_||_t|||_|||||_dSr|)rrrsrtr. _entrysize make_helper_set)ryr r!r"set_vals r'rzzSetInstance.__init__vsE  ((;;''7CC r)c|jjSr|)rtrers r'rezSetInstance.dtype}s x~r)c|j}|j}|jj||j}t |||j|S)z/ The _SetPayload for this set. )rrrsnrt meminfo_datameminforort)ryr r!r#s r'r&zSetInstance.payloadsD --m,,WdlCC7GTXs;;;r)c4|jSr|)r _getvaluers r'r?zSetInstance.valuesy""$$$r)c|jjSr|)rrrs r'rzSetInstance.meminfos y  r)c|jjSr|rparentrs r'rzSetInstance.parents yr)c||j_dSr|rrs r'rzSetInstance.parents  r)c|jjS)z< Return the number of elements in the size. )r&rrs r'get_sizezSetInstance.get_sizes|  r)cj|jjr&|r tjn tj|j_dSdSr|)rt reflectedrrrr&rryvals r' set_dirtyzSetInstance.set_dirtys< 8  P58!O!1!1g>ODL    P Pr)Tc|j}|j}|j}||_||||_|j} t j| jd} | | | x} |_| t|||d5| |j | |_ dddn #1swxYwY|r| | |ddS)NrTlikely)rrrsr6 incref_valuerrrr:r;rrrSrupsizer) ryr&rlrrD do_resizer r!old_hashrrs r' _add_entryzSetInstance._add_entrys7--:  $ |k$)Q''%kk$444w| __]7GXFF$(** : :";;w|S99GL : : : : : : : : : : : : : : :   KK    ts!C  CCc|j}|j}|||d\}} ||} || 5|| } | j} || _|r|||| _|j } tj | j d}| | |x} |_ |t||| d5| |j||_dddn #1swxYwY|r|| |dddddS#1swxYwYdS)NTrrr)rrrsrnot_rrr6rrrrr:r;rrSrrr)ryr&rrDr do_increfr r!rr not_foundrlrrrs r'_add_keyzSetInstance._add_keys--??4t?<<qLL'' __Y ' ' ! !%%a((EzHEJ (!!$'''EI >&{{7<==  > > > > > > > > > > > > > > > " D!!! NN4 ' ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !s7B&E+:!D' E+'D+ +E+.D+ //E++E/2E/cntj|jjt|_|r||j|j}tj|jd}|j ||x}|_|r| || ddS)NrT) rr:r6r;rV decref_valuerrrssubdownsizer)ryr&rlr do_decrefrrs r' _remove_entryzSetInstance._remove_entrys['::  )   ei ( ( (|k$)Q''"m//c:::w|  MM$    tr)c|j}|j}|||\}}||5||} ||| |dddn #1swxYwY|Sr|)rrrsrrrr) ryr&rrDrr r!rrrls r' _remove_keyzSetInstance._remove_keys--??4++q __U # # : :%%a((E   wy 9 9 9 : : : : : : : : : : : : : : : s-A66A:=A:c|j}|j}|j}t|||jj|}|||||dSr|)rrrsr&rGrtrer )ryrrr r!r&rDs r'rzSetInstance.addsL--, 7GTX^T B B gtQ 22222r)c|j}|j}|j}|||||}|||||dS)z`A version of .add for use inside functions following Python calling convention. N)rrrsr&_pyapi_get_hash_valuer )rypyapirrr r!r&rDs r' add_pyapizSetInstance.add_pyapisP--,  & &ugw E E gtQ 22222r)cH|jjg}tj}d}|g}t j|g|R} ||| |\} } || d5|| dddn #1swxYwY| S)z=Python API compatible version of `get_hash_value()`. c t|Sr|)rM)rs r'wrapperz2SetInstance._pyapi_get_hash_value..wrapper s,S11 1r)FrN) rtrerrr signature call_jit_coderretget_null_object) ryrr r!rargtypesrestyrrKrBis_errorretvals r'rz!SetInstance._pyapi_get_hash_valuesHN#  2 2 2vu0x000 ..wTBB& __Xe_ 4 4 1 1 KK--// 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 s#(BBBc|j}|j}|j}t|||jj|}|||\}}|Sr|)rrrsr&rGrtrer)ryrr r!r&rDrrs r'containszSetInstance.containssJ--, 7GTX^T B B??4++q r)c|j}|j}|j}t|||jj|}||||}|Sr|)rrrsr&rGrtrer)ryrr r!r&rDrs r'discardzSetInstance.discard sJ--, 7GTX^T B B  $22 r)c||j}|j}||jj}t j||}|j}|5}| |j || ||ddddn #1swxYwY| |S)NF)r) rrrsrrtrer alloca_oncer&rrrrr)ryr r!ltyrr&rls r'popzSetInstance.pop)s--$$TX^44!'3//,  " " @e MM%)S ) ) )   w  ? ? ? @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ ||C   s4BB"%B"c|j}|j}|tj}t j|t}||| ddS)NT) rrrsrrrrr:MINSIZE_replace_payloadr)ryr r!rminsizes r'clearzSetInstance.clear8sb--'' 33+fg.. g&&& tr)c N|j}|j}|j}|j}|j}t ||||jd}|d||}||d5\}} |5| |} | | | d5|j |tddddn #1swxYwYdddn #1swxYwY| 5||||} || } | | | d5|j |tddddn #1swxYwY|j} |5} | j}|| |j|jddddn #1swxYwYdddn #1swxYwYdddn #1swxYwY|S)z, Return a copy of this set. NrOTrF)zcannot copy setr)rrrsr&rrr;rtrQif_else _copy_payloadrr call_convreturn_user_exc MemoryErrorchoose_alloc_size_allocate_payloadrrlr rr6)ryr r!r&rrotherno_deleted_entries if_no_deleted if_deletedoknentries other_payloadrrls r'copyzSetInstance.copyAs--,||T 7GTXt<<$224tDD __/_ = = 4* z L L((11__W\\"%5%5e_DDLL%55g{6JLLLLLLLLLLLLLLLLLL L L L L L L L L L L L L L L L 4 411'7DII,,X66__W\\"%5%5e_DDLL%55g{6JLLLLLLLLLLLLLLLLLL!& %%''44 JENN=%)UZ-2#444444444444444444 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4. s2H8AC=8"C& C=&C* *C=-C* .C=1 H=D HD H AH!"F HF HF H5+G, H,G0 0H3G0 4H7 HH H H HH!Hc|j}|j}|j}|j}|5}|j}||j|j\}} || |5| ||ddddn #1swxYwYdddn #1swxYwY| |j dS)z9 In-place intersection with *other* set. Fr0N) rrrsr&rrlrrr6rrrr r) ryr8r r!r&r>rrlr_s r' intersectzSetInstance.intersecthss--,      D4JE$,,UY CCHE1e!4!455 D D""7EU"CCC D D D D D D D D D D D D D D D D D D D D D D D D D D D D D D gl#####s6ACB) C)B- -C0B- 1CCCc|j}|j}|j}|j}|5}|j}|||j|jddddn #1swxYwY||j dS)z7 In-place difference with *other* set. Fr0N) rrrsr&rrlrrr6r r)ryr8r r!r&r>rrls r' differencezSetInstance.differencezs--,  # # % % NJE   Weiu  M M M N N N N N N N N N N N N N N N gl#####s+A((A,/A,c |j}|j}|j}|5}|jj}|jj}|j}|||d\} } || } | | 5\} } | 5| || ddddn #1swxYwY| 5| || ||dddn #1swxYwYdddn #1swxYwYdddn #1swxYwY| |jj dS)zA In-place symmetric difference with *other* set. TrFr0N)rrrsr&rrlrr6rrr1rrr r)ryr8r r!r>rrrDr&rrrl if_common if_not_commons r'symmetric_differencez SetInstance.symmetric_differencesF--  # # % % <*.C AlGsA$??HE1%%a((E'' <+EI}HH&&w&GGGHHHHHHHHHHHHHHH"<<OOGUC;;;<<<<<<<<<<<<<<< < < < < < < < < < < < < < < < < < < < < < < < < < < < < < < dl'(((((s~A%DDB:. D:B> >DB> DC, D,C0 0D3C0 4D7 DD D D DD!DFc |j}|j}|j}|j}|rdnd}tj|tj}||||j|j5\} } | 5| tj |dddn #1swxYwY| 5| 5} | j } | | j| j\} }||| 5| tj || dddn #1swxYwYdddn #1swxYwYdddn #1swxYwYdddn #1swxYwY||S)Nr[<=)rrrsr&rrrr1rQrrrrrlrrr6rrrmr)ryr8strictr r!r&r>cmp_opres if_smaller if_largerrrlrrAs r'issubsetzSetInstance.issubsets--, (D'1ABB __  ! !&', 8J K K ((*i 6 6 g/555 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 ( (%%''(4 JE,44UY KKHE1 e)<)<==(( g&7=== (((((((((((((((((((((((((((((( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (||C   s6F$F  F$ F F$F F$$F(+F(c |j}|j|j}|j}tjtj  fd}d|j|j5\}}|5|||dddn #1swxYwY|5|||dddn #1swxYwYdddn #1swxYwY S)Ncx|5}|j}||j|j\}}|5tj| dddn #1swxYwYddddS#1swxYwYdSr|) rrlrrr6rrrrrm)smallerlargerrrlrrAr!rMs r'checkz%SetInstance.isdisjoint..checks#!!## $t !>>%)UZ@@q__U++$$MM'"3S999MMOOO$$$$$$$$$$$$$$$ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $s6AB/5B B/B B/B B//B36B3>) rrrsr&rrrr1rQrr) ryr8r r&r>rUrO otherwiser!rMs @@r' isdisjointzSetInstance.isdisjoints--, '1ABB $ $ $ $ $ $__  ! !#w|]5G H H .')Y . .mW--- . . . . . . . . . . . . . . . . .g}--- . . . . . . . . . . . . . . . . . . . . . . . . . . . . . .||C   sZ8C> B CB CB C$ B=1 C=C CC CCCc ||j}|j}|j}|j}tj|tj}||d|j|j5\}}|5| 5} | j } | | j | j \} } ||| 5|tj|| dddn #1swxYwYdddn #1swxYwYdddn #1swxYwY|5|tj|dddn #1swxYwYdddn #1swxYwY||S)NrO)rrrsr&rrrr1rQrrrlrrr6rrrrrmr) ryr8r r!r&r>rM if_same_sizerWrrlrrAs r'equalszSetInstance.equalss--, '1ABB __  ! !$ m6H I I 6*,  ( (%%''(4 JE,44UY KKHE1 e)<)<==(( g&7=== (((((((((((((((((((((((((((((( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( 6 6 g/555 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6||C   s0F6E  AD65D  D6D# #D6&D# 'D6* E 6D: :E =D: >E  F E FE F!F; FF FF FF"%F"NcJ|tj}|tj|t }nAt |trtj||}||||}||||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) rrrrr:r+r_intr6r7) clsr r!r"nitemsrr=ryr<s r' allocate_exzSetInstance.allocate_exs'' 33 >{6733HH&#&& 5VV44,,WgvFFHs7GXt44  # #H - -4xr)c|||||\}}|||d5|j|t ddddn #1swxYwY|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`rrr3r4r5)r^r r!r"r_r<rys r'allocatezSetInstance.allocates??7GXvFFD __W\\"--e_ < < H H   - -g{.F H H H H H H H H H H H H H H H H H H s"A44A8;A8c||||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^r r!r"rrys r' from_meminfozSetInstance.from_meminfos^s7GXt44# "44U^DD  7Hdj999 r)cR|j}tj|d}tj|t}|||}t j||}|d} |d} || | | 5| |} | d| |} | | d5|| dddn #1swxYwY|| |} | | ||| dddn #1swxYwY|| | |S)zT Choose a suitable number of entries for the given number of items. r calcsize.body calcsize.end>=FrN)r;rr:r+shlrrrrrrrQrrr)r^r r!r_rrr- min_entriessize_prrris_large_enough next_sizes r'r6zSetInstance.choose_alloc_size$s k&!$$+fg..kk&#.. *7G<<,,_==++N;;w    ( ( $ $<<''D%33D$ LLO?? ' 'v&&& ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' D#..I MM)V , , , NN7 # # # $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ '''||F###s86AE4:D E4D E4#D $AE44E8;E8c &|j}|j}|j}tj|d}tj|d}|j}|||}||j|} | d|| } | | d5tj || } | d} | d} || || 5|| }|||}||| | d||}||| | dddn #1swxYwY|| || }t(r||d || |||||d ddddS#1swxYwYdS) zs When adding to the set, ensure it is properly sized for the given number of used entries. rrkFrrirjNzKupsize to %zd items: current size = %zd, min entries = %zd, new size = %zd zcannot grow set)rrrsr;rr:r&rlrr~rQrrrrrrrrrr DEBUG_ALLOCSprintf_resize)ryr_r r!rrtwor&rmr need_resize new_size_prrnew_size is_too_smalls r'rzSetInstance.upsizeCs --k&!$$k&!$$,kk&#.. {{7<--++D+tDD __[_ 7 7 ? ? 27DAAJ00AAG//??F NN7 # # ###G,, ? ?"<< 33";;x55 h 333&44T;QQ  gv>>>  ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?  # #F + + +||J//H Dw E%t[(DDD LL(,= > > >3 ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?s9&A*HA0F  H F HF A%HH  H c |j}|j}|j}tj|d}tj|d}tj|t }|j}|||} || d| || |} || |} | |j |} | | d| | | d|| } | | d5tj|| } |d}|d }||||5|| }|||}| d | |}| |5||d d d n #1swxYwY||| ||d d d n #1swxYwY|||| }t.r||d || | ||||d d d d d S#1swxYwYd S)zw When removing from the set, ensure it is properly sized for the given number of used entries. rrrrkrJr[FrrirjrVNzMdownsize to %zd items: current size = %zd, min entries = %zd, new size = %zd zcannot shrink set)rrrsr;rr:r+r&rlr=rQrr~rrrrrrrrrrrrsrtru)ryr_r r!rrrvr-r&rmmax_sizerrwrxrrryrzs r'r zSetInstance.downsizeqs --k&!$$k&!$$+fg..,kk&#.. nnW%:%:4g%V%V%0';; ;;{C00{{7<--ll  ! !$$ 7 7  ! !#w 5 577 __[_ 7 7 A A 27DAAJ00AAG//??F NN7 # # ###G,, ( ("<< 33"<<#66&44S+xPP __\22++NN6***+++++++++++++++ h 333w''' ( ( ( ( ( ( ( ( ( ( ( ( ( ( (  # #F + + +||J//H Dw E%t[(DDD LL(,? @ @ @A A A A A A A A A A A A A A A A A A As]+A*K AI-H IH IH .I K I K I A%K  KKc |j}|j}|}||d}|||d5|j|t|fdddn #1swxYwY|j}| 5}|j } | || j | j dddddn #1swxYwY||jdS)zw Resize the payload to the given number of entries. CAUTION: *nentries* must be a power of 2! TreallocFrN)rr)rrrsr7rrr3r4r5r&rrlr rr6 _free_payloadr#) ryr&r=errmsgr r! old_payloadr<rrls r'ruzSetInstance._resizes --  # #Hd # ; ; __W\\"--e_ < < 9 9   - -g{/5i 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 ,  ! ! # # >!+HHYg>O(P(PRR"?7G<<(-!// I2C8U99MM'"3R888999999999999999II"U,3 )+2+D+DU^+T+T ("lGN7GKDIII#'GL#'GL%)GN&{{8S99H#+GL#Iw'X'0'+xIIIIIIIIIIIIIIIIII  I I I I I I I I I I I I I I I" I" I" I" I" I" I" I" I" I" I" I" I" I" I" IH||Bs!EE E =CM,M!I3' M3I7 7M:I7 ;MB2L>2 M>M MM M M,M M,M M,,M03M0c\|jj|j|j|dS)z9 Free an allocated old payload at *ptr*. N)rrrmeminfo_varsize_freersr)ryr#s r'rzSetInstance._free_payload"s+ ..t}dlCPPPPPr)c |j}|j}tj|tj}|t j}tj |d}tj |d}| t j |j }| |} |j} | | z} |j} ||| } |tj || |tj || | } |||d5|||j}|j|| ||tj}tj||}||d5\}}|5|tj|dddn #1swxYwY|5||j_|j }|j!|_!|j"|_"||_#| |_tj$||j%|j%| | |&5}|'|j(j)dddn #1swxYwYtTr|+|d| |j,| dddn #1swxYwYdddn #1swxYwYdddn #1swxYwY||S)z7 Raw-copy the given payload into self. rrTrFNr)-rrrsrrrrrrrr:rrrtr,rr~rrrrrrrrrrrr1rrrrr&rrr raw_memcpyrqrrrlrrsrtr#)ry src_payloadr r!r<rrrr$rrr~r=rrrrrrr&rs r'r2zSetInstance._copy_payload(s^--  &w0@ A A'' 33{61%%k&!$$,,U-=dh-G-GHH --l;; _  " ;;sD)) KK FL A A ' BK ,K,K,4!6!677 __W\\"--d_ ; ; E E>>'7>::DkDDGOOD':K$L$LNNGw88H%88 E!!!0[(   [   [ $$[$<,?,?,?\:A:A:Ax,,,6JJJ.A A A A FQQQ ? ? ? B8EEE EEEEEr)rceZdZdZedZedZedZedZ e j dZ dZ dS) SetIterInstancec||_||_||_|||||_|jj||j}t|||jj ||_ dSr|) rrrsrtr_iterrrrro containerru)ryr r! iter_typeiter_valr#s r'rzzSetIterInstance.__init__sj  (()XFF m,,WdlCC#GWdh6H#NN r)ct|||j|}||||d}|tjd}t j|||j_|j |j_ |SrJ) rr get_constantrrrrrrkr)r^r r!rrset_instryrks r'from_setzSetIterInstance.from_setsmw1DgNNs7GY55$$UZ33"4WeDD %-  r)c4|jSr|)rrrs r'r?zSetIterInstance.valuesz##%%%r)c|jjSr|)rrrs r'rzSetIterInstance.meminfos z!!r)cJ|j|jjSr|)rsrrrkrs r'rkzSetIterInstance.indexs}!!$*"2333r)cP|j||jjdSr|)rsrrrkrs r'rkzSetIterInstance.indexs% E4:#344444r)c|j}|j}tj|jd}|||5}|j}|| |j |j |j||_| ddddS#1swxYwYdS)Nrr)rkrurr:r; set_exhaustedrrl set_validyield_rrsrrm)ryresultrkr&rrrls r'iternextzSetIterInstance.iternexts -k%*a((   E  * * dJE       MM%) $ $ $**4:s;;DJ MMOOO                   sA/CCCN) rrrrzrrrr?rrkrrr^r)r'rrsOOO[&&X&""X"44X4 \55\5     r)rc  t|}t||||}|dkrtj||}tj||}|tj|}tj ||5} | tj ||d| j } | | dddn #1swxYwYt||||jS)zD Build a set of the given type, containing the given items. rN)lenrrbr pack_arrayrrrrrrrrkrr r?) r r!r"itemsr_instarray array_ptrcountrrs r' build_setrsZZF   (F C CD zz"7E22-gu== $$UZ88  w . . $<< GY4: N NOOD HHTNNN                GWh C CCsAC##C'*C'cv|j}t|||}t||||jSr|) return_typerrbr r?)r r!rBrKr"rs r'set_empty_constructorrs6H   ( ; ;D GWh C CCr)c|j}|j\}|\}t||||}t||||}t ||||5} || j|j ||j | jdddn #1swxYwYt||||jSr|) rrKrrrbrrr?rrrer ) r r!rBrKr" items_typernrrs r'set_constructorrsH(KJ FE ':u55A   (A > >D '7J 6 6@$  7HNDJ???@@@@@@@@@@@@@@@ GWh C CCsAB##B'*B'cpt|||jd|d}|SrJ)rrKrr r!rBrKrs r'set_lenrs- w!d1g > >D ==??r)c~t|||jd|d}||dSNrr)rrKr#rs r'in_setrs5 w!d1g > >D ==a ! !!r)getiterct|||j|d}t|||j|jSrJ)rrrr r?rs r' getiter_setrs9  # #GWcotAw O OD Wgs K KKr)rcvt|||jd|d}||dSrJ)rrKr)r r!rBrKrrs r'iternext_listiterr s9 7GSXa[$q' B BDMM&r)zset.addct|||jd|d}|d}|||Sr)rrKrget_dummy_valuer r!rBrKrrs r'set_addrsI w!d1g > >D 7DHHTNNN  " " $ $$r)c:tj||}d}||fS)Nct|||jd|d}|d}|||Sr)rrKr%rrs r' set_discardz!_set_discard..set_discard"sL7GSXa[$q'BBAw T&&(((r)rnone)r@rfrrBrs r' _set_discardrs/ *Q  C)))  r)r%cdS)Nc"t||Sr|)rrfrs r'z ol_set_discard...s<400r)r^rs r'ol_set_discardr,s 0 00r)c:||}d}||fS)NcNt|||jd|d}|jj}|t j||d5|j|tddddn #1swxYwY| S)NrFr)zset.pop(): empty set) rrKr&rrrrr3r4KeyErrorr))r r!rBrKrrs r'set_popz_set_pop..set_pop5s7GSXa[$q'BB|  __W_Wd;;E_ J J G G   - -gx,E G G G G G G G G G G G G G G G G G Gxxzzs"BB B rd)r@rfrBrs r'_set_popr1s, ''!**C <r)r)cdS)Nc t|Sr|)rrfs r'rzol_set_pop..Cs Xa[[r)r^rs r' ol_set_poprAs  r)c:tj||}d}||fS)Ncpt|||jd|d}|d}||}|||d5|j|tddddn #1swxYwY|S)NrrFr)zset.remove(): key not in set) rrKr%rrr3r4rr)r r!rBrKrrrs r' set_removez_set_remove..set_removeJs7GSXa[$q'BBAw T"" __W\\%00_ ? ? O O   - -gx,M O O O O O O O O O O O O O O O O O O&&(((s+"BB Br)r@rfrrBrs r' _set_removerFs. *Q  C)))  ?r)removec"|j|krdSdS)Nc"t||Sr|)rrs r'rzol_set_remove..Zs{1d33r)rdrs r' ol_set_removerWsw$333r)c8tj|}d}||fS)Nct|||jd|d}||SrJ)rrKr.rrs r' set_clearz_set_clear..set_clearcs>7GSXa[$q'BB &&(((r)r)r@rfrBrs r' _set_clearr_s* *Q--C)))  >r)r.cdS)Nc t|Sr|)rrs r'rzol_set_clear..ms Z]]r)r^rs r' ol_set_clearrks " ""r)c&||}d}||fS)Nct|||jd|d}|}t|||j|jSrJ)rrKr?r rr?r r!rBrKrr8s r'set_copyz_set_copy..set_copytsD7GSXa[$q'BB #/5;OOOr)r^)r@rfrBr s r' _set_copyr ps- !A$$CPPP =r)r?cdS)Nc t|Sr|)r rs r'rzol_set_copy..~s Yq\\r)r^rs r' ol_set_copyr|s ! !!r)ct|||jd|d}t|||jd|d}|||Sr)rrKrDrr s r'set_difference_updaters` w!d1g > >D #(1+tAw ? ?EOOE  " " $ $$r)c>tj||}|tfSr|)rrrr@abrBs r'_set_difference_updaters *Q  C % %%r)difference_updatec(t||dS)Nc"t||Sr|)rrrs r'rz,set_difference_update_impl..s.q!44r)rhrs r'set_difference_update_implrs!Q 4 44r)ct|||jd|d}t|||jd|d}|||Sr)rrKrBrr s r'set_intersection_updaters` w!d1g > >D #(1+tAw ? ?ENN5  " " $ $$r)c>tj||}|tfSr|)rrrrs r'_set_intersection_updaters *Q  C ' ''r)intersection_updatec(t||dS)Nc"t||Sr|)rrs r'rz.set_intersection_update_impl..s0A66r)rrs r'set_intersection_update_implr#s!Q 6 66r)ct|||jd|d}t|||jd|d}|||Sr)rrKrHrr s r'set_symmetric_difference_updater%sb w!d1g > >D #(1+tAw ? ?Ee$$$  " " $ $$r)c>tj||}|tfSr|)rrr%rs r' _set_symmetric_difference_updater's *Q  C / //r)symmetric_difference_updatec(t||dS)Nc"t||Sr|)r'rs r'rz6set_symmetric_difference_update_impl..s8A>>r)rrs r'$set_symmetric_difference_update_implr+s!Q > >>r)z set.updatect|||jd|d}|jd}|d}t||||}|5||jj|}||t||||5} ||| j |j |j } || |j ||j | j dddn #1swxYwY|| |jj|Sr)rrKrrr&rrrcastr?rerrr r) r r!rBrKrrrrryrcasteds r' set_updater/sp w!d1g > >D!J GE ':u55A};;t|0!44 H '7J 6 6 B$gtz:3CTZPP   7J$4djAAA B B B B B B B B B B B B B B B } dl'(((  " " $ $$sA#DD Dcbtfdt|fd}dS)Nc.|||}fd}||fS)Nc|j|jdksJ||||t|||jd|dSrJ)rrKr )r r!rBrKrLs r'codegenz>gen_operator_impl.._set_operator_intr..codegensP?chqk1111 D'3 - - -$Wgsx{DGLL Lr)r^)r@rrrBr3rLs r'_set_operator_intrz-gen_operator_impl.._set_operator_intrs<a1gg M M M M MG|r)c.t||fdS)Nc||Sr|r^rrr4s r'rz=gen_operator_impl.._ol_set_operator..s..q!44r)rr7s r'_ol_set_operatorz+gen_operator_impl.._ol_set_operators"a44444r))rr)oprLr8r4s ` @r'gen_operator_implr:s\Yb\\5555\555r)rDc,t||d}|S)NcX|}|||Sr|)r?rrrrfs r'difference_implz,impl_set_difference..difference_impls( FFHH Ar)r)rrr>s r'impl_set_differencer?s+!Q r) intersectionc,t||d}|S)Nct|t|kr+|}|||S|}|||Sr|)rr?r r=s r'intersection_implz+set_intersection..intersection_implsc q66CFF??A ! !! $ $ $HA ! !! $ $ $Hr)r)rrrCs r'set_intersectionrDs+!Q r)rHc,t||d}|S)Nct|t|kr+|}|||S|}|||Sr|)rr?r(r=s r'symmetric_difference_implz;set_symmetric_difference..symmetric_difference_implsc q66CFF??A ) )! , , ,HA ) )! , , ,Hr)r)rrrGs r'set_symmetric_differencerHs+!Q %$r)unionc,t||d}|S)Nct|t|kr+|}|||S|}|||Sr|)rr?updater=s r' union_implzset_union..union_impl(sY q66CFF??A HHQKKKHA HHQKKKHr)r)rrrMs r' set_unionrN#s+!Q r)c:tj||}d}||fS)Nct|||jd|d}t|||jd|d}||Sr)rrKrXr s r'r3z _set_isdisjoint..codegen;sN7GSXa[$q'BBGWchqk47CCu%%%r)rbooleanr@rrrBr3s r'_set_isdisjointrT7s. -1  C&&& <r)rXc(t||dS)Nc"t||Sr|)rTrs r'rz set_isdisjoint..Hs1--r)rrs r'set_isdisjointrWDs!Q - --r)c:tj||}d}||fS)Nct|||jd|d}t|||jd|d}||SrrrKrPr s r'r3z_set_issubset..codegenOsN7GSXa[$q'BBGWchqk47CC}}U###r)rQrSs r' _set_issubsetr[Ks. -1  C$$$ <r)rPc(t||dS)Nc"t||Sr|)r[rs r'rzset_issubset..\s a++r)rrs r' set_issubsetr^Ws!Q + ++r) issupersetc,t||d}|S)Nc,||Sr|)rPrs r' superset_implz%set_issuperset..superset_impldszz!}}r)r)rrrbs r'set_issupersetrc_s+!Q r)c:tj||}d}||fS)Nct|||jd|d}t|||jd|d}||Sr)rrKr[r s r'r3z_set_eq..codegenmsN7GSXa[$q'BBGWchqk47CC{{5!!!r)rQrSs r'_set_eqrfis. -1  C""" <r)c(t||dS)Nc"t||Sr|)rfrs r'rzset_eq..y1 r)rrs r'set_eqrju!Q % %%r)c,t||d}|S)Nc||k Sr|r^rs r'ne_implzset_ne..ne_impls6zr)r)rrrns r'set_nero{s(!Q Nr)c:tj||}d}||fS)Nct|||jd|d}t|||jd|d}||dS)NrrT)rKrZr s r'r3z_set_lt..codegensR7GSXa[$q'BBGWchqk47CC}}U4}000r)rQrSs r'_set_ltrrs. -1  C111 <r)c(t||dS)Nc"t||Sr|)rrrs r'rzset_lt..rir)rrs r'set_ltrurkr)c,t||d}|S)Nc||kSr|r^rs r'gt_implzset_gt..gt_impls 1u r)r)rrrxs r'set_gtrys(!Q Nr)cPt|||jd|d}t|||jd|d}||jtj}||jtj}|d||S)NrrrO)rrKptrtointrrr icmp_signed)r r!rBrKrrmambs r'set_isrsGWchqk47;;AGWchqk47;;A   !)W^ 4 4B   !)W^ 4 4B   tR , ,,r)c*|j|jksJ|Sr|rd)r r!fromtytotyrs r' set_to_setrs <4: % % % % Jr))}__doc__ collectionsrmathr functoolsrllvmliter numba.corerrrnumba.core.imputilsrr r r r r rrr numba.miscr numba.cpythonrnumba.core.errorsrrnumba.core.extendingrrrr(r.rPrVr<r+rrsrGrMrSrYr9rh namedtupleriobjectrorrrsetr IterableTyperrr`rr#AnyrrSetIterBORROWEDrrrrrrrrrrr rrrrrrr#r%r'r+r/r:iandiorisubixorop_op_implr r?rrDxorrHor_rNrTrWr[ler^gercrfrrjnerorrltrugtryis_rrr^r)r'rs/   %%%%%%----------AAAAAAAAAAAAAAAAAAAAAA! !!!!!!::::::::EEEEEEEEEEHHH(((       . . .  111333222LLL !+ ,J K K| | | | | &| | | ~Y EY EY EY EY E&Y EY EY Ex00000f000lDDD,sDDD sE&''DD('D.sEIx %)UY77""87"y%)$$LL%$Lz5=))w  ! *)y%)UY//%%0/%     I&&11'&1     E""!!#"!      H%%44&%4  G$$##%$#  F##""$#"%%% && & /0055105 %%% (( ( 12277327 %%% 00 0 9::??;:? |UY(:;;%%<;%: 5 5 5"]+, \: ])* ]34 $$LC c7####  (,L))*) (-N++  ,+  (,233 % %43 % (,G$$  %$ $     L))..*).       (+J'',,(',  (+L))*)      (+&&&  (+      (+&&&  (+x|UY 22--32- EIuy!!"!r)