Changeset 1a121ca

Timestamp:

Feb 23, 2014, 9:32:49 PM (10 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:

060a3135

Parents:

31a5c405

Message:

python/ext/py-sink.c: add channels and do_multi

File:

• python/ext/py-sink.c (modified) (6 diffs)

python/ext/py-sink.c

@@ -7 +7 @@
   char_t* uri;
   uint_t samplerate;
+  uint_t channels;
 } Py_sink;
 
@@ -17 +18 @@
   char_t* uri = NULL;
   uint_t samplerate = 0;
-  static char *kwlist[] = { "uri", "samplerate", NULL };
+  uint_t channels = 0;
+  static char *kwlist[] = { "uri", "samplerate", "channels", NULL };
 
-  if (!PyArg_ParseTupleAndKeywords (args, kwds, "|sI", kwlist,
-          &uri, &samplerate)) {
+  if (!PyArg_ParseTupleAndKeywords (args, kwds, "|sII", kwlist,
+          &uri, &samplerate, &channels)) {
     return NULL;
   }
@@ -44 +46 @@
   }
 
+  self->channels = 1;
+  if ((sint_t)channels > 0) {
+    self->channels = channels;
+  } else if ((sint_t)channels < 0) {
+    PyErr_SetString (PyExc_ValueError,
+        "can not use negative or null value for channels");
+    return NULL;
+  }
+
   return (PyObject *) self;
 }
 
-AUBIO_INIT(sink , self->uri, self->samplerate)
+static int
+Py_sink_init (Py_sink * self, PyObject * args, PyObject * kwds)
+{
+  if (self->channels == 1) {
+    self->o = new_aubio_sink ( self->uri, self->samplerate );
+  } else {
+    self->o = new_aubio_sink ( self->uri, 0 );
+    aubio_sink_preset_channels ( self->o, self->channels );
+    aubio_sink_preset_samplerate ( self->o, self->samplerate );
+  }
+  if (self->o == NULL) {
+    PyErr_SetString (PyExc_StandardError, "error creating sink with this uri");
+    return -1;
+  }
+  self->samplerate = aubio_sink_get_samplerate ( self->o );
+  self->channels = aubio_sink_get_channels ( self->o );
+
+  return 0;
+}
 
 AUBIO_DEL(sink)
 
 /* function Py_sink_do */
-static PyObject * 
+static PyObject *
 Py_sink_do(Py_sink * self, PyObject * args)
 {
@@ -75 +104 @@
   }
 
-  
-  
+
+
 
 
@@ -85 +114 @@
 }
 
+/* function Py_sink_do */
+static PyObject *
+Py_sink_do_multi(Py_sink * self, PyObject * args)
+{
+  /* input vectors python prototypes */
+  PyObject * write_data_obj;
+
+  /* input vectors prototypes */
+  fmat_t * write_data;
+  uint_t write;
+
+
+  if (!PyArg_ParseTuple (args, "OI", &write_data_obj, &write)) {
+    return NULL;
+  }
+
+
+  /* input vectors parsing */
+  write_data = PyAubio_ArrayToCFmat (write_data_obj);
+
+  if (write_data == NULL) {
+    return NULL;
+  }
+
+
+
+
+
+  /* compute _do function */
+  aubio_sink_do_multi (self->o, write_data, write);
+  Py_RETURN_NONE;
+}
+
 AUBIO_MEMBERS_START(sink)
   {"uri", T_STRING, offsetof (Py_sink, uri), READONLY, ""},
   {"samplerate", T_INT, offsetof (Py_sink, samplerate), READONLY, ""},
+  {"channels", T_INT, offsetof (Py_sink, channels), READONLY, ""},
 AUBIO_MEMBERS_STOP(sink)
 
@@ -98 +161 @@
 
 static PyMethodDef Py_sink_methods[] = {
-  {"close", (PyCFunction) Pyaubio_sink_close,
-    METH_NOARGS, ""},
+  {"__call__", (PyCFunction) Py_sink_do, METH_VARARGS,
+    "x.__call__(vec, write)\n"
+    "write monophonic vector to sink"
+    ""},
+  {"do", (PyCFunction) Py_sink_do, METH_VARARGS,
+    "x.do(vec, write)\n"
+    "write monophonic vector to sink"
+    ""},
+  {"do_multi", (PyCFunction) Py_sink_do_multi, METH_VARARGS,
+    "x.do_multi(mat, write)\n"
+    "write polyphonic vector to sink"},
+  {"close", (PyCFunction) Pyaubio_sink_close, METH_NOARGS,
+    "x.close()\n"
+    "close this sink now"},
   {NULL} /* sentinel */
 };