source: python/ext/py-fft.c @ 164980f

feature/autosinkfeature/constantqfeature/pitchshiftfeature/pydocstringsfeature/timestretchpitchshiftsamplertimestretchyinfft+
Last change on this file since 164980f was 164980f, checked in by Paul Brossier <piem@piem.org>, 8 years ago

python/: move source files to ext/

  • Property mode set to 100644
File size: 2.0 KB
Line 
1#include "aubiowraphell.h"
2
3static char Py_fft_doc[] = "fft object";
4
5AUBIO_DECLARE(fft, uint_t win_s)
6
7//AUBIO_NEW(fft)
8static PyObject *
9Py_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
44AUBIO_INIT(fft, self->win_s)
45
46AUBIO_DEL(fft)
47
48static PyObject * 
49Py_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
72AUBIO_MEMBERS_START(fft) 
73  {"win_s", T_INT, offsetof (Py_fft, win_s), READONLY,
74    "size of the window"},
75AUBIO_MEMBERS_STOP(fft)
76
77static PyObject * 
78Py_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
101static PyMethodDef Py_fft_methods[] = {
102  {"rdo", (PyCFunction) Py_fft_rdo, METH_VARARGS,
103    "synthesis of spectral grain"},
104  {NULL}
105};
106
107AUBIO_TYPEOBJECT(fft, "aubio.fft")
Note: See TracBrowser for help on using the repository browser.