12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364 |
- #!/usr/bin/env python3
- """
- C declarations, CPP macros, and C functions for f2py2e.
- Only required declarations/macros/functions will be used.
- Copyright 1999,2000 Pearu Peterson all rights reserved,
- Pearu Peterson <pearu@ioc.ee>
- Permission to use, modify, and distribute this software is given under the
- terms of the NumPy License.
- NO WARRANTY IS EXPRESSED OR IMPLIED. USE AT YOUR OWN RISK.
- $Date: 2005/05/06 11:42:34 $
- Pearu Peterson
- """
- import sys
- import copy
- from . import __version__
- f2py_version = __version__.version
- errmess = sys.stderr.write
- ##################### Definitions ##################
- outneeds = {'includes0': [], 'includes': [], 'typedefs': [], 'typedefs_generated': [],
- 'userincludes': [],
- 'cppmacros': [], 'cfuncs': [], 'callbacks': [], 'f90modhooks': [],
- 'commonhooks': []}
- needs = {}
- includes0 = {'includes0': '/*need_includes0*/'}
- includes = {'includes': '/*need_includes*/'}
- userincludes = {'userincludes': '/*need_userincludes*/'}
- typedefs = {'typedefs': '/*need_typedefs*/'}
- typedefs_generated = {'typedefs_generated': '/*need_typedefs_generated*/'}
- cppmacros = {'cppmacros': '/*need_cppmacros*/'}
- cfuncs = {'cfuncs': '/*need_cfuncs*/'}
- callbacks = {'callbacks': '/*need_callbacks*/'}
- f90modhooks = {'f90modhooks': '/*need_f90modhooks*/',
- 'initf90modhooksstatic': '/*initf90modhooksstatic*/',
- 'initf90modhooksdynamic': '/*initf90modhooksdynamic*/',
- }
- commonhooks = {'commonhooks': '/*need_commonhooks*/',
- 'initcommonhooks': '/*need_initcommonhooks*/',
- }
- ############ Includes ###################
- includes0['math.h'] = '#include <math.h>'
- includes0['string.h'] = '#include <string.h>'
- includes0['setjmp.h'] = '#include <setjmp.h>'
- includes['Python.h'] = '#include "Python.h"'
- needs['arrayobject.h'] = ['Python.h']
- includes['arrayobject.h'] = '''#define PY_ARRAY_UNIQUE_SYMBOL PyArray_API
- #include "arrayobject.h"'''
- includes['arrayobject.h'] = '#include "fortranobject.h"'
- includes['stdarg.h'] = '#include <stdarg.h>'
- ############# Type definitions ###############
- typedefs['unsigned_char'] = 'typedef unsigned char unsigned_char;'
- typedefs['unsigned_short'] = 'typedef unsigned short unsigned_short;'
- typedefs['unsigned_long'] = 'typedef unsigned long unsigned_long;'
- typedefs['signed_char'] = 'typedef signed char signed_char;'
- typedefs['long_long'] = """\
- #ifdef _WIN32
- typedef __int64 long_long;
- #else
- typedef long long long_long;
- typedef unsigned long long unsigned_long_long;
- #endif
- """
- typedefs['unsigned_long_long'] = """\
- #ifdef _WIN32
- typedef __uint64 long_long;
- #else
- typedef unsigned long long unsigned_long_long;
- #endif
- """
- typedefs['long_double'] = """\
- #ifndef _LONG_DOUBLE
- typedef long double long_double;
- #endif
- """
- typedefs[
- 'complex_long_double'] = 'typedef struct {long double r,i;} complex_long_double;'
- typedefs['complex_float'] = 'typedef struct {float r,i;} complex_float;'
- typedefs['complex_double'] = 'typedef struct {double r,i;} complex_double;'
- typedefs['string'] = """typedef char * string;"""
- ############### CPP macros ####################
- cppmacros['CFUNCSMESS'] = """\
- #ifdef DEBUGCFUNCS
- #define CFUNCSMESS(mess) fprintf(stderr,\"debug-capi:\"mess);
- #define CFUNCSMESSPY(mess,obj) CFUNCSMESS(mess) \\
- PyObject_Print((PyObject *)obj,stderr,Py_PRINT_RAW);\\
- fprintf(stderr,\"\\n\");
- #else
- #define CFUNCSMESS(mess)
- #define CFUNCSMESSPY(mess,obj)
- #endif
- """
- cppmacros['F_FUNC'] = """\
- #if defined(PREPEND_FORTRAN)
- #if defined(NO_APPEND_FORTRAN)
- #if defined(UPPERCASE_FORTRAN)
- #define F_FUNC(f,F) _##F
- #else
- #define F_FUNC(f,F) _##f
- #endif
- #else
- #if defined(UPPERCASE_FORTRAN)
- #define F_FUNC(f,F) _##F##_
- #else
- #define F_FUNC(f,F) _##f##_
- #endif
- #endif
- #else
- #if defined(NO_APPEND_FORTRAN)
- #if defined(UPPERCASE_FORTRAN)
- #define F_FUNC(f,F) F
- #else
- #define F_FUNC(f,F) f
- #endif
- #else
- #if defined(UPPERCASE_FORTRAN)
- #define F_FUNC(f,F) F##_
- #else
- #define F_FUNC(f,F) f##_
- #endif
- #endif
- #endif
- #if defined(UNDERSCORE_G77)
- #define F_FUNC_US(f,F) F_FUNC(f##_,F##_)
- #else
- #define F_FUNC_US(f,F) F_FUNC(f,F)
- #endif
- """
- cppmacros['F_WRAPPEDFUNC'] = """\
- #if defined(PREPEND_FORTRAN)
- #if defined(NO_APPEND_FORTRAN)
- #if defined(UPPERCASE_FORTRAN)
- #define F_WRAPPEDFUNC(f,F) _F2PYWRAP##F
- #else
- #define F_WRAPPEDFUNC(f,F) _f2pywrap##f
- #endif
- #else
- #if defined(UPPERCASE_FORTRAN)
- #define F_WRAPPEDFUNC(f,F) _F2PYWRAP##F##_
- #else
- #define F_WRAPPEDFUNC(f,F) _f2pywrap##f##_
- #endif
- #endif
- #else
- #if defined(NO_APPEND_FORTRAN)
- #if defined(UPPERCASE_FORTRAN)
- #define F_WRAPPEDFUNC(f,F) F2PYWRAP##F
- #else
- #define F_WRAPPEDFUNC(f,F) f2pywrap##f
- #endif
- #else
- #if defined(UPPERCASE_FORTRAN)
- #define F_WRAPPEDFUNC(f,F) F2PYWRAP##F##_
- #else
- #define F_WRAPPEDFUNC(f,F) f2pywrap##f##_
- #endif
- #endif
- #endif
- #if defined(UNDERSCORE_G77)
- #define F_WRAPPEDFUNC_US(f,F) F_WRAPPEDFUNC(f##_,F##_)
- #else
- #define F_WRAPPEDFUNC_US(f,F) F_WRAPPEDFUNC(f,F)
- #endif
- """
- cppmacros['F_MODFUNC'] = """\
- #if defined(F90MOD2CCONV1) /*E.g. Compaq Fortran */
- #if defined(NO_APPEND_FORTRAN)
- #define F_MODFUNCNAME(m,f) $ ## m ## $ ## f
- #else
- #define F_MODFUNCNAME(m,f) $ ## m ## $ ## f ## _
- #endif
- #endif
- #if defined(F90MOD2CCONV2) /*E.g. IBM XL Fortran, not tested though */
- #if defined(NO_APPEND_FORTRAN)
- #define F_MODFUNCNAME(m,f) __ ## m ## _MOD_ ## f
- #else
- #define F_MODFUNCNAME(m,f) __ ## m ## _MOD_ ## f ## _
- #endif
- #endif
- #if defined(F90MOD2CCONV3) /*E.g. MIPSPro Compilers */
- #if defined(NO_APPEND_FORTRAN)
- #define F_MODFUNCNAME(m,f) f ## .in. ## m
- #else
- #define F_MODFUNCNAME(m,f) f ## .in. ## m ## _
- #endif
- #endif
- /*
- #if defined(UPPERCASE_FORTRAN)
- #define F_MODFUNC(m,M,f,F) F_MODFUNCNAME(M,F)
- #else
- #define F_MODFUNC(m,M,f,F) F_MODFUNCNAME(m,f)
- #endif
- */
- #define F_MODFUNC(m,f) (*(f2pymodstruct##m##.##f))
- """
- cppmacros['SWAPUNSAFE'] = """\
- #define SWAP(a,b) (size_t)(a) = ((size_t)(a) ^ (size_t)(b));\\
- (size_t)(b) = ((size_t)(a) ^ (size_t)(b));\\
- (size_t)(a) = ((size_t)(a) ^ (size_t)(b))
- """
- cppmacros['SWAP'] = """\
- #define SWAP(a,b,t) {\\
- t *c;\\
- c = a;\\
- a = b;\\
- b = c;}
- """
- # cppmacros['ISCONTIGUOUS']='#define ISCONTIGUOUS(m) (PyArray_FLAGS(m) &
- # NPY_ARRAY_C_CONTIGUOUS)'
- cppmacros['PRINTPYOBJERR'] = """\
- #define PRINTPYOBJERR(obj)\\
- fprintf(stderr,\"#modulename#.error is related to \");\\
- PyObject_Print((PyObject *)obj,stderr,Py_PRINT_RAW);\\
- fprintf(stderr,\"\\n\");
- """
- cppmacros['MINMAX'] = """\
- #ifndef max
- #define max(a,b) ((a > b) ? (a) : (b))
- #endif
- #ifndef min
- #define min(a,b) ((a < b) ? (a) : (b))
- #endif
- #ifndef MAX
- #define MAX(a,b) ((a > b) ? (a) : (b))
- #endif
- #ifndef MIN
- #define MIN(a,b) ((a < b) ? (a) : (b))
- #endif
- """
- needs['len..'] = ['f2py_size']
- cppmacros['len..'] = """\
- #define rank(var) var ## _Rank
- #define shape(var,dim) var ## _Dims[dim]
- #define old_rank(var) (PyArray_NDIM((PyArrayObject *)(capi_ ## var ## _tmp)))
- #define old_shape(var,dim) PyArray_DIM(((PyArrayObject *)(capi_ ## var ## _tmp)),dim)
- #define fshape(var,dim) shape(var,rank(var)-dim-1)
- #define len(var) shape(var,0)
- #define flen(var) fshape(var,0)
- #define old_size(var) PyArray_SIZE((PyArrayObject *)(capi_ ## var ## _tmp))
- /* #define index(i) capi_i ## i */
- #define slen(var) capi_ ## var ## _len
- #define size(var, ...) f2py_size((PyArrayObject *)(capi_ ## var ## _tmp), ## __VA_ARGS__, -1)
- """
- needs['f2py_size'] = ['stdarg.h']
- cfuncs['f2py_size'] = """\
- static int f2py_size(PyArrayObject* var, ...)
- {
- npy_int sz = 0;
- npy_int dim;
- npy_int rank;
- va_list argp;
- va_start(argp, var);
- dim = va_arg(argp, npy_int);
- if (dim==-1)
- {
- sz = PyArray_SIZE(var);
- }
- else
- {
- rank = PyArray_NDIM(var);
- if (dim>=1 && dim<=rank)
- sz = PyArray_DIM(var, dim-1);
- else
- fprintf(stderr, \"f2py_size: 2nd argument value=%d fails to satisfy 1<=value<=%d. Result will be 0.\\n\", dim, rank);
- }
- va_end(argp);
- return sz;
- }
- """
- cppmacros[
- 'pyobj_from_char1'] = '#define pyobj_from_char1(v) (PyLong_FromLong(v))'
- cppmacros[
- 'pyobj_from_short1'] = '#define pyobj_from_short1(v) (PyLong_FromLong(v))'
- needs['pyobj_from_int1'] = ['signed_char']
- cppmacros['pyobj_from_int1'] = '#define pyobj_from_int1(v) (PyLong_FromLong(v))'
- cppmacros[
- 'pyobj_from_long1'] = '#define pyobj_from_long1(v) (PyLong_FromLong(v))'
- needs['pyobj_from_long_long1'] = ['long_long']
- cppmacros['pyobj_from_long_long1'] = """\
- #ifdef HAVE_LONG_LONG
- #define pyobj_from_long_long1(v) (PyLong_FromLongLong(v))
- #else
- #warning HAVE_LONG_LONG is not available. Redefining pyobj_from_long_long.
- #define pyobj_from_long_long1(v) (PyLong_FromLong(v))
- #endif
- """
- needs['pyobj_from_long_double1'] = ['long_double']
- cppmacros[
- 'pyobj_from_long_double1'] = '#define pyobj_from_long_double1(v) (PyFloat_FromDouble(v))'
- cppmacros[
- 'pyobj_from_double1'] = '#define pyobj_from_double1(v) (PyFloat_FromDouble(v))'
- cppmacros[
- 'pyobj_from_float1'] = '#define pyobj_from_float1(v) (PyFloat_FromDouble(v))'
- needs['pyobj_from_complex_long_double1'] = ['complex_long_double']
- cppmacros[
- 'pyobj_from_complex_long_double1'] = '#define pyobj_from_complex_long_double1(v) (PyComplex_FromDoubles(v.r,v.i))'
- needs['pyobj_from_complex_double1'] = ['complex_double']
- cppmacros[
- 'pyobj_from_complex_double1'] = '#define pyobj_from_complex_double1(v) (PyComplex_FromDoubles(v.r,v.i))'
- needs['pyobj_from_complex_float1'] = ['complex_float']
- cppmacros[
- 'pyobj_from_complex_float1'] = '#define pyobj_from_complex_float1(v) (PyComplex_FromDoubles(v.r,v.i))'
- needs['pyobj_from_string1'] = ['string']
- cppmacros[
- 'pyobj_from_string1'] = '#define pyobj_from_string1(v) (PyUnicode_FromString((char *)v))'
- needs['pyobj_from_string1size'] = ['string']
- cppmacros[
- 'pyobj_from_string1size'] = '#define pyobj_from_string1size(v,len) (PyUnicode_FromStringAndSize((char *)v, len))'
- needs['TRYPYARRAYTEMPLATE'] = ['PRINTPYOBJERR']
- cppmacros['TRYPYARRAYTEMPLATE'] = """\
- /* New SciPy */
- #define TRYPYARRAYTEMPLATECHAR case NPY_STRING: *(char *)(PyArray_DATA(arr))=*v; break;
- #define TRYPYARRAYTEMPLATELONG case NPY_LONG: *(long *)(PyArray_DATA(arr))=*v; break;
- #define TRYPYARRAYTEMPLATEOBJECT case NPY_OBJECT: PyArray_SETITEM(arr,PyArray_DATA(arr),pyobj_from_ ## ctype ## 1(*v)); break;
- #define TRYPYARRAYTEMPLATE(ctype,typecode) \\
- PyArrayObject *arr = NULL;\\
- if (!obj) return -2;\\
- if (!PyArray_Check(obj)) return -1;\\
- if (!(arr=(PyArrayObject *)obj)) {fprintf(stderr,\"TRYPYARRAYTEMPLATE:\");PRINTPYOBJERR(obj);return 0;}\\
- if (PyArray_DESCR(arr)->type==typecode) {*(ctype *)(PyArray_DATA(arr))=*v; return 1;}\\
- switch (PyArray_TYPE(arr)) {\\
- case NPY_DOUBLE: *(double *)(PyArray_DATA(arr))=*v; break;\\
- case NPY_INT: *(int *)(PyArray_DATA(arr))=*v; break;\\
- case NPY_LONG: *(long *)(PyArray_DATA(arr))=*v; break;\\
- case NPY_FLOAT: *(float *)(PyArray_DATA(arr))=*v; break;\\
- case NPY_CDOUBLE: *(double *)(PyArray_DATA(arr))=*v; break;\\
- case NPY_CFLOAT: *(float *)(PyArray_DATA(arr))=*v; break;\\
- case NPY_BOOL: *(npy_bool *)(PyArray_DATA(arr))=(*v!=0); break;\\
- case NPY_UBYTE: *(unsigned char *)(PyArray_DATA(arr))=*v; break;\\
- case NPY_BYTE: *(signed char *)(PyArray_DATA(arr))=*v; break;\\
- case NPY_SHORT: *(short *)(PyArray_DATA(arr))=*v; break;\\
- case NPY_USHORT: *(npy_ushort *)(PyArray_DATA(arr))=*v; break;\\
- case NPY_UINT: *(npy_uint *)(PyArray_DATA(arr))=*v; break;\\
- case NPY_ULONG: *(npy_ulong *)(PyArray_DATA(arr))=*v; break;\\
- case NPY_LONGLONG: *(npy_longlong *)(PyArray_DATA(arr))=*v; break;\\
- case NPY_ULONGLONG: *(npy_ulonglong *)(PyArray_DATA(arr))=*v; break;\\
- case NPY_LONGDOUBLE: *(npy_longdouble *)(PyArray_DATA(arr))=*v; break;\\
- case NPY_CLONGDOUBLE: *(npy_longdouble *)(PyArray_DATA(arr))=*v; break;\\
- case NPY_OBJECT: PyArray_SETITEM(arr, PyArray_DATA(arr), pyobj_from_ ## ctype ## 1(*v)); break;\\
- default: return -2;\\
- };\\
- return 1
- """
- needs['TRYCOMPLEXPYARRAYTEMPLATE'] = ['PRINTPYOBJERR']
- cppmacros['TRYCOMPLEXPYARRAYTEMPLATE'] = """\
- #define TRYCOMPLEXPYARRAYTEMPLATEOBJECT case NPY_OBJECT: PyArray_SETITEM(arr, PyArray_DATA(arr), pyobj_from_complex_ ## ctype ## 1((*v))); break;
- #define TRYCOMPLEXPYARRAYTEMPLATE(ctype,typecode)\\
- PyArrayObject *arr = NULL;\\
- if (!obj) return -2;\\
- if (!PyArray_Check(obj)) return -1;\\
- if (!(arr=(PyArrayObject *)obj)) {fprintf(stderr,\"TRYCOMPLEXPYARRAYTEMPLATE:\");PRINTPYOBJERR(obj);return 0;}\\
- if (PyArray_DESCR(arr)->type==typecode) {\\
- *(ctype *)(PyArray_DATA(arr))=(*v).r;\\
- *(ctype *)(PyArray_DATA(arr)+sizeof(ctype))=(*v).i;\\
- return 1;\\
- }\\
- switch (PyArray_TYPE(arr)) {\\
- case NPY_CDOUBLE: *(double *)(PyArray_DATA(arr))=(*v).r;*(double *)(PyArray_DATA(arr)+sizeof(double))=(*v).i;break;\\
- case NPY_CFLOAT: *(float *)(PyArray_DATA(arr))=(*v).r;*(float *)(PyArray_DATA(arr)+sizeof(float))=(*v).i;break;\\
- case NPY_DOUBLE: *(double *)(PyArray_DATA(arr))=(*v).r; break;\\
- case NPY_LONG: *(long *)(PyArray_DATA(arr))=(*v).r; break;\\
- case NPY_FLOAT: *(float *)(PyArray_DATA(arr))=(*v).r; break;\\
- case NPY_INT: *(int *)(PyArray_DATA(arr))=(*v).r; break;\\
- case NPY_SHORT: *(short *)(PyArray_DATA(arr))=(*v).r; break;\\
- case NPY_UBYTE: *(unsigned char *)(PyArray_DATA(arr))=(*v).r; break;\\
- case NPY_BYTE: *(signed char *)(PyArray_DATA(arr))=(*v).r; break;\\
- case NPY_BOOL: *(npy_bool *)(PyArray_DATA(arr))=((*v).r!=0 && (*v).i!=0); break;\\
- case NPY_USHORT: *(npy_ushort *)(PyArray_DATA(arr))=(*v).r; break;\\
- case NPY_UINT: *(npy_uint *)(PyArray_DATA(arr))=(*v).r; break;\\
- case NPY_ULONG: *(npy_ulong *)(PyArray_DATA(arr))=(*v).r; break;\\
- case NPY_LONGLONG: *(npy_longlong *)(PyArray_DATA(arr))=(*v).r; break;\\
- case NPY_ULONGLONG: *(npy_ulonglong *)(PyArray_DATA(arr))=(*v).r; break;\\
- case NPY_LONGDOUBLE: *(npy_longdouble *)(PyArray_DATA(arr))=(*v).r; break;\\
- case NPY_CLONGDOUBLE: *(npy_longdouble *)(PyArray_DATA(arr))=(*v).r;*(npy_longdouble *)(PyArray_DATA(arr)+sizeof(npy_longdouble))=(*v).i;break;\\
- case NPY_OBJECT: PyArray_SETITEM(arr, PyArray_DATA(arr), pyobj_from_complex_ ## ctype ## 1((*v))); break;\\
- default: return -2;\\
- };\\
- return -1;
- """
- # cppmacros['NUMFROMARROBJ']="""\
- # define NUMFROMARROBJ(typenum,ctype) \\
- # if (PyArray_Check(obj)) arr = (PyArrayObject *)obj;\\
- # else arr = (PyArrayObject *)PyArray_ContiguousFromObject(obj,typenum,0,0);\\
- # if (arr) {\\
- # if (PyArray_TYPE(arr)==NPY_OBJECT) {\\
- # if (!ctype ## _from_pyobj(v,(PyArray_DESCR(arr)->getitem)(PyArray_DATA(arr)),\"\"))\\
- # goto capi_fail;\\
- # } else {\\
- # (PyArray_DESCR(arr)->cast[typenum])(PyArray_DATA(arr),1,(char*)v,1,1);\\
- # }\\
- # if ((PyObject *)arr != obj) { Py_DECREF(arr); }\\
- # return 1;\\
- # }
- # """
- # XXX: Note that CNUMFROMARROBJ is identical with NUMFROMARROBJ
- # cppmacros['CNUMFROMARROBJ']="""\
- # define CNUMFROMARROBJ(typenum,ctype) \\
- # if (PyArray_Check(obj)) arr = (PyArrayObject *)obj;\\
- # else arr = (PyArrayObject *)PyArray_ContiguousFromObject(obj,typenum,0,0);\\
- # if (arr) {\\
- # if (PyArray_TYPE(arr)==NPY_OBJECT) {\\
- # if (!ctype ## _from_pyobj(v,(PyArray_DESCR(arr)->getitem)(PyArray_DATA(arr)),\"\"))\\
- # goto capi_fail;\\
- # } else {\\
- # (PyArray_DESCR(arr)->cast[typenum])((void *)(PyArray_DATA(arr)),1,(void *)(v),1,1);\\
- # }\\
- # if ((PyObject *)arr != obj) { Py_DECREF(arr); }\\
- # return 1;\\
- # }
- # """
- needs['GETSTRFROMPYTUPLE'] = ['STRINGCOPYN', 'PRINTPYOBJERR']
- cppmacros['GETSTRFROMPYTUPLE'] = """\
- #define GETSTRFROMPYTUPLE(tuple,index,str,len) {\\
- PyObject *rv_cb_str = PyTuple_GetItem((tuple),(index));\\
- if (rv_cb_str == NULL)\\
- goto capi_fail;\\
- if (PyBytes_Check(rv_cb_str)) {\\
- str[len-1]='\\0';\\
- STRINGCOPYN((str),PyBytes_AS_STRING((PyBytesObject*)rv_cb_str),(len));\\
- } else {\\
- PRINTPYOBJERR(rv_cb_str);\\
- PyErr_SetString(#modulename#_error,\"string object expected\");\\
- goto capi_fail;\\
- }\\
- }
- """
- cppmacros['GETSCALARFROMPYTUPLE'] = """\
- #define GETSCALARFROMPYTUPLE(tuple,index,var,ctype,mess) {\\
- if ((capi_tmp = PyTuple_GetItem((tuple),(index)))==NULL) goto capi_fail;\\
- if (!(ctype ## _from_pyobj((var),capi_tmp,mess)))\\
- goto capi_fail;\\
- }
- """
- cppmacros['FAILNULL'] = """\\
- #define FAILNULL(p) do { \\
- if ((p) == NULL) { \\
- PyErr_SetString(PyExc_MemoryError, "NULL pointer found"); \\
- goto capi_fail; \\
- } \\
- } while (0)
- """
- needs['MEMCOPY'] = ['string.h', 'FAILNULL']
- cppmacros['MEMCOPY'] = """\
- #define MEMCOPY(to,from,n)\\
- do { FAILNULL(to); FAILNULL(from); (void)memcpy(to,from,n); } while (0)
- """
- cppmacros['STRINGMALLOC'] = """\
- #define STRINGMALLOC(str,len)\\
- if ((str = (string)malloc(sizeof(char)*(len+1))) == NULL) {\\
- PyErr_SetString(PyExc_MemoryError, \"out of memory\");\\
- goto capi_fail;\\
- } else {\\
- (str)[len] = '\\0';\\
- }
- """
- cppmacros['STRINGFREE'] = """\
- #define STRINGFREE(str) do {if (!(str == NULL)) free(str);} while (0)
- """
- needs['STRINGCOPYN'] = ['string.h', 'FAILNULL']
- cppmacros['STRINGCOPYN'] = """\
- #define STRINGCOPYN(to,from,buf_size) \\
- do { \\
- int _m = (buf_size); \\
- char *_to = (to); \\
- char *_from = (from); \\
- FAILNULL(_to); FAILNULL(_from); \\
- (void)strncpy(_to, _from, sizeof(char)*_m); \\
- _to[_m-1] = '\\0'; \\
- /* Padding with spaces instead of nulls */ \\
- for (_m -= 2; _m >= 0 && _to[_m] == '\\0'; _m--) { \\
- _to[_m] = ' '; \\
- } \\
- } while (0)
- """
- needs['STRINGCOPY'] = ['string.h', 'FAILNULL']
- cppmacros['STRINGCOPY'] = """\
- #define STRINGCOPY(to,from)\\
- do { FAILNULL(to); FAILNULL(from); (void)strcpy(to,from); } while (0)
- """
- cppmacros['CHECKGENERIC'] = """\
- #define CHECKGENERIC(check,tcheck,name) \\
- if (!(check)) {\\
- PyErr_SetString(#modulename#_error,\"(\"tcheck\") failed for \"name);\\
- /*goto capi_fail;*/\\
- } else """
- cppmacros['CHECKARRAY'] = """\
- #define CHECKARRAY(check,tcheck,name) \\
- if (!(check)) {\\
- PyErr_SetString(#modulename#_error,\"(\"tcheck\") failed for \"name);\\
- /*goto capi_fail;*/\\
- } else """
- cppmacros['CHECKSTRING'] = """\
- #define CHECKSTRING(check,tcheck,name,show,var)\\
- if (!(check)) {\\
- char errstring[256];\\
- sprintf(errstring, \"%s: \"show, \"(\"tcheck\") failed for \"name, slen(var), var);\\
- PyErr_SetString(#modulename#_error, errstring);\\
- /*goto capi_fail;*/\\
- } else """
- cppmacros['CHECKSCALAR'] = """\
- #define CHECKSCALAR(check,tcheck,name,show,var)\\
- if (!(check)) {\\
- char errstring[256];\\
- sprintf(errstring, \"%s: \"show, \"(\"tcheck\") failed for \"name, var);\\
- PyErr_SetString(#modulename#_error,errstring);\\
- /*goto capi_fail;*/\\
- } else """
- # cppmacros['CHECKDIMS']="""\
- # define CHECKDIMS(dims,rank) \\
- # for (int i=0;i<(rank);i++)\\
- # if (dims[i]<0) {\\
- # fprintf(stderr,\"Unspecified array argument requires a complete dimension specification.\\n\");\\
- # goto capi_fail;\\
- # }
- # """
- cppmacros[
- 'ARRSIZE'] = '#define ARRSIZE(dims,rank) (_PyArray_multiply_list(dims,rank))'
- cppmacros['OLDPYNUM'] = """\
- #ifdef OLDPYNUM
- #error You need to install NumPy version 0.13 or higher. See https://scipy.org/install.html
- #endif
- """
- cppmacros["F2PY_THREAD_LOCAL_DECL"] = """\
- #ifndef F2PY_THREAD_LOCAL_DECL
- #if defined(_MSC_VER) \\
- || defined(_WIN32) || defined(_WIN64) \\
- || defined(__MINGW32__) || defined(__MINGW64__)
- #define F2PY_THREAD_LOCAL_DECL __declspec(thread)
- #elif defined(__STDC_VERSION__) \\
- && (__STDC_VERSION__ >= 201112L) \\
- && !defined(__STDC_NO_THREADS__) \\
- && (!defined(__GLIBC__) || __GLIBC__ > 2 || (__GLIBC__ == 2 && __GLIBC_MINOR__ > 12))
- /* __STDC_NO_THREADS__ was first defined in a maintenance release of glibc 2.12,
- see https://lists.gnu.org/archive/html/commit-hurd/2012-07/msg00180.html,
- so `!defined(__STDC_NO_THREADS__)` may give false positive for the existence
- of `threads.h` when using an older release of glibc 2.12 */
- #include <threads.h>
- #define F2PY_THREAD_LOCAL_DECL thread_local
- #elif defined(__GNUC__) \\
- && (__GNUC__ > 4 || (__GNUC__ == 4 && (__GNUC_MINOR__ >= 4)))
- #define F2PY_THREAD_LOCAL_DECL __thread
- #endif
- #endif
- """
- ################# C functions ###############
- cfuncs['calcarrindex'] = """\
- static int calcarrindex(int *i,PyArrayObject *arr) {
- int k,ii = i[0];
- for (k=1; k < PyArray_NDIM(arr); k++)
- ii += (ii*(PyArray_DIM(arr,k) - 1)+i[k]); /* assuming contiguous arr */
- return ii;
- }"""
- cfuncs['calcarrindextr'] = """\
- static int calcarrindextr(int *i,PyArrayObject *arr) {
- int k,ii = i[PyArray_NDIM(arr)-1];
- for (k=1; k < PyArray_NDIM(arr); k++)
- ii += (ii*(PyArray_DIM(arr,PyArray_NDIM(arr)-k-1) - 1)+i[PyArray_NDIM(arr)-k-1]); /* assuming contiguous arr */
- return ii;
- }"""
- cfuncs['forcomb'] = """\
- static struct { int nd;npy_intp *d;int *i,*i_tr,tr; } forcombcache;
- static int initforcomb(npy_intp *dims,int nd,int tr) {
- int k;
- if (dims==NULL) return 0;
- if (nd<0) return 0;
- forcombcache.nd = nd;
- forcombcache.d = dims;
- forcombcache.tr = tr;
- if ((forcombcache.i = (int *)malloc(sizeof(int)*nd))==NULL) return 0;
- if ((forcombcache.i_tr = (int *)malloc(sizeof(int)*nd))==NULL) return 0;
- for (k=1;k<nd;k++) {
- forcombcache.i[k] = forcombcache.i_tr[nd-k-1] = 0;
- }
- forcombcache.i[0] = forcombcache.i_tr[nd-1] = -1;
- return 1;
- }
- static int *nextforcomb(void) {
- int j,*i,*i_tr,k;
- int nd=forcombcache.nd;
- if ((i=forcombcache.i) == NULL) return NULL;
- if ((i_tr=forcombcache.i_tr) == NULL) return NULL;
- if (forcombcache.d == NULL) return NULL;
- i[0]++;
- if (i[0]==forcombcache.d[0]) {
- j=1;
- while ((j<nd) && (i[j]==forcombcache.d[j]-1)) j++;
- if (j==nd) {
- free(i);
- free(i_tr);
- return NULL;
- }
- for (k=0;k<j;k++) i[k] = i_tr[nd-k-1] = 0;
- i[j]++;
- i_tr[nd-j-1]++;
- } else
- i_tr[nd-1]++;
- if (forcombcache.tr) return i_tr;
- return i;
- }"""
- needs['try_pyarr_from_string'] = ['STRINGCOPYN', 'PRINTPYOBJERR', 'string']
- cfuncs['try_pyarr_from_string'] = """\
- static int try_pyarr_from_string(PyObject *obj,const string str) {
- PyArrayObject *arr = NULL;
- if (PyArray_Check(obj) && (!((arr = (PyArrayObject *)obj) == NULL)))
- { STRINGCOPYN(PyArray_DATA(arr),str,PyArray_NBYTES(arr)); }
- return 1;
- capi_fail:
- PRINTPYOBJERR(obj);
- PyErr_SetString(#modulename#_error,\"try_pyarr_from_string failed\");
- return 0;
- }
- """
- needs['string_from_pyobj'] = ['string', 'STRINGMALLOC', 'STRINGCOPYN']
- cfuncs['string_from_pyobj'] = """\
- static int
- string_from_pyobj(string *str,int *len,const string inistr,PyObject *obj,const char *errmess)
- {
- PyArrayObject *arr = NULL;
- PyObject *tmp = NULL;
- #ifdef DEBUGCFUNCS
- fprintf(stderr,\"string_from_pyobj(str='%s',len=%d,inistr='%s',obj=%p)\\n\",(char*)str,*len,(char *)inistr,obj);
- #endif
- if (obj == Py_None) {
- if (*len == -1)
- *len = strlen(inistr); /* Will this cause problems? */
- STRINGMALLOC(*str,*len);
- STRINGCOPYN(*str,inistr,*len+1);
- return 1;
- }
- if (PyArray_Check(obj)) {
- if ((arr = (PyArrayObject *)obj) == NULL)
- goto capi_fail;
- if (!ISCONTIGUOUS(arr)) {
- PyErr_SetString(PyExc_ValueError,\"array object is non-contiguous.\");
- goto capi_fail;
- }
- if (*len == -1)
- *len = (PyArray_ITEMSIZE(arr))*PyArray_SIZE(arr);
- STRINGMALLOC(*str,*len);
- STRINGCOPYN(*str,PyArray_DATA(arr),*len+1);
- return 1;
- }
- if (PyBytes_Check(obj)) {
- tmp = obj;
- Py_INCREF(tmp);
- }
- else if (PyUnicode_Check(obj)) {
- tmp = PyUnicode_AsASCIIString(obj);
- }
- else {
- PyObject *tmp2;
- tmp2 = PyObject_Str(obj);
- if (tmp2) {
- tmp = PyUnicode_AsASCIIString(tmp2);
- Py_DECREF(tmp2);
- }
- else {
- tmp = NULL;
- }
- }
- if (tmp == NULL) goto capi_fail;
- if (*len == -1)
- *len = PyBytes_GET_SIZE(tmp);
- STRINGMALLOC(*str,*len);
- STRINGCOPYN(*str,PyBytes_AS_STRING(tmp),*len+1);
- Py_DECREF(tmp);
- return 1;
- capi_fail:
- Py_XDECREF(tmp);
- {
- PyObject* err = PyErr_Occurred();
- if (err == NULL) {
- err = #modulename#_error;
- }
- PyErr_SetString(err, errmess);
- }
- return 0;
- }
- """
- needs['char_from_pyobj'] = ['int_from_pyobj']
- cfuncs['char_from_pyobj'] = """\
- static int
- char_from_pyobj(char* v, PyObject *obj, const char *errmess) {
- int i = 0;
- if (int_from_pyobj(&i, obj, errmess)) {
- *v = (char)i;
- return 1;
- }
- return 0;
- }
- """
- needs['signed_char_from_pyobj'] = ['int_from_pyobj', 'signed_char']
- cfuncs['signed_char_from_pyobj'] = """\
- static int
- signed_char_from_pyobj(signed_char* v, PyObject *obj, const char *errmess) {
- int i = 0;
- if (int_from_pyobj(&i, obj, errmess)) {
- *v = (signed_char)i;
- return 1;
- }
- return 0;
- }
- """
- needs['short_from_pyobj'] = ['int_from_pyobj']
- cfuncs['short_from_pyobj'] = """\
- static int
- short_from_pyobj(short* v, PyObject *obj, const char *errmess) {
- int i = 0;
- if (int_from_pyobj(&i, obj, errmess)) {
- *v = (short)i;
- return 1;
- }
- return 0;
- }
- """
- cfuncs['int_from_pyobj'] = """\
- static int
- int_from_pyobj(int* v, PyObject *obj, const char *errmess)
- {
- PyObject* tmp = NULL;
- if (PyLong_Check(obj)) {
- *v = Npy__PyLong_AsInt(obj);
- return !(*v == -1 && PyErr_Occurred());
- }
- tmp = PyNumber_Long(obj);
- if (tmp) {
- *v = Npy__PyLong_AsInt(tmp);
- Py_DECREF(tmp);
- return !(*v == -1 && PyErr_Occurred());
- }
- if (PyComplex_Check(obj))
- tmp = PyObject_GetAttrString(obj,\"real\");
- else if (PyBytes_Check(obj) || PyUnicode_Check(obj))
- /*pass*/;
- else if (PySequence_Check(obj))
- tmp = PySequence_GetItem(obj, 0);
- if (tmp) {
- PyErr_Clear();
- if (int_from_pyobj(v, tmp, errmess)) {
- Py_DECREF(tmp);
- return 1;
- }
- Py_DECREF(tmp);
- }
- {
- PyObject* err = PyErr_Occurred();
- if (err == NULL) {
- err = #modulename#_error;
- }
- PyErr_SetString(err, errmess);
- }
- return 0;
- }
- """
- cfuncs['long_from_pyobj'] = """\
- static int
- long_from_pyobj(long* v, PyObject *obj, const char *errmess) {
- PyObject* tmp = NULL;
- if (PyLong_Check(obj)) {
- *v = PyLong_AsLong(obj);
- return !(*v == -1 && PyErr_Occurred());
- }
- tmp = PyNumber_Long(obj);
- if (tmp) {
- *v = PyLong_AsLong(tmp);
- Py_DECREF(tmp);
- return !(*v == -1 && PyErr_Occurred());
- }
- if (PyComplex_Check(obj))
- tmp = PyObject_GetAttrString(obj,\"real\");
- else if (PyBytes_Check(obj) || PyUnicode_Check(obj))
- /*pass*/;
- else if (PySequence_Check(obj))
- tmp = PySequence_GetItem(obj,0);
- if (tmp) {
- PyErr_Clear();
- if (long_from_pyobj(v, tmp, errmess)) {
- Py_DECREF(tmp);
- return 1;
- }
- Py_DECREF(tmp);
- }
- {
- PyObject* err = PyErr_Occurred();
- if (err == NULL) {
- err = #modulename#_error;
- }
- PyErr_SetString(err, errmess);
- }
- return 0;
- }
- """
- needs['long_long_from_pyobj'] = ['long_long']
- cfuncs['long_long_from_pyobj'] = """\
- static int
- long_long_from_pyobj(long_long* v, PyObject *obj, const char *errmess)
- {
- PyObject* tmp = NULL;
- if (PyLong_Check(obj)) {
- *v = PyLong_AsLongLong(obj);
- return !(*v == -1 && PyErr_Occurred());
- }
- tmp = PyNumber_Long(obj);
- if (tmp) {
- *v = PyLong_AsLongLong(tmp);
- Py_DECREF(tmp);
- return !(*v == -1 && PyErr_Occurred());
- }
- if (PyComplex_Check(obj))
- tmp = PyObject_GetAttrString(obj,\"real\");
- else if (PyBytes_Check(obj) || PyUnicode_Check(obj))
- /*pass*/;
- else if (PySequence_Check(obj))
- tmp = PySequence_GetItem(obj,0);
- if (tmp) {
- PyErr_Clear();
- if (long_long_from_pyobj(v, tmp, errmess)) {
- Py_DECREF(tmp);
- return 1;
- }
- Py_DECREF(tmp);
- }
- {
- PyObject* err = PyErr_Occurred();
- if (err == NULL) {
- err = #modulename#_error;
- }
- PyErr_SetString(err,errmess);
- }
- return 0;
- }
- """
- needs['long_double_from_pyobj'] = ['double_from_pyobj', 'long_double']
- cfuncs['long_double_from_pyobj'] = """\
- static int
- long_double_from_pyobj(long_double* v, PyObject *obj, const char *errmess)
- {
- double d=0;
- if (PyArray_CheckScalar(obj)){
- if PyArray_IsScalar(obj, LongDouble) {
- PyArray_ScalarAsCtype(obj, v);
- return 1;
- }
- else if (PyArray_Check(obj) && PyArray_TYPE(obj) == NPY_LONGDOUBLE) {
- (*v) = *((npy_longdouble *)PyArray_DATA(obj));
- return 1;
- }
- }
- if (double_from_pyobj(&d, obj, errmess)) {
- *v = (long_double)d;
- return 1;
- }
- return 0;
- }
- """
- cfuncs['double_from_pyobj'] = """\
- static int
- double_from_pyobj(double* v, PyObject *obj, const char *errmess)
- {
- PyObject* tmp = NULL;
- if (PyFloat_Check(obj)) {
- *v = PyFloat_AsDouble(obj);
- return !(*v == -1.0 && PyErr_Occurred());
- }
- tmp = PyNumber_Float(obj);
- if (tmp) {
- *v = PyFloat_AsDouble(tmp);
- Py_DECREF(tmp);
- return !(*v == -1.0 && PyErr_Occurred());
- }
- if (PyComplex_Check(obj))
- tmp = PyObject_GetAttrString(obj,\"real\");
- else if (PyBytes_Check(obj) || PyUnicode_Check(obj))
- /*pass*/;
- else if (PySequence_Check(obj))
- tmp = PySequence_GetItem(obj,0);
- if (tmp) {
- PyErr_Clear();
- if (double_from_pyobj(v,tmp,errmess)) {Py_DECREF(tmp); return 1;}
- Py_DECREF(tmp);
- }
- {
- PyObject* err = PyErr_Occurred();
- if (err==NULL) err = #modulename#_error;
- PyErr_SetString(err,errmess);
- }
- return 0;
- }
- """
- needs['float_from_pyobj'] = ['double_from_pyobj']
- cfuncs['float_from_pyobj'] = """\
- static int
- float_from_pyobj(float* v, PyObject *obj, const char *errmess)
- {
- double d=0.0;
- if (double_from_pyobj(&d,obj,errmess)) {
- *v = (float)d;
- return 1;
- }
- return 0;
- }
- """
- needs['complex_long_double_from_pyobj'] = ['complex_long_double', 'long_double',
- 'complex_double_from_pyobj']
- cfuncs['complex_long_double_from_pyobj'] = """\
- static int
- complex_long_double_from_pyobj(complex_long_double* v, PyObject *obj, const char *errmess)
- {
- complex_double cd = {0.0,0.0};
- if (PyArray_CheckScalar(obj)){
- if PyArray_IsScalar(obj, CLongDouble) {
- PyArray_ScalarAsCtype(obj, v);
- return 1;
- }
- else if (PyArray_Check(obj) && PyArray_TYPE(obj)==NPY_CLONGDOUBLE) {
- (*v).r = ((npy_clongdouble *)PyArray_DATA(obj))->real;
- (*v).i = ((npy_clongdouble *)PyArray_DATA(obj))->imag;
- return 1;
- }
- }
- if (complex_double_from_pyobj(&cd,obj,errmess)) {
- (*v).r = (long_double)cd.r;
- (*v).i = (long_double)cd.i;
- return 1;
- }
- return 0;
- }
- """
- needs['complex_double_from_pyobj'] = ['complex_double']
- cfuncs['complex_double_from_pyobj'] = """\
- static int
- complex_double_from_pyobj(complex_double* v, PyObject *obj, const char *errmess) {
- Py_complex c;
- if (PyComplex_Check(obj)) {
- c = PyComplex_AsCComplex(obj);
- (*v).r = c.real;
- (*v).i = c.imag;
- return 1;
- }
- if (PyArray_IsScalar(obj, ComplexFloating)) {
- if (PyArray_IsScalar(obj, CFloat)) {
- npy_cfloat new;
- PyArray_ScalarAsCtype(obj, &new);
- (*v).r = (double)new.real;
- (*v).i = (double)new.imag;
- }
- else if (PyArray_IsScalar(obj, CLongDouble)) {
- npy_clongdouble new;
- PyArray_ScalarAsCtype(obj, &new);
- (*v).r = (double)new.real;
- (*v).i = (double)new.imag;
- }
- else { /* if (PyArray_IsScalar(obj, CDouble)) */
- PyArray_ScalarAsCtype(obj, v);
- }
- return 1;
- }
- if (PyArray_CheckScalar(obj)) { /* 0-dim array or still array scalar */
- PyObject *arr;
- if (PyArray_Check(obj)) {
- arr = PyArray_Cast((PyArrayObject *)obj, NPY_CDOUBLE);
- }
- else {
- arr = PyArray_FromScalar(obj, PyArray_DescrFromType(NPY_CDOUBLE));
- }
- if (arr == NULL) {
- return 0;
- }
- (*v).r = ((npy_cdouble *)PyArray_DATA(arr))->real;
- (*v).i = ((npy_cdouble *)PyArray_DATA(arr))->imag;
- Py_DECREF(arr);
- return 1;
- }
- /* Python does not provide PyNumber_Complex function :-( */
- (*v).i = 0.0;
- if (PyFloat_Check(obj)) {
- (*v).r = PyFloat_AsDouble(obj);
- return !((*v).r == -1.0 && PyErr_Occurred());
- }
- if (PyLong_Check(obj)) {
- (*v).r = PyLong_AsDouble(obj);
- return !((*v).r == -1.0 && PyErr_Occurred());
- }
- if (PySequence_Check(obj) && !(PyBytes_Check(obj) || PyUnicode_Check(obj))) {
- PyObject *tmp = PySequence_GetItem(obj,0);
- if (tmp) {
- if (complex_double_from_pyobj(v,tmp,errmess)) {
- Py_DECREF(tmp);
- return 1;
- }
- Py_DECREF(tmp);
- }
- }
- {
- PyObject* err = PyErr_Occurred();
- if (err==NULL)
- err = PyExc_TypeError;
- PyErr_SetString(err,errmess);
- }
- return 0;
- }
- """
- needs['complex_float_from_pyobj'] = [
- 'complex_float', 'complex_double_from_pyobj']
- cfuncs['complex_float_from_pyobj'] = """\
- static int
- complex_float_from_pyobj(complex_float* v,PyObject *obj,const char *errmess)
- {
- complex_double cd={0.0,0.0};
- if (complex_double_from_pyobj(&cd,obj,errmess)) {
- (*v).r = (float)cd.r;
- (*v).i = (float)cd.i;
- return 1;
- }
- return 0;
- }
- """
- needs['try_pyarr_from_char'] = ['pyobj_from_char1', 'TRYPYARRAYTEMPLATE']
- cfuncs[
- 'try_pyarr_from_char'] = 'static int try_pyarr_from_char(PyObject* obj,char* v) {\n TRYPYARRAYTEMPLATE(char,\'c\');\n}\n'
- needs['try_pyarr_from_signed_char'] = ['TRYPYARRAYTEMPLATE', 'unsigned_char']
- cfuncs[
- 'try_pyarr_from_unsigned_char'] = 'static int try_pyarr_from_unsigned_char(PyObject* obj,unsigned_char* v) {\n TRYPYARRAYTEMPLATE(unsigned_char,\'b\');\n}\n'
- needs['try_pyarr_from_signed_char'] = ['TRYPYARRAYTEMPLATE', 'signed_char']
- cfuncs[
- 'try_pyarr_from_signed_char'] = 'static int try_pyarr_from_signed_char(PyObject* obj,signed_char* v) {\n TRYPYARRAYTEMPLATE(signed_char,\'1\');\n}\n'
- needs['try_pyarr_from_short'] = ['pyobj_from_short1', 'TRYPYARRAYTEMPLATE']
- cfuncs[
- 'try_pyarr_from_short'] = 'static int try_pyarr_from_short(PyObject* obj,short* v) {\n TRYPYARRAYTEMPLATE(short,\'s\');\n}\n'
- needs['try_pyarr_from_int'] = ['pyobj_from_int1', 'TRYPYARRAYTEMPLATE']
- cfuncs[
- 'try_pyarr_from_int'] = 'static int try_pyarr_from_int(PyObject* obj,int* v) {\n TRYPYARRAYTEMPLATE(int,\'i\');\n}\n'
- needs['try_pyarr_from_long'] = ['pyobj_from_long1', 'TRYPYARRAYTEMPLATE']
- cfuncs[
- 'try_pyarr_from_long'] = 'static int try_pyarr_from_long(PyObject* obj,long* v) {\n TRYPYARRAYTEMPLATE(long,\'l\');\n}\n'
- needs['try_pyarr_from_long_long'] = [
- 'pyobj_from_long_long1', 'TRYPYARRAYTEMPLATE', 'long_long']
- cfuncs[
- 'try_pyarr_from_long_long'] = 'static int try_pyarr_from_long_long(PyObject* obj,long_long* v) {\n TRYPYARRAYTEMPLATE(long_long,\'L\');\n}\n'
- needs['try_pyarr_from_float'] = ['pyobj_from_float1', 'TRYPYARRAYTEMPLATE']
- cfuncs[
- 'try_pyarr_from_float'] = 'static int try_pyarr_from_float(PyObject* obj,float* v) {\n TRYPYARRAYTEMPLATE(float,\'f\');\n}\n'
- needs['try_pyarr_from_double'] = ['pyobj_from_double1', 'TRYPYARRAYTEMPLATE']
- cfuncs[
- 'try_pyarr_from_double'] = 'static int try_pyarr_from_double(PyObject* obj,double* v) {\n TRYPYARRAYTEMPLATE(double,\'d\');\n}\n'
- needs['try_pyarr_from_complex_float'] = [
- 'pyobj_from_complex_float1', 'TRYCOMPLEXPYARRAYTEMPLATE', 'complex_float']
- cfuncs[
- 'try_pyarr_from_complex_float'] = 'static int try_pyarr_from_complex_float(PyObject* obj,complex_float* v) {\n TRYCOMPLEXPYARRAYTEMPLATE(float,\'F\');\n}\n'
- needs['try_pyarr_from_complex_double'] = [
- 'pyobj_from_complex_double1', 'TRYCOMPLEXPYARRAYTEMPLATE', 'complex_double']
- cfuncs[
- 'try_pyarr_from_complex_double'] = 'static int try_pyarr_from_complex_double(PyObject* obj,complex_double* v) {\n TRYCOMPLEXPYARRAYTEMPLATE(double,\'D\');\n}\n'
- needs['create_cb_arglist'] = ['CFUNCSMESS', 'PRINTPYOBJERR', 'MINMAX']
- # create the list of arguments to be used when calling back to python
- cfuncs['create_cb_arglist'] = """\
- static int
- create_cb_arglist(PyObject* fun, PyTupleObject* xa , const int maxnofargs,
- const int nofoptargs, int *nofargs, PyTupleObject **args,
- const char *errmess)
- {
- PyObject *tmp = NULL;
- PyObject *tmp_fun = NULL;
- Py_ssize_t tot, opt, ext, siz, i, di = 0;
- CFUNCSMESS(\"create_cb_arglist\\n\");
- tot=opt=ext=siz=0;
- /* Get the total number of arguments */
- if (PyFunction_Check(fun)) {
- tmp_fun = fun;
- Py_INCREF(tmp_fun);
- }
- else {
- di = 1;
- if (PyObject_HasAttrString(fun,\"im_func\")) {
- tmp_fun = PyObject_GetAttrString(fun,\"im_func\");
- }
- else if (PyObject_HasAttrString(fun,\"__call__\")) {
- tmp = PyObject_GetAttrString(fun,\"__call__\");
- if (PyObject_HasAttrString(tmp,\"im_func\"))
- tmp_fun = PyObject_GetAttrString(tmp,\"im_func\");
- else {
- tmp_fun = fun; /* built-in function */
- Py_INCREF(tmp_fun);
- tot = maxnofargs;
- if (PyCFunction_Check(fun)) {
- /* In case the function has a co_argcount (like on PyPy) */
- di = 0;
- }
- if (xa != NULL)
- tot += PyTuple_Size((PyObject *)xa);
- }
- Py_XDECREF(tmp);
- }
- else if (PyFortran_Check(fun) || PyFortran_Check1(fun)) {
- tot = maxnofargs;
- if (xa != NULL)
- tot += PyTuple_Size((PyObject *)xa);
- tmp_fun = fun;
- Py_INCREF(tmp_fun);
- }
- else if (F2PyCapsule_Check(fun)) {
- tot = maxnofargs;
- if (xa != NULL)
- ext = PyTuple_Size((PyObject *)xa);
- if(ext>0) {
- fprintf(stderr,\"extra arguments tuple cannot be used with CObject call-back\\n\");
- goto capi_fail;
- }
- tmp_fun = fun;
- Py_INCREF(tmp_fun);
- }
- }
- if (tmp_fun == NULL) {
- fprintf(stderr,
- \"Call-back argument must be function|instance|instance.__call__|f2py-function \"
- \"but got %s.\\n\",
- ((fun == NULL) ? \"NULL\" : Py_TYPE(fun)->tp_name));
- goto capi_fail;
- }
- if (PyObject_HasAttrString(tmp_fun,\"__code__\")) {
- if (PyObject_HasAttrString(tmp = PyObject_GetAttrString(tmp_fun,\"__code__\"),\"co_argcount\")) {
- PyObject *tmp_argcount = PyObject_GetAttrString(tmp,\"co_argcount\");
- Py_DECREF(tmp);
- if (tmp_argcount == NULL) {
- goto capi_fail;
- }
- tot = PyLong_AsSsize_t(tmp_argcount) - di;
- Py_DECREF(tmp_argcount);
- }
- }
- /* Get the number of optional arguments */
- if (PyObject_HasAttrString(tmp_fun,\"__defaults__\")) {
- if (PyTuple_Check(tmp = PyObject_GetAttrString(tmp_fun,\"__defaults__\")))
- opt = PyTuple_Size(tmp);
- Py_XDECREF(tmp);
- }
- /* Get the number of extra arguments */
- if (xa != NULL)
- ext = PyTuple_Size((PyObject *)xa);
- /* Calculate the size of call-backs argument list */
- siz = MIN(maxnofargs+ext,tot);
- *nofargs = MAX(0,siz-ext);
- #ifdef DEBUGCFUNCS
- fprintf(stderr,
- \"debug-capi:create_cb_arglist:maxnofargs(-nofoptargs),\"
- \"tot,opt,ext,siz,nofargs = %d(-%d), %zd, %zd, %zd, %zd, %d\\n\",
- maxnofargs, nofoptargs, tot, opt, ext, siz, *nofargs);
- #endif
- if (siz < tot-opt) {
- fprintf(stderr,
- \"create_cb_arglist: Failed to build argument list \"
- \"(siz) with enough arguments (tot-opt) required by \"
- \"user-supplied function (siz,tot,opt=%zd, %zd, %zd).\\n\",
- siz, tot, opt);
- goto capi_fail;
- }
- /* Initialize argument list */
- *args = (PyTupleObject *)PyTuple_New(siz);
- for (i=0;i<*nofargs;i++) {
- Py_INCREF(Py_None);
- PyTuple_SET_ITEM((PyObject *)(*args),i,Py_None);
- }
- if (xa != NULL)
- for (i=(*nofargs);i<siz;i++) {
- tmp = PyTuple_GetItem((PyObject *)xa,i-(*nofargs));
- Py_INCREF(tmp);
- PyTuple_SET_ITEM(*args,i,tmp);
- }
- CFUNCSMESS(\"create_cb_arglist-end\\n\");
- Py_DECREF(tmp_fun);
- return 1;
- capi_fail:
- if (PyErr_Occurred() == NULL)
- PyErr_SetString(#modulename#_error, errmess);
- Py_XDECREF(tmp_fun);
- return 0;
- }
- """
- def buildcfuncs():
- from .capi_maps import c2capi_map
- for k in c2capi_map.keys():
- m = 'pyarr_from_p_%s1' % k
- cppmacros[
- m] = '#define %s(v) (PyArray_SimpleNewFromData(0,NULL,%s,(char *)v))' % (m, c2capi_map[k])
- k = 'string'
- m = 'pyarr_from_p_%s1' % k
- # NPY_CHAR compatibility, NPY_STRING with itemsize 1
- cppmacros[
- m] = '#define %s(v,dims) (PyArray_New(&PyArray_Type, 1, dims, NPY_STRING, NULL, v, 1, NPY_ARRAY_CARRAY, NULL))' % (m)
- ############ Auxiliary functions for sorting needs ###################
- def append_needs(need, flag=1):
- # This function modifies the contents of the global `outneeds` dict.
- if isinstance(need, list):
- for n in need:
- append_needs(n, flag)
- elif isinstance(need, str):
- if not need:
- return
- if need in includes0:
- n = 'includes0'
- elif need in includes:
- n = 'includes'
- elif need in typedefs:
- n = 'typedefs'
- elif need in typedefs_generated:
- n = 'typedefs_generated'
- elif need in cppmacros:
- n = 'cppmacros'
- elif need in cfuncs:
- n = 'cfuncs'
- elif need in callbacks:
- n = 'callbacks'
- elif need in f90modhooks:
- n = 'f90modhooks'
- elif need in commonhooks:
- n = 'commonhooks'
- else:
- errmess('append_needs: unknown need %s\n' % (repr(need)))
- return
- if need in outneeds[n]:
- return
- if flag:
- tmp = {}
- if need in needs:
- for nn in needs[need]:
- t = append_needs(nn, 0)
- if isinstance(t, dict):
- for nnn in t.keys():
- if nnn in tmp:
- tmp[nnn] = tmp[nnn] + t[nnn]
- else:
- tmp[nnn] = t[nnn]
- for nn in tmp.keys():
- for nnn in tmp[nn]:
- if nnn not in outneeds[nn]:
- outneeds[nn] = [nnn] + outneeds[nn]
- outneeds[n].append(need)
- else:
- tmp = {}
- if need in needs:
- for nn in needs[need]:
- t = append_needs(nn, flag)
- if isinstance(t, dict):
- for nnn in t.keys():
- if nnn in tmp:
- tmp[nnn] = t[nnn] + tmp[nnn]
- else:
- tmp[nnn] = t[nnn]
- if n not in tmp:
- tmp[n] = []
- tmp[n].append(need)
- return tmp
- else:
- errmess('append_needs: expected list or string but got :%s\n' %
- (repr(need)))
- def get_needs():
- # This function modifies the contents of the global `outneeds` dict.
- res = {}
- for n in outneeds.keys():
- out = []
- saveout = copy.copy(outneeds[n])
- while len(outneeds[n]) > 0:
- if outneeds[n][0] not in needs:
- out.append(outneeds[n][0])
- del outneeds[n][0]
- else:
- flag = 0
- for k in outneeds[n][1:]:
- if k in needs[outneeds[n][0]]:
- flag = 1
- break
- if flag:
- outneeds[n] = outneeds[n][1:] + [outneeds[n][0]]
- else:
- out.append(outneeds[n][0])
- del outneeds[n][0]
- if saveout and (0 not in map(lambda x, y: x == y, saveout, outneeds[n])) \
- and outneeds[n] != []:
- print(n, saveout)
- errmess(
- 'get_needs: no progress in sorting needs, probably circular dependence, skipping.\n')
- out = out + saveout
- break
- saveout = copy.copy(outneeds[n])
- if out == []:
- out = [n]
- res[n] = out
- return res
|