[1fe822d] | 1 | #include "aubio_priv.h" |
---|
| 2 | #include "fmat.h" |
---|
| 3 | #include "tensor.h" |
---|
| 4 | |
---|
| 5 | aubio_tensor_t *new_aubio_tensor(uint_t n_dims, uint_t *dims) |
---|
| 6 | { |
---|
| 7 | aubio_tensor_t *c = AUBIO_NEW(aubio_tensor_t); |
---|
| 8 | uint_t i; |
---|
| 9 | |
---|
| 10 | if ((sint_t)n_dims <= 0) goto failure; |
---|
| 11 | for (i = 0; i < n_dims; i++) { |
---|
| 12 | if ((sint_t)dims[i] <= 0) goto failure; |
---|
| 13 | } |
---|
| 14 | |
---|
| 15 | c->n_dims = n_dims; |
---|
| 16 | c->items_per_row = 1; |
---|
| 17 | //c->dims = AUBIO_ARRAY(uint_t, n_dims); |
---|
| 18 | c->dims[0] = dims[0]; |
---|
| 19 | for (i = 1; i < n_dims; i++) { |
---|
| 20 | c->dims[i] = dims[i]; |
---|
| 21 | c->items_per_row *= dims[i]; |
---|
| 22 | } |
---|
| 23 | c->n_items = c->items_per_row * dims[0]; |
---|
| 24 | c->data = AUBIO_ARRAY(smpl_t*, dims[0]); |
---|
| 25 | c->data[0] = AUBIO_ARRAY(smpl_t, c->n_items); |
---|
| 26 | for (i = 1; i < c->dims[0]; i++) { |
---|
| 27 | c->data[i] = c->data[0] + i * c->items_per_row; |
---|
| 28 | } |
---|
| 29 | |
---|
| 30 | return c; |
---|
| 31 | |
---|
| 32 | failure: |
---|
| 33 | del_aubio_tensor(c); |
---|
| 34 | return NULL; |
---|
| 35 | } |
---|
| 36 | |
---|
| 37 | void del_aubio_tensor(aubio_tensor_t *c) |
---|
| 38 | { |
---|
| 39 | AUBIO_ASSERT(c); |
---|
| 40 | if (c->data) { |
---|
| 41 | if (c->data[0]) { |
---|
| 42 | AUBIO_FREE(c->data[0]); |
---|
| 43 | } |
---|
| 44 | AUBIO_FREE(c->data); |
---|
| 45 | } |
---|
| 46 | //if (c->dims) |
---|
| 47 | // AUBIO_FREE(c->dims); |
---|
| 48 | AUBIO_FREE(c); |
---|
| 49 | } |
---|
| 50 | |
---|
| 51 | uint_t aubio_tensor_as_fvec(aubio_tensor_t *c, fvec_t *o) { |
---|
| 52 | if (c->n_dims != 1) return AUBIO_FAIL; |
---|
| 53 | if (c->dims[0] <= 0) return AUBIO_FAIL; |
---|
| 54 | o->length = c->dims[0]; |
---|
| 55 | o->data = c->data[0]; |
---|
| 56 | return AUBIO_OK; |
---|
| 57 | } |
---|
| 58 | |
---|
| 59 | uint_t aubio_fvec_as_tensor(fvec_t *o, aubio_tensor_t *c) { |
---|
| 60 | if (o == NULL) return AUBIO_FAIL; |
---|
| 61 | c->n_dims = 1; |
---|
| 62 | c->dims[0] = o->length; |
---|
| 63 | c->data = &o->data; |
---|
| 64 | return AUBIO_OK; |
---|
| 65 | } |
---|
| 66 | |
---|
| 67 | uint_t aubio_tensor_as_fmat(aubio_tensor_t *c, fmat_t *o) { |
---|
| 68 | if (c->n_dims != 2) return AUBIO_FAIL; |
---|
| 69 | if (c->dims[0] <= 0) return AUBIO_FAIL; |
---|
| 70 | if (c->dims[1] <= 0) return AUBIO_FAIL; |
---|
| 71 | o->height = c->dims[0]; |
---|
| 72 | o->length = c->dims[1]; |
---|
| 73 | o->data = c->data; |
---|
| 74 | return AUBIO_OK; |
---|
| 75 | } |
---|
| 76 | |
---|
| 77 | uint_t aubio_fmat_as_tensor(fmat_t *o, aubio_tensor_t *c) { |
---|
| 78 | if (o == NULL) return AUBIO_FAIL; |
---|
| 79 | if (c == NULL) return AUBIO_FAIL; |
---|
| 80 | c->n_dims = 2; |
---|
| 81 | c->dims[0] = o->height; |
---|
| 82 | c->dims[1] = o->length; |
---|
| 83 | c->data = o->data; |
---|
| 84 | return AUBIO_OK; |
---|
| 85 | } |
---|
| 86 | |
---|
| 87 | smpl_t aubio_tensor_max(aubio_tensor_t *t) |
---|
| 88 | { |
---|
| 89 | uint_t i; |
---|
| 90 | smpl_t max = -1000000; |
---|
| 91 | for (i = 0; i < t->n_items; i++) { |
---|
| 92 | max = MAX(t->data[0][i], max); |
---|
| 93 | } |
---|
| 94 | return max; |
---|
| 95 | } |
---|