source: python/ext/py-filterbank.c @ b055b4e

feature/autosinkfeature/constantqfeature/pitchshiftfeature/pydocstringsfeature/timestretchpitchshiftsamplertimestretchyinfft+
Last change on this file since b055b4e was b055b4e, checked in by Paul Brossier <piem@piem.org>, 3 years ago

python/ext/py-filterbank.c: use new_py_fvec

  • Property mode set to 100644
File size: 5.1 KB
RevLine 
[5652a8c]1#include "aubio-types.h"
[615ac7d]2
3static char Py_filterbank_doc[] = "filterbank object";
4
[59cb451]5typedef struct
6{
7  PyObject_HEAD
8  aubio_filterbank_t * o;
9  uint_t n_filters;
10  uint_t win_s;
[569b363]11  cvec_t vec;
12  fvec_t freqs;
13  fmat_t coeffs;
[b055b4e]14  PyObject *out;
15  fvec_t c_out;
[59cb451]16} Py_filterbank;
[615ac7d]17
18static PyObject *
19Py_filterbank_new (PyTypeObject * type, PyObject * args, PyObject * kwds)
20{
21  int win_s = 0, n_filters = 0;
22  Py_filterbank *self;
23  static char *kwlist[] = { "n_filters", "win_s", NULL };
24
25  if (!PyArg_ParseTupleAndKeywords (args, kwds, "|II", kwlist,
26          &n_filters, &win_s)) {
27    return NULL;
28  }
29
30  self = (Py_filterbank *) type->tp_alloc (type, 0);
31
32  if (self == NULL) {
33    return NULL;
34  }
35
36  self->win_s = Py_default_vector_length;
37  if (win_s > 0) {
38    self->win_s = win_s;
39  } else if (win_s < 0) {
40    PyErr_SetString (PyExc_ValueError,
41        "can not use negative window size");
42    return NULL;
43  }
44
45  self->n_filters = 40;
46  if (n_filters > 0) {
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");
51    return NULL;
52  }
53
54  return (PyObject *) self;
55}
56
[59cb451]57static int
58Py_filterbank_init (Py_filterbank * self, PyObject * args, PyObject * kwds)
59{
60  self->o = new_aubio_filterbank (self->n_filters, self->win_s);
61  if (self->o == NULL) {
62    char_t errstr[30];
63    sprintf(errstr, "error creating filterbank with n_filters=%d, win_s=%d",
64        self->n_filters, self->win_s);
[5c1200a]65    PyErr_SetString (PyExc_RuntimeError, errstr);
[59cb451]66    return -1;
67  }
[b055b4e]68  self->out = new_py_fvec(self->n_filters);
[615ac7d]69
[59cb451]70  return 0;
71}
[615ac7d]72
[59cb451]73static void
74Py_filterbank_del (Py_filterbank *self, PyObject *unused)
75{
76  del_aubio_filterbank(self->o);
[b055b4e]77  Py_DECREF(self->out);
[569b363]78  free(self->coeffs.data);
[5c1200a]79  Py_TYPE(self)->tp_free((PyObject *) self);
[59cb451]80}
[615ac7d]81
[f6cbefe]82static PyObject *
[615ac7d]83Py_filterbank_do(Py_filterbank * self, PyObject * args)
84{
85  PyObject *input;
86
87  if (!PyArg_ParseTuple (args, "O", &input)) {
88    return NULL;
89  }
90
[92a8800]91  if (!PyAubio_PyCvecToCCvec(input, &(self->vec) )) {
[615ac7d]92    return NULL;
93  }
94
[b055b4e]95  Py_INCREF(self->out);
96  if (!PyAubio_ArrayToCFvec(self->out, &(self->c_out))) {
97    return NULL;
98  }
[615ac7d]99  // compute the function
[b055b4e]100  aubio_filterbank_do (self->o, &(self->vec), &(self->c_out));
101  return self->out;
[615ac7d]102}
103
[5652a8c]104static PyMemberDef Py_filterbank_members[] = {
[615ac7d]105  {"win_s", T_INT, offsetof (Py_filterbank, win_s), READONLY,
106    "size of the window"},
107  {"n_filters", T_INT, offsetof (Py_filterbank, n_filters), READONLY,
108    "number of filters"},
[5652a8c]109  {NULL} /* sentinel */
110};
[615ac7d]111
[f6cbefe]112static PyObject *
[615ac7d]113Py_filterbank_set_triangle_bands (Py_filterbank * self, PyObject *args)
114{
115  uint_t err = 0;
116
117  PyObject *input;
118  uint_t samplerate;
119  if (!PyArg_ParseTuple (args, "OI", &input, &samplerate)) {
120    return NULL;
121  }
122
123  if (input == NULL) {
124    return NULL;
125  }
126
[569b363]127  if (!PyAubio_ArrayToCFvec(input, &(self->freqs) )) {
[615ac7d]128    return NULL;
129  }
130
131  err = aubio_filterbank_set_triangle_bands (self->o,
[569b363]132      &(self->freqs), samplerate);
[615ac7d]133  if (err > 0) {
134    PyErr_SetString (PyExc_ValueError,
135        "error when setting filter to A-weighting");
136    return NULL;
137  }
[9e6695d]138  Py_RETURN_NONE;
[615ac7d]139}
140
[f6cbefe]141static PyObject *
[615ac7d]142Py_filterbank_set_mel_coeffs_slaney (Py_filterbank * self, PyObject *args)
143{
144  uint_t err = 0;
145
146  uint_t samplerate;
147  if (!PyArg_ParseTuple (args, "I", &samplerate)) {
148    return NULL;
149  }
150
151  err = aubio_filterbank_set_mel_coeffs_slaney (self->o, samplerate);
152  if (err > 0) {
153    PyErr_SetString (PyExc_ValueError,
154        "error when setting filter to A-weighting");
155    return NULL;
156  }
[9e6695d]157  Py_RETURN_NONE;
[615ac7d]158}
159
[f6cbefe]160static PyObject *
161Py_filterbank_set_coeffs (Py_filterbank * self, PyObject *args)
162{
163  uint_t err = 0;
164
165  PyObject *input;
166  if (!PyArg_ParseTuple (args, "O", &input)) {
167    return NULL;
168  }
169
[569b363]170  if (!PyAubio_ArrayToCFmat(input, &(self->coeffs))) {
[f6cbefe]171    return NULL;
172  }
173
[569b363]174  err = aubio_filterbank_set_coeffs (self->o, &(self->coeffs));
[f6cbefe]175
176  if (err > 0) {
177    PyErr_SetString (PyExc_ValueError,
178        "error when setting filter coefficients");
179    return NULL;
180  }
[9e6695d]181  Py_RETURN_NONE;
[f6cbefe]182}
183
184static PyObject *
[615ac7d]185Py_filterbank_get_coeffs (Py_filterbank * self, PyObject *unused)
186{
[363ce7a]187  return (PyObject *)PyAubio_CFmatToArray(
188      aubio_filterbank_get_coeffs (self->o) );
[615ac7d]189}
190
191static PyMethodDef Py_filterbank_methods[] = {
192  {"set_triangle_bands", (PyCFunction) Py_filterbank_set_triangle_bands,
193    METH_VARARGS, "set coefficients of filterbanks"},
194  {"set_mel_coeffs_slaney", (PyCFunction) Py_filterbank_set_mel_coeffs_slaney,
195    METH_VARARGS, "set coefficients of filterbank as in Auditory Toolbox"},
196  {"get_coeffs", (PyCFunction) Py_filterbank_get_coeffs,
197    METH_NOARGS, "get coefficients of filterbank"},
[f6cbefe]198  {"set_coeffs", (PyCFunction) Py_filterbank_set_coeffs,
199    METH_VARARGS, "set coefficients of filterbank"},
[615ac7d]200  {NULL}
201};
202
[5652a8c]203PyTypeObject Py_filterbankType = {
204  PyVarObject_HEAD_INIT (NULL, 0)
205  "aubio.filterbank",
206  sizeof (Py_filterbank),
207  0,
208  (destructor) Py_filterbank_del,
209  0,
210  0,
211  0,
212  0,
213  0,
214  0,
215  0,
216  0,
217  0,
218  (ternaryfunc)Py_filterbank_do,
219  0,
220  0,
221  0,
222  0,
223  Py_TPFLAGS_DEFAULT,
224  Py_filterbank_doc,
225  0,
226  0,
227  0,
228  0,
229  0,
230  0,
231  Py_filterbank_methods,
232  Py_filterbank_members,
233  0,
234  0,
235  0,
236  0,
237  0,
238  0,
239  (initproc) Py_filterbank_init,
240  0,
241  Py_filterbank_new,
242};
Note: See TracBrowser for help on using the repository browser.