Path: blob/master/ invest-robot-contest_TinkoffBotTwitch-main/venv/lib/python3.8/site-packages/numpy/f2py/rules.py
7757 views
#!/usr/bin/env python31"""23Rules for building C/API module with f2py2e.45Here is a skeleton of a new wrapper function (13Dec2001):67wrapper_function(args)8declarations9get_python_arguments, say, `a' and `b'1011get_a_from_python12if (successful) {1314get_b_from_python15if (successful) {1617callfortran18if (successful) {1920put_a_to_python21if (successful) {2223put_b_to_python24if (successful) {2526buildvalue = ...2728}2930}3132}3334}35cleanup_b3637}38cleanup_a3940return buildvalue4142Copyright 1999,2000 Pearu Peterson all rights reserved,43Pearu Peterson <[email protected]>44Permission to use, modify, and distribute this software is given under the45terms of the NumPy License.4647NO WARRANTY IS EXPRESSED OR IMPLIED. USE AT YOUR OWN RISK.48$Date: 2005/08/30 08:58:42 $49Pearu Peterson5051"""52import os53import time54import copy5556# __version__.version is now the same as the NumPy version57from . import __version__58f2py_version = __version__.version59numpy_version = __version__.version6061from .auxfuncs import (62applyrules, debugcapi, dictappend, errmess, gentitle, getargs2,63hascallstatement, hasexternals, hasinitvalue, hasnote, hasresultnote,64isarray, isarrayofstrings, iscomplex, iscomplexarray,65iscomplexfunction, iscomplexfunction_warn, isdummyroutine, isexternal,66isfunction, isfunction_wrap, isint1array, isintent_aux, isintent_c,67isintent_callback, isintent_copy, isintent_hide, isintent_inout,68isintent_nothide, isintent_out, isintent_overwrite, islogical,69islong_complex, islong_double, islong_doublefunction, islong_long,70islong_longfunction, ismoduleroutine, isoptional, isrequired, isscalar,71issigned_long_longarray, isstring, isstringarray, isstringfunction,72issubroutine, issubroutine_wrap, isthreadsafe, isunsigned,73isunsigned_char, isunsigned_chararray, isunsigned_long_long,74isunsigned_long_longarray, isunsigned_short, isunsigned_shortarray,75l_and, l_not, l_or, outmess, replace, stripcomma, requiresf90wrapper76)7778from . import capi_maps79from . import cfuncs80from . import common_rules81from . import use_rules82from . import f90mod_rules83from . import func2subr8485options = {}86sepdict = {}87#for k in ['need_cfuncs']: sepdict[k]=','88for k in ['decl',89'frompyobj',90'cleanupfrompyobj',91'topyarr', 'method',92'pyobjfrom', 'closepyobjfrom',93'freemem',94'userincludes',95'includes0', 'includes', 'typedefs', 'typedefs_generated',96'cppmacros', 'cfuncs', 'callbacks',97'latexdoc',98'restdoc',99'routine_defs', 'externroutines',100'initf2pywraphooks',101'commonhooks', 'initcommonhooks',102'f90modhooks', 'initf90modhooks']:103sepdict[k] = '\n'104105#################### Rules for C/API module #################106107generationtime = int(os.environ.get('SOURCE_DATE_EPOCH', time.time()))108module_rules = {109'modulebody': """\110/* File: #modulename#module.c111* This file is auto-generated with f2py (version:#f2py_version#).112* f2py is a Fortran to Python Interface Generator (FPIG), Second Edition,113* written by Pearu Peterson <[email protected]>.114* Generation date: """ + time.asctime(time.gmtime(generationtime)) + """115* Do not edit this file directly unless you know what you are doing!!!116*/117118#ifdef __cplusplus119extern \"C\" {120#endif121122#ifndef PY_SSIZE_T_CLEAN123#define PY_SSIZE_T_CLEAN124#endif /* PY_SSIZE_T_CLEAN */125126""" + gentitle("See f2py2e/cfuncs.py: includes") + """127#includes#128#includes0#129130""" + gentitle("See f2py2e/rules.py: mod_rules['modulebody']") + """131static PyObject *#modulename#_error;132static PyObject *#modulename#_module;133134""" + gentitle("See f2py2e/cfuncs.py: typedefs") + """135#typedefs#136137""" + gentitle("See f2py2e/cfuncs.py: typedefs_generated") + """138#typedefs_generated#139140""" + gentitle("See f2py2e/cfuncs.py: cppmacros") + """141#cppmacros#142143""" + gentitle("See f2py2e/cfuncs.py: cfuncs") + """144#cfuncs#145146""" + gentitle("See f2py2e/cfuncs.py: userincludes") + """147#userincludes#148149""" + gentitle("See f2py2e/capi_rules.py: usercode") + """150#usercode#151152/* See f2py2e/rules.py */153#externroutines#154155""" + gentitle("See f2py2e/capi_rules.py: usercode1") + """156#usercode1#157158""" + gentitle("See f2py2e/cb_rules.py: buildcallback") + """159#callbacks#160161""" + gentitle("See f2py2e/rules.py: buildapi") + """162#body#163164""" + gentitle("See f2py2e/f90mod_rules.py: buildhooks") + """165#f90modhooks#166167""" + gentitle("See f2py2e/rules.py: module_rules['modulebody']") + """168169""" + gentitle("See f2py2e/common_rules.py: buildhooks") + """170#commonhooks#171172""" + gentitle("See f2py2e/rules.py") + """173174static FortranDataDef f2py_routine_defs[] = {175#routine_defs#176{NULL}177};178179static PyMethodDef f2py_module_methods[] = {180#pymethoddef#181{NULL,NULL}182};183184static struct PyModuleDef moduledef = {185PyModuleDef_HEAD_INIT,186"#modulename#",187NULL,188-1,189f2py_module_methods,190NULL,191NULL,192NULL,193NULL194};195196PyMODINIT_FUNC PyInit_#modulename#(void) {197int i;198PyObject *m,*d, *s, *tmp;199m = #modulename#_module = PyModule_Create(&moduledef);200Py_SET_TYPE(&PyFortran_Type, &PyType_Type);201import_array();202if (PyErr_Occurred())203{PyErr_SetString(PyExc_ImportError, \"can't initialize module #modulename# (failed to import numpy)\"); return m;}204d = PyModule_GetDict(m);205s = PyUnicode_FromString(\"#f2py_version#\");206PyDict_SetItemString(d, \"__version__\", s);207Py_DECREF(s);208s = PyUnicode_FromString(209\"This module '#modulename#' is auto-generated with f2py (version:#f2py_version#).\\nFunctions:\\n\"\n#docs#\".\");210PyDict_SetItemString(d, \"__doc__\", s);211Py_DECREF(s);212s = PyUnicode_FromString(\"""" + numpy_version + """\");213PyDict_SetItemString(d, \"__f2py_numpy_version__\", s);214Py_DECREF(s);215#modulename#_error = PyErr_NewException (\"#modulename#.error\", NULL, NULL);216/*217* Store the error object inside the dict, so that it could get deallocated.218* (in practice, this is a module, so it likely will not and cannot.)219*/220PyDict_SetItemString(d, \"_#modulename#_error\", #modulename#_error);221Py_DECREF(#modulename#_error);222for(i=0;f2py_routine_defs[i].name!=NULL;i++) {223tmp = PyFortranObject_NewAsAttr(&f2py_routine_defs[i]);224PyDict_SetItemString(d, f2py_routine_defs[i].name, tmp);225Py_DECREF(tmp);226}227#initf2pywraphooks#228#initf90modhooks#229#initcommonhooks#230#interface_usercode#231232#ifdef F2PY_REPORT_ATEXIT233if (! PyErr_Occurred())234on_exit(f2py_report_on_exit,(void*)\"#modulename#\");235#endif236return m;237}238#ifdef __cplusplus239}240#endif241""",242'separatorsfor': {'latexdoc': '\n\n',243'restdoc': '\n\n'},244'latexdoc': ['\\section{Module \\texttt{#texmodulename#}}\n',245'#modnote#\n',246'#latexdoc#'],247'restdoc': ['Module #modulename#\n' + '=' * 80,248'\n#restdoc#']249}250251defmod_rules = [252{'body': '/*eof body*/',253'method': '/*eof method*/',254'externroutines': '/*eof externroutines*/',255'routine_defs': '/*eof routine_defs*/',256'initf90modhooks': '/*eof initf90modhooks*/',257'initf2pywraphooks': '/*eof initf2pywraphooks*/',258'initcommonhooks': '/*eof initcommonhooks*/',259'latexdoc': '',260'restdoc': '',261'modnote': {hasnote: '#note#', l_not(hasnote): ''},262}263]264265routine_rules = {266'separatorsfor': sepdict,267'body': """268#begintitle#269static char doc_#apiname#[] = \"\\\n#docreturn##name#(#docsignatureshort#)\\n\\nWrapper for ``#name#``.\\\n\\n#docstrsigns#\";270/* #declfortranroutine# */271static PyObject *#apiname#(const PyObject *capi_self,272PyObject *capi_args,273PyObject *capi_keywds,274#functype# (*f2py_func)(#callprotoargument#)) {275PyObject * volatile capi_buildvalue = NULL;276volatile int f2py_success = 1;277#decl#278static char *capi_kwlist[] = {#kwlist##kwlistopt##kwlistxa#NULL};279#usercode#280#routdebugenter#281#ifdef F2PY_REPORT_ATEXIT282f2py_start_clock();283#endif284if (!PyArg_ParseTupleAndKeywords(capi_args,capi_keywds,\\285\"#argformat#|#keyformat##xaformat#:#pyname#\",\\286capi_kwlist#args_capi##keys_capi##keys_xa#))\n return NULL;287#frompyobj#288/*end of frompyobj*/289#ifdef F2PY_REPORT_ATEXIT290f2py_start_call_clock();291#endif292#callfortranroutine#293if (PyErr_Occurred())294f2py_success = 0;295#ifdef F2PY_REPORT_ATEXIT296f2py_stop_call_clock();297#endif298/*end of callfortranroutine*/299if (f2py_success) {300#pyobjfrom#301/*end of pyobjfrom*/302CFUNCSMESS(\"Building return value.\\n\");303capi_buildvalue = Py_BuildValue(\"#returnformat#\"#return#);304/*closepyobjfrom*/305#closepyobjfrom#306} /*if (f2py_success) after callfortranroutine*/307/*cleanupfrompyobj*/308#cleanupfrompyobj#309if (capi_buildvalue == NULL) {310#routdebugfailure#311} else {312#routdebugleave#313}314CFUNCSMESS(\"Freeing memory.\\n\");315#freemem#316#ifdef F2PY_REPORT_ATEXIT317f2py_stop_clock();318#endif319return capi_buildvalue;320}321#endtitle#322""",323'routine_defs': '#routine_def#',324'initf2pywraphooks': '#initf2pywraphook#',325'externroutines': '#declfortranroutine#',326'doc': '#docreturn##name#(#docsignature#)',327'docshort': '#docreturn##name#(#docsignatureshort#)',328'docs': '" #docreturn##name#(#docsignature#)\\n"\n',329'need': ['arrayobject.h', 'CFUNCSMESS', 'MINMAX'],330'cppmacros': {debugcapi: '#define DEBUGCFUNCS'},331'latexdoc': ['\\subsection{Wrapper function \\texttt{#texname#}}\n',332"""333\\noindent{{}\\verb@#docreturn##name#@{}}\\texttt{(#latexdocsignatureshort#)}334#routnote#335336#latexdocstrsigns#337"""],338'restdoc': ['Wrapped function ``#name#``\n' + '-' * 80,339340]341}342343################## Rules for C/API function ##############344345rout_rules = [346{ # Init347'separatorsfor': {'callfortranroutine': '\n', 'routdebugenter': '\n', 'decl': '\n',348'routdebugleave': '\n', 'routdebugfailure': '\n',349'setjmpbuf': ' || ',350'docstrreq': '\n', 'docstropt': '\n', 'docstrout': '\n',351'docstrcbs': '\n', 'docstrsigns': '\\n"\n"',352'latexdocstrsigns': '\n',353'latexdocstrreq': '\n', 'latexdocstropt': '\n',354'latexdocstrout': '\n', 'latexdocstrcbs': '\n',355},356'kwlist': '', 'kwlistopt': '', 'callfortran': '', 'callfortranappend': '',357'docsign': '', 'docsignopt': '', 'decl': '/*decl*/',358'freemem': '/*freemem*/',359'docsignshort': '', 'docsignoptshort': '',360'docstrsigns': '', 'latexdocstrsigns': '',361'docstrreq': '\\nParameters\\n----------',362'docstropt': '\\nOther Parameters\\n----------------',363'docstrout': '\\nReturns\\n-------',364'docstrcbs': '\\nNotes\\n-----\\nCall-back functions::\\n',365'latexdocstrreq': '\\noindent Required arguments:',366'latexdocstropt': '\\noindent Optional arguments:',367'latexdocstrout': '\\noindent Return objects:',368'latexdocstrcbs': '\\noindent Call-back functions:',369'args_capi': '', 'keys_capi': '', 'functype': '',370'frompyobj': '/*frompyobj*/',371# this list will be reversed372'cleanupfrompyobj': ['/*end of cleanupfrompyobj*/'],373'pyobjfrom': '/*pyobjfrom*/',374# this list will be reversed375'closepyobjfrom': ['/*end of closepyobjfrom*/'],376'topyarr': '/*topyarr*/', 'routdebugleave': '/*routdebugleave*/',377'routdebugenter': '/*routdebugenter*/',378'routdebugfailure': '/*routdebugfailure*/',379'callfortranroutine': '/*callfortranroutine*/',380'argformat': '', 'keyformat': '', 'need_cfuncs': '',381'docreturn': '', 'return': '', 'returnformat': '', 'rformat': '',382'kwlistxa': '', 'keys_xa': '', 'xaformat': '', 'docsignxa': '', 'docsignxashort': '',383'initf2pywraphook': '',384'routnote': {hasnote: '--- #note#', l_not(hasnote): ''},385}, {386'apiname': 'f2py_rout_#modulename#_#name#',387'pyname': '#modulename#.#name#',388'decl': '',389'_check': l_not(ismoduleroutine)390}, {391'apiname': 'f2py_rout_#modulename#_#f90modulename#_#name#',392'pyname': '#modulename#.#f90modulename#.#name#',393'decl': '',394'_check': ismoduleroutine395}, { # Subroutine396'functype': 'void',397'declfortranroutine': {l_and(l_not(l_or(ismoduleroutine, isintent_c)), l_not(isdummyroutine)): 'extern void #F_FUNC#(#fortranname#,#FORTRANNAME#)(#callprotoargument#);',398l_and(l_not(ismoduleroutine), isintent_c, l_not(isdummyroutine)): 'extern void #fortranname#(#callprotoargument#);',399ismoduleroutine: '',400isdummyroutine: ''401},402'routine_def': {l_not(l_or(ismoduleroutine, isintent_c, isdummyroutine)): ' {\"#name#\",-1,{{-1}},0,(char *)#F_FUNC#(#fortranname#,#FORTRANNAME#),(f2py_init_func)#apiname#,doc_#apiname#},',403l_and(l_not(ismoduleroutine), isintent_c, l_not(isdummyroutine)): ' {\"#name#\",-1,{{-1}},0,(char *)#fortranname#,(f2py_init_func)#apiname#,doc_#apiname#},',404l_and(l_not(ismoduleroutine), isdummyroutine): ' {\"#name#\",-1,{{-1}},0,NULL,(f2py_init_func)#apiname#,doc_#apiname#},',405},406'need': {l_and(l_not(l_or(ismoduleroutine, isintent_c)), l_not(isdummyroutine)): 'F_FUNC'},407'callfortranroutine': [408{debugcapi: [409""" fprintf(stderr,\"debug-capi:Fortran subroutine `#fortranname#(#callfortran#)\'\\n\");"""]},410{hasexternals: """\411if (#setjmpbuf#) {412f2py_success = 0;413} else {"""},414{isthreadsafe: ' Py_BEGIN_ALLOW_THREADS'},415{hascallstatement: ''' #callstatement#;416/*(*f2py_func)(#callfortran#);*/'''},417{l_not(l_or(hascallstatement, isdummyroutine))418: ' (*f2py_func)(#callfortran#);'},419{isthreadsafe: ' Py_END_ALLOW_THREADS'},420{hasexternals: """ }"""}421],422'_check': l_and(issubroutine, l_not(issubroutine_wrap)),423}, { # Wrapped function424'functype': 'void',425'declfortranroutine': {l_not(l_or(ismoduleroutine, isdummyroutine)): 'extern void #F_WRAPPEDFUNC#(#name_lower#,#NAME#)(#callprotoargument#);',426isdummyroutine: '',427},428429'routine_def': {l_not(l_or(ismoduleroutine, isdummyroutine)): ' {\"#name#\",-1,{{-1}},0,(char *)#F_WRAPPEDFUNC#(#name_lower#,#NAME#),(f2py_init_func)#apiname#,doc_#apiname#},',430isdummyroutine: ' {\"#name#\",-1,{{-1}},0,NULL,(f2py_init_func)#apiname#,doc_#apiname#},',431},432'initf2pywraphook': {l_not(l_or(ismoduleroutine, isdummyroutine)): '''433{434extern #ctype# #F_FUNC#(#name_lower#,#NAME#)(void);435PyObject* o = PyDict_GetItemString(d,"#name#");436tmp = F2PyCapsule_FromVoidPtr((void*)#F_FUNC#(#name_lower#,#NAME#),NULL);437PyObject_SetAttrString(o,"_cpointer", tmp);438Py_DECREF(tmp);439s = PyUnicode_FromString("#name#");440PyObject_SetAttrString(o,"__name__", s);441Py_DECREF(s);442}443'''},444'need': {l_not(l_or(ismoduleroutine, isdummyroutine)): ['F_WRAPPEDFUNC', 'F_FUNC']},445'callfortranroutine': [446{debugcapi: [447""" fprintf(stderr,\"debug-capi:Fortran subroutine `f2pywrap#name_lower#(#callfortran#)\'\\n\");"""]},448{hasexternals: """\449if (#setjmpbuf#) {450f2py_success = 0;451} else {"""},452{isthreadsafe: ' Py_BEGIN_ALLOW_THREADS'},453{l_not(l_or(hascallstatement, isdummyroutine))454: ' (*f2py_func)(#callfortran#);'},455{hascallstatement:456' #callstatement#;\n /*(*f2py_func)(#callfortran#);*/'},457{isthreadsafe: ' Py_END_ALLOW_THREADS'},458{hasexternals: ' }'}459],460'_check': isfunction_wrap,461}, { # Wrapped subroutine462'functype': 'void',463'declfortranroutine': {l_not(l_or(ismoduleroutine, isdummyroutine)): 'extern void #F_WRAPPEDFUNC#(#name_lower#,#NAME#)(#callprotoargument#);',464isdummyroutine: '',465},466467'routine_def': {l_not(l_or(ismoduleroutine, isdummyroutine)): ' {\"#name#\",-1,{{-1}},0,(char *)#F_WRAPPEDFUNC#(#name_lower#,#NAME#),(f2py_init_func)#apiname#,doc_#apiname#},',468isdummyroutine: ' {\"#name#\",-1,{{-1}},0,NULL,(f2py_init_func)#apiname#,doc_#apiname#},',469},470'initf2pywraphook': {l_not(l_or(ismoduleroutine, isdummyroutine)): '''471{472extern void #F_FUNC#(#name_lower#,#NAME#)(void);473PyObject* o = PyDict_GetItemString(d,"#name#");474tmp = F2PyCapsule_FromVoidPtr((void*)#F_FUNC#(#name_lower#,#NAME#),NULL);475PyObject_SetAttrString(o,"_cpointer", tmp);476Py_DECREF(tmp);477s = PyUnicode_FromString("#name#");478PyObject_SetAttrString(o,"__name__", s);479Py_DECREF(s);480}481'''},482'need': {l_not(l_or(ismoduleroutine, isdummyroutine)): ['F_WRAPPEDFUNC', 'F_FUNC']},483'callfortranroutine': [484{debugcapi: [485""" fprintf(stderr,\"debug-capi:Fortran subroutine `f2pywrap#name_lower#(#callfortran#)\'\\n\");"""]},486{hasexternals: """\487if (#setjmpbuf#) {488f2py_success = 0;489} else {"""},490{isthreadsafe: ' Py_BEGIN_ALLOW_THREADS'},491{l_not(l_or(hascallstatement, isdummyroutine))492: ' (*f2py_func)(#callfortran#);'},493{hascallstatement:494' #callstatement#;\n /*(*f2py_func)(#callfortran#);*/'},495{isthreadsafe: ' Py_END_ALLOW_THREADS'},496{hasexternals: ' }'}497],498'_check': issubroutine_wrap,499}, { # Function500'functype': '#ctype#',501'docreturn': {l_not(isintent_hide): '#rname#,'},502'docstrout': '#pydocsignout#',503'latexdocstrout': ['\\item[]{{}\\verb@#pydocsignout#@{}}',504{hasresultnote: '--- #resultnote#'}],505'callfortranroutine': [{l_and(debugcapi, isstringfunction): """\506#ifdef USESCOMPAQFORTRAN507fprintf(stderr,\"debug-capi:Fortran function #ctype# #fortranname#(#callcompaqfortran#)\\n\");508#else509fprintf(stderr,\"debug-capi:Fortran function #ctype# #fortranname#(#callfortran#)\\n\");510#endif511"""},512{l_and(debugcapi, l_not(isstringfunction)): """\513fprintf(stderr,\"debug-capi:Fortran function #ctype# #fortranname#(#callfortran#)\\n\");514"""}515],516'_check': l_and(isfunction, l_not(isfunction_wrap))517}, { # Scalar function518'declfortranroutine': {l_and(l_not(l_or(ismoduleroutine, isintent_c)), l_not(isdummyroutine)): 'extern #ctype# #F_FUNC#(#fortranname#,#FORTRANNAME#)(#callprotoargument#);',519l_and(l_not(ismoduleroutine), isintent_c, l_not(isdummyroutine)): 'extern #ctype# #fortranname#(#callprotoargument#);',520isdummyroutine: ''521},522'routine_def': {l_and(l_not(l_or(ismoduleroutine, isintent_c)), l_not(isdummyroutine)): ' {\"#name#\",-1,{{-1}},0,(char *)#F_FUNC#(#fortranname#,#FORTRANNAME#),(f2py_init_func)#apiname#,doc_#apiname#},',523l_and(l_not(ismoduleroutine), isintent_c, l_not(isdummyroutine)): ' {\"#name#\",-1,{{-1}},0,(char *)#fortranname#,(f2py_init_func)#apiname#,doc_#apiname#},',524isdummyroutine: ' {\"#name#\",-1,{{-1}},0,NULL,(f2py_init_func)#apiname#,doc_#apiname#},',525},526'decl': [{iscomplexfunction_warn: ' #ctype# #name#_return_value={0,0};',527l_not(iscomplexfunction): ' #ctype# #name#_return_value=0;'},528{iscomplexfunction:529' PyObject *#name#_return_value_capi = Py_None;'}530],531'callfortranroutine': [532{hasexternals: """\533if (#setjmpbuf#) {534f2py_success = 0;535} else {"""},536{isthreadsafe: ' Py_BEGIN_ALLOW_THREADS'},537{hascallstatement: ''' #callstatement#;538/* #name#_return_value = (*f2py_func)(#callfortran#);*/539'''},540{l_not(l_or(hascallstatement, isdummyroutine))541: ' #name#_return_value = (*f2py_func)(#callfortran#);'},542{isthreadsafe: ' Py_END_ALLOW_THREADS'},543{hasexternals: ' }'},544{l_and(debugcapi, iscomplexfunction)545: ' fprintf(stderr,"#routdebugshowvalue#\\n",#name#_return_value.r,#name#_return_value.i);'},546{l_and(debugcapi, l_not(iscomplexfunction)): ' fprintf(stderr,"#routdebugshowvalue#\\n",#name#_return_value);'}],547'pyobjfrom': {iscomplexfunction: ' #name#_return_value_capi = pyobj_from_#ctype#1(#name#_return_value);'},548'need': [{l_not(isdummyroutine): 'F_FUNC'},549{iscomplexfunction: 'pyobj_from_#ctype#1'},550{islong_longfunction: 'long_long'},551{islong_doublefunction: 'long_double'}],552'returnformat': {l_not(isintent_hide): '#rformat#'},553'return': {iscomplexfunction: ',#name#_return_value_capi',554l_not(l_or(iscomplexfunction, isintent_hide)): ',#name#_return_value'},555'_check': l_and(isfunction, l_not(isstringfunction), l_not(isfunction_wrap))556}, { # String function # in use for --no-wrap557'declfortranroutine': 'extern void #F_FUNC#(#fortranname#,#FORTRANNAME#)(#callprotoargument#);',558'routine_def': {l_not(l_or(ismoduleroutine, isintent_c)):559' {\"#name#\",-1,{{-1}},0,(char *)#F_FUNC#(#fortranname#,#FORTRANNAME#),(f2py_init_func)#apiname#,doc_#apiname#},',560l_and(l_not(ismoduleroutine), isintent_c):561' {\"#name#\",-1,{{-1}},0,(char *)#fortranname#,(f2py_init_func)#apiname#,doc_#apiname#},'562},563'decl': [' #ctype# #name#_return_value = NULL;',564' int #name#_return_value_len = 0;'],565'callfortran':'#name#_return_value,#name#_return_value_len,',566'callfortranroutine':[' #name#_return_value_len = #rlength#;',567' if ((#name#_return_value = (string)malloc('568+ '#name#_return_value_len+1) == NULL) {',569' PyErr_SetString(PyExc_MemoryError, \"out of memory\");',570' f2py_success = 0;',571' } else {',572" (#name#_return_value)[#name#_return_value_len] = '\\0';",573' }',574' if (f2py_success) {',575{hasexternals: """\576if (#setjmpbuf#) {577f2py_success = 0;578} else {"""},579{isthreadsafe: ' Py_BEGIN_ALLOW_THREADS'},580"""\581#ifdef USESCOMPAQFORTRAN582(*f2py_func)(#callcompaqfortran#);583#else584(*f2py_func)(#callfortran#);585#endif586""",587{isthreadsafe: ' Py_END_ALLOW_THREADS'},588{hasexternals: ' }'},589{debugcapi:590' fprintf(stderr,"#routdebugshowvalue#\\n",#name#_return_value_len,#name#_return_value);'},591' } /* if (f2py_success) after (string)malloc */',592],593'returnformat': '#rformat#',594'return': ',#name#_return_value',595'freemem': ' STRINGFREE(#name#_return_value);',596'need': ['F_FUNC', '#ctype#', 'STRINGFREE'],597'_check':l_and(isstringfunction, l_not(isfunction_wrap)) # ???obsolete598},599{ # Debugging600'routdebugenter': ' fprintf(stderr,"debug-capi:Python C/API function #modulename#.#name#(#docsignature#)\\n");',601'routdebugleave': ' fprintf(stderr,"debug-capi:Python C/API function #modulename#.#name#: successful.\\n");',602'routdebugfailure': ' fprintf(stderr,"debug-capi:Python C/API function #modulename#.#name#: failure.\\n");',603'_check': debugcapi604}605]606607################ Rules for arguments ##################608609typedef_need_dict = {islong_long: 'long_long',610islong_double: 'long_double',611islong_complex: 'complex_long_double',612isunsigned_char: 'unsigned_char',613isunsigned_short: 'unsigned_short',614isunsigned: 'unsigned',615isunsigned_long_long: 'unsigned_long_long',616isunsigned_chararray: 'unsigned_char',617isunsigned_shortarray: 'unsigned_short',618isunsigned_long_longarray: 'unsigned_long_long',619issigned_long_longarray: 'long_long',620}621622aux_rules = [623{624'separatorsfor': sepdict625},626{ # Common627'frompyobj': [' /* Processing auxiliary variable #varname# */',628{debugcapi: ' fprintf(stderr,"#vardebuginfo#\\n");'}, ],629'cleanupfrompyobj': ' /* End of cleaning variable #varname# */',630'need': typedef_need_dict,631},632# Scalars (not complex)633{ # Common634'decl': ' #ctype# #varname# = 0;',635'need': {hasinitvalue: 'math.h'},636'frompyobj': {hasinitvalue: ' #varname# = #init#;'},637'_check': l_and(isscalar, l_not(iscomplex)),638},639{640'return': ',#varname#',641'docstrout': '#pydocsignout#',642'docreturn': '#outvarname#,',643'returnformat': '#varrformat#',644'_check': l_and(isscalar, l_not(iscomplex), isintent_out),645},646# Complex scalars647{ # Common648'decl': ' #ctype# #varname#;',649'frompyobj': {hasinitvalue: ' #varname#.r = #init.r#, #varname#.i = #init.i#;'},650'_check': iscomplex651},652# String653{ # Common654'decl': [' #ctype# #varname# = NULL;',655' int slen(#varname#);',656],657'need':['len..'],658'_check':isstring659},660# Array661{ # Common662'decl': [' #ctype# *#varname# = NULL;',663' npy_intp #varname#_Dims[#rank#] = {#rank*[-1]#};',664' const int #varname#_Rank = #rank#;',665],666'need':['len..', {hasinitvalue: 'forcomb'}, {hasinitvalue: 'CFUNCSMESS'}],667'_check': isarray668},669# Scalararray670{ # Common671'_check': l_and(isarray, l_not(iscomplexarray))672}, { # Not hidden673'_check': l_and(isarray, l_not(iscomplexarray), isintent_nothide)674},675# Integer*1 array676{'need': '#ctype#',677'_check': isint1array,678'_depend': ''679},680# Integer*-1 array681{'need': '#ctype#',682'_check': isunsigned_chararray,683'_depend': ''684},685# Integer*-2 array686{'need': '#ctype#',687'_check': isunsigned_shortarray,688'_depend': ''689},690# Integer*-8 array691{'need': '#ctype#',692'_check': isunsigned_long_longarray,693'_depend': ''694},695# Complexarray696{'need': '#ctype#',697'_check': iscomplexarray,698'_depend': ''699},700# Stringarray701{702'callfortranappend': {isarrayofstrings: 'flen(#varname#),'},703'need': 'string',704'_check': isstringarray705}706]707708arg_rules = [709{710'separatorsfor': sepdict711},712{ # Common713'frompyobj': [' /* Processing variable #varname# */',714{debugcapi: ' fprintf(stderr,"#vardebuginfo#\\n");'}, ],715'cleanupfrompyobj': ' /* End of cleaning variable #varname# */',716'_depend': '',717'need': typedef_need_dict,718},719# Doc signatures720{721'docstropt': {l_and(isoptional, isintent_nothide): '#pydocsign#'},722'docstrreq': {l_and(isrequired, isintent_nothide): '#pydocsign#'},723'docstrout': {isintent_out: '#pydocsignout#'},724'latexdocstropt': {l_and(isoptional, isintent_nothide): ['\\item[]{{}\\verb@#pydocsign#@{}}',725{hasnote: '--- #note#'}]},726'latexdocstrreq': {l_and(isrequired, isintent_nothide): ['\\item[]{{}\\verb@#pydocsign#@{}}',727{hasnote: '--- #note#'}]},728'latexdocstrout': {isintent_out: ['\\item[]{{}\\verb@#pydocsignout#@{}}',729{l_and(hasnote, isintent_hide): '--- #note#',730l_and(hasnote, isintent_nothide): '--- See above.'}]},731'depend': ''732},733# Required/Optional arguments734{735'kwlist': '"#varname#",',736'docsign': '#varname#,',737'_check': l_and(isintent_nothide, l_not(isoptional))738},739{740'kwlistopt': '"#varname#",',741'docsignopt': '#varname#=#showinit#,',742'docsignoptshort': '#varname#,',743'_check': l_and(isintent_nothide, isoptional)744},745# Docstring/BuildValue746{747'docreturn': '#outvarname#,',748'returnformat': '#varrformat#',749'_check': isintent_out750},751# Externals (call-back functions)752{ # Common753'docsignxa': {isintent_nothide: '#varname#_extra_args=(),'},754'docsignxashort': {isintent_nothide: '#varname#_extra_args,'},755'docstropt': {isintent_nothide: '#varname#_extra_args : input tuple, optional\\n Default: ()'},756'docstrcbs': '#cbdocstr#',757'latexdocstrcbs': '\\item[] #cblatexdocstr#',758'latexdocstropt': {isintent_nothide: '\\item[]{{}\\verb@#varname#_extra_args := () input tuple@{}} --- Extra arguments for call-back function {{}\\verb@#varname#@{}}.'},759'decl': [' #cbname#_t #varname#_cb = { Py_None, NULL, 0 };',760' #cbname#_t *#varname#_cb_ptr = &#varname#_cb;',761' PyTupleObject *#varname#_xa_capi = NULL;',762{l_not(isintent_callback):763' #cbname#_typedef #varname#_cptr;'}764],765'kwlistxa': {isintent_nothide: '"#varname#_extra_args",'},766'argformat': {isrequired: 'O'},767'keyformat': {isoptional: 'O'},768'xaformat': {isintent_nothide: 'O!'},769'args_capi': {isrequired: ',&#varname#_cb.capi'},770'keys_capi': {isoptional: ',&#varname#_cb.capi'},771'keys_xa': ',&PyTuple_Type,&#varname#_xa_capi',772'setjmpbuf': '(setjmp(#varname#_cb.jmpbuf))',773'callfortran': {l_not(isintent_callback): '#varname#_cptr,'},774'need': ['#cbname#', 'setjmp.h'],775'_check':isexternal776},777{778'frompyobj': [{l_not(isintent_callback): """\779if(F2PyCapsule_Check(#varname#_cb.capi)) {780#varname#_cptr = F2PyCapsule_AsVoidPtr(#varname#_cb.capi);781} else {782#varname#_cptr = #cbname#;783}784"""}, {isintent_callback: """\785if (#varname#_cb.capi==Py_None) {786#varname#_cb.capi = PyObject_GetAttrString(#modulename#_module,\"#varname#\");787if (#varname#_cb.capi) {788if (#varname#_xa_capi==NULL) {789if (PyObject_HasAttrString(#modulename#_module,\"#varname#_extra_args\")) {790PyObject* capi_tmp = PyObject_GetAttrString(#modulename#_module,\"#varname#_extra_args\");791if (capi_tmp) {792#varname#_xa_capi = (PyTupleObject *)PySequence_Tuple(capi_tmp);793Py_DECREF(capi_tmp);794}795else {796#varname#_xa_capi = (PyTupleObject *)Py_BuildValue(\"()\");797}798if (#varname#_xa_capi==NULL) {799PyErr_SetString(#modulename#_error,\"Failed to convert #modulename#.#varname#_extra_args to tuple.\\n\");800return NULL;801}802}803}804}805if (#varname#_cb.capi==NULL) {806PyErr_SetString(#modulename#_error,\"Callback #varname# not defined (as an argument or module #modulename# attribute).\\n\");807return NULL;808}809}810"""},811"""\812if (create_cb_arglist(#varname#_cb.capi,#varname#_xa_capi,#maxnofargs#,#nofoptargs#,&#varname#_cb.nofargs,&#varname#_cb.args_capi,\"failed in processing argument list for call-back #varname#.\")) {813""",814{debugcapi: ["""\815fprintf(stderr,\"debug-capi:Assuming %d arguments; at most #maxnofargs#(-#nofoptargs#) is expected.\\n\",#varname#_cb.nofargs);816CFUNCSMESSPY(\"for #varname#=\",#varname#_cb.capi);""",817{l_not(isintent_callback): """ fprintf(stderr,\"#vardebugshowvalue# (call-back in C).\\n\",#cbname#);"""}]},818"""\819CFUNCSMESS(\"Saving callback variables for `#varname#`.\\n\");820#varname#_cb_ptr = swap_active_#cbname#(#varname#_cb_ptr);""",821],822'cleanupfrompyobj':823"""\824CFUNCSMESS(\"Restoring callback variables for `#varname#`.\\n\");825#varname#_cb_ptr = swap_active_#cbname#(#varname#_cb_ptr);826Py_DECREF(#varname#_cb.args_capi);827}""",828'need': ['SWAP', 'create_cb_arglist'],829'_check':isexternal,830'_depend':''831},832# Scalars (not complex)833{ # Common834'decl': ' #ctype# #varname# = 0;',835'pyobjfrom': {debugcapi: ' fprintf(stderr,"#vardebugshowvalue#\\n",#varname#);'},836'callfortran': {isintent_c: '#varname#,', l_not(isintent_c): '&#varname#,'},837'return': {isintent_out: ',#varname#'},838'_check': l_and(isscalar, l_not(iscomplex))839}, {840'need': {hasinitvalue: 'math.h'},841'_check': l_and(isscalar, l_not(iscomplex)),842}, { # Not hidden843'decl': ' PyObject *#varname#_capi = Py_None;',844'argformat': {isrequired: 'O'},845'keyformat': {isoptional: 'O'},846'args_capi': {isrequired: ',&#varname#_capi'},847'keys_capi': {isoptional: ',&#varname#_capi'},848'pyobjfrom': {isintent_inout: """\849f2py_success = try_pyarr_from_#ctype#(#varname#_capi,&#varname#);850if (f2py_success) {"""},851'closepyobjfrom': {isintent_inout: " } /*if (f2py_success) of #varname# pyobjfrom*/"},852'need': {isintent_inout: 'try_pyarr_from_#ctype#'},853'_check': l_and(isscalar, l_not(iscomplex), isintent_nothide)854}, {855'frompyobj': [856# hasinitvalue...857# if pyobj is None:858# varname = init859# else860# from_pyobj(varname)861#862# isoptional and noinitvalue...863# if pyobj is not None:864# from_pyobj(varname)865# else:866# varname is uninitialized867#868# ...869# from_pyobj(varname)870#871{hasinitvalue: ' if (#varname#_capi == Py_None) #varname# = #init#; else',872'_depend': ''},873{l_and(isoptional, l_not(hasinitvalue)): ' if (#varname#_capi != Py_None)',874'_depend': ''},875{l_not(islogical): '''\876f2py_success = #ctype#_from_pyobj(&#varname#,#varname#_capi,"#pyname#() #nth# (#varname#) can\'t be converted to #ctype#");877if (f2py_success) {'''},878{islogical: '''\879#varname# = (#ctype#)PyObject_IsTrue(#varname#_capi);880f2py_success = 1;881if (f2py_success) {'''},882],883'cleanupfrompyobj': ' } /*if (f2py_success) of #varname#*/',884'need': {l_not(islogical): '#ctype#_from_pyobj'},885'_check': l_and(isscalar, l_not(iscomplex), isintent_nothide),886'_depend': ''887}, { # Hidden888'frompyobj': {hasinitvalue: ' #varname# = #init#;'},889'need': typedef_need_dict,890'_check': l_and(isscalar, l_not(iscomplex), isintent_hide),891'_depend': ''892}, { # Common893'frompyobj': {debugcapi: ' fprintf(stderr,"#vardebugshowvalue#\\n",#varname#);'},894'_check': l_and(isscalar, l_not(iscomplex)),895'_depend': ''896},897# Complex scalars898{ # Common899'decl': ' #ctype# #varname#;',900'callfortran': {isintent_c: '#varname#,', l_not(isintent_c): '&#varname#,'},901'pyobjfrom': {debugcapi: ' fprintf(stderr,"#vardebugshowvalue#\\n",#varname#.r,#varname#.i);'},902'return': {isintent_out: ',#varname#_capi'},903'_check': iscomplex904}, { # Not hidden905'decl': ' PyObject *#varname#_capi = Py_None;',906'argformat': {isrequired: 'O'},907'keyformat': {isoptional: 'O'},908'args_capi': {isrequired: ',&#varname#_capi'},909'keys_capi': {isoptional: ',&#varname#_capi'},910'need': {isintent_inout: 'try_pyarr_from_#ctype#'},911'pyobjfrom': {isintent_inout: """\912f2py_success = try_pyarr_from_#ctype#(#varname#_capi,&#varname#);913if (f2py_success) {"""},914'closepyobjfrom': {isintent_inout: " } /*if (f2py_success) of #varname# pyobjfrom*/"},915'_check': l_and(iscomplex, isintent_nothide)916}, {917'frompyobj': [{hasinitvalue: ' if (#varname#_capi==Py_None) {#varname#.r = #init.r#, #varname#.i = #init.i#;} else'},918{l_and(isoptional, l_not(hasinitvalue))919: ' if (#varname#_capi != Py_None)'},920' f2py_success = #ctype#_from_pyobj(&#varname#,#varname#_capi,"#pyname#() #nth# (#varname#) can\'t be converted to #ctype#");'921'\n if (f2py_success) {'],922'cleanupfrompyobj': ' } /*if (f2py_success) of #varname# frompyobj*/',923'need': ['#ctype#_from_pyobj'],924'_check': l_and(iscomplex, isintent_nothide),925'_depend': ''926}, { # Hidden927'decl': {isintent_out: ' PyObject *#varname#_capi = Py_None;'},928'_check': l_and(iscomplex, isintent_hide)929}, {930'frompyobj': {hasinitvalue: ' #varname#.r = #init.r#, #varname#.i = #init.i#;'},931'_check': l_and(iscomplex, isintent_hide),932'_depend': ''933}, { # Common934'pyobjfrom': {isintent_out: ' #varname#_capi = pyobj_from_#ctype#1(#varname#);'},935'need': ['pyobj_from_#ctype#1'],936'_check': iscomplex937}, {938'frompyobj': {debugcapi: ' fprintf(stderr,"#vardebugshowvalue#\\n",#varname#.r,#varname#.i);'},939'_check': iscomplex,940'_depend': ''941},942# String943{ # Common944'decl': [' #ctype# #varname# = NULL;',945' int slen(#varname#);',946' PyObject *#varname#_capi = Py_None;'],947'callfortran':'#varname#,',948'callfortranappend':'slen(#varname#),',949'pyobjfrom':[950{debugcapi:951' fprintf(stderr,'952'"#vardebugshowvalue#\\n",slen(#varname#),#varname#);'},953# The trailing null value for Fortran is blank.954{l_and(isintent_out, l_not(isintent_c)):955" STRINGPADN(#varname#, slen(#varname#), ' ', '\\0');"},956],957'return': {isintent_out: ',#varname#'},958'need': ['len..',959{l_and(isintent_out, l_not(isintent_c)): 'STRINGPADN'}],960'_check':isstring961}, { # Common962'frompyobj': [963"""\964slen(#varname#) = #length#;965f2py_success = #ctype#_from_pyobj(&#varname#,&slen(#varname#),#init#,"""966"""#varname#_capi,\"#ctype#_from_pyobj failed in converting #nth#"""967"""`#varname#\' of #pyname# to C #ctype#\");968if (f2py_success) {""",969# The trailing null value for Fortran is blank.970{l_not(isintent_c):971" STRINGPADN(#varname#, slen(#varname#), '\\0', ' ');"},972],973'cleanupfrompyobj': """\974STRINGFREE(#varname#);975} /*if (f2py_success) of #varname#*/""",976'need': ['#ctype#_from_pyobj', 'len..', 'STRINGFREE',977{l_not(isintent_c): 'STRINGPADN'}],978'_check':isstring,979'_depend':''980}, { # Not hidden981'argformat': {isrequired: 'O'},982'keyformat': {isoptional: 'O'},983'args_capi': {isrequired: ',&#varname#_capi'},984'keys_capi': {isoptional: ',&#varname#_capi'},985'pyobjfrom': [986{l_and(isintent_inout, l_not(isintent_c)):987" STRINGPADN(#varname#, slen(#varname#), ' ', '\\0');"},988{isintent_inout: '''\989f2py_success = try_pyarr_from_#ctype#(#varname#_capi, #varname#,990slen(#varname#));991if (f2py_success) {'''}],992'closepyobjfrom': {isintent_inout: ' } /*if (f2py_success) of #varname# pyobjfrom*/'},993'need': {isintent_inout: 'try_pyarr_from_#ctype#',994l_and(isintent_inout, l_not(isintent_c)): 'STRINGPADN'},995'_check': l_and(isstring, isintent_nothide)996}, { # Hidden997'_check': l_and(isstring, isintent_hide)998}, {999'frompyobj': {debugcapi: ' fprintf(stderr,"#vardebugshowvalue#\\n",slen(#varname#),#varname#);'},1000'_check': isstring,1001'_depend': ''1002},1003# Array1004{ # Common1005'decl': [' #ctype# *#varname# = NULL;',1006' npy_intp #varname#_Dims[#rank#] = {#rank*[-1]#};',1007' const int #varname#_Rank = #rank#;',1008' PyArrayObject *capi_#varname#_tmp = NULL;',1009' int capi_#varname#_intent = 0;',1010],1011'callfortran':'#varname#,',1012'return':{isintent_out: ',capi_#varname#_tmp'},1013'need': 'len..',1014'_check': isarray1015}, { # intent(overwrite) array1016'decl': ' int capi_overwrite_#varname# = 1;',1017'kwlistxa': '"overwrite_#varname#",',1018'xaformat': 'i',1019'keys_xa': ',&capi_overwrite_#varname#',1020'docsignxa': 'overwrite_#varname#=1,',1021'docsignxashort': 'overwrite_#varname#,',1022'docstropt': 'overwrite_#varname# : input int, optional\\n Default: 1',1023'_check': l_and(isarray, isintent_overwrite),1024}, {1025'frompyobj': ' capi_#varname#_intent |= (capi_overwrite_#varname#?0:F2PY_INTENT_COPY);',1026'_check': l_and(isarray, isintent_overwrite),1027'_depend': '',1028},1029{ # intent(copy) array1030'decl': ' int capi_overwrite_#varname# = 0;',1031'kwlistxa': '"overwrite_#varname#",',1032'xaformat': 'i',1033'keys_xa': ',&capi_overwrite_#varname#',1034'docsignxa': 'overwrite_#varname#=0,',1035'docsignxashort': 'overwrite_#varname#,',1036'docstropt': 'overwrite_#varname# : input int, optional\\n Default: 0',1037'_check': l_and(isarray, isintent_copy),1038}, {1039'frompyobj': ' capi_#varname#_intent |= (capi_overwrite_#varname#?0:F2PY_INTENT_COPY);',1040'_check': l_and(isarray, isintent_copy),1041'_depend': '',1042}, {1043'need': [{hasinitvalue: 'forcomb'}, {hasinitvalue: 'CFUNCSMESS'}],1044'_check': isarray,1045'_depend': ''1046}, { # Not hidden1047'decl': ' PyObject *#varname#_capi = Py_None;',1048'argformat': {isrequired: 'O'},1049'keyformat': {isoptional: 'O'},1050'args_capi': {isrequired: ',&#varname#_capi'},1051'keys_capi': {isoptional: ',&#varname#_capi'},1052'_check': l_and(isarray, isintent_nothide)1053}, {1054'frompyobj': [' #setdims#;',1055' capi_#varname#_intent |= #intent#;',1056{isintent_hide:1057' capi_#varname#_tmp = array_from_pyobj(#atype#,#varname#_Dims,#varname#_Rank,capi_#varname#_intent,Py_None);'},1058{isintent_nothide:1059' capi_#varname#_tmp = array_from_pyobj(#atype#,#varname#_Dims,#varname#_Rank,capi_#varname#_intent,#varname#_capi);'},1060"""\1061if (capi_#varname#_tmp == NULL) {1062PyObject *exc, *val, *tb;1063PyErr_Fetch(&exc, &val, &tb);1064PyErr_SetString(exc ? exc : #modulename#_error,\"failed in converting #nth# `#varname#\' of #pyname# to C/Fortran array\" );1065npy_PyErr_ChainExceptionsCause(exc, val, tb);1066} else {1067#varname# = (#ctype# *)(PyArray_DATA(capi_#varname#_tmp));1068""",1069{hasinitvalue: [1070{isintent_nothide:1071' if (#varname#_capi == Py_None) {'},1072{isintent_hide: ' {'},1073{iscomplexarray: ' #ctype# capi_c;'},1074"""\1075int *_i,capi_i=0;1076CFUNCSMESS(\"#name#: Initializing #varname#=#init#\\n\");1077if (initforcomb(PyArray_DIMS(capi_#varname#_tmp),PyArray_NDIM(capi_#varname#_tmp),1)) {1078while ((_i = nextforcomb()))1079#varname#[capi_i++] = #init#; /* fortran way */1080} else {1081PyObject *exc, *val, *tb;1082PyErr_Fetch(&exc, &val, &tb);1083PyErr_SetString(exc ? exc : #modulename#_error,\"Initialization of #nth# #varname# failed (initforcomb).\");1084npy_PyErr_ChainExceptionsCause(exc, val, tb);1085f2py_success = 0;1086}1087}1088if (f2py_success) {"""]},1089],1090'cleanupfrompyobj': [ # note that this list will be reversed1091' } /*if (capi_#varname#_tmp == NULL) ... else of #varname#*/',1092{l_not(l_or(isintent_out, isintent_hide)): """\1093if((PyObject *)capi_#varname#_tmp!=#varname#_capi) {1094Py_XDECREF(capi_#varname#_tmp); }"""},1095{l_and(isintent_hide, l_not(isintent_out))1096: """ Py_XDECREF(capi_#varname#_tmp);"""},1097{hasinitvalue: ' } /*if (f2py_success) of #varname# init*/'},1098],1099'_check': isarray,1100'_depend': ''1101},1102# Scalararray1103{ # Common1104'_check': l_and(isarray, l_not(iscomplexarray))1105}, { # Not hidden1106'_check': l_and(isarray, l_not(iscomplexarray), isintent_nothide)1107},1108# Integer*1 array1109{'need': '#ctype#',1110'_check': isint1array,1111'_depend': ''1112},1113# Integer*-1 array1114{'need': '#ctype#',1115'_check': isunsigned_chararray,1116'_depend': ''1117},1118# Integer*-2 array1119{'need': '#ctype#',1120'_check': isunsigned_shortarray,1121'_depend': ''1122},1123# Integer*-8 array1124{'need': '#ctype#',1125'_check': isunsigned_long_longarray,1126'_depend': ''1127},1128# Complexarray1129{'need': '#ctype#',1130'_check': iscomplexarray,1131'_depend': ''1132},1133# Stringarray1134{1135'callfortranappend': {isarrayofstrings: 'flen(#varname#),'},1136'need': 'string',1137'_check': isstringarray1138}1139]11401141################# Rules for checking ###############11421143check_rules = [1144{1145'frompyobj': {debugcapi: ' fprintf(stderr,\"debug-capi:Checking `#check#\'\\n\");'},1146'need': 'len..'1147}, {1148'frompyobj': ' CHECKSCALAR(#check#,\"#check#\",\"#nth# #varname#\",\"#varshowvalue#\",#varname#) {',1149'cleanupfrompyobj': ' } /*CHECKSCALAR(#check#)*/',1150'need': 'CHECKSCALAR',1151'_check': l_and(isscalar, l_not(iscomplex)),1152'_break': ''1153}, {1154'frompyobj': ' CHECKSTRING(#check#,\"#check#\",\"#nth# #varname#\",\"#varshowvalue#\",#varname#) {',1155'cleanupfrompyobj': ' } /*CHECKSTRING(#check#)*/',1156'need': 'CHECKSTRING',1157'_check': isstring,1158'_break': ''1159}, {1160'need': 'CHECKARRAY',1161'frompyobj': ' CHECKARRAY(#check#,\"#check#\",\"#nth# #varname#\") {',1162'cleanupfrompyobj': ' } /*CHECKARRAY(#check#)*/',1163'_check': isarray,1164'_break': ''1165}, {1166'need': 'CHECKGENERIC',1167'frompyobj': ' CHECKGENERIC(#check#,\"#check#\",\"#nth# #varname#\") {',1168'cleanupfrompyobj': ' } /*CHECKGENERIC(#check#)*/',1169}1170]11711172########## Applying the rules. No need to modify what follows #############11731174#################### Build C/API module #######################117511761177def buildmodule(m, um):1178"""1179Return1180"""1181outmess(' Building module "%s"...\n' % (m['name']))1182ret = {}1183mod_rules = defmod_rules[:]1184vrd = capi_maps.modsign2map(m)1185rd = dictappend({'f2py_version': f2py_version}, vrd)1186funcwrappers = []1187funcwrappers2 = [] # F90 codes1188for n in m['interfaced']:1189nb = None1190for bi in m['body']:1191if bi['block'] not in ['interface', 'abstract interface']:1192errmess('buildmodule: Expected interface block. Skipping.\n')1193continue1194for b in bi['body']:1195if b['name'] == n:1196nb = b1197break11981199if not nb:1200errmess(1201'buildmodule: Could not found the body of interfaced routine "%s". Skipping.\n' % (n))1202continue1203nb_list = [nb]1204if 'entry' in nb:1205for k, a in nb['entry'].items():1206nb1 = copy.deepcopy(nb)1207del nb1['entry']1208nb1['name'] = k1209nb1['args'] = a1210nb_list.append(nb1)1211for nb in nb_list:1212# requiresf90wrapper must be called before buildapi as it1213# rewrites assumed shape arrays as automatic arrays.1214isf90 = requiresf90wrapper(nb)1215api, wrap = buildapi(nb)1216if wrap:1217if isf90:1218funcwrappers2.append(wrap)1219else:1220funcwrappers.append(wrap)1221ar = applyrules(api, vrd)1222rd = dictappend(rd, ar)12231224# Construct COMMON block support1225cr, wrap = common_rules.buildhooks(m)1226if wrap:1227funcwrappers.append(wrap)1228ar = applyrules(cr, vrd)1229rd = dictappend(rd, ar)12301231# Construct F90 module support1232mr, wrap = f90mod_rules.buildhooks(m)1233if wrap:1234funcwrappers2.append(wrap)1235ar = applyrules(mr, vrd)1236rd = dictappend(rd, ar)12371238for u in um:1239ar = use_rules.buildusevars(u, m['use'][u['name']])1240rd = dictappend(rd, ar)12411242needs = cfuncs.get_needs()1243code = {}1244for n in needs.keys():1245code[n] = []1246for k in needs[n]:1247c = ''1248if k in cfuncs.includes0:1249c = cfuncs.includes0[k]1250elif k in cfuncs.includes:1251c = cfuncs.includes[k]1252elif k in cfuncs.userincludes:1253c = cfuncs.userincludes[k]1254elif k in cfuncs.typedefs:1255c = cfuncs.typedefs[k]1256elif k in cfuncs.typedefs_generated:1257c = cfuncs.typedefs_generated[k]1258elif k in cfuncs.cppmacros:1259c = cfuncs.cppmacros[k]1260elif k in cfuncs.cfuncs:1261c = cfuncs.cfuncs[k]1262elif k in cfuncs.callbacks:1263c = cfuncs.callbacks[k]1264elif k in cfuncs.f90modhooks:1265c = cfuncs.f90modhooks[k]1266elif k in cfuncs.commonhooks:1267c = cfuncs.commonhooks[k]1268else:1269errmess('buildmodule: unknown need %s.\n' % (repr(k)))1270continue1271code[n].append(c)1272mod_rules.append(code)1273for r in mod_rules:1274if ('_check' in r and r['_check'](m)) or ('_check' not in r):1275ar = applyrules(r, vrd, m)1276rd = dictappend(rd, ar)1277ar = applyrules(module_rules, rd)12781279fn = os.path.join(options['buildpath'], vrd['coutput'])1280ret['csrc'] = fn1281with open(fn, 'w') as f:1282f.write(ar['modulebody'].replace('\t', 2 * ' '))1283outmess(' Wrote C/API module "%s" to file "%s"\n' % (m['name'], fn))12841285if options['dorestdoc']:1286fn = os.path.join(1287options['buildpath'], vrd['modulename'] + 'module.rest')1288with open(fn, 'w') as f:1289f.write('.. -*- rest -*-\n')1290f.write('\n'.join(ar['restdoc']))1291outmess(' ReST Documentation is saved to file "%s/%smodule.rest"\n' %1292(options['buildpath'], vrd['modulename']))1293if options['dolatexdoc']:1294fn = os.path.join(1295options['buildpath'], vrd['modulename'] + 'module.tex')1296ret['ltx'] = fn1297with open(fn, 'w') as f:1298f.write(1299'%% This file is auto-generated with f2py (version:%s)\n' % (f2py_version))1300if 'shortlatex' not in options:1301f.write(1302'\\documentclass{article}\n\\usepackage{a4wide}\n\\begin{document}\n\\tableofcontents\n\n')1303f.write('\n'.join(ar['latexdoc']))1304if 'shortlatex' not in options:1305f.write('\\end{document}')1306outmess(' Documentation is saved to file "%s/%smodule.tex"\n' %1307(options['buildpath'], vrd['modulename']))1308if funcwrappers:1309wn = os.path.join(options['buildpath'], vrd['f2py_wrapper_output'])1310ret['fsrc'] = wn1311with open(wn, 'w') as f:1312f.write('C -*- fortran -*-\n')1313f.write(1314'C This file is autogenerated with f2py (version:%s)\n' % (f2py_version))1315f.write(1316'C It contains Fortran 77 wrappers to fortran functions.\n')1317lines = []1318for l in ('\n\n'.join(funcwrappers) + '\n').split('\n'):1319if 0 <= l.find('!') < 66:1320# don't split comment lines1321lines.append(l + '\n')1322elif l and l[0] == ' ':1323while len(l) >= 66:1324lines.append(l[:66] + '\n &')1325l = l[66:]1326lines.append(l + '\n')1327else:1328lines.append(l + '\n')1329lines = ''.join(lines).replace('\n &\n', '\n')1330f.write(lines)1331outmess(' Fortran 77 wrappers are saved to "%s"\n' % (wn))1332if funcwrappers2:1333wn = os.path.join(1334options['buildpath'], '%s-f2pywrappers2.f90' % (vrd['modulename']))1335ret['fsrc'] = wn1336with open(wn, 'w') as f:1337f.write('! -*- f90 -*-\n')1338f.write(1339'! This file is autogenerated with f2py (version:%s)\n' % (f2py_version))1340f.write(1341'! It contains Fortran 90 wrappers to fortran functions.\n')1342lines = []1343for l in ('\n\n'.join(funcwrappers2) + '\n').split('\n'):1344if 0 <= l.find('!') < 72:1345# don't split comment lines1346lines.append(l + '\n')1347elif len(l) > 72 and l[0] == ' ':1348lines.append(l[:72] + '&\n &')1349l = l[72:]1350while len(l) > 66:1351lines.append(l[:66] + '&\n &')1352l = l[66:]1353lines.append(l + '\n')1354else:1355lines.append(l + '\n')1356lines = ''.join(lines).replace('\n &\n', '\n')1357f.write(lines)1358outmess(' Fortran 90 wrappers are saved to "%s"\n' % (wn))1359return ret13601361################## Build C/API function #############13621363stnd = {1: 'st', 2: 'nd', 3: 'rd', 4: 'th', 5: 'th',13646: 'th', 7: 'th', 8: 'th', 9: 'th', 0: 'th'}136513661367def buildapi(rout):1368rout, wrap = func2subr.assubr(rout)1369args, depargs = getargs2(rout)1370capi_maps.depargs = depargs1371var = rout['vars']13721373if ismoduleroutine(rout):1374outmess(' Constructing wrapper function "%s.%s"...\n' %1375(rout['modulename'], rout['name']))1376else:1377outmess(' Constructing wrapper function "%s"...\n' % (rout['name']))1378# Routine1379vrd = capi_maps.routsign2map(rout)1380rd = dictappend({}, vrd)1381for r in rout_rules:1382if ('_check' in r and r['_check'](rout)) or ('_check' not in r):1383ar = applyrules(r, vrd, rout)1384rd = dictappend(rd, ar)13851386# Args1387nth, nthk = 0, 01388savevrd = {}1389for a in args:1390vrd = capi_maps.sign2map(a, var[a])1391if isintent_aux(var[a]):1392_rules = aux_rules1393else:1394_rules = arg_rules1395if not isintent_hide(var[a]):1396if not isoptional(var[a]):1397nth = nth + 11398vrd['nth'] = repr(nth) + stnd[nth % 10] + ' argument'1399else:1400nthk = nthk + 11401vrd['nth'] = repr(nthk) + stnd[nthk % 10] + ' keyword'1402else:1403vrd['nth'] = 'hidden'1404savevrd[a] = vrd1405for r in _rules:1406if '_depend' in r:1407continue1408if ('_check' in r and r['_check'](var[a])) or ('_check' not in r):1409ar = applyrules(r, vrd, var[a])1410rd = dictappend(rd, ar)1411if '_break' in r:1412break1413for a in depargs:1414if isintent_aux(var[a]):1415_rules = aux_rules1416else:1417_rules = arg_rules1418vrd = savevrd[a]1419for r in _rules:1420if '_depend' not in r:1421continue1422if ('_check' in r and r['_check'](var[a])) or ('_check' not in r):1423ar = applyrules(r, vrd, var[a])1424rd = dictappend(rd, ar)1425if '_break' in r:1426break1427if 'check' in var[a]:1428for c in var[a]['check']:1429vrd['check'] = c1430ar = applyrules(check_rules, vrd, var[a])1431rd = dictappend(rd, ar)1432if isinstance(rd['cleanupfrompyobj'], list):1433rd['cleanupfrompyobj'].reverse()1434if isinstance(rd['closepyobjfrom'], list):1435rd['closepyobjfrom'].reverse()1436rd['docsignature'] = stripcomma(replace('#docsign##docsignopt##docsignxa#',1437{'docsign': rd['docsign'],1438'docsignopt': rd['docsignopt'],1439'docsignxa': rd['docsignxa']}))1440optargs = stripcomma(replace('#docsignopt##docsignxa#',1441{'docsignxa': rd['docsignxashort'],1442'docsignopt': rd['docsignoptshort']}1443))1444if optargs == '':1445rd['docsignatureshort'] = stripcomma(1446replace('#docsign#', {'docsign': rd['docsign']}))1447else:1448rd['docsignatureshort'] = replace('#docsign#[#docsignopt#]',1449{'docsign': rd['docsign'],1450'docsignopt': optargs,1451})1452rd['latexdocsignatureshort'] = rd['docsignatureshort'].replace('_', '\\_')1453rd['latexdocsignatureshort'] = rd[1454'latexdocsignatureshort'].replace(',', ', ')1455cfs = stripcomma(replace('#callfortran##callfortranappend#', {1456'callfortran': rd['callfortran'], 'callfortranappend': rd['callfortranappend']}))1457if len(rd['callfortranappend']) > 1:1458rd['callcompaqfortran'] = stripcomma(replace('#callfortran# 0,#callfortranappend#', {1459'callfortran': rd['callfortran'], 'callfortranappend': rd['callfortranappend']}))1460else:1461rd['callcompaqfortran'] = cfs1462rd['callfortran'] = cfs1463if isinstance(rd['docreturn'], list):1464rd['docreturn'] = stripcomma(1465replace('#docreturn#', {'docreturn': rd['docreturn']})) + ' = '1466rd['docstrsigns'] = []1467rd['latexdocstrsigns'] = []1468for k in ['docstrreq', 'docstropt', 'docstrout', 'docstrcbs']:1469if k in rd and isinstance(rd[k], list):1470rd['docstrsigns'] = rd['docstrsigns'] + rd[k]1471k = 'latex' + k1472if k in rd and isinstance(rd[k], list):1473rd['latexdocstrsigns'] = rd['latexdocstrsigns'] + rd[k][0:1] +\1474['\\begin{description}'] + rd[k][1:] +\1475['\\end{description}']14761477ar = applyrules(routine_rules, rd)1478if ismoduleroutine(rout):1479outmess(' %s\n' % (ar['docshort']))1480else:1481outmess(' %s\n' % (ar['docshort']))1482return ar, wrap148314841485#################### EOF rules.py #######################148614871488