[96fb8ad] | 1 | /* |
---|
[a6db140] | 2 | Copyright (C) 2003-2009 Paul Brossier <piem@aubio.org> |
---|
[96fb8ad] | 3 | |
---|
[a6db140] | 4 | This file is part of aubio. |
---|
[96fb8ad] | 5 | |
---|
[a6db140] | 6 | aubio is free software: you can redistribute it and/or modify |
---|
| 7 | it under the terms of the GNU General Public License as published by |
---|
| 8 | the Free Software Foundation, either version 3 of the License, or |
---|
| 9 | (at your option) any later version. |
---|
[96fb8ad] | 10 | |
---|
[a6db140] | 11 | aubio is distributed in the hope that it will be useful, |
---|
| 12 | but WITHOUT ANY WARRANTY; without even the implied warranty of |
---|
| 13 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
---|
| 14 | GNU General Public License for more details. |
---|
| 15 | |
---|
| 16 | You should have received a copy of the GNU General Public License |
---|
| 17 | along with aubio. If not, see <http://www.gnu.org/licenses/>. |
---|
[96fb8ad] | 18 | |
---|
| 19 | */ |
---|
| 20 | |
---|
| 21 | #include "aubio_priv.h" |
---|
[d1ec8cb] | 22 | #include "cvec.h" |
---|
[96fb8ad] | 23 | |
---|
[1120f86] | 24 | cvec_t * new_cvec(uint_t length) { |
---|
[c21acb9] | 25 | cvec_t * s; |
---|
[767990e] | 26 | if ((sint_t)length <= 0) { |
---|
| 27 | return NULL; |
---|
| 28 | } |
---|
[c21acb9] | 29 | s = AUBIO_NEW(cvec_t); |
---|
[96fb8ad] | 30 | s->length = length/2 + 1; |
---|
[66fb3ea] | 31 | s->norm = AUBIO_ARRAY(smpl_t,s->length); |
---|
| 32 | s->phas = AUBIO_ARRAY(smpl_t,s->length); |
---|
[96fb8ad] | 33 | return s; |
---|
| 34 | } |
---|
| 35 | |
---|
| 36 | void del_cvec(cvec_t *s) { |
---|
| 37 | AUBIO_FREE(s->norm); |
---|
| 38 | AUBIO_FREE(s->phas); |
---|
| 39 | AUBIO_FREE(s); |
---|
| 40 | } |
---|
[7c206df] | 41 | |
---|
[5d10ac1] | 42 | void cvec_norm_set_sample (cvec_t *s, smpl_t data, uint_t position) { |
---|
[66fb3ea] | 43 | s->norm[position] = data; |
---|
[7c206df] | 44 | } |
---|
[5d10ac1] | 45 | |
---|
| 46 | void cvec_phas_set_sample (cvec_t *s, smpl_t data, uint_t position) { |
---|
[66fb3ea] | 47 | s->phas[position] = data; |
---|
[7c206df] | 48 | } |
---|
[5d10ac1] | 49 | |
---|
| 50 | smpl_t cvec_norm_get_sample (cvec_t *s, uint_t position) { |
---|
[66fb3ea] | 51 | return s->norm[position]; |
---|
[7c206df] | 52 | } |
---|
[5d10ac1] | 53 | |
---|
| 54 | smpl_t cvec_phas_get_sample (cvec_t *s, uint_t position) { |
---|
[66fb3ea] | 55 | return s->phas[position]; |
---|
[7c206df] | 56 | } |
---|
[5d10ac1] | 57 | |
---|
[1120f86] | 58 | smpl_t * cvec_norm_get_data (const cvec_t *s) { |
---|
[7c206df] | 59 | return s->norm; |
---|
| 60 | } |
---|
[5d10ac1] | 61 | |
---|
[1120f86] | 62 | smpl_t * cvec_phas_get_data (const cvec_t *s) { |
---|
[7c206df] | 63 | return s->phas; |
---|
| 64 | } |
---|
[d1ec8cb] | 65 | |
---|
[55b7cb4] | 66 | /* helper functions */ |
---|
| 67 | |
---|
[1120f86] | 68 | void cvec_print(const cvec_t *s) { |
---|
[66fb3ea] | 69 | uint_t j; |
---|
| 70 | AUBIO_MSG("norm: "); |
---|
| 71 | for (j=0; j< s->length; j++) { |
---|
| 72 | AUBIO_MSG(AUBIO_SMPL_FMT " ", s->norm[j]); |
---|
| 73 | } |
---|
| 74 | AUBIO_MSG("\n"); |
---|
| 75 | AUBIO_MSG("phas: "); |
---|
| 76 | for (j=0; j< s->length; j++) { |
---|
| 77 | AUBIO_MSG(AUBIO_SMPL_FMT " ", s->phas[j]); |
---|
[55b7cb4] | 78 | } |
---|
[66fb3ea] | 79 | AUBIO_MSG("\n"); |
---|
[55b7cb4] | 80 | } |
---|
| 81 | |
---|
[1120f86] | 82 | void cvec_copy(const cvec_t *s, cvec_t *t) { |
---|
[39a7b26] | 83 | if (s->length != t->length) { |
---|
| 84 | AUBIO_ERR("trying to copy %d elements to %d elements \n", |
---|
| 85 | s->length, t->length); |
---|
| 86 | return; |
---|
| 87 | } |
---|
| 88 | #if HAVE_MEMCPY_HACKS |
---|
| 89 | memcpy(t->norm, s->norm, t->length * sizeof(smpl_t)); |
---|
| 90 | memcpy(t->phas, s->phas, t->length * sizeof(smpl_t)); |
---|
| 91 | #else |
---|
| 92 | uint_t j; |
---|
| 93 | for (j=0; j< t->length; j++) { |
---|
| 94 | t->norm[j] = s->norm[j]; |
---|
| 95 | t->phas[j] = s->phas[j]; |
---|
| 96 | } |
---|
| 97 | #endif |
---|
| 98 | } |
---|
| 99 | |
---|
[5d10ac1] | 100 | void cvec_norm_set_all (cvec_t *s, smpl_t val) { |
---|
[66fb3ea] | 101 | uint_t j; |
---|
| 102 | for (j=0; j< s->length; j++) { |
---|
| 103 | s->norm[j] = val; |
---|
[012466b] | 104 | } |
---|
| 105 | } |
---|
| 106 | |
---|
[5d10ac1] | 107 | void cvec_norm_zeros(cvec_t *s) { |
---|
[39a7b26] | 108 | #if HAVE_MEMCPY_HACKS |
---|
| 109 | memset(s->norm, 0, s->length * sizeof(smpl_t)); |
---|
| 110 | #else |
---|
[5d10ac1] | 111 | cvec_norm_set_all (s, 0.); |
---|
[39a7b26] | 112 | #endif |
---|
[012466b] | 113 | } |
---|
| 114 | |
---|
[5d10ac1] | 115 | void cvec_norm_ones(cvec_t *s) { |
---|
| 116 | cvec_norm_set_all (s, 1.); |
---|
[012466b] | 117 | } |
---|
| 118 | |
---|
[5d10ac1] | 119 | void cvec_phas_set_all (cvec_t *s, smpl_t val) { |
---|
[39a7b26] | 120 | uint_t j; |
---|
| 121 | for (j=0; j< s->length; j++) { |
---|
| 122 | s->phas[j] = val; |
---|
| 123 | } |
---|
| 124 | } |
---|
| 125 | |
---|
[5d10ac1] | 126 | void cvec_phas_zeros(cvec_t *s) { |
---|
[39a7b26] | 127 | #if HAVE_MEMCPY_HACKS |
---|
| 128 | memset(s->phas, 0, s->length * sizeof(smpl_t)); |
---|
| 129 | #else |
---|
[5d10ac1] | 130 | cvec_phas_set_all (s, 0.); |
---|
[39a7b26] | 131 | #endif |
---|
| 132 | } |
---|
| 133 | |
---|
[5d10ac1] | 134 | void cvec_phas_ones(cvec_t *s) { |
---|
| 135 | cvec_phas_set_all (s, 1.); |
---|
[39a7b26] | 136 | } |
---|
| 137 | |
---|
| 138 | void cvec_zeros(cvec_t *s) { |
---|
[5d10ac1] | 139 | cvec_norm_zeros(s); |
---|
| 140 | cvec_phas_zeros(s); |
---|
[39a7b26] | 141 | } |
---|