rh!(0ddlZddlmZddlmZmZmZmZddl m Z m Z edZ edZ edZedZed Zed Zed Zed Zed ZedZedZedZedZedZedZy)N)register_jitable) UINT32_MAX UINT64_MAX UINT16_MAX UINT8_MAX) next_uint32 next_uint64ctj|}||dz z}||dz z}||dz z}||dz z}||dz z}||dz z}|S)N )npuint64)maxmasks \/var/www/html/engine/venv/lib/python3.12/site-packages/numba/np/random/new_random_methods.pygen_maskr sh 99S>DDAIDDAIDDAIDDAIDDBJDDBJD Kc`|dk(r|||fS|st|}d}n |dz}|dz}|dzdk7||fS)Nrr )r)bitgenoffrngbcntbufs rbuffered_bounded_boolrsN qD#~ &!     1WNT3 &&rcf|st|}d}n |dz}|dz}tj|||fS)Nrr )rruint8rrrs rbuffered_uint8r$'s; &!     88C=$ ##rcf|st|}d}n |dz}|dz}tj|||fS)Nr r)rruint16r#s rbuffered_uint16r'3s; &!     99S>4 $$rctj|tjdz}|dk7sJt|||\}}}tj||z}|dz}||krXtjt|z |z}||kr4t|||\}}}tj||z}|dz}||kr4|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. r r)rr"r$r&r rrrrrng_exclnmleftover thresholds rbuffered_bounded_lemire_uint8r0Asxx}rxx{*H 4KK"&$4LAtS !h,A4xH8hhy)C/8; )#)&$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. rFrr6)rrHrIrr7rJs 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. rrFr2)rrHrIr'r4 rrKrrLrGrrrMrNvals 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. rrFr))rrHrIr$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. rrF)rrHrIrrSs rrandom_bounded_bool_fillrY.sb C D ((4u %C ZZ .vsCsKT3sA Jrcl||kr td|dkDrtj|}|s|tjdz}tj|}|dkDrtj|}||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 boundsrr zhigh is out of boundsz*low is greater than high in given intervalN) ValueErrorrr)rKhighendpoint lower_bound upper_bounds r_randint_arg_checkr`<s [/00  axyy BIIaL Dii , 7))C.C + 45 5 :IJ J  + 45 5 :IJ J rc|dk(rytj|}tjt|}|dkrNtjt||z}||kDrOtjt||z}||kDr'n't ||z}||kDrt ||z}||kDrtj|S)Nrr6)rrrrr )rmax_valrvalues rrandom_intervalrd[s1 ii G 99Xg& 'D: +f-.5goIIk&12T9EgoF#d*go'$.Ego 99U r)numpyrnumba.core.extendingrnumba.np.random._constantsrrrrnumba.np.random.generator_corerr rrr$r'r0r4r7rDrOrQrUrWrYr`rdrrrjsI1??C  ' '$$%%  F  F4D6&64  KK<r