source: python/ext/py-filterbank.c @ 59cb451

feature/autosinkfeature/cnnfeature/cnn_orgfeature/constantqfeature/crepefeature/crepe_orgfeature/pitchshiftfeature/pydocstringsfeature/timestretchfix/ffmpeg5pitchshiftsamplertimestretchyinfft+
Last change on this file since 59cb451 was 59cb451, checked in by Paul Brossier <piem@piem.org>, 8 years ago

ext/py-filterbank.c: continue fixing memory leak (#49)

  • Property mode set to 100644
File size: 4.7 KB
RevLine 
[615ac7d]1#include "aubiowraphell.h"
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;
11  fvec_t *out;
12} Py_filterbank;
[615ac7d]13
14//AUBIO_NEW(filterbank)
15static PyObject *
16Py_filterbank_new (PyTypeObject * type, PyObject * args, PyObject * kwds)
17{
18  int win_s = 0, n_filters = 0;
19  Py_filterbank *self;
20  static char *kwlist[] = { "n_filters", "win_s", NULL };
21
22  if (!PyArg_ParseTupleAndKeywords (args, kwds, "|II", kwlist,
23          &n_filters, &win_s)) {
24    return NULL;
25  }
26
27  self = (Py_filterbank *) type->tp_alloc (type, 0);
28
29  if (self == NULL) {
30    return NULL;
31  }
32
33  self->win_s = Py_default_vector_length;
34  if (win_s > 0) {
35    self->win_s = win_s;
36  } else if (win_s < 0) {
37    PyErr_SetString (PyExc_ValueError,
38        "can not use negative window size");
39    return NULL;
40  }
41
42  self->n_filters = 40;
43  if (n_filters > 0) {
44    self->n_filters = n_filters;
45  } else if (n_filters < 0) {
46    PyErr_SetString (PyExc_ValueError,
47        "can not use negative number of filters");
48    return NULL;
49  }
50
51  return (PyObject *) self;
52}
53
[59cb451]54static int
55Py_filterbank_init (Py_filterbank * self, PyObject * args, PyObject * kwds)
56{
57  self->o = new_aubio_filterbank (self->n_filters, self->win_s);
58  if (self->o == NULL) {
59    char_t errstr[30];
60    sprintf(errstr, "error creating filterbank with n_filters=%d, win_s=%d",
61        self->n_filters, self->win_s);
62    PyErr_SetString (PyExc_StandardError, errstr);
63    return -1;
64  }
65  self->out = new_fvec(self->n_filters);
[615ac7d]66
[59cb451]67  return 0;
68}
[615ac7d]69
[59cb451]70static void
71Py_filterbank_del (Py_filterbank *self, PyObject *unused)
72{
73  del_aubio_filterbank(self->o);
74  del_fvec(self->out);
75  self->ob_type->tp_free((PyObject *) self);
76}
[615ac7d]77
[f6cbefe]78static PyObject *
[615ac7d]79Py_filterbank_do(Py_filterbank * self, PyObject * args)
80{
81  PyObject *input;
[3e882a8]82  cvec_t *vec;
[615ac7d]83
84  if (!PyArg_ParseTuple (args, "O", &input)) {
85    return NULL;
86  }
87
[3e882a8]88  vec = PyAubio_ArrayToCCvec (input);
[615ac7d]89
90  if (vec == NULL) {
91    return NULL;
92  }
93
94  // compute the function
[59cb451]95  aubio_filterbank_do (self->o, vec, self->out);
96  return (PyObject *)PyAubio_CFvecToArray(self->out);
[615ac7d]97}
98
[f6cbefe]99AUBIO_MEMBERS_START(filterbank)
[615ac7d]100  {"win_s", T_INT, offsetof (Py_filterbank, win_s), READONLY,
101    "size of the window"},
102  {"n_filters", T_INT, offsetof (Py_filterbank, n_filters), READONLY,
103    "number of filters"},
104AUBIO_MEMBERS_STOP(filterbank)
105
[f6cbefe]106static PyObject *
[615ac7d]107Py_filterbank_set_triangle_bands (Py_filterbank * self, PyObject *args)
108{
109  uint_t err = 0;
110
111  PyObject *input;
112  uint_t samplerate;
[363ce7a]113  fvec_t *freqs;
[615ac7d]114  if (!PyArg_ParseTuple (args, "OI", &input, &samplerate)) {
115    return NULL;
116  }
117
118  if (input == NULL) {
119    return NULL;
120  }
121
[363ce7a]122  freqs = PyAubio_ArrayToCFvec (input);
[615ac7d]123
124  if (freqs == NULL) {
125    return NULL;
126  }
127
128  err = aubio_filterbank_set_triangle_bands (self->o,
[363ce7a]129      freqs, samplerate);
[615ac7d]130  if (err > 0) {
131    PyErr_SetString (PyExc_ValueError,
132        "error when setting filter to A-weighting");
133    return NULL;
134  }
[9e6695d]135  Py_RETURN_NONE;
[615ac7d]136}
137
[f6cbefe]138static PyObject *
[615ac7d]139Py_filterbank_set_mel_coeffs_slaney (Py_filterbank * self, PyObject *args)
140{
141  uint_t err = 0;
142
143  uint_t samplerate;
144  if (!PyArg_ParseTuple (args, "I", &samplerate)) {
145    return NULL;
146  }
147
148  err = aubio_filterbank_set_mel_coeffs_slaney (self->o, samplerate);
149  if (err > 0) {
150    PyErr_SetString (PyExc_ValueError,
151        "error when setting filter to A-weighting");
152    return NULL;
153  }
[9e6695d]154  Py_RETURN_NONE;
[615ac7d]155}
156
[f6cbefe]157static PyObject *
158Py_filterbank_set_coeffs (Py_filterbank * self, PyObject *args)
159{
160  uint_t err = 0;
161
162  PyObject *input;
163  fmat_t *coeffs;
164
165  if (!PyArg_ParseTuple (args, "O", &input)) {
166    return NULL;
167  }
168
169  coeffs = PyAubio_ArrayToCFmat (input);
170
171  if (coeffs == NULL) {
172    PyErr_SetString (PyExc_ValueError,
173        "unable to parse input array");
174    return NULL;
175  }
176
177  err = aubio_filterbank_set_coeffs (self->o, coeffs);
178
179  if (err > 0) {
180    PyErr_SetString (PyExc_ValueError,
181        "error when setting filter coefficients");
182    return NULL;
183  }
[9e6695d]184  Py_RETURN_NONE;
[f6cbefe]185}
186
187static PyObject *
[615ac7d]188Py_filterbank_get_coeffs (Py_filterbank * self, PyObject *unused)
189{
[363ce7a]190  return (PyObject *)PyAubio_CFmatToArray(
191      aubio_filterbank_get_coeffs (self->o) );
[615ac7d]192}
193
194static PyMethodDef Py_filterbank_methods[] = {
195  {"set_triangle_bands", (PyCFunction) Py_filterbank_set_triangle_bands,
196    METH_VARARGS, "set coefficients of filterbanks"},
197  {"set_mel_coeffs_slaney", (PyCFunction) Py_filterbank_set_mel_coeffs_slaney,
198    METH_VARARGS, "set coefficients of filterbank as in Auditory Toolbox"},
199  {"get_coeffs", (PyCFunction) Py_filterbank_get_coeffs,
200    METH_NOARGS, "get coefficients of filterbank"},
[f6cbefe]201  {"set_coeffs", (PyCFunction) Py_filterbank_set_coeffs,
202    METH_VARARGS, "set coefficients of filterbank"},
[615ac7d]203  {NULL}
204};
205
206AUBIO_TYPEOBJECT(filterbank, "aubio.filterbank")
Note: See TracBrowser for help on using the repository browser.