g'ddlZddlmZmZmZmZddlmZddl m Z m Z m Z m Z ddlmZmZedZedZedZed Zed Zed Zed Zed ZedZedZedZedZedZedZedZdS)N)uint64uint32uint16uint8)register_jitable) UINT32_MAX UINT64_MAX UINT16_MAX UINT8_MAX) next_uint32 next_uint64ct|}||dz z}||dz z}||dz z}||dz z}||dz z}||dz z}|S)N )r)maxmasks T/granary/edev/quantum/lib/python3.11/site-packages/numba/np/random/random_methods.pygen_maskrsf #;;DDAIDDAIDDAIDDAIDDBJDDBJD Kcl|dkr|||fS|st|}d}n |dz}|dz}|dzdk||fS)Nrr)r )bitgenoffrngbcntbufs rbuffered_bounded_boolr!sZ qD#~ &!!     1WNT3 &&rcb|st|}d}n |dz}|dz}t|||fS)Nrr)r rrrr s rbuffered_uint8r%(sC &!!     ::tS  rcb|st|}d}n |dz}|dz}t|||fS)Nrr)r rr$s rbuffered_uint16r'4sC &!!     #;;c !!rcht|tdz}|dksJt|||\}}}t||z}|dz}||krRtt|z |z}||kr2t|||\}}}t||z}|dz}||k2|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_uint8r0BsSzzE!HH$H 4KKKK"&$44LAtSq8|A4xH8I&&,8 )##)&$<.>>A:~H)## Grc t|tdz}|dksJt|}t|t|z}||krGt|z |z}||kr4t|}t|t|z}||k4|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{{VAYY&H % % % % %FAayy6(+++H8#%1 )##F##Aayy6(#3#33H)## VJ   B bB6*---IBI iB Y"(#A VJ   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. dtyperr5r8)npemptyndindexr r6r rCrlowrsizerFoutis 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. rErr5)rGrHrIr r6rJs 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. rrEr2)rGrHrIr'r3 rrKrrLrFr rrMrNvals 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. rrEr))rGrHrIr%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. rrE)rGrHrIr!rSs rrandom_bounded_bool_fillrY/sj C D (4u % % %C Z  .vsCsKKT3sA Jrcv||krtd|dkrst|}|s|tdz}t|}|dkrt|}||krtd||krtddS||krtd||krtddS)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 [/000  axxd||  F1II D[)) 77++C +  455 5 ::IJJ J : +  455 5 ::IJJ J :rc|dkrdSt|}tt|}|dkrKtt||z}||kr%tt||z}||k%n0t||z}||krt||z}||kt|S)Nrr5)rrr r )rmax_valrvalues rrandom_intervalrd\s1 qWooG (7## $ $D:{6**++d2goo;v..//$6EgooF##d*goo''$.Egoo %==r) numpyrGnumbarrrrnumba.core.extendingrnumba.np.random._constantsrr r r numba.np.random.generator_corer r rr!r%r'r0r3r6rCrOrQrUrWrYr`rdrrrksO////////////111111????????????CCCCCCCC  ' ' '!!!"""   F   F4D6&64   KKK<r