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