1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
|
__all__ = ['PythonCAPISubProgram']
import sys
from parser.api import TypeDecl, TypeStmt, Module
from wrapper_base import *
from py_wrap_type import *
class PythonCAPISubProgram(WrapperBase):
"""
Fortran subprogram hooks.
"""
header_template_f77 = '''\
#define %(name)s_f F_FUNC(%(name)s, %(NAME)s)
'''
extern_template_f77 = '''\
extern void %(name)s_f();
'''
objdecl_template_doc = '''\
static char %(cname)s__doc[] = "";
'''
module_method_template = '''\
{"%(pyname)s", (PyCFunction)%(cname)s, METH_VARARGS | METH_KEYWORDS, %(cname)s__doc},'''
capi_code_template = '''\
static PyObject* %(cname)s(PyObject *capi_self, PyObject *capi_args, PyObject *capi_keywds) {
PyObject * volatile capi_buildvalue = NULL;
volatile int f2py_success = 1;
%(decl_list)s
static char *capi_kwlist[] = {%(kw_clist+optkw_clist+extrakw_clist+["NULL"])s};
if (PyArg_ParseTupleAndKeywords(capi_args,capi_keywds,
"%(pyarg_format_elist)s",
%(["capi_kwlist"]+pyarg_obj_clist)s)) {
%(frompyobj_list)s
%(call_list)s
f2py_success = !PyErr_Occurred();
if (f2py_success) {
%(pyobjfrom_list)s
capi_buildvalue = Py_BuildValue("%(return_format_elist)s"
%(return_obj_clist)s);
%(clean_pyobjfrom_list)s
}
%(clean_call_list)s
%(clean_frompyobj_list)s
}
return capi_buildvalue;
}
'''
header_template_module = '''
#define %(name)s_f (*%(name)s_func_ptr)
#define %(init_func)s_f F_FUNC(%(init_func)s, %(INIT_FUNC)s)
'''
typedef_template_module = '''
typedef void (*%(name)s_functype)();
'''
extern_template_module = '''\
static %(name)s_functype %(name)s_func_ptr;
'''
objdecl_template_module = '''
'''
fortran_code_template_module = '''
subroutine %(init_func)s(init_func_c)
use %(mname)s
external init_func_c
call init_func_c(%(name)s)
end
'''
c_code_template_module = '''
static void %(init_func)s_c(%(name)s_functype func_ptr) {
%(name)s_func_ptr = func_ptr;
}
'''
module_init_template_module = '''
%(init_func)s_f(%(init_func)s_c);
'''
_defined = []
def __init__(self, parent, block):
WrapperBase.__init__(self)
self.name = name = pyname = block.name
self.cname = cname = '%s_%s' % (parent.cname,name)
if cname in self._defined:
return
self._defined.append(cname)
self.info('Generating interface for %s: %s' % (block.__class__, cname))
if pyname.startswith('f2pywrap_'):
pyname = pyname[9:]
self.pyname = pyname
self.header_template = ''
self.extern_template = ''
self.module_init_template = ''
self.typedef_template = ''
self.c_code_template = ''
self.objdecl_template = ''
self.fortran_code_template = ''
if isinstance(block.parent, Module):
self.mname = block.parent.name
self.init_func = '%s_init' % (name)
self.typedef_template += self.typedef_template_module
self.header_template += self.header_template_module
self.fortran_code_template += self.fortran_code_template_module
self.module_init_template += self.module_init_template_module
self.objdecl_template += self.objdecl_template_module
self.c_code_template += self.c_code_template_module
self.extern_template += self.extern_template_module
else:
self.extern_template += self.extern_template_f77
self.header_template += self.header_template_f77
self.objdecl_template += self.objdecl_template_doc
self.decl_list = []
self.kw_list = []
self.optkw_list = []
self.extrakw_list = []
self.pyarg_format_list = []
self.pyarg_obj_list = []
self.frompyobj_list = []
self.call_list = []
self.pyobjfrom_list = []
self.return_format_list = []
self.return_obj_list = []
self.buildvalue_list = []
self.clean_pyobjfrom_list = []
self.clean_call_list = []
self.clean_frompyobj_list = []
args_f = []
extra_args_f = []
argindex = -1
for argname in block.args:
argindex += 1
var = block.a.variables[argname]
assert var.is_scalar(),'array support not implemented: "%s"' % (var)
typedecl = var.get_typedecl()
PythonCAPIType(parent, typedecl)
ti = PyTypeInterface(typedecl)
if var.is_intent_in():
self.kw_list.append('"%s"' % (argname))
if isinstance(typedecl, TypeStmt):
if var.is_intent_in():
self.pyarg_format_list.append('O&')
self.pyarg_obj_list.append('\npyobj_to_%s_inplace, &%s' % (ti.ctype, argname))
else:
self.frompyobj_list.append('%s = (%s*)pyobj_from_%s(NULL);' % (argname,ti.otype,ti.ctype))
if not var.is_intent_out():
self.clean_frompyobj_list.append('Py_DECREF(%s);' % (argname))
self.decl_list.append('%s* %s = NULL;' % (ti.otype, argname))
args_f.append('%s->data' % (argname)) # is_scalar
else:
if var.is_intent_in():
self.pyarg_format_list.append('O&')
self.pyarg_obj_list.append('\npyobj_to_%s, &%s' % (ti.ctype, argname))
assert not isinstance(typedecl, TypeDecl)
if ti.ctype=='f2py_string0':
if not var.is_intent_in():
assert not var.is_intent_out(),'intent(out) not implemented for "%s"' % (var)
self.decl_list.append('%s %s = {NULL,0};' % (ti.ctype, argname))
args_f.append('%s.data' % argname) # is_scalar
extra_args_f.append('%s.len' % argname)
self.clean_frompyobj_list.append(\
'if (%s.len) free(%s.data);' % (argname,argname))
else:
self.decl_list.append('%s %s;' % (ti.ctype, argname))
args_f.append('&'+argname) # is_scalar
if var.is_intent_out(): # and is_scalar
if isinstance(typedecl, TypeStmt):
self.return_format_list.append('N')
self.return_obj_list.append('\n%s' % (argname))
else:
self.return_format_list.append('O&')
self.return_obj_list.append('\npyobj_from_%s, &%s' % (ti.ctype, argname))
WrapperCPPMacro(parent, 'F_FUNC')
self.call_list.append('%s_f(%s);' % (name,', '.join(args_f+extra_args_f)))
self.clean_pyobjfrom_list.reverse()
self.clean_call_list.reverse()
self.clean_frompyobj_list.reverse()
if self.return_obj_list: self.return_obj_list.insert(0,'')
parent.apply_templates(self)
return
|
