ref: 621f1ff6aed41ad258484d939eff9d97a0e25c5e
dir: /python/aubiomodule.c/
#include <Python.h> #define PY_ARRAY_UNIQUE_SYMBOL PyArray_API //#define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION #include <numpy/arrayobject.h> #include "aubio-types.h" #include "generated/aubio-generated.h" static char Py_alpha_norm_doc[] = "compute alpha normalisation factor"; static PyObject * Py_alpha_norm (PyObject * self, PyObject * args) { PyObject *input; fvec_t *vec; smpl_t alpha; PyObject *result; if (!PyArg_ParseTuple (args, "Of:alpha_norm", &input, &alpha)) { return NULL; } if (input == NULL) { return NULL; } vec = PyAubio_ArrayToCFvec (input); if (vec == NULL) { return NULL; } // compute the function result = Py_BuildValue ("f", fvec_alpha_norm (vec, alpha)); if (result == NULL) { return NULL; } return result; } static char Py_zero_crossing_rate_doc[] = "compute zero crossing rate"; static PyObject * Py_zero_crossing_rate (PyObject * self, PyObject * args) { PyObject *input; fvec_t *vec; PyObject *result; if (!PyArg_ParseTuple (args, "O:zero_crossing_rate", &input)) { return NULL; } if (input == NULL) { return NULL; } vec = PyAubio_ArrayToCFvec (input); if (vec == NULL) { return NULL; } // compute the function result = Py_BuildValue ("f", aubio_zero_crossing_rate (vec)); if (result == NULL) { return NULL; } return result; } static char Py_min_removal_doc[] = "compute zero crossing rate"; static PyObject * Py_min_removal(PyObject * self, PyObject * args) { PyObject *input; fvec_t *vec; if (!PyArg_ParseTuple (args, "O:min_removal", &input)) { return NULL; } if (input == NULL) { return NULL; } vec = PyAubio_ArrayToCFvec (input); if (vec == NULL) { return NULL; } // compute the function fvec_min_removal (vec); // since this function does not return, we could return None //return Py_None; // however it is convenient to return the modified vector return (PyObject *) PyAubio_CFvecToArray(vec); // or even without converting it back to an array //Py_INCREF(vec); //return (PyObject *)vec; } static PyMethodDef aubio_methods[] = { {"alpha_norm", Py_alpha_norm, METH_VARARGS, Py_alpha_norm_doc}, {"zero_crossing_rate", Py_zero_crossing_rate, METH_VARARGS, Py_zero_crossing_rate_doc}, {"min_removal", Py_min_removal, METH_VARARGS, Py_min_removal_doc}, {NULL, NULL} /* Sentinel */ }; static char aubio_module_doc[] = "Python module for the aubio library"; PyMODINIT_FUNC init_aubio (void) { PyObject *m; int err; if ( (PyType_Ready (&Py_cvecType) < 0) || (PyType_Ready (&Py_filterType) < 0) || (PyType_Ready (&Py_filterbankType) < 0) || (PyType_Ready (&Py_fftType) < 0) || (PyType_Ready (&Py_pvocType) < 0) // generated objects || (generated_types_ready() < 0 ) ) { return; } m = Py_InitModule3 ("_aubio", aubio_methods, aubio_module_doc); if (m == NULL) { return; } err = _import_array (); if (err != 0) { fprintf (stderr, "Unable to import Numpy C API from aubio module (error %d)\n", err); } Py_INCREF (&Py_cvecType); PyModule_AddObject (m, "cvec", (PyObject *) & Py_cvecType); Py_INCREF (&Py_filterType); PyModule_AddObject (m, "digital_filter", (PyObject *) & Py_filterType); Py_INCREF (&Py_filterbankType); PyModule_AddObject (m, "filterbank", (PyObject *) & Py_filterbankType); Py_INCREF (&Py_fftType); PyModule_AddObject (m, "fft", (PyObject *) & Py_fftType); Py_INCREF (&Py_pvocType); PyModule_AddObject (m, "pvoc", (PyObject *) & Py_pvocType); // generated objects add_generated_objects(m); }