o ۾i!(@sddlZddlmZddlmZmZmZmZddl m Z m Z eddZ eddZ ed d Zed d Zed dZeddZeddZeddZeddZeddZeddZeddZeddZedd Zed!d"ZdS)#N)register_jitable) UINT32_MAX UINT64_MAX UINT16_MAX UINT8_MAX) next_uint32 next_uint64cCsVt|}||d?O}||d?O}||d?O}||d?O}||d?O}||d?O}|S)N )npuint64)maxmaskrV/home/ubuntu/.local/lib/python3.10/site-packages/numba/np/random/new_random_methods.pygen_mask s       rcCsF|dkr |||fS|st|}d}n|dL}|d8}|d@dk||fS)Nrr )r)bitgenoffrngbcntbufrrrbuffered_bounded_bools rcCs2|s t|}d}n|dL}|d8}t|||fS)Nr r )rruint8rrrrrrbuffered_uint8' r cCs2|s t|}d}n|dL}|d8}t|||fS)Nr r )rruint16rrrrbuffered_uint163r!r#c Ct|td}|dksJt|||\}}}t||}|d@}||krMtt||}||krMt|||\}}}t||}|d@}||ks5|d?||fS)a? Generates a random unsigned 8 bit integer bounded within a given interval using Lemire's rejection. The buffer acts as storage for a 32 bit integer drawn from the associated BitGenerator so that multiple integers of smaller bitsize can be generated from a single draw of the BitGenerator. r r )rrr r"r rrrrrng_exclnmleftover thresholdrrrbuffered_bounded_lemire_uint8A r,c Cr$)a@ Generates a random unsigned 16 bit integer bounded within a given interval using Lemire's rejection. The buffer acts as storage for a 32 bit integer drawn from the associated BitGenerator so that multiple integers of smaller bitsize can be generated from a single draw of the BitGenerator. r r )rr"r#uint32rr&rrrbuffered_bounded_lemire_uint16er-r0cCst|td}|dksJtt|t|}|d@}||krBt||}||krBtt|t|}|d@}||ks.|d?S)zr Generates a random unsigned 32 bit integer bounded within a given interval using Lemire's rejection. r r)rr/rrr)rrr'r)r*r+rrrbuffered_bounded_lemire_uint32s  r2cCst|td}|dksJt|}t|t|}||kr>t||}||kr>t|}t|t|}||ks,|td@}|d?}|td@}|d?} ||} ||| d?} | td@} | d?} | || 7} || | | d?}|S)zr Generates a random unsigned 64 bit integer bounded within a given interval using Lemire's rejection. r r1r)rrrr)rrr'xr*r+x0x1 rng_excl0 rng_excl1w0tw1w2m1rrrbounded_lemire_uint64s*   r>cCstj||d}|dkrt|D]}|||<q|S|dkrF|dkr3t|D] }|t|||<q&|St|D] }|t||||<q8|S|dkr\t|D] }|t|||<qO|St|D] }|t||||<qa|S)z_ Returns a new array of given size with 64 bit integers bounded by given interval. dtyperr1r3)remptyndindexrr2rr>rlowrsizer@outirrrrandom_bounded_uint64_fills(   rHcCstj||d}|dkrt|D]}|||<q|S|dkr/t|D] }|t|||<q"|St|D] }|t||||<q4|S)z_ Returns a new array of given size with 32 bit integers bounded by given interval. r?rr1)rrArBrr2rCrrrrandom_bounded_uint32_fills rIc Cd}d}tj||d}|dkrt|D]}|||<q|S|dkr:t|D]}t|||\} }}|| ||<q&|St|D]}t||||\} }}|| ||<q?|S)z_ Returns a new array of given size with 16 bit integers bounded by given interval. rr?r.)rrArBr#r0 rrDrrEr@rrrFrGvalrrrrandom_bounded_uint16_fills&  rMc CrJ)z^ Returns a new array of given size with 8 bit integers bounded by given interval. rr?r%)rrArBr r,rKrrrrandom_bounded_uint8_fills&  rNc CsLd}d}tj||d}t|D]}t|||||\} }}|| ||<q|S)z@ Returns a new array of given size with boolean values. rr?)rrArBrrKrrrrandom_bounded_bool_fill.srOcCs||krtd|dkr:t|}|s|td8}t|}|dkr(t|}||kr0td||kr8tddS||krBtd||krJtddS)zS Check that low and high are within the bounds for the given datatype. zlow is out of boundsrr zhigh is out of boundsz*low is greater than high in given intervalN) ValueErrorrr)rDhighendpoint lower_bound upper_boundrrr_randint_arg_check<s&   rUcCs|dkrdSt|}tt|}|dkr1tt||@}||kr0tt||@}||ks#nt||@}||krEt||@}||ks;t|S)Nrr1)rrrrr)rmax_valrvaluerrrrandom_interval[s    rX)numpyrnumba.core.extendingrnumba.np.random._constantsrrrrnumba.np.random.generator_corerrrrr r#r,r0r2r>rHrIrMrNrOrUrXrrrrsD      # #  "