J/Ph!(ddlZddlmZddlmZmZmZmZddl m Z m Z edZ edZ edZedZed Zed Zed Zed Zed ZedZedZedZedZedZedZdS)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 b/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numba/np/random/new_random_methods.pygen_maskr sh 9S>>DDAIDDAIDDAIDDAIDDBJDDBJD Kcl|dkr|||fS|st|}d}n |dz}|dz}|dzdk||fS)Nrr )r)bitgenoffrngbcntbufs rbuffered_bounded_boolrsZ qD#~ &!!     1WNT3 &&rcl|st|}d}n |dz}|dz}tj|||fS)Nrr )rruint8rrrs rbuffered_uint8r$'sE &!!     8C==$ ##rcl|st|}d}n |dz}|dz}tj|||fS)Nr r)rruint16r#s rbuffered_uint16r'3sE &!!     9S>>4 $$rctj|tjdz}|dksJt|||\}}}tj||z}|dz}||kr\tjt|z |z}||kr7t|||\}}}tj||z}|dz}||k7|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_uint8r0Asx}}rx{{*H 4KKKK"&$44LAtS !h,A4xH8hy))C/8; )##)&$<t|}tj|tj|z}||k>|tjdz}|dz }|tjdz}|dz } ||z} ||z| dz z} | tjdz} | dz } | || zz } || z| z| dz z}|S)zr Generates a random unsigned 64 bit integer bounded within a given interval using Lemire's rejection. r r6r)rrr r)rrr+xr.r/x0x1 rng_excl0 rng_excl1w0tw1w2m1s rbounded_lemire_uint64rDsR y~~ ! ,H % % % % %FAy||bi111H8#%1 )##F##Ay||bi&9&99H)## RYz " " "B bB29Z000IBI iB Y"(#A RYz " " "B bB"y.B i" b )B Irctj||}|dkrtj|D]}|||<n|dkra|dkr-tj|D]}|t|z||<ntj|D]}|t ||z||<n`|dkr-tj|D]}|t |z||<n-tj|D]}|t ||z||<|S)z_ Returns a new array of given size with 64 bit integers bounded by given interval. dtyperr6r9)remptyndindexrr7r rDrlowrsizerGoutis rrandom_bounded_uint64_fillrOsS (4u % % %C axxD!!  ACFF    :  Z%% 3 3{6222A 3Z%% K K=fcJJJA K # # #D!! / /A;v...CFF /D!! > >A0===CFF Jrc8tj||}|dkrtj|D]}|||<n`|dkr-tj|D]}|t|z||<n-tj|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 axxD!!  ACFF   D!! / /A;v...CFF /D!! G GA9&#FFFCFF Jrc`d}d}tj||}|dkrtj|D]}|||<np|dkr5tj|D]}t|||\} }}|| z||< n5tj|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 axxD!!  ACFF  D!!  A,VT3??NCs3YCFF  D!!  A.vs/3S:: Cs3YCFF Jrc`d}d}tj||}|dkrtj|D]}|||<np|dkr5tj|D]}t|||\} }}|| z||< n5tj|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 axxD!!  ACFF  D!!  A+FD#>>NCs3YCFF D!!  A-fc.2C99 Cs3YCFF 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.sj C D (4u % % %C Z  .vsCsKKT3sA Jrc||krtd|dkrtj|}|s|tjdz}tj|}|dkrtj|}||krtd||krtddS||krtd||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) ValueErrorrr)rKhighendpoint lower_bound upper_bounds r_randint_arg_checkr`<s [/000  axxy ! BIaLL Di ,, 77)C..C +  455 5 ::IJJ J : +  455 5 ::IJJ J :rc|dkrdStj|}tjt|}|dkrUtjt||z}||kr*tjt||z}||k*n0t ||z}||krt ||z}||ktj|S)Nrr6)rrrrr )rmax_valrvalues rrandom_intervalrd[s1 qi  G 9Xg&& ' 'D: +f--..5gooIk&1122T9EgooF##d*goo''$.Egoo 9U  r)numpyrnumba.core.extendingrnumba.np.random._constantsrrrrnumba.np.random.generator_corerr rrr$r'r0r4r7rDrOrQrUrWrYr`rdrrrjs+111111????????????CCCCCCCC  ' ' '$$$%%%   F   F4D6&64   KKK<r