1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
#if !defined(__cfunc__)
#define __cfunc__ 1
typedef PyObject *(*CFUNCasPyValue)(void *);
typedef int (*UFUNC)(long, long, long, void **, long*);
/* typedef void (*CFUNC_2ARG)(long, void *, void *); */
/* typedef void (*CFUNC_3ARG)(long, void *, void *, void *); */
typedef int (*CFUNCfromPyValue)(PyObject *, void *);
typedef int (*CFUNC_STRIDE_CONV_FUNC)(long, long, maybelong *,
void *, long, maybelong*, void *, long, maybelong *);
typedef int (*CFUNC_STRIDED_FUNC)(PyObject *, long, PyArrayObject **,
char **data);
#define MAXARRAYS 16
typedef enum {
CFUNC_UFUNC,
CFUNC_STRIDING,
CFUNC_NSTRIDING,
CFUNC_AS_PY_VALUE,
CFUNC_FROM_PY_VALUE
} eCfuncType;
typedef struct {
char *name;
void *fptr; /* Pointer to "un-wrapped" c function */
eCfuncType type; /* UFUNC, STRIDING, AsPyValue, FromPyValue */
Bool chkself; /* CFUNC does own alignment/bounds checking */
Bool align; /* CFUNC requires aligned buffer pointers */
Int8 wantIn, wantOut; /* required input/output arg counts. */
Int8 sizes[MAXARRAYS]; /* array of align/itemsizes. */
Int8 iters[MAXARRAYS]; /* array of element counts. 0 --> niter. */
} CfuncDescriptor;
typedef struct {
PyObject_HEAD
CfuncDescriptor descr;
} CfuncObject;
#define SELF_CHECKED_CFUNC_DESCR(name, type) \
static CfuncDescriptor name##_descr = { #name, (void *) name, type, 1 }
#define CHECK_ALIGN 1
#define CFUNC_DESCR(name, type, align, iargs, oargs, s1, s2, s3, i1, i2, i3) \
static CfuncDescriptor name##_descr = \
{ #name, (void *)name, type, 0, align, iargs, oargs, {s1, s2, s3}, {i1, i2, i3} }
#define UFUNC_DESCR1(name, s1) \
CFUNC_DESCR(name, CFUNC_UFUNC, CHECK_ALIGN, 0, 1, s1, 0, 0, 0, 0, 0)
#define UFUNC_DESCR2(name, s1, s2) \
CFUNC_DESCR(name, CFUNC_UFUNC, CHECK_ALIGN, 1, 1, s1, s2, 0, 0, 0, 0)
#define UFUNC_DESCR3(name, s1, s2, s3) \
CFUNC_DESCR(name, CFUNC_UFUNC, CHECK_ALIGN, 2, 1, s1, s2, s3, 0, 0, 0)
#define UFUNC_DESCR3sv(name, s1, s2, s3) \
CFUNC_DESCR(name, CFUNC_UFUNC, CHECK_ALIGN, 2, 1, s1, s2, s3, 1, 0, 0)
#define UFUNC_DESCR3vs(name, s1, s2, s3) \
CFUNC_DESCR(name, CFUNC_UFUNC, CHECK_ALIGN, 2, 1, s1, s2, s3, 0, 1, 0)
#define STRIDING_DESCR2(name, align, s1, s2) \
CFUNC_DESCR(name, CFUNC_STRIDING, align, 1, 1, s1, s2, 0, 0, 0, 0)
#define NSTRIDING_DESCR1(name) \
CFUNC_DESCR(name, CFUNC_NSTRIDING, 0, 0, 1, 0, 0, 0, 0, 0, 0)
#define NSTRIDING_DESCR2(name) \
CFUNC_DESCR(name, CFUNC_NSTRIDING, 0, 1, 1, 0, 0, 0, 0, 0, 0)
#define NSTRIDING_DESCR3(name) \
CFUNC_DESCR(name, CFUNC_NSTRIDING, 0, 2, 1, 0, 0, 0, 0, 0, 0)
#endif
|
