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
  45       //compute mfcics
  46       aubio_mffc_do(fftgrain->norm, nframes, filterbank, outbuf);
  47
  48
  49
  50       /* end of block loop */
  51       pos = -1; /* so it will be zero next j loop */
  52     }
  53     pos++;
  54   }
  55   return 1;
  56 }
  57
  58 void process_print (void);
  59 void process_print (void) {
  60       /* output times in seconds
  61          write extracted mfccs
  62       */
  63
  64       if (output_filename == NULL) {
  65         if(frames >= 4) {
  66           outmsg("%f\n",(frames-4)*overlap_size/(float)samplerate);
  67         } else if (frames < 4) {
  68           outmsg("%f\n",0.);
  69         }
  70       }
  71 }
  72
  73 int main(int argc, char **argv) {
  74   examples_common_init(argc,argv);
  75
  76   //allocate and initialize mel filter bank
  77   uint_t n_filters=20;
  78   uint_t nyquist= samplerate / 2.;
  79
  80   uint_t banksize = (uint) ( sizeof(aubio_mel_filter));
  81   aubio_mel_filter * mf = (aubio_mel_filter *)getbytes(banksize);
  82
  83   mfilterbank->n_filters = 20;
  84   mfilterbank->filters = (smpl_t **)getbytes(mf->n_filters * sizeof(smpl_t *));
  85   for(n = 0; n < mf->n_filters; n++)
  86     mf->filters[n] = (smpl_t *)getbytes((buffer_size/2+1) * sizeof(smpl_t));
  87
  88   //populating the filter
  89   new_aubio_mfcc(buffer_size, nyquist, XTRACT_EQUAL_GAIN, 80.0f, 18000.0f, mf->n_filters, mf->filters);
  90
  91   //process
  92   examples_common_process(aubio_process,process_print);
  93   examples_common_del();
  94   debug("End of program.\n");
  95   fflush(stderr);
  96
  97   //destroying filterbank
  98   free(mf);
  99
 100   return 0;
 101 }
 102