a ;jga"@sdZddlmZddlmZmZmZmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!ddlm"Z"ej#Z$ddgddd d d d d Z%dddddddddddddddddddddgddddgddgddddddddddddddddde d ee did!d"ed#id$ed%igd&ed'id(gd)eeeeee d*ed+id,d-d.ed/id0ed1igd&ed'id2d3gd4ed5d6d7d8d9ed#id:ed;igdd?d@e d igdAedBed4dCgZ&eeedDieeedDied?ieeedEe d igieeedEe d igied@ee ed ee edFigieeedGieeedGiddH eeedIeeeedJe dKedIieeedLeeeedMe dLedLieeed=eeeedNe dNed=ieee ed=iedOiedPiedQidReedSieeee!dTdiedUiedVieeeee edWieeeee eedXieeee edYieeee eedZiged&d(gied'iged[ed\ied]iged^ied_ie d=igeeedd`edaidbedcigd=d3ed'id2geeeddedeiedfiedgigedhied_igeeedd`didje ddkedliedmeednidogee eeeeddpedUidqedridsedtigdue d=igee edddGedvg Z'iZ(dwdxZ)dydzZ*d{S)|a3 Build call-back mechanism for f2py2e. Copyright 1999 -- 2011 Pearu Peterson all rights reserved. Copyright 2011 -- present NumPy Developers. 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. ) __version__) applyrules debugcapi dictappenderrmessgetargshasnoteisarray iscomplexiscomplexarrayiscomplexfunction isfunction isintent_c isintent_hide isintent_inisintent_inoutisintent_nothide isintent_out isoptional isrequiredisscalarisstringisstringfunction issubroutinel_andl_notl_oroutmessreplace stripcomma throw_error)cfuncszPtypedef #rctype#(*#name#_typedef)(#optargs_td##args_td##strarglens_td##noargs#);aj #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); Py_DECREF(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((PyObject *)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# )zsetjmp.h CFUNCSMESSZF2PY_THREAD_LOCAL_DECLz #maxnofargs#z #nofoptargs#zF def #argname#(#docsignature#): return #docreturn#\n\ #docstrsigns#zc {{}\verb@def #argname#(#latexdocsignature#): return #docreturn#@{}} #routnote# #latexdocstrsigns#z1def #argname#(#docsignature#): return #docreturn#) cbtypedefsbodyneed maxnofargs nofoptargsdocstr latexdocstr docstrshort ,z\n" ")declargsoptargs pyobjfromfreememargs_td optargs_tdargs_nm optargs_nm frompyobjsetdims docstrsignslatexdocstrsignslatexdocstrreqlatexdocstroptlatexdocstroutlatexdocstrcbsz/*decl*/z /*pyobjfrom*/z /*frompyobj*/z /*freemem*/z /*setdims*/z Required arguments:z Optional arguments:z Return objects:z Call-back functions:z\noindent Required arguments:z\noindent Optional arguments:z\noindent Return objects:z\noindent Call-back functions:z --- #note#)Z separatorsforr.r1r7r/r0return strarglensr2r3r4 strarglens_tdr5r6 strarglens_nmnoargsr8r9r: docstrreq docstropt docstrout docstrcbs docreturndocsign docsignoptr;r<r=r>Zroutnotez #ctype# return_value = 0;z, CFUNCSMESS("cb:Getting return_value->");a if (capi_j>capi_i) { 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 { fprintf(stderr,"Warning: call-back function #name# did not provide" " return value (index=%d, type=#ctype#)\n",capi_i); }z8 fprintf(stderr,"#showvalueformat#.\n",return_value);z#ctype#_from_pyobjr"ZGETSCALARFROMPYTUPLEz return return_value;)r.r7r%r?_checkzB fprintf(stderr,"debug-capi:cb:#name#:%d:\n",return_value_len);z)#ctype# return_value,int return_value_lenzreturn_value,&return_value_lenz #ctype# ,intz. CFUNCSMESS("cb:Getting return_value->\"");a if (capi_j>capi_i) { GETSTRFROMPYTUPLE(capi_return,capi_i++,return_value,return_value_len); } else { fprintf(stderr,"Warning: call-back function #name# did not provide" " return value (index=%d, type=#ctype#)\n",capi_i); }z: fprintf(stderr,"#showvalueformat#\".\n",return_value);zstring.hZGETSTRFROMPYTUPLEzreturn;)r1r/r5r3r7r%r?rKz< #ifndef F2PY_CB_RETURNCOMPLEX #ctype# *return_value #endif z3 #ifndef F2PY_CB_RETURNCOMPLEX return_value #endif z0 #ifndef F2PY_CB_RETURNCOMPLEX #ctype# * #endif zH #ifdef F2PY_CB_RETURNCOMPLEX #ctype# return_value = {0, 0}; #endif a 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 } else { fprintf(stderr, "Warning: call-back function #name# did not provide" " return value (index=%d, type=#ctype#)\n",capi_i); }z#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 zP #ifdef F2PY_CB_RETURNCOMPLEX return return_value; #else return; #endif z#ctype#)r0r6r4r.r7r?r%rKz #pydocsignout#z"\item[]{{}\verb@#pydocsignout#@{}}z#rname#,)rFr=rHrK)rKr?z #pydocsign#z\item[]{{}\verb@#pydocsign#@{}}z--- See above.z #varname#,) rErDrFr<r;r=rIrJZdependz#ctype# #varname_i#z#ctype# *#varname_i#_cb_capiz#ctype# *#varname_i#z #varname_i#z#varname_i#_cb_capiz #ctype# *z,int #varname_i#_cb_lenz,intz,#varname_i#_cb_len)r/r5r3r%r@rArBz/ #ctype# #varname_i#=(*#varname_i#_cb_capi);z8intent(c,out) is forbidden for callback scalar argumentsz) CFUNCSMESS("cb:Getting #varname#->");z if (capi_j>capi_i) 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");z7 fprintf(stderr,"#showvalueformat#.\n",#varname_i#);z@ fprintf(stderr,"#showvalueformat#.\n",*#varname_i#_cb_capi);zK fprintf(stderr,"#showvalueformat#.\n",(#varname_i#).r,(#varname_i#).i);z] fprintf(stderr,"#showvalueformat#.\n",(*#varname_i#_cb_capi).r,(*#varname_i#_cb_capi).i);)r.errorr7r%rKz if (cb->nofargs>capi_i) if (CAPI_ARGLIST_SETITEM(capi_i++,pyobj_from_#ctype#1(#varname_i#))) goto capi_fail;z if (cb->nofargs>capi_i) if (CAPI_ARGLIST_SETITEM(capi_i++,pyarr_from_p_#ctype#1(#varname_i#_cb_capi))) goto capi_fail;zpyobj_from_#ctype#1zpyarr_from_p_#ctype#1)r1r%rK _optionalz+ CFUNCSMESS("cb:Getting #varname#->\"");zf if (capi_j>capi_i) GETSTRFROMPYTUPLE(capi_return,capi_i++,#varname_i#,#varname_i#_cb_len);zP fprintf(stderr,"#showvalueformat#\":%d:.\n",#varname_i#,#varname_i#_cb_len);)r7r%rKze fprintf(stderr,"debug-capi:cb:#varname#=#showvalueformat#:%d:\n",#varname_i#,#varname_i#_cb_len);z if (cb->nofargs>capi_i) if (CAPI_ARGLIST_SETITEM(capi_i++,pyobj_from_#ctype#1size(#varname_i#,#varname_i#_cb_len))) goto capi_fail;z 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; }zpyobj_from_#ctype#1sizez6 npy_intp #varname_i#_Dims[#rank#] = {#rank*[-1]#};z #cbsetdims#;)r.r8rK_dependz0 fprintf(stderr,"debug-capi:cb:#varname#\n");a if (cb->nofargs>capi_i) { /* tmp_arr will be inserted to capi_arglist_list that will be destroyed when leaving callback function wrapper together with tmp_arr. */ PyArrayObject *tmp_arr = (PyArrayObject *)PyArray_New(&PyArray_Type, #rank#,#varname_i#_Dims,#atype#,NULL,(char*)#varname_i#,#elsize#, NPY_ARRAY_CARRAY,NULL); a if (cb->nofargs>capi_i) { /* tmp_arr will be inserted to capi_arglist_list that will be destroyed when leaving callback function wrapper together with tmp_arr. */ PyArrayObject *tmp_arr = (PyArrayObject *)PyArray_New(&PyArray_Type, #rank#,#varname_i#_Dims,#atype#,NULL,(char*)#varname_i#,#elsize#, NPY_ARRAY_FARRAY,NULL); z if (tmp_arr==NULL) goto capi_fail; if (CAPI_ARGLIST_SETITEM(capi_i++,(PyObject *)tmp_arr)) goto capi_fail; })r1rKrMz 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_INz|F2PY_INTENT_Ca8,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); } }z fprintf(stderr,"<-.\n");ZMEMCOPY)rHrKcCs\gt|d<|dD]B}|ddkr|dD](}|rDt||dq,td|dq,qdS)Nnamer$blockZ interfacezwarning: empty body for %s )cb_map buildcallbackr)mZbibrU=/usr/local/lib/python3.9/site-packages/numpy/f2py/cb_rules.pybuildcallbackss    rWcCs`ddlm}td|d|ft|\}}||_|d}|||}ti|}t||d|dgt D]6}d|vr|d|sd|vrnt |||} t|| }qni} t |D]\} } |j | || | d}|| | <t D]n}d|vrqd |vrt|| rqd|vr|d|| s"d|vrt |||| } t|| }d |vrqqq|D]} | | }t D]}d|vrrq`d |vs`d |vrt|| rq`d|vr|d|| sd|vr`t |||| } t|| }d |vr`qPq`qP|D]} | | }t D]r}d|vrqd |vr$qd|vr@|d|| sJd|vrt |||| } t|| }d |vrqqqd |vrd |vrt|d tr|d d g|d <|dd g|d<|dd g|d<t|dtrttdd|di|d<ttdd|di} | dkrs  .     :                 +