12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460 |
- #!/usr/bin/env python3
- """
- Rules for building C/API module with f2py2e.
- Here is a skeleton of a new wrapper function (13Dec2001):
- wrapper_function(args)
- declarations
- get_python_arguments, say, `a' and `b'
- get_a_from_python
- if (successful) {
- get_b_from_python
- if (successful) {
- callfortran
- if (successful) {
- put_a_to_python
- if (successful) {
- put_b_to_python
- if (successful) {
- buildvalue = ...
- }
- }
- }
- }
- cleanup_b
- }
- cleanup_a
- return buildvalue
- Copyright 1999,2000 Pearu Peterson all rights reserved,
- Pearu Peterson <pearu@ioc.ee>
- Permission to use, modify, and distribute this software is given under the
- terms of the NumPy License.
- NO WARRANTY IS EXPRESSED OR IMPLIED. USE AT YOUR OWN RISK.
- $Date: 2005/08/30 08:58:42 $
- Pearu Peterson
- """
- import os
- import time
- import copy
- # __version__.version is now the same as the NumPy version
- from . import __version__
- f2py_version = __version__.version
- numpy_version = __version__.version
- from .auxfuncs import (
- applyrules, debugcapi, dictappend, errmess, gentitle, getargs2,
- hascallstatement, hasexternals, hasinitvalue, hasnote, hasresultnote,
- isarray, isarrayofstrings, iscomplex, iscomplexarray,
- iscomplexfunction, iscomplexfunction_warn, isdummyroutine, isexternal,
- isfunction, isfunction_wrap, isint1array, isintent_aux, isintent_c,
- isintent_callback, isintent_copy, isintent_hide, isintent_inout,
- isintent_nothide, isintent_out, isintent_overwrite, islogical,
- islong_complex, islong_double, islong_doublefunction, islong_long,
- islong_longfunction, ismoduleroutine, isoptional, isrequired, isscalar,
- issigned_long_longarray, isstring, isstringarray, isstringfunction,
- issubroutine, issubroutine_wrap, isthreadsafe, isunsigned,
- isunsigned_char, isunsigned_chararray, isunsigned_long_long,
- isunsigned_long_longarray, isunsigned_short, isunsigned_shortarray,
- l_and, l_not, l_or, outmess, replace, stripcomma, requiresf90wrapper
- )
- from . import capi_maps
- from . import cfuncs
- from . import common_rules
- from . import use_rules
- from . import f90mod_rules
- from . import func2subr
- options = {}
- sepdict = {}
- #for k in ['need_cfuncs']: sepdict[k]=','
- for k in ['decl',
- 'frompyobj',
- 'cleanupfrompyobj',
- 'topyarr', 'method',
- 'pyobjfrom', 'closepyobjfrom',
- 'freemem',
- 'userincludes',
- 'includes0', 'includes', 'typedefs', 'typedefs_generated',
- 'cppmacros', 'cfuncs', 'callbacks',
- 'latexdoc',
- 'restdoc',
- 'routine_defs', 'externroutines',
- 'initf2pywraphooks',
- 'commonhooks', 'initcommonhooks',
- 'f90modhooks', 'initf90modhooks']:
- sepdict[k] = '\n'
- #################### Rules for C/API module #################
- generationtime = int(os.environ.get('SOURCE_DATE_EPOCH', time.time()))
- module_rules = {
- 'modulebody': """\
- /* File: #modulename#module.c
- * This file is auto-generated with f2py (version:#f2py_version#).
- * f2py is a Fortran to Python Interface Generator (FPIG), Second Edition,
- * written by Pearu Peterson <pearu@cens.ioc.ee>.
- * Generation date: """ + time.asctime(time.gmtime(generationtime)) + """
- * Do not edit this file directly unless you know what you are doing!!!
- */
- #ifdef __cplusplus
- extern \"C\" {
- #endif
- """ + gentitle("See f2py2e/cfuncs.py: includes") + """
- #includes#
- #includes0#
- """ + gentitle("See f2py2e/rules.py: mod_rules['modulebody']") + """
- static PyObject *#modulename#_error;
- static PyObject *#modulename#_module;
- """ + gentitle("See f2py2e/cfuncs.py: typedefs") + """
- #typedefs#
- """ + gentitle("See f2py2e/cfuncs.py: typedefs_generated") + """
- #typedefs_generated#
- """ + gentitle("See f2py2e/cfuncs.py: cppmacros") + """
- #cppmacros#
- """ + gentitle("See f2py2e/cfuncs.py: cfuncs") + """
- #cfuncs#
- """ + gentitle("See f2py2e/cfuncs.py: userincludes") + """
- #userincludes#
- """ + gentitle("See f2py2e/capi_rules.py: usercode") + """
- #usercode#
- /* See f2py2e/rules.py */
- #externroutines#
- """ + gentitle("See f2py2e/capi_rules.py: usercode1") + """
- #usercode1#
- """ + gentitle("See f2py2e/cb_rules.py: buildcallback") + """
- #callbacks#
- """ + gentitle("See f2py2e/rules.py: buildapi") + """
- #body#
- """ + gentitle("See f2py2e/f90mod_rules.py: buildhooks") + """
- #f90modhooks#
- """ + gentitle("See f2py2e/rules.py: module_rules['modulebody']") + """
- """ + gentitle("See f2py2e/common_rules.py: buildhooks") + """
- #commonhooks#
- """ + gentitle("See f2py2e/rules.py") + """
- static FortranDataDef f2py_routine_defs[] = {
- #routine_defs#
- \t{NULL}
- };
- static PyMethodDef f2py_module_methods[] = {
- #pymethoddef#
- \t{NULL,NULL}
- };
- static struct PyModuleDef moduledef = {
- \tPyModuleDef_HEAD_INIT,
- \t"#modulename#",
- \tNULL,
- \t-1,
- \tf2py_module_methods,
- \tNULL,
- \tNULL,
- \tNULL,
- \tNULL
- };
- PyMODINIT_FUNC PyInit_#modulename#(void) {
- \tint i;
- \tPyObject *m,*d, *s, *tmp;
- \tm = #modulename#_module = PyModule_Create(&moduledef);
- \tPy_SET_TYPE(&PyFortran_Type, &PyType_Type);
- \timport_array();
- \tif (PyErr_Occurred())
- \t\t{PyErr_SetString(PyExc_ImportError, \"can't initialize module #modulename# (failed to import numpy)\"); return m;}
- \td = PyModule_GetDict(m);
- \ts = PyUnicode_FromString(\"#f2py_version#\");
- \tPyDict_SetItemString(d, \"__version__\", s);
- \tPy_DECREF(s);
- \ts = PyUnicode_FromString(
- \t\t\"This module '#modulename#' is auto-generated with f2py (version:#f2py_version#).\\nFunctions:\\n\"\n#docs#\".\");
- \tPyDict_SetItemString(d, \"__doc__\", s);
- \tPy_DECREF(s);
- \ts = PyUnicode_FromString(\"""" + numpy_version + """\");
- \tPyDict_SetItemString(d, \"__f2py_numpy_version__\", s);
- \tPy_DECREF(s);
- \t#modulename#_error = PyErr_NewException (\"#modulename#.error\", NULL, NULL);
- \t/*
- \t * Store the error object inside the dict, so that it could get deallocated.
- \t * (in practice, this is a module, so it likely will not and cannot.)
- \t */
- \tPyDict_SetItemString(d, \"_#modulename#_error\", #modulename#_error);
- \tPy_DECREF(#modulename#_error);
- \tfor(i=0;f2py_routine_defs[i].name!=NULL;i++) {
- \t\ttmp = PyFortranObject_NewAsAttr(&f2py_routine_defs[i]);
- \t\tPyDict_SetItemString(d, f2py_routine_defs[i].name, tmp);
- \t\tPy_DECREF(tmp);
- \t}
- #initf2pywraphooks#
- #initf90modhooks#
- #initcommonhooks#
- #interface_usercode#
- #ifdef F2PY_REPORT_ATEXIT
- \tif (! PyErr_Occurred())
- \t\ton_exit(f2py_report_on_exit,(void*)\"#modulename#\");
- #endif
- \treturn m;
- }
- #ifdef __cplusplus
- }
- #endif
- """,
- 'separatorsfor': {'latexdoc': '\n\n',
- 'restdoc': '\n\n'},
- 'latexdoc': ['\\section{Module \\texttt{#texmodulename#}}\n',
- '#modnote#\n',
- '#latexdoc#'],
- 'restdoc': ['Module #modulename#\n' + '=' * 80,
- '\n#restdoc#']
- }
- defmod_rules = [
- {'body': '/*eof body*/',
- 'method': '/*eof method*/',
- 'externroutines': '/*eof externroutines*/',
- 'routine_defs': '/*eof routine_defs*/',
- 'initf90modhooks': '/*eof initf90modhooks*/',
- 'initf2pywraphooks': '/*eof initf2pywraphooks*/',
- 'initcommonhooks': '/*eof initcommonhooks*/',
- 'latexdoc': '',
- 'restdoc': '',
- 'modnote': {hasnote: '#note#', l_not(hasnote): ''},
- }
- ]
- routine_rules = {
- 'separatorsfor': sepdict,
- 'body': """
- #begintitle#
- static char doc_#apiname#[] = \"\\\n#docreturn##name#(#docsignatureshort#)\\n\\nWrapper for ``#name#``.\\\n\\n#docstrsigns#\";
- /* #declfortranroutine# */
- static PyObject *#apiname#(const PyObject *capi_self,
- PyObject *capi_args,
- PyObject *capi_keywds,
- #functype# (*f2py_func)(#callprotoargument#)) {
- PyObject * volatile capi_buildvalue = NULL;
- volatile int f2py_success = 1;
- #decl#
- static char *capi_kwlist[] = {#kwlist##kwlistopt##kwlistxa#NULL};
- #usercode#
- #routdebugenter#
- #ifdef F2PY_REPORT_ATEXIT
- f2py_start_clock();
- #endif
- if (!PyArg_ParseTupleAndKeywords(capi_args,capi_keywds,\\
- \"#argformat#|#keyformat##xaformat#:#pyname#\",\\
- capi_kwlist#args_capi##keys_capi##keys_xa#))\n return NULL;
- #frompyobj#
- /*end of frompyobj*/
- #ifdef F2PY_REPORT_ATEXIT
- f2py_start_call_clock();
- #endif
- #callfortranroutine#
- if (PyErr_Occurred())
- f2py_success = 0;
- #ifdef F2PY_REPORT_ATEXIT
- f2py_stop_call_clock();
- #endif
- /*end of callfortranroutine*/
- if (f2py_success) {
- #pyobjfrom#
- /*end of pyobjfrom*/
- CFUNCSMESS(\"Building return value.\\n\");
- capi_buildvalue = Py_BuildValue(\"#returnformat#\"#return#);
- /*closepyobjfrom*/
- #closepyobjfrom#
- } /*if (f2py_success) after callfortranroutine*/
- /*cleanupfrompyobj*/
- #cleanupfrompyobj#
- if (capi_buildvalue == NULL) {
- #routdebugfailure#
- } else {
- #routdebugleave#
- }
- CFUNCSMESS(\"Freeing memory.\\n\");
- #freemem#
- #ifdef F2PY_REPORT_ATEXIT
- f2py_stop_clock();
- #endif
- return capi_buildvalue;
- }
- #endtitle#
- """,
- 'routine_defs': '#routine_def#',
- 'initf2pywraphooks': '#initf2pywraphook#',
- 'externroutines': '#declfortranroutine#',
- 'doc': '#docreturn##name#(#docsignature#)',
- 'docshort': '#docreturn##name#(#docsignatureshort#)',
- 'docs': '"\t#docreturn##name#(#docsignature#)\\n"\n',
- 'need': ['arrayobject.h', 'CFUNCSMESS', 'MINMAX'],
- 'cppmacros': {debugcapi: '#define DEBUGCFUNCS'},
- 'latexdoc': ['\\subsection{Wrapper function \\texttt{#texname#}}\n',
- """
- \\noindent{{}\\verb@#docreturn##name#@{}}\\texttt{(#latexdocsignatureshort#)}
- #routnote#
- #latexdocstrsigns#
- """],
- 'restdoc': ['Wrapped function ``#name#``\n' + '-' * 80,
- ]
- }
- ################## Rules for C/API function ##############
- rout_rules = [
- { # Init
- 'separatorsfor': {'callfortranroutine': '\n', 'routdebugenter': '\n', 'decl': '\n',
- 'routdebugleave': '\n', 'routdebugfailure': '\n',
- 'setjmpbuf': ' || ',
- 'docstrreq': '\n', 'docstropt': '\n', 'docstrout': '\n',
- 'docstrcbs': '\n', 'docstrsigns': '\\n"\n"',
- 'latexdocstrsigns': '\n',
- 'latexdocstrreq': '\n', 'latexdocstropt': '\n',
- 'latexdocstrout': '\n', 'latexdocstrcbs': '\n',
- },
- 'kwlist': '', 'kwlistopt': '', 'callfortran': '', 'callfortranappend': '',
- 'docsign': '', 'docsignopt': '', 'decl': '/*decl*/',
- 'freemem': '/*freemem*/',
- 'docsignshort': '', 'docsignoptshort': '',
- 'docstrsigns': '', 'latexdocstrsigns': '',
- 'docstrreq': '\\nParameters\\n----------',
- 'docstropt': '\\nOther Parameters\\n----------------',
- 'docstrout': '\\nReturns\\n-------',
- 'docstrcbs': '\\nNotes\\n-----\\nCall-back functions::\\n',
- 'latexdocstrreq': '\\noindent Required arguments:',
- 'latexdocstropt': '\\noindent Optional arguments:',
- 'latexdocstrout': '\\noindent Return objects:',
- 'latexdocstrcbs': '\\noindent Call-back functions:',
- 'args_capi': '', 'keys_capi': '', 'functype': '',
- 'frompyobj': '/*frompyobj*/',
- # this list will be reversed
- 'cleanupfrompyobj': ['/*end of cleanupfrompyobj*/'],
- 'pyobjfrom': '/*pyobjfrom*/',
- # this list will be reversed
- 'closepyobjfrom': ['/*end of closepyobjfrom*/'],
- 'topyarr': '/*topyarr*/', 'routdebugleave': '/*routdebugleave*/',
- 'routdebugenter': '/*routdebugenter*/',
- 'routdebugfailure': '/*routdebugfailure*/',
- 'callfortranroutine': '/*callfortranroutine*/',
- 'argformat': '', 'keyformat': '', 'need_cfuncs': '',
- 'docreturn': '', 'return': '', 'returnformat': '', 'rformat': '',
- 'kwlistxa': '', 'keys_xa': '', 'xaformat': '', 'docsignxa': '', 'docsignxashort': '',
- 'initf2pywraphook': '',
- 'routnote': {hasnote: '--- #note#', l_not(hasnote): ''},
- }, {
- 'apiname': 'f2py_rout_#modulename#_#name#',
- 'pyname': '#modulename#.#name#',
- 'decl': '',
- '_check': l_not(ismoduleroutine)
- }, {
- 'apiname': 'f2py_rout_#modulename#_#f90modulename#_#name#',
- 'pyname': '#modulename#.#f90modulename#.#name#',
- 'decl': '',
- '_check': ismoduleroutine
- }, { # Subroutine
- 'functype': 'void',
- 'declfortranroutine': {l_and(l_not(l_or(ismoduleroutine, isintent_c)), l_not(isdummyroutine)): 'extern void #F_FUNC#(#fortranname#,#FORTRANNAME#)(#callprotoargument#);',
- l_and(l_not(ismoduleroutine), isintent_c, l_not(isdummyroutine)): 'extern void #fortranname#(#callprotoargument#);',
- ismoduleroutine: '',
- isdummyroutine: ''
- },
- 'routine_def': {l_not(l_or(ismoduleroutine, isintent_c, isdummyroutine)): '\t{\"#name#\",-1,{{-1}},0,(char *)#F_FUNC#(#fortranname#,#FORTRANNAME#),(f2py_init_func)#apiname#,doc_#apiname#},',
- l_and(l_not(ismoduleroutine), isintent_c, l_not(isdummyroutine)): '\t{\"#name#\",-1,{{-1}},0,(char *)#fortranname#,(f2py_init_func)#apiname#,doc_#apiname#},',
- l_and(l_not(ismoduleroutine), isdummyroutine): '\t{\"#name#\",-1,{{-1}},0,NULL,(f2py_init_func)#apiname#,doc_#apiname#},',
- },
- 'need': {l_and(l_not(l_or(ismoduleroutine, isintent_c)), l_not(isdummyroutine)): 'F_FUNC'},
- 'callfortranroutine': [
- {debugcapi: [
- """\tfprintf(stderr,\"debug-capi:Fortran subroutine `#fortranname#(#callfortran#)\'\\n\");"""]},
- {hasexternals: """\
- \t\tif (#setjmpbuf#) {
- \t\t\tf2py_success = 0;
- \t\t} else {"""},
- {isthreadsafe: '\t\t\tPy_BEGIN_ALLOW_THREADS'},
- {hascallstatement: '''\t\t\t\t#callstatement#;
- \t\t\t\t/*(*f2py_func)(#callfortran#);*/'''},
- {l_not(l_or(hascallstatement, isdummyroutine))
- : '\t\t\t\t(*f2py_func)(#callfortran#);'},
- {isthreadsafe: '\t\t\tPy_END_ALLOW_THREADS'},
- {hasexternals: """\t\t}"""}
- ],
- '_check': l_and(issubroutine, l_not(issubroutine_wrap)),
- }, { # Wrapped function
- 'functype': 'void',
- 'declfortranroutine': {l_not(l_or(ismoduleroutine, isdummyroutine)): 'extern void #F_WRAPPEDFUNC#(#name_lower#,#NAME#)(#callprotoargument#);',
- isdummyroutine: '',
- },
- 'routine_def': {l_not(l_or(ismoduleroutine, isdummyroutine)): '\t{\"#name#\",-1,{{-1}},0,(char *)#F_WRAPPEDFUNC#(#name_lower#,#NAME#),(f2py_init_func)#apiname#,doc_#apiname#},',
- isdummyroutine: '\t{\"#name#\",-1,{{-1}},0,NULL,(f2py_init_func)#apiname#,doc_#apiname#},',
- },
- 'initf2pywraphook': {l_not(l_or(ismoduleroutine, isdummyroutine)): '''
- {
- extern #ctype# #F_FUNC#(#name_lower#,#NAME#)(void);
- PyObject* o = PyDict_GetItemString(d,"#name#");
- tmp = F2PyCapsule_FromVoidPtr((void*)#F_FUNC#(#name_lower#,#NAME#),NULL);
- PyObject_SetAttrString(o,"_cpointer", tmp);
- Py_DECREF(tmp);
- s = PyUnicode_FromString("#name#");
- PyObject_SetAttrString(o,"__name__", s);
- Py_DECREF(s);
- }
- '''},
- 'need': {l_not(l_or(ismoduleroutine, isdummyroutine)): ['F_WRAPPEDFUNC', 'F_FUNC']},
- 'callfortranroutine': [
- {debugcapi: [
- """\tfprintf(stderr,\"debug-capi:Fortran subroutine `f2pywrap#name_lower#(#callfortran#)\'\\n\");"""]},
- {hasexternals: """\
- \tif (#setjmpbuf#) {
- \t\tf2py_success = 0;
- \t} else {"""},
- {isthreadsafe: '\tPy_BEGIN_ALLOW_THREADS'},
- {l_not(l_or(hascallstatement, isdummyroutine))
- : '\t(*f2py_func)(#callfortran#);'},
- {hascallstatement:
- '\t#callstatement#;\n\t/*(*f2py_func)(#callfortran#);*/'},
- {isthreadsafe: '\tPy_END_ALLOW_THREADS'},
- {hasexternals: '\t}'}
- ],
- '_check': isfunction_wrap,
- }, { # Wrapped subroutine
- 'functype': 'void',
- 'declfortranroutine': {l_not(l_or(ismoduleroutine, isdummyroutine)): 'extern void #F_WRAPPEDFUNC#(#name_lower#,#NAME#)(#callprotoargument#);',
- isdummyroutine: '',
- },
- 'routine_def': {l_not(l_or(ismoduleroutine, isdummyroutine)): '\t{\"#name#\",-1,{{-1}},0,(char *)#F_WRAPPEDFUNC#(#name_lower#,#NAME#),(f2py_init_func)#apiname#,doc_#apiname#},',
- isdummyroutine: '\t{\"#name#\",-1,{{-1}},0,NULL,(f2py_init_func)#apiname#,doc_#apiname#},',
- },
- 'initf2pywraphook': {l_not(l_or(ismoduleroutine, isdummyroutine)): '''
- {
- extern void #F_FUNC#(#name_lower#,#NAME#)(void);
- PyObject* o = PyDict_GetItemString(d,"#name#");
- tmp = F2PyCapsule_FromVoidPtr((void*)#F_FUNC#(#name_lower#,#NAME#),NULL);
- PyObject_SetAttrString(o,"_cpointer", tmp);
- Py_DECREF(tmp);
- s = PyUnicode_FromString("#name#");
- PyObject_SetAttrString(o,"__name__", s);
- Py_DECREF(s);
- }
- '''},
- 'need': {l_not(l_or(ismoduleroutine, isdummyroutine)): ['F_WRAPPEDFUNC', 'F_FUNC']},
- 'callfortranroutine': [
- {debugcapi: [
- """\tfprintf(stderr,\"debug-capi:Fortran subroutine `f2pywrap#name_lower#(#callfortran#)\'\\n\");"""]},
- {hasexternals: """\
- \tif (#setjmpbuf#) {
- \t\tf2py_success = 0;
- \t} else {"""},
- {isthreadsafe: '\tPy_BEGIN_ALLOW_THREADS'},
- {l_not(l_or(hascallstatement, isdummyroutine))
- : '\t(*f2py_func)(#callfortran#);'},
- {hascallstatement:
- '\t#callstatement#;\n\t/*(*f2py_func)(#callfortran#);*/'},
- {isthreadsafe: '\tPy_END_ALLOW_THREADS'},
- {hasexternals: '\t}'}
- ],
- '_check': issubroutine_wrap,
- }, { # Function
- 'functype': '#ctype#',
- 'docreturn': {l_not(isintent_hide): '#rname#,'},
- 'docstrout': '#pydocsignout#',
- 'latexdocstrout': ['\\item[]{{}\\verb@#pydocsignout#@{}}',
- {hasresultnote: '--- #resultnote#'}],
- 'callfortranroutine': [{l_and(debugcapi, isstringfunction): """\
- #ifdef USESCOMPAQFORTRAN
- \tfprintf(stderr,\"debug-capi:Fortran function #ctype# #fortranname#(#callcompaqfortran#)\\n\");
- #else
- \tfprintf(stderr,\"debug-capi:Fortran function #ctype# #fortranname#(#callfortran#)\\n\");
- #endif
- """},
- {l_and(debugcapi, l_not(isstringfunction)): """\
- \tfprintf(stderr,\"debug-capi:Fortran function #ctype# #fortranname#(#callfortran#)\\n\");
- """}
- ],
- '_check': l_and(isfunction, l_not(isfunction_wrap))
- }, { # Scalar function
- 'declfortranroutine': {l_and(l_not(l_or(ismoduleroutine, isintent_c)), l_not(isdummyroutine)): 'extern #ctype# #F_FUNC#(#fortranname#,#FORTRANNAME#)(#callprotoargument#);',
- l_and(l_not(ismoduleroutine), isintent_c, l_not(isdummyroutine)): 'extern #ctype# #fortranname#(#callprotoargument#);',
- isdummyroutine: ''
- },
- 'routine_def': {l_and(l_not(l_or(ismoduleroutine, isintent_c)), l_not(isdummyroutine)): '\t{\"#name#\",-1,{{-1}},0,(char *)#F_FUNC#(#fortranname#,#FORTRANNAME#),(f2py_init_func)#apiname#,doc_#apiname#},',
- l_and(l_not(ismoduleroutine), isintent_c, l_not(isdummyroutine)): '\t{\"#name#\",-1,{{-1}},0,(char *)#fortranname#,(f2py_init_func)#apiname#,doc_#apiname#},',
- isdummyroutine: '\t{\"#name#\",-1,{{-1}},0,NULL,(f2py_init_func)#apiname#,doc_#apiname#},',
- },
- 'decl': [{iscomplexfunction_warn: '\t#ctype# #name#_return_value={0,0};',
- l_not(iscomplexfunction): '\t#ctype# #name#_return_value=0;'},
- {iscomplexfunction:
- '\tPyObject *#name#_return_value_capi = Py_None;'}
- ],
- 'callfortranroutine': [
- {hasexternals: """\
- \tif (#setjmpbuf#) {
- \t\tf2py_success = 0;
- \t} else {"""},
- {isthreadsafe: '\tPy_BEGIN_ALLOW_THREADS'},
- {hascallstatement: '''\t#callstatement#;
- /*\t#name#_return_value = (*f2py_func)(#callfortran#);*/
- '''},
- {l_not(l_or(hascallstatement, isdummyroutine))
- : '\t#name#_return_value = (*f2py_func)(#callfortran#);'},
- {isthreadsafe: '\tPy_END_ALLOW_THREADS'},
- {hasexternals: '\t}'},
- {l_and(debugcapi, iscomplexfunction)
- : '\tfprintf(stderr,"#routdebugshowvalue#\\n",#name#_return_value.r,#name#_return_value.i);'},
- {l_and(debugcapi, l_not(iscomplexfunction)): '\tfprintf(stderr,"#routdebugshowvalue#\\n",#name#_return_value);'}],
- 'pyobjfrom': {iscomplexfunction: '\t#name#_return_value_capi = pyobj_from_#ctype#1(#name#_return_value);'},
- 'need': [{l_not(isdummyroutine): 'F_FUNC'},
- {iscomplexfunction: 'pyobj_from_#ctype#1'},
- {islong_longfunction: 'long_long'},
- {islong_doublefunction: 'long_double'}],
- 'returnformat': {l_not(isintent_hide): '#rformat#'},
- 'return': {iscomplexfunction: ',#name#_return_value_capi',
- l_not(l_or(iscomplexfunction, isintent_hide)): ',#name#_return_value'},
- '_check': l_and(isfunction, l_not(isstringfunction), l_not(isfunction_wrap))
- }, { # String function # in use for --no-wrap
- 'declfortranroutine': 'extern void #F_FUNC#(#fortranname#,#FORTRANNAME#)(#callprotoargument#);',
- 'routine_def': {l_not(l_or(ismoduleroutine, isintent_c)):
- '\t{\"#name#\",-1,{{-1}},0,(char *)#F_FUNC#(#fortranname#,#FORTRANNAME#),(f2py_init_func)#apiname#,doc_#apiname#},',
- l_and(l_not(ismoduleroutine), isintent_c):
- '\t{\"#name#\",-1,{{-1}},0,(char *)#fortranname#,(f2py_init_func)#apiname#,doc_#apiname#},'
- },
- 'decl': ['\t#ctype# #name#_return_value = NULL;',
- '\tint #name#_return_value_len = 0;'],
- 'callfortran':'#name#_return_value,#name#_return_value_len,',
- 'callfortranroutine':['\t#name#_return_value_len = #rlength#;',
- '\tif ((#name#_return_value = (string)malloc(sizeof(char)*(#name#_return_value_len+1))) == NULL) {',
- '\t\tPyErr_SetString(PyExc_MemoryError, \"out of memory\");',
- '\t\tf2py_success = 0;',
- '\t} else {',
- "\t\t(#name#_return_value)[#name#_return_value_len] = '\\0';",
- '\t}',
- '\tif (f2py_success) {',
- {hasexternals: """\
- \t\tif (#setjmpbuf#) {
- \t\t\tf2py_success = 0;
- \t\t} else {"""},
- {isthreadsafe: '\t\tPy_BEGIN_ALLOW_THREADS'},
- """\
- #ifdef USESCOMPAQFORTRAN
- \t\t(*f2py_func)(#callcompaqfortran#);
- #else
- \t\t(*f2py_func)(#callfortran#);
- #endif
- """,
- {isthreadsafe: '\t\tPy_END_ALLOW_THREADS'},
- {hasexternals: '\t\t}'},
- {debugcapi:
- '\t\tfprintf(stderr,"#routdebugshowvalue#\\n",#name#_return_value_len,#name#_return_value);'},
- '\t} /* if (f2py_success) after (string)malloc */',
- ],
- 'returnformat': '#rformat#',
- 'return': ',#name#_return_value',
- 'freemem': '\tSTRINGFREE(#name#_return_value);',
- 'need': ['F_FUNC', '#ctype#', 'STRINGFREE'],
- '_check':l_and(isstringfunction, l_not(isfunction_wrap)) # ???obsolete
- },
- { # Debugging
- 'routdebugenter': '\tfprintf(stderr,"debug-capi:Python C/API function #modulename#.#name#(#docsignature#)\\n");',
- 'routdebugleave': '\tfprintf(stderr,"debug-capi:Python C/API function #modulename#.#name#: successful.\\n");',
- 'routdebugfailure': '\tfprintf(stderr,"debug-capi:Python C/API function #modulename#.#name#: failure.\\n");',
- '_check': debugcapi
- }
- ]
- ################ Rules for arguments ##################
- typedef_need_dict = {islong_long: 'long_long',
- islong_double: 'long_double',
- islong_complex: 'complex_long_double',
- isunsigned_char: 'unsigned_char',
- isunsigned_short: 'unsigned_short',
- isunsigned: 'unsigned',
- isunsigned_long_long: 'unsigned_long_long',
- isunsigned_chararray: 'unsigned_char',
- isunsigned_shortarray: 'unsigned_short',
- isunsigned_long_longarray: 'unsigned_long_long',
- issigned_long_longarray: 'long_long',
- }
- aux_rules = [
- {
- 'separatorsfor': sepdict
- },
- { # Common
- 'frompyobj': ['\t/* Processing auxiliary variable #varname# */',
- {debugcapi: '\tfprintf(stderr,"#vardebuginfo#\\n");'}, ],
- 'cleanupfrompyobj': '\t/* End of cleaning variable #varname# */',
- 'need': typedef_need_dict,
- },
- # Scalars (not complex)
- { # Common
- 'decl': '\t#ctype# #varname# = 0;',
- 'need': {hasinitvalue: 'math.h'},
- 'frompyobj': {hasinitvalue: '\t#varname# = #init#;'},
- '_check': l_and(isscalar, l_not(iscomplex)),
- },
- {
- 'return': ',#varname#',
- 'docstrout': '#pydocsignout#',
- 'docreturn': '#outvarname#,',
- 'returnformat': '#varrformat#',
- '_check': l_and(isscalar, l_not(iscomplex), isintent_out),
- },
- # Complex scalars
- { # Common
- 'decl': '\t#ctype# #varname#;',
- 'frompyobj': {hasinitvalue: '\t#varname#.r = #init.r#, #varname#.i = #init.i#;'},
- '_check': iscomplex
- },
- # String
- { # Common
- 'decl': ['\t#ctype# #varname# = NULL;',
- '\tint slen(#varname#);',
- ],
- 'need':['len..'],
- '_check':isstring
- },
- # Array
- { # Common
- 'decl': ['\t#ctype# *#varname# = NULL;',
- '\tnpy_intp #varname#_Dims[#rank#] = {#rank*[-1]#};',
- '\tconst int #varname#_Rank = #rank#;',
- ],
- 'need':['len..', {hasinitvalue: 'forcomb'}, {hasinitvalue: 'CFUNCSMESS'}],
- '_check': isarray
- },
- # Scalararray
- { # Common
- '_check': l_and(isarray, l_not(iscomplexarray))
- }, { # Not hidden
- '_check': l_and(isarray, l_not(iscomplexarray), isintent_nothide)
- },
- # Integer*1 array
- {'need': '#ctype#',
- '_check': isint1array,
- '_depend': ''
- },
- # Integer*-1 array
- {'need': '#ctype#',
- '_check': isunsigned_chararray,
- '_depend': ''
- },
- # Integer*-2 array
- {'need': '#ctype#',
- '_check': isunsigned_shortarray,
- '_depend': ''
- },
- # Integer*-8 array
- {'need': '#ctype#',
- '_check': isunsigned_long_longarray,
- '_depend': ''
- },
- # Complexarray
- {'need': '#ctype#',
- '_check': iscomplexarray,
- '_depend': ''
- },
- # Stringarray
- {
- 'callfortranappend': {isarrayofstrings: 'flen(#varname#),'},
- 'need': 'string',
- '_check': isstringarray
- }
- ]
- arg_rules = [
- {
- 'separatorsfor': sepdict
- },
- { # Common
- 'frompyobj': ['\t/* Processing variable #varname# */',
- {debugcapi: '\tfprintf(stderr,"#vardebuginfo#\\n");'}, ],
- 'cleanupfrompyobj': '\t/* End of cleaning variable #varname# */',
- '_depend': '',
- 'need': typedef_need_dict,
- },
- # Doc signatures
- {
- 'docstropt': {l_and(isoptional, isintent_nothide): '#pydocsign#'},
- 'docstrreq': {l_and(isrequired, isintent_nothide): '#pydocsign#'},
- 'docstrout': {isintent_out: '#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.'}]},
- 'depend': ''
- },
- # Required/Optional arguments
- {
- 'kwlist': '"#varname#",',
- 'docsign': '#varname#,',
- '_check': l_and(isintent_nothide, l_not(isoptional))
- },
- {
- 'kwlistopt': '"#varname#",',
- 'docsignopt': '#varname#=#showinit#,',
- 'docsignoptshort': '#varname#,',
- '_check': l_and(isintent_nothide, isoptional)
- },
- # Docstring/BuildValue
- {
- 'docreturn': '#outvarname#,',
- 'returnformat': '#varrformat#',
- '_check': isintent_out
- },
- # Externals (call-back functions)
- { # Common
- 'docsignxa': {isintent_nothide: '#varname#_extra_args=(),'},
- 'docsignxashort': {isintent_nothide: '#varname#_extra_args,'},
- 'docstropt': {isintent_nothide: '#varname#_extra_args : input tuple, optional\\n Default: ()'},
- 'docstrcbs': '#cbdocstr#',
- 'latexdocstrcbs': '\\item[] #cblatexdocstr#',
- 'latexdocstropt': {isintent_nothide: '\\item[]{{}\\verb@#varname#_extra_args := () input tuple@{}} --- Extra arguments for call-back function {{}\\verb@#varname#@{}}.'},
- 'decl': [' #cbname#_t #varname#_cb = { Py_None, NULL, 0 };',
- ' #cbname#_t *#varname#_cb_ptr = &#varname#_cb;',
- ' PyTupleObject *#varname#_xa_capi = NULL;',
- {l_not(isintent_callback):
- ' #cbname#_typedef #varname#_cptr;'}
- ],
- 'kwlistxa': {isintent_nothide: '"#varname#_extra_args",'},
- 'argformat': {isrequired: 'O'},
- 'keyformat': {isoptional: 'O'},
- 'xaformat': {isintent_nothide: 'O!'},
- 'args_capi': {isrequired: ',&#varname#_cb.capi'},
- 'keys_capi': {isoptional: ',&#varname#_cb.capi'},
- 'keys_xa': ',&PyTuple_Type,&#varname#_xa_capi',
- 'setjmpbuf': '(setjmp(#varname#_cb.jmpbuf))',
- 'callfortran': {l_not(isintent_callback): '#varname#_cptr,'},
- 'need': ['#cbname#', 'setjmp.h'],
- '_check':isexternal
- },
- {
- 'frompyobj': [{l_not(isintent_callback): """\
- if(F2PyCapsule_Check(#varname#_cb.capi)) {
- #varname#_cptr = F2PyCapsule_AsVoidPtr(#varname#_cb.capi);
- } else {
- #varname#_cptr = #cbname#;
- }
- """}, {isintent_callback: """\
- if (#varname#_cb.capi==Py_None) {
- #varname#_cb.capi = PyObject_GetAttrString(#modulename#_module,\"#varname#\");
- if (#varname#_cb.capi) {
- if (#varname#_xa_capi==NULL) {
- if (PyObject_HasAttrString(#modulename#_module,\"#varname#_extra_args\")) {
- PyObject* capi_tmp = PyObject_GetAttrString(#modulename#_module,\"#varname#_extra_args\");
- if (capi_tmp) {
- #varname#_xa_capi = (PyTupleObject *)PySequence_Tuple(capi_tmp);
- Py_DECREF(capi_tmp);
- }
- else {
- #varname#_xa_capi = (PyTupleObject *)Py_BuildValue(\"()\");
- }
- if (#varname#_xa_capi==NULL) {
- PyErr_SetString(#modulename#_error,\"Failed to convert #modulename#.#varname#_extra_args to tuple.\\n\");
- return NULL;
- }
- }
- }
- }
- if (#varname#_cb.capi==NULL) {
- PyErr_SetString(#modulename#_error,\"Callback #varname# not defined (as an argument or module #modulename# attribute).\\n\");
- return NULL;
- }
- }
- """},
- """\
- if (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#.\")) {
- """,
- {debugcapi: ["""\
- fprintf(stderr,\"debug-capi:Assuming %d arguments; at most #maxnofargs#(-#nofoptargs#) is expected.\\n\",#varname#_cb.nofargs);
- CFUNCSMESSPY(\"for #varname#=\",#varname#_cb.capi);""",
- {l_not(isintent_callback): """ fprintf(stderr,\"#vardebugshowvalue# (call-back in C).\\n\",#cbname#);"""}]},
- """\
- CFUNCSMESS(\"Saving callback variables for `#varname#`.\\n\");
- #varname#_cb_ptr = swap_active_#cbname#(#varname#_cb_ptr);""",
- ],
- 'cleanupfrompyobj':
- """\
- CFUNCSMESS(\"Restoring callback variables for `#varname#`.\\n\");
- #varname#_cb_ptr = swap_active_#cbname#(#varname#_cb_ptr);
- Py_DECREF(#varname#_cb.args_capi);
- }""",
- 'need': ['SWAP', 'create_cb_arglist'],
- '_check':isexternal,
- '_depend':''
- },
- # Scalars (not complex)
- { # Common
- 'decl': '\t#ctype# #varname# = 0;',
- 'pyobjfrom': {debugcapi: '\tfprintf(stderr,"#vardebugshowvalue#\\n",#varname#);'},
- 'callfortran': {isintent_c: '#varname#,', l_not(isintent_c): '&#varname#,'},
- 'return': {isintent_out: ',#varname#'},
- '_check': l_and(isscalar, l_not(iscomplex))
- }, {
- 'need': {hasinitvalue: 'math.h'},
- '_check': l_and(isscalar, l_not(iscomplex)),
- }, { # Not hidden
- 'decl': '\tPyObject *#varname#_capi = Py_None;',
- 'argformat': {isrequired: 'O'},
- 'keyformat': {isoptional: 'O'},
- 'args_capi': {isrequired: ',&#varname#_capi'},
- 'keys_capi': {isoptional: ',&#varname#_capi'},
- 'pyobjfrom': {isintent_inout: """\
- \tf2py_success = try_pyarr_from_#ctype#(#varname#_capi,&#varname#);
- \tif (f2py_success) {"""},
- 'closepyobjfrom': {isintent_inout: "\t} /*if (f2py_success) of #varname# pyobjfrom*/"},
- 'need': {isintent_inout: 'try_pyarr_from_#ctype#'},
- '_check': l_and(isscalar, l_not(iscomplex), isintent_nothide)
- }, {
- 'frompyobj': [
- # hasinitvalue...
- # if pyobj is None:
- # varname = init
- # else
- # from_pyobj(varname)
- #
- # isoptional and noinitvalue...
- # if pyobj is not None:
- # from_pyobj(varname)
- # else:
- # varname is uninitialized
- #
- # ...
- # from_pyobj(varname)
- #
- {hasinitvalue: '\tif (#varname#_capi == Py_None) #varname# = #init#; else',
- '_depend': ''},
- {l_and(isoptional, l_not(hasinitvalue)): '\tif (#varname#_capi != Py_None)',
- '_depend': ''},
- {l_not(islogical): '''\
- \t\tf2py_success = #ctype#_from_pyobj(&#varname#,#varname#_capi,"#pyname#() #nth# (#varname#) can\'t be converted to #ctype#");
- \tif (f2py_success) {'''},
- {islogical: '''\
- \t\t#varname# = (#ctype#)PyObject_IsTrue(#varname#_capi);
- \t\tf2py_success = 1;
- \tif (f2py_success) {'''},
- ],
- 'cleanupfrompyobj': '\t} /*if (f2py_success) of #varname#*/',
- 'need': {l_not(islogical): '#ctype#_from_pyobj'},
- '_check': l_and(isscalar, l_not(iscomplex), isintent_nothide),
- '_depend': ''
- }, { # Hidden
- 'frompyobj': {hasinitvalue: '\t#varname# = #init#;'},
- 'need': typedef_need_dict,
- '_check': l_and(isscalar, l_not(iscomplex), isintent_hide),
- '_depend': ''
- }, { # Common
- 'frompyobj': {debugcapi: '\tfprintf(stderr,"#vardebugshowvalue#\\n",#varname#);'},
- '_check': l_and(isscalar, l_not(iscomplex)),
- '_depend': ''
- },
- # Complex scalars
- { # Common
- 'decl': '\t#ctype# #varname#;',
- 'callfortran': {isintent_c: '#varname#,', l_not(isintent_c): '&#varname#,'},
- 'pyobjfrom': {debugcapi: '\tfprintf(stderr,"#vardebugshowvalue#\\n",#varname#.r,#varname#.i);'},
- 'return': {isintent_out: ',#varname#_capi'},
- '_check': iscomplex
- }, { # Not hidden
- 'decl': '\tPyObject *#varname#_capi = Py_None;',
- 'argformat': {isrequired: 'O'},
- 'keyformat': {isoptional: 'O'},
- 'args_capi': {isrequired: ',&#varname#_capi'},
- 'keys_capi': {isoptional: ',&#varname#_capi'},
- 'need': {isintent_inout: 'try_pyarr_from_#ctype#'},
- 'pyobjfrom': {isintent_inout: """\
- \t\tf2py_success = try_pyarr_from_#ctype#(#varname#_capi,&#varname#);
- \t\tif (f2py_success) {"""},
- 'closepyobjfrom': {isintent_inout: "\t\t} /*if (f2py_success) of #varname# pyobjfrom*/"},
- '_check': l_and(iscomplex, isintent_nothide)
- }, {
- 'frompyobj': [{hasinitvalue: '\tif (#varname#_capi==Py_None) {#varname#.r = #init.r#, #varname#.i = #init.i#;} else'},
- {l_and(isoptional, l_not(hasinitvalue))
- : '\tif (#varname#_capi != Py_None)'},
- '\t\tf2py_success = #ctype#_from_pyobj(&#varname#,#varname#_capi,"#pyname#() #nth# (#varname#) can\'t be converted to #ctype#");'
- '\n\tif (f2py_success) {'],
- 'cleanupfrompyobj': '\t} /*if (f2py_success) of #varname# frompyobj*/',
- 'need': ['#ctype#_from_pyobj'],
- '_check': l_and(iscomplex, isintent_nothide),
- '_depend': ''
- }, { # Hidden
- 'decl': {isintent_out: '\tPyObject *#varname#_capi = Py_None;'},
- '_check': l_and(iscomplex, isintent_hide)
- }, {
- 'frompyobj': {hasinitvalue: '\t#varname#.r = #init.r#, #varname#.i = #init.i#;'},
- '_check': l_and(iscomplex, isintent_hide),
- '_depend': ''
- }, { # Common
- 'pyobjfrom': {isintent_out: '\t#varname#_capi = pyobj_from_#ctype#1(#varname#);'},
- 'need': ['pyobj_from_#ctype#1'],
- '_check': iscomplex
- }, {
- 'frompyobj': {debugcapi: '\tfprintf(stderr,"#vardebugshowvalue#\\n",#varname#.r,#varname#.i);'},
- '_check': iscomplex,
- '_depend': ''
- },
- # String
- { # Common
- 'decl': ['\t#ctype# #varname# = NULL;',
- '\tint slen(#varname#);',
- '\tPyObject *#varname#_capi = Py_None;'],
- 'callfortran':'#varname#,',
- 'callfortranappend':'slen(#varname#),',
- 'pyobjfrom':{debugcapi: '\tfprintf(stderr,"#vardebugshowvalue#\\n",slen(#varname#),#varname#);'},
- 'return': {isintent_out: ',#varname#'},
- 'need': ['len..'], # 'STRINGFREE'],
- '_check':isstring
- }, { # Common
- 'frompyobj': """\
- \tslen(#varname#) = #length#;
- \tf2py_success = #ctype#_from_pyobj(&#varname#,&slen(#varname#),#init#,#varname#_capi,\"#ctype#_from_pyobj failed in converting #nth# `#varname#\' of #pyname# to C #ctype#\");
- \tif (f2py_success) {""",
- 'cleanupfrompyobj': """\
- \t\tSTRINGFREE(#varname#);
- \t} /*if (f2py_success) of #varname#*/""",
- 'need': ['#ctype#_from_pyobj', 'len..', 'STRINGFREE'],
- '_check':isstring,
- '_depend':''
- }, { # Not hidden
- 'argformat': {isrequired: 'O'},
- 'keyformat': {isoptional: 'O'},
- 'args_capi': {isrequired: ',&#varname#_capi'},
- 'keys_capi': {isoptional: ',&#varname#_capi'},
- 'pyobjfrom': {isintent_inout: '''\
- \tf2py_success = try_pyarr_from_#ctype#(#varname#_capi,#varname#);
- \tif (f2py_success) {'''},
- 'closepyobjfrom': {isintent_inout: '\t} /*if (f2py_success) of #varname# pyobjfrom*/'},
- 'need': {isintent_inout: 'try_pyarr_from_#ctype#'},
- '_check': l_and(isstring, isintent_nothide)
- }, { # Hidden
- '_check': l_and(isstring, isintent_hide)
- }, {
- 'frompyobj': {debugcapi: '\tfprintf(stderr,"#vardebugshowvalue#\\n",slen(#varname#),#varname#);'},
- '_check': isstring,
- '_depend': ''
- },
- # Array
- { # Common
- 'decl': ['\t#ctype# *#varname# = NULL;',
- '\tnpy_intp #varname#_Dims[#rank#] = {#rank*[-1]#};',
- '\tconst int #varname#_Rank = #rank#;',
- '\tPyArrayObject *capi_#varname#_tmp = NULL;',
- '\tint capi_#varname#_intent = 0;',
- ],
- 'callfortran':'#varname#,',
- 'return':{isintent_out: ',capi_#varname#_tmp'},
- 'need': 'len..',
- '_check': isarray
- }, { # intent(overwrite) array
- 'decl': '\tint capi_overwrite_#varname# = 1;',
- 'kwlistxa': '"overwrite_#varname#",',
- 'xaformat': 'i',
- 'keys_xa': ',&capi_overwrite_#varname#',
- 'docsignxa': 'overwrite_#varname#=1,',
- 'docsignxashort': 'overwrite_#varname#,',
- 'docstropt': 'overwrite_#varname# : input int, optional\\n Default: 1',
- '_check': l_and(isarray, isintent_overwrite),
- }, {
- 'frompyobj': '\tcapi_#varname#_intent |= (capi_overwrite_#varname#?0:F2PY_INTENT_COPY);',
- '_check': l_and(isarray, isintent_overwrite),
- '_depend': '',
- },
- { # intent(copy) array
- 'decl': '\tint capi_overwrite_#varname# = 0;',
- 'kwlistxa': '"overwrite_#varname#",',
- 'xaformat': 'i',
- 'keys_xa': ',&capi_overwrite_#varname#',
- 'docsignxa': 'overwrite_#varname#=0,',
- 'docsignxashort': 'overwrite_#varname#,',
- 'docstropt': 'overwrite_#varname# : input int, optional\\n Default: 0',
- '_check': l_and(isarray, isintent_copy),
- }, {
- 'frompyobj': '\tcapi_#varname#_intent |= (capi_overwrite_#varname#?0:F2PY_INTENT_COPY);',
- '_check': l_and(isarray, isintent_copy),
- '_depend': '',
- }, {
- 'need': [{hasinitvalue: 'forcomb'}, {hasinitvalue: 'CFUNCSMESS'}],
- '_check': isarray,
- '_depend': ''
- }, { # Not hidden
- 'decl': '\tPyObject *#varname#_capi = Py_None;',
- 'argformat': {isrequired: 'O'},
- 'keyformat': {isoptional: 'O'},
- 'args_capi': {isrequired: ',&#varname#_capi'},
- 'keys_capi': {isoptional: ',&#varname#_capi'},
- '_check': l_and(isarray, isintent_nothide)
- }, {
- 'frompyobj': ['\t#setdims#;',
- '\tcapi_#varname#_intent |= #intent#;',
- {isintent_hide:
- '\tcapi_#varname#_tmp = array_from_pyobj(#atype#,#varname#_Dims,#varname#_Rank,capi_#varname#_intent,Py_None);'},
- {isintent_nothide:
- '\tcapi_#varname#_tmp = array_from_pyobj(#atype#,#varname#_Dims,#varname#_Rank,capi_#varname#_intent,#varname#_capi);'},
- """\
- \tif (capi_#varname#_tmp == NULL) {
- \t\tPyObject *exc, *val, *tb;
- \t\tPyErr_Fetch(&exc, &val, &tb);
- \t\tPyErr_SetString(exc ? exc : #modulename#_error,\"failed in converting #nth# `#varname#\' of #pyname# to C/Fortran array\" );
- \t\tnpy_PyErr_ChainExceptionsCause(exc, val, tb);
- \t} else {
- \t\t#varname# = (#ctype# *)(PyArray_DATA(capi_#varname#_tmp));
- """,
- {hasinitvalue: [
- {isintent_nothide:
- '\tif (#varname#_capi == Py_None) {'},
- {isintent_hide: '\t{'},
- {iscomplexarray: '\t\t#ctype# capi_c;'},
- """\
- \t\tint *_i,capi_i=0;
- \t\tCFUNCSMESS(\"#name#: Initializing #varname#=#init#\\n\");
- \t\tif (initforcomb(PyArray_DIMS(capi_#varname#_tmp),PyArray_NDIM(capi_#varname#_tmp),1)) {
- \t\t\twhile ((_i = nextforcomb()))
- \t\t\t\t#varname#[capi_i++] = #init#; /* fortran way */
- \t\t} else {
- \t\t\tPyObject *exc, *val, *tb;
- \t\t\tPyErr_Fetch(&exc, &val, &tb);
- \t\t\tPyErr_SetString(exc ? exc : #modulename#_error,\"Initialization of #nth# #varname# failed (initforcomb).\");
- \t\t\tnpy_PyErr_ChainExceptionsCause(exc, val, tb);
- \t\t\tf2py_success = 0;
- \t\t}
- \t}
- \tif (f2py_success) {"""]},
- ],
- 'cleanupfrompyobj': [ # note that this list will be reversed
- '\t} /*if (capi_#varname#_tmp == NULL) ... else of #varname#*/',
- {l_not(l_or(isintent_out, isintent_hide)): """\
- \tif((PyObject *)capi_#varname#_tmp!=#varname#_capi) {
- \t\tPy_XDECREF(capi_#varname#_tmp); }"""},
- {l_and(isintent_hide, l_not(isintent_out))
- : """\t\tPy_XDECREF(capi_#varname#_tmp);"""},
- {hasinitvalue: '\t} /*if (f2py_success) of #varname# init*/'},
- ],
- '_check': isarray,
- '_depend': ''
- },
- # Scalararray
- { # Common
- '_check': l_and(isarray, l_not(iscomplexarray))
- }, { # Not hidden
- '_check': l_and(isarray, l_not(iscomplexarray), isintent_nothide)
- },
- # Integer*1 array
- {'need': '#ctype#',
- '_check': isint1array,
- '_depend': ''
- },
- # Integer*-1 array
- {'need': '#ctype#',
- '_check': isunsigned_chararray,
- '_depend': ''
- },
- # Integer*-2 array
- {'need': '#ctype#',
- '_check': isunsigned_shortarray,
- '_depend': ''
- },
- # Integer*-8 array
- {'need': '#ctype#',
- '_check': isunsigned_long_longarray,
- '_depend': ''
- },
- # Complexarray
- {'need': '#ctype#',
- '_check': iscomplexarray,
- '_depend': ''
- },
- # Stringarray
- {
- 'callfortranappend': {isarrayofstrings: 'flen(#varname#),'},
- 'need': 'string',
- '_check': isstringarray
- }
- ]
- ################# Rules for checking ###############
- check_rules = [
- {
- 'frompyobj': {debugcapi: '\tfprintf(stderr,\"debug-capi:Checking `#check#\'\\n\");'},
- 'need': 'len..'
- }, {
- 'frompyobj': '\tCHECKSCALAR(#check#,\"#check#\",\"#nth# #varname#\",\"#varshowvalue#\",#varname#) {',
- 'cleanupfrompyobj': '\t} /*CHECKSCALAR(#check#)*/',
- 'need': 'CHECKSCALAR',
- '_check': l_and(isscalar, l_not(iscomplex)),
- '_break': ''
- }, {
- 'frompyobj': '\tCHECKSTRING(#check#,\"#check#\",\"#nth# #varname#\",\"#varshowvalue#\",#varname#) {',
- 'cleanupfrompyobj': '\t} /*CHECKSTRING(#check#)*/',
- 'need': 'CHECKSTRING',
- '_check': isstring,
- '_break': ''
- }, {
- 'need': 'CHECKARRAY',
- 'frompyobj': '\tCHECKARRAY(#check#,\"#check#\",\"#nth# #varname#\") {',
- 'cleanupfrompyobj': '\t} /*CHECKARRAY(#check#)*/',
- '_check': isarray,
- '_break': ''
- }, {
- 'need': 'CHECKGENERIC',
- 'frompyobj': '\tCHECKGENERIC(#check#,\"#check#\",\"#nth# #varname#\") {',
- 'cleanupfrompyobj': '\t} /*CHECKGENERIC(#check#)*/',
- }
- ]
- ########## Applying the rules. No need to modify what follows #############
- #################### Build C/API module #######################
- def buildmodule(m, um):
- """
- Return
- """
- outmess('\tBuilding module "%s"...\n' % (m['name']))
- ret = {}
- mod_rules = defmod_rules[:]
- vrd = capi_maps.modsign2map(m)
- rd = dictappend({'f2py_version': f2py_version}, vrd)
- funcwrappers = []
- funcwrappers2 = [] # F90 codes
- for n in m['interfaced']:
- nb = None
- for bi in m['body']:
- if bi['block'] not in ['interface', 'abstract interface']:
- errmess('buildmodule: Expected interface block. Skipping.\n')
- continue
- for b in bi['body']:
- if b['name'] == n:
- nb = b
- break
- if not nb:
- errmess(
- 'buildmodule: Could not found the body of interfaced routine "%s". Skipping.\n' % (n))
- continue
- nb_list = [nb]
- if 'entry' in nb:
- for k, a in nb['entry'].items():
- nb1 = copy.deepcopy(nb)
- del nb1['entry']
- nb1['name'] = k
- nb1['args'] = a
- nb_list.append(nb1)
- for nb in nb_list:
- # requiresf90wrapper must be called before buildapi as it
- # rewrites assumed shape arrays as automatic arrays.
- isf90 = requiresf90wrapper(nb)
- api, wrap = buildapi(nb)
- if wrap:
- if isf90:
- funcwrappers2.append(wrap)
- else:
- funcwrappers.append(wrap)
- ar = applyrules(api, vrd)
- rd = dictappend(rd, ar)
- # Construct COMMON block support
- cr, wrap = common_rules.buildhooks(m)
- if wrap:
- funcwrappers.append(wrap)
- ar = applyrules(cr, vrd)
- rd = dictappend(rd, ar)
- # Construct F90 module support
- mr, wrap = f90mod_rules.buildhooks(m)
- if wrap:
- funcwrappers2.append(wrap)
- ar = applyrules(mr, vrd)
- rd = dictappend(rd, ar)
- for u in um:
- ar = use_rules.buildusevars(u, m['use'][u['name']])
- rd = dictappend(rd, ar)
- needs = cfuncs.get_needs()
- code = {}
- for n in needs.keys():
- code[n] = []
- for k in needs[n]:
- c = ''
- if k in cfuncs.includes0:
- c = cfuncs.includes0[k]
- elif k in cfuncs.includes:
- c = cfuncs.includes[k]
- elif k in cfuncs.userincludes:
- c = cfuncs.userincludes[k]
- elif k in cfuncs.typedefs:
- c = cfuncs.typedefs[k]
- elif k in cfuncs.typedefs_generated:
- c = cfuncs.typedefs_generated[k]
- elif k in cfuncs.cppmacros:
- c = cfuncs.cppmacros[k]
- elif k in cfuncs.cfuncs:
- c = cfuncs.cfuncs[k]
- elif k in cfuncs.callbacks:
- c = cfuncs.callbacks[k]
- elif k in cfuncs.f90modhooks:
- c = cfuncs.f90modhooks[k]
- elif k in cfuncs.commonhooks:
- c = cfuncs.commonhooks[k]
- else:
- errmess('buildmodule: unknown need %s.\n' % (repr(k)))
- continue
- code[n].append(c)
- mod_rules.append(code)
- for r in mod_rules:
- if ('_check' in r and r['_check'](m)) or ('_check' not in r):
- ar = applyrules(r, vrd, m)
- rd = dictappend(rd, ar)
- ar = applyrules(module_rules, rd)
- fn = os.path.join(options['buildpath'], vrd['coutput'])
- ret['csrc'] = fn
- with open(fn, 'w') as f:
- f.write(ar['modulebody'].replace('\t', 2 * ' '))
- outmess('\tWrote C/API module "%s" to file "%s"\n' % (m['name'], fn))
- if options['dorestdoc']:
- fn = os.path.join(
- options['buildpath'], vrd['modulename'] + 'module.rest')
- with open(fn, 'w') as f:
- f.write('.. -*- rest -*-\n')
- f.write('\n'.join(ar['restdoc']))
- outmess('\tReST Documentation is saved to file "%s/%smodule.rest"\n' %
- (options['buildpath'], vrd['modulename']))
- if options['dolatexdoc']:
- fn = os.path.join(
- options['buildpath'], vrd['modulename'] + 'module.tex')
- ret['ltx'] = fn
- with open(fn, 'w') as f:
- f.write(
- '%% This file is auto-generated with f2py (version:%s)\n' % (f2py_version))
- if 'shortlatex' not in options:
- f.write(
- '\\documentclass{article}\n\\usepackage{a4wide}\n\\begin{document}\n\\tableofcontents\n\n')
- f.write('\n'.join(ar['latexdoc']))
- if 'shortlatex' not in options:
- f.write('\\end{document}')
- outmess('\tDocumentation is saved to file "%s/%smodule.tex"\n' %
- (options['buildpath'], vrd['modulename']))
- if funcwrappers:
- wn = os.path.join(options['buildpath'], vrd['f2py_wrapper_output'])
- ret['fsrc'] = wn
- with open(wn, 'w') as f:
- f.write('C -*- fortran -*-\n')
- f.write(
- 'C This file is autogenerated with f2py (version:%s)\n' % (f2py_version))
- f.write(
- 'C It contains Fortran 77 wrappers to fortran functions.\n')
- lines = []
- for l in ('\n\n'.join(funcwrappers) + '\n').split('\n'):
- if 0 <= l.find('!') < 66:
- # don't split comment lines
- lines.append(l + '\n')
- elif l and l[0] == ' ':
- while len(l) >= 66:
- lines.append(l[:66] + '\n &')
- l = l[66:]
- lines.append(l + '\n')
- else:
- lines.append(l + '\n')
- lines = ''.join(lines).replace('\n &\n', '\n')
- f.write(lines)
- outmess('\tFortran 77 wrappers are saved to "%s"\n' % (wn))
- if funcwrappers2:
- wn = os.path.join(
- options['buildpath'], '%s-f2pywrappers2.f90' % (vrd['modulename']))
- ret['fsrc'] = wn
- with open(wn, 'w') as f:
- f.write('! -*- f90 -*-\n')
- f.write(
- '! This file is autogenerated with f2py (version:%s)\n' % (f2py_version))
- f.write(
- '! It contains Fortran 90 wrappers to fortran functions.\n')
- lines = []
- for l in ('\n\n'.join(funcwrappers2) + '\n').split('\n'):
- if 0 <= l.find('!') < 72:
- # don't split comment lines
- lines.append(l + '\n')
- elif len(l) > 72 and l[0] == ' ':
- lines.append(l[:72] + '&\n &')
- l = l[72:]
- while len(l) > 66:
- lines.append(l[:66] + '&\n &')
- l = l[66:]
- lines.append(l + '\n')
- else:
- lines.append(l + '\n')
- lines = ''.join(lines).replace('\n &\n', '\n')
- f.write(lines)
- outmess('\tFortran 90 wrappers are saved to "%s"\n' % (wn))
- return ret
- ################## Build C/API function #############
- stnd = {1: 'st', 2: 'nd', 3: 'rd', 4: 'th', 5: 'th',
- 6: 'th', 7: 'th', 8: 'th', 9: 'th', 0: 'th'}
- def buildapi(rout):
- rout, wrap = func2subr.assubr(rout)
- args, depargs = getargs2(rout)
- capi_maps.depargs = depargs
- var = rout['vars']
- if ismoduleroutine(rout):
- outmess('\t\t\tConstructing wrapper function "%s.%s"...\n' %
- (rout['modulename'], rout['name']))
- else:
- outmess('\t\tConstructing wrapper function "%s"...\n' % (rout['name']))
- # Routine
- vrd = capi_maps.routsign2map(rout)
- rd = dictappend({}, vrd)
- for r in rout_rules:
- if ('_check' in r and r['_check'](rout)) or ('_check' not in r):
- ar = applyrules(r, vrd, rout)
- rd = dictappend(rd, ar)
- # Args
- nth, nthk = 0, 0
- savevrd = {}
- for a in args:
- vrd = capi_maps.sign2map(a, var[a])
- if isintent_aux(var[a]):
- _rules = aux_rules
- else:
- _rules = arg_rules
- if not isintent_hide(var[a]):
- if not isoptional(var[a]):
- nth = nth + 1
- vrd['nth'] = repr(nth) + stnd[nth % 10] + ' argument'
- else:
- nthk = nthk + 1
- vrd['nth'] = repr(nthk) + stnd[nthk % 10] + ' keyword'
- else:
- vrd['nth'] = 'hidden'
- savevrd[a] = vrd
- for r in _rules:
- if '_depend' 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
- for a in depargs:
- if isintent_aux(var[a]):
- _rules = aux_rules
- else:
- _rules = arg_rules
- vrd = savevrd[a]
- for r in _rules:
- if '_depend' not 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 'check' in var[a]:
- for c in var[a]['check']:
- vrd['check'] = c
- ar = applyrules(check_rules, vrd, var[a])
- rd = dictappend(rd, ar)
- if isinstance(rd['cleanupfrompyobj'], list):
- rd['cleanupfrompyobj'].reverse()
- if isinstance(rd['closepyobjfrom'], list):
- rd['closepyobjfrom'].reverse()
- rd['docsignature'] = stripcomma(replace('#docsign##docsignopt##docsignxa#',
- {'docsign': rd['docsign'],
- 'docsignopt': rd['docsignopt'],
- 'docsignxa': rd['docsignxa']}))
- optargs = stripcomma(replace('#docsignopt##docsignxa#',
- {'docsignxa': rd['docsignxashort'],
- 'docsignopt': rd['docsignoptshort']}
- ))
- if optargs == '':
- rd['docsignatureshort'] = stripcomma(
- replace('#docsign#', {'docsign': rd['docsign']}))
- else:
- rd['docsignatureshort'] = replace('#docsign#[#docsignopt#]',
- {'docsign': rd['docsign'],
- 'docsignopt': optargs,
- })
- rd['latexdocsignatureshort'] = rd['docsignatureshort'].replace('_', '\\_')
- rd['latexdocsignatureshort'] = rd[
- 'latexdocsignatureshort'].replace(',', ', ')
- cfs = stripcomma(replace('#callfortran##callfortranappend#', {
- 'callfortran': rd['callfortran'], 'callfortranappend': rd['callfortranappend']}))
- if len(rd['callfortranappend']) > 1:
- rd['callcompaqfortran'] = stripcomma(replace('#callfortran# 0,#callfortranappend#', {
- 'callfortran': rd['callfortran'], 'callfortranappend': rd['callfortranappend']}))
- else:
- rd['callcompaqfortran'] = cfs
- rd['callfortran'] = cfs
- if isinstance(rd['docreturn'], list):
- rd['docreturn'] = stripcomma(
- replace('#docreturn#', {'docreturn': rd['docreturn']})) + ' = '
- 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}']
- ar = applyrules(routine_rules, rd)
- if ismoduleroutine(rout):
- outmess('\t\t\t %s\n' % (ar['docshort']))
- else:
- outmess('\t\t %s\n' % (ar['docshort']))
- return ar, wrap
- #################### EOF rules.py #######################
|