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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
/* -*- mode: c++; c-basic-offset: 4 -*- */
/* Small utilities that are shared by most extension modules. */
#ifndef MPLUTILS_H
#define MPLUTILS_H
#define PY_SSIZE_T_CLEAN
#include <Python.h>
#include <stdint.h>
#ifdef _POSIX_C_SOURCE
# undef _POSIX_C_SOURCE
#endif
#ifndef _AIX
#ifdef _XOPEN_SOURCE
# undef _XOPEN_SOURCE
#endif
#endif
// Prevent multiple conflicting definitions of swab from stdlib.h and unistd.h
#if defined(__sun) || defined(sun)
#if defined(_XPG4)
#undef _XPG4
#endif
#if defined(_XPG3)
#undef _XPG3
#endif
#endif
inline int mpl_round_to_int(double v)
{
return (int)(v + ((v >= 0.0) ? 0.5 : -0.5));
}
inline double mpl_round(double v)
{
return (double)mpl_round_to_int(v);
}
// 'kind' codes for paths.
enum {
STOP = 0,
MOVETO = 1,
LINETO = 2,
CURVE3 = 3,
CURVE4 = 4,
CLOSEPOLY = 0x4f
};
const size_t NUM_VERTICES[] = { 1, 1, 1, 2, 3, 1 };
inline int prepare_and_add_type(PyTypeObject *type, PyObject *module)
{
if (PyType_Ready(type)) {
return -1;
}
char const* ptr = strrchr(type->tp_name, '.');
if (!ptr) {
PyErr_SetString(PyExc_ValueError, "tp_name should be a qualified name");
return -1;
}
if (PyModule_AddObject(module, ptr + 1, (PyObject *)type)) {
return -1;
}
return 0;
}
#ifdef __cplusplus // not for macosx.m
// Check that array has shape (N, d1) or (N, d1, d2). We cast d1, d2 to longs
// so that we don't need to access the NPY_INTP_FMT macro here.
template<typename T>
inline bool check_trailing_shape(T array, char const* name, long d1)
{
if (array.dim(1) != d1) {
PyErr_Format(PyExc_ValueError,
"%s must have shape (N, %ld), got (%ld, %ld)",
name, d1, array.dim(0), array.dim(1));
return false;
}
return true;
}
template<typename T>
inline bool check_trailing_shape(T array, char const* name, long d1, long d2)
{
if (array.dim(1) != d1 || array.dim(2) != d2) {
PyErr_Format(PyExc_ValueError,
"%s must have shape (N, %ld, %ld), got (%ld, %ld, %ld)",
name, d1, d2, array.dim(0), array.dim(1), array.dim(2));
return false;
}
return true;
}
#endif
#endif
|
