static PyObject *Py_GenericFilter1D(PyObject *obj, PyObject *args)
{
PyArrayObject *input = NULL, *output = NULL;
PyObject *fnc = NULL, *extra_arguments = NULL, *extra_keywords = NULL;
void *func = Py_Filter1DFunc, *data = NULL;
NI_PythonCallbackData cbdata;
int axis, mode;
#if PY_VERSION_HEX < 0x02050000
long origin, filter_size;
#define FMT "l"
#else
npy_intp origin, filter_size;
#define FMT "n"
#endif
double cval;
if (!PyArg_ParseTuple(args, "O&O" FMT "iO&id" FMT "OO",
NI_ObjectToInputArray, &input,
&fnc, &filter_size, &axis,
NI_ObjectToOutputArray, &output,
&mode, &cval, &origin,
&extra_arguments, &extra_keywords))
goto exit;
#undef FMT
if (!PyTuple_Check(extra_arguments)) {
PyErr_SetString(PyExc_RuntimeError, "extra_arguments must be a tuple");
goto exit;
}
if (!PyDict_Check(extra_keywords)) {
PyErr_SetString(PyExc_RuntimeError,
"extra_keywords must be a dictionary");
goto exit;
}
if (NpyCapsule_Check(fnc)) {
func = NpyCapsule_AsVoidPtr(fnc);
data = NpyCapsule_GetDesc(fnc);
} else if (PyCallable_Check(fnc)) {
cbdata.function = fnc;
cbdata.extra_arguments = extra_arguments;
cbdata.extra_keywords = extra_keywords;
data = (void*)&cbdata;
} else {
PyErr_SetString(PyExc_RuntimeError,
"function parameter is not callable");
goto exit;
}
if (!NI_GenericFilter1D(input, func, data, filter_size, axis, output,
(NI_ExtendMode)mode, cval, origin))
goto exit;
exit:
Py_XDECREF(input);
Py_XDECREF(output);
return PyErr_Occurred() ? NULL : Py_BuildValue("");
}
static PyObject *Py_GenericFilter1D(PyObject *obj, PyObject *args)
{
PyArrayObject *input = NULL, *output = NULL;
PyObject *fnc = NULL, *extra_arguments = NULL, *extra_keywords = NULL;
void *func = NULL, *data = NULL;
NI_PythonCallbackData cbdata;
int axis, mode;
npy_intp origin, filter_size;
double cval;
ccallback_t callback;
static ccallback_signature_t callback_signatures[] = {
{"int (double *, intptr_t, double *, intptr_t, void *)"},
{"int (double *, npy_intp, double *, npy_intp, void *)"},
#if NPY_SIZEOF_INTP == NPY_SIZEOF_SHORT
{"int (double *, short, double *, short, void *)"},
#endif
#if NPY_SIZEOF_INTP == NPY_SIZEOF_INT
{"int (double *, int, double *, int, void *)"},
#endif
#if NPY_SIZEOF_INTP == NPY_SIZEOF_LONG
{"int (double *, long, double *, long, void *)"},
#endif
#if NPY_SIZEOF_INTP == NPY_SIZEOF_LONGLONG
{"int (double *, long long, double *, long long, void *)"},
#endif
{NULL}
};
callback.py_function = NULL;
callback.c_function = NULL;
if (!PyArg_ParseTuple(args, "O&OniO&idnOO",
NI_ObjectToInputArray, &input,
&fnc, &filter_size, &axis,
NI_ObjectToOutputArray, &output,
&mode, &cval, &origin,
&extra_arguments, &extra_keywords))
goto exit;
if (!PyTuple_Check(extra_arguments)) {
PyErr_SetString(PyExc_RuntimeError, "extra_arguments must be a tuple");
goto exit;
}
if (!PyDict_Check(extra_keywords)) {
PyErr_SetString(PyExc_RuntimeError,
"extra_keywords must be a dictionary");
goto exit;
}
if (PyCapsule_CheckExact(fnc) && PyCapsule_GetName(fnc) == NULL) {
/* 'Legacy' low-level callable */
func = PyCapsule_GetPointer(fnc, NULL);
data = PyCapsule_GetContext(fnc);
#if PY_VERSION_HEX < 0x03000000
} else if (PyCObject_Check(fnc)) {
/* 'Legacy' low-level callable on Py2 */
func = PyCObject_AsVoidPtr(fnc);
data = PyCObject_GetDesc(fnc);
#endif
} else {
int ret;
ret = ccallback_prepare(&callback, callback_signatures, fnc, CCALLBACK_DEFAULTS);
if (ret == -1) {
goto exit;
}
if (callback.py_function != NULL) {
cbdata.extra_arguments = extra_arguments;
cbdata.extra_keywords = extra_keywords;
callback.info_p = (void*)&cbdata;
func = Py_Filter1DFunc;
data = (void*)&callback;
}
else {
func = callback.c_function;
data = callback.user_data;
}
}
NI_GenericFilter1D(input, func, data, filter_size, axis, output,
(NI_ExtendMode)mode, cval, origin);
#ifdef HAVE_WRITEBACKIFCOPY
PyArray_ResolveWritebackIfCopy(output);
#endif
exit:
if (callback.py_function != NULL || callback.c_function != NULL) {
ccallback_release(&callback);
}
Py_XDECREF(input);
Py_XDECREF(output);
return PyErr_Occurred() ? NULL : Py_BuildValue("");
}
