[96fb8ad] | 1 | /* |
---|
| 2 | Copyright (C) 2003 Paul Brossier |
---|
| 3 | |
---|
| 4 | This program is free software; you can redistribute it and/or modify |
---|
| 5 | it under the terms of the GNU General Public License as published by |
---|
| 6 | the Free Software Foundation; either version 2 of the License, or |
---|
| 7 | (at your option) any later version. |
---|
| 8 | |
---|
| 9 | This program is distributed in the hope that it will be useful, |
---|
| 10 | but WITHOUT ANY WARRANTY; without even the implied warranty of |
---|
| 11 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
---|
| 12 | GNU General Public License for more details. |
---|
| 13 | |
---|
| 14 | You should have received a copy of the GNU General Public License |
---|
| 15 | along with this program; if not, write to the Free Software |
---|
| 16 | Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. |
---|
| 17 | |
---|
| 18 | */ |
---|
| 19 | |
---|
| 20 | #include "aubio_priv.h" |
---|
| 21 | #include "sample.h" |
---|
| 22 | #include "fft.h" |
---|
| 23 | #include "mathutils.h" |
---|
| 24 | #include "phasevoc.h" |
---|
| 25 | |
---|
| 26 | /** phasevocoder internal object */ |
---|
| 27 | struct _aubio_pvoc_t { |
---|
[685412e] | 28 | uint_t win_s; /** grain length */ |
---|
| 29 | uint_t hop_s; /** overlap step */ |
---|
| 30 | uint_t channels; /** number of channels */ |
---|
| 31 | aubio_mfft_t * fft; /** spectral data */ |
---|
| 32 | fvec_t * synth; /**cur output grain [win_s] */ |
---|
| 33 | fvec_t * synthold; /**last input frame [win_s-hop_s] */ |
---|
| 34 | fvec_t * data; /**current input grain [win_s] */ |
---|
| 35 | fvec_t * dataold; /**last input frame [win_s-hop_s] */ |
---|
[b1c948b] | 36 | smpl_t * w; /** grain window [win_s] */ |
---|
[96fb8ad] | 37 | }; |
---|
| 38 | |
---|
| 39 | |
---|
| 40 | /** returns data and dataold slided by hop_s */ |
---|
[685412e] | 41 | static void aubio_pvoc_swapbuffers(smpl_t * data, smpl_t * dataold, const |
---|
| 42 | smpl_t * datanew, uint_t win_s, uint_t hop_s); |
---|
[96fb8ad] | 43 | |
---|
[685412e] | 44 | /** do additive synthesis from 'old' and 'cur' */ |
---|
| 45 | static void aubio_pvoc_addsynth(const smpl_t * synth, smpl_t * synthold, |
---|
| 46 | smpl_t * synthnew, uint_t win_s, uint_t hop_s); |
---|
[96fb8ad] | 47 | |
---|
| 48 | void aubio_pvoc_do(aubio_pvoc_t *pv, fvec_t * datanew, cvec_t *fftgrain) { |
---|
[685412e] | 49 | uint_t i,j; |
---|
| 50 | for (i=0; i<pv->channels; i++) { |
---|
| 51 | /* slide */ |
---|
| 52 | aubio_pvoc_swapbuffers(pv->data->data[i],pv->dataold->data[i], |
---|
| 53 | datanew->data[i],pv->win_s,pv->hop_s); |
---|
| 54 | /* windowing */ |
---|
| 55 | for (j=0; j<pv->win_s; j++) pv->data->data[i][j] *= pv->w[j]; |
---|
| 56 | } |
---|
| 57 | /* shift */ |
---|
| 58 | vec_shift(pv->data); |
---|
| 59 | /* calculate fft */ |
---|
| 60 | aubio_mfft_do (pv->fft,pv->data,fftgrain); |
---|
[96fb8ad] | 61 | } |
---|
| 62 | |
---|
| 63 | void aubio_pvoc_rdo(aubio_pvoc_t *pv,cvec_t * fftgrain, fvec_t * synthnew) { |
---|
[685412e] | 64 | uint_t i; |
---|
| 65 | /* calculate rfft */ |
---|
| 66 | aubio_mfft_rdo(pv->fft,fftgrain,pv->synth); |
---|
| 67 | /* unshift */ |
---|
| 68 | vec_shift(pv->synth); |
---|
| 69 | for (i=0; i<pv->channels; i++) { |
---|
| 70 | aubio_pvoc_addsynth(pv->synth->data[i],pv->synthold->data[i], |
---|
| 71 | synthnew->data[i],pv->win_s,pv->hop_s); |
---|
| 72 | } |
---|
[96fb8ad] | 73 | } |
---|
| 74 | |
---|
| 75 | aubio_pvoc_t * new_aubio_pvoc (uint_t win_s, uint_t hop_s, uint_t channels) { |
---|
[685412e] | 76 | aubio_pvoc_t * pv = AUBIO_NEW(aubio_pvoc_t); |
---|
| 77 | |
---|
| 78 | if (win_s < 2*hop_s) { |
---|
| 79 | AUBIO_ERR("Hop size bigger than half the window size!\n"); |
---|
| 80 | AUBIO_ERR("Resetting hop size to half the window size.\n"); |
---|
| 81 | hop_s = win_s / 2; |
---|
| 82 | } |
---|
| 83 | |
---|
| 84 | if (hop_s < 1) { |
---|
| 85 | AUBIO_ERR("Hop size is smaller than 1!\n"); |
---|
| 86 | AUBIO_ERR("Resetting hop size to half the window size.\n"); |
---|
| 87 | hop_s = win_s / 2; |
---|
| 88 | } |
---|
| 89 | |
---|
| 90 | pv->fft = new_aubio_mfft(win_s,channels); |
---|
| 91 | |
---|
| 92 | /* remember old */ |
---|
| 93 | pv->data = new_fvec (win_s, channels); |
---|
| 94 | pv->synth = new_fvec (win_s, channels); |
---|
| 95 | |
---|
| 96 | /* new input output */ |
---|
| 97 | pv->dataold = new_fvec (win_s-hop_s, channels); |
---|
| 98 | pv->synthold = new_fvec (win_s-hop_s, channels); |
---|
| 99 | pv->w = AUBIO_ARRAY(smpl_t,win_s); |
---|
| 100 | aubio_window(pv->w,win_s,aubio_win_hanningz); |
---|
| 101 | |
---|
| 102 | pv->channels = channels; |
---|
| 103 | pv->hop_s = hop_s; |
---|
| 104 | pv->win_s = win_s; |
---|
| 105 | |
---|
| 106 | return pv; |
---|
[96fb8ad] | 107 | } |
---|
| 108 | |
---|
[f88a326] | 109 | void del_aubio_pvoc(aubio_pvoc_t *pv) { |
---|
[685412e] | 110 | del_fvec(pv->data); |
---|
| 111 | del_fvec(pv->synth); |
---|
| 112 | del_fvec(pv->dataold); |
---|
| 113 | del_fvec(pv->synthold); |
---|
| 114 | del_aubio_mfft(pv->fft); |
---|
| 115 | AUBIO_FREE(pv->w); |
---|
| 116 | AUBIO_FREE(pv); |
---|
[96fb8ad] | 117 | } |
---|
| 118 | |
---|
[f88a326] | 119 | static void aubio_pvoc_swapbuffers(smpl_t * data, smpl_t * dataold, |
---|
[685412e] | 120 | const smpl_t * datanew, uint_t win_s, uint_t hop_s) |
---|
[96fb8ad] | 121 | { |
---|
[685412e] | 122 | uint_t i; |
---|
| 123 | for (i=0;i<win_s-hop_s;i++) |
---|
| 124 | data[i] = dataold[i]; |
---|
| 125 | for (i=0;i<hop_s;i++) |
---|
| 126 | data[win_s-hop_s+i] = datanew[i]; |
---|
| 127 | for (i=0;i<win_s-hop_s;i++) |
---|
| 128 | dataold[i] = data[i+hop_s]; |
---|
[96fb8ad] | 129 | } |
---|
| 130 | |
---|
[f88a326] | 131 | static void aubio_pvoc_addsynth(const smpl_t * synth, smpl_t * synthold, |
---|
| 132 | smpl_t * synthnew, uint_t win_s, uint_t hop_s) |
---|
[96fb8ad] | 133 | { |
---|
[685412e] | 134 | uint_t i; |
---|
| 135 | smpl_t scale = 2*hop_s/(win_s+.0); |
---|
| 136 | /* add new synth to old one and put result in synthnew */ |
---|
| 137 | for (i=0;i<hop_s;i++) |
---|
| 138 | synthnew[i] = synthold[i]+synth[i]*scale; |
---|
| 139 | /* shift synthold */ |
---|
| 140 | for (i=0;i<win_s-2*hop_s;i++) |
---|
| 141 | synthold[i] = synthold[i+hop_s]; |
---|
| 142 | /* erase last frame in synthold */ |
---|
| 143 | for (i=win_s-hop_s;i<win_s;i++) |
---|
| 144 | synthold[i-hop_s]=0.; |
---|
| 145 | /* additive synth */ |
---|
| 146 | for (i=0;i<win_s-hop_s;i++) |
---|
| 147 | synthold[i] += synth[i+hop_s]*scale; |
---|
[96fb8ad] | 148 | } |
---|
| 149 | |
---|