o i'@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_/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/numba/np/random/old_random_methods.pygen_masks      rcCsF|dkr |||fS|st|}d}n|dL}|d8}|d@dk||fS)Nrr )r )bitgenoffrngbcntbufrrrbuffered_bounded_bools rcCs0|s t|}d}n|dL}|d8}t|||fS)Nrr )r rrrrrrrbuffered_uint8( r!cCs0|s t|}d}n|dL}|d8}t|||fS)Nr r)r rr rrrbuffered_uint164r"r#c Ct|td}|dksJt|||\}}}t||}|d@}||krHtt||}||krHt|||\}}}t||}|d@}||ks1|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)rr!rr rrrrrng_exclnmleftover thresholdrrrbuffered_bounded_lemire_uint8B   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#rr r&rrrbuffered_bounded_lemire_uint16fr-r/cCst|td}|dksJtt|t|}|d@}||krrr0)r@rArBr r1rCrrrrandom_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.)r@rArBr#r/ 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%)r@rArBr!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>)r@rArBrrKrrrrandom_bounded_bool_fill/srOcCs||krtd|dkr6t|}|s|td8}t|}|dkr$t|}||kr,td||kr4tddS||kr>td||krFtddS)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)rDhighendpoint lower_bound upper_boundrrr_randint_arg_check=s& rUcCs|dkrdSt|}tt|}|dkr0tt||@}||kr,tt||@}||ks t|St||@}||krDt||@}||ks:t|S)Nrr0)rrr r )rmax_valrvaluerrrrandom_interval\s   rX) numpyr@numbarrrrnumba.core.extendingrnumba.np.random._constantsrrr r numba.np.random.generator_corer r rrr!r#r,r/r1r=rHrIrMrNrOrUrXrrrrsF      # #  "