1 #include "aubio-types.h"
10 static char Py_filter_doc[] = "filter object";
13 Py_filter_new (PyTypeObject * type, PyObject * args, PyObject * kwds)
17 static char *kwlist[] = { "order", NULL };
19 if (!PyArg_ParseTupleAndKeywords (args, kwds, "|I", kwlist,
24 self = (Py_filter *) type->tp_alloc (type, 0);
34 } else if (order < 0) {
35 PyErr_SetString (PyExc_ValueError,
36 "can not use negative order");
40 return (PyObject *) self;
44 Py_filter_init (Py_filter * self, PyObject * args, PyObject * kwds)
46 self->o = new_aubio_filter (self->order);
47 if (self->o == NULL) {
55 Py_filter_del (Py_filter * self)
57 del_aubio_filter (self->o);
58 Py_TYPE(self)->tp_free ((PyObject *) self);
62 Py_filter_do(Py_filter * self, PyObject * args)
67 if (!PyArg_ParseTuple (args, "O:digital_filter.do", &input)) {
75 vec = PyAubio_ArrayToCFvec (input);
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);
88 Py_filter_set_c_weighting (Py_filter * self, PyObject *args)
92 if (!PyArg_ParseTuple (args, "I", &samplerate)) {
96 err = aubio_filter_set_c_weighting (self->o, samplerate);
98 PyErr_SetString (PyExc_ValueError,
99 "error when setting filter to C-weighting");
106 Py_filter_set_a_weighting (Py_filter * self, PyObject *args)
110 if (!PyArg_ParseTuple (args, "I", &samplerate)) {
114 err = aubio_filter_set_a_weighting (self->o, samplerate);
116 PyErr_SetString (PyExc_ValueError,
117 "error when setting filter to A-weighting");
124 Py_filter_set_biquad(Py_filter * self, PyObject *args)
127 lsmp_t b0, b1, b2, a1, a2;
128 if (!PyArg_ParseTuple (args, "ddddd", &b0, &b1, &b2, &a1, &a2)) {
132 err = aubio_filter_set_biquad (self->o, b0, b1, b2, a1, a2);
134 PyErr_SetString (PyExc_ValueError,
135 "error when setting filter with biquad coefficients");
141 static 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 */
148 static 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"},
158 PyTypeObject Py_filterType = {
159 PyObject_HEAD_INIT (NULL)
161 "aubio.digital_filter", /* tp_name */
162 sizeof (Py_filter), /* tp_basicsize */
164 (destructor) Py_filter_del, /* tp_dealloc */
169 0, //(reprfunc) Py_filter_repr, /* tp_repr */
170 0, /* tp_as_number */
171 0, /* tp_as_sequence */
172 0, /* tp_as_mapping */
174 (ternaryfunc)Py_filter_do, /* tp_call */
178 0, /* tp_as_buffer */
179 Py_TPFLAGS_DEFAULT, /* tp_flags */
180 Py_filter_doc, /* tp_doc */
183 0, /* tp_richcompare */
184 0, /* tp_weaklistoffset */
187 Py_filter_methods, /* tp_methods */
188 Py_filter_members, /* tp_members */
192 0, /* tp_descr_get */
193 0, /* tp_descr_set */
194 0, /* tp_dictoffset */
195 (initproc) Py_filter_init, /* tp_init */
197 Py_filter_new, /* tp_new */