1 | /*\file FetchData.cpp:
|
---|
2 | * \brief: general I/O interface to fetch data in python
|
---|
3 | */
|
---|
4 |
|
---|
5 | #ifdef HAVE_CONFIG_H
|
---|
6 | #include <config.h>
|
---|
7 | #else
|
---|
8 | #error "Cannot compile with HAVE_CONFIG_H symbol! run configure first!"
|
---|
9 | #endif
|
---|
10 |
|
---|
11 | #define PY_ARRAY_UNIQUE_SYMBOL PythonIOSymbol
|
---|
12 | #define NO_IMPORT
|
---|
13 |
|
---|
14 | #include "./pythonio.h"
|
---|
15 | #include "../../c/include/include.h"
|
---|
16 | #include "../../c/shared/shared.h"
|
---|
17 |
|
---|
18 | /*Primitive data types*/
|
---|
19 | /*FUNCTION FetchData(double* pscalar,PyObject* py_float){{{*/
|
---|
20 | void FetchData(double* pscalar,PyObject* py_float){
|
---|
21 |
|
---|
22 | double dscalar;
|
---|
23 |
|
---|
24 | /*return internal value: */
|
---|
25 | if (PyFloat_Check(py_float))
|
---|
26 | dscalar=PyFloat_AsDouble(py_float);
|
---|
27 | else if (PyInt_Check(py_float))
|
---|
28 | dscalar=(double)PyInt_AsLong(py_float);
|
---|
29 | else if (PyLong_Check(py_float))
|
---|
30 | dscalar=PyLong_AsDouble(py_float);
|
---|
31 | else if (PyBool_Check(py_float))
|
---|
32 | dscalar=(double)PyLong_AsLong(py_float);
|
---|
33 | else if (PyTuple_Check(py_float) && (int)PyTuple_Size(py_float)==1)
|
---|
34 | FetchData(&dscalar,PyTuple_GetItem(py_float,(Py_ssize_t)0));
|
---|
35 | else if (PyList_Check(py_float) && (int)PyList_Size(py_float)==1)
|
---|
36 | FetchData(&dscalar,PyList_GetItem(py_float,(Py_ssize_t)0));
|
---|
37 | else
|
---|
38 | _error_("unrecognized float type in input!");
|
---|
39 |
|
---|
40 | /*output: */
|
---|
41 | *pscalar=dscalar;
|
---|
42 | }
|
---|
43 | /*}}}*/
|
---|
44 | /*FUNCTION FetchData(int* pscalar,PyObject* py_long){{{*/
|
---|
45 | void FetchData(int* pscalar, PyObject* py_long){
|
---|
46 |
|
---|
47 | int iscalar;
|
---|
48 |
|
---|
49 | /*return internal value: */
|
---|
50 | if (PyInt_Check(py_long))
|
---|
51 | iscalar=(int)PyInt_AsLong(py_long);
|
---|
52 | else if (PyLong_Check(py_long))
|
---|
53 | iscalar=(int)PyLong_AsLong(py_long);
|
---|
54 | else if (PyFloat_Check(py_long))
|
---|
55 | iscalar=(int)PyFloat_AsDouble(py_long);
|
---|
56 | else if (PyBool_Check(py_long))
|
---|
57 | iscalar=(int)PyLong_AsLong(py_long);
|
---|
58 | else if (PyTuple_Check(py_long) && (int)PyTuple_Size(py_long)==1)
|
---|
59 | FetchData(&iscalar,PyTuple_GetItem(py_long,(Py_ssize_t)0));
|
---|
60 | else if (PyList_Check(py_long) && (int)PyList_Size(py_long)==1)
|
---|
61 | FetchData(&iscalar,PyList_GetItem(py_long,(Py_ssize_t)0));
|
---|
62 | else
|
---|
63 | _error_("unrecognized long type in input!");
|
---|
64 |
|
---|
65 | /*output: */
|
---|
66 | *pscalar=iscalar;
|
---|
67 | }
|
---|
68 | /*}}}*/
|
---|
69 | /*FUNCTION FetchData(bool* pscalar,PyObject* py_boolean){{{*/
|
---|
70 | void FetchData(bool* pscalar,PyObject* py_boolean){
|
---|
71 |
|
---|
72 | bool bscalar;
|
---|
73 |
|
---|
74 | /*return internal value: */
|
---|
75 | if (PyBool_Check(py_boolean))
|
---|
76 | bscalar=(bool)PyLong_AsLong(py_boolean);
|
---|
77 | else if (PyInt_Check(py_boolean))
|
---|
78 | bscalar=(bool)PyInt_AsLong(py_boolean);
|
---|
79 | else if (PyLong_Check(py_boolean))
|
---|
80 | bscalar=(bool)PyLong_AsLong(py_boolean);
|
---|
81 | else if (PyTuple_Check(py_boolean) && (int)PyTuple_Size(py_boolean)==1)
|
---|
82 | FetchData(&bscalar,PyTuple_GetItem(py_boolean,(Py_ssize_t)0));
|
---|
83 | else if (PyList_Check(py_boolean) && (int)PyList_Size(py_boolean)==1)
|
---|
84 | FetchData(&bscalar,PyList_GetItem(py_boolean,(Py_ssize_t)0));
|
---|
85 | else
|
---|
86 | _error_("unrecognized boolean type in input!");
|
---|
87 |
|
---|
88 | /*output: */
|
---|
89 | *pscalar=bscalar;
|
---|
90 | }
|
---|
91 | /*}}}*/
|
---|
92 | /*FUNCTION FetchData(double** pmatrix,int* pM, int* pN, PyObject* py_matrix){{{*/
|
---|
93 | void FetchData(double** pmatrix,int* pM,int *pN,PyObject* py_matrix){
|
---|
94 |
|
---|
95 | /*output: */
|
---|
96 | double* dmatrix=NULL;
|
---|
97 | double* matrix=NULL;
|
---|
98 | int M,N;
|
---|
99 | int ndim;
|
---|
100 | npy_intp* dims=NULL;
|
---|
101 |
|
---|
102 | /*intermediary:*/
|
---|
103 | long* lmatrix=NULL;
|
---|
104 | bool* bmatrix=NULL;
|
---|
105 | int i;
|
---|
106 |
|
---|
107 | if (PyArray_Check((PyArrayObject*)py_matrix)) {
|
---|
108 | /*retrieve dimensions: */
|
---|
109 | ndim=PyArray_NDIM((const PyArrayObject*)py_matrix);
|
---|
110 | if (ndim==2) {
|
---|
111 | dims=PyArray_DIMS((PyArrayObject*)py_matrix);
|
---|
112 | M=dims[0]; N=dims[1];
|
---|
113 | }
|
---|
114 | else if (ndim==1) {
|
---|
115 | dims=PyArray_DIMS((PyArrayObject*)py_matrix);
|
---|
116 | M=dims[0]; N=1;
|
---|
117 | }
|
---|
118 | else
|
---|
119 | _error_("expecting an MxN matrix or M vector in input!");
|
---|
120 |
|
---|
121 | if (M && N) {
|
---|
122 | if (PyArray_TYPE((PyArrayObject*)py_matrix) == NPY_DOUBLE) {
|
---|
123 | /*retrieve internal value: */
|
---|
124 | dmatrix=(double*)PyArray_DATA((PyArrayObject*)py_matrix);
|
---|
125 |
|
---|
126 | /*copy matrix: */
|
---|
127 | matrix=xNew<double>(M*N);
|
---|
128 | memcpy(matrix,dmatrix,(M*N)*sizeof(double));
|
---|
129 | }
|
---|
130 |
|
---|
131 | else if (PyArray_TYPE((PyArrayObject*)py_matrix) == NPY_INT64) {
|
---|
132 | /*retrieve internal value: */
|
---|
133 | lmatrix=(long*)PyArray_DATA((PyArrayObject*)py_matrix);
|
---|
134 |
|
---|
135 | /*transform into double matrix: */
|
---|
136 | matrix=xNew<double>(M*N);
|
---|
137 | for(i=0;i<M*N;i++)matrix[i]=(double)lmatrix[i];
|
---|
138 | }
|
---|
139 |
|
---|
140 | else if (PyArray_TYPE((PyArrayObject*)py_matrix) == NPY_BOOL) {
|
---|
141 | /*retrieve internal value: */
|
---|
142 | bmatrix=(bool*)PyArray_DATA((PyArrayObject*)py_matrix);
|
---|
143 |
|
---|
144 | /*transform into double matrix: */
|
---|
145 | matrix=xNew<double>(M*N);
|
---|
146 | for(i=0;i<M*N;i++)matrix[i]=(double)bmatrix[i];
|
---|
147 | }
|
---|
148 |
|
---|
149 | else
|
---|
150 | _error_("unrecognized double pyarray type in input!");
|
---|
151 | }
|
---|
152 | else
|
---|
153 | matrix=NULL;
|
---|
154 | }
|
---|
155 |
|
---|
156 | else {
|
---|
157 | M=1;
|
---|
158 | N=1;
|
---|
159 | matrix=xNew<double>(M*N);
|
---|
160 | FetchData(&(matrix[0]),py_matrix);
|
---|
161 | }
|
---|
162 |
|
---|
163 | /*output: */
|
---|
164 | if(pM)*pM=M;
|
---|
165 | if(pN)*pN=N;
|
---|
166 | if(pmatrix)*pmatrix=matrix;
|
---|
167 | }
|
---|
168 | /*}}}*/
|
---|
169 | /*FUNCTION FetchData(int** pmatrix,int* pM, int* pN, PyObject* py_matrix){{{*/
|
---|
170 | void FetchData(int** pmatrix,int* pM,int *pN,PyObject* py_matrix){
|
---|
171 |
|
---|
172 | /*output: */
|
---|
173 | int* matrix=NULL;
|
---|
174 | int M,N;
|
---|
175 | int ndim;
|
---|
176 | npy_intp* dims=NULL;
|
---|
177 |
|
---|
178 | /*intermediary:*/
|
---|
179 | double* dmatrix=NULL;
|
---|
180 | long* lmatrix=NULL;
|
---|
181 | bool* bmatrix=NULL;
|
---|
182 | int i;
|
---|
183 |
|
---|
184 | if (PyArray_Check((PyArrayObject*)py_matrix)) {
|
---|
185 | /*retrieve dimensions: */
|
---|
186 | ndim=PyArray_NDIM((const PyArrayObject*)py_matrix);
|
---|
187 | if (ndim==2) {
|
---|
188 | dims=PyArray_DIMS((PyArrayObject*)py_matrix);
|
---|
189 | M=dims[0]; N=dims[1];
|
---|
190 | }
|
---|
191 | else if (ndim==1) {
|
---|
192 | dims=PyArray_DIMS((PyArrayObject*)py_matrix);
|
---|
193 | M=dims[0]; N=1;
|
---|
194 | }
|
---|
195 | else
|
---|
196 | _error_("expecting an MxN matrix or M vector in input!");
|
---|
197 |
|
---|
198 | if (M && N) {
|
---|
199 | if (PyArray_TYPE((PyArrayObject*)py_matrix) == NPY_DOUBLE) {
|
---|
200 | /*retrieve internal value: */
|
---|
201 | dmatrix=(double*)PyArray_DATA((PyArrayObject*)py_matrix);
|
---|
202 |
|
---|
203 | /*transform into integer matrix: */
|
---|
204 | matrix=xNew<int>(M*N);
|
---|
205 | for(i=0;i<M*N;i++)matrix[i]=(int)dmatrix[i];
|
---|
206 | }
|
---|
207 |
|
---|
208 | else if (PyArray_TYPE((PyArrayObject*)py_matrix) == NPY_INT64) {
|
---|
209 | /*retrieve internal value: */
|
---|
210 | lmatrix=(long*)PyArray_DATA((PyArrayObject*)py_matrix);
|
---|
211 |
|
---|
212 | /*transform into integer matrix: */
|
---|
213 | matrix=xNew<int>(M*N);
|
---|
214 | for(i=0;i<M*N;i++)matrix[i]=(int)lmatrix[i];
|
---|
215 | }
|
---|
216 |
|
---|
217 | else if (PyArray_TYPE((PyArrayObject*)py_matrix) == NPY_BOOL) {
|
---|
218 | /*retrieve internal value: */
|
---|
219 | bmatrix=(bool*)PyArray_DATA((PyArrayObject*)py_matrix);
|
---|
220 |
|
---|
221 | /*transform into integer matrix: */
|
---|
222 | matrix=xNew<int>(M*N);
|
---|
223 | for(i=0;i<M*N;i++)matrix[i]=(int)bmatrix[i];
|
---|
224 | }
|
---|
225 |
|
---|
226 | else
|
---|
227 | _error_("unrecognized int pyarray type in input!");
|
---|
228 | }
|
---|
229 | else
|
---|
230 | matrix=NULL;
|
---|
231 | }
|
---|
232 |
|
---|
233 | else {
|
---|
234 | M=1;
|
---|
235 | N=1;
|
---|
236 | matrix=xNew<int>(M*N);
|
---|
237 | FetchData(&(matrix[0]),py_matrix);
|
---|
238 | }
|
---|
239 |
|
---|
240 | /*output: */
|
---|
241 | if(pM)*pM=M;
|
---|
242 | if(pN)*pN=N;
|
---|
243 | if(pmatrix)*pmatrix=matrix;
|
---|
244 | }
|
---|
245 | /*}}}*/
|
---|
246 | /*FUNCTION FetchData(bool** pmatrix,int* pM, int* pN, PyObject* py_matrix){{{*/
|
---|
247 | void FetchData(bool** pmatrix,int* pM,int *pN,PyObject* py_matrix){
|
---|
248 |
|
---|
249 | /*output: */
|
---|
250 | bool* bmatrix=NULL;
|
---|
251 | bool* matrix=NULL;
|
---|
252 | int M,N;
|
---|
253 | int ndim;
|
---|
254 | npy_intp* dims=NULL;
|
---|
255 |
|
---|
256 | /*intermediary:*/
|
---|
257 | double* dmatrix=NULL;
|
---|
258 | long* lmatrix=NULL;
|
---|
259 | int i;
|
---|
260 |
|
---|
261 | if (PyArray_Check((PyArrayObject*)py_matrix)) {
|
---|
262 | /*retrieve dimensions: */
|
---|
263 | ndim=PyArray_NDIM((const PyArrayObject*)py_matrix);
|
---|
264 | if (ndim==2) {
|
---|
265 | dims=PyArray_DIMS((PyArrayObject*)py_matrix);
|
---|
266 | M=dims[0]; N=dims[1];
|
---|
267 | }
|
---|
268 | else if (ndim==1) {
|
---|
269 | dims=PyArray_DIMS((PyArrayObject*)py_matrix);
|
---|
270 | M=dims[0]; N=1;
|
---|
271 | }
|
---|
272 | else
|
---|
273 | _error_("expecting an MxN matrix or M vector in input!");
|
---|
274 |
|
---|
275 | if (M && N) {
|
---|
276 | if (PyArray_TYPE((PyArrayObject*)py_matrix) == NPY_DOUBLE) {
|
---|
277 | /*retrieve internal value: */
|
---|
278 | dmatrix=(double*)PyArray_DATA((PyArrayObject*)py_matrix);
|
---|
279 |
|
---|
280 | /*transform into bool matrix: */
|
---|
281 | matrix=xNew<bool>(M*N);
|
---|
282 | for(i=0;i<M*N;i++)matrix[i]=(bool)dmatrix[i];
|
---|
283 | }
|
---|
284 |
|
---|
285 | else if (PyArray_TYPE((PyArrayObject*)py_matrix) == NPY_INT64) {
|
---|
286 | /*retrieve internal value: */
|
---|
287 | lmatrix=(long*)PyArray_DATA((PyArrayObject*)py_matrix);
|
---|
288 |
|
---|
289 | /*transform into bool matrix: */
|
---|
290 | matrix=xNew<bool>(M*N);
|
---|
291 | for(i=0;i<M*N;i++)matrix[i]=(bool)lmatrix[i];
|
---|
292 | }
|
---|
293 |
|
---|
294 | else if (PyArray_TYPE((PyArrayObject*)py_matrix) == NPY_BOOL) {
|
---|
295 | /*retrieve internal value: */
|
---|
296 | bmatrix=(bool*)PyArray_DATA((PyArrayObject*)py_matrix);
|
---|
297 |
|
---|
298 | /*copy matrix: */
|
---|
299 | matrix=xNew<bool>(M*N);
|
---|
300 | memcpy(matrix,bmatrix,(M*N)*sizeof(bool));
|
---|
301 | }
|
---|
302 |
|
---|
303 | else
|
---|
304 | _error_("unrecognized bool pyarray type in input!");
|
---|
305 | }
|
---|
306 | else
|
---|
307 | matrix=NULL;
|
---|
308 | }
|
---|
309 |
|
---|
310 | else {
|
---|
311 | M=1;
|
---|
312 | N=1;
|
---|
313 | matrix=xNew<bool>(M*N);
|
---|
314 | FetchData(&(matrix[0]),py_matrix);
|
---|
315 | }
|
---|
316 |
|
---|
317 | /*output: */
|
---|
318 | if(pM)*pM=M;
|
---|
319 | if(pN)*pN=N;
|
---|
320 | if(pmatrix)*pmatrix=matrix;
|
---|
321 | }
|
---|
322 | /*}}}*/
|
---|
323 | /*FUNCTION FetchData(double** pvector,int* pM, PyObject* py_vector){{{*/
|
---|
324 | void FetchData(double** pvector,int* pM,PyObject* py_vector){
|
---|
325 |
|
---|
326 | /*output: */
|
---|
327 | double* dvector=NULL;
|
---|
328 | double* vector=NULL;
|
---|
329 | int M;
|
---|
330 | int ndim;
|
---|
331 | npy_intp* dims=NULL;
|
---|
332 |
|
---|
333 | /*intermediary:*/
|
---|
334 | long* lvector=NULL;
|
---|
335 | bool* bvector=NULL;
|
---|
336 | int i;
|
---|
337 |
|
---|
338 | /*retrieve dimensions: */
|
---|
339 | ndim=PyArray_NDIM((const PyArrayObject*)py_vector);
|
---|
340 | if(ndim!=1)_error_("expecting an Mx1 vector in input!");
|
---|
341 | dims=PyArray_DIMS((PyArrayObject*)py_vector);
|
---|
342 | M=dims[0];
|
---|
343 |
|
---|
344 | if (M) {
|
---|
345 | if (PyArray_TYPE((PyArrayObject*)py_vector) == NPY_DOUBLE) {
|
---|
346 | /*retrieve internal value: */
|
---|
347 | dvector=(double*)PyArray_DATA((PyArrayObject*)py_vector);
|
---|
348 |
|
---|
349 | /*copy vector: */
|
---|
350 | vector=xNew<double>(M);
|
---|
351 | memcpy(vector,dvector,(M)*sizeof(double));
|
---|
352 | }
|
---|
353 |
|
---|
354 | else if (PyArray_TYPE((PyArrayObject*)py_vector) == NPY_INT64) {
|
---|
355 | /*retrieve internal value: */
|
---|
356 | lvector=(long*)PyArray_DATA((PyArrayObject*)py_vector);
|
---|
357 |
|
---|
358 | /*transform into double vector: */
|
---|
359 | vector=xNew<double>(M);
|
---|
360 | for(i=0;i<M;i++)vector[i]=(double)lvector[i];
|
---|
361 | }
|
---|
362 |
|
---|
363 | else if (PyArray_TYPE((PyArrayObject*)py_vector) == NPY_BOOL) {
|
---|
364 | /*retrieve internal value: */
|
---|
365 | bvector=(bool*)PyArray_DATA((PyArrayObject*)py_vector);
|
---|
366 |
|
---|
367 | /*transform into double vector: */
|
---|
368 | vector=xNew<double>(M);
|
---|
369 | for(i=0;i<M;i++)vector[i]=(double)bvector[i];
|
---|
370 | }
|
---|
371 |
|
---|
372 | else
|
---|
373 | _error_("unrecognized double pyarray type in input!");
|
---|
374 | }
|
---|
375 | else
|
---|
376 | vector=NULL;
|
---|
377 |
|
---|
378 | /*output: */
|
---|
379 | if(pM)*pM=M;
|
---|
380 | if(pvector)*pvector=vector;
|
---|
381 | }
|
---|
382 | /*}}}*/
|
---|
383 | /*FUNCTION FetchData(int** pvector,int* pM, PyObject* py_vector){{{*/
|
---|
384 | void FetchData(int** pvector,int* pM,PyObject* py_vector){
|
---|
385 |
|
---|
386 | /*output: */
|
---|
387 | int* vector=NULL;
|
---|
388 | int M;
|
---|
389 | int ndim;
|
---|
390 | npy_intp* dims=NULL;
|
---|
391 |
|
---|
392 | /*intermediary:*/
|
---|
393 | long* lvector=NULL;
|
---|
394 | bool* bvector=NULL;
|
---|
395 | double* dvector=NULL;
|
---|
396 | int i;
|
---|
397 |
|
---|
398 | /*retrieve dimensions: */
|
---|
399 | ndim=PyArray_NDIM((const PyArrayObject*)py_vector);
|
---|
400 | if(ndim!=1)_error_("expecting an Mx1 vector in input!");
|
---|
401 | dims=PyArray_DIMS((PyArrayObject*)py_vector);
|
---|
402 | M=dims[0];
|
---|
403 |
|
---|
404 | if (M) {
|
---|
405 | if (PyArray_TYPE((PyArrayObject*)py_vector) == NPY_DOUBLE) {
|
---|
406 | /*retrieve internal value: */
|
---|
407 | dvector=(double*)PyArray_DATA((PyArrayObject*)py_vector);
|
---|
408 |
|
---|
409 | /*transform into int vector: */
|
---|
410 | vector=xNew<int>(M);
|
---|
411 | for(i=0;i<M;i++)vector[i]=(int)lvector[i];
|
---|
412 | }
|
---|
413 |
|
---|
414 | else if (PyArray_TYPE((PyArrayObject*)py_vector) == NPY_INT64) {
|
---|
415 | /*retrieve internal value: */
|
---|
416 | lvector=(long*)PyArray_DATA((PyArrayObject*)py_vector);
|
---|
417 |
|
---|
418 | /*transform into int vector: */
|
---|
419 | vector=xNew<int>(M);
|
---|
420 | for(i=0;i<M;i++)vector[i]=(int)lvector[i];
|
---|
421 | }
|
---|
422 |
|
---|
423 | else if (PyArray_TYPE((PyArrayObject*)py_vector) == NPY_BOOL) {
|
---|
424 | /*retrieve internal value: */
|
---|
425 | bvector=(bool*)PyArray_DATA((PyArrayObject*)py_vector);
|
---|
426 |
|
---|
427 | /*transform into int vector: */
|
---|
428 | vector=xNew<int>(M);
|
---|
429 | for(i=0;i<M;i++)vector[i]=(int)bvector[i];
|
---|
430 | }
|
---|
431 |
|
---|
432 | else
|
---|
433 | _error_("unrecognized int pyarray type in input!");
|
---|
434 | }
|
---|
435 | else
|
---|
436 | vector=NULL;
|
---|
437 |
|
---|
438 | /*output: */
|
---|
439 | if(pM)*pM=M;
|
---|
440 | if(pvector)*pvector=vector;
|
---|
441 | }
|
---|
442 | /*}}}*/
|
---|
443 | /*FUNCTION FetchData(bool** pvector,int* pM, PyObject* py_vector){{{*/
|
---|
444 | void FetchData(bool** pvector,int* pM,PyObject* py_vector){
|
---|
445 |
|
---|
446 | /*output: */
|
---|
447 | bool* bvector=NULL;
|
---|
448 | bool* vector=NULL;
|
---|
449 | int M;
|
---|
450 | int ndim;
|
---|
451 | npy_intp* dims=NULL;
|
---|
452 |
|
---|
453 | /*intermediary:*/
|
---|
454 | double* dvector=NULL;
|
---|
455 | long* lvector=NULL;
|
---|
456 | int i;
|
---|
457 |
|
---|
458 | /*retrieve dimensions: */
|
---|
459 | ndim=PyArray_NDIM((const PyArrayObject*)py_vector);
|
---|
460 | if(ndim!=1)_error_("expecting an Mx1 vector in input!");
|
---|
461 | dims=PyArray_DIMS((PyArrayObject*)py_vector);
|
---|
462 | M=dims[0];
|
---|
463 |
|
---|
464 | if (M) {
|
---|
465 | if (PyArray_TYPE((PyArrayObject*)py_vector) == NPY_DOUBLE) {
|
---|
466 | /*retrieve internal value: */
|
---|
467 | dvector=(double*)PyArray_DATA((PyArrayObject*)py_vector);
|
---|
468 |
|
---|
469 | /*transform into bool vector: */
|
---|
470 | vector=xNew<bool>(M);
|
---|
471 | for(i=0;i<M;i++)vector[i]=(bool)dvector[i];
|
---|
472 | }
|
---|
473 |
|
---|
474 | else if (PyArray_TYPE((PyArrayObject*)py_vector) == NPY_INT64) {
|
---|
475 | /*retrieve internal value: */
|
---|
476 | lvector=(long*)PyArray_DATA((PyArrayObject*)py_vector);
|
---|
477 |
|
---|
478 | /*transform into bool vector: */
|
---|
479 | vector=xNew<bool>(M);
|
---|
480 | for(i=0;i<M;i++)vector[i]=(bool)lvector[i];
|
---|
481 | }
|
---|
482 |
|
---|
483 | else if (PyArray_TYPE((PyArrayObject*)py_vector) == NPY_BOOL) {
|
---|
484 | /*retrieve internal value: */
|
---|
485 | bvector=(bool*)PyArray_DATA((PyArrayObject*)py_vector);
|
---|
486 |
|
---|
487 | /*copy vector: */
|
---|
488 | vector=xNew<bool>(M);
|
---|
489 | memcpy(vector,bvector,(M)*sizeof(bool));
|
---|
490 | }
|
---|
491 |
|
---|
492 | else
|
---|
493 | _error_("unrecognized bool pyarray type in input!");
|
---|
494 | }
|
---|
495 | else
|
---|
496 | vector=NULL;
|
---|
497 |
|
---|
498 | /*output: */
|
---|
499 | if(pM)*pM=M;
|
---|
500 | if(pvector)*pvector=vector;
|
---|
501 | }
|
---|
502 | /*}}}*/
|
---|
503 |
|
---|
504 | /*ISSM objects*/
|
---|
505 | /*FUNCTION FetchData(BamgGeom** pbamggeom,PyObject* py_dict){{{*/
|
---|
506 | void FetchData(BamgGeom** pbamggeom,PyObject* py_dict){
|
---|
507 |
|
---|
508 | /*Initialize output*/
|
---|
509 | BamgGeom* bamggeom=new BamgGeom();
|
---|
510 |
|
---|
511 | /*Fetch all fields*/
|
---|
512 | FetchData(&bamggeom->Vertices,&bamggeom->VerticesSize[0],&bamggeom->VerticesSize[1],PyDict_GetItemString(py_dict,"Vertices"));
|
---|
513 | FetchData(&bamggeom->Edges, &bamggeom->EdgesSize[0], &bamggeom->EdgesSize[1], PyDict_GetItemString(py_dict,"Edges"));
|
---|
514 | FetchData(&bamggeom->Corners, &bamggeom->CornersSize[0], &bamggeom->CornersSize[1], PyDict_GetItemString(py_dict,"Corners"));
|
---|
515 | FetchData(&bamggeom->RequiredVertices,&bamggeom->RequiredVerticesSize[0],&bamggeom->RequiredVerticesSize[1],PyDict_GetItemString(py_dict,"RequiredVertices"));
|
---|
516 | FetchData(&bamggeom->RequiredEdges, &bamggeom->RequiredEdgesSize[0], &bamggeom->RequiredEdgesSize[1], PyDict_GetItemString(py_dict,"RequiredEdges"));
|
---|
517 | FetchData(&bamggeom->CrackedEdges,&bamggeom->CrackedEdgesSize[0],&bamggeom->CrackedEdgesSize[1],PyDict_GetItemString(py_dict,"CrackedEdges"));
|
---|
518 | FetchData(&bamggeom->SubDomains,&bamggeom->SubDomainsSize[0],&bamggeom->SubDomainsSize[1],PyDict_GetItemString(py_dict,"SubDomains"));
|
---|
519 |
|
---|
520 | /*Assign output pointers:*/
|
---|
521 | *pbamggeom=bamggeom;
|
---|
522 | }
|
---|
523 | /*}}}*/
|
---|
524 | /*FUNCTION FetchData(BamgMesh** pbamgmesh,PyObject* py_dict){{{*/
|
---|
525 | void FetchData(BamgMesh** pbamgmesh,PyObject* py_dict){
|
---|
526 |
|
---|
527 | /*Initialize output*/
|
---|
528 | BamgMesh* bamgmesh=new BamgMesh();
|
---|
529 |
|
---|
530 | /*Fetch all fields*/
|
---|
531 | FetchData(&bamgmesh->Vertices,&bamgmesh->VerticesSize[0],&bamgmesh->VerticesSize[1],PyDict_GetItemString(py_dict,"Vertices"));
|
---|
532 | FetchData(&bamgmesh->Edges, &bamgmesh->EdgesSize[0], &bamgmesh->EdgesSize[1], PyDict_GetItemString(py_dict,"Edges"));
|
---|
533 | FetchData(&bamgmesh->Triangles, &bamgmesh->TrianglesSize[0], &bamgmesh->TrianglesSize[1], PyDict_GetItemString(py_dict,"Triangles"));
|
---|
534 | FetchData(&bamgmesh->CrackedEdges,&bamgmesh->CrackedEdgesSize[0],&bamgmesh->CrackedEdgesSize[1],PyDict_GetItemString(py_dict,"CrackedEdges"));
|
---|
535 | FetchData(&bamgmesh->VerticesOnGeomEdge,&bamgmesh->VerticesOnGeomEdgeSize[0],&bamgmesh->VerticesOnGeomEdgeSize[1],PyDict_GetItemString(py_dict,"VerticesOnGeomEdge"));
|
---|
536 | FetchData(&bamgmesh->VerticesOnGeomVertex,&bamgmesh->VerticesOnGeomVertexSize[0],&bamgmesh->VerticesOnGeomVertexSize[1],PyDict_GetItemString(py_dict,"VerticesOnGeomVertex"));
|
---|
537 | FetchData(&bamgmesh->EdgesOnGeomEdge, &bamgmesh->EdgesOnGeomEdgeSize[0], &bamgmesh->EdgesOnGeomEdgeSize[1], PyDict_GetItemString(py_dict,"EdgesOnGeomEdge"));
|
---|
538 | FetchData(&bamgmesh->IssmSegments,&bamgmesh->IssmSegmentsSize[0],&bamgmesh->IssmSegmentsSize[1],PyDict_GetItemString(py_dict,"IssmSegments"));
|
---|
539 |
|
---|
540 | /*Assign output pointers:*/
|
---|
541 | *pbamgmesh=bamgmesh;
|
---|
542 | }
|
---|
543 | /*}}}*/
|
---|
544 | /*FUNCTION FetchData(BamgOpts** pbamgopts,PyObject* py_dict){{{*/
|
---|
545 | void FetchData(BamgOpts** pbamgopts,PyObject* py_dict){
|
---|
546 |
|
---|
547 | /*Initialize output*/
|
---|
548 | BamgOpts* bamgopts=new BamgOpts();
|
---|
549 |
|
---|
550 | /*Fetch all fields*/
|
---|
551 | FetchData(&bamgopts->anisomax,PyDict_GetItemString(py_dict,"anisomax"));
|
---|
552 | FetchData(&bamgopts->cutoff,PyDict_GetItemString(py_dict,"cutoff"));
|
---|
553 | FetchData(&bamgopts->coeff,PyDict_GetItemString(py_dict,"coeff"));
|
---|
554 | FetchData(&bamgopts->errg,PyDict_GetItemString(py_dict,"errg"));
|
---|
555 | FetchData(&bamgopts->gradation,PyDict_GetItemString(py_dict,"gradation"));
|
---|
556 | FetchData(&bamgopts->Hessiantype,PyDict_GetItemString(py_dict,"Hessiantype"));
|
---|
557 | FetchData(&bamgopts->MaxCornerAngle,PyDict_GetItemString(py_dict,"MaxCornerAngle"));
|
---|
558 | FetchData(&bamgopts->maxnbv,PyDict_GetItemString(py_dict,"maxnbv"));
|
---|
559 | FetchData(&bamgopts->maxsubdiv,PyDict_GetItemString(py_dict,"maxsubdiv"));
|
---|
560 | FetchData(&bamgopts->Metrictype,PyDict_GetItemString(py_dict,"Metrictype"));
|
---|
561 | FetchData(&bamgopts->nbjacobi,PyDict_GetItemString(py_dict,"nbjacobi"));
|
---|
562 | FetchData(&bamgopts->nbsmooth,PyDict_GetItemString(py_dict,"nbsmooth"));
|
---|
563 | FetchData(&bamgopts->omega,PyDict_GetItemString(py_dict,"omega"));
|
---|
564 | FetchData(&bamgopts->power,PyDict_GetItemString(py_dict,"power"));
|
---|
565 | FetchData(&bamgopts->verbose,PyDict_GetItemString(py_dict,"verbose"));
|
---|
566 |
|
---|
567 | FetchData(&bamgopts->Crack,PyDict_GetItemString(py_dict,"Crack"));
|
---|
568 | FetchData(&bamgopts->geometricalmetric,PyDict_GetItemString(py_dict,"geometricalmetric"));
|
---|
569 | FetchData(&bamgopts->KeepVertices,PyDict_GetItemString(py_dict,"KeepVertices"));
|
---|
570 | FetchData(&bamgopts->splitcorners,PyDict_GetItemString(py_dict,"splitcorners"));
|
---|
571 |
|
---|
572 | FetchData(&bamgopts->hmin,PyDict_GetItemString(py_dict,"hmin"));
|
---|
573 | FetchData(&bamgopts->hmax,PyDict_GetItemString(py_dict,"hmax"));
|
---|
574 | FetchData(&bamgopts->hminVertices,&bamgopts->hminVerticesSize[0],&bamgopts->hminVerticesSize[1],PyDict_GetItemString(py_dict,"hminVertices"));
|
---|
575 | FetchData(&bamgopts->hmaxVertices,&bamgopts->hmaxVerticesSize[0],&bamgopts->hmaxVerticesSize[1],PyDict_GetItemString(py_dict,"hmaxVertices"));
|
---|
576 | FetchData(&bamgopts->hVertices,&bamgopts->hVerticesSize[0],&bamgopts->hVerticesSize[1],PyDict_GetItemString(py_dict,"hVertices"));
|
---|
577 | FetchData(&bamgopts->metric,&bamgopts->metricSize[0],&bamgopts->metricSize[1],PyDict_GetItemString(py_dict,"metric"));
|
---|
578 | FetchData(&bamgopts->field,&bamgopts->fieldSize[0],&bamgopts->fieldSize[1],PyDict_GetItemString(py_dict,"field"));
|
---|
579 | FetchData(&bamgopts->err,&bamgopts->errSize[0],&bamgopts->errSize[1],PyDict_GetItemString(py_dict,"err"));
|
---|
580 |
|
---|
581 | /*Additional checks*/
|
---|
582 | bamgopts->Check();
|
---|
583 |
|
---|
584 | /*Assign output pointers:*/
|
---|
585 | *pbamgopts=bamgopts;
|
---|
586 | }
|
---|
587 | /*}}}*/
|
---|
588 | /*FUNCTION FetchData(Options** poptions,int istart, int nrhs,PyObject* py_tuple){{{*/
|
---|
589 | void FetchData(Options** poptions,int istart, int nrhs,PyObject* py_tuple){
|
---|
590 |
|
---|
591 | char *name = NULL;
|
---|
592 | Option *option = NULL;
|
---|
593 |
|
---|
594 | /*Initialize output*/
|
---|
595 | Options* options=new Options();
|
---|
596 |
|
---|
597 | /*Fetch all options*/
|
---|
598 | for (int i=istart; i<nrhs; i=i+2){
|
---|
599 | if (!PyString_Check(PyTuple_GetItem(py_tuple,(Py_ssize_t)i))) _error_("Argument " << i+1 << " must be name of option");
|
---|
600 |
|
---|
601 | FetchData(&name,PyTuple_GetItem(py_tuple,(Py_ssize_t)i));
|
---|
602 | if(i+1 == nrhs) _error_("Argument " << i+2 << " must exist and be value of option \"" << name << "\".");
|
---|
603 |
|
---|
604 | _pprintLine_("FetchData for Options not implemented yet, ignoring option \"" << name << "\"!");
|
---|
605 |
|
---|
606 | // option=(Option*)OptionParse(name,&PyTuple_GetItem(py_tuple,(Py_ssize_t)(i+1)));
|
---|
607 | // options->AddOption(option);
|
---|
608 | // option=NULL;
|
---|
609 | }
|
---|
610 |
|
---|
611 | /*Assign output pointers:*/
|
---|
612 | *poptions=options;
|
---|
613 | }
|
---|
614 | /*}}}*/
|
---|
615 | /*FUNCTION FetchData(DataSet** pcontours,PyObject* py_list){{{*/
|
---|
616 | void FetchData(DataSet** pcontours,PyObject* py_list){
|
---|
617 |
|
---|
618 | int numcontours,test1,test2;
|
---|
619 | char *contourname = NULL;
|
---|
620 | DataSet *contours = NULL;
|
---|
621 | Contour<double> *contouri = NULL;
|
---|
622 | PyObject *py_dicti = NULL;
|
---|
623 | PyObject *py_item = NULL;
|
---|
624 |
|
---|
625 | if (PyString_Check(py_list)){
|
---|
626 | FetchData(&contourname,py_list);
|
---|
627 | contours=DomainOutlineRead<double>(contourname);
|
---|
628 | }
|
---|
629 | else if(PyList_Check(py_list)){
|
---|
630 |
|
---|
631 | contours=new DataSet(0);
|
---|
632 | numcontours=(int)PyList_Size(py_list);
|
---|
633 |
|
---|
634 | for(int i=0;i<numcontours;i++){
|
---|
635 |
|
---|
636 | contouri=xNew<Contour<double> >(1);
|
---|
637 | py_dicti=PyList_GetItem(py_list,(Py_ssize_t)i);
|
---|
638 |
|
---|
639 | py_item = PyDict_GetItemString(py_dicti,"nods");
|
---|
640 | if(!py_item) _error_("input structure does not have a 'nods' field");
|
---|
641 | FetchData(&contouri->nods,py_item);
|
---|
642 |
|
---|
643 | py_item = PyDict_GetItemString(py_dicti,"x");
|
---|
644 | if(!py_item) _error_("input structure does not have a 'x' field");
|
---|
645 | FetchData(&contouri->x,&test1,&test2,py_item);
|
---|
646 | if(test1!=contouri->nods || test2!=1) _error_("field x should be of size ["<<contouri->nods<<" 1]");
|
---|
647 |
|
---|
648 | py_item = PyDict_GetItemString(py_dicti,"y");
|
---|
649 | if(!py_item) _error_("input structure does not have a 'y' field");
|
---|
650 | FetchData(&contouri->y,&test1,&test2,py_item);
|
---|
651 | if(test1!=contouri->nods || test2!=1) _error_("field y should be of size ["<<contouri->nods<<" 1]");
|
---|
652 |
|
---|
653 | contours->AddObject(contouri);
|
---|
654 | }
|
---|
655 | }
|
---|
656 | else{
|
---|
657 | _error_("Contour is neither a string nor a structure and cannot be loaded");
|
---|
658 | }
|
---|
659 |
|
---|
660 | /*clean-up and assign output pointer*/
|
---|
661 | xDelete<char>(contourname);
|
---|
662 | *pcontours=contours;
|
---|
663 | }
|
---|
664 | /*}}}*/
|
---|
665 |
|
---|
666 | /*Python version dependent: */
|
---|
667 | #if _PYTHON_MAJOR_ >= 3
|
---|
668 | /*FUNCTION FetchData(char** pstring,PyObject* py_unicode){{{*/
|
---|
669 | void FetchData(char** pstring,PyObject* py_unicode){
|
---|
670 |
|
---|
671 | PyObject* py_bytes;
|
---|
672 | char* string=NULL;
|
---|
673 |
|
---|
674 | /*convert to bytes format: */
|
---|
675 | PyUnicode_FSConverter(py_unicode,&py_bytes);
|
---|
676 |
|
---|
677 | /*convert from bytes to string: */
|
---|
678 | string=PyBytes_AS_STRING(py_bytes);
|
---|
679 |
|
---|
680 | *pstring=string;
|
---|
681 | }
|
---|
682 | /*}}}*/
|
---|
683 | #else
|
---|
684 | /*FUNCTION FetchData(char** pstring,PyObject* py_string){{{*/
|
---|
685 | void FetchData(char** pstring,PyObject* py_string){
|
---|
686 |
|
---|
687 | char* string=NULL;
|
---|
688 |
|
---|
689 | /*extract internal string: */
|
---|
690 | string=PyString_AsString(py_string);
|
---|
691 |
|
---|
692 | /*copy string (note strlen does not include trailing NULL): */
|
---|
693 | *pstring=xNew<char>(strlen(string)+1);
|
---|
694 | memcpy(*pstring,string,(strlen(string)+1)*sizeof(char));
|
---|
695 | }
|
---|
696 | /*}}}*/
|
---|
697 | #endif
|
---|