o
    Td4j                  
   @   s  d Z ddlZddlZddlZddlZddlZddlmZ ddl	m
Z ddlmZ ddlmZ ddlmZmZmZmZ ddlmZ ddlmZmZ dd	lmZmZ dd
lmZmZmZm Z  ej!dkZ"ej#j$Z%e&ede% Z'e&ede% Z(ej#j)Z*ej#j+Z,e(ej#j-Z.ej/j0dkrdndZ1dZ2e2Z3ej4j5Z6ej#j7Z8ej#j9Z:dd Z;ee;dd Z<ee=dd Z>edd Z?ee(e(ej@ejAejAe(ejAdddd ZBedd ZCed d! ZDee(e(e(e(e(e(ejEejEd"dd#d$ ZFeejGd%eejHd%d&d' ZIeejJd%d(d) ZKeejLd%d*d+ ZMe(j0d, d-kr/e(d.ZNe(d/ZOe(d0ZPed1ejQidd2d3 ZRne(d4ZNe(d5ZOe(d6ZPed1ejQidd7d3 ZRee(e(e(e(e(d8dd9d: ZSeejTd%d;d< ZUdd=lmVZVmWZWmXZXmYZYmZZZm[Z[ G d>d? d?eZZ\G d@dA dAeZZ]G dBdC dCeZZ^G dDdE dEeZZ_G dFdG dGe[Z`edHdIdJgZadKdL Zbeb Zce:dMv re:dNkrdOZdeeed eejQejQdPddQdR ZfeejQejQejQejQejQejQdSddTdU ZgeejQejQejQejQdVddWdX ZheejQejQejQejQdVddYdZ Zid[d\ Zjejd]Zkejd^Zle:d]kreknelZmnd_e: Zdeneded`da Zodbdc Zpeepddde Zqedfe(iddgdh Zreejsd%didj ZtdS )kz&
Hash implementations for Numba types
    N)
namedtuple)ir)literal_unroll)overloadoverload_method	intrinsicregister_jitable)errors)typesutils)	grab_bytegrab_uint64_t)	const_intget_next_intget_next_int32get_state_ptr)   
   zint%szuint%s       =   iCB c                 C   s   d S N )	hash_funcr   r   h/home/ncw/WWW/www-new/content/articles/pi-bbp/venv/lib/python3.10/site-packages/numba/cpython/hashing.py_defer_hash1   s   r   c                    s   d|  d  fdd}|S )Nzunhashable type: ''c                    s   |d u rt  | S r   )	TypeError)objr   err_msgr   r   impl9   s   zol_defer_hash.<locals>.implr   )r   r   r!   r   r   r   ol_defer_hash5   s   r"   c                    s   d|  d  fdd}|S )Nz+No __hash__ is defined for object of type 'zn' and a generic hash() cannot be performed as there is no suitable object represention in Numba compiled code!c                    s"   t | drt| t| dS t )N__hash__)hasattrr   getattrr   )r   attempt_generic_msgr   r   r!   I   s   
zhash_overload.<locals>.implr   )r   r!   r   r&   r   hash_overloadB   s
   r(   c                 C   s    t | }|tdkrtd}|S )N)
_Py_hash_tint)valasintr   r   r   process_returnQ   s   r/   )xymesign_PyHASH_MODULUS_PyHASH_BITSlocalsc                 C   s
  t | st | r| dkrtS t S trt }t|S tS t	| \}}d}|dk r0d}| }d}|r^|d> t
@ |td ? B }|d9 }|d8 }t|}||8 }||7 }|t
kr\|t
8 }|s4|dkrg|t }n
td d| t  }||> t
@ |t| ? B }|| }t|S )Nr      r)      g      A)npisfiniteisinf_PyHASH_INF_py310_or_later_prng_random_hashr/   _PyHASH_NANmathfrexpr5   r6   r,   )vr0   r2   r3   r4   r1   r   r   r   _Py_HashDouble`   s>   



rE   c                 C   s   dd }t t j}||fS )Nc                 S   s   |d }| |t S )Nr   )fpextr   
DoubleType)cgctxbuilder	signatureargsr-   r   r   r   r!      s   z_fpext.<locals>.impl)r
   float64float32)tyctxr-   r!   sigr   r   r   _fpext   s   rP   c                 C   s   dd }t  }||fS )Nc                 S   s@   t | |d}tt}tdkrt| ||}|S t| |||d}|S )Ninternalr   F)r   r   _hash_widthr   r   )rH   rI   rJ   rK   	state_ptrbitsvaluer   r   r   r!      s   	z_prng_random_hash.<locals>.impl)r+   )rN   r!   rO   r   r   r   r@      s   r@   )r0   p1p2p3p4r5   r6   _PyLong_SHIFTc           
      C   s   dt  }td |? }dt  d }d}tt  }t|d ddD ]'}|t > }|t@ }||? }	||	B }|t| |t  ? |@ 7 }|tkrG|t8 }q t|S )Nr   r   @   r9   r)   )rZ   r
   uint32r6   ranger5   r+   )
r-   
_tmp_shift
mask_shiftir0   rX   idxrV   rW   rY   r   r   r   
_long_impl   s   rb   r#   c                    sH   t jdkrdndtdt| ddrtjntj  fdd}|S )	Nl        r*   l         signedFc                    s    | } t | }|tk r*| dkrd}t|S | kr"t}t|S t| }t|S d}| dk r5|  } d}t| }|r>| }t|S )Nr   FT)absr5   r+   rb   r/   )r-   magretneeds_negate_BIG_HASH_I64_MIN_SIGNED_MINr   r   r!      s&   
	zint_hash.<locals>.impl)sysmaxsizer
   int64r%   uint64r-   r!   r   ri   r   int_hash   s
   
rr   c                 C   s"   | j dkrdd }|S dd }|S )Nr[   c                 S   s   t | }|S r   )rE   )r-   hashedr   r   r   r!     s   zfloat_hash.<locals>.implc                 S   s   t t| }t|}|S r   )r;   rL   rP   rE   )r-   
fpextendedrs   r   r   r   r!     s   )bitwidthrq   r   r   r   
float_hash  s
   
	rv   c                 C      dd }|S )Nc                 S   s(   t | j}t | j}|t|  }t|S r   )hashrealimag_PyHASH_IMAGr/   )r-   hashrealhashimagcombinedr   r   r   r!   (  s   

zcomplex_hash.<locals>.implr   rq   r   r   r   complex_hash&  s   
r         l   Jfq	 l   OkO8 l   g,,> r0   c                 C      | t d> | t d? B S )Nr   !   r
   rp   r0   r   r   r   _PyHASH_XXROTATE@     r   l   yn< l   wJ igVc                 C   r   )N      r   r   r   r   r   r   I  r   )acclane_PyHASH_XXPRIME_5_PyHASH_XXPRIME_1tlc                 C   s   t | }t}t| D ]}t|}|tdkr dS ||t 7 }t|}|t9 }q
||ttdA A 7 }|tdkr<tdS t|S )Nr)   is5 iK*\)	lenr   r   rx   _Py_uhash_t_PyHASH_XXPRIME_2r   r   r/   )tupr   r   r0   r   r   r   r   _tuple_hashO  s   
r   c                 C   rw   )Nc                 S      t | S r   )r   )r-   r   r   r   r!   h     ztuple_hash.<locals>.implr   rq   r   r   r   
tuple_hashf     r   )c_size_tc_ubytec_uint64	pythonapi	StructureUnionc                   @      e Zd ZdefdefgZdS )FNVprefixsuffixN)__name__
__module____qualname__r   _fields_r   r   r   r   r         r   c                   @   r   )SIPHASHk0k1N)r   r   r   r   r   r   r   r   r   r     r   r   c                   @       e Zd Zded fdefgZdS )DJBX33Apadding   r   Nr   r   r   r   r   r   r   r   r   r   r         
r   c                   @   r   )EXPATr   r   hashsaltNr   r   r   r   r   r     r   r   c                   @   s2   e Zd Zded fdefdefdefdefgZdS )_Py_HashSecret_tuc   fnvsiphashdjbx33aexpatN)	r   r   r   r   r   r   r   r   r   r   r   r   r   r     s    
r   _hashsecret_entrysymbolrU   c                     sJ   t td} i   fdd}|d| jj |d| jj |d| jj  S )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_HashSecretc                    s>   d | }t|}t|}t|| t||d | < d S )Nz_numba_hashsecret_{})r   rU   )formatctypesr   	addressofll
add_symbolr   )namer-   symbol_nameaddrinfor   r   inject  s
   


z!_build_hashsecret.<locals>.injectdjbx33a_suffix
siphash_k0
siphash_k1)r   in_dllr   r   r   r   r   r   )pyhashsecretr   r   r   r   _build_hashsecret  s   r   )	siphash13	siphash24r   r   a  FNV 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.r0   bc                 C   s    t | |> | t d| ? B S )Nr[   r   r   r   r   r   _ROTATE  s    r   ar   cdstc                 C   sB   | |7 } ||7 }t ||| A }t |||A }t | d} | |||fS )Nr   )r   r   r   r   r   _HALF_ROUND  s   
r   v0v1v2v3c                 C   s@   t | |||dd\} }}}t ||| |dd\}}} }| |||fS )Nr   r         )r   r   r   r   r   _SINGLE_ROUND$  s   r   c                 C   s8   t | |||\} }}}t | |||\} }}}| |||fS r   )r   r   r   r   r   _DOUBLE_ROUND-  s   r   c                    sr   | dkr	t d n| dkrtd nJ dttjtjtjtjtjtjtjtjtjtjd
d fd	d
}|S )Nr   Tr   Fr   unreachable)
r   r   r   r   r   mir   maskjmpohexefefr7   c                    s   t |d> }| t dA }|t dA }| t dA }|t dA }d}	|dkrMt||	}
|	d7 }	|d8 }||
N }||||\}}}}||
N }|dks)t d}|	d }t d	}|d
kryd}t ||>  }||@ t t||d |> B }|dkrd}t ||>  }||@ t t||d |> B }|dkrd}t ||>  }||@ t t||d |> B }|dkr|t dM }tdD ]}|d }t ||>  }||@ t t||| |> B }q|dkrd}t ||>  }||@ t t||d |> B }|dkr!d}t ||>  }||@ t t||d |> B }|dkr7| }||@ t t||d B }||O }||N }||||\}}}}||N }||N }||||\}}}}||||\}}}} rv||||\}}}}||A ||A A }|S )N8   l   ue`5{ l   mo\I{# l   ar\c l   ser+# r   r   r9         0      (      r   r   l        r   r      )r
   rp   r   r   r]   )r   r   srcsrc_szr   r   r   r   r   ra   r   r   boffsetr   r   r   r`   _EXTRA_ROUND_ROUNDERr   r   _siphash@  s   






z_gen_siphash.<locals>._siphash)r   r   r   r
   rp   )algr   r   r   r   _gen_siphash6  s(   
Er   r   r   z'Unsupported hashing algorithm in use %sc                    sD   t |tjstdt|j j tj}||} fdd}||fS )z&Emit code to load the hashsecret.
    zrequires literal stringc                    sL   |j }z| }W n ty   tj|td d}Y nw ||}|S )Nr[   r   )module
get_globalKeyErrorr   GlobalVariableIntTypeload)rH   rI   rO   rK   modgvrD   symr   r   r!     s   
z%_inject_hashsecret_read.<locals>.impl)	
isinstancer
   StringLiteralr	   TypingError_hashsecretliteral_valuer   rp   )rN   r   restyrO   r!   r   r  r   _inject_hashsecret_read  s   
r	  c                 C   s
   t |  jS r   )r  rU   r   r   r   r   _load_hashsecret  s   
r
  c                 C   rw   )Nc                 S   r   r   )r	  r   r   r   r   imp  r   z"_impl_load_hashsecret.<locals>.impr   )r   r  r   r   r   _impl_load_hashsecret  r   r  _hashc                 C   s   |dkrt dS |tk r3td}t|D ]}|d> | tt| | }q||N }|tdN }t |S tt	
tdt	
td| |}t |}t |S )Nr   i  r   r   r   r   )r/   _Py_HASH_CUTOFFr   r]   r;   uint8r   r
  
_siphasherr
   rp   )r-   _lenr  ra   tmpr   r   r   _Py_HashBytes  s   r  c                    s   ddl m   fdd}|S )Nr   _kind_to_byte_widthc                    s4    | j }t| }| j}|dkr|S t| j|| S )Nr)   )_kindr   r  r  _data)r-   	kindwidthr  current_hashr  r   r   r!     s   
zunicode_hash.<locals>.impl)numba.cpython.unicoder  rq   r   r  r   unicode_hash  s   r  )u__doc__rB   numpyr;   rm   r   warningscollectionsr   llvmlite.bindingbindingr   llvmliter   numbar   numba.core.extendingr   r   r   r   
numba.corer	   r
   r   numba.core.unsafe.bytesr   r   numba.cpython.randomimplr   r   r   r   	PYVERSIONr?   	hash_infowidthrR   r%   r+   r   infr>   nanrA   modulusr5   intpru   r6   _PyHASH_MULTIPLIERr{   int_infobits_per_digitrZ   cutoffr  	algorithm_Py_hashfunc_namer   r"   rx   r(   r/   doubleintcrE   rP   r@   int32rb   IntegerBooleanrr   Floatrv   Complexr   r   r   r   rp   r   r   	BaseTupler   r   r   r   r   r   r   r   r   r   r   r   r   r   r  msgwarnr   r   r   r   r   
_siphash13
_siphash24r  
ValueErrorr	  r
  r  r  UnicodeTyper  r   r   r   r   <module>   s   




1





,






 



/



[




