1 #include "aubio-types.h"
11 static char Py_filter_doc[] = "filter object";
14 Py_filter_new (PyTypeObject * type, PyObject * args, PyObject * kwds)
18 static char *kwlist[] = { "order", NULL };
20 if (!PyArg_ParseTupleAndKeywords (args, kwds, "|I", kwlist,
25 self = (Py_filter *) type->tp_alloc (type, 0);
35 } else if (order < 0) {
36 PyErr_SetString (PyExc_ValueError,
37 "can not use negative order");
41 return (PyObject *) self;
45 Py_filter_init (Py_filter * self, PyObject * args, PyObject * kwds)
47 self->o = new_aubio_filter (self->order);
48 if (self->o == NULL) {
51 self->out = new_fvec(Py_default_vector_length);
56 Py_filter_del (Py_filter * self)
59 del_aubio_filter (self->o);
60 self->ob_type->tp_free ((PyObject *) self);
64 Py_filter_do(Py_filter * self, PyObject * args)
69 if (!PyArg_ParseTuple (args, "O:digital_filter.do", &input)) {
77 vec = PyAubio_ArrayToCFvec (input);
83 // reallocate the output if needed
84 if (vec->length != self->out->length) {
86 self->out = new_fvec(vec->length);
88 // compute the function
89 aubio_filter_do_outplace (self->o, vec, self->out);
90 return PyAubio_CFvecToArray(self->out);
94 Py_filter_set_c_weighting (Py_filter * self, PyObject *args)
98 if (!PyArg_ParseTuple (args, "I", &samplerate)) {
102 err = aubio_filter_set_c_weighting (self->o, samplerate);
104 PyErr_SetString (PyExc_ValueError,
105 "error when setting filter to C-weighting");
112 Py_filter_set_a_weighting (Py_filter * self, PyObject *args)
116 if (!PyArg_ParseTuple (args, "I", &samplerate)) {
120 err = aubio_filter_set_a_weighting (self->o, samplerate);
122 PyErr_SetString (PyExc_ValueError,
123 "error when setting filter to A-weighting");
130 Py_filter_set_biquad(Py_filter * self, PyObject *args)
133 lsmp_t b0, b1, b2, a1, a2;
134 if (!PyArg_ParseTuple (args, "ddddd", &b0, &b1, &b2, &a1, &a2)) {
138 err = aubio_filter_set_biquad (self->o, b0, b1, b2, a1, a2);
140 PyErr_SetString (PyExc_ValueError,
141 "error when setting filter with biquad coefficients");
147 static PyMemberDef Py_filter_members[] = {
148 // TODO remove READONLY flag and define getter/setter
149 {"order", T_INT, offsetof (Py_filter, order), READONLY,
150 "order of the filter"},
151 {NULL} /* Sentinel */
154 static PyMethodDef Py_filter_methods[] = {
155 {"set_c_weighting", (PyCFunction) Py_filter_set_c_weighting, METH_VARARGS,
156 "set filter coefficients to C-weighting"},
157 {"set_a_weighting", (PyCFunction) Py_filter_set_a_weighting, METH_VARARGS,
158 "set filter coefficients to A-weighting"},
159 {"set_biquad", (PyCFunction) Py_filter_set_biquad, METH_VARARGS,
160 "set b0, b1, b2, a1, a2 biquad coefficients"},
164 PyTypeObject Py_filterType = {
165 PyObject_HEAD_INIT (NULL)
167 "aubio.digital_filter", /* tp_name */
168 sizeof (Py_filter), /* tp_basicsize */
170 (destructor) Py_filter_del, /* tp_dealloc */
175 0, //(reprfunc) Py_filter_repr, /* tp_repr */
176 0, /* tp_as_number */
177 0, /* tp_as_sequence */
178 0, /* tp_as_mapping */
180 (ternaryfunc)Py_filter_do, /* tp_call */
184 0, /* tp_as_buffer */
185 Py_TPFLAGS_DEFAULT, /* tp_flags */
186 Py_filter_doc, /* tp_doc */
189 0, /* tp_richcompare */
190 0, /* tp_weaklistoffset */
193 Py_filter_methods, /* tp_methods */
194 Py_filter_members, /* tp_members */
198 0, /* tp_descr_get */
199 0, /* tp_descr_set */
200 0, /* tp_dictoffset */
201 (initproc) Py_filter_init, /* tp_init */
203 Py_filter_new, /* tp_new */