Changes in / [1da7e08:c721874]
- Files:
-
- 6 added
- 1 deleted
- 7 edited
Legend:
- Unmodified
- Added
- Removed
-
src/aubio_priv.h
r1da7e08 rc721874 1 1 /* 2 Copyright (C) 2003Paul Brossier2 Copyright (C) 2003-2007 Paul Brossier 3 3 4 5 6 7 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 8 9 10 11 12 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 13 14 15 16 17 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 18 */ 19 19 … … 71 71 72 72 /* Memory management */ 73 #define AUBIO_MALLOC(_n) 74 #define AUBIO_REALLOC(_p,_n) 75 #define AUBIO_NEW(_t) 76 #define AUBIO_ARRAY(_t,_n) 77 #define AUBIO_MEMCPY(_dst,_src,_n) 78 #define AUBIO_MEMSET(_dst,_src,_t) 79 #define AUBIO_FREE(_p) free(_p)73 #define AUBIO_MALLOC(_n) malloc(_n) 74 #define AUBIO_REALLOC(_p,_n) realloc(_p,_n) 75 #define AUBIO_NEW(_t) (_t*)malloc(sizeof(_t)) 76 #define AUBIO_ARRAY(_t,_n) (_t*)malloc((_n)*sizeof(_t)) 77 #define AUBIO_MEMCPY(_dst,_src,_n) memcpy(_dst,_src,_n) 78 #define AUBIO_MEMSET(_dst,_src,_t) memset(_dst,_src,_t) 79 #define AUBIO_FREE(_p) free(_p) 80 80 81 81 -
src/fft.c
r1da7e08 rc721874 24 24 25 25 #if FFTW3F_SUPPORT 26 #define fftw_malloc 27 #define fftw_free 28 #define fftw_execute 29 #define fftw_plan_dft_r2c_1d 30 #define fftw_plan_dft_c2r_1d 31 #define fftw_plan_r2r_1d fftwf_plan_r2r_1d32 #define fftw_plan 33 #define fftw_destroy_plan 26 #define fftw_malloc fftwf_malloc 27 #define fftw_free fftwf_free 28 #define fftw_execute fftwf_execute 29 #define fftw_plan_dft_r2c_1d fftwf_plan_dft_r2c_1d 30 #define fftw_plan_dft_c2r_1d fftwf_plan_dft_c2r_1d 31 #define fftw_plan_r2r_1d fftwf_plan_r2r_1d 32 #define fftw_plan fftwf_plan 33 #define fftw_destroy_plan fftwf_destroy_plan 34 34 #endif 35 35 … … 41 41 42 42 struct _aubio_fft_t { 43 44 45 real_t*in, *out;46 fft_data_t*specdata;47 fftw_planpfw, pbw;43 uint_t fft_size; 44 uint_t channels; 45 real_t *in, *out; 46 fft_data_t *specdata; 47 fftw_plan pfw, pbw; 48 48 }; 49 49 … … 51 51 52 52 aubio_fft_t * new_aubio_fft(uint_t size) { 53 aubio_fft_t * s = AUBIO_NEW(aubio_fft_t); 54 /* allocate memory */ 55 s->in = AUBIO_ARRAY(real_t,size); 56 s->out = AUBIO_ARRAY(real_t,size); 57 s->specdata = (fft_data_t*)fftw_malloc(sizeof(fft_data_t)*size); 58 /* create plans */ 53 aubio_fft_t * s = AUBIO_NEW(aubio_fft_t); 54 /* allocate memory */ 55 s->in = AUBIO_ARRAY(real_t,size); 56 s->out = AUBIO_ARRAY(real_t,size); 57 /* create plans */ 59 58 #ifdef HAVE_COMPLEX_H 60 s->pfw = fftw_plan_dft_r2c_1d(size, s->in, s->specdata, FFTW_ESTIMATE); 61 s->pbw = fftw_plan_dft_c2r_1d(size, s->specdata, s->out, FFTW_ESTIMATE); 59 s->fft_size = size/2+1; 60 s->specdata = (fft_data_t*)fftw_malloc(sizeof(fft_data_t)*s->fft_size); 61 s->pfw = fftw_plan_dft_r2c_1d(size, s->in, s->specdata, FFTW_ESTIMATE); 62 s->pbw = fftw_plan_dft_c2r_1d(size, s->specdata, s->out, FFTW_ESTIMATE); 62 63 #else 63 s->pfw = fftw_plan_r2r_1d(size, s->in, s->specdata, FFTW_R2HC, FFTW_ESTIMATE); 64 s->pbw = fftw_plan_r2r_1d(size, s->specdata, s->out, FFTW_HC2R, FFTW_ESTIMATE); 64 s->fft_size = size; 65 s->specdata = (fft_data_t*)fftw_malloc(sizeof(fft_data_t)*s->fft_size); 66 s->pfw = fftw_plan_r2r_1d(size, s->in, s->specdata, FFTW_R2HC, FFTW_ESTIMATE); 67 s->pbw = fftw_plan_r2r_1d(size, s->specdata, s->out, FFTW_HC2R, FFTW_ESTIMATE); 65 68 #endif 66 69 return s; 67 70 } 68 71 69 72 void del_aubio_fft(aubio_fft_t * s) { 70 71 72 73 74 75 76 73 /* destroy data */ 74 fftw_destroy_plan(s->pfw); 75 fftw_destroy_plan(s->pbw); 76 fftw_free(s->specdata); 77 AUBIO_FREE(s->out); 78 AUBIO_FREE(s->in ); 79 AUBIO_FREE(s); 77 80 } 78 81 79 82 void aubio_fft_do(const aubio_fft_t * s, 80 const smpl_t * data, fft_data_t * spectrum, 81 const uint_t size) { 82 uint_t i; 83 for (i=0;i<size;i++) s->in[i] = data[i]; 84 fftw_execute(s->pfw); 85 for (i=0;i<size;i++) spectrum[i] = s->specdata[i]; 83 const smpl_t * data, fft_data_t * spectrum, const uint_t size) { 84 uint_t i; 85 for (i=0;i<size;i++) s->in[i] = data[i]; 86 fftw_execute(s->pfw); 87 for (i=0; i < s->fft_size; i++) spectrum[i] = s->specdata[i]; 86 88 } 87 89 88 90 void aubio_fft_rdo(const aubio_fft_t * s, 89 const fft_data_t * spectrum, 90 smpl_t * data, 91 const uint_t size) { 92 uint_t i; 93 const smpl_t renorm = 1./(smpl_t)size; 94 for (i=0;i<size;i++) s->specdata[i] = spectrum[i]; 95 fftw_execute(s->pbw); 96 for (i=0;i<size;i++) data[i] = s->out[i]*renorm; 91 const fft_data_t * spectrum, smpl_t * data, const uint_t size) { 92 uint_t i; 93 const smpl_t renorm = 1./(smpl_t)size; 94 for (i=0; i < s->fft_size; i++) s->specdata[i] = spectrum[i]; 95 fftw_execute(s->pbw); 96 for (i=0;i<size;i++) data[i] = s->out[i]*renorm; 97 97 } 98 98 … … 100 100 101 101 void aubio_fft_getnorm(smpl_t * norm, fft_data_t * spectrum, uint_t size) { 102 uint_t i; 103 for (i=0;i<size/2+1;i++) norm[i] = ABSC(spectrum[i]); 104 //for (i=0;i<size/2+1;i++) AUBIO_DBG("%f\n", norm[i]); 102 uint_t i; 103 for (i=0;i<size/2+1;i++) norm[i] = ABSC(spectrum[i]); 105 104 } 106 105 107 106 void aubio_fft_getphas(smpl_t * phas, fft_data_t * spectrum, uint_t size) { 108 uint_t i; 109 for (i=0;i<size/2+1;i++) phas[i] = ARGC(spectrum[i]); 110 //for (i=0;i<size/2+1;i++) AUBIO_DBG("%f\n", phas[i]); 107 uint_t i; 108 for (i=0;i<size/2+1;i++) phas[i] = ARGC(spectrum[i]); 111 109 } 112 110 … … 122 120 123 121 void aubio_fft_getnorm(smpl_t * norm, fft_data_t * spectrum, uint_t size) { 124 125 norm[0] = -spectrum[0];126 for (i=1;i<size/2+1;i++) norm[i] = SQRT(SQR(spectrum[i]) + SQR(spectrum[size-i]));127 //for (i=0;i<size/2+1;i++) AUBIO_DBG("%f\n", norm[i]);122 uint_t i; 123 norm[0] = SQR(spectrum[0]); 124 for (i=1;i<size/2;i++) norm[i] = (SQR(spectrum[i]) + SQR(spectrum[size-i])); 125 norm[size/2] = SQR(spectrum[size/2]); 128 126 } 129 127 130 128 void aubio_fft_getphas(smpl_t * phas, fft_data_t * spectrum, uint_t size) { 131 132 phas[0] = PI;133 134 //for (i=0;i<size/2+1;i++) AUBIO_DBG("%f\n", phas[i]);129 uint_t i; 130 phas[0] = 0; 131 for (i=1;i<size/2+1;i++) phas[i] = atan2f(spectrum[size-i] , spectrum[i]); 132 phas[size/2] = 0; 135 133 } 136 134 … … 156 154 157 155 aubio_mfft_t * new_aubio_mfft(uint_t winsize, uint_t channels){ 158 159 160 161 162 163 164 165 166 156 uint_t i; 157 aubio_mfft_t * fft = AUBIO_NEW(aubio_mfft_t); 158 fft->winsize = winsize; 159 fft->channels = channels; 160 fft->fft = new_aubio_fft(winsize); 161 fft->spec = AUBIO_ARRAY(fft_data_t*,channels); 162 for (i=0; i < channels; i++) 163 fft->spec[i] = AUBIO_ARRAY(fft_data_t,winsize); 164 return fft; 167 165 } 168 166 169 167 /* execute stft */ 170 168 void aubio_mfft_do (aubio_mfft_t * fft,fvec_t * in,cvec_t * fftgrain){ 171 172 173 174 175 176 177 178 169 uint_t i=0; 170 /* execute stft */ 171 for (i=0; i < fft->channels; i++) { 172 aubio_fft_do (fft->fft,in->data[i],fft->spec[i],fft->winsize); 173 /* put norm and phase into fftgrain */ 174 aubio_fft_getnorm(fftgrain->norm[i], fft->spec[i], fft->winsize); 175 aubio_fft_getphas(fftgrain->phas[i], fft->spec[i], fft->winsize); 176 } 179 177 } 180 178 181 179 /* execute inverse fourier transform */ 182 180 void aubio_mfft_rdo(aubio_mfft_t * fft,cvec_t * fftgrain, fvec_t * out){ 183 184 185 186 187 181 uint_t i=0; 182 for (i=0; i < fft->channels; i++) { 183 aubio_fft_getspectrum(fft->spec[i],fftgrain->norm[i],fftgrain->phas[i],fft->winsize); 184 aubio_fft_rdo(fft->fft,fft->spec[i],out->data[i],fft->winsize); 185 } 188 186 } 189 187 190 188 void del_aubio_mfft(aubio_mfft_t * fft) { 191 192 193 194 195 196 189 uint_t i; 190 for (i=0; i < fft->channels; i++) 191 AUBIO_FREE(fft->spec[i]); 192 AUBIO_FREE(fft->spec); 193 del_aubio_fft(fft->fft); 194 AUBIO_FREE(fft); 197 195 } -
src/fft.h
r1da7e08 rc721874 1 1 /* 2 2 Copyright (C) 2003 Paul Brossier 3 3 4 5 6 7 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 8 9 10 11 12 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 13 14 15 16 17 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 18 */ 19 19 -
src/phasevoc.c
r1da7e08 rc721874 26 26 /** phasevocoder internal object */ 27 27 struct _aubio_pvoc_t { 28 /** grain length */ 29 uint_t win_s; 30 /** overlap step */ 31 uint_t hop_s; 32 /** number of channels */ 33 uint_t channels; 34 /** spectral data */ 35 aubio_mfft_t * fft; 36 /**cur output grain [win_s] */ 37 fvec_t * synth; 38 /**last input frame [win_s-hop_s] */ 39 fvec_t * synthold; 40 /**current input grain [win_s] */ 41 fvec_t * data; 42 /**last input frame [win_s-hop_s] */ 43 fvec_t * dataold; 44 /** grain window [win_s] */ 45 float * w; 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] */ 36 smpl_t * w; /** grain window [win_s] */ 46 37 }; 47 38 48 39 49 40 /** returns data and dataold slided by hop_s */ 50 static void aubio_pvoc_swapbuffers( 51 smpl_t * data, 52 smpl_t * dataold, 53 const smpl_t * datanew, 54 uint_t win_s, uint_t hop_s); 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); 43 55 44 /** do additive synthesis from 'old' and 'cur' */ 56 static void aubio_pvoc_addsynth( 57 const smpl_t * synth, 58 smpl_t * synthold, 59 smpl_t * synthnew, 60 uint_t win_s, uint_t hop_s); 61 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); 62 47 63 48 void aubio_pvoc_do(aubio_pvoc_t *pv, fvec_t * datanew, cvec_t *fftgrain) { 64 65 66 67 68 69 70 71 72 73 74 75 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); 76 61 } 77 62 78 63 void aubio_pvoc_rdo(aubio_pvoc_t *pv,cvec_t * fftgrain, fvec_t * synthnew) { 79 uint_t i,j; 80 /* calculate rfft */ 81 aubio_mfft_rdo(pv->fft,fftgrain,pv->synth); 82 /* unshift */ 83 vec_shift(pv->synth); 84 for (i=0; i<pv->channels; i++) { 85 for (j=0; j<pv->win_s; j++) pv->synth->data[i][j] *= pv->w[j]; 86 aubio_pvoc_addsynth(pv->synth->data[i],pv->synthold->data[i], 87 synthnew->data[i],pv->win_s,pv->hop_s); 88 } 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 } 89 73 } 90 74 91 75 aubio_pvoc_t * new_aubio_pvoc (uint_t win_s, uint_t hop_s, uint_t channels) { 92 76 aubio_pvoc_t * pv = AUBIO_NEW(aubio_pvoc_t); 93 77 94 95 96 97 98 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 } 99 83 100 if (hop_s < 1) { 101 AUBIO_ERR("Hop size is smaller than 1!\n"); 102 AUBIO_ERR("Resetting hop size to half the window size.\n"); 103 hop_s = win_s / 2; 104 } 105 106 pv->fft = new_aubio_mfft(win_s,channels); 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 } 107 89 108 /* remember old */ 109 pv->data = new_fvec (win_s, channels); 110 pv->synth = new_fvec (win_s, channels); 90 pv->fft = new_aubio_mfft(win_s,channels); 111 91 112 /* new input output */ 113 pv->dataold = new_fvec (win_s-hop_s, channels); 114 pv->synthold = new_fvec (win_s-hop_s, channels); 115 pv->w = AUBIO_ARRAY(smpl_t,win_s); 116 aubio_window(pv->w,win_s,aubio_win_hanningz); 92 /* remember old */ 93 pv->data = new_fvec (win_s, channels); 94 pv->synth = new_fvec (win_s, channels); 117 95 118 pv->channels = channels; 119 pv->hop_s = hop_s; 120 pv->win_s = win_s; 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); 121 101 122 return pv; 102 pv->channels = channels; 103 pv->hop_s = hop_s; 104 pv->win_s = win_s; 105 106 return pv; 123 107 } 124 108 125 109 void del_aubio_pvoc(aubio_pvoc_t *pv) { 126 127 128 129 130 131 132 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); 133 117 } 134 118 135 119 static void aubio_pvoc_swapbuffers(smpl_t * data, smpl_t * dataold, 136 120 const smpl_t * datanew, uint_t win_s, uint_t hop_s) 137 121 { 138 139 140 141 142 143 144 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]; 145 129 } 146 130 … … 148 132 smpl_t * synthnew, uint_t win_s, uint_t hop_s) 149 133 { 150 151 152 153 for (i=0;i<hop_s;i++) 154 155 156 for (i=0;i<win_s-2*hop_s;i++) 157 158 159 for (i=win_s-hop_s;i<win_s;i++) 160 161 162 for (i=0;i<win_s-hop_s;i++) 163 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; 164 148 } 165 149 -
src/types.h
r1da7e08 rc721874 40 40 41 41 /** short sample format (32 or 64 bits) */ 42 typedef float smpl_t;43 //typedef double smpl_t;42 //typedef float smpl_t; 43 typedef double smpl_t; 44 44 /** long sample format (64 bits or more) */ 45 45 typedef double lsmp_t; -
tests/python/fvec.py
r1da7e08 rc721874 22 22 for index in range(buf_size): 23 23 for channel in range(channels): 24 self.assertEqual( fvec_read_sample(self.vector,channel,index),0.)24 self.assertEqual(0., fvec_read_sample(self.vector,channel,index)) 25 25 26 26 def test_fvec_write_sample(self): … … 31 31 for index in range(buf_size): 32 32 for channel in range(channels): 33 self.assertEqual( fvec_read_sample(self.vector,channel,index),1.)33 self.assertEqual(1., fvec_read_sample(self.vector,channel,index)) 34 34 35 35 if __name__ == '__main__': -
tests/src/Makefile.am
r1da7e08 rc721874 10 10 test-hist \ 11 11 test-scale \ 12 test-sample \ 12 test-cvec \ 13 test-fvec \ 13 14 test-window \ 14 15 test-filter \
Note: See TracChangeset
for help on using the changeset viewer.