|
14 | 14 |
|
15 | 15 | using namespace dynd; |
16 | 16 |
|
17 | | - |
18 | 17 | // Unpack and convert a single element to a PyObject. |
19 | | -template <typename T, typename std::enable_if_t<std::is_signed<T>::value && std::numeric_limits<T>::max() <= INT64_MAX, int> = 0> |
20 | | -inline PyObject * |
21 | | -unpack_single(const char *data) |
| 18 | +template <typename T, |
| 19 | + typename std::enable_if< |
| 20 | + std::is_integral<T>::value && std::is_signed<T>::value && sizeof(T) <= sizeof(long long), int>::type = 0> |
| 21 | +inline PyObject *unpack_single(const char *data) |
22 | 22 | { |
23 | | - return PyLong_FromLongLong(*reinterpret_cast<const T *>(data)); |
| 23 | + return PyLong_FromLongLong(*reinterpret_cast<const T *>(data)); |
24 | 24 | } |
25 | 25 |
|
26 | | -template <typename T, std::enable_if_t<std::is_unsigned<T>::value && std::numeric_limits<T>::max() <= UINT64_MAX, int> = 0> |
27 | | -inline PyObject * |
28 | | -unpack_single(const char *data) |
| 26 | +template <typename T, typename std::enable_if_t<std::is_integral<T>::value && std::is_unsigned<T>::value && |
| 27 | + sizeof(T) <= sizeof(unsigned long long), |
| 28 | + int> = 0> |
| 29 | +inline PyObject *unpack_single(const char *data) |
29 | 30 | { |
30 | | - return PyLong_FromUnsignedLongLong(*reinterpret_cast<const T *>(data)); |
| 31 | + return PyLong_FromUnsignedLongLong(*reinterpret_cast<const T *>(data)); |
31 | 32 | } |
32 | 33 |
|
33 | 34 | template <typename T, typename std::enable_if_t<std::is_same<T, bool1>::value, int> = 0> |
34 | | -inline PyObject * |
35 | | -unpack_single(const char *data) |
| 35 | +inline PyObject *unpack_single(const char *data) |
36 | 36 | { |
37 | | - return PyBool_FromLong(*reinterpret_cast<const T *>(data)); |
| 37 | + return PyBool_FromLong(*reinterpret_cast<const T *>(data)); |
38 | 38 | } |
39 | 39 |
|
40 | 40 | template <typename T, typename std::enable_if_t<std::is_same<T, float64>::value, int> = 0> |
41 | | -inline PyObject * |
42 | | -unpack_single(const char *data) |
| 41 | +inline PyObject *unpack_single(const char *data) |
43 | 42 | { |
44 | | - return PyFloat_FromDouble(*reinterpret_cast<const T *>(data)); |
| 43 | + return PyFloat_FromDouble(*reinterpret_cast<const T *>(data)); |
45 | 44 | } |
46 | 45 |
|
47 | 46 | template <typename T, typename std::enable_if_t<std::is_same<T, complex128>::value, int> = 0> |
48 | | -inline PyObject * |
49 | | -unpack_single(const char *data) |
| 47 | +inline PyObject *unpack_single(const char *data) |
50 | 48 | { |
51 | | - complex128 c = *reinterpret_cast<const T *>(data); |
52 | | - return PyComplex_FromDoubles(c.real(), c.imag()); |
| 49 | + complex128 c = *reinterpret_cast<const T *>(data); |
| 50 | + return PyComplex_FromDoubles(c.real(), c.imag()); |
53 | 51 | } |
54 | 52 |
|
55 | 53 | template <typename T, typename std::enable_if_t<std::is_same<T, std::string>::value, int> = 0> |
56 | | -inline PyObject * |
57 | | -unpack_single(const char *data) |
| 54 | +inline PyObject *unpack_single(const char *data) |
58 | 55 | { |
59 | | - const std::string &s = *reinterpret_cast<const T *>(data); |
60 | | - return PyUnicode_FromString(s.data()); |
| 56 | + const std::string &s = *reinterpret_cast<const T *>(data); |
| 57 | + return PyUnicode_FromString(s.data()); |
61 | 58 | } |
62 | 59 |
|
63 | 60 | template <typename T, typename std::enable_if_t<std::is_same<T, ndt::type>::value, int> = 0> |
64 | | -inline PyObject * |
65 | | -unpack_single(const char *data) |
| 61 | +inline PyObject *unpack_single(const char *data) |
66 | 62 | { |
67 | | - return pydynd::type_from_cpp(*reinterpret_cast<const T *>(data)); |
| 63 | + return pydynd::type_from_cpp(*reinterpret_cast<const T *>(data)); |
68 | 64 | } |
69 | 65 |
|
70 | | - |
71 | 66 | // Convert a single element to a PyObject. |
72 | | -template <typename T, typename std::enable_if_t<std::is_signed<T>::value && std::numeric_limits<T>::max() <= INT64_MAX, int> = 0> |
73 | | -inline PyObject * |
74 | | -convert_single(T value) |
| 67 | +template <typename T, |
| 68 | + typename std::enable_if_t< |
| 69 | + std::is_integral<T>::value && std::is_signed<T>::value && sizeof(T) <= sizeof(long long), int> = 0> |
| 70 | +inline PyObject *convert_single(T value) |
75 | 71 | { |
76 | | - return PyLong_FromLongLong(value); |
| 72 | + return PyLong_FromLongLong(value); |
77 | 73 | } |
78 | 74 |
|
79 | | -template <typename T, typename std::enable_if_t<std::is_unsigned<T>::value && std::numeric_limits<T>::max() <= UINT64_MAX, int> = 0> |
80 | | -inline PyObject * |
81 | | -convert_single(T value) |
| 75 | +template <typename T, typename std::enable_if_t<std::is_integral<T>::value && std::is_unsigned<T>::value && |
| 76 | + sizeof(T) <= sizeof(unsigned long long), |
| 77 | + int> = 0> |
| 78 | +inline PyObject *convert_single(T value) |
82 | 79 | { |
83 | | - return PyLong_FromUnsignedLongLong(value); |
| 80 | + return PyLong_FromUnsignedLongLong(value); |
84 | 81 | } |
85 | 82 |
|
86 | 83 | template <typename T, typename std::enable_if_t<std::is_same<T, bool1>::value, int> = 0> |
87 | | -inline PyObject * |
88 | | -convert_single(T value) |
| 84 | +inline PyObject *convert_single(T value) |
89 | 85 | { |
90 | | - return PyBool_FromLong(value); |
| 86 | + return PyBool_FromLong(value); |
91 | 87 | } |
92 | 88 |
|
93 | 89 | template <typename T, typename std::enable_if_t<std::is_same<T, float64>::value, int> = 0> |
94 | | -inline PyObject * |
95 | | -convert_single(T value) |
| 90 | +inline PyObject *convert_single(T value) |
96 | 91 | { |
97 | | - return PyFloat_FromDouble(value); |
| 92 | + return PyFloat_FromDouble(value); |
98 | 93 | } |
99 | 94 |
|
100 | 95 | template <typename T, typename std::enable_if_t<std::is_same<T, complex128>::value, int> = 0> |
101 | | -inline PyObject * |
102 | | -convert_single(T value) |
| 96 | +inline PyObject *convert_single(T value) |
103 | 97 | { |
104 | | - return PyComplex_FromDoubles(value.real(), value.imag()); |
| 98 | + return PyComplex_FromDoubles(value.real(), value.imag()); |
105 | 99 | } |
106 | 100 |
|
107 | 101 | template <typename T, typename std::enable_if_t<std::is_same<T, std::string>::value, int> = 0> |
108 | | -inline PyObject * |
109 | | -convert_single(const T &value) |
| 102 | +inline PyObject *convert_single(const T &value) |
110 | 103 | { |
111 | | - return PyUnicode_FromString(value.data()); |
| 104 | + return PyUnicode_FromString(value.data()); |
112 | 105 | } |
113 | 106 |
|
114 | 107 | template <typename T, typename std::enable_if_t<std::is_same<T, ndt::type>::value, int> = 0> |
115 | | -inline PyObject * |
116 | | -convert_single(const T &value) |
| 108 | +inline PyObject *convert_single(const T &value) |
117 | 109 | { |
118 | | - return pydynd::type_from_cpp(value); |
| 110 | + return pydynd::type_from_cpp(value); |
119 | 111 | } |
120 | 112 |
|
121 | | - |
122 | 113 | template <typename T> |
123 | 114 | PyObject *unpack_vector(const char *data) |
124 | 115 | { |
125 | | - auto &vec = *reinterpret_cast<const std::vector<T> *>(data); |
126 | | - PyObject *lst, *item; |
| 116 | + auto &vec = *reinterpret_cast<const std::vector<T> *>(data); |
| 117 | + PyObject *lst, *item; |
127 | 118 |
|
128 | | - lst = PyList_New(vec.size()); |
129 | | - if (lst == NULL) { |
130 | | - return NULL; |
131 | | - } |
| 119 | + lst = PyList_New(vec.size()); |
| 120 | + if (lst == NULL) { |
| 121 | + return NULL; |
| 122 | + } |
132 | 123 |
|
133 | | - for (size_t i = 0; i < vec.size(); ++i) { |
134 | | - item = convert_single<T>(vec[i]); |
135 | | - if (item == NULL) { |
136 | | - Py_DECREF(lst); |
137 | | - return NULL; |
138 | | - } |
139 | | - PyList_SET_ITEM(lst, i, item); |
| 124 | + for (size_t i = 0; i < vec.size(); ++i) { |
| 125 | + item = convert_single<T>(vec[i]); |
| 126 | + if (item == NULL) { |
| 127 | + Py_DECREF(lst); |
| 128 | + return NULL; |
140 | 129 | } |
| 130 | + PyList_SET_ITEM(lst, i, item); |
| 131 | + } |
141 | 132 |
|
142 | | - return lst; |
| 133 | + return lst; |
143 | 134 | } |
144 | 135 |
|
145 | 136 | template <typename T> |
146 | 137 | inline PyObject *unpack(bool is_vector, const char *data) |
147 | 138 | { |
148 | | - if (is_vector) { |
149 | | - return unpack_vector<T>(data); |
150 | | - } |
151 | | - else { |
152 | | - return unpack_single<T>(data); |
153 | | - } |
| 139 | + if (is_vector) { |
| 140 | + return unpack_vector<T>(data); |
| 141 | + } |
| 142 | + else { |
| 143 | + return unpack_single<T>(data); |
| 144 | + } |
154 | 145 | } |
155 | 146 |
|
156 | 147 | PyObject *from_type_property(const std::pair<ndt::type, const char *> &pair) |
|
0 commit comments