python/ext/py-fft.c

   1 #include "aubio-types.h"
   2
   3 static char Py_fft_doc[] = "fft object";
   4
   5 typedef struct
   6 {
   7   PyObject_HEAD
   8   aubio_fft_t * o;
   9   uint_t win_s;
  10   // do / rdo input vectors
  11   fvec_t vecin;
  12   cvec_t cvecin;
  13   // do / rdo output results
  14   PyObject *doout;
  15   PyObject *rdoout;
  16 } Py_fft;
  17
  18 static PyObject *
  19 Py_fft_new (PyTypeObject * type, PyObject * args, PyObject * kwds)
  20 {
  21   int win_s = 0;
  22   Py_fft *self;
  23   static char *kwlist[] = { "win_s", NULL };
  24
  25   if (!PyArg_ParseTupleAndKeywords (args, kwds, "|I", kwlist,
  26           &win_s)) {
  27     return NULL;
  28   }
  29
  30   self = (Py_fft *) type->tp_alloc (type, 0);
  31
  32   if (self == NULL) {
  33     return NULL;
  34   }
  35
  36   self->win_s = Py_default_vector_length;
  37
  38   if (win_s > 0) {
  39     self->win_s = win_s;
  40   } else if (win_s < 0) {
  41     PyErr_SetString (PyExc_ValueError,
  42         "can not use negative window size");
  43     return NULL;
  44   }
  45
  46   return (PyObject *) self;
  47 }
  48
  49 static int
  50 Py_fft_init (Py_fft * self, PyObject * args, PyObject * kwds)
  51 {
  52   self->o = new_aubio_fft (self->win_s);
  53   if (self->o == NULL) {
  54     PyErr_Format(PyExc_RuntimeError,
  55         "error creating fft with win_s=%d "
  56         "(should be a power of 2 greater than 1; "
  57         "try recompiling aubio with --enable-fftw3)",
  58         self->win_s);
  59     return -1;
  60   }
  61
  62   self->doout = new_py_cvec(self->win_s);
  63   self->rdoout = new_py_fvec(self->win_s);
  64
  65   return 0;
  66 }
  67
  68 static void
  69 Py_fft_del (Py_fft *self, PyObject *unused)
  70 {
  71   Py_XDECREF(self->doout);
  72   Py_XDECREF(self->rdoout);
  73   if (self->o) {
  74     del_aubio_fft(self->o);
  75   }
  76   Py_TYPE(self)->tp_free((PyObject *) self);
  77 }
  78
  79 static PyObject *
  80 Py_fft_do(Py_fft * self, PyObject * args)
  81 {
  82   PyObject *input;
  83
  84   if (!PyArg_ParseTuple (args, "O", &input)) {
  85     return NULL;
  86   }
  87
  88   if (!PyAubio_ArrayToCFvec(input, &(self->vecin))) {
  89     return NULL;
  90   }
  91
  92   if (self->vecin.length != self->win_s) {
  93     PyErr_Format(PyExc_ValueError,
  94                  "input array has length %d, but fft expects length %d",
  95                  self->vecin.length, self->win_s);
  96     return NULL;
  97   }
  98
  99   cvec_t c_out;
 100   Py_INCREF(self->doout);
 101   if (!PyAubio_PyCvecToCCvec(self->doout, &c_out)) {
 102     return NULL;
 103   }
 104   // compute the function
 105   aubio_fft_do (self->o, &(self->vecin), &c_out);
 106   return self->doout;
 107 }
 108
 109 static PyMemberDef Py_fft_members[] = {
 110   {"win_s", T_INT, offsetof (Py_fft, win_s), READONLY,
 111     "size of the window"},
 112   {NULL}
 113 };
 114
 115 static PyObject *
 116 Py_fft_rdo(Py_fft * self, PyObject * args)
 117 {
 118   PyObject *input;
 119
 120   if (!PyArg_ParseTuple (args, "O", &input)) {
 121     return NULL;
 122   }
 123
 124   if (!PyAubio_PyCvecToCCvec (input, &(self->cvecin)) ) {
 125     return NULL;
 126   }
 127
 128   if (self->cvecin.length != self->win_s / 2 + 1) {
 129     PyErr_Format(PyExc_ValueError,
 130                  "input cvec has length %d, but fft expects length %d",
 131                  self->cvecin.length, self->win_s / 2 + 1);
 132     return NULL;
 133   }
 134
 135   fvec_t out;
 136   Py_INCREF(self->rdoout);
 137   if (!PyAubio_ArrayToCFvec(self->rdoout, &out) ) {
 138     return NULL;
 139   }
 140   // compute the function
 141   aubio_fft_rdo (self->o, &(self->cvecin), &out);
 142   return self->rdoout;
 143 }
 144
 145 static PyMethodDef Py_fft_methods[] = {
 146   {"rdo", (PyCFunction) Py_fft_rdo, METH_VARARGS,
 147     "synthesis of spectral grain"},
 148   {NULL}
 149 };
 150
 151 PyTypeObject Py_fftType = {
 152   PyVarObject_HEAD_INIT (NULL, 0)
 153   "aubio.fft",
 154   sizeof (Py_fft),
 155   0,
 156   (destructor) Py_fft_del,
 157   0,
 158   0,
 159   0,
 160   0,
 161   0,
 162   0,
 163   0,
 164   0,
 165   0,
 166   (ternaryfunc)Py_fft_do,
 167   0,
 168   0,
 169   0,
 170   0,
 171   Py_TPFLAGS_DEFAULT,
 172   Py_fft_doc,
 173   0,
 174   0,
 175   0,
 176   0,
 177   0,
 178   0,
 179   Py_fft_methods,
 180   Py_fft_members,
 181   0,
 182   0,
 183   0,
 184   0,
 185   0,
 186   0,
 187   (initproc) Py_fft_init,
 188   0,
 189   Py_fft_new,
 190 };