Changeset 59cb451 for python/ext

Timestamp:

Apr 18, 2016, 8:31:20 PM (9 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/autosink, feature/cnn, feature/cnn_org, feature/constantq, feature/crepe, feature/crepe_org, feature/pitchshift, feature/pydocstrings, feature/timestretch, fix/ffmpeg5, master, pitchshift, sampler, timestretch, yinfft+

Children:

202697a

Parents:

a28dab6

Message:

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

File:

• python/ext/py-filterbank.c (modified) (4 diffs)

python/ext/py-filterbank.c

@@ -3 +3 @@
 static char Py_filterbank_doc[] = "filterbank object";
 
-AUBIO_DECLARE(filterbank, uint_t n_filters; uint_t win_s)
+typedef struct
+{
+  PyObject_HEAD
+  aubio_filterbank_t * o;
+  uint_t n_filters;
+  uint_t win_s;
+  fvec_t *out;
+} Py_filterbank;
 
 //AUBIO_NEW(filterbank)
@@ -45 +52 @@
 }
 
-
-AUBIO_INIT(filterbank, self->n_filters, self->win_s)
-
-AUBIO_DEL(filterbank)
+static int
+Py_filterbank_init (Py_filterbank * self, PyObject * args, PyObject * kwds)
+{
+  self->o = new_aubio_filterbank (self->n_filters, self->win_s);
+  if (self->o == NULL) {
+    char_t errstr[30];
+    sprintf(errstr, "error creating filterbank with n_filters=%d, win_s=%d",
+        self->n_filters, self->win_s);
+    PyErr_SetString (PyExc_StandardError, errstr);
+    return -1;
+  }
+  self->out = new_fvec(self->n_filters);
+
+  return 0;
+}
+
+static void
+Py_filterbank_del (Py_filterbank *self, PyObject *unused)
+{
+  del_aubio_filterbank(self->o);
+  del_fvec(self->out);
+  self->ob_type->tp_free((PyObject *) self);
+}
 
 static PyObject *
@@ -55 +81 @@
   PyObject *input;
   cvec_t *vec;
-  fvec_t *out;
 
   if (!PyArg_ParseTuple (args, "O", &input)) {
@@ -67 +92 @@
   }
 
-  out = new_fvec (self->n_filters);
-
   // compute the function
-  aubio_filterbank_do (self->o, vec, out);
-  return (PyObject *)PyAubio_CFvecToArray(out);
+  aubio_filterbank_do (self->o, vec, self->out);
+  return (PyObject *)PyAubio_CFvecToArray(self->out);
 }