codekingpro/portable-devtools

1#cython: language_level=3
2
3from libc.stdint cimport uint32_t, uint64_t, int32_t, int64_t
4
5import numpy as np
6cimport numpy as np
7
8from numpy.random cimport bitgen_t
9
10cdef double POISSON_LAM_MAX
11cdef double LEGACY_POISSON_LAM_MAX
12cdef uint64_t MAXSIZE
13
14cdef enum ConstraintType:
15    CONS_NONE
16    CONS_NON_NEGATIVE
17    CONS_POSITIVE
18    CONS_POSITIVE_NOT_NAN
19    CONS_BOUNDED_0_1
20    CONS_BOUNDED_GT_0_1
21    CONS_BOUNDED_LT_0_1
22    CONS_GT_1
23    CONS_GTE_1
24    CONS_POISSON
25    LEGACY_CONS_POISSON
26    LEGACY_CONS_NON_NEGATIVE_INBOUNDS_LONG
27
28ctypedef ConstraintType constraint_type
29ctypedef fused double_or_int64:
30    double
31    int64_t
32
33cdef object benchmark(bitgen_t *bitgen, object lock, Py_ssize_t cnt, object method)
34cdef object random_raw(bitgen_t *bitgen, object lock, object size, object output)
35cdef object prepare_cffi(bitgen_t *bitgen)
36cdef object prepare_ctypes(bitgen_t *bitgen)
37cdef int check_constraint(double_or_int64 val, object name, constraint_type cons) except -1
38cdef int check_array_constraint(np.ndarray val, object name, constraint_type cons) except -1
39
40cdef extern from "include/aligned_malloc.h":
41    cdef void *PyArray_realloc_aligned(void *p, size_t n)
42    cdef void *PyArray_malloc_aligned(size_t n)
43    cdef void *PyArray_calloc_aligned(size_t n, size_t s)
44    cdef void PyArray_free_aligned(void *p)
45
46ctypedef void (*random_double_fill)(bitgen_t *state, np.npy_intp count, double* out)  noexcept nogil
47ctypedef double (*random_double_0)(void *state)  noexcept nogil
48ctypedef double (*random_double_1)(void *state, double a)  noexcept nogil
49ctypedef double (*random_double_2)(void *state, double a, double b)  noexcept nogil
50ctypedef double (*random_double_3)(void *state, double a, double b, double c)  noexcept nogil
51
52ctypedef void (*random_float_fill)(bitgen_t *state, np.npy_intp count, float* out)  noexcept nogil
53ctypedef float (*random_float_0)(bitgen_t *state)  noexcept nogil
54ctypedef float (*random_float_1)(bitgen_t *state, float a)  noexcept nogil
55
56ctypedef int64_t (*random_uint_0)(void *state)  noexcept nogil
57ctypedef int64_t (*random_uint_d)(void *state, double a)  noexcept nogil
58ctypedef int64_t (*random_uint_dd)(void *state, double a, double b)  noexcept nogil
59ctypedef int64_t (*random_uint_di)(void *state, double a, uint64_t b)  noexcept nogil
60ctypedef int64_t (*random_uint_i)(void *state, int64_t a)  noexcept nogil
61ctypedef int64_t (*random_uint_iii)(void *state, int64_t a, int64_t b, int64_t c)  noexcept nogil
62
63ctypedef uint32_t (*random_uint_0_32)(bitgen_t *state)  noexcept nogil
64ctypedef uint32_t (*random_uint_1_i_32)(bitgen_t *state, uint32_t a)  noexcept nogil
65
66ctypedef int32_t (*random_int_2_i_32)(bitgen_t *state, int32_t a, int32_t b)  noexcept nogil
67ctypedef int64_t (*random_int_2_i)(bitgen_t *state, int64_t a, int64_t b)  noexcept nogil
68
69cdef double kahan_sum(double *darr, np.npy_intp n) noexcept
70
71cdef inline double uint64_to_double(uint64_t rnd) noexcept nogil:
72    return (rnd >> 11) * (1.0 / 9007199254740992.0)
73
74cdef object double_fill(void *func, bitgen_t *state, object size, object lock, object out)
75
76cdef object float_fill(void *func, bitgen_t *state, object size, object lock, object out)
77
78cdef object float_fill_from_double(void *func, bitgen_t *state, object size, object lock, object out)
79
80cdef object wrap_int(object val, object bits)
81
82cdef np.ndarray int_to_array(object value, object name, object bits, object uint_size)
83
84cdef validate_output_shape(iter_shape, np.ndarray output)
85
86cdef object cont(void *func, void *state, object size, object lock, int narg,
87                 object a, object a_name, constraint_type a_constraint,
88                 object b, object b_name, constraint_type b_constraint,
89                 object c, object c_name, constraint_type c_constraint,
90                 object out)
91
92cdef object disc(void *func, void *state, object size, object lock,
93                 int narg_double, int narg_int64,
94                 object a, object a_name, constraint_type a_constraint,
95                 object b, object b_name, constraint_type b_constraint,
96                 object c, object c_name, constraint_type c_constraint)
97
98cdef object cont_f(void *func, bitgen_t *state, object size, object lock,
99                   object a, object a_name, constraint_type a_constraint,
100                   object out)
101
102cdef object cont_broadcast_3(void *func, void *state, object size, object lock,
103                             np.ndarray a_arr, object a_name, constraint_type a_constraint,
104                             np.ndarray b_arr, object b_name, constraint_type b_constraint,
105                             np.ndarray c_arr, object c_name, constraint_type c_constraint)
106
107cdef object discrete_broadcast_iii(void *func, void *state, object size, object lock,
108                                   np.ndarray a_arr, object a_name, constraint_type a_constraint,
109                                   np.ndarray b_arr, object b_name, constraint_type b_constraint,
110                                   np.ndarray c_arr, object c_name, constraint_type c_constraint)
111