source: python/ext/aubiomodule.c @ 6203a70

feature/constantq
Last change on this file since 6203a70 was b532275, checked in by Paul Brossier <piem@piem.org>, 6 years ago

python/ext/py-musicutils.*: add shift(fvec) and ishift(fvec)

  • Property mode set to 100644
File size: 9.2 KB
Line 
1#define PY_AUBIO_MODULE_MAIN
2#include "aubio-types.h"
3#include "py-musicutils.h"
4
5// this dummy macro is used to convince windows that a string passed as -D flag
6// is just that, a string, and not a double.
7#define REDEFINESTRING(x) #x
8#define DEFINEDSTRING(x) REDEFINESTRING(x)
9
10static char aubio_module_doc[] = "Python module for the aubio library";
11
12static char Py_alpha_norm_doc[] = ""
13"alpha_norm(fvec, integer) -> float\n"
14"\n"
15"Compute alpha normalisation factor on vector, given alpha\n"
16"\n"
17"Example\n"
18"-------\n"
19"\n"
20">>> b = alpha_norm(a, 9)";
21
22static char Py_bintomidi_doc[] = ""
23"bintomidi(float, samplerate = integer, fftsize = integer) -> float\n"
24"\n"
25"Convert bin (float) to midi (float), given the sampling rate and the FFT size\n"
26"\n"
27"Example\n"
28"-------\n"
29"\n"
30">>> midi = bintomidi(float, samplerate = 44100, fftsize = 1024)";
31
32static char Py_miditobin_doc[] = ""
33"miditobin(float, samplerate = integer, fftsize = integer) -> float\n"
34"\n"
35"Convert midi (float) to bin (float), given the sampling rate and the FFT size\n"
36"\n"
37"Example\n"
38"-------\n"
39"\n"
40">>> bin = miditobin(midi, samplerate = 44100, fftsize = 1024)";
41
42static char Py_bintofreq_doc[] = ""
43"bintofreq(float, samplerate = integer, fftsize = integer) -> float\n"
44"\n"
45"Convert bin number (float) in frequency (Hz), given the sampling rate and the FFT size\n"
46"\n"
47"Example\n"
48"-------\n"
49"\n"
50">>> freq = bintofreq(bin, samplerate = 44100, fftsize = 1024)";
51
52static char Py_freqtobin_doc[] = ""
53"freqtobin(float, samplerate = integer, fftsize = integer) -> float\n"
54"\n"
55"Convert frequency (Hz) in bin number (float), given the sampling rate and the FFT size\n"
56"\n"
57"Example\n"
58"-------\n"
59"\n"
60">>> bin = freqtobin(freq, samplerate = 44100, fftsize = 1024)";
61
62static char Py_zero_crossing_rate_doc[] = ""
63"zero_crossing_rate(fvec) -> float\n"
64"\n"
65"Compute Zero crossing rate of a vector\n"
66"\n"
67"Example\n"
68"-------\n"
69"\n"
70">>> z = zero_crossing_rate(a)";
71
72static char Py_min_removal_doc[] = ""
73"min_removal(fvec) -> float\n"
74"\n"
75"Remove the minimum value of a vector, in-place modification\n"
76"\n"
77"Example\n"
78"-------\n"
79"\n"
80">>> min_removal(a)";
81
82extern void add_ufuncs ( PyObject *m );
83extern int generated_types_ready(void);
84
85static PyObject *
86Py_alpha_norm (PyObject * self, PyObject * args)
87{
88  PyObject *input;
89  fvec_t vec;
90  smpl_t alpha;
91  PyObject *result;
92
93  if (!PyArg_ParseTuple (args, "O" AUBIO_NPY_SMPL_CHR ":alpha_norm", &input, &alpha)) {
94    return NULL;
95  }
96
97  if (input == NULL) {
98    return NULL;
99  }
100
101  if (!PyAubio_ArrayToCFvec(input, &vec)) {
102    return NULL;
103  }
104
105  // compute the function
106  result = Py_BuildValue (AUBIO_NPY_SMPL_CHR, fvec_alpha_norm (&vec, alpha));
107  if (result == NULL) {
108    return NULL;
109  }
110
111  return result;
112}
113
114static PyObject *
115Py_bintomidi (PyObject * self, PyObject * args)
116{
117  smpl_t input, samplerate, fftsize;
118  smpl_t output;
119
120  if (!PyArg_ParseTuple (args, "|" AUBIO_NPY_SMPL_CHR AUBIO_NPY_SMPL_CHR AUBIO_NPY_SMPL_CHR , &input, &samplerate, &fftsize)) {
121    return NULL;
122  }
123
124  output = aubio_bintomidi (input, samplerate, fftsize);
125
126  return (PyObject *)PyFloat_FromDouble (output);
127}
128
129static PyObject *
130Py_miditobin (PyObject * self, PyObject * args)
131{
132  smpl_t input, samplerate, fftsize;
133  smpl_t output;
134
135  if (!PyArg_ParseTuple (args, "|" AUBIO_NPY_SMPL_CHR AUBIO_NPY_SMPL_CHR AUBIO_NPY_SMPL_CHR , &input, &samplerate, &fftsize)) {
136    return NULL;
137  }
138
139  output = aubio_miditobin (input, samplerate, fftsize);
140
141  return (PyObject *)PyFloat_FromDouble (output);
142}
143
144static PyObject *
145Py_bintofreq (PyObject * self, PyObject * args)
146{
147  smpl_t input, samplerate, fftsize;
148  smpl_t output;
149
150  if (!PyArg_ParseTuple (args, "|" AUBIO_NPY_SMPL_CHR AUBIO_NPY_SMPL_CHR AUBIO_NPY_SMPL_CHR, &input, &samplerate, &fftsize)) {
151    return NULL;
152  }
153
154  output = aubio_bintofreq (input, samplerate, fftsize);
155
156  return (PyObject *)PyFloat_FromDouble (output);
157}
158
159static PyObject *
160Py_freqtobin (PyObject * self, PyObject * args)
161{
162  smpl_t input, samplerate, fftsize;
163  smpl_t output;
164
165  if (!PyArg_ParseTuple (args, "|" AUBIO_NPY_SMPL_CHR AUBIO_NPY_SMPL_CHR AUBIO_NPY_SMPL_CHR, &input, &samplerate, &fftsize)) {
166    return NULL;
167  }
168
169  output = aubio_freqtobin (input, samplerate, fftsize);
170
171  return (PyObject *)PyFloat_FromDouble (output);
172}
173
174static PyObject *
175Py_zero_crossing_rate (PyObject * self, PyObject * args)
176{
177  PyObject *input;
178  fvec_t vec;
179  PyObject *result;
180
181  if (!PyArg_ParseTuple (args, "O:zero_crossing_rate", &input)) {
182    return NULL;
183  }
184
185  if (input == NULL) {
186    return NULL;
187  }
188
189  if (!PyAubio_ArrayToCFvec(input, &vec)) {
190    return NULL;
191  }
192
193  // compute the function
194  result = Py_BuildValue (AUBIO_NPY_SMPL_CHR, aubio_zero_crossing_rate (&vec));
195  if (result == NULL) {
196    return NULL;
197  }
198
199  return result;
200}
201
202static PyObject *
203Py_min_removal(PyObject * self, PyObject * args)
204{
205  PyObject *input;
206  fvec_t vec;
207
208  if (!PyArg_ParseTuple (args, "O:min_removal", &input)) {
209    return NULL;
210  }
211
212  if (input == NULL) {
213    return NULL;
214  }
215
216  if (!PyAubio_ArrayToCFvec(input, &vec)) {
217    return NULL;
218  }
219
220  // compute the function
221  fvec_min_removal (&vec);
222
223  // since this function does not return, we could return None
224  //Py_RETURN_NONE;
225  // however it is convenient to return the modified vector
226  return (PyObject *) PyAubio_CFvecToArray(&vec);
227  // or even without converting it back to an array
228  //Py_INCREF(vec);
229  //return (PyObject *)vec;
230}
231
232static PyMethodDef aubio_methods[] = {
233  {"bintomidi", Py_bintomidi, METH_VARARGS, Py_bintomidi_doc},
234  {"miditobin", Py_miditobin, METH_VARARGS, Py_miditobin_doc},
235  {"bintofreq", Py_bintofreq, METH_VARARGS, Py_bintofreq_doc},
236  {"freqtobin", Py_freqtobin, METH_VARARGS, Py_freqtobin_doc},
237  {"alpha_norm", Py_alpha_norm, METH_VARARGS, Py_alpha_norm_doc},
238  {"zero_crossing_rate", Py_zero_crossing_rate, METH_VARARGS, Py_zero_crossing_rate_doc},
239  {"min_removal", Py_min_removal, METH_VARARGS, Py_min_removal_doc},
240  {"level_lin", Py_aubio_level_lin, METH_VARARGS, Py_aubio_level_lin_doc},
241  {"db_spl", Py_aubio_db_spl, METH_VARARGS, Py_aubio_db_spl_doc},
242  {"silence_detection", Py_aubio_silence_detection, METH_VARARGS, Py_aubio_silence_detection_doc},
243  {"level_detection", Py_aubio_level_detection, METH_VARARGS, Py_aubio_level_detection_doc},
244  {"window", Py_aubio_window, METH_VARARGS, Py_aubio_window_doc},
245  {"shift", Py_aubio_shift, METH_VARARGS, Py_aubio_shift_doc},
246  {"ishift", Py_aubio_ishift, METH_VARARGS, Py_aubio_ishift_doc},
247  {NULL, NULL, 0, NULL} /* Sentinel */
248};
249
250#if PY_MAJOR_VERSION >= 3
251// Python3 module definition
252static struct PyModuleDef moduledef = {
253   PyModuleDef_HEAD_INIT,
254   "_aubio",          /* m_name */
255   aubio_module_doc,  /* m_doc */
256   -1,                /* m_size */
257   aubio_methods,     /* m_methods */
258   NULL,              /* m_reload */
259   NULL,              /* m_traverse */
260   NULL,              /* m_clear */
261   NULL,              /* m_free */
262};
263#endif
264
265void
266aubio_log_function(int level, const char *message, void *data)
267{
268  // remove trailing \n
269  char *pos;
270  if ((pos=strchr(message, '\n')) != NULL) {
271        *pos = '\0';
272  }
273  // warning or error
274  if (level == AUBIO_LOG_ERR) {
275    PyErr_Format(PyExc_RuntimeError, "%s", message);
276  } else {
277    PyErr_WarnEx(PyExc_UserWarning, message, 1);
278  }
279}
280
281static PyObject *
282initaubio (void)
283{
284  PyObject *m = NULL;
285  int err;
286
287  // fvec is defined in __init__.py
288  if (   (PyType_Ready (&Py_cvecType) < 0)
289      || (PyType_Ready (&Py_filterType) < 0)
290      || (PyType_Ready (&Py_filterbankType) < 0)
291      || (PyType_Ready (&Py_fftType) < 0)
292      || (PyType_Ready (&Py_pvocType) < 0)
293      || (PyType_Ready (&Py_sourceType) < 0)
294      || (PyType_Ready (&Py_sinkType) < 0)
295      // generated objects
296      || (generated_types_ready() < 0 )
297  ) {
298    return m;
299  }
300
301#if PY_MAJOR_VERSION >= 3
302  m = PyModule_Create(&moduledef);
303#else
304  m = Py_InitModule3 ("_aubio", aubio_methods, aubio_module_doc);
305#endif
306
307  if (m == NULL) {
308    return m;
309  }
310
311  err = _import_array ();
312  if (err != 0) {
313    fprintf (stderr,
314        "Unable to import Numpy array from aubio module (error %d)\n", err);
315  }
316
317  Py_INCREF (&Py_cvecType);
318  PyModule_AddObject (m, "cvec", (PyObject *) & Py_cvecType);
319  Py_INCREF (&Py_filterType);
320  PyModule_AddObject (m, "digital_filter", (PyObject *) & Py_filterType);
321  Py_INCREF (&Py_filterbankType);
322  PyModule_AddObject (m, "filterbank", (PyObject *) & Py_filterbankType);
323  Py_INCREF (&Py_fftType);
324  PyModule_AddObject (m, "fft", (PyObject *) & Py_fftType);
325  Py_INCREF (&Py_pvocType);
326  PyModule_AddObject (m, "pvoc", (PyObject *) & Py_pvocType);
327  Py_INCREF (&Py_sourceType);
328  PyModule_AddObject (m, "source", (PyObject *) & Py_sourceType);
329  Py_INCREF (&Py_sinkType);
330  PyModule_AddObject (m, "sink", (PyObject *) & Py_sinkType);
331
332  PyModule_AddStringConstant(m, "float_type", AUBIO_NPY_SMPL_STR);
333  PyModule_AddStringConstant(m, "__version__", DEFINEDSTRING(AUBIO_VERSION));
334
335  // add generated objects
336  add_generated_objects(m);
337
338  // add ufunc
339  add_ufuncs(m);
340
341  aubio_log_set_level_function(AUBIO_LOG_ERR, aubio_log_function, NULL);
342  aubio_log_set_level_function(AUBIO_LOG_WRN, aubio_log_function, NULL);
343  return m;
344}
345
346#if PY_MAJOR_VERSION >= 3
347    // Python3 init
348    PyMODINIT_FUNC PyInit__aubio(void)
349    {
350        return initaubio();
351    }
352#else
353    // Python 2 init
354    PyMODINIT_FUNC init_aubio(void)
355    {
356        initaubio();
357    }
358#endif
Note: See TracBrowser for help on using the repository browser.