U |e'@s ddlZddlmZmZmZmZddlmZddl m Z m Z m Z m Z ddlmZmZeddZedd Zed d Zed d ZeddZeddZeddZeddZeddZeddZeddZeddZeddZed d!Zed"d#ZdS)$N)uint64uint32uint16uint8)register_jitable) UINT32_MAX UINT64_MAX UINT16_MAX UINT8_MAX) next_uint32 next_uint64cCsTt|}||d?O}||d?O}||d?O}||d?O}||d?O}||d?O}|S)N )r)maxmaskr[/var/www/website-v5/atlas_env/lib/python3.8/site-packages/numba/np/random/random_methods.pygen_masks      rcCsF|dkr|||fS|s$t|}d}n|dL}|d8}|d@dk||fS)Nrr )r )bitgenoffrngbcntbufrrrbuffered_bounded_bools rcCs0|st|}d}n|dL}|d8}t|||fS)Nrr )r rrrrrrrbuffered_uint8(s r!cCs0|st|}d}n|dL}|d8}t|||fS)Nr r)r rr rrrbuffered_uint164s r"c Cst|td}|dkstt|||\}}}t||}|d@}||krtt||}||krt|||\}}}t||}|d@}qZ|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)rAssertionErrorr!rr rrrrrng_exclnmleftover thresholdrrrbuffered_bounded_lemire_uint8Bs     r+c Cst|td}|dkstt|||\}}}t||}|d@}||krtt||}||krt|||\}}}t||}|d@}qZ|d?||fS)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"rr r%rrrbuffered_bounded_lemire_uint16fs     r-cCszt|td}|dksttt|t|}|d@}||krrt||}||krrtt|t|}|d@}qL|d?S)zr Generates a random unsigned 32 bit integer bounded within a given interval using Lemire's rejection. r r)rr$rr r)rrr&r(r)r*rrrbuffered_bounded_lemire_uint32s   r/cCst|td}|dkstt|}t|t|}||krjt||}||krjt|}t|t|}qH|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 r.r)rr$r r)rrr&xr)r*x0x1Z rng_excl0Z rng_excl1w0tw1w2m1rrrbounded_lemire_uint64s(      r9cCstj||d}|dkr0t|D] }|||<q n|dkr|dkrbt|D]}|t|||<qJqt|D]}|t||||<qlnL|dkrt|D]}|t|||<qn"t|D]}|t||||<q|S)z_ Returns a new array of given size with 64 bit integers bounded by given interval. dtyperr.r0)npemptyndindexr r/r r9rlowrsizer;outirrrrandom_bounded_uint64_fills  rDcCstj||d}|dkr0t|D] }|||<q nL|dkrZt|D]}|t|||<qBn"t|D]}|t||||<qd|S)z_ Returns a new array of given size with 32 bit integers bounded by given interval. r:rr.)r<r=r>r r/r?rrrrandom_bounded_uint32_fills rEc Csd}d}tj||d}|dkr8t|D] }|||<q(nh|dkrpt|D]"}t|||\} }}|| ||<qJn0t|D]$}t||||\} }}|| ||<qz|S)z_ Returns a new array of given size with 16 bit integers bounded by given interval. rr:r,)r<r=r>r"r- rr@rrAr;rrrBrCvalrrrrandom_bounded_uint16_fills$ rHc Csd}d}tj||d}|dkr8t|D] }|||<q(nh|dkrpt|D]"}t|||\} }}|| ||<qJn0t|D]$}t||||\} }}|| ||<qz|S)z^ Returns a new array of given size with 8 bit integers bounded by given interval. rr:r#)r<r=r>r!r+rFrrrrandom_bounded_uint8_fills$ rIc CsLd}d}tj||d}t|D]&}t|||||\} }}|| ||<q |S)z@ Returns a new array of given size with boolean values. rr:)r<r=r>rrFrrrrandom_bounded_bool_fill/srJcCs||krtd|dkrjt|}|s0|td8}t|}|dkrHt|}||krXtd||krtdn ||krztd||krtddS)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) ValueErrorr)r@highendpoint lower_bound upper_boundrrr_randint_arg_check=s"  rPcCs~|dkr dSt|}tt|}|dkrTtt||@}||krvtt||@}q8n"t||@}||krvt||@}q`t|S)Nrr.)rrr r )rmax_valrvaluerrrrandom_interval\s  rS) numpyr<numbarrrrnumba.core.extendingrnumba.np.random._constantsrrr r numba.np.random.generator_corer r rrr!r"r+r-r/r9rDrErHrIrJrPrSrrrrsD      # #  "