:j'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 s/home/mcse/projects/srt_converter/srt-converter-venv/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_fillrOs4 ((4u %C axD! ACF " J  : ZZ% 3{622A 3 JZZ% K=fcJJA K 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! GA9&#FFCF G 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/sb 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