python/ext/py-fft.c

   1 #include "aubiowraphell.h"
   2
   3 static char Py_fft_doc[] = "fft object";
   4
   5 AUBIO_DECLARE(fft, uint_t win_s)
   6
   7 //AUBIO_NEW(fft)
   8 static PyObject *
   9 Py_fft_new (PyTypeObject * type, PyObject * args, PyObject * kwds)
  10 {
  11   int win_s = 0;
  12   Py_fft *self;
  13   static char *kwlist[] = { "win_s", NULL };
  14
  15   if (!PyArg_ParseTupleAndKeywords (args, kwds, "|I", kwlist,
  16           &win_s)) {
  17     return NULL;
  18   }
  19
  20   self = (Py_fft *) type->tp_alloc (type, 0);
  21
  22   if (self == NULL) {
  23     return NULL;
  24   }
  25
  26   self->win_s = Py_default_vector_length;
  27
  28   if (self == NULL) {
  29     return NULL;
  30   }
  31
  32   if (win_s > 0) {
  33     self->win_s = win_s;
  34   } else if (win_s < 0) {
  35     PyErr_SetString (PyExc_ValueError,
  36         "can not use negative window size");
  37     return NULL;
  38   }
  39
  40   return (PyObject *) self;
  41 }
  42
  43
  44 AUBIO_INIT(fft, self->win_s)
  45
  46 AUBIO_DEL(fft)
  47
  48 static PyObject *
  49 Py_fft_do(PyObject * self, PyObject * args)
  50 {
  51   PyObject *input;
  52   fvec_t *vec;
  53   cvec_t *output;
  54
  55   if (!PyArg_ParseTuple (args, "O", &input)) {
  56     return NULL;
  57   }
  58
  59   vec = PyAubio_ArrayToCFvec (input);
  60
  61   if (vec == NULL) {
  62     return NULL;
  63   }
  64
  65   output = new_cvec(((Py_fft *) self)->win_s);
  66
  67   // compute the function
  68   aubio_fft_do (((Py_fft *)self)->o, vec, output);
  69   return (PyObject *)PyAubio_CCvecToPyCvec(output);
  70 }
  71
  72 AUBIO_MEMBERS_START(fft)
  73   {"win_s", T_INT, offsetof (Py_fft, win_s), READONLY,
  74     "size of the window"},
  75 AUBIO_MEMBERS_STOP(fft)
  76
  77 static PyObject *
  78 Py_fft_rdo(Py_fft * self, PyObject * args)
  79 {
  80   PyObject *input;
  81   cvec_t *vec;
  82   fvec_t *output;
  83
  84   if (!PyArg_ParseTuple (args, "O", &input)) {
  85     return NULL;
  86   }
  87
  88   vec = PyAubio_ArrayToCCvec (input);
  89
  90   if (vec == NULL) {
  91     return NULL;
  92   }
  93
  94   output = new_fvec(self->win_s);
  95
  96   // compute the function
  97   aubio_fft_rdo (((Py_fft *)self)->o, vec, output);
  98   return (PyObject *)PyAubio_CFvecToArray(output);
  99 }
 100
 101 static PyMethodDef Py_fft_methods[] = {
 102   {"rdo", (PyCFunction) Py_fft_rdo, METH_VARARGS,
 103     "synthesis of spectral grain"},
 104   {NULL}
 105 };
 106
 107 AUBIO_TYPEOBJECT(fft, "aubio.fft")