source: python/ext/aubioproxy.c @ bfe8256

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

ext/aubio-types.h: rewrite array to fvec conversions to reduce memory allocations

  • Property mode set to 100644
File size: 5.0 KB
Line 
1#include "aubio-types.h"
2
3PyObject *
4PyAubio_CFvecToArray (fvec_t * self)
5{
6  npy_intp dims[] = { self->length, 1 };
7  return PyArray_SimpleNewFromData (1, dims, AUBIO_NPY_SMPL, self->data);
8}
9
10int
11PyAubio_ArrayToCFvec (PyObject *input, fvec_t *out) {
12  if (input == NULL) {
13    PyErr_SetString (PyExc_ValueError, "input array is not a python object");
14    return 0;
15  }
16  // parsing input object into a Py_fvec
17  if (PyArray_Check(input)) {
18
19    // we got an array, convert it to an fvec
20    if (PyArray_NDIM ((PyArrayObject *)input) == 0) {
21      PyErr_SetString (PyExc_ValueError, "input array is a scalar");
22      return 0;
23    } else if (PyArray_NDIM ((PyArrayObject *)input) > 1) {
24      PyErr_SetString (PyExc_ValueError,
25          "input array has more than one dimensions");
26      return 0;
27    }
28
29    if (!PyArray_ISFLOAT ((PyArrayObject *)input)) {
30      PyErr_SetString (PyExc_ValueError, "input array should be float");
31      return 0;
32    } else if (PyArray_TYPE ((PyArrayObject *)input) != AUBIO_NPY_SMPL) {
33      PyErr_SetString (PyExc_ValueError, "input array should be float32");
34      return 0;
35    }
36
37    // vec = new_fvec (vec->length);
38    // no need to really allocate fvec, just its struct member
39    long length = PyArray_SIZE ((PyArrayObject *)input);
40    if (length <= 0) {
41      PyErr_SetString (PyExc_ValueError, "input array size should be greater than 0");
42      return 0;
43    }
44
45  } else if (PyObject_TypeCheck (input, &PyList_Type)) {
46    PyErr_SetString (PyExc_ValueError, "does not convert from list yet");
47    return 0;
48  } else {
49    PyErr_SetString (PyExc_ValueError, "can only accept vector of float as input");
50    return 0;
51  }
52
53  out->length = (uint_t) PyArray_SIZE ((PyArrayObject *)input);
54  out->data = (smpl_t *) PyArray_GETPTR1 ((PyArrayObject *)input, 0);
55  return 1;
56}
57
58PyObject *
59PyAubio_CCvecToPyCvec (cvec_t * input, Py_cvec *vec) {
60  vec->length = input->length;
61  vec->o = input;
62  // keep a reference to re-use after returning it
63  Py_INCREF(vec);
64  return (PyObject *)vec;
65}
66
67int
68PyAubio_ArrayToCCvec (PyObject *input, cvec_t *i) {
69  if (PyObject_TypeCheck (input, &Py_cvecType)) {
70      //*i = *(((Py_cvec*)input)->o);
71      i->norm = ((Py_cvec*)input)->o->norm;
72      i->phas = ((Py_cvec*)input)->o->phas;
73      i->length = ((Py_cvec*)input)->o->length;
74      return 1;
75  } else {
76      PyErr_SetString (PyExc_ValueError, "input array should be float32");
77      return 0;
78  }
79}
80
81PyObject *
82PyAubio_CFmatToArray (fmat_t * input)
83{
84  PyObject *array = NULL;
85  uint_t i;
86  npy_intp dims[] = { input->length, 1 };
87  PyObject *concat = PyList_New (0), *tmp = NULL;
88  for (i = 0; i < input->height; i++) {
89    tmp = PyArray_SimpleNewFromData (1, dims, AUBIO_NPY_SMPL, input->data[i]);
90    PyList_Append (concat, tmp);
91    Py_DECREF (tmp);
92  }
93  array = PyArray_FromObject (concat, AUBIO_NPY_SMPL, 2, 2);
94  Py_DECREF (concat);
95  return array;
96}
97
98int
99PyAubio_ArrayToCFmat (PyObject *input, fmat_t *mat) {
100  uint_t i;
101  if (input == NULL) {
102    PyErr_SetString (PyExc_ValueError, "input array is not a python object");
103    return 0;
104  }
105  // parsing input object into a Py_fvec
106  if (PyArray_Check(input)) {
107
108    // we got an array, convert it to an fvec
109    if (PyArray_NDIM ((PyArrayObject *)input) == 0) {
110      PyErr_SetString (PyExc_ValueError, "input array is a scalar");
111      return 0;
112    } else if (PyArray_NDIM ((PyArrayObject *)input) > 2) {
113      PyErr_SetString (PyExc_ValueError,
114          "input array has more than two dimensions");
115      return 0;
116    }
117
118    if (!PyArray_ISFLOAT ((PyArrayObject *)input)) {
119      PyErr_SetString (PyExc_ValueError, "input array should be float");
120      return 0;
121    } else if (PyArray_TYPE ((PyArrayObject *)input) != AUBIO_NPY_SMPL) {
122      PyErr_SetString (PyExc_ValueError, "input array should be float32");
123      return 0;
124    }
125
126    // no need to really allocate fvec, just its struct member
127    long length = PyArray_DIM ((PyArrayObject *)input, 1);
128    if (length <= 0) {
129      PyErr_SetString (PyExc_ValueError, "input array dimension 1 should be greater than 0");
130      return 0;
131    }
132    long height = PyArray_DIM ((PyArrayObject *)input, 0);
133    if (height <= 0) {
134      PyErr_SetString (PyExc_ValueError, "input array dimension 0 should be greater than 0");
135      return 0;
136    }
137
138  } else if (PyObject_TypeCheck (input, &PyList_Type)) {
139    PyErr_SetString (PyExc_ValueError, "can not convert list to fmat");
140    return 0;
141  } else {
142    PyErr_SetString (PyExc_ValueError, "can only accept matrix of float as input");
143    return 0;
144  }
145
146  if (mat->height != (uint_t)PyArray_DIM ((PyArrayObject *)input, 0)) {
147    /*
148    free(mat->data);
149    mat->height = (uint_t)PyArray_DIM ((PyArrayObject *)input, 0);
150    mat->data = (smpl_t **)malloc(sizeof(smpl_t*) * mat->height);
151    */
152    PyErr_Format(PyExc_ValueError, "too many rows, %d but %ld expected",
153                      mat->height, PyArray_DIM ((PyArrayObject *)input, 0) );
154    return 0;
155  }
156
157  mat->length = (uint_t)PyArray_DIM ((PyArrayObject *)input, 1);
158  for (i=0; i< mat->height; i++) {
159    mat->data[i] = (smpl_t*)PyArray_GETPTR1 ((PyArrayObject *)input, i);
160  }
161  return 1;
162}
Note: See TracBrowser for help on using the repository browser.