src/fvec.h: clean up fvec api
[aubio.git] / examples / aubiopitch.c
1 /*
2   Copyright (C) 2003-2013 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 "utils.h"
22 #define PROG_HAS_PITCH 1
23 #define PROG_HAS_OUTPUT 1
24 #define PROG_HAS_JACK 1
25 #include "parse_args.h"
26
27 aubio_pitch_t *o;
28 aubio_wavetable_t *wavetable;
29 fvec_t *pitch;
30
31 void
32 process_block(fvec_t * ibuf, fvec_t * obuf) {
33   fvec_zeros(obuf);
34   aubio_pitch_do (o, ibuf, pitch);
35   smpl_t freq = fvec_get_sample(pitch, 0);
36   aubio_wavetable_set_amp ( wavetable, aubio_level_lin (ibuf) );
37   aubio_wavetable_set_freq ( wavetable, freq );
38
39   if (mix_input)
40     aubio_wavetable_do (wavetable, ibuf, obuf);
41   else
42     aubio_wavetable_do (wavetable, obuf, obuf);
43 }
44
45 void
46 process_print (void) {
47   smpl_t pitch_found = fvec_get_sample(pitch, 0);
48   outmsg("%f %f\n",(blocks)
49       *hop_size/(float)samplerate, pitch_found);
50 }
51
52 int main(int argc, char **argv) {
53
54   buffer_size = 2048;
55
56   examples_common_init(argc,argv);
57
58   verbmsg ("using source: %s at %dHz\n", source_uri, samplerate);
59   verbmsg ("pitch method: %s, ", pitch_method);
60   verbmsg ("pitch unit: %s, ", pitch_unit);
61   verbmsg ("buffer_size: %d, ", buffer_size);
62   verbmsg ("hop_size: %d, ", hop_size);
63   verbmsg ("tolerance: %f\n", pitch_tolerance);
64
65   o = new_aubio_pitch (pitch_method, buffer_size, hop_size, samplerate);
66   if (pitch_tolerance != 0.)
67     aubio_pitch_set_tolerance (o, pitch_tolerance);
68   if (silence_threshold != -90.)
69     aubio_pitch_set_silence (o, silence_threshold);
70   if (pitch_unit != NULL)
71     aubio_pitch_set_unit (o, pitch_unit);
72
73   pitch = new_fvec (1);
74
75   wavetable = new_aubio_wavetable (samplerate, hop_size);
76   aubio_wavetable_play ( wavetable );
77
78   examples_common_process((aubio_process_func_t)process_block,process_print);
79
80   del_aubio_pitch (o);
81   del_aubio_wavetable (wavetable);
82   del_fvec (pitch);
83
84   examples_common_del();
85   return 0;
86 }
87