source: python/ext/py-cvec.c @ 1ad9dc3

feature/autosinkfeature/cnnfeature/cnn_orgfeature/constantqfeature/crepefeature/crepe_orgfeature/pitchshiftfeature/pydocstringsfeature/timestretchfix/ffmpeg5pitchshiftsamplertimestretchyinfft+
Last change on this file since 1ad9dc3 was 1ad9dc3, checked in by Paul Brossier <piem@piem.org>, 9 years ago

python/ext/py-cvec.c: remove unused function

  • Property mode set to 100644
File size: 6.6 KB
RevLine 
[f826349]1#include "aubio-types.h"
2
[92a8800]3/* cvec type definition
[f826349]4
5class cvec():
[92a8800]6    def __new__(self, length = 1024):
7        self.length = length / 2 + 1
8        self.norm = np.zeros(length / 2 + 1)
9        self.phas = np.zeros(length / 2 + 1)
[f826349]10
[7a7b00f]11*/
[f826349]12
[92a8800]13// special python type for cvec
14typedef struct
15{
16  PyObject_HEAD
17  PyObject *norm;
18  PyObject *phas;
19  uint_t length;
20} Py_cvec;
21
[f826349]22static char Py_cvec_doc[] = "cvec object";
23
[ede5d38]24
25PyObject *
26new_py_cvec(uint_t length) {
27  Py_cvec* vec = (Py_cvec*) PyObject_New (Py_cvec, &Py_cvecType);
28  npy_intp dims[] = { length / 2 + 1, 1 };
29  vec->norm = PyArray_ZEROS(1, dims, AUBIO_NPY_SMPL, 0);
30  vec->phas = PyArray_ZEROS(1, dims, AUBIO_NPY_SMPL, 0);
31  vec->length = length / 2 + 1;
32  return (PyObject*)vec;
33}
34
[92a8800]35int
36PyAubio_PyCvecToCCvec (PyObject *input, cvec_t *i) {
37  if (PyObject_TypeCheck (input, &Py_cvecType)) {
38      Py_cvec * in = (Py_cvec *)input;
39      i->norm = (smpl_t *) PyArray_GETPTR1 ((PyArrayObject *)(in->norm), 0);
40      i->phas = (smpl_t *) PyArray_GETPTR1 ((PyArrayObject *)(in->phas), 0);
41      i->length = ((Py_cvec*)input)->length;
42      return 1;
43  } else {
44      PyErr_SetString (PyExc_ValueError, "input array should be aubio.cvec");
45      return 0;
46  }
47}
48
[f826349]49static PyObject *
50Py_cvec_new (PyTypeObject * type, PyObject * args, PyObject * kwds)
51{
[7a7b00f]52  int length= 0;
[f826349]53  Py_cvec *self;
[7a7b00f]54  static char *kwlist[] = { "length", NULL };
[f826349]55
[7a7b00f]56  if (!PyArg_ParseTupleAndKeywords (args, kwds, "|I", kwlist,
57          &length)) {
[f826349]58    return NULL;
59  }
60
61  self = (Py_cvec *) type->tp_alloc (type, 0);
62
[eb93592]63  self->length = Py_default_vector_length / 2 + 1;
[f826349]64
65  if (self == NULL) {
66    return NULL;
67  }
68
69  if (length > 0) {
[eb93592]70    self->length = length / 2 + 1;
[f826349]71  } else if (length < 0) {
72    PyErr_SetString (PyExc_ValueError,
73        "can not use negative number of elements");
74    return NULL;
75  }
76
77  return (PyObject *) self;
78}
79
80static int
81Py_cvec_init (Py_cvec * self, PyObject * args, PyObject * kwds)
82{
[ede5d38]83  npy_intp dims[] = { self->length, 1 };
84  self->phas = PyArray_ZEROS(1, dims, AUBIO_NPY_SMPL, 0);
85  self->norm = PyArray_ZEROS(1, dims, AUBIO_NPY_SMPL, 0);
[f826349]86  return 0;
87}
88
89static void
90Py_cvec_del (Py_cvec * self)
91{
[ede5d38]92  Py_DECREF(self->norm);
93  Py_DECREF(self->phas);
[770b9e7]94  Py_TYPE(self)->tp_free ((PyObject *) self);
[f826349]95}
96
97static PyObject *
98Py_cvec_repr (Py_cvec * self, PyObject * unused)
99{
100  PyObject *format = NULL;
101  PyObject *args = NULL;
102  PyObject *result = NULL;
103
[5c1200a]104  format = PyUnicode_FromString ("aubio cvec of %d elements");
[f826349]105  if (format == NULL) {
106    goto fail;
107  }
108
[7a7b00f]109  args = Py_BuildValue ("I", self->length);
[f826349]110  if (args == NULL) {
111    goto fail;
112  }
[92a8800]113  // hide actual norm / phas content
[f826349]114
[5c1200a]115  result = PyUnicode_Format (format, args);
[f826349]116
117fail:
118  Py_XDECREF (format);
119  Py_XDECREF (args);
120
121  return result;
122}
123
[615ac7d]124PyObject *
125Py_cvec_get_norm (Py_cvec * self, void *closure)
126{
[ede5d38]127  // we want self->norm to still exist after our caller return it
[92a8800]128  Py_INCREF(self->norm);
129  return (PyObject*)(self->norm);
[615ac7d]130}
131
132PyObject *
133Py_cvec_get_phas (Py_cvec * self, void *closure)
134{
[ede5d38]135  // we want self->phas to still exist after our caller return it
[92a8800]136  Py_INCREF(self->phas);
137  return (PyObject *)(self->phas);
[615ac7d]138}
139
140static int
[a8aaef3]141Py_cvec_set_norm (Py_cvec * vec, PyObject *input, void * closure)
[615ac7d]142{
[34d0c25]143  if (!PyAubio_IsValidVector(input)) {
144    return 1;
[a8aaef3]145  }
[34d0c25]146  long length = PyArray_SIZE ((PyArrayObject *)input);
147  if (length != vec->length) {
148    PyErr_Format (PyExc_ValueError,
149        "input array has length %ld, but cvec has length %d", length,
150        vec->length);
[a8aaef3]151    return 1;
152  }
153
[34d0c25]154  Py_XDECREF(vec->norm);
155  vec->norm = input;
156  Py_INCREF(vec->norm);
[615ac7d]157  return 0;
158}
159
160static int
[a8aaef3]161Py_cvec_set_phas (Py_cvec * vec, PyObject *input, void * closure)
[615ac7d]162{
[34d0c25]163  if (!PyAubio_IsValidVector(input)) {
164    return 1;
[a8aaef3]165  }
[34d0c25]166  long length = PyArray_SIZE ((PyArrayObject *)input);
167  if (length != vec->length) {
168    PyErr_Format (PyExc_ValueError,
169        "input array has length %ld, but cvec has length %d", length,
170        vec->length);
[a8aaef3]171    return 1;
172  }
173
[34d0c25]174  Py_XDECREF(vec->phas);
175  vec->phas = input;
176  Py_INCREF(vec->phas);
[615ac7d]177  return 0;
178}
179
[f826349]180static PyMemberDef Py_cvec_members[] = {
181  // TODO remove READONLY flag and define getter/setter
182  {"length", T_INT, offsetof (Py_cvec, length), READONLY,
183      "length attribute"},
184  {NULL}                        /* Sentinel */
185};
186
187static PyMethodDef Py_cvec_methods[] = {
188  {NULL}
189};
190
[615ac7d]191static PyGetSetDef Py_cvec_getseters[] = {
192  {"norm", (getter)Py_cvec_get_norm, (setter)Py_cvec_set_norm, 
[474f297]193      "Numpy vector of shape (length,) containing the magnitude",
[615ac7d]194      NULL},
195  {"phas", (getter)Py_cvec_get_phas, (setter)Py_cvec_set_phas, 
[474f297]196      "Numpy vector of shape (length,) containing the phase",
[615ac7d]197      NULL},
198  {NULL} /* sentinel */
199};
200
[f826349]201PyTypeObject Py_cvecType = {
[5c1200a]202  PyVarObject_HEAD_INIT(NULL, 0)
[f826349]203  "aubio.cvec",                 /* tp_name           */
204  sizeof (Py_cvec),             /* tp_basicsize      */
205  0,                            /* tp_itemsize       */
206  (destructor) Py_cvec_del,     /* tp_dealloc        */
207  0,                            /* tp_print          */
208  0,                            /* tp_getattr        */
209  0,                            /* tp_setattr        */
210  0,                            /* tp_compare        */
211  (reprfunc) Py_cvec_repr,      /* tp_repr           */
212  0,                            /* tp_as_number      */
[f2ce0fc]213  0, //&Py_cvec_tp_as_sequence, /* tp_as_sequence    */
[f826349]214  0,                            /* tp_as_mapping     */
215  0,                            /* tp_hash           */
216  0,                            /* tp_call           */
217  0,                            /* tp_str            */
218  0,                            /* tp_getattro       */
219  0,                            /* tp_setattro       */
220  0,                            /* tp_as_buffer      */
221  Py_TPFLAGS_DEFAULT,           /* tp_flags          */
222  Py_cvec_doc,                  /* tp_doc            */
223  0,                            /* tp_traverse       */
224  0,                            /* tp_clear          */
225  0,                            /* tp_richcompare    */
226  0,                            /* tp_weaklistoffset */
227  0,                            /* tp_iter           */
228  0,                            /* tp_iternext       */
229  Py_cvec_methods,              /* tp_methods        */
230  Py_cvec_members,              /* tp_members        */
[615ac7d]231  Py_cvec_getseters,            /* tp_getset         */
[f826349]232  0,                            /* tp_base           */
233  0,                            /* tp_dict           */
234  0,                            /* tp_descr_get      */
235  0,                            /* tp_descr_set      */
236  0,                            /* tp_dictoffset     */
237  (initproc) Py_cvec_init,      /* tp_init           */
238  0,                            /* tp_alloc          */
239  Py_cvec_new,                  /* tp_new            */
240};
Note: See TracBrowser for help on using the repository browser.