5
#pragma warning(disable: 4275)
6
#pragma warning(disable: 4101)
10
#include "frameobject.h"
14
#include "scxx/object.h"
15
#include "scxx/list.h"
16
#include "scxx/tuple.h"
17
#include "scxx/dict.h"
20
#include "numpy/arrayobject.h"
25
// global None value for use in functions.
27
object None = object(Py_None);
30
char* find_type(PyObject* py_obj)
32
if(py_obj == NULL) return "C NULL value";
33
if(PyCallable_Check(py_obj)) return "callable";
34
if(PyString_Check(py_obj)) return "string";
35
if(PyInt_Check(py_obj)) return "int";
36
if(PyFloat_Check(py_obj)) return "float";
37
if(PyDict_Check(py_obj)) return "dict";
38
if(PyList_Check(py_obj)) return "list";
39
if(PyTuple_Check(py_obj)) return "tuple";
40
if(PyFile_Check(py_obj)) return "file";
41
if(PyModule_Check(py_obj)) return "module";
43
//should probably do more intergation (and thinking) on these.
44
if(PyCallable_Check(py_obj) && PyInstance_Check(py_obj)) return "callable";
45
if(PyInstance_Check(py_obj)) return "instance";
46
if(PyCallable_Check(py_obj)) return "callable";
50
void throw_error(PyObject* exc, const char* msg)
52
//printf("setting python error: %s\n",msg);
53
PyErr_SetString(exc, msg);
54
//printf("throwing error\n");
58
void handle_bad_type(PyObject* py_obj, const char* good_type, const char* var_name)
61
sprintf(msg,"received '%s' type instead of '%s' for variable '%s'",
62
find_type(py_obj),good_type,var_name);
63
throw_error(PyExc_TypeError,msg);
66
void handle_conversion_error(PyObject* py_obj, const char* good_type, const char* var_name)
69
sprintf(msg,"Conversion Error:, received '%s' type instead of '%s' for variable '%s'",
70
find_type(py_obj),good_type,var_name);
71
throw_error(PyExc_TypeError,msg);
78
int convert_to_int(PyObject* py_obj, const char* name)
80
// Incref occurs even if conversion fails so that
81
// the decref in cleanup_code has a matching incref.
83
if (!py_obj || !PyInt_Check(py_obj))
84
handle_conversion_error(py_obj,"int", name);
85
return (int) PyInt_AsLong(py_obj);
88
int py_to_int(PyObject* py_obj, const char* name)
90
// !! Pretty sure INCREF should only be called on success since
91
// !! py_to_xxx is used by the user -- not the code generator.
92
if (!py_obj || !PyInt_Check(py_obj))
93
handle_bad_type(py_obj,"int", name);
95
return (int) PyInt_AsLong(py_obj);
99
int_handler x__int_handler = int_handler();
100
#define convert_to_int(py_obj,name) \
101
x__int_handler.convert_to_int(py_obj,name)
102
#define py_to_int(py_obj,name) \
103
x__int_handler.py_to_int(py_obj,name)
106
PyObject* int_to_py(PyObject* obj)
108
return (PyObject*) obj;
115
double convert_to_float(PyObject* py_obj, const char* name)
117
// Incref occurs even if conversion fails so that
118
// the decref in cleanup_code has a matching incref.
120
if (!py_obj || !PyFloat_Check(py_obj))
121
handle_conversion_error(py_obj,"float", name);
122
return PyFloat_AsDouble(py_obj);
125
double py_to_float(PyObject* py_obj, const char* name)
127
// !! Pretty sure INCREF should only be called on success since
128
// !! py_to_xxx is used by the user -- not the code generator.
129
if (!py_obj || !PyFloat_Check(py_obj))
130
handle_bad_type(py_obj,"float", name);
132
return PyFloat_AsDouble(py_obj);
136
float_handler x__float_handler = float_handler();
137
#define convert_to_float(py_obj,name) \
138
x__float_handler.convert_to_float(py_obj,name)
139
#define py_to_float(py_obj,name) \
140
x__float_handler.py_to_float(py_obj,name)
143
PyObject* float_to_py(PyObject* obj)
145
return (PyObject*) obj;
149
class complex_handler
152
std::complex<double> convert_to_complex(PyObject* py_obj, const char* name)
154
// Incref occurs even if conversion fails so that
155
// the decref in cleanup_code has a matching incref.
157
if (!py_obj || !PyComplex_Check(py_obj))
158
handle_conversion_error(py_obj,"complex", name);
159
return std::complex<double>(PyComplex_RealAsDouble(py_obj),PyComplex_ImagAsDouble(py_obj));
162
std::complex<double> py_to_complex(PyObject* py_obj, const char* name)
164
// !! Pretty sure INCREF should only be called on success since
165
// !! py_to_xxx is used by the user -- not the code generator.
166
if (!py_obj || !PyComplex_Check(py_obj))
167
handle_bad_type(py_obj,"complex", name);
169
return std::complex<double>(PyComplex_RealAsDouble(py_obj),PyComplex_ImagAsDouble(py_obj));
173
complex_handler x__complex_handler = complex_handler();
174
#define convert_to_complex(py_obj,name) \
175
x__complex_handler.convert_to_complex(py_obj,name)
176
#define py_to_complex(py_obj,name) \
177
x__complex_handler.py_to_complex(py_obj,name)
180
PyObject* complex_to_py(PyObject* obj)
182
return (PyObject*) obj;
186
class unicode_handler
189
Py_UNICODE* convert_to_unicode(PyObject* py_obj, const char* name)
191
// Incref occurs even if conversion fails so that
192
// the decref in cleanup_code has a matching incref.
194
if (!py_obj || !PyUnicode_Check(py_obj))
195
handle_conversion_error(py_obj,"unicode", name);
196
return PyUnicode_AS_UNICODE(py_obj);
199
Py_UNICODE* py_to_unicode(PyObject* py_obj, const char* name)
201
// !! Pretty sure INCREF should only be called on success since
202
// !! py_to_xxx is used by the user -- not the code generator.
203
if (!py_obj || !PyUnicode_Check(py_obj))
204
handle_bad_type(py_obj,"unicode", name);
206
return PyUnicode_AS_UNICODE(py_obj);
210
unicode_handler x__unicode_handler = unicode_handler();
211
#define convert_to_unicode(py_obj,name) \
212
x__unicode_handler.convert_to_unicode(py_obj,name)
213
#define py_to_unicode(py_obj,name) \
214
x__unicode_handler.py_to_unicode(py_obj,name)
217
PyObject* unicode_to_py(PyObject* obj)
219
return (PyObject*) obj;
226
std::string convert_to_string(PyObject* py_obj, const char* name)
228
// Incref occurs even if conversion fails so that
229
// the decref in cleanup_code has a matching incref.
231
if (!py_obj || !PyString_Check(py_obj))
232
handle_conversion_error(py_obj,"string", name);
233
return std::string(PyString_AsString(py_obj));
236
std::string py_to_string(PyObject* py_obj, const char* name)
238
// !! Pretty sure INCREF should only be called on success since
239
// !! py_to_xxx is used by the user -- not the code generator.
240
if (!py_obj || !PyString_Check(py_obj))
241
handle_bad_type(py_obj,"string", name);
243
return std::string(PyString_AsString(py_obj));
247
string_handler x__string_handler = string_handler();
248
#define convert_to_string(py_obj,name) \
249
x__string_handler.convert_to_string(py_obj,name)
250
#define py_to_string(py_obj,name) \
251
x__string_handler.py_to_string(py_obj,name)
254
PyObject* string_to_py(std::string s)
256
return PyString_FromString(s.c_str());
262
py::list convert_to_list(PyObject* py_obj, const char* name)
264
// Incref occurs even if conversion fails so that
265
// the decref in cleanup_code has a matching incref.
267
if (!py_obj || !PyList_Check(py_obj))
268
handle_conversion_error(py_obj,"list", name);
269
return py::list(py_obj);
272
py::list py_to_list(PyObject* py_obj, const char* name)
274
// !! Pretty sure INCREF should only be called on success since
275
// !! py_to_xxx is used by the user -- not the code generator.
276
if (!py_obj || !PyList_Check(py_obj))
277
handle_bad_type(py_obj,"list", name);
279
return py::list(py_obj);
283
list_handler x__list_handler = list_handler();
284
#define convert_to_list(py_obj,name) \
285
x__list_handler.convert_to_list(py_obj,name)
286
#define py_to_list(py_obj,name) \
287
x__list_handler.py_to_list(py_obj,name)
290
PyObject* list_to_py(PyObject* obj)
292
return (PyObject*) obj;
299
py::dict convert_to_dict(PyObject* py_obj, const char* name)
301
// Incref occurs even if conversion fails so that
302
// the decref in cleanup_code has a matching incref.
304
if (!py_obj || !PyDict_Check(py_obj))
305
handle_conversion_error(py_obj,"dict", name);
306
return py::dict(py_obj);
309
py::dict py_to_dict(PyObject* py_obj, const char* name)
311
// !! Pretty sure INCREF should only be called on success since
312
// !! py_to_xxx is used by the user -- not the code generator.
313
if (!py_obj || !PyDict_Check(py_obj))
314
handle_bad_type(py_obj,"dict", name);
316
return py::dict(py_obj);
320
dict_handler x__dict_handler = dict_handler();
321
#define convert_to_dict(py_obj,name) \
322
x__dict_handler.convert_to_dict(py_obj,name)
323
#define py_to_dict(py_obj,name) \
324
x__dict_handler.py_to_dict(py_obj,name)
327
PyObject* dict_to_py(PyObject* obj)
329
return (PyObject*) obj;
336
py::tuple convert_to_tuple(PyObject* py_obj, const char* name)
338
// Incref occurs even if conversion fails so that
339
// the decref in cleanup_code has a matching incref.
341
if (!py_obj || !PyTuple_Check(py_obj))
342
handle_conversion_error(py_obj,"tuple", name);
343
return py::tuple(py_obj);
346
py::tuple py_to_tuple(PyObject* py_obj, const char* name)
348
// !! Pretty sure INCREF should only be called on success since
349
// !! py_to_xxx is used by the user -- not the code generator.
350
if (!py_obj || !PyTuple_Check(py_obj))
351
handle_bad_type(py_obj,"tuple", name);
353
return py::tuple(py_obj);
357
tuple_handler x__tuple_handler = tuple_handler();
358
#define convert_to_tuple(py_obj,name) \
359
x__tuple_handler.convert_to_tuple(py_obj,name)
360
#define py_to_tuple(py_obj,name) \
361
x__tuple_handler.py_to_tuple(py_obj,name)
364
PyObject* tuple_to_py(PyObject* obj)
366
return (PyObject*) obj;
373
FILE* convert_to_file(PyObject* py_obj, const char* name)
375
// Incref occurs even if conversion fails so that
376
// the decref in cleanup_code has a matching incref.
378
if (!py_obj || !PyFile_Check(py_obj))
379
handle_conversion_error(py_obj,"file", name);
380
return PyFile_AsFile(py_obj);
383
FILE* py_to_file(PyObject* py_obj, const char* name)
385
// !! Pretty sure INCREF should only be called on success since
386
// !! py_to_xxx is used by the user -- not the code generator.
387
if (!py_obj || !PyFile_Check(py_obj))
388
handle_bad_type(py_obj,"file", name);
390
return PyFile_AsFile(py_obj);
394
file_handler x__file_handler = file_handler();
395
#define convert_to_file(py_obj,name) \
396
x__file_handler.convert_to_file(py_obj,name)
397
#define py_to_file(py_obj,name) \
398
x__file_handler.py_to_file(py_obj,name)
401
PyObject* file_to_py(FILE* file, char* name, char* mode)
403
PyObject* py_obj = NULL;
404
//extern int fclose(FILE *);
405
return (PyObject*) PyFile_FromFile(file, name, mode, fclose);
408
class instance_handler
411
py::object convert_to_instance(PyObject* py_obj, const char* name)
413
// Incref occurs even if conversion fails so that
414
// the decref in cleanup_code has a matching incref.
416
if (!py_obj || !PyInstance_Check(py_obj))
417
handle_conversion_error(py_obj,"instance", name);
418
return py::object(py_obj);
421
py::object py_to_instance(PyObject* py_obj, const char* name)
423
// !! Pretty sure INCREF should only be called on success since
424
// !! py_to_xxx is used by the user -- not the code generator.
425
if (!py_obj || !PyInstance_Check(py_obj))
426
handle_bad_type(py_obj,"instance", name);
428
return py::object(py_obj);
432
instance_handler x__instance_handler = instance_handler();
433
#define convert_to_instance(py_obj,name) \
434
x__instance_handler.convert_to_instance(py_obj,name)
435
#define py_to_instance(py_obj,name) \
436
x__instance_handler.py_to_instance(py_obj,name)
439
PyObject* instance_to_py(PyObject* obj)
441
return (PyObject*) obj;
445
class numpy_size_handler
448
void conversion_numpy_check_size(PyArrayObject* arr_obj, int Ndims,
451
if (arr_obj->nd != Ndims)
454
sprintf(msg,"Conversion Error: received '%d' dimensional array instead of '%d' dimensional array for variable '%s'",
455
arr_obj->nd,Ndims,name);
456
throw_error(PyExc_TypeError,msg);
460
void numpy_check_size(PyArrayObject* arr_obj, int Ndims, const char* name)
462
if (arr_obj->nd != Ndims)
465
sprintf(msg,"received '%d' dimensional array instead of '%d' dimensional array for variable '%s'",
466
arr_obj->nd,Ndims,name);
467
throw_error(PyExc_TypeError,msg);
472
numpy_size_handler x__numpy_size_handler = numpy_size_handler();
473
#define conversion_numpy_check_size x__numpy_size_handler.conversion_numpy_check_size
474
#define numpy_check_size x__numpy_size_handler.numpy_check_size
477
class numpy_type_handler
480
void conversion_numpy_check_type(PyArrayObject* arr_obj, int numeric_type,
483
// Make sure input has correct numeric type.
484
int arr_type = arr_obj->descr->type_num;
485
if (PyTypeNum_ISEXTENDED(numeric_type))
488
sprintf(msg, "Conversion Error: extended types not supported for variable '%s'",
490
throw_error(PyExc_TypeError, msg);
492
if (!PyArray_EquivTypenums(arr_type, numeric_type))
495
char* type_names[23] = {"bool", "byte", "ubyte","short", "ushort",
496
"int", "uint", "long", "ulong", "longlong", "ulonglong",
497
"float", "double", "longdouble", "cfloat", "cdouble",
498
"clongdouble", "object", "string", "unicode", "void", "ntype",
501
sprintf(msg,"Conversion Error: received '%s' typed array instead of '%s' typed array for variable '%s'",
502
type_names[arr_type],type_names[numeric_type],name);
503
throw_error(PyExc_TypeError,msg);
507
void numpy_check_type(PyArrayObject* arr_obj, int numeric_type, const char* name)
509
// Make sure input has correct numeric type.
510
int arr_type = arr_obj->descr->type_num;
511
if (PyTypeNum_ISEXTENDED(numeric_type))
514
sprintf(msg, "Conversion Error: extended types not supported for variable '%s'",
516
throw_error(PyExc_TypeError, msg);
518
if (!PyArray_EquivTypenums(arr_type, numeric_type))
520
char* type_names[23] = {"bool", "byte", "ubyte","short", "ushort",
521
"int", "uint", "long", "ulong", "longlong", "ulonglong",
522
"float", "double", "longdouble", "cfloat", "cdouble",
523
"clongdouble", "object", "string", "unicode", "void", "ntype",
526
sprintf(msg,"received '%s' typed array instead of '%s' typed array for variable '%s'",
527
type_names[arr_type],type_names[numeric_type],name);
528
throw_error(PyExc_TypeError,msg);
533
numpy_type_handler x__numpy_type_handler = numpy_type_handler();
534
#define conversion_numpy_check_type x__numpy_type_handler.conversion_numpy_check_type
535
#define numpy_check_type x__numpy_type_handler.numpy_check_type
541
PyArrayObject* convert_to_numpy(PyObject* py_obj, const char* name)
543
// Incref occurs even if conversion fails so that
544
// the decref in cleanup_code has a matching incref.
546
if (!py_obj || !PyArray_Check(py_obj))
547
handle_conversion_error(py_obj,"numpy", name);
548
return (PyArrayObject*) py_obj;
551
PyArrayObject* py_to_numpy(PyObject* py_obj, const char* name)
553
// !! Pretty sure INCREF should only be called on success since
554
// !! py_to_xxx is used by the user -- not the code generator.
555
if (!py_obj || !PyArray_Check(py_obj))
556
handle_bad_type(py_obj,"numpy", name);
558
return (PyArrayObject*) py_obj;
562
numpy_handler x__numpy_handler = numpy_handler();
563
#define convert_to_numpy(py_obj,name) \
564
x__numpy_handler.convert_to_numpy(py_obj,name)
565
#define py_to_numpy(py_obj,name) \
566
x__numpy_handler.py_to_numpy(py_obj,name)
569
PyObject* numpy_to_py(PyObject* obj)
571
return (PyObject*) obj;
575
class catchall_handler
578
py::object convert_to_catchall(PyObject* py_obj, const char* name)
580
// Incref occurs even if conversion fails so that
581
// the decref in cleanup_code has a matching incref.
583
if (!py_obj || !(py_obj))
584
handle_conversion_error(py_obj,"catchall", name);
585
return py::object(py_obj);
588
py::object py_to_catchall(PyObject* py_obj, const char* name)
590
// !! Pretty sure INCREF should only be called on success since
591
// !! py_to_xxx is used by the user -- not the code generator.
592
if (!py_obj || !(py_obj))
593
handle_bad_type(py_obj,"catchall", name);
595
return py::object(py_obj);
599
catchall_handler x__catchall_handler = catchall_handler();
600
#define convert_to_catchall(py_obj,name) \
601
x__catchall_handler.convert_to_catchall(py_obj,name)
602
#define py_to_catchall(py_obj,name) \
603
x__catchall_handler.py_to_catchall(py_obj,name)
606
PyObject* catchall_to_py(PyObject* obj)
608
return (PyObject*) obj;
617
return fib1(a-2) + fib1(a-1);
622
int last, next_to_last, result;
626
last = next_to_last = 1;
627
for(int i = 2; i < a; i++ )
629
result = last + next_to_last;
638
static PyObject* c_fib1(PyObject*self, PyObject* args, PyObject* kywds)
640
py::object return_val;
641
int exception_occured = 0;
642
PyObject *py_local_dict = NULL;
643
static char *kwlist[] = {"a","local_dict", NULL};
649
if(!PyArg_ParseTupleAndKeywords(args,kywds,"O|O:c_fib1",kwlist,&py_a, &py_local_dict))
654
int a = convert_to_int(py_a,"a");
656
/*<function call here>*/
658
return_val = fib1(a);
661
py::dict local_dict = py::dict(py_local_dict);
667
return_val = py::object();
668
exception_occured = 1;
671
if(!(PyObject*)return_val && !exception_occured)
674
return_val = Py_None;
677
return return_val.disown();
679
static PyObject* c_fib2(PyObject*self, PyObject* args, PyObject* kywds)
681
py::object return_val;
682
int exception_occured = 0;
683
PyObject *py_local_dict = NULL;
684
static char *kwlist[] = {"a","local_dict", NULL};
690
if(!PyArg_ParseTupleAndKeywords(args,kywds,"O|O:c_fib2",kwlist,&py_a, &py_local_dict))
695
int a = convert_to_int(py_a,"a");
697
/*<function call here>*/
699
return_val = fib2(a);
702
py::dict local_dict = py::dict(py_local_dict);
708
return_val = py::object();
709
exception_occured = 1;
712
if(!(PyObject*)return_val && !exception_occured)
715
return_val = Py_None;
718
return return_val.disown();
722
static PyMethodDef compiled_methods[] =
724
{"c_fib1",(PyCFunction)c_fib1 , METH_VARARGS|METH_KEYWORDS},
725
{"c_fib2",(PyCFunction)c_fib2 , METH_VARARGS|METH_KEYWORDS},
726
{NULL, NULL} /* Sentinel */
729
PyMODINIT_FUNC initfibonacci_ext(void)
734
PyImport_ImportModule("numpy");
735
(void) Py_InitModule("fibonacci_ext", compiled_methods);
738
#ifdef __CPLUSCPLUS__
added
removed
Lib/weave/examples/fibonacci_ext.cpp
