source: python/ext/py-source.c @ 5652a8c

feature/autosinkfeature/cnnfeature/cnn_orgfeature/constantqfeature/crepefeature/crepe_orgfeature/pitchshiftfeature/pydocstringsfeature/timestretchfix/ffmpeg5pitchshiftsamplertimestretchyinfft+
Last change on this file since 5652a8c was 5652a8c, checked in by Paul Brossier <piem@piem.org>, 8 years ago

ext/: no more hell, use plain c

  • Property mode set to 100644
File size: 7.4 KB
RevLine 
[5652a8c]1#include "aubio-types.h"
[d27634d]2
3typedef struct
4{
5  PyObject_HEAD
6  aubio_source_t * o;
7  char_t* uri;
8  uint_t samplerate;
[1164bcdf]9  uint_t channels;
[d27634d]10  uint_t hop_size;
[a28dab6]11  fvec_t *read_to;
12  fmat_t *mread_to;
[d27634d]13} Py_source;
14
[a79ec76]15static char Py_source_doc[] = ""
16"   __new__(path, samplerate = 0, hop_size = 512, channels = 1)\n"
17"\n"
18"       Create a new source, opening the given path for reading.\n"
19"\n"
20"       Examples\n"
21"       --------\n"
22"\n"
23"       Create a new source, using the original samplerate, with hop_size = 512:\n"
24"\n"
25"       >>> source('/tmp/t.wav')\n"
26"\n"
27"       Create a new source, resampling the original to 8000Hz:\n"
28"\n"
29"       >>> source('/tmp/t.wav', samplerate = 8000)\n"
30"\n"
31"       Create a new source, resampling it at 32000Hz, hop_size = 32:\n"
32"\n"
33"       >>> source('/tmp/t.wav', samplerate = 32000, hop_size = 32)\n"
34"\n"
35"       Create a new source, using its original samplerate:\n"
36"\n"
37"       >>> source('/tmp/t.wav', samplerate = 0)\n"
38"\n"
39"   __call__()\n"
40"       vec, read = x() <==> vec, read = x.do()\n"
41"\n"
42"       Read vector from source.\n"
43"\n"
44"       See also\n"
45"       --------\n"
46"       aubio.source.do\n"
47"\n";
48
49static char Py_source_get_samplerate_doc[] = ""
50"x.get_samplerate() -> source samplerate\n"
51"\n"
52"Get samplerate of source.";
53
54static char Py_source_get_channels_doc[] = ""
55"x.get_channels() -> number of channels\n"
56"\n"
57"Get number of channels in source.";
58
59static char Py_source_do_doc[] = ""
60"vec, read = x.do() <==> vec, read = x()\n"
61"\n"
62"Read monophonic vector from source.";
63
64static char Py_source_do_multi_doc[] = ""
65"mat, read = x.do_multi()\n"
66"\n"
67"Read polyphonic vector from source.";
68
69static char Py_source_close_doc[] = ""
70"x.close()\n"
71"\n"
72"Close this source now.";
[d27634d]73
[11b49d7]74static char Py_source_seek_doc[] = ""
75"x.seek(position)\n"
76"\n"
77"Seek to resampled frame position.";
78
[d27634d]79static PyObject *
80Py_source_new (PyTypeObject * pytype, PyObject * args, PyObject * kwds)
81{
82  Py_source *self;
83  char_t* uri = NULL;
84  uint_t samplerate = 0;
85  uint_t hop_size = 0;
[24931d5]86  uint_t channels = 0;
87  static char *kwlist[] = { "uri", "samplerate", "hop_size", "channels", NULL };
[d27634d]88
[24931d5]89  if (!PyArg_ParseTupleAndKeywords (args, kwds, "|sIII", kwlist,
90          &uri, &samplerate, &hop_size, &channels)) {
[d27634d]91    return NULL;
92  }
93
94  self = (Py_source *) pytype->tp_alloc (pytype, 0);
95
96  if (self == NULL) {
97    return NULL;
98  }
99
100  self->uri = "none";
101  if (uri != NULL) {
102    self->uri = uri;
103  }
104
[2f9af5d]105  self->samplerate = 0;
[18e22f9]106  if ((sint_t)samplerate > 0) {
[d27634d]107    self->samplerate = samplerate;
[18e22f9]108  } else if ((sint_t)samplerate < 0) {
109    PyErr_SetString (PyExc_ValueError,
110        "can not use negative value for samplerate");
111    return NULL;
[d27634d]112  }
113
114  self->hop_size = Py_default_vector_length / 2;
[18e22f9]115  if ((sint_t)hop_size > 0) {
[d27634d]116    self->hop_size = hop_size;
[18e22f9]117  } else if ((sint_t)hop_size < 0) {
118    PyErr_SetString (PyExc_ValueError,
119        "can not use negative value for hop_size");
120    return NULL;
[d27634d]121  }
122
[24931d5]123  self->channels = 1;
124  if ((sint_t)channels >= 0) {
125    self->channels = channels;
126  } else if ((sint_t)channels < 0) {
127    PyErr_SetString (PyExc_ValueError,
128        "can not use negative value for channels");
129    return NULL;
130  }
131
[d27634d]132  return (PyObject *) self;
133}
134
135static int
136Py_source_init (Py_source * self, PyObject * args, PyObject * kwds)
137{
138  self->o = new_aubio_source ( self->uri, self->samplerate, self->hop_size );
139  if (self->o == NULL) {
[24931d5]140    char_t errstr[30 + strlen(self->uri)];
141    sprintf(errstr, "error creating source with %s", self->uri);
[ac7e49b]142    PyErr_SetString (PyExc_RuntimeError, errstr);
[d27634d]143    return -1;
144  }
145  self->samplerate = aubio_source_get_samplerate ( self->o );
[24931d5]146  if (self->channels == 0) {
147    self->channels = aubio_source_get_channels ( self->o );
148  }
[d27634d]149
[a28dab6]150  self->read_to = new_fvec(self->hop_size);
151  self->mread_to = new_fmat (self->channels, self->hop_size);
152
[d27634d]153  return 0;
154}
155
[a28dab6]156static void
157Py_source_del (Py_source *self, PyObject *unused)
158{
159  del_aubio_source(self->o);
160  del_fvec(self->read_to);
161  del_fmat(self->mread_to);
[5c1200a]162  Py_TYPE(self)->tp_free((PyObject *) self);
[a28dab6]163}
164
[d27634d]165
166/* function Py_source_do */
[1164bcdf]167static PyObject *
[d27634d]168Py_source_do(Py_source * self, PyObject * args)
169{
170
171
172  /* output vectors prototypes */
173  uint_t read;
174
175
176
177
178
[1164bcdf]179
[d27634d]180  /* creating output read_to as a new_fvec of length self->hop_size */
181  read = 0;
182
183
184  /* compute _do function */
[a28dab6]185  aubio_source_do (self->o, self->read_to, &read);
[d27634d]186
187  PyObject *outputs = PyList_New(0);
[a28dab6]188  PyList_Append( outputs, (PyObject *)PyAubio_CFvecToArray (self->read_to));
[5c1200a]189  PyList_Append( outputs, (PyObject *)PyLong_FromLong(read));
[d27634d]190  return outputs;
191}
192
[1164bcdf]193/* function Py_source_do_multi */
194static PyObject *
195Py_source_do_multi(Py_source * self, PyObject * args)
196{
197
198
199  /* output vectors prototypes */
200  uint_t read;
201
202
203
204
205
206
[a28dab6]207  /* creating output mread_to as a new_fvec of length self->hop_size */
[1164bcdf]208  read = 0;
209
210
211  /* compute _do function */
[a28dab6]212  aubio_source_do_multi (self->o, self->mread_to, &read);
[1164bcdf]213
214  PyObject *outputs = PyList_New(0);
[a28dab6]215  PyList_Append( outputs, (PyObject *)PyAubio_CFmatToArray (self->mread_to));
[5c1200a]216  PyList_Append( outputs, (PyObject *)PyLong_FromLong(read));
[1164bcdf]217  return outputs;
218}
219
[5652a8c]220static PyMemberDef Py_source_members[] = {
[a79ec76]221  {"uri", T_STRING, offsetof (Py_source, uri), READONLY,
222    "path at which the source was created"},
223  {"samplerate", T_INT, offsetof (Py_source, samplerate), READONLY,
224    "samplerate at which the source is viewed"},
225  {"channels", T_INT, offsetof (Py_source, channels), READONLY,
226    "number of channels found in the source"},
227  {"hop_size", T_INT, offsetof (Py_source, hop_size), READONLY,
228    "number of consecutive frames that will be read at each do or do_multi call"},
[5652a8c]229  { NULL } // sentinel
230};
[d27634d]231
232static PyObject *
233Pyaubio_source_get_samplerate (Py_source *self, PyObject *unused)
234{
235  uint_t tmp = aubio_source_get_samplerate (self->o);
[5c1200a]236  return (PyObject *)PyLong_FromLong (tmp);
[d27634d]237}
238
239static PyObject *
240Pyaubio_source_get_channels (Py_source *self, PyObject *unused)
241{
242  uint_t tmp = aubio_source_get_channels (self->o);
[5c1200a]243  return (PyObject *)PyLong_FromLong (tmp);
[d27634d]244}
245
[7b56229]246static PyObject *
247Pyaubio_source_close (Py_source *self, PyObject *unused)
248{
[3cc3fd8]249  aubio_source_close (self->o);
[7b56229]250  Py_RETURN_NONE;
251}
252
[11b49d7]253static PyObject *
254Pyaubio_source_seek (Py_source *self, PyObject *args)
255{
256  uint_t err = 0;
257
258  uint_t position;
259  if (!PyArg_ParseTuple (args, "I", &position)) {
260    return NULL;
261  }
262
263  err = aubio_source_seek(self->o, position);
264  if (err != 0) {
265    PyErr_SetString (PyExc_ValueError,
266        "error when seeking in source");
267    return NULL;
268  }
269  Py_RETURN_NONE;
270}
271
[d27634d]272static PyMethodDef Py_source_methods[] = {
273  {"get_samplerate", (PyCFunction) Pyaubio_source_get_samplerate,
[a79ec76]274    METH_NOARGS, Py_source_get_samplerate_doc},
[d27634d]275  {"get_channels", (PyCFunction) Pyaubio_source_get_channels,
[a79ec76]276    METH_NOARGS, Py_source_get_channels_doc},
277  {"do", (PyCFunction) Py_source_do,
278    METH_NOARGS, Py_source_do_doc},
[1164bcdf]279  {"do_multi", (PyCFunction) Py_source_do_multi,
[a79ec76]280    METH_NOARGS, Py_source_do_multi_doc},
[7b56229]281  {"close", (PyCFunction) Pyaubio_source_close,
[a79ec76]282    METH_NOARGS, Py_source_close_doc},
[11b49d7]283  {"seek", (PyCFunction) Pyaubio_source_seek,
284    METH_VARARGS, Py_source_seek_doc},
[d27634d]285  {NULL} /* sentinel */
286};
287
[5652a8c]288PyTypeObject Py_sourceType = {
289  PyVarObject_HEAD_INIT (NULL, 0)
290  "aubio.source",
291  sizeof (Py_source),
292  0,
293  (destructor) Py_source_del,
294  0,
295  0,
296  0,
297  0,
298  0,
299  0,
300  0,
301  0,
302  0,
303  (ternaryfunc)Py_source_do,
304  0,
305  0,
306  0,
307  0,
308  Py_TPFLAGS_DEFAULT,
309  Py_source_doc,
310  0,
311  0,
312  0,
313  0,
314  0,
315  0,
316  Py_source_methods,
317  Py_source_members,
318  0,
319  0,
320  0,
321  0,
322  0,
323  0,
324  (initproc) Py_source_init,
325  0,
326  Py_source_new,
327};
Note: See TracBrowser for help on using the repository browser.