1 #include "aubio-types.h"
18 static char Py_source_doc[] = ""
19 " __new__(path, samplerate = 0, hop_size = 512, channels = 1)\n"
21 " Create a new source, opening the given path for reading.\n"
26 " Create a new source, using the original samplerate, with hop_size = 512:\n"
28 " >>> source('/tmp/t.wav')\n"
30 " Create a new source, resampling the original to 8000Hz:\n"
32 " >>> source('/tmp/t.wav', samplerate = 8000)\n"
34 " Create a new source, resampling it at 32000Hz, hop_size = 32:\n"
36 " >>> source('/tmp/t.wav', samplerate = 32000, hop_size = 32)\n"
38 " Create a new source, using its original samplerate:\n"
40 " >>> source('/tmp/t.wav', samplerate = 0)\n"
43 " vec, read = x() <==> vec, read = x.do()\n"
45 " Read vector from source.\n"
52 static char Py_source_get_samplerate_doc[] = ""
53 "x.get_samplerate() -> source samplerate\n"
55 "Get samplerate of source.";
57 static char Py_source_get_channels_doc[] = ""
58 "x.get_channels() -> number of channels\n"
60 "Get number of channels in source.";
62 static char Py_source_do_doc[] = ""
63 "vec, read = x.do() <==> vec, read = x()\n"
65 "Read monophonic vector from source.";
67 static char Py_source_do_multi_doc[] = ""
68 "mat, read = x.do_multi()\n"
70 "Read polyphonic vector from source.";
72 static char Py_source_close_doc[] = ""
75 "Close this source now.";
77 static char Py_source_seek_doc[] = ""
80 "Seek to resampled frame position.";
83 Py_source_new (PyTypeObject * pytype, PyObject * args, PyObject * kwds)
87 uint_t samplerate = 0;
90 static char *kwlist[] = { "uri", "samplerate", "hop_size", "channels", NULL };
92 if (!PyArg_ParseTupleAndKeywords (args, kwds, "|sIII", kwlist,
93 &uri, &samplerate, &hop_size, &channels)) {
97 self = (Py_source *) pytype->tp_alloc (pytype, 0);
105 self->uri = (char_t *)malloc(sizeof(char_t) * (strnlen(uri, PATH_MAX) + 1));
106 strncpy(self->uri, uri, strnlen(uri, PATH_MAX) + 1);
109 self->samplerate = 0;
110 if ((sint_t)samplerate > 0) {
111 self->samplerate = samplerate;
112 } else if ((sint_t)samplerate < 0) {
113 PyErr_SetString (PyExc_ValueError,
114 "can not use negative value for samplerate");
118 self->hop_size = Py_default_vector_length / 2;
119 if ((sint_t)hop_size > 0) {
120 self->hop_size = hop_size;
121 } else if ((sint_t)hop_size < 0) {
122 PyErr_SetString (PyExc_ValueError,
123 "can not use negative value for hop_size");
128 if ((sint_t)channels >= 0) {
129 self->channels = channels;
130 } else if ((sint_t)channels < 0) {
131 PyErr_SetString (PyExc_ValueError,
132 "can not use negative value for channels");
136 return (PyObject *) self;
140 Py_source_init (Py_source * self, PyObject * args, PyObject * kwds)
142 self->o = new_aubio_source ( self->uri, self->samplerate, self->hop_size );
143 if (self->o == NULL) {
144 // PyErr_Format(PyExc_RuntimeError, ...) was set above by new_ which called
145 // AUBIO_ERR when failing
148 self->samplerate = aubio_source_get_samplerate ( self->o );
149 if (self->channels == 0) {
150 self->channels = aubio_source_get_channels ( self->o );
152 self->duration = aubio_source_get_duration ( self->o );
154 self->read_to = new_py_fvec(self->hop_size);
155 self->mread_to = new_py_fmat(self->channels, self->hop_size);
161 Py_source_del (Py_source *self, PyObject *unused)
164 del_aubio_source(self->o);
165 free(self->c_mread_to.data);
170 Py_XDECREF(self->read_to);
171 Py_XDECREF(self->mread_to);
172 Py_TYPE(self)->tp_free((PyObject *) self);
176 /* function Py_source_do */
178 Py_source_do(Py_source * self, PyObject * args)
184 Py_INCREF(self->read_to);
185 if (!PyAubio_ArrayToCFvec(self->read_to, &(self->c_read_to))) {
188 /* compute _do function */
189 aubio_source_do (self->o, &(self->c_read_to), &read);
191 outputs = PyTuple_New(2);
192 PyTuple_SetItem( outputs, 0, self->read_to );
193 PyTuple_SetItem( outputs, 1, (PyObject *)PyLong_FromLong(read));
197 /* function Py_source_do_multi */
199 Py_source_do_multi(Py_source * self, PyObject * args)
205 Py_INCREF(self->mread_to);
206 if (!PyAubio_ArrayToCFmat(self->mread_to, &(self->c_mread_to))) {
209 /* compute _do function */
210 aubio_source_do_multi (self->o, &(self->c_mread_to), &read);
212 outputs = PyTuple_New(2);
213 PyTuple_SetItem( outputs, 0, self->mread_to);
214 PyTuple_SetItem( outputs, 1, (PyObject *)PyLong_FromLong(read));
218 static PyMemberDef Py_source_members[] = {
219 {"uri", T_STRING, offsetof (Py_source, uri), READONLY,
220 "path at which the source was created"},
221 {"samplerate", T_INT, offsetof (Py_source, samplerate), READONLY,
222 "samplerate at which the source is viewed"},
223 {"channels", T_INT, offsetof (Py_source, channels), READONLY,
224 "number of channels found in the source"},
225 {"hop_size", T_INT, offsetof (Py_source, hop_size), READONLY,
226 "number of consecutive frames that will be read at each do or do_multi call"},
227 {"duration", T_INT, offsetof (Py_source, duration), READONLY,
228 "total number of frames in the source (estimated)"},
233 Pyaubio_source_get_samplerate (Py_source *self, PyObject *unused)
235 uint_t tmp = aubio_source_get_samplerate (self->o);
236 return (PyObject *)PyLong_FromLong (tmp);
240 Pyaubio_source_get_channels (Py_source *self, PyObject *unused)
242 uint_t tmp = aubio_source_get_channels (self->o);
243 return (PyObject *)PyLong_FromLong (tmp);
247 Pyaubio_source_close (Py_source *self, PyObject *unused)
249 if (aubio_source_close(self->o) != 0) return NULL;
254 Pyaubio_source_seek (Py_source *self, PyObject *args)
259 if (!PyArg_ParseTuple (args, "I", &position)) {
264 PyErr_Format(PyExc_ValueError,
265 "error when seeking in source: can not seek to negative value %d",
270 err = aubio_source_seek(self->o, position);
272 PyErr_SetString (PyExc_ValueError,
273 "error when seeking in source");
279 static char Pyaubio_source_enter_doc[] = "";
280 static PyObject* Pyaubio_source_enter(Py_source *self, PyObject *unused) {
282 return (PyObject*)self;
285 static char Pyaubio_source_exit_doc[] = "";
286 static PyObject* Pyaubio_source_exit(Py_source *self, PyObject *unused) {
287 return Pyaubio_source_close(self, unused);
290 static PyObject* Pyaubio_source_iter(PyObject *self) {
292 return (PyObject*)self;
295 static PyObject* Pyaubio_source_iter_next(Py_source *self) {
296 PyObject *done, *size;
297 if (self->channels == 1) {
298 done = Py_source_do(self, NULL);
300 done = Py_source_do_multi(self, NULL);
302 if (!PyTuple_Check(done)) {
303 PyErr_Format(PyExc_ValueError,
304 "error when reading source: not opened?");
307 size = PyTuple_GetItem(done, 1);
308 if (size != NULL && PyLong_Check(size)) {
309 if (PyLong_AsLong(size) == (long)self->hop_size) {
310 PyObject *vec = PyTuple_GetItem(done, 0);
312 } else if (PyLong_AsLong(size) > 0) {
313 // short read, return a shorter array
314 PyArrayObject *shortread = (PyArrayObject*)PyTuple_GetItem(done, 0);
315 PyArray_Dims newdims;
317 newdims.len = PyArray_NDIM(shortread);
318 newdims.ptr = PyArray_DIMS(shortread);
319 // mono or multiple channels?
320 if (newdims.len == 1) {
321 newdims.ptr[0] = PyLong_AsLong(size);
323 newdims.ptr[1] = PyLong_AsLong(size);
325 reshaped = PyArray_Newshape(shortread, &newdims, NPY_CORDER);
326 Py_DECREF(shortread);
329 PyErr_SetNone(PyExc_StopIteration);
333 PyErr_SetNone(PyExc_StopIteration);
338 static PyMethodDef Py_source_methods[] = {
339 {"get_samplerate", (PyCFunction) Pyaubio_source_get_samplerate,
340 METH_NOARGS, Py_source_get_samplerate_doc},
341 {"get_channels", (PyCFunction) Pyaubio_source_get_channels,
342 METH_NOARGS, Py_source_get_channels_doc},
343 {"do", (PyCFunction) Py_source_do,
344 METH_NOARGS, Py_source_do_doc},
345 {"do_multi", (PyCFunction) Py_source_do_multi,
346 METH_NOARGS, Py_source_do_multi_doc},
347 {"close", (PyCFunction) Pyaubio_source_close,
348 METH_NOARGS, Py_source_close_doc},
349 {"seek", (PyCFunction) Pyaubio_source_seek,
350 METH_VARARGS, Py_source_seek_doc},
351 {"__enter__", (PyCFunction)Pyaubio_source_enter, METH_NOARGS,
352 Pyaubio_source_enter_doc},
353 {"__exit__", (PyCFunction)Pyaubio_source_exit, METH_VARARGS,
354 Pyaubio_source_exit_doc},
355 {NULL} /* sentinel */
358 PyTypeObject Py_sourceType = {
359 PyVarObject_HEAD_INIT (NULL, 0)
363 (destructor) Py_source_del,
373 (ternaryfunc)Py_source_do,
385 (unaryfunc)Pyaubio_source_iter_next,
394 (initproc) Py_source_init,