merge from develop
[aubio.git] / examples / aubiomfcc.c
1 /*
2   Copyright (C) 2007-2009 Paul Brossier <piem@aubio.org>
3
4   This file is part of aubio.
5
6   aubio is free software: you can redistribute it and/or modify
7   it under the terms of the GNU General Public License as published by
8   the Free Software Foundation, either version 3 of the License, or
9   (at your option) any later version.
10
11   aubio is distributed in the hope that it will be useful,
12   but WITHOUT ANY WARRANTY; without even the implied warranty of
13   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14   GNU General Public License for more details.
15
16   You should have received a copy of the GNU General Public License
17   along with aubio.  If not, see <http://www.gnu.org/licenses/>.
18
19 */
20
21 #include "utils.h"
22
23 /* mfcc objects */
24 fvec_t * mfcc_out;
25 aubio_mfcc_t * mfcc;
26 aubio_pvoc_t *pv;
27 cvec_t *fftgrain;
28
29 uint_t n_filters = 40;
30 uint_t n_coefs = 13;
31
32 unsigned int pos = 0; /*frames%dspblocksize*/
33
34 static int aubio_process(smpl_t **input, smpl_t **output, int nframes) {
35   unsigned int j;       /*frames*/
36   
37   for (j=0;j<(unsigned)nframes;j++) {
38     if(usejack) {
39       /* write input to datanew */
40       fvec_write_sample(ibuf, input[0][j], pos);
41       /* put synthnew in output */
42       output[0][j] = fvec_read_sample(obuf, pos);
43     }
44     /*time for fft*/
45     if (pos == overlap_size-1) {         
46       /* block loop */
47       
48       //compute mag spectrum
49       aubio_pvoc_do (pv, ibuf, fftgrain);
50      
51       //compute mfccs
52       aubio_mfcc_do(mfcc, fftgrain, mfcc_out);
53       
54       /* end of block loop */
55       pos = -1; /* so it will be zero next j loop */
56     }
57     pos++;
58   }
59   return 1;
60 }
61
62 static void process_print (void) {
63       /* output times in seconds
64          write extracted mfccs
65       */
66       
67       uint_t coef_cnt;
68       if (sink_uri == NULL) {
69         outmsg("%f\t",frames*overlap_size/(float)samplerate);
70         for (coef_cnt = 0; coef_cnt < n_coefs; coef_cnt++) {
71             outmsg("%f ", fvec_read_sample (mfcc_out, coef_cnt) );
72         }
73         outmsg("\n");
74       }
75 }
76
77 int main(int argc, char **argv) {
78   // params
79   buffer_size  = 512;
80   overlap_size = 256;
81   
82   examples_common_init(argc,argv);
83
84   /* phase vocoder */
85   pv = new_aubio_pvoc (buffer_size, overlap_size);
86
87   fftgrain = new_cvec (buffer_size);
88
89   //populating the filter
90   mfcc = new_aubio_mfcc(buffer_size, n_filters, n_coefs, samplerate);
91   
92   mfcc_out = new_fvec(n_coefs);
93   
94   //process
95   examples_common_process(aubio_process,process_print);
96   
97   //destroying mfcc 
98   del_aubio_pvoc (pv);
99   del_cvec (fftgrain);
100   del_aubio_mfcc(mfcc);
101   del_fvec(mfcc_out);
102
103   examples_common_del();
104   debug("End of program.\n");
105   fflush(stderr);
106   
107   return 0;
108 }
109