pull from master
[aubio.git] / examples / aubiotrack.c
1 /*
2   Copyright (C) 2003-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 <aubio.h>
22 #include "utils.h"
23
24 uint_t pos = 0;    /* frames%dspblocksize */
25 fvec_t * tempo_out = NULL;
26 aubio_tempo_t * bt = NULL;
27 smpl_t istactus = 0;
28 smpl_t isonset = 0;
29
30 static int aubio_process(smpl_t **input, smpl_t **output, int nframes) {
31   unsigned int j;       /*frames*/
32   for (j=0;j<(unsigned)nframes;j++) {
33     if(usejack) {
34       /* write input to datanew */
35       fvec_write_sample(ibuf, input[0][j], pos);
36       /* put synthnew in output */
37       output[0][j] = fvec_read_sample(obuf, pos);
38     }
39     /*time for fft*/
40     if (pos == overlap_size-1) {         
41       /* block loop */
42       aubio_tempo_do (bt,ibuf,tempo_out);
43       istactus = fvec_read_sample (tempo_out, 0);
44       isonset = fvec_read_sample (tempo_out, 1);
45       if (istactus > 0.) {
46         fvec_copy (woodblock, obuf);
47       } else {
48         fvec_zeros (obuf);
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 static void process_print (void) {
59         if (output_filename == NULL) {
60                 if (istactus) {
61                         outmsg("%f\n",((smpl_t)(frames*overlap_size)+(istactus-1.)*overlap_size)/(smpl_t)samplerate); 
62                 }
63                 if (isonset && verbose)
64                         outmsg(" \t \t%f\n",(frames)*overlap_size/(float)samplerate);
65         }
66 }
67
68 int main(int argc, char **argv) {
69   
70   buffer_size = 1024;
71   overlap_size = 512;
72   /* override default settings */
73   examples_common_init(argc,argv);
74
75   tempo_out = new_fvec(2);
76   bt = new_aubio_tempo(onset_mode,buffer_size,overlap_size, samplerate);
77   if (threshold != 0.) aubio_tempo_set_threshold (bt, threshold);
78
79   examples_common_process(aubio_process,process_print);
80
81   del_aubio_tempo(bt);
82   del_fvec(tempo_out);
83
84   examples_common_del();
85
86   debug("End of program.\n");
87
88   fflush(stderr);
89
90   return 0;
91 }
92