[88199ce] | 1 | /* |
---|
| 2 | Copyright (C) 2007 Amaury Hazan |
---|
| 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 | #include "utils.h" |
---|
| 20 | |
---|
| 21 | unsigned int pos = 0; /*frames%dspblocksize*/ |
---|
| 22 | uint_t usepitch = 0; |
---|
| 23 | |
---|
| 24 | int aubio_process(float **input, float **output, int nframes); |
---|
| 25 | int aubio_process(float **input, float **output, int nframes) { |
---|
| 26 | unsigned int i; /*channels*/ |
---|
| 27 | unsigned int j; /*frames*/ |
---|
[dcc649c] | 28 | |
---|
| 29 | // declare the mel filter bank |
---|
| 30 | // TODO: should be done only once |
---|
| 31 | aubio_mfcc_init(int N, NYQUIST, int style, float freq_min, float freq_max, int freq_bands, float **fft_tables); |
---|
| 32 | |
---|
[88199ce] | 33 | for (j=0;j<(unsigned)nframes;j++) { |
---|
| 34 | if(usejack) { |
---|
| 35 | for (i=0;i<channels;i++) { |
---|
| 36 | /* write input to datanew */ |
---|
| 37 | fvec_write_sample(ibuf, input[i][j], i, pos); |
---|
| 38 | /* put synthnew in output */ |
---|
| 39 | output[i][j] = fvec_read_sample(obuf, i, pos); |
---|
| 40 | } |
---|
| 41 | } |
---|
| 42 | /*time for fft*/ |
---|
| 43 | if (pos == overlap_size-1) { |
---|
| 44 | /* block loop */ |
---|
| 45 | |
---|
| 46 | //compute mag spectrum |
---|
| 47 | aubio_pvoc_do (pv,ibuf, fftgrain); |
---|
[71d3bf0] | 48 | |
---|
| 49 | uint_t coef_cnt; |
---|
| 50 | uint_t n_filters=20; |
---|
| 51 | smpl_t outbuf[20]; |
---|
[88199ce] | 52 | |
---|
[71d3bf0] | 53 | for (coef_cnt=0; coef_cnt<n_filters ; coef_cnt++) |
---|
| 54 | outbuf[coef_cnt]=0.f; |
---|
[fe28ff3] | 55 | |
---|
[88199ce] | 56 | //compute mfccs |
---|
[97886fa] | 57 | aubio_mffc_do(fftgrain->norm, nframes, mf, outbuf); |
---|
[88199ce] | 58 | |
---|
[71d3bf0] | 59 | for (coef_cnt=0; coef_cnt<n_filters ; coef_cnt++) |
---|
| 60 | outmsg("%f ",outbuf[coef_cnt]); |
---|
| 61 | outmsg("\n"); |
---|
[88199ce] | 62 | |
---|
| 63 | |
---|
| 64 | |
---|
| 65 | /* end of block loop */ |
---|
| 66 | pos = -1; /* so it will be zero next j loop */ |
---|
| 67 | } |
---|
| 68 | pos++; |
---|
| 69 | } |
---|
| 70 | return 1; |
---|
| 71 | } |
---|
| 72 | |
---|
| 73 | void process_print (void); |
---|
| 74 | void process_print (void) { |
---|
| 75 | /* output times in seconds |
---|
| 76 | write extracted mfccs |
---|
| 77 | */ |
---|
| 78 | |
---|
| 79 | if (output_filename == NULL) { |
---|
| 80 | if(frames >= 4) { |
---|
| 81 | outmsg("%f\n",(frames-4)*overlap_size/(float)samplerate); |
---|
| 82 | } else if (frames < 4) { |
---|
| 83 | outmsg("%f\n",0.); |
---|
| 84 | } |
---|
| 85 | } |
---|
| 86 | } |
---|
| 87 | |
---|
| 88 | int main(int argc, char **argv) { |
---|
| 89 | examples_common_init(argc,argv); |
---|
[fe28ff3] | 90 | |
---|
| 91 | //allocate and initialize mel filter bank |
---|
| 92 | uint_t n_filters=20; |
---|
| 93 | uint_t nyquist= samplerate / 2.; |
---|
[71d3bf0] | 94 | smpl_t lowfreq=80.f; |
---|
| 95 | smpl_t highfreq=18000.f; |
---|
[fe28ff3] | 96 | |
---|
| 97 | uint_t banksize = (uint) ( sizeof(aubio_mel_filter)); |
---|
| 98 | aubio_mel_filter * mf = (aubio_mel_filter *)getbytes(banksize); |
---|
| 99 | |
---|
[97886fa] | 100 | mf->n_filters = 20; |
---|
| 101 | mf->filters = (smpl_t **)getbytes(mf->n_filters * sizeof(smpl_t *)); |
---|
[fe28ff3] | 102 | for(n = 0; n < mf->n_filters; n++) |
---|
| 103 | mf->filters[n] = (smpl_t *)getbytes((buffer_size/2+1) * sizeof(smpl_t)); |
---|
| 104 | |
---|
| 105 | //populating the filter |
---|
[71d3bf0] | 106 | aubio_mfcc_init(buffer_size, nyquist, XTRACT_EQUAL_GAIN, lowfreq, highfreq, mf->n_filters, mf->filters); |
---|
[fe28ff3] | 107 | |
---|
| 108 | //process |
---|
[88199ce] | 109 | examples_common_process(aubio_process,process_print); |
---|
| 110 | examples_common_del(); |
---|
| 111 | debug("End of program.\n"); |
---|
| 112 | fflush(stderr); |
---|
[fe28ff3] | 113 | |
---|
| 114 | //destroying filterbank |
---|
| 115 | free(mf); |
---|
| 116 | |
---|
[88199ce] | 117 | return 0; |
---|
| 118 | } |
---|
| 119 | |
---|