ext/py-source.c: add channels
[aubio.git] / python / ext / py-source.c
1 #include "aubiowraphell.h"
2
3 typedef struct
4 {
5   PyObject_HEAD
6   aubio_source_t * o;
7   char_t* uri;
8   uint_t samplerate;
9   uint_t channels;
10   uint_t hop_size;
11 } Py_source;
12
13 static char Py_source_doc[] = ""
14 "   __new__(path, samplerate = 0, hop_size = 512, channels = 1)\n"
15 "\n"
16 "       Create a new source, opening the given path for reading.\n"
17 "\n"
18 "       Examples\n"
19 "       --------\n"
20 "\n"
21 "       Create a new source, using the original samplerate, with hop_size = 512:\n"
22 "\n"
23 "       >>> source('/tmp/t.wav')\n"
24 "\n"
25 "       Create a new source, resampling the original to 8000Hz:\n"
26 "\n"
27 "       >>> source('/tmp/t.wav', samplerate = 8000)\n"
28 "\n"
29 "       Create a new source, resampling it at 32000Hz, hop_size = 32:\n"
30 "\n"
31 "       >>> source('/tmp/t.wav', samplerate = 32000, hop_size = 32)\n"
32 "\n"
33 "       Create a new source, using its original samplerate:\n"
34 "\n"
35 "       >>> source('/tmp/t.wav', samplerate = 0)\n"
36 "\n"
37 "   __call__()\n"
38 "       vec, read = x() <==> vec, read = x.do()\n"
39 "\n"
40 "       Read vector from source.\n"
41 "\n"
42 "       See also\n"
43 "       --------\n"
44 "       aubio.source.do\n"
45 "\n";
46
47 static char Py_source_get_samplerate_doc[] = ""
48 "x.get_samplerate() -> source samplerate\n"
49 "\n"
50 "Get samplerate of source.";
51
52 static char Py_source_get_channels_doc[] = ""
53 "x.get_channels() -> number of channels\n"
54 "\n"
55 "Get number of channels in source.";
56
57 static char Py_source_do_doc[] = ""
58 "vec, read = x.do() <==> vec, read = x()\n"
59 "\n"
60 "Read monophonic vector from source.";
61
62 static char Py_source_do_multi_doc[] = ""
63 "mat, read = x.do_multi()\n"
64 "\n"
65 "Read polyphonic vector from source.";
66
67 static char Py_source_close_doc[] = ""
68 "x.close()\n"
69 "\n"
70 "Close this source now.";
71
72 static PyObject *
73 Py_source_new (PyTypeObject * pytype, PyObject * args, PyObject * kwds)
74 {
75   Py_source *self;
76   char_t* uri = NULL;
77   uint_t samplerate = 0;
78   uint_t hop_size = 0;
79   uint_t channels = 0;
80   static char *kwlist[] = { "uri", "samplerate", "hop_size", "channels", NULL };
81
82   if (!PyArg_ParseTupleAndKeywords (args, kwds, "|sIII", kwlist,
83           &uri, &samplerate, &hop_size, &channels)) {
84     return NULL;
85   }
86
87   self = (Py_source *) pytype->tp_alloc (pytype, 0);
88
89   if (self == NULL) {
90     return NULL;
91   }
92
93   self->uri = "none";
94   if (uri != NULL) {
95     self->uri = uri;
96   }
97
98   self->samplerate = 0;
99   if ((sint_t)samplerate > 0) {
100     self->samplerate = samplerate;
101   } else if ((sint_t)samplerate < 0) {
102     PyErr_SetString (PyExc_ValueError,
103         "can not use negative value for samplerate");
104     return NULL;
105   }
106
107   self->hop_size = Py_default_vector_length / 2;
108   if ((sint_t)hop_size > 0) {
109     self->hop_size = hop_size;
110   } else if ((sint_t)hop_size < 0) {
111     PyErr_SetString (PyExc_ValueError,
112         "can not use negative value for hop_size");
113     return NULL;
114   }
115
116   self->channels = 1;
117   if ((sint_t)channels >= 0) {
118     self->channels = channels;
119   } else if ((sint_t)channels < 0) {
120     PyErr_SetString (PyExc_ValueError,
121         "can not use negative value for channels");
122     return NULL;
123   }
124
125   return (PyObject *) self;
126 }
127
128 static int
129 Py_source_init (Py_source * self, PyObject * args, PyObject * kwds)
130 {
131   self->o = new_aubio_source ( self->uri, self->samplerate, self->hop_size );
132   if (self->o == NULL) {
133     char_t errstr[30 + strlen(self->uri)];
134     sprintf(errstr, "error creating source with %s", self->uri);
135     PyErr_SetString (PyExc_StandardError, errstr);
136     return -1;
137   }
138   self->samplerate = aubio_source_get_samplerate ( self->o );
139   if (self->channels == 0) {
140     self->channels = aubio_source_get_channels ( self->o );
141   }
142
143   return 0;
144 }
145
146 AUBIO_DEL(source)
147
148 /* function Py_source_do */
149 static PyObject *
150 Py_source_do(Py_source * self, PyObject * args)
151 {
152
153
154   /* output vectors prototypes */
155   fvec_t* read_to;
156   uint_t read;
157
158
159
160
161
162
163   /* creating output read_to as a new_fvec of length self->hop_size */
164   read_to = new_fvec (self->hop_size);
165   read = 0;
166
167
168   /* compute _do function */
169   aubio_source_do (self->o, read_to, &read);
170
171   PyObject *outputs = PyList_New(0);
172   PyList_Append( outputs, (PyObject *)PyAubio_CFvecToArray (read_to));
173   //del_fvec (read_to);
174   PyList_Append( outputs, (PyObject *)PyInt_FromLong (read));
175   return outputs;
176 }
177
178 /* function Py_source_do_multi */
179 static PyObject *
180 Py_source_do_multi(Py_source * self, PyObject * args)
181 {
182
183
184   /* output vectors prototypes */
185   fmat_t* read_to;
186   uint_t read;
187
188
189
190
191
192
193   /* creating output read_to as a new_fvec of length self->hop_size */
194   read_to = new_fmat (self->channels, self->hop_size);
195   read = 0;
196
197
198   /* compute _do function */
199   aubio_source_do_multi (self->o, read_to, &read);
200
201   PyObject *outputs = PyList_New(0);
202   PyList_Append( outputs, (PyObject *)PyAubio_CFmatToArray (read_to));
203   //del_fvec (read_to);
204   PyList_Append( outputs, (PyObject *)PyInt_FromLong (read));
205   return outputs;
206 }
207
208 AUBIO_MEMBERS_START(source)
209   {"uri", T_STRING, offsetof (Py_source, uri), READONLY,
210     "path at which the source was created"},
211   {"samplerate", T_INT, offsetof (Py_source, samplerate), READONLY,
212     "samplerate at which the source is viewed"},
213   {"channels", T_INT, offsetof (Py_source, channels), READONLY,
214     "number of channels found in the source"},
215   {"hop_size", T_INT, offsetof (Py_source, hop_size), READONLY,
216     "number of consecutive frames that will be read at each do or do_multi call"},
217 AUBIO_MEMBERS_STOP(source)
218
219
220 static PyObject *
221 Pyaubio_source_get_samplerate (Py_source *self, PyObject *unused)
222 {
223   uint_t tmp = aubio_source_get_samplerate (self->o);
224   return (PyObject *)PyInt_FromLong (tmp);
225 }
226
227 static PyObject *
228 Pyaubio_source_get_channels (Py_source *self, PyObject *unused)
229 {
230   uint_t tmp = aubio_source_get_channels (self->o);
231   return (PyObject *)PyInt_FromLong (tmp);
232 }
233
234 static PyObject *
235 Pyaubio_source_close (Py_source *self, PyObject *unused)
236 {
237   aubio_source_close (self->o);
238   Py_RETURN_NONE;
239 }
240
241 static PyMethodDef Py_source_methods[] = {
242   {"get_samplerate", (PyCFunction) Pyaubio_source_get_samplerate,
243     METH_NOARGS, Py_source_get_samplerate_doc},
244   {"get_channels", (PyCFunction) Pyaubio_source_get_channels,
245     METH_NOARGS, Py_source_get_channels_doc},
246   {"do", (PyCFunction) Py_source_do,
247     METH_NOARGS, Py_source_do_doc},
248   {"do_multi", (PyCFunction) Py_source_do_multi,
249     METH_NOARGS, Py_source_do_multi_doc},
250   {"close", (PyCFunction) Pyaubio_source_close,
251     METH_NOARGS, Py_source_close_doc},
252   {NULL} /* sentinel */
253 };
254
255 AUBIO_TYPEOBJECT(source, "aubio.source")