xKg'HddlZddlmZmZmZmZddlmZddl m Z m Z m Z m Z ddlmZmZedZedZedZed Zed Zed Zed Zed ZedZedZedZedZedZedZedZy)N)uint64uint32uint16uint8)register_jitable) UINT32_MAX UINT64_MAX UINT16_MAX UINT8_MAX) next_uint32 next_uint64c|t|}||dz z}||dz z}||dz z}||dz z}||dz z}||dz z}|S)N )r)maxmasks b/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/numba/np/random/random_methods.pygen_maskrsd #;DDAIDDAIDDAIDDAIDDBJDDBJD Kc`|dk(r|||fS|st|}d}n |dz}|dz}|dzdk7||fS)Nrr)r )bitgenoffrngbcntbufs rbuffered_bounded_boolr!sN qD#~ &!     1WNT3 &&rcR|st|}d}n |dz}|dz}t|||fS)Nrr)r rrrr s rbuffered_uint8r%(s7 &!     :tS  rcR|st|}d}n |dz}|dz}t|||fS)Nrr)r rr$s rbuffered_uint16r'4s7 &!     #;c !!rc(t|tdz}|dk7sJt|||\}}}t||z}|dz}||krDtt|z |z}||kr*t|||\}}}t||z}|dz}||kr*|dz ||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. rr)rr%rr rrrr rng_exclnmleftover thresholds rbuffered_bounded_lemire_uint8r0BsSzE!H$H 4KK"&$4LAtSq8|A4xH8I&,8 )#)&$>A:~H)# Grct|tdz}|dk7sJt|}t|t|z}||kr9t|z |z}||kr(t|}t|t|z}||kr(|tdz}|dz }|tdz}|dz } ||z} ||z| dz z} | tdz} | dz } | || zz } || z| z| dz z}|S)zr Generates a random unsigned 64 bit integer bounded within a given interval using Lemire's rejection. rr5r)rr r )rrr+xr.r/x0x1 rng_excl0 rng_excl1w0tw1w2m1s rbounded_lemire_uint64rCs c{VAY&H % %& %FAay6(++H8#%1 )#F#Aay6(#33H)# VJ  B bB6*--IBI iB Y"(#A VJ  B bB"y.B i" b )B Irctj||}|dk(r!tj|D]}|||< |S|dkr`|dk(r-tj|D]}|t|z||<|Stj|D]}|t ||z||<|S|dk(r-tj|D]}|t |z||<|Stj|D]}|t ||z||<|S)z_ Returns a new array of given size with 64 bit integers bounded by given interval. dtyperr5r8)npemptyndindexr r6r rCrlowrsizerFoutis rrandom_bounded_uint64_fillrOs ((4u %C axD!ACF"" J  : ZZ%{622A& JZZ%=fcJJA& J # #D!A;v..CF" JD!A0==CF" Jrc<tj||}|dk(r!tj|D]}|||< |S|dk(r-tj|D]}|t|z||<|Stj|D]}|t ||z||<|S)z_ Returns a new array of given size with 32 bit integers bounded by given interval. rErr5)rGrHrIr r6rJs rrandom_bounded_uint32_fillrQs ((4u %C axD!ACF" J D!A;v..CF" JD!A9&#FFCF" Jrcdd}d}tj||}|dk(r!tj|D]}|||< |S|dk(r5tj|D]}t|||\} }}|| z||<|Stj|D]}t ||||\} }}|| z||<|S)z_ Returns a new array of given size with 16 bit integers bounded by given interval. rrEr2)rGrHrIr'r3 rrKrrLrFr rrMrNvals rrandom_bounded_uint16_fillrUs C D ((4u %C axD!ACF" J D!A,VT3?NCs3YCF" J D!A.vs/3S: Cs3YCF " Jrcdd}d}tj||}|dk(r!tj|D]}|||< |S|dk(r5tj|D]}t|||\} }}|| z||<|Stj|D]}t ||||\} }}|| z||<|S)z^ Returns a new array of given size with 8 bit integers bounded by given interval. rrEr))rGrHrIr%r0rSs rrandom_bounded_uint8_fillrWs C D ((4u %C axD!ACF" J D!A+FD#>NCs3YCF" J D!A-fc.2C9 Cs3YCF " Jrcd}d}tj||}tj|D]}t|||||\} }}|| z||<|S)z@ Returns a new array of given size with boolean values. rrE)rGrHrIr!rSs rrandom_bounded_bool_fillrY/s] C D ((4u %C ZZ .vsCsKT3sA Jrc||kr td|dkDrWt|}|s|tdz}t|}|dkDr t|}||kDr td||kDr tdy||kDr td||kDr tdy)zS Check that low and high are within the bounds for the given datatype. zlow is out of boundsrrzhigh is out of boundsz*low is greater than high in given intervalN) ValueErrorr)rKhighendpoint lower_bound upper_bounds r_randint_arg_checkr`=s [/00  axd| F1I D[) 7+C + 45 5 :IJ J  + 45 5 :IJ J rcB|dk(ryt|}tt|}|dkrDtt||z}||kDrOtt||z}||kDrt|St||z}||kDrt||z}||kDrt|S)Nrr5)rrr r )rmax_valrvalues rrandom_intervalrd\s1 WoG (7# $D:{6*+d2go;v./$6Ego %= F#d*go'$.Ego %=r) numpyrGnumbarrrrnumba.core.extendingrnumba.np.random._constantsrr r r numba.np.random.generator_corer r rr!r%r'r0r3r6rCrOrQrUrWrYr`rdrrrksO//1??C  ' '!!""  F  F4D6&64  KK<r