Movatterモバイル変換


[0]ホーム

URL:


Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Sign up
Appearance settings

Commit25fd516

Browse files
authored
Merge pull request#10726 from charris/backport-10657
BUG: Fix f2py generated code to work on PyPy
2 parents02c1bc5 +0fbeba2 commit25fd516

File tree

1 file changed

+25
-5
lines changed

1 file changed

+25
-5
lines changed

‎numpy/f2py/cb_rules.py‎

Lines changed: 25 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@
4444
\tPyTupleObject *capi_arglist = #name#_args_capi;
4545
\tPyObject *capi_return = NULL;
4646
\tPyObject *capi_tmp = NULL;
47+
\tPyObject *capi_arglist_list = NULL;
4748
\tint capi_j,capi_i = 0;
4849
\tint capi_longjmp_ok = 1;
4950
#decl#
@@ -85,13 +86,31 @@
8586
\t\tgoto capi_fail;
8687
\t}
8788
#setdims#
89+
#ifdef PYPY_VERSION
90+
#define CAPI_ARGLIST_SETITEM(idx, value) PyList_SetItem((PyObject *)capi_arglist_list, idx, value)
91+
\tcapi_arglist_list = PySequence_List(capi_arglist);
92+
\tif (capi_arglist_list == NULL) goto capi_fail;
93+
#else
94+
#define CAPI_ARGLIST_SETITEM(idx, value) PyTuple_SetItem((PyObject *)capi_arglist, idx, value)
95+
#endif
8896
#pyobjfrom#
97+
#undef CAPI_ARGLIST_SETITEM
98+
#ifdef PYPY_VERSION
99+
\tCFUNCSMESSPY(\"cb:capi_arglist=\",capi_arglist_list);
100+
#else
89101
\tCFUNCSMESSPY(\"cb:capi_arglist=\",capi_arglist);
102+
#endif
90103
\tCFUNCSMESS(\"cb:Call-back calling Python function #argname#.\\n\");
91104
#ifdef F2PY_REPORT_ATEXIT
92105
f2py_cb_start_call_clock();
93106
#endif
107+
#ifdef PYPY_VERSION
108+
\tcapi_return = PyObject_CallObject(#name#_capi,(PyObject *)capi_arglist_list);
109+
\tPy_DECREF(capi_arglist_list);
110+
\tcapi_arglist_list = NULL;
111+
#else
94112
\tcapi_return = PyObject_CallObject(#name#_capi,(PyObject *)capi_arglist);
113+
#endif
95114
#ifdef F2PY_REPORT_ATEXIT
96115
f2py_cb_stop_call_clock();
97116
#endif
@@ -119,6 +138,7 @@
119138
capi_fail:
120139
\tfprintf(stderr,\"Call-back #name# failed.\\n\");
121140
\tPy_XDECREF(capi_return);
141+
\tPy_XDECREF(capi_arglist_list);
122142
\tif (capi_longjmp_ok)
123143
\t\tlongjmp(#name#_jmpbuf,-1);
124144
capi_return_pt:
@@ -318,11 +338,11 @@
318338
}, {
319339
'pyobjfrom': [{isintent_in:"""\
320340
\tif (#name#_nofargs>capi_i)
321-
\t\tif (PyTuple_SetItem((PyObject *)capi_arglist,capi_i++,pyobj_from_#ctype#1(#varname_i#)))
341+
\t\tif (CAPI_ARGLIST_SETITEM(capi_i++,pyobj_from_#ctype#1(#varname_i#)))
322342
\t\t\tgoto capi_fail;"""},
323343
{isintent_inout:"""\
324344
\tif (#name#_nofargs>capi_i)
325-
\t\tif (PyTuple_SetItem((PyObject *)capi_arglist,capi_i++,pyarr_from_p_#ctype#1(#varname_i#_cb_capi)))
345+
\t\tif (CAPI_ARGLIST_SETITEM(capi_i++,pyarr_from_p_#ctype#1(#varname_i#_cb_capi)))
326346
\t\t\tgoto capi_fail;"""}],
327347
'need': [{isintent_in:'pyobj_from_#ctype#1'},
328348
{isintent_inout:'pyarr_from_p_#ctype#1'},
@@ -343,12 +363,12 @@
343363
'pyobjfrom': [{debugcapi:'\tfprintf(stderr,"debug-capi:cb:#varname#=\\"#showvalueformat#\\":%d:\\n",#varname_i#,#varname_i#_cb_len);'},
344364
{isintent_in:"""\
345365
\tif (#name#_nofargs>capi_i)
346-
\t\tif (PyTuple_SetItem((PyObject *)capi_arglist,capi_i++,pyobj_from_#ctype#1size(#varname_i#,#varname_i#_cb_len)))
366+
\t\tif (CAPI_ARGLIST_SETITEM(capi_i++,pyobj_from_#ctype#1size(#varname_i#,#varname_i#_cb_len)))
347367
\t\t\tgoto capi_fail;"""},
348368
{isintent_inout:"""\
349369
\tif (#name#_nofargs>capi_i) {
350370
\t\tint #varname_i#_cb_dims[] = {#varname_i#_cb_len};
351-
\t\tif (PyTuple_SetItem((PyObject *)capi_arglist,capi_i++,pyarr_from_p_#ctype#1(#varname_i#,#varname_i#_cb_dims)))
371+
\t\tif (CAPI_ARGLIST_SETITEM(capi_i++,pyarr_from_p_#ctype#1(#varname_i#,#varname_i#_cb_dims)))
352372
\t\t\tgoto capi_fail;
353373
\t}"""}],
354374
'need': [{isintent_in:'pyobj_from_#ctype#1size'},
@@ -381,7 +401,7 @@
381401
"""
382402
\t\tif (tmp_arr==NULL)
383403
\t\t\tgoto capi_fail;
384-
\t\tif (PyTuple_SetItem((PyObject *)capi_arglist,capi_i++,(PyObject *)tmp_arr))
404+
\t\tif (CAPI_ARGLIST_SETITEM(capi_i++,(PyObject *)tmp_arr))
385405
\t\t\tgoto capi_fail;
386406
}"""],
387407
'_check':l_and(isarray,isintent_nothide,l_or(isintent_in,isintent_inout)),

0 commit comments

Comments
 (0)

[8]ページ先頭

©2009-2025 Movatter.jp