o 8ήc#m @sdZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddlmZmZmZmZddlmZddlmZmZddlmZmZdd lmZmZmZmZejd kZ ejd kZ!ej"j#Z$e%ed e$Z&e%ed e$Z'ej"j(Z)ej"j*Z+e'ej"j,Z-ej.j/dkrdndZ0dZ1e1Z2ej3j4Z5ej"j6Z7ej"j8Z9ee:ddZ;eddZej>e'ej>ddddZ?eddZ@eddZAee'e'e'e'e'e'ejBejBdddd ZCeejDd!eejEd!d"d#ZFeejGd!d$d%ZHeejId!d&d'ZJe rSe'j/d(d)kr%e'd*ZKe'd+ZLe'd,ZMed-ejNidd.d/ZOne'd0ZKe'd1ZLe'd2ZMed-ejNidd3d/ZOdd4lPmQZQee'e'e'e'e'd5dd6d7ZRnee'e&e'e&d8dd9d7ZRed:d;ZSeejTd!dlmVZVmWZWmXZXmYZYmZZZm[Z[Gd?d@d@eZZ\GdAdBdBeZZ]GdCdDdDeZZ^GdEdFdFeZZ_GdGdHdHe[Z`edIdJdKgZadLdMZbebZce9dNvr$e9dOkrdPZdeeedeejNejNdQddRdSZfeejNejNejNejNejNejNdTddUdVZgeejNejNejNejNdWddXdYZheejNejNejNejNejNejNeiejNdZd[ejNejNejNejNd\ dd]d^Zjnd_e9Zdekeded`daZldbdcZmeemdddeZnedfe'iddgdhZoeejpd!didjZqdS)kz& Hash implementations for Numba types N) namedtuple)ir)overloadoverload_method intrinsicregister_jitable)errors)typesutils) grab_byte grab_uint64_t) const_int get_next_intget_next_int32 get_state_ptr))r zint%szuint%s =iCBcC dd}|S)NcSs|SN)__hash__)objr.implr)rrrrr hash_overload0rcCs t|}|tdkrtd}|S)N) _Py_hash_tint)valasintrrrprocess_return7s r')xymesign_PyHASH_MODULUS _PyHASH_BITS)localscCs t|st|r|dkrtSt Strt}t|StSt |\}}d}|dkr0d}| }d}|r^|d>t @|t d?B}|d9}|d8}t |}||8}||7}|t kr\|t 8}|s4|dkrg|t }n t dd|t }||>t @|t |?B}||}t|S)Nrr!gA) npisfiniteisinf _PyHASH_INF_py310_or_later_prng_random_hashr' _PyHASH_NANmathfrexpr-r.r$)vr(r*r+r,r)rrr_Py_HashDoubleFs>   r<cCsdd}ttj}||fS)NcSs|d}||tS)Nr)fpextr DoubleType)cgctxbuilder signatureargsr%rrrrsz_fpext..impl)r float64float32tyctxr%rsigrrr_fpext~s rHcCsdd}t}||fS)NcSs@t||d}tt}tdkrt|||}|St||||d}|S)NinternalrF)rr _hash_widthrr)r?r@rArB state_ptrbitsvaluerrrrs  z_prng_random_hash..implr#)rFrrGrrrr7sr7)r(p1p2p3p4r-r. _PyLong_SHIFTc Csdt}td|?}dtd}d}tt}t|dddD]'}|t>}|t@}||?} || B}|t||t?|@7}|tkrG|t8}q t|S)Nrr@r0r!)rSr uint32r.ranger-r#) r% _tmp_shift mask_shiftir(rQidxrOrPrRrrr _long_impls r[rcsHtjdkrdndtdt|ddrtjntjfdd}|S) Nlr"lsignedFcs|}t|}|tkr*|dkrd}t|S|kr"t}t|St|}t|Sd}|dkr5| }d}t|}|r>| }t|S)NrFT)absr-r#r[r')r%magret needs_negate_BIG _HASH_I64_MIN _SIGNED_MINrrrs&    zint_hash..impl)sysmaxsizer int64getattruint64r%rrrbrint_hashs  rlcCs"|jdkr dd}|Sdd}|S)NrTcSs t|}|Sr)r<r%hashedrrrrszfloat_hash..implcSstt|}t|}|Sr)r2rCrHr<)r% fpextendedrnrrrrs)bitwidthrkrrr float_hashs  rqcCr)NcSs(t|j}t|j}|t|}t|Sr)hashrealimag _PyHASH_IMAGr')r%hashrealhashimagcombinedrrrrs   zcomplex_hash..implrrkrrr complex_hash s ryrlJ fq lOkO8 lg,,>r(cC|td>|td?BS)Nr!r rjr(rrr_PyHASH_XXROTATE'rlyn<lwJ igVcCr{)N r}r~rrrr0r)literal_unroll)acclane_PyHASH_XXPRIME_5_PyHASH_XXPRIME_1tlcCst|}t}t|D]}t|}|tdkrdS||t7}t|}|t9}q ||ttdAA7}|tdkr.implrNrErrr_tuple_hash_resolvels rcCs(tst|tjrdd}|Sdd}|S)NcSt|Sr)r)r%rrrrrztuple_hash..implcSstt|}t|Sr)r#rr'rmrrrrs )_py38_or_later isinstancer Sequencerkrrr tuple_hashs r)c_size_tc_ubytec_uint64 pythonapi StructureUnionc@eZdZdefdefgZdS)FNVprefixsuffixN)__name__ __module__ __qualname__r_fields_rrrrrrc@r)SIPHASHk0k1N)rrrrrrrrrrrrc@ eZdZdedfdefgZdS)DJBX33ApaddingrNrrrrrrrrrrr rc@r)EXPATrrhashsaltNrrrrrrrrc@s2eZdZdedfdefdefdefdefgZdS)_Py_HashSecret_tucfnvsiphashdjbx33aexpatN) rrrrrrrrrrrrrrs r_hashsecret_entrysymbolrMcsJttd}ifdd}|d|jj|d|jj|d|jjS)zRead hash secret from the Python process Returns ------- info : dict - keys are "djbx33a_suffix", "siphash_k0", siphash_k1". - values are the namedtuple[symbol:str, value:int] _Py_HashSecretcs>d|}t|}t|}t||t||d|<dS)Nz_numba_hashsecret_{})rrM)formatctypesr addressofll add_symbolr)namer% symbol_nameaddrinforrinjects    z!_build_hashsecret..injectdjbx33a_suffix siphash_k0 siphash_k1)rin_dllrrrrrr) pyhashsecretrrrr_build_hashsecrets  r) siphash24rraFNV hashing is not implemented in Numba. See PEP 456 https://www.python.org/dev/peps/pep-0456/ for rationale over not using FNV. Numba will continue to work, but hashes for built in types will be computed using siphash24. This will permit e.g. dictionaries to continue to behave as expected, however anything relying on the value of the hash opposed to hash as a derived property is likely to not work as expected.r(bcCs t||>|td|?BS)NrTr}rrrr_ROTATE6s rarcdstcCsB||7}||7}t|||A}t|||A}t|d}||||fS)Nr)rrrrr _HALF_ROUND;s   rv0v1v2v3cCstt||||dd\}}}}t||||dd\}}}}t||||dd\}}}}t||||dd\}}}}||||fS)Nrr)rrrrr _DOUBLE_ROUNDIs  rr0C) rrrrrmitmprmaskjmpohexefefcCst|d>}|tdA}|tdA}|tdA}|tdA}d} |dkrMt|| } | d7} |d8}|| N}t||||\}}}}|| N}|dks)td} | d} td } |d kryd }t| |>}| |@tt|| d |>B} |d krd }t| |>}| |@tt|| d|>B} |dkrd}t| |>}| |@tt|| d|>B} |dkr| tdM} tdD]}|d}t| |>}| |@tt|| ||>B} q|dkrd}t| |>}| |@tt|| d|>B} |dkr!d}t| |>}| |@tt|| d|>B} |dkr7| }| |@tt|| dB} || O}||N}t||||\}}}}||N}|| N}t||||\}}}}t||||\}}}}||A||AA} | S)N8lue`5{lmo\I{#lar\clser+#rrr00(rrzlrr)r rjr rr rV)rrsrcsrc_szrrrrrrZrrboffsetrrrrYrrr _siphash24Ts        r z'Unsupported hashing algorithm in use %scsDt|tjs tdt|jjtj}||}fdd}||fS)z&Emit code to load the hashsecret. zrequires literal stringcsL|j}z|}Wntytj|tdd}Ynw||}|S)NrTr)module get_globalKeyErrorrGlobalVariableIntTypeload)r?r@rGrBmodgvr;symrrrs  z%_inject_hashsecret_read..impl) rr StringLiteralr TypingError _hashsecret literal_valuerrj)rFrrestyrGrrrr_inject_hashsecret_reads     rcCs t|jSr)rrMr rrr_load_hashsecrets rcCr)NcSrr)rr rrrimprz"_impl_load_hashsecret..impr)rrrrr_impl_load_hashsecretr r_hashcCs|dkrtdS|tkr3td}t|D]}|d>|tt||}q||N}|tdN}t|Stt tdt td||}t|}t|S)Nrirrrr) r'_Py_HASH_CUTOFFrrVr2uint8r rr r rj)r%_lenr rZrrrr _Py_HashBytess   r$csddlmfdd}|S)Nr_kind_to_byte_widthcs4|j}t|}|j}|dkr|St|j||S)Nr!)_kindrr r$_data)r% kindwidthr# current_hashr%rrrs zunicode_hash..impl)numba.cpython.unicoder&rkrr%r unicode_hashs   r,)r__doc__r9numpyr2rfrwarnings collectionsrllvmlite.bindingbindingrllvmliternumba.core.extendingrrrr numba.corerr r numba.core.unsafe.bytesr r numba.cpython.randomimplr rrr PYVERSIONrr6 hash_infowidthrJrir#rinfr5nanr8modulusr-intprpr.rruint_infobits_per_digitrScutoffr! algorithm_Py_hashfunc_namerrrr'doubleintcr<rHr7int32r[IntegerBooleanrlFloatrqComplexryrrrrjrnumbarrr BaseTuplerrrrrrrrrrrrrrrmsgwarnrrrArrayr  ValueErrorrrrr$ UnicodeTyper,rrrrs$          1       ,                *    C