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/html/Darija-Ai-Train/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*Zx0x1Z rng_excl0Z rng_excl1Zw0tZw1Zw2m1rrrbounded_lemire_uint64s(      r5cCstj||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 r5rlowrsizer7outirrrrandom_bounded_uint64_fills  r@cCstj||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. r6rr.)r8r9r:r r/r;rrrrandom_bounded_uint32_fills rAc 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. rr6r,)r8r9r:r"r- rr<rr=r7rrr>r?valrrrrandom_bounded_uint16_fills$ rDc 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. rr6r#)r8r9r:r!r+rBrrrrandom_bounded_uint8_fills$ rEc CsLd}d}tj||d}t|D]&}t|||||\} }}|| ||<q |S)z@ Returns a new array of given size with boolean values. rr6)r8r9r:rrBrrrrandom_bounded_bool_fill/srFcCs||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<highZendpoint lower_bound upper_boundrrr_randint_arg_check=s"  rKcCs~|dkr dSt|}tt|}|dkrTtt||@}||krvtt||@}q8n"t||@}||krvt||@}q`t|S)Nrr.)rrr r )rZmax_valrvaluerrrrandom_interval\s  rM) numpyr8ZnumbarrrrZnumba.core.extendingrZnumba.np.random._constantsrrr r Znumba.np.random.generator_corer r rrr!r"r+r-r/r5r@rArDrErFrKrMrrrrsD      # #  "