source: python/ext/py-filter.c @ 770b9e7

feature/autosinkfeature/cnnfeature/cnn_orgfeature/constantqfeature/crepefeature/crepe_orgfeature/pitchshiftfeature/pydocstringsfeature/timestretchfix/ffmpeg5pitchshiftsamplertimestretchyinfft+
Last change on this file since 770b9e7 was 770b9e7, checked in by Nils Philippsen <nils@tiptoe.de>, 8 years ago

Python 3: use Py_TYPE() instead of ob_type member

  • Property mode set to 100644
File size: 5.5 KB
Line 
1#include "aubio-types.h"
2
3typedef struct
4{
5  PyObject_HEAD
6  aubio_filter_t * o;
7  uint_t order;
8} Py_filter;
9
10static char Py_filter_doc[] = "filter object";
11
12static PyObject *
13Py_filter_new (PyTypeObject * type, PyObject * args, PyObject * kwds)
14{
15  int order= 0;
16  Py_filter *self;
17  static char *kwlist[] = { "order", NULL };
18
19  if (!PyArg_ParseTupleAndKeywords (args, kwds, "|I", kwlist,
20          &order)) {
21    return NULL;
22  }
23
24  self = (Py_filter *) type->tp_alloc (type, 0);
25
26  if (self == NULL) {
27    return NULL;
28  }
29
30  self->order = 7;
31
32  if (order > 0) {
33    self->order = order;
34  } else if (order < 0) {
35    PyErr_SetString (PyExc_ValueError,
36        "can not use negative order");
37    return NULL;
38  }
39
40  return (PyObject *) self;
41}
42
43static int
44Py_filter_init (Py_filter * self, PyObject * args, PyObject * kwds)
45{
46  self->o = new_aubio_filter (self->order);
47  if (self->o == NULL) {
48    return -1;
49  }
50
51  return 0;
52}
53
54static void
55Py_filter_del (Py_filter * self)
56{
57  del_aubio_filter (self->o);
58  Py_TYPE(self)->tp_free ((PyObject *) self);
59}
60
61static PyObject * 
62Py_filter_do(Py_filter * self, PyObject * args)
63{
64  PyObject *input;
65  fvec_t *vec;
66
67  if (!PyArg_ParseTuple (args, "O:digital_filter.do", &input)) {
68    return NULL;
69  }
70
71  if (input == NULL) {
72    return NULL;
73  }
74
75  vec = PyAubio_ArrayToCFvec (input);
76
77  if (vec == NULL) {
78    return NULL;
79  }
80
81  // compute the function
82  fvec_t * out = new_fvec(vec->length);
83  aubio_filter_do_outplace (self->o, vec, out);
84  return PyAubio_CFvecToArray(out);
85}
86
87static PyObject * 
88Py_filter_set_c_weighting (Py_filter * self, PyObject *args)
89{
90  uint_t err = 0;
91  uint_t samplerate;
92  if (!PyArg_ParseTuple (args, "I", &samplerate)) {
93    return NULL;
94  }
95
96  err = aubio_filter_set_c_weighting (self->o, samplerate);
97  if (err > 0) {
98    PyErr_SetString (PyExc_ValueError,
99        "error when setting filter to C-weighting");
100    return NULL;
101  }
102  Py_RETURN_NONE;
103}
104
105static PyObject * 
106Py_filter_set_a_weighting (Py_filter * self, PyObject *args)
107{
108  uint_t err = 0;
109  uint_t samplerate;
110  if (!PyArg_ParseTuple (args, "I", &samplerate)) {
111    return NULL;
112  }
113
114  err = aubio_filter_set_a_weighting (self->o, samplerate);
115  if (err > 0) {
116    PyErr_SetString (PyExc_ValueError,
117        "error when setting filter to A-weighting");
118    return NULL;
119  }
120  Py_RETURN_NONE;
121}
122
123static PyObject *
124Py_filter_set_biquad(Py_filter * self, PyObject *args)
125{
126  uint_t err = 0;
127  lsmp_t b0, b1, b2, a1, a2;
128  if (!PyArg_ParseTuple (args, "ddddd", &b0, &b1, &b2, &a1, &a2)) {
129    return NULL;
130  }
131
132  err = aubio_filter_set_biquad (self->o, b0, b1, b2, a1, a2);
133  if (err > 0) {
134    PyErr_SetString (PyExc_ValueError,
135        "error when setting filter with biquad coefficients");
136    return NULL;
137  }
138  Py_RETURN_NONE;
139}
140
141static PyMemberDef Py_filter_members[] = {
142  // TODO remove READONLY flag and define getter/setter
143  {"order", T_INT, offsetof (Py_filter, order), READONLY,
144      "order of the filter"},
145  {NULL}                        /* Sentinel */
146};
147
148static PyMethodDef Py_filter_methods[] = {
149  {"set_c_weighting", (PyCFunction) Py_filter_set_c_weighting, METH_VARARGS,
150      "set filter coefficients to C-weighting"},
151  {"set_a_weighting", (PyCFunction) Py_filter_set_a_weighting, METH_VARARGS,
152      "set filter coefficients to A-weighting"},
153  {"set_biquad", (PyCFunction) Py_filter_set_biquad, METH_VARARGS,
154      "set b0, b1, b2, a1, a2 biquad coefficients"},
155  {NULL}
156};
157
158PyTypeObject Py_filterType = {
159  PyObject_HEAD_INIT (NULL)
160  0,                            /* ob_size           */
161  "aubio.digital_filter",       /* tp_name           */
162  sizeof (Py_filter),           /* tp_basicsize      */
163  0,                            /* tp_itemsize       */
164  (destructor) Py_filter_del,   /* tp_dealloc        */
165  0,                            /* tp_print          */
166  0,                            /* tp_getattr        */
167  0,                            /* tp_setattr        */
168  0,                            /* tp_compare        */
169  0, //(reprfunc) Py_filter_repr,    /* tp_repr           */
170  0,                            /* tp_as_number      */
171  0,                            /* tp_as_sequence    */
172  0,                            /* tp_as_mapping     */
173  0,                            /* tp_hash           */
174  (ternaryfunc)Py_filter_do,    /* tp_call           */
175  0,                            /* tp_str            */
176  0,                            /* tp_getattro       */
177  0,                            /* tp_setattro       */
178  0,                            /* tp_as_buffer      */
179  Py_TPFLAGS_DEFAULT,           /* tp_flags          */
180  Py_filter_doc,                /* tp_doc            */
181  0,                            /* tp_traverse       */
182  0,                            /* tp_clear          */
183  0,                            /* tp_richcompare    */
184  0,                            /* tp_weaklistoffset */
185  0,                            /* tp_iter           */
186  0,                            /* tp_iternext       */
187  Py_filter_methods,            /* tp_methods        */
188  Py_filter_members,            /* tp_members        */
189  0,                            /* tp_getset         */
190  0,                            /* tp_base           */
191  0,                            /* tp_dict           */
192  0,                            /* tp_descr_get      */
193  0,                            /* tp_descr_set      */
194  0,                            /* tp_dictoffset     */
195  (initproc) Py_filter_init,    /* tp_init           */
196  0,                            /* tp_alloc          */
197  Py_filter_new,                /* tp_new            */
198};
Note: See TracBrowser for help on using the repository browser.