source: python/ext/ufuncs.c @ c2e0aef

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

ext/aubiomodule.c: more ufuncs

  • Property mode set to 100644
File size: 3.3 KB
Line 
1#define PY_AUBIO_MODULE_UFUNC
2#include "aubio-types.h"
3
4typedef smpl_t (*aubio_unary_func_t)(smpl_t input);
5
6static void aubio_PyUFunc_d_d(char **args, npy_intp *dimensions,
7                            npy_intp* steps, void* data)
8{
9    npy_intp i;
10    npy_intp n = dimensions[0];
11    char *in = args[0], *out = args[1];
12    npy_intp in_step = steps[0], out_step = steps[1];
13    aubio_unary_func_t func = (aubio_unary_func_t)(data);
14
15    for (i = 0; i < n; i++) {
16        /*BEGIN main ufunc computation*/
17        *((double *)out) = func(*(double *)in);
18        /*END main ufunc computation*/
19
20        in += in_step;
21        out += out_step;
22    }
23}
24
25static void aubio_PyUFunc_f_f_As_d_d(char **args, npy_intp *dimensions,
26                            npy_intp* steps, void* data)
27{
28    npy_intp i;
29    npy_intp n = dimensions[0];
30    char *in = args[0], *out = args[1];
31    npy_intp in_step = steps[0], out_step = steps[1];
32    aubio_unary_func_t func = (aubio_unary_func_t)(data);
33
34    for (i = 0; i < n; i++) {
35        /*BEGIN main ufunc computation*/
36        *((float *)out) = func(*(float *)in);
37        /*END main ufunc computation*/
38
39        in += in_step;
40        out += out_step;
41    }
42}
43
44static int Py_aubio_unary_n_types = 2;
45static int Py_aubio_unary_n_inputs = 1;
46static int Py_aubio_unary_n_outputs = 1;
47PyUFuncGenericFunction Py_aubio_unary_functions[] = {
48  &aubio_PyUFunc_f_f_As_d_d,
49  &aubio_PyUFunc_d_d,
50  //PyUFunc_f_f_As_d_d, PyUFunc_d_d,
51  //PyUFunc_g_g, PyUFunc_OO_O_method,
52};
53
54static char Py_aubio_unary_types[] = {
55  NPY_FLOAT, NPY_FLOAT,
56  NPY_DOUBLE, NPY_DOUBLE,
57  //NPY_LONGDOUBLE, NPY_LONGDOUBLE,
58  //NPY_OBJECT, NPY_OBJECT,
59};
60
61static char Py_unwrap2pi_doc[] = "map angle to unit circle [-pi, pi[";
62
63static void* Py_unwrap2pi_data[] = {
64  (void *)aubio_unwrap2pi,
65  (void *)aubio_unwrap2pi,
66  //(void *)unwrap2pil,
67  //(void *)unwrap2pio,
68};
69
70static char Py_freqtomidi_doc[] = "convert frequency to midi";
71
72static void* Py_freqtomidi_data[] = {
73  (void *)aubio_freqtomidi,
74  (void *)aubio_freqtomidi,
75};
76
77static char Py_miditofreq_doc[] = "convert midi to frequency";
78
79static void* Py_miditofreq_data[] = {
80  (void *)aubio_miditofreq,
81  (void *)aubio_miditofreq,
82};
83
84void add_ufuncs ( PyObject *m )
85{
86  int err = 0;
87
88  err = _import_umath ();
89  if (err != 0) {
90    fprintf (stderr,
91        "Unable to import Numpy umath from aubio module (error %d)\n", err);
92  }
93
94  PyObject *f, *dict;
95  dict = PyModule_GetDict(m);
96  f = PyUFunc_FromFuncAndData(Py_aubio_unary_functions, Py_unwrap2pi_data, Py_aubio_unary_types,
97          Py_aubio_unary_n_types, Py_aubio_unary_n_inputs, Py_aubio_unary_n_outputs,
98          PyUFunc_None, "unwrap2pi", Py_unwrap2pi_doc, 0);
99  PyDict_SetItemString(dict, "unwrap2pi", f);
100  Py_DECREF(f);
101
102  PyObject *g;
103  g = PyUFunc_FromFuncAndData(Py_aubio_unary_functions, Py_freqtomidi_data, Py_aubio_unary_types,
104          Py_aubio_unary_n_types, Py_aubio_unary_n_inputs, Py_aubio_unary_n_outputs,
105          PyUFunc_None, "freqtomidi", Py_freqtomidi_doc, 0);
106  PyDict_SetItemString(dict, "freqtomidi", g);
107  Py_DECREF(g);
108
109  PyObject *h;
110  h = PyUFunc_FromFuncAndData(Py_aubio_unary_functions, Py_miditofreq_data, Py_aubio_unary_types,
111          Py_aubio_unary_n_types, Py_aubio_unary_n_inputs, Py_aubio_unary_n_outputs,
112          PyUFunc_None, "miditofreq", Py_miditofreq_doc, 0);
113  PyDict_SetItemString(dict, "miditofreq", h);
114  Py_DECREF(h);
115  return;
116}
Note: See TracBrowser for help on using the repository browser.