Spaces:
Running
Running
| #!/usr/bin/env python3 | |
| """ | |
| Build call-back mechanism for f2py2e. | |
| Copyright 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/07/20 11:27:58 $ | |
| Pearu Peterson | |
| """ | |
| from . import __version__ | |
| from .auxfuncs import ( | |
| applyrules, debugcapi, dictappend, errmess, getargs, hasnote, isarray, | |
| iscomplex, iscomplexarray, iscomplexfunction, isfunction, isintent_c, | |
| isintent_hide, isintent_in, isintent_inout, isintent_nothide, | |
| isintent_out, isoptional, isrequired, isscalar, isstring, | |
| isstringfunction, issubroutine, l_and, l_not, l_or, outmess, replace, | |
| stripcomma, throw_error | |
| ) | |
| from . import cfuncs | |
| f2py_version = __version__.version | |
| ################## Rules for callback function ############## | |
| cb_routine_rules = { | |
| 'cbtypedefs': 'typedef #rctype#(*#name#_typedef)(#optargs_td##args_td##strarglens_td##noargs#);', | |
| 'body': """ | |
| #begintitle# | |
| typedef struct { | |
| PyObject *capi; | |
| PyTupleObject *args_capi; | |
| int nofargs; | |
| jmp_buf jmpbuf; | |
| } #name#_t; | |
| #if defined(F2PY_THREAD_LOCAL_DECL) && !defined(F2PY_USE_PYTHON_TLS) | |
| static F2PY_THREAD_LOCAL_DECL #name#_t *_active_#name# = NULL; | |
| static #name#_t *swap_active_#name#(#name#_t *ptr) { | |
| #name#_t *prev = _active_#name#; | |
| _active_#name# = ptr; | |
| return prev; | |
| } | |
| static #name#_t *get_active_#name#(void) { | |
| return _active_#name#; | |
| } | |
| #else | |
| static #name#_t *swap_active_#name#(#name#_t *ptr) { | |
| char *key = "__f2py_cb_#name#"; | |
| return (#name#_t *)F2PySwapThreadLocalCallbackPtr(key, ptr); | |
| } | |
| static #name#_t *get_active_#name#(void) { | |
| char *key = "__f2py_cb_#name#"; | |
| return (#name#_t *)F2PyGetThreadLocalCallbackPtr(key); | |
| } | |
| #endif | |
| /*typedef #rctype#(*#name#_typedef)(#optargs_td##args_td##strarglens_td##noargs#);*/ | |
| #static# #rctype# #callbackname# (#optargs##args##strarglens##noargs#) { | |
| #name#_t cb_local = { NULL, NULL, 0 }; | |
| #name#_t *cb = NULL; | |
| PyTupleObject *capi_arglist = NULL; | |
| PyObject *capi_return = NULL; | |
| PyObject *capi_tmp = NULL; | |
| PyObject *capi_arglist_list = NULL; | |
| int capi_j,capi_i = 0; | |
| int capi_longjmp_ok = 1; | |
| #decl# | |
| #ifdef F2PY_REPORT_ATEXIT | |
| f2py_cb_start_clock(); | |
| #endif | |
| cb = get_active_#name#(); | |
| if (cb == NULL) { | |
| capi_longjmp_ok = 0; | |
| cb = &cb_local; | |
| } | |
| capi_arglist = cb->args_capi; | |
| CFUNCSMESS(\"cb:Call-back function #name# (maxnofargs=#maxnofargs#(-#nofoptargs#))\\n\"); | |
| CFUNCSMESSPY(\"cb:#name#_capi=\",cb->capi); | |
| if (cb->capi==NULL) { | |
| capi_longjmp_ok = 0; | |
| cb->capi = PyObject_GetAttrString(#modulename#_module,\"#argname#\"); | |
| CFUNCSMESSPY(\"cb:#name#_capi=\",cb->capi); | |
| } | |
| if (cb->capi==NULL) { | |
| PyErr_SetString(#modulename#_error,\"cb: Callback #argname# not defined (as an argument or module #modulename# attribute).\\n\"); | |
| goto capi_fail; | |
| } | |
| if (F2PyCapsule_Check(cb->capi)) { | |
| #name#_typedef #name#_cptr; | |
| #name#_cptr = F2PyCapsule_AsVoidPtr(cb->capi); | |
| #returncptr#(*#name#_cptr)(#optargs_nm##args_nm##strarglens_nm#); | |
| #return# | |
| } | |
| if (capi_arglist==NULL) { | |
| capi_longjmp_ok = 0; | |
| capi_tmp = PyObject_GetAttrString(#modulename#_module,\"#argname#_extra_args\"); | |
| if (capi_tmp) { | |
| capi_arglist = (PyTupleObject *)PySequence_Tuple(capi_tmp); | |
| if (capi_arglist==NULL) { | |
| PyErr_SetString(#modulename#_error,\"Failed to convert #modulename#.#argname#_extra_args to tuple.\\n\"); | |
| goto capi_fail; | |
| } | |
| } else { | |
| PyErr_Clear(); | |
| capi_arglist = (PyTupleObject *)Py_BuildValue(\"()\"); | |
| } | |
| } | |
| if (capi_arglist == NULL) { | |
| PyErr_SetString(#modulename#_error,\"Callback #argname# argument list is not set.\\n\"); | |
| goto capi_fail; | |
| } | |
| #setdims# | |
| #ifdef PYPY_VERSION | |
| #define CAPI_ARGLIST_SETITEM(idx, value) PyList_SetItem((PyObject *)capi_arglist_list, idx, value) | |
| capi_arglist_list = PySequence_List(capi_arglist); | |
| if (capi_arglist_list == NULL) goto capi_fail; | |
| #else | |
| #define CAPI_ARGLIST_SETITEM(idx, value) PyTuple_SetItem((PyObject *)capi_arglist, idx, value) | |
| #endif | |
| #pyobjfrom# | |
| #undef CAPI_ARGLIST_SETITEM | |
| #ifdef PYPY_VERSION | |
| CFUNCSMESSPY(\"cb:capi_arglist=\",capi_arglist_list); | |
| #else | |
| CFUNCSMESSPY(\"cb:capi_arglist=\",capi_arglist); | |
| #endif | |
| CFUNCSMESS(\"cb:Call-back calling Python function #argname#.\\n\"); | |
| #ifdef F2PY_REPORT_ATEXIT | |
| f2py_cb_start_call_clock(); | |
| #endif | |
| #ifdef PYPY_VERSION | |
| capi_return = PyObject_CallObject(cb->capi,(PyObject *)capi_arglist_list); | |
| Py_DECREF(capi_arglist_list); | |
| capi_arglist_list = NULL; | |
| #else | |
| capi_return = PyObject_CallObject(cb->capi,(PyObject *)capi_arglist); | |
| #endif | |
| #ifdef F2PY_REPORT_ATEXIT | |
| f2py_cb_stop_call_clock(); | |
| #endif | |
| CFUNCSMESSPY(\"cb:capi_return=\",capi_return); | |
| if (capi_return == NULL) { | |
| fprintf(stderr,\"capi_return is NULL\\n\"); | |
| goto capi_fail; | |
| } | |
| if (capi_return == Py_None) { | |
| Py_DECREF(capi_return); | |
| capi_return = Py_BuildValue(\"()\"); | |
| } | |
| else if (!PyTuple_Check(capi_return)) { | |
| capi_return = Py_BuildValue(\"(N)\",capi_return); | |
| } | |
| capi_j = PyTuple_Size(capi_return); | |
| capi_i = 0; | |
| #frompyobj# | |
| CFUNCSMESS(\"cb:#name#:successful\\n\"); | |
| Py_DECREF(capi_return); | |
| #ifdef F2PY_REPORT_ATEXIT | |
| f2py_cb_stop_clock(); | |
| #endif | |
| goto capi_return_pt; | |
| capi_fail: | |
| fprintf(stderr,\"Call-back #name# failed.\\n\"); | |
| Py_XDECREF(capi_return); | |
| Py_XDECREF(capi_arglist_list); | |
| if (capi_longjmp_ok) { | |
| longjmp(cb->jmpbuf,-1); | |
| } | |
| capi_return_pt: | |
| ; | |
| #return# | |
| } | |
| #endtitle# | |
| """, | |
| 'need': ['setjmp.h', 'CFUNCSMESS', 'F2PY_THREAD_LOCAL_DECL'], | |
| 'maxnofargs': '#maxnofargs#', | |
| 'nofoptargs': '#nofoptargs#', | |
| 'docstr': """\ | |
| \tdef #argname#(#docsignature#): return #docreturn#\\n\\ | |
| #docstrsigns#""", | |
| 'latexdocstr': """ | |
| {{}\\verb@def #argname#(#latexdocsignature#): return #docreturn#@{}} | |
| #routnote# | |
| #latexdocstrsigns#""", | |
| 'docstrshort': 'def #argname#(#docsignature#): return #docreturn#' | |
| } | |
| cb_rout_rules = [ | |
| { # Init | |
| 'separatorsfor': {'decl': '\n', | |
| 'args': ',', 'optargs': '', 'pyobjfrom': '\n', 'freemem': '\n', | |
| 'args_td': ',', 'optargs_td': '', | |
| 'args_nm': ',', 'optargs_nm': '', | |
| 'frompyobj': '\n', 'setdims': '\n', | |
| 'docstrsigns': '\\n"\n"', | |
| 'latexdocstrsigns': '\n', | |
| 'latexdocstrreq': '\n', 'latexdocstropt': '\n', | |
| 'latexdocstrout': '\n', 'latexdocstrcbs': '\n', | |
| }, | |
| 'decl': '/*decl*/', 'pyobjfrom': '/*pyobjfrom*/', 'frompyobj': '/*frompyobj*/', | |
| 'args': [], 'optargs': '', 'return': '', 'strarglens': '', 'freemem': '/*freemem*/', | |
| 'args_td': [], 'optargs_td': '', 'strarglens_td': '', | |
| 'args_nm': [], 'optargs_nm': '', 'strarglens_nm': '', | |
| 'noargs': '', | |
| 'setdims': '/*setdims*/', | |
| 'docstrsigns': '', 'latexdocstrsigns': '', | |
| 'docstrreq': '\tRequired arguments:', | |
| 'docstropt': '\tOptional arguments:', | |
| 'docstrout': '\tReturn objects:', | |
| 'docstrcbs': '\tCall-back functions:', | |
| 'docreturn': '', 'docsign': '', 'docsignopt': '', | |
| 'latexdocstrreq': '\\noindent Required arguments:', | |
| 'latexdocstropt': '\\noindent Optional arguments:', | |
| 'latexdocstrout': '\\noindent Return objects:', | |
| 'latexdocstrcbs': '\\noindent Call-back functions:', | |
| 'routnote': {hasnote: '--- #note#', l_not(hasnote): ''}, | |
| }, { # Function | |
| 'decl': ' #ctype# return_value;', | |
| 'frompyobj': [{debugcapi: ' CFUNCSMESS("cb:Getting return_value->");'}, | |
| ' if (capi_j>capi_i)\n GETSCALARFROMPYTUPLE(capi_return,capi_i++,&return_value,#ctype#,"#ctype#_from_pyobj failed in converting return_value of call-back function #name# to C #ctype#\\n");', | |
| {debugcapi: | |
| ' fprintf(stderr,"#showvalueformat#.\\n",return_value);'} | |
| ], | |
| 'need': ['#ctype#_from_pyobj', {debugcapi: 'CFUNCSMESS'}, 'GETSCALARFROMPYTUPLE'], | |
| 'return': ' return return_value;', | |
| '_check': l_and(isfunction, l_not(isstringfunction), l_not(iscomplexfunction)) | |
| }, | |
| { # String function | |
| 'pyobjfrom': {debugcapi: ' fprintf(stderr,"debug-capi:cb:#name#:%d:\\n",return_value_len);'}, | |
| 'args': '#ctype# return_value,int return_value_len', | |
| 'args_nm': 'return_value,&return_value_len', | |
| 'args_td': '#ctype# ,int', | |
| 'frompyobj': [{debugcapi: ' CFUNCSMESS("cb:Getting return_value->\\"");'}, | |
| """ if (capi_j>capi_i) | |
| GETSTRFROMPYTUPLE(capi_return,capi_i++,return_value,return_value_len);""", | |
| {debugcapi: | |
| ' fprintf(stderr,"#showvalueformat#\\".\\n",return_value);'} | |
| ], | |
| 'need': ['#ctype#_from_pyobj', {debugcapi: 'CFUNCSMESS'}, | |
| 'string.h', 'GETSTRFROMPYTUPLE'], | |
| 'return': 'return;', | |
| '_check': isstringfunction | |
| }, | |
| { # Complex function | |
| 'optargs': """ | |
| #ifndef F2PY_CB_RETURNCOMPLEX | |
| #ctype# *return_value | |
| #endif | |
| """, | |
| 'optargs_nm': """ | |
| #ifndef F2PY_CB_RETURNCOMPLEX | |
| return_value | |
| #endif | |
| """, | |
| 'optargs_td': """ | |
| #ifndef F2PY_CB_RETURNCOMPLEX | |
| #ctype# * | |
| #endif | |
| """, | |
| 'decl': """ | |
| #ifdef F2PY_CB_RETURNCOMPLEX | |
| #ctype# return_value; | |
| #endif | |
| """, | |
| 'frompyobj': [{debugcapi: ' CFUNCSMESS("cb:Getting return_value->");'}, | |
| """\ | |
| if (capi_j>capi_i) | |
| #ifdef F2PY_CB_RETURNCOMPLEX | |
| GETSCALARFROMPYTUPLE(capi_return,capi_i++,&return_value,#ctype#,\"#ctype#_from_pyobj failed in converting return_value of call-back function #name# to C #ctype#\\n\"); | |
| #else | |
| GETSCALARFROMPYTUPLE(capi_return,capi_i++,return_value,#ctype#,\"#ctype#_from_pyobj failed in converting return_value of call-back function #name# to C #ctype#\\n\"); | |
| #endif | |
| """, | |
| {debugcapi: """ | |
| #ifdef F2PY_CB_RETURNCOMPLEX | |
| fprintf(stderr,\"#showvalueformat#.\\n\",(return_value).r,(return_value).i); | |
| #else | |
| fprintf(stderr,\"#showvalueformat#.\\n\",(*return_value).r,(*return_value).i); | |
| #endif | |
| """} | |
| ], | |
| 'return': """ | |
| #ifdef F2PY_CB_RETURNCOMPLEX | |
| return return_value; | |
| #else | |
| return; | |
| #endif | |
| """, | |
| 'need': ['#ctype#_from_pyobj', {debugcapi: 'CFUNCSMESS'}, | |
| 'string.h', 'GETSCALARFROMPYTUPLE', '#ctype#'], | |
| '_check': iscomplexfunction | |
| }, | |
| {'docstrout': '\t\t#pydocsignout#', | |
| 'latexdocstrout': ['\\item[]{{}\\verb@#pydocsignout#@{}}', | |
| {hasnote: '--- #note#'}], | |
| 'docreturn': '#rname#,', | |
| '_check': isfunction}, | |
| {'_check': issubroutine, 'return': 'return;'} | |
| ] | |
| cb_arg_rules = [ | |
| { # Doc | |
| 'docstropt': {l_and(isoptional, isintent_nothide): '\t\t#pydocsign#'}, | |
| 'docstrreq': {l_and(isrequired, isintent_nothide): '\t\t#pydocsign#'}, | |
| 'docstrout': {isintent_out: '\t\t#pydocsignout#'}, | |
| 'latexdocstropt': {l_and(isoptional, isintent_nothide): ['\\item[]{{}\\verb@#pydocsign#@{}}', | |
| {hasnote: '--- #note#'}]}, | |
| 'latexdocstrreq': {l_and(isrequired, isintent_nothide): ['\\item[]{{}\\verb@#pydocsign#@{}}', | |
| {hasnote: '--- #note#'}]}, | |
| 'latexdocstrout': {isintent_out: ['\\item[]{{}\\verb@#pydocsignout#@{}}', | |
| {l_and(hasnote, isintent_hide): '--- #note#', | |
| l_and(hasnote, isintent_nothide): '--- See above.'}]}, | |
| 'docsign': {l_and(isrequired, isintent_nothide): '#varname#,'}, | |
| 'docsignopt': {l_and(isoptional, isintent_nothide): '#varname#,'}, | |
| 'depend': '' | |
| }, | |
| { | |
| 'args': { | |
| l_and(isscalar, isintent_c): '#ctype# #varname_i#', | |
| l_and(isscalar, l_not(isintent_c)): '#ctype# *#varname_i#_cb_capi', | |
| isarray: '#ctype# *#varname_i#', | |
| isstring: '#ctype# #varname_i#' | |
| }, | |
| 'args_nm': { | |
| l_and(isscalar, isintent_c): '#varname_i#', | |
| l_and(isscalar, l_not(isintent_c)): '#varname_i#_cb_capi', | |
| isarray: '#varname_i#', | |
| isstring: '#varname_i#' | |
| }, | |
| 'args_td': { | |
| l_and(isscalar, isintent_c): '#ctype#', | |
| l_and(isscalar, l_not(isintent_c)): '#ctype# *', | |
| isarray: '#ctype# *', | |
| isstring: '#ctype#' | |
| }, | |
| 'need': {l_or(isscalar, isarray, isstring): '#ctype#'}, | |
| # untested with multiple args | |
| 'strarglens': {isstring: ',int #varname_i#_cb_len'}, | |
| 'strarglens_td': {isstring: ',int'}, # untested with multiple args | |
| # untested with multiple args | |
| 'strarglens_nm': {isstring: ',#varname_i#_cb_len'}, | |
| }, | |
| { # Scalars | |
| 'decl': {l_not(isintent_c): ' #ctype# #varname_i#=(*#varname_i#_cb_capi);'}, | |
| 'error': {l_and(isintent_c, isintent_out, | |
| throw_error('intent(c,out) is forbidden for callback scalar arguments')): | |
| ''}, | |
| 'frompyobj': [{debugcapi: ' CFUNCSMESS("cb:Getting #varname#->");'}, | |
| {isintent_out: | |
| ' if (capi_j>capi_i)\n GETSCALARFROMPYTUPLE(capi_return,capi_i++,#varname_i#_cb_capi,#ctype#,"#ctype#_from_pyobj failed in converting argument #varname# of call-back function #name# to C #ctype#\\n");'}, | |
| {l_and(debugcapi, l_and(l_not(iscomplex), isintent_c)): | |
| ' fprintf(stderr,"#showvalueformat#.\\n",#varname_i#);'}, | |
| {l_and(debugcapi, l_and(l_not(iscomplex), l_not( isintent_c))): | |
| ' fprintf(stderr,"#showvalueformat#.\\n",*#varname_i#_cb_capi);'}, | |
| {l_and(debugcapi, l_and(iscomplex, isintent_c)): | |
| ' fprintf(stderr,"#showvalueformat#.\\n",(#varname_i#).r,(#varname_i#).i);'}, | |
| {l_and(debugcapi, l_and(iscomplex, l_not( isintent_c))): | |
| ' fprintf(stderr,"#showvalueformat#.\\n",(*#varname_i#_cb_capi).r,(*#varname_i#_cb_capi).i);'}, | |
| ], | |
| 'need': [{isintent_out: ['#ctype#_from_pyobj', 'GETSCALARFROMPYTUPLE']}, | |
| {debugcapi: 'CFUNCSMESS'}], | |
| '_check': isscalar | |
| }, { | |
| 'pyobjfrom': [{isintent_in: """\ | |
| if (cb->nofargs>capi_i) | |
| if (CAPI_ARGLIST_SETITEM(capi_i++,pyobj_from_#ctype#1(#varname_i#))) | |
| goto capi_fail;"""}, | |
| {isintent_inout: """\ | |
| if (cb->nofargs>capi_i) | |
| if (CAPI_ARGLIST_SETITEM(capi_i++,pyarr_from_p_#ctype#1(#varname_i#_cb_capi))) | |
| goto capi_fail;"""}], | |
| 'need': [{isintent_in: 'pyobj_from_#ctype#1'}, | |
| {isintent_inout: 'pyarr_from_p_#ctype#1'}, | |
| {iscomplex: '#ctype#'}], | |
| '_check': l_and(isscalar, isintent_nothide), | |
| '_optional': '' | |
| }, { # String | |
| 'frompyobj': [{debugcapi: ' CFUNCSMESS("cb:Getting #varname#->\\"");'}, | |
| """ if (capi_j>capi_i) | |
| GETSTRFROMPYTUPLE(capi_return,capi_i++,#varname_i#,#varname_i#_cb_len);""", | |
| {debugcapi: | |
| ' fprintf(stderr,"#showvalueformat#\\":%d:.\\n",#varname_i#,#varname_i#_cb_len);'}, | |
| ], | |
| 'need': ['#ctype#', 'GETSTRFROMPYTUPLE', | |
| {debugcapi: 'CFUNCSMESS'}, 'string.h'], | |
| '_check': l_and(isstring, isintent_out) | |
| }, { | |
| 'pyobjfrom': [{debugcapi: ' fprintf(stderr,"debug-capi:cb:#varname#=\\"#showvalueformat#\\":%d:\\n",#varname_i#,#varname_i#_cb_len);'}, | |
| {isintent_in: """\ | |
| if (cb->nofargs>capi_i) | |
| if (CAPI_ARGLIST_SETITEM(capi_i++,pyobj_from_#ctype#1size(#varname_i#,#varname_i#_cb_len))) | |
| goto capi_fail;"""}, | |
| {isintent_inout: """\ | |
| if (cb->nofargs>capi_i) { | |
| int #varname_i#_cb_dims[] = {#varname_i#_cb_len}; | |
| if (CAPI_ARGLIST_SETITEM(capi_i++,pyarr_from_p_#ctype#1(#varname_i#,#varname_i#_cb_dims))) | |
| goto capi_fail; | |
| }"""}], | |
| 'need': [{isintent_in: 'pyobj_from_#ctype#1size'}, | |
| {isintent_inout: 'pyarr_from_p_#ctype#1'}], | |
| '_check': l_and(isstring, isintent_nothide), | |
| '_optional': '' | |
| }, | |
| # Array ... | |
| { | |
| 'decl': ' npy_intp #varname_i#_Dims[#rank#] = {#rank*[-1]#};', | |
| 'setdims': ' #cbsetdims#;', | |
| '_check': isarray, | |
| '_depend': '' | |
| }, | |
| { | |
| 'pyobjfrom': [{debugcapi: ' fprintf(stderr,"debug-capi:cb:#varname#\\n");'}, | |
| {isintent_c: """\ | |
| if (cb->nofargs>capi_i) { | |
| int itemsize_ = #atype# == NPY_STRING ? 1 : 0; | |
| /*XXX: Hmm, what will destroy this array??? */ | |
| PyArrayObject *tmp_arr = (PyArrayObject *)PyArray_New(&PyArray_Type,#rank#,#varname_i#_Dims,#atype#,NULL,(char*)#varname_i#,itemsize_,NPY_ARRAY_CARRAY,NULL); | |
| """, | |
| l_not(isintent_c): """\ | |
| if (cb->nofargs>capi_i) { | |
| int itemsize_ = #atype# == NPY_STRING ? 1 : 0; | |
| /*XXX: Hmm, what will destroy this array??? */ | |
| PyArrayObject *tmp_arr = (PyArrayObject *)PyArray_New(&PyArray_Type,#rank#,#varname_i#_Dims,#atype#,NULL,(char*)#varname_i#,itemsize_,NPY_ARRAY_FARRAY,NULL); | |
| """, | |
| }, | |
| """ | |
| if (tmp_arr==NULL) | |
| goto capi_fail; | |
| if (CAPI_ARGLIST_SETITEM(capi_i++,(PyObject *)tmp_arr)) | |
| goto capi_fail; | |
| }"""], | |
| '_check': l_and(isarray, isintent_nothide, l_or(isintent_in, isintent_inout)), | |
| '_optional': '', | |
| }, { | |
| 'frompyobj': [{debugcapi: ' CFUNCSMESS("cb:Getting #varname#->");'}, | |
| """ if (capi_j>capi_i) { | |
| PyArrayObject *rv_cb_arr = NULL; | |
| if ((capi_tmp = PyTuple_GetItem(capi_return,capi_i++))==NULL) goto capi_fail; | |
| rv_cb_arr = array_from_pyobj(#atype#,#varname_i#_Dims,#rank#,F2PY_INTENT_IN""", | |
| {isintent_c: '|F2PY_INTENT_C'}, | |
| """,capi_tmp); | |
| if (rv_cb_arr == NULL) { | |
| fprintf(stderr,\"rv_cb_arr is NULL\\n\"); | |
| goto capi_fail; | |
| } | |
| MEMCOPY(#varname_i#,PyArray_DATA(rv_cb_arr),PyArray_NBYTES(rv_cb_arr)); | |
| if (capi_tmp != (PyObject *)rv_cb_arr) { | |
| Py_DECREF(rv_cb_arr); | |
| } | |
| }""", | |
| {debugcapi: ' fprintf(stderr,"<-.\\n");'}, | |
| ], | |
| 'need': ['MEMCOPY', {iscomplexarray: '#ctype#'}], | |
| '_check': l_and(isarray, isintent_out) | |
| }, { | |
| 'docreturn': '#varname#,', | |
| '_check': isintent_out | |
| } | |
| ] | |
| ################## Build call-back module ############# | |
| cb_map = {} | |
| def buildcallbacks(m): | |
| cb_map[m['name']] = [] | |
| for bi in m['body']: | |
| if bi['block'] == 'interface': | |
| for b in bi['body']: | |
| if b: | |
| buildcallback(b, m['name']) | |
| else: | |
| errmess('warning: empty body for %s\n' % (m['name'])) | |
| def buildcallback(rout, um): | |
| from . import capi_maps | |
| outmess('\tConstructing call-back function "cb_%s_in_%s"\n' % | |
| (rout['name'], um)) | |
| args, depargs = getargs(rout) | |
| capi_maps.depargs = depargs | |
| var = rout['vars'] | |
| vrd = capi_maps.cb_routsign2map(rout, um) | |
| rd = dictappend({}, vrd) | |
| cb_map[um].append([rout['name'], rd['name']]) | |
| for r in cb_rout_rules: | |
| if ('_check' in r and r['_check'](rout)) or ('_check' not in r): | |
| ar = applyrules(r, vrd, rout) | |
| rd = dictappend(rd, ar) | |
| savevrd = {} | |
| for i, a in enumerate(args): | |
| vrd = capi_maps.cb_sign2map(a, var[a], index=i) | |
| savevrd[a] = vrd | |
| for r in cb_arg_rules: | |
| if '_depend' in r: | |
| continue | |
| if '_optional' in r and isoptional(var[a]): | |
| continue | |
| if ('_check' in r and r['_check'](var[a])) or ('_check' not in r): | |
| ar = applyrules(r, vrd, var[a]) | |
| rd = dictappend(rd, ar) | |
| if '_break' in r: | |
| break | |
| for a in args: | |
| vrd = savevrd[a] | |
| for r in cb_arg_rules: | |
| if '_depend' in r: | |
| continue | |
| if ('_optional' not in r) or ('_optional' in r and isrequired(var[a])): | |
| continue | |
| if ('_check' in r and r['_check'](var[a])) or ('_check' not in r): | |
| ar = applyrules(r, vrd, var[a]) | |
| rd = dictappend(rd, ar) | |
| if '_break' in r: | |
| break | |
| for a in depargs: | |
| vrd = savevrd[a] | |
| for r in cb_arg_rules: | |
| if '_depend' not in r: | |
| continue | |
| if '_optional' in r: | |
| continue | |
| if ('_check' in r and r['_check'](var[a])) or ('_check' not in r): | |
| ar = applyrules(r, vrd, var[a]) | |
| rd = dictappend(rd, ar) | |
| if '_break' in r: | |
| break | |
| if 'args' in rd and 'optargs' in rd: | |
| if isinstance(rd['optargs'], list): | |
| rd['optargs'] = rd['optargs'] + [""" | |
| #ifndef F2PY_CB_RETURNCOMPLEX | |
| , | |
| #endif | |
| """] | |
| rd['optargs_nm'] = rd['optargs_nm'] + [""" | |
| #ifndef F2PY_CB_RETURNCOMPLEX | |
| , | |
| #endif | |
| """] | |
| rd['optargs_td'] = rd['optargs_td'] + [""" | |
| #ifndef F2PY_CB_RETURNCOMPLEX | |
| , | |
| #endif | |
| """] | |
| if isinstance(rd['docreturn'], list): | |
| rd['docreturn'] = stripcomma( | |
| replace('#docreturn#', {'docreturn': rd['docreturn']})) | |
| optargs = stripcomma(replace('#docsignopt#', | |
| {'docsignopt': rd['docsignopt']} | |
| )) | |
| if optargs == '': | |
| rd['docsignature'] = stripcomma( | |
| replace('#docsign#', {'docsign': rd['docsign']})) | |
| else: | |
| rd['docsignature'] = replace('#docsign#[#docsignopt#]', | |
| {'docsign': rd['docsign'], | |
| 'docsignopt': optargs, | |
| }) | |
| rd['latexdocsignature'] = rd['docsignature'].replace('_', '\\_') | |
| rd['latexdocsignature'] = rd['latexdocsignature'].replace(',', ', ') | |
| rd['docstrsigns'] = [] | |
| rd['latexdocstrsigns'] = [] | |
| for k in ['docstrreq', 'docstropt', 'docstrout', 'docstrcbs']: | |
| if k in rd and isinstance(rd[k], list): | |
| rd['docstrsigns'] = rd['docstrsigns'] + rd[k] | |
| k = 'latex' + k | |
| if k in rd and isinstance(rd[k], list): | |
| rd['latexdocstrsigns'] = rd['latexdocstrsigns'] + rd[k][0:1] +\ | |
| ['\\begin{description}'] + rd[k][1:] +\ | |
| ['\\end{description}'] | |
| if 'args' not in rd: | |
| rd['args'] = '' | |
| rd['args_td'] = '' | |
| rd['args_nm'] = '' | |
| if not (rd.get('args') or rd.get('optargs') or rd.get('strarglens')): | |
| rd['noargs'] = 'void' | |
| ar = applyrules(cb_routine_rules, rd) | |
| cfuncs.callbacks[rd['name']] = ar['body'] | |
| if isinstance(ar['need'], str): | |
| ar['need'] = [ar['need']] | |
| if 'need' in rd: | |
| for t in cfuncs.typedefs.keys(): | |
| if t in rd['need']: | |
| ar['need'].append(t) | |
| cfuncs.typedefs_generated[rd['name'] + '_typedef'] = ar['cbtypedefs'] | |
| ar['need'].append(rd['name'] + '_typedef') | |
| cfuncs.needs[rd['name']] = ar['need'] | |
| capi_maps.lcb2_map[rd['name']] = {'maxnofargs': ar['maxnofargs'], | |
| 'nofoptargs': ar['nofoptargs'], | |
| 'docstr': ar['docstr'], | |
| 'latexdocstr': ar['latexdocstr'], | |
| 'argname': rd['argname'] | |
| } | |
| outmess('\t %s\n' % (ar['docstrshort'])) | |
| return | |
| ################## Build call-back function ############# | |