1 #include "aubio-types.h"
3 static char Py_filterbank_doc[] = "filterbank object";
8 aubio_filterbank_t * o;
19 Py_filterbank_new (PyTypeObject * type, PyObject * args, PyObject * kwds)
21 int win_s = 0, n_filters = 0;
23 static char *kwlist[] = { "n_filters", "win_s", NULL };
25 if (!PyArg_ParseTupleAndKeywords (args, kwds, "|II", kwlist,
26 &n_filters, &win_s)) {
30 self = (Py_filterbank *) type->tp_alloc (type, 0);
36 self->win_s = Py_default_vector_length;
39 } else if (win_s < 0) {
40 PyErr_SetString (PyExc_ValueError,
41 "can not use negative window size");
47 self->n_filters = n_filters;
48 } else if (n_filters < 0) {
49 PyErr_SetString (PyExc_ValueError,
50 "can not use negative number of filters");
54 return (PyObject *) self;
58 Py_filterbank_init (Py_filterbank * self, PyObject * args, PyObject * kwds)
60 self->o = new_aubio_filterbank (self->n_filters, self->win_s);
61 if (self->o == NULL) {
62 PyErr_Format(PyExc_RuntimeError, "error creating filterbank with"
63 " n_filters=%d, win_s=%d", self->n_filters, self->win_s);
66 self->out = new_py_fvec(self->n_filters);
72 Py_filterbank_del (Py_filterbank *self, PyObject *unused)
75 free(self->coeffs.data);
76 del_aubio_filterbank(self->o);
78 Py_XDECREF(self->out);
79 Py_TYPE(self)->tp_free((PyObject *) self);
83 Py_filterbank_do(Py_filterbank * self, PyObject * args)
87 if (!PyArg_ParseTuple (args, "O", &input)) {
91 if (!PyAubio_PyCvecToCCvec(input, &(self->vec) )) {
95 if (self->vec.length != self->win_s / 2 + 1) {
96 PyErr_Format(PyExc_ValueError,
97 "input cvec has length %d, but filterbank expects length %d",
98 self->vec.length, self->win_s / 2 + 1);
102 Py_INCREF(self->out);
103 if (!PyAubio_ArrayToCFvec(self->out, &(self->c_out))) {
106 // compute the function
107 aubio_filterbank_do (self->o, &(self->vec), &(self->c_out));
111 static PyMemberDef Py_filterbank_members[] = {
112 {"win_s", T_INT, offsetof (Py_filterbank, win_s), READONLY,
113 "size of the window"},
114 {"n_filters", T_INT, offsetof (Py_filterbank, n_filters), READONLY,
115 "number of filters"},
116 {NULL} /* sentinel */
120 Py_filterbank_set_triangle_bands (Py_filterbank * self, PyObject *args)
126 if (!PyArg_ParseTuple (args, "OI", &input, &samplerate)) {
134 if (!PyAubio_ArrayToCFvec(input, &(self->freqs) )) {
138 err = aubio_filterbank_set_triangle_bands (self->o,
139 &(self->freqs), samplerate);
141 if (PyErr_Occurred() == NULL) {
142 PyErr_SetString (PyExc_ValueError, "error running set_triangle_bands");
144 // change the RuntimeError into ValueError
145 PyObject *type, *value, *traceback;
146 PyErr_Fetch(&type, &value, &traceback);
147 PyErr_Restore(PyExc_ValueError, value, traceback);
155 Py_filterbank_set_mel_coeffs_slaney (Py_filterbank * self, PyObject *args)
160 if (!PyArg_ParseTuple (args, "I", &samplerate)) {
164 err = aubio_filterbank_set_mel_coeffs_slaney (self->o, samplerate);
166 if (PyErr_Occurred() == NULL) {
167 PyErr_SetString (PyExc_ValueError, "error running set_mel_coeffs_slaney");
169 // change the RuntimeError into ValueError
170 PyObject *type, *value, *traceback;
171 PyErr_Fetch(&type, &value, &traceback);
172 PyErr_Restore(PyExc_ValueError, value, traceback);
180 Py_filterbank_set_mel_coeffs (Py_filterbank * self, PyObject *args)
187 if (!PyArg_ParseTuple (args, "I" AUBIO_NPY_SMPL_CHR AUBIO_NPY_SMPL_CHR,
188 &samplerate, &freq_min, &freq_max)) {
192 err = aubio_filterbank_set_mel_coeffs (self->o, samplerate,
195 if (PyErr_Occurred() == NULL) {
196 PyErr_SetString (PyExc_ValueError, "error running set_mel_coeffs");
198 // change the RuntimeError into ValueError
199 PyObject *type, *value, *traceback;
200 PyErr_Fetch(&type, &value, &traceback);
201 PyErr_Restore(PyExc_ValueError, value, traceback);
209 Py_filterbank_set_mel_coeffs_htk (Py_filterbank * self, PyObject *args)
216 if (!PyArg_ParseTuple (args, "I" AUBIO_NPY_SMPL_CHR AUBIO_NPY_SMPL_CHR,
217 &samplerate, &freq_min, &freq_max)) {
221 err = aubio_filterbank_set_mel_coeffs_htk (self->o,
222 freq_min, freq_max, samplerate);
224 if (PyErr_Occurred() == NULL) {
225 PyErr_SetString (PyExc_ValueError, "error running set_mel_coeffs_htk");
227 // change the RuntimeError into ValueError
228 PyObject *type, *value, *traceback;
229 PyErr_Fetch(&type, &value, &traceback);
230 PyErr_Restore(PyExc_ValueError, value, traceback);
238 Py_filterbank_set_coeffs (Py_filterbank * self, PyObject *args)
243 if (!PyArg_ParseTuple (args, "O", &input)) {
247 if (!PyAubio_ArrayToCFmat(input, &(self->coeffs))) {
251 err = aubio_filterbank_set_coeffs (self->o, &(self->coeffs));
254 PyErr_SetString (PyExc_ValueError,
255 "error when setting filter coefficients");
262 Py_filterbank_get_coeffs (Py_filterbank * self, PyObject *unused)
264 return (PyObject *)PyAubio_CFmatToArray(
265 aubio_filterbank_get_coeffs (self->o) );
268 static PyMethodDef Py_filterbank_methods[] = {
269 {"set_triangle_bands", (PyCFunction) Py_filterbank_set_triangle_bands,
270 METH_VARARGS, "set coefficients of filterbanks"},
271 {"set_mel_coeffs_slaney", (PyCFunction) Py_filterbank_set_mel_coeffs_slaney,
272 METH_VARARGS, "set coefficients of filterbank as in Auditory Toolbox"},
273 {"set_mel_coeffs", (PyCFunction) Py_filterbank_set_mel_coeffs,
274 METH_VARARGS, "set coefficients of filterbank to linearly spaced mel scale"},
275 {"set_mel_coeffs_htk", (PyCFunction) Py_filterbank_set_mel_coeffs_htk,
276 METH_VARARGS, "set coefficients of filterbank to linearly spaced mel scale"},
277 {"get_coeffs", (PyCFunction) Py_filterbank_get_coeffs,
278 METH_NOARGS, "get coefficients of filterbank"},
279 {"set_coeffs", (PyCFunction) Py_filterbank_set_coeffs,
280 METH_VARARGS, "set coefficients of filterbank"},
284 PyTypeObject Py_filterbankType = {
285 PyVarObject_HEAD_INIT (NULL, 0)
287 sizeof (Py_filterbank),
289 (destructor) Py_filterbank_del,
299 (ternaryfunc)Py_filterbank_do,
312 Py_filterbank_methods,
313 Py_filterbank_members,
320 (initproc) Py_filterbank_init,