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