examples/aubiopitch.c: add pitch_unit
[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_read_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_read_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   examples_common_init(argc,argv);
54
55   verbmsg ("using source: %s at %dHz\n", source_uri, samplerate);
56   verbmsg ("pitch method: %s, ", pitch_method);
57   verbmsg ("pitch unit: %s, ", pitch_unit);
58   verbmsg ("buffer_size: %d, ", buffer_size);
59   verbmsg ("hop_size: %d, ", hop_size);
60   verbmsg ("tolerance: %f\n", pitch_tolerance);
61
62   o = new_aubio_pitch (pitch_method, buffer_size, hop_size, samplerate);
63   if (pitch_tolerance != 0.) aubio_pitch_set_tolerance (o, pitch_tolerance);
64   if (pitch_unit != NULL) aubio_pitch_set_unit (o, pitch_unit);
65   pitch = new_fvec (1);
66
67   wavetable = new_aubio_wavetable (samplerate, hop_size);
68   aubio_wavetable_play ( wavetable );
69
70   examples_common_process((aubio_process_func_t)process_block,process_print);
71
72   del_aubio_pitch (o);
73   del_aubio_wavetable (wavetable);
74   del_fvec (pitch);
75
76   examples_common_del();
77   return 0;
78 }
79