o
    6a^                     @   s,  d Z ddlmZ ddlmZmZmZmZmZm	Z	m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z! ddlm"Z" ej#Z$ddg ddd	d
dddZ%i di ddddddddddddddddddddddddddd dd!dd"dd#ddd$dd%dd&dg ddd'dd(ddd)dg ddd*ddg ddd+dd,ddd-ddd.d/d0d1dddd2d3d4d5e	d6ee	did7d8ed9id:ed;igd<ed=id>gd?eeeeeed@edAidBdCdDedEidFedGigd<ed=idHdIgdJedKdLdMdNdOed9idPedQigdRd<ed=idHd>dSgedTdUdVe	d6igdWedXedJdYgZ&eeedZieeedZiedUieeed[e	d6igieeed[e	d6igiedVee	ed6ee	ed\igieeed]ieeed]idd^	eeed_eeeed`e
daed_ieeedbeeeedce
dbedbieeedSeeeedde
ddedSieee
edSiedeiedfiedgidheediieeee!djdiedkiedlieeeeeedmieeeeeeednieeeeedoieeeeeedpiged<d>gied=igedqedriedsigedtieduiedSigeeeddvedwidxedyigdSdIed=idHgeeedzed{ied|ied}iged~ieduigeeeddvdde
ddediedeedidgee
eeeeddedkidedidedigdedSigee
edzd]edg
Z'i Z(dd Z)dd Z*dS )aK  

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

   )__version__)
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)cfuncszPtypedef #rctype#(*#name#_typedef)(#optargs_td##args_td##strarglens_td##noargs#);a=  
#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#
)zsetjmp.h
CFUNCSMESSF2PY_THREAD_LOCAL_DECLz#maxnofargs#z#nofoptargs#zC	def #argname#(#docsignature#): return #docreturn#\n\
#docstrsigns#zc
{{}\verb@def #argname#(#latexdocsignature#): return #docreturn#@{}}
#routnote#

#latexdocstrsigns#z1def #argname#(#docsignature#): return #docreturn#)
cbtypedefsbodyneed
maxnofargs
nofoptargsdocstrlatexdocstrdocstrshortseparatorsfordecl
args,optargs 	pyobjfromfreememargs_td
optargs_tdargs_nm
optargs_nm	frompyobjsetdimsdocstrsignsz\n"
"latexdocstrsignslatexdocstrreqlatexdocstroptlatexdocstroutlatexdocstrcbsz/*decl*/z/*pyobjfrom*/z/*frompyobj*/return
strarglensz/*freemem*/strarglens_tdstrarglens_nmnoargsz/*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#)r;   r<   	docstrreq	docstropt	docstrout	docstrcbs	docreturndocsign
docsignoptr=   r>   r?   r@   routnotez    #ctype# return_value;z,    CFUNCSMESS("cb:Getting return_value->");z    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");z8    fprintf(stderr,"#showvalueformat#.\n",return_value);z#ctype#_from_pyobjr"   GETSCALARFROMPYTUPLEz    return return_value;)r-   r9   r&   rA   _checkzB    fprintf(stderr,"debug-capi:cb:#name#:%d:\n",return_value_len);z)#ctype# return_value,int return_value_lenzreturn_value,&return_value_lenz#ctype# ,intz.    CFUNCSMESS("cb:Getting return_value->\"");ze    if (capi_j>capi_i)
        GETSTRFROMPYTUPLE(capi_return,capi_i++,return_value,return_value_len);z:    fprintf(stderr,"#showvalueformat#\".\n",return_value);zstring.hGETSTRFROMPYTUPLEzreturn;)r3   r/   r7   r5   r9   r&   rA   rO   z<
#ifndef F2PY_CB_RETURNCOMPLEX
#ctype# *return_value
#endif
z3
#ifndef F2PY_CB_RETURNCOMPLEX
return_value
#endif
z0
#ifndef F2PY_CB_RETURNCOMPLEX
#ctype# *
#endif
z?
#ifdef F2PY_CB_RETURNCOMPLEX
    #ctype# return_value;
#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
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#)r1   r8   r6   r-   r9   rA   r&   rO   z		#pydocsignout#z"\item[]{{}\verb@#pydocsignout#@{}}z#rname#,)rH   r?   rJ   rO   )rO   rA   z		#pydocsign#z\item[]{{}\verb@#pydocsign#@{}}z--- See above.z
#varname#,)	rG   rF   rH   r>   r=   r?   rK   rL   dependz#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/   r7   r5   r&   rB   rC   rD   z/    #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-   errorr9   r&   rO   z    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;zpyobj_from_#ctype#1zpyarr_from_p_#ctype#1)r3   r&   rO   	_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);)r9   r&   rO   zi    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;
    }zpyobj_from_#ctype#1sizez6    npy_intp #varname_i#_Dims[#rank#] = {#rank*[-1]#};z    #cbsetdims#;)r-   r:   rO   _dependz0    fprintf(stderr,"debug-capi:cb:#varname#\n");a2      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);
a2      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);
z
        if (tmp_arr==NULL)
            goto capi_fail;
        if (CAPI_ARGLIST_SETITEM(capi_i++,(PyObject *)tmp_arr))
            goto capi_fail;
})r3   rO   rS   z    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");MEMCOPY)rJ   rO   c                 C   s\   g t | d < | d D ]!}|d dkr+|d D ]}|r"t|| d  qtd| d   qq
d S )Nnamer%   block	interfacezwarning: empty body for %s
)cb_mapbuildcallbackr   )mbib r^   5/usr/lib/python3/dist-packages/numpy/f2py/cb_rules.pybuildcallbacks  s   r`   c                 C   s4  ddl m} td| d |f  t| \}}||_| d }|| |}ti |}t| | d |d g t	D ]}d|v rC|d | sGd|vrRt
||| }	t||	}q7i }
t|D ]J\}}|j||| |d}||
|< tD ]5}d|v rtqmd	|v rt|| rqmd|v r|d || sd|vrt
|||| }	t||	}d
|v r nqmqY|D ]B}|
| }tD ]9}d|v rqd	|vsd	|v rt|| rqd|v r|d || sd|vrt
|||| }	t||	}d
|v r nqq|D ]<}|
| }tD ]3}d|vrqd	|v rqd|v r|d || sd|vr&t
|||| }	t||	}d
|v r& nqqd|v rUd|v rUt|d trU|d dg |d< |d dg |d< |d dg |d< t|d trjttdd|d i|d< ttdd|d i}|dkrttdd|d i|d< ntd|d |d|d< |d dd|d< |d dd|d< g |d< g |d < d!D ]G}||v rt|| tr|d ||  |d< d"| }||v rt|| tr|d  || d#d  d$g || dd   d%g |d < qd|vrd|d< d|d&< d|d'< |ds!|ds!|d(s!d)|d*< t
t|}	|	d+ tj|d < t|	d, tr>|	d, g|	d,< d,|v rZtj D ]}||d, v rX|	d, | qH|	d- tj|d d. < |	d, |d d.  |	d, tj|d < |	d/ |	d0 |	d1 |	d2 |d3 d4|j|d < td5|	d6   d S )7Nr   )	capi_mapsz/	Constructing call-back function "cb_%s_in_%s"
rV   varsrO   )indexrT   rS   _breakr/   r1   z(
#ifndef F2PY_CB_RETURNCOMPLEX
,
#endif
r8   r6   rJ   z#docreturn#z#docsignopt#rL   r2   z	#docsign#rK   docsignaturez#docsign#[#docsignopt#])rK   rL   _z\_latexdocsignaturer0   z, r;   r<   )rF   rG   rH   rI   latex    z\begin{description}z\end{description}r5   r7   rB   voidrE   r%   r&   r$   _typedefr'   r(   r)   r*   argname)r'   r(   r)   r*   rl   z	  %s
r+   )r2   ra   r   r   depargscb_routsign2mapr   rY   appendcb_rout_rulesr   	enumeratecb_sign2mapcb_arg_rulesr   r   
isinstancelistr   r   getcb_routine_rulesr!   	callbacksstrtypedefskeystypedefs_generatedneedslcb2_map)routumra   r/   rm   varvrdrdrarsavevrdiar1   ktr^   r^   r_   rZ     s   


 
 
&





$

rZ   N)+__doc__r2   r   auxfuncsr   r   r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   versionf2py_versionrw   rp   rs   rY   r`   rZ   r^   r^   r^   r_   <module>   s&     ,
2t










  "