[88199ce] | 1 | /* |
---|
[06cae6c] | 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 |
---|
[06cae6c] | 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, |
---|
[06cae6c] | 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. |
---|
[88199ce] | 16 | |
---|
[06cae6c] | 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 | |
---|
[06cae6c] | 20 | */ |
---|
[f918cb9] | 21 | |
---|
[7c6c806] | 22 | #include "aubio_priv.h" |
---|
[6c7d49b] | 23 | #include "fvec.h" |
---|
[d95ff38] | 24 | #include "fmat.h" |
---|
[6c7d49b] | 25 | #include "cvec.h" |
---|
[0e30a12] | 26 | #include "vecutils.h" |
---|
[32d6958] | 27 | #include "spectral/filterbank.h" |
---|
[17961b0] | 28 | #include "mathutils.h" |
---|
[88199ce] | 29 | |
---|
[8708556] | 30 | /** \brief A structure to store a set of n_filters filters of lenghts win_s */ |
---|
[e6c11e3] | 31 | struct _aubio_filterbank_t |
---|
[3b3b03e] | 32 | { |
---|
| 33 | uint_t win_s; |
---|
| 34 | uint_t n_filters; |
---|
[d95ff38] | 35 | fmat_t *filters; |
---|
[0e30a12] | 36 | smpl_t norm; |
---|
| 37 | smpl_t power; |
---|
[7c6c806] | 38 | }; |
---|
| 39 | |
---|
[3b3b03e] | 40 | aubio_filterbank_t * |
---|
| 41 | new_aubio_filterbank (uint_t n_filters, uint_t win_s) |
---|
| 42 | { |
---|
[06cae6c] | 43 | /* allocate space for filterbank object */ |
---|
[3b3b03e] | 44 | aubio_filterbank_t *fb = AUBIO_NEW (aubio_filterbank_t); |
---|
| 45 | fb->win_s = win_s; |
---|
| 46 | fb->n_filters = n_filters; |
---|
[88199ce] | 47 | |
---|
[d95ff38] | 48 | /* allocate filter tables, a matrix of length win_s and of height n_filters */ |
---|
[4ed0ed1] | 49 | fb->filters = new_fmat (n_filters, win_s / 2 + 1); |
---|
[88199ce] | 50 | |
---|
[0e30a12] | 51 | fb->norm = 1; |
---|
| 52 | |
---|
| 53 | fb->power = 1; |
---|
| 54 | |
---|
[b276dee] | 55 | return fb; |
---|
| 56 | } |
---|
| 57 | |
---|
[3b3b03e] | 58 | void |
---|
| 59 | del_aubio_filterbank (aubio_filterbank_t * fb) |
---|
| 60 | { |
---|
[d95ff38] | 61 | del_fmat (fb->filters); |
---|
[3b3b03e] | 62 | AUBIO_FREE (fb); |
---|
[06cae6c] | 63 | } |
---|
[95a64c7] | 64 | |
---|
[3b3b03e] | 65 | void |
---|
[feb694b] | 66 | aubio_filterbank_do (aubio_filterbank_t * f, const cvec_t * in, fvec_t * out) |
---|
[3b3b03e] | 67 | { |
---|
[06cae6c] | 68 | /* apply filter to all input channel, provided out has enough channels */ |
---|
[fc2d7fb] | 69 | //uint_t max_filters = MIN (f->n_filters, out->length); |
---|
| 70 | //uint_t max_length = MIN (in->length, f->filters->length); |
---|
| 71 | |
---|
| 72 | // view cvec->norm as fvec->data |
---|
| 73 | fvec_t tmp; |
---|
| 74 | tmp.length = in->length; |
---|
| 75 | tmp.data = in->norm; |
---|
| 76 | |
---|
[0e30a12] | 77 | if (f->power != 1.) fvec_pow(&tmp, f->power); |
---|
| 78 | |
---|
[fc2d7fb] | 79 | fmat_vecmul(f->filters, &tmp, out); |
---|
[b276dee] | 80 | |
---|
| 81 | return; |
---|
[fe28ff3] | 82 | } |
---|
[17961b0] | 83 | |
---|
[d95ff38] | 84 | fmat_t * |
---|
[feb694b] | 85 | aubio_filterbank_get_coeffs (const aubio_filterbank_t * f) |
---|
[3b3b03e] | 86 | { |
---|
[06cae6c] | 87 | return f->filters; |
---|
[17961b0] | 88 | } |
---|
[33916b8] | 89 | |
---|
| 90 | uint_t |
---|
[feb694b] | 91 | aubio_filterbank_set_coeffs (aubio_filterbank_t * f, const fmat_t * filter_coeffs) |
---|
[33916b8] | 92 | { |
---|
[d95ff38] | 93 | fmat_copy(filter_coeffs, f->filters); |
---|
[33916b8] | 94 | return 0; |
---|
| 95 | } |
---|
[0e30a12] | 96 | |
---|
| 97 | uint_t |
---|
| 98 | aubio_filterbank_set_norm (aubio_filterbank_t *f, smpl_t norm) |
---|
| 99 | { |
---|
| 100 | if (norm != 0 && norm != 1) return AUBIO_FAIL; |
---|
| 101 | f->norm = norm; |
---|
| 102 | return AUBIO_OK; |
---|
| 103 | } |
---|
| 104 | |
---|
| 105 | smpl_t |
---|
| 106 | aubio_filterbank_get_norm (aubio_filterbank_t *f) |
---|
| 107 | { |
---|
| 108 | return f->norm; |
---|
| 109 | } |
---|
| 110 | |
---|
| 111 | uint_t |
---|
| 112 | aubio_filterbank_set_power (aubio_filterbank_t *f, smpl_t power) |
---|
| 113 | { |
---|
| 114 | f->power = power; |
---|
| 115 | return AUBIO_OK; |
---|
| 116 | } |
---|
| 117 | |
---|
| 118 | smpl_t |
---|
| 119 | aubio_filterbank_get_power (aubio_filterbank_t *f) |
---|
| 120 | { |
---|
| 121 | return f->norm; |
---|
| 122 | } |
---|