/ lib / numpy / random / _bounded_integers.pxd
_bounded_integers.pxd
 1  from libc.stdint cimport (uint8_t, uint16_t, uint32_t, uint64_t,
 2                            int8_t, int16_t, int32_t, int64_t, intptr_t)
 3  import numpy as np
 4  cimport numpy as np
 5  ctypedef np.npy_bool bool_t
 6  
 7  from numpy.random cimport bitgen_t
 8  
 9  cdef inline uint64_t _gen_mask(uint64_t max_val) nogil:
10      """Mask generator for use in bounded random numbers"""
11      # Smallest bit mask >= max
12      cdef uint64_t mask = max_val
13      mask |= mask >> 1
14      mask |= mask >> 2
15      mask |= mask >> 4
16      mask |= mask >> 8
17      mask |= mask >> 16
18      mask |= mask >> 32
19      return mask
20  
21  cdef object _rand_uint64(object low, object high, object size, bint use_masked, bint closed, bitgen_t *state, object lock)
22  cdef object _rand_uint32(object low, object high, object size, bint use_masked, bint closed, bitgen_t *state, object lock)
23  cdef object _rand_uint16(object low, object high, object size, bint use_masked, bint closed, bitgen_t *state, object lock)
24  cdef object _rand_uint8(object low, object high, object size, bint use_masked, bint closed, bitgen_t *state, object lock)
25  cdef object _rand_bool(object low, object high, object size, bint use_masked, bint closed, bitgen_t *state, object lock)
26  cdef object _rand_int64(object low, object high, object size, bint use_masked, bint closed, bitgen_t *state, object lock)
27  cdef object _rand_int32(object low, object high, object size, bint use_masked, bint closed, bitgen_t *state, object lock)
28  cdef object _rand_int16(object low, object high, object size, bint use_masked, bint closed, bitgen_t *state, object lock)
29  cdef object _rand_int8(object low, object high, object size, bint use_masked, bint closed, bitgen_t *state, object lock)