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)maxmaskrm/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/numba/np/random/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@}||krrr?r9rrr@rAvalrrrrandom_bounded_uint16_fills&  rGc CrD)z^ Returns a new array of given size with 8 bit integers bounded by given interval. rr8r%)r:r;r<r!r,rErrrrandom_bounded_uint8_fills&  rHc CsLd}d}tj||d}t|D]}t|||||\} }}|| ||<q|S)z@ Returns a new array of given size with boolean values. rr8)r:r;r<rrErrrrandom_bounded_bool_fill/srIcCs||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)r>highZendpoint lower_bound upper_boundrrr_randint_arg_check=s& rNcCs|dkrdSt|}tt|}|dkr0tt||@}||kr,tt||@}||ks t|St||@}||krDt||@}||ks:t|S)Nrr0)rrr r )rZmax_valrvaluerrrrandom_interval\s   rP) numpyr:ZnumbarrrrZnumba.core.extendingrZnumba.np.random._constantsrrr r Znumba.np.random.generator_corer r rrr!r#r,r/r1r7rBrCrGrHrIrNrPrrrrsF      # #  "