ref: 25db68c4eefe354f18c58a47eeb5643fc534eb81
dir: /python/ext/py-sink.c/
#include "aubio-types.h"
typedef struct
{
PyObject_HEAD
aubio_sink_t * o;
char_t* uri;
uint_t samplerate;
uint_t channels;
fvec_t write_data;
fmat_t mwrite_data;
} Py_sink;
static char Py_sink_doc[] = ""
" __new__(path, samplerate = 44100, channels = 1)\n"
"\n"
" Create a new sink, opening the given path for writing.\n"
"\n"
" Examples\n"
" --------\n"
"\n"
" Create a new sink at 44100Hz, mono:\n"
"\n"
" >>> sink('/tmp/t.wav')\n"
"\n"
" Create a new sink at 8000Hz, mono:\n"
"\n"
" >>> sink('/tmp/t.wav', samplerate = 8000)\n"
"\n"
" Create a new sink at 32000Hz, stereo:\n"
"\n"
" >>> sink('/tmp/t.wav', samplerate = 32000, channels = 2)\n"
"\n"
" Create a new sink at 32000Hz, 5 channels:\n"
"\n"
" >>> sink('/tmp/t.wav', channels = 5, samplerate = 32000)\n"
"\n"
" __call__(vec, write)\n"
" x(vec,write) <==> x.do(vec, write)\n"
"\n"
" Write vector to sink.\n"
"\n"
" See also\n"
" --------\n"
" aubio.sink.do\n"
"\n";
static char Py_sink_do_doc[] = ""
"x.do(vec, write) <==> x(vec, write)\n"
"\n"
"write monophonic vector to sink";
static char Py_sink_do_multi_doc[] = ""
"x.do_multi(mat, write)\n"
"\n"
"write polyphonic vector to sink";
static char Py_sink_close_doc[] = ""
"x.close()\n"
"\n"
"close this sink now";
static PyObject *
Py_sink_new (PyTypeObject * pytype, PyObject * args, PyObject * kwds)
{
Py_sink *self;
char_t* uri = NULL;
uint_t samplerate = 0;
uint_t channels = 0;
static char *kwlist[] = { "uri", "samplerate", "channels", NULL };
if (!PyArg_ParseTupleAndKeywords (args, kwds, "|sII", kwlist,
&uri, &samplerate, &channels)) {
return NULL;
}
self = (Py_sink *) pytype->tp_alloc (pytype, 0);
if (self == NULL) {
return NULL;
}
self->uri = NULL;
if (uri != NULL) {
self->uri = (char_t *)malloc(sizeof(char_t) * (strnlen(uri, PATH_MAX) + 1));
strncpy(self->uri, uri, strnlen(uri, PATH_MAX) + 1);
}
self->samplerate = Py_aubio_default_samplerate;
if (samplerate != 0) {
self->samplerate = samplerate;
}
self->channels = 1;
if (channels != 0) {
self->channels = channels;
}
return (PyObject *) self;
}
static int
Py_sink_init (Py_sink * self, PyObject * args, PyObject * kwds)
{
self->o = new_aubio_sink ( self->uri, 0 );
if (self->o == NULL) {
// error string was set in new_aubio_sink
return -1;
}
if (aubio_sink_preset_channels(self->o, self->channels) != 0) {
// error string was set in aubio_sink_preset_channels
return -1;
}
if (aubio_sink_preset_samplerate(self->o, self->samplerate) != 0) {
// error string was set in aubio_sink_preset_samplerate
return -1;
}
self->samplerate = aubio_sink_get_samplerate ( self->o );
self->channels = aubio_sink_get_channels ( self->o );
return 0;
}
static void
Py_sink_del (Py_sink *self, PyObject *unused)
{
del_aubio_sink(self->o);
free(self->mwrite_data.data);
if (self->uri) {
free(self->uri);
}
Py_TYPE(self)->tp_free((PyObject *) self);
}
/* function Py_sink_do */
static PyObject *
Py_sink_do(Py_sink * self, PyObject * args)
{
/* input vectors python prototypes */
PyObject * write_data_obj;
/* input vectors prototypes */
uint_t write;
if (!PyArg_ParseTuple (args, "OI", &write_data_obj, &write)) {
return NULL;
}
/* input vectors parsing */
if (!PyAubio_ArrayToCFvec(write_data_obj, &(self->write_data))) {
return NULL;
}
/* compute _do function */
aubio_sink_do (self->o, &(self->write_data), write);
Py_RETURN_NONE;
}
/* function Py_sink_do_multi */
static PyObject *
Py_sink_do_multi(Py_sink * self, PyObject * args)
{
/* input vectors python prototypes */
PyObject * write_data_obj;
/* input vectors prototypes */
uint_t write;
if (!PyArg_ParseTuple (args, "OI", &write_data_obj, &write)) {
return NULL;
}
/* input vectors parsing */
if (!PyAubio_ArrayToCFmat(write_data_obj, &(self->mwrite_data))) {
return NULL;
}
/* compute _do function */
aubio_sink_do_multi (self->o, &(self->mwrite_data), write);
Py_RETURN_NONE;
}
static PyMemberDef Py_sink_members[] = {
{"uri", T_STRING, offsetof (Py_sink, uri), READONLY,
"path at which the sink was created"},
{"samplerate", T_INT, offsetof (Py_sink, samplerate), READONLY,
"samplerate at which the sink was created"},
{"channels", T_INT, offsetof (Py_sink, channels), READONLY,
"number of channels with which the sink was created"},
{ NULL } // sentinel
};
static PyObject *
Pyaubio_sink_close (Py_sink *self, PyObject *unused)
{
aubio_sink_close (self->o);
Py_RETURN_NONE;
}
static char Pyaubio_sink_enter_doc[] = "";
static PyObject* Pyaubio_sink_enter(Py_sink *self, PyObject *unused) {
Py_INCREF(self);
return (PyObject*)self;
}
static char Pyaubio_sink_exit_doc[] = "";
static PyObject* Pyaubio_sink_exit(Py_sink *self, PyObject *unused) {
return Pyaubio_sink_close(self, unused);
}
static PyMethodDef Py_sink_methods[] = {
{"do", (PyCFunction) Py_sink_do, METH_VARARGS, Py_sink_do_doc},
{"do_multi", (PyCFunction) Py_sink_do_multi, METH_VARARGS, Py_sink_do_multi_doc},
{"close", (PyCFunction) Pyaubio_sink_close, METH_NOARGS, Py_sink_close_doc},
{"__enter__", (PyCFunction)Pyaubio_sink_enter, METH_NOARGS,
Pyaubio_sink_enter_doc},
{"__exit__", (PyCFunction)Pyaubio_sink_exit, METH_VARARGS,
Pyaubio_sink_exit_doc},
{NULL} /* sentinel */
};
PyTypeObject Py_sinkType = {
PyVarObject_HEAD_INIT (NULL, 0)
"aubio.sink",
sizeof (Py_sink),
0,
(destructor) Py_sink_del,
0,
0,
0,
0,
0,
0,
0,
0,
0,
(ternaryfunc)Py_sink_do,
0,
0,
0,
0,
Py_TPFLAGS_DEFAULT,
Py_sink_doc,
0,
0,
0,
0,
0,
0,
Py_sink_methods,
Py_sink_members,
0,
0,
0,
0,
0,
0,
(initproc) Py_sink_init,
0,
Py_sink_new,
0,
0,
0,
0,
0,
0,
0,
0,
0,
};