[88199ce] | 1 | /* |
---|
[7a84b21] | 2 | Copyright (C) 2007-2009 Paul Brossier <piem@aubio.org> |
---|
| 3 | and Amaury Hazan <ahazan@iua.upf.edu> |
---|
[88199ce] | 4 | |
---|
[1c2e186] | 5 | This file is part of aubio. |
---|
[88199ce] | 6 | |
---|
[1c2e186] | 7 | aubio is free software: you can redistribute it and/or modify |
---|
[7a84b21] | 8 | it under the terms of the GNU General Public License as published by |
---|
| 9 | the Free Software Foundation, either version 3 of the License, or |
---|
| 10 | (at your option) any later version. |
---|
[88199ce] | 11 | |
---|
[1c2e186] | 12 | aubio is distributed in the hope that it will be useful, |
---|
[7a84b21] | 13 | but WITHOUT ANY WARRANTY; without even the implied warranty of |
---|
| 14 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
---|
| 15 | GNU General Public License for more details. |
---|
| 16 | |
---|
| 17 | You should have received a copy of the GNU General Public License |
---|
[1c2e186] | 18 | along with aubio. If not, see <http://www.gnu.org/licenses/>. |
---|
[88199ce] | 19 | |
---|
| 20 | */ |
---|
| 21 | |
---|
[7c6c806d] | 22 | #include "aubio_priv.h" |
---|
[6c7d49b] | 23 | #include "fvec.h" |
---|
[d95ff38] | 24 | #include "fmat.h" |
---|
[6c7d49b] | 25 | #include "cvec.h" |
---|
[d37400c] | 26 | #include "mathutils.h" |
---|
[d0dca26] | 27 | #include "vecutils.h" |
---|
[32d6958] | 28 | #include "spectral/fft.h" |
---|
[d90db7d4] | 29 | #include "spectral/filterbank.h" |
---|
[06cae6c] | 30 | #include "spectral/filterbank_mel.h" |
---|
[d8b1161] | 31 | #include "spectral/dct.h" |
---|
[32d6958] | 32 | #include "spectral/mfcc.h" |
---|
[88199ce] | 33 | |
---|
[7a84b21] | 34 | /** Internal structure for mfcc object */ |
---|
[88199ce] | 35 | |
---|
[d99d819] | 36 | struct _aubio_mfcc_t |
---|
[7a84b21] | 37 | { |
---|
[34d72f0] | 38 | uint_t win_s; /** grain length */ |
---|
| 39 | uint_t samplerate; /** sample rate (needed?) */ |
---|
[f2a0769] | 40 | uint_t n_filters; /** number of filters */ |
---|
[7a46bf6] | 41 | uint_t n_coefs; /** number of coefficients (<= n_filters/2 +1) */ |
---|
[7a84b21] | 42 | aubio_filterbank_t *fb; /** filter bank */ |
---|
| 43 | fvec_t *in_dct; /** input buffer for dct * [fb->n_filters] */ |
---|
[e744416] | 44 | aubio_dct_t *dct; /** dct object */ |
---|
| 45 | fvec_t *output; /** dct output */ |
---|
[d66d2ac] | 46 | smpl_t scale; |
---|
[8708556] | 47 | }; |
---|
| 48 | |
---|
| 49 | |
---|
[7a84b21] | 50 | aubio_mfcc_t * |
---|
[41bf913] | 51 | new_aubio_mfcc (uint_t win_s, uint_t n_filters, uint_t n_coefs, |
---|
| 52 | uint_t samplerate) |
---|
[7a84b21] | 53 | { |
---|
| 54 | |
---|
| 55 | /* allocate space for mfcc object */ |
---|
| 56 | aubio_mfcc_t *mfcc = AUBIO_NEW (aubio_mfcc_t); |
---|
[c185ebb] | 57 | |
---|
[7a84b21] | 58 | mfcc->win_s = win_s; |
---|
| 59 | mfcc->samplerate = samplerate; |
---|
| 60 | mfcc->n_filters = n_filters; |
---|
| 61 | mfcc->n_coefs = n_coefs; |
---|
[cfe4038] | 62 | |
---|
[7a84b21] | 63 | /* filterbank allocation */ |
---|
| 64 | mfcc->fb = new_aubio_filterbank (n_filters, mfcc->win_s); |
---|
[1c1dae7] | 65 | if (n_filters == 40) |
---|
| 66 | aubio_filterbank_set_mel_coeffs_slaney (mfcc->fb, samplerate); |
---|
| 67 | else |
---|
| 68 | aubio_filterbank_set_mel_coeffs(mfcc->fb, samplerate, |
---|
| 69 | 0, samplerate/2.); |
---|
[8708556] | 70 | |
---|
[7a84b21] | 71 | /* allocating buffers */ |
---|
[d95ff38] | 72 | mfcc->in_dct = new_fvec (n_filters); |
---|
[8708556] | 73 | |
---|
[d8b1161] | 74 | mfcc->dct = new_aubio_dct (n_filters); |
---|
| 75 | mfcc->output = new_fvec (n_filters); |
---|
[8708556] | 76 | |
---|
[d66d2ac] | 77 | mfcc->scale = 1.; |
---|
| 78 | |
---|
[8708556] | 79 | return mfcc; |
---|
[163d159] | 80 | } |
---|
[8708556] | 81 | |
---|
[7a84b21] | 82 | void |
---|
| 83 | del_aubio_mfcc (aubio_mfcc_t * mf) |
---|
| 84 | { |
---|
| 85 | |
---|
| 86 | /* delete filterbank */ |
---|
| 87 | del_aubio_filterbank (mf->fb); |
---|
| 88 | |
---|
| 89 | /* delete buffers */ |
---|
| 90 | del_fvec (mf->in_dct); |
---|
[d8b1161] | 91 | del_aubio_dct (mf->dct); |
---|
| 92 | del_fvec (mf->output); |
---|
[7a84b21] | 93 | |
---|
| 94 | /* delete mfcc object */ |
---|
| 95 | AUBIO_FREE (mf); |
---|
[88199ce] | 96 | } |
---|
| 97 | |
---|
[177f09a] | 98 | |
---|
[7a84b21] | 99 | void |
---|
[feb694b] | 100 | aubio_mfcc_do (aubio_mfcc_t * mf, const cvec_t * in, fvec_t * out) |
---|
[7a84b21] | 101 | { |
---|
[3aa60b2] | 102 | fvec_t tmp; |
---|
[d66d2ac] | 103 | |
---|
[7a84b21] | 104 | /* compute filterbank */ |
---|
| 105 | aubio_filterbank_do (mf->fb, in, mf->in_dct); |
---|
[c185ebb] | 106 | |
---|
[d37400c] | 107 | /* compute log10 */ |
---|
| 108 | fvec_log10 (mf->in_dct); |
---|
| 109 | |
---|
[d66d2ac] | 110 | if (mf->scale != 1) fvec_mul (mf->in_dct, mf->scale); |
---|
[d37400c] | 111 | |
---|
[73e8f65] | 112 | /* compute mfccs */ |
---|
[d8b1161] | 113 | aubio_dct_do(mf->dct, mf->in_dct, mf->output); |
---|
| 114 | // copy only first n_coeffs elements |
---|
| 115 | // TODO assert mf->output->length == n_coeffs |
---|
| 116 | tmp.data = mf->output->data; |
---|
| 117 | tmp.length = out->length; |
---|
| 118 | fvec_copy(&tmp, out); |
---|
[cfe4038] | 119 | |
---|
[c185ebb] | 120 | return; |
---|
| 121 | } |
---|
[517630f] | 122 | |
---|
| 123 | uint_t aubio_mfcc_set_power (aubio_mfcc_t *mf, smpl_t power) |
---|
| 124 | { |
---|
| 125 | return aubio_filterbank_set_power(mf->fb, power); |
---|
| 126 | } |
---|
| 127 | |
---|
| 128 | uint_t aubio_mfcc_get_power (aubio_mfcc_t *mf) |
---|
| 129 | { |
---|
| 130 | return aubio_filterbank_get_power(mf->fb); |
---|
| 131 | } |
---|
| 132 | |
---|
[d66d2ac] | 133 | uint_t aubio_mfcc_set_scale (aubio_mfcc_t *mf, smpl_t scale) |
---|
| 134 | { |
---|
| 135 | mf->scale = scale; |
---|
| 136 | return AUBIO_OK; |
---|
| 137 | } |
---|
| 138 | |
---|
| 139 | uint_t aubio_mfcc_get_scale (aubio_mfcc_t *mf) |
---|
| 140 | { |
---|
| 141 | return mf->scale; |
---|
| 142 | } |
---|
| 143 | |
---|
[517630f] | 144 | uint_t aubio_mfcc_set_mel_coeffs (aubio_mfcc_t *mf, smpl_t freq_min, |
---|
| 145 | smpl_t freq_max) |
---|
| 146 | { |
---|
| 147 | return aubio_filterbank_set_mel_coeffs(mf->fb, mf->samplerate, |
---|
| 148 | freq_min, freq_max); |
---|
| 149 | } |
---|
| 150 | |
---|
| 151 | uint_t aubio_mfcc_set_mel_coeffs_htk (aubio_mfcc_t *mf, smpl_t freq_min, |
---|
| 152 | smpl_t freq_max) |
---|
| 153 | { |
---|
| 154 | return aubio_filterbank_set_mel_coeffs_htk(mf->fb, mf->samplerate, |
---|
| 155 | freq_min, freq_max); |
---|
| 156 | } |
---|
[10fafc2] | 157 | |
---|
[c879811] | 158 | uint_t aubio_mfcc_set_mel_coeffs_slaney (aubio_mfcc_t *mf) |
---|
[10fafc2] | 159 | { |
---|
| 160 | return aubio_filterbank_set_mel_coeffs_slaney (mf->fb, mf->samplerate); |
---|
| 161 | } |
---|