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     char_t errstr[30];
  55     sprintf(errstr, "error creating fft with win_s=%d", self->win_s);
  56     PyErr_SetString (PyExc_Exception, errstr);
  57     return -1;
  58   }
  59
  60   self->doout = new_py_cvec(self->win_s);
  61   self->rdoout = new_py_fvec(self->win_s);
  62
  63   return 0;
  64 }
  65
  66 static void
  67 Py_fft_del (Py_fft *self, PyObject *unused)
  68 {
  69   Py_XDECREF(self->doout);
  70   Py_XDECREF(self->rdoout);
  71   del_aubio_fft(self->o);
  72   Py_TYPE(self)->tp_free((PyObject *) self);
  73 }
  74
  75 static PyObject *
  76 Py_fft_do(Py_fft * self, PyObject * args)
  77 {
  78   PyObject *input;
  79
  80   if (!PyArg_ParseTuple (args, "O", &input)) {
  81     return NULL;
  82   }
  83
  84   if (!PyAubio_ArrayToCFvec(input, &(self->vecin))) {
  85     return NULL;
  86   }
  87
  88   if (self->vecin.length != self->win_s) {
  89     PyErr_Format(PyExc_ValueError,
  90                  "input array has length %d, but fft has size %d",
  91                  self->vecin.length, self->win_s);
  92     return NULL;
  93   }
  94
  95   cvec_t c_out;
  96   Py_INCREF(self->doout);
  97   if (!PyAubio_PyCvecToCCvec(self->doout, &c_out)) {
  98     return NULL;
  99   }
 100   // compute the function
 101   aubio_fft_do (self->o, &(self->vecin), &c_out);
 102   return self->doout;
 103 }
 104
 105 static PyMemberDef Py_fft_members[] = {
 106   {"win_s", T_INT, offsetof (Py_fft, win_s), READONLY,
 107     "size of the window"},
 108   {NULL}
 109 };
 110
 111 static PyObject *
 112 Py_fft_rdo(Py_fft * self, PyObject * args)
 113 {
 114   PyObject *input;
 115
 116   if (!PyArg_ParseTuple (args, "O", &input)) {
 117     return NULL;
 118   }
 119
 120   if (!PyAubio_PyCvecToCCvec (input, &(self->cvecin)) ) {
 121     return NULL;
 122   }
 123
 124   fvec_t out;
 125   Py_INCREF(self->rdoout);
 126   if (!PyAubio_ArrayToCFvec(self->rdoout, &out) ) {
 127     return NULL;
 128   }
 129   // compute the function
 130   aubio_fft_rdo (self->o, &(self->cvecin), &out);
 131   return self->rdoout;
 132 }
 133
 134 static PyMethodDef Py_fft_methods[] = {
 135   {"rdo", (PyCFunction) Py_fft_rdo, METH_VARARGS,
 136     "synthesis of spectral grain"},
 137   {NULL}
 138 };
 139
 140 PyTypeObject Py_fftType = {
 141   PyVarObject_HEAD_INIT (NULL, 0)
 142   "aubio.fft",
 143   sizeof (Py_fft),
 144   0,
 145   (destructor) Py_fft_del,
 146   0,
 147   0,
 148   0,
 149   0,
 150   0,
 151   0,
 152   0,
 153   0,
 154   0,
 155   (ternaryfunc)Py_fft_do,
 156   0,
 157   0,
 158   0,
 159   0,
 160   Py_TPFLAGS_DEFAULT,
 161   Py_fft_doc,
 162   0,
 163   0,
 164   0,
 165   0,
 166   0,
 167   0,
 168   Py_fft_methods,
 169   Py_fft_members,
 170   0,
 171   0,
 172   0,
 173   0,
 174   0,
 175   0,
 176   (initproc) Py_fft_init,
 177   0,
 178   Py_fft_new,
 179 };