examples/aubioonset.c

   1 /*
   2    Copyright (C) 2003 Paul Brossier
   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
  23 aubio_onset_t *o;
  24 fvec_t *onset;
  25
  26 static int aubio_process(smpl_t **input, smpl_t **output, int nframes) {
  27   unsigned int i;       /*channels*/
  28   unsigned int j;       /*frames*/
  29   for (j=0;j<(unsigned)nframes;j++) {
  30     if(usejack) {
  31       for (i=0;i<channels;i++) {
  32         /* write input to datanew */
  33         fvec_write_sample(ibuf, input[i][j], i, pos);
  34         /* put synthnew in output */
  35         output[i][j] = fvec_read_sample(obuf, i, pos);
  36       }
  37     }
  38     /*time for fft*/
  39     if (pos == overlap_size-1) {
  40       /* block loop */
  41       aubio_onset_do (o, ibuf, onset);
  42       if (fvec_read_sample(onset, 0, 0)) {
  43         for (pos = 0; pos < overlap_size; pos++){
  44           obuf->data[0][pos] = woodblock->data[0][pos];
  45         }
  46       } else {
  47         for (pos = 0; pos < overlap_size; pos++) {
  48           obuf->data[0][pos] = 0.;
  49         }
  50       }
  51       /* end of block loop */
  52       pos = -1; /* so it will be zero next j loop */
  53     }
  54     pos++;
  55   }
  56   return 1;
  57 }
  58
  59 static void process_print (void) {
  60       /* output times in seconds, taking back some
  61        * delay to ensure the label is _before_ the
  62        * actual onset */
  63       if (!verbose && usejack) return;
  64       smpl_t onset_found = fvec_read_sample(onset, 0, 0);
  65       if (onset_found) {
  66         if(frames >= 4) {
  67           outmsg("%f\n",(frames - frames_delay + onset_found)
  68                   *overlap_size/(float)samplerate);
  69         } else if (frames < frames_delay) {
  70           outmsg("%f\n",0.);
  71         }
  72       }
  73 }
  74
  75 int main(int argc, char **argv) {
  76   frames_delay = 3;
  77   examples_common_init(argc,argv);
  78
  79   o = new_aubio_onset (onset_mode, buffer_size, overlap_size, channels,
  80           samplerate);
  81   if (threshold != 0.) aubio_onset_set_threshold (o, threshold);
  82   onset = new_fvec (1, channels);
  83
  84   examples_common_process(aubio_process,process_print);
  85
  86   del_aubio_onset (o);
  87   del_fvec (onset);
  88
  89   examples_common_del();
  90   debug("End of program.\n");
  91   fflush(stderr);
  92   return 0;
  93 }
  94