[tests] add run_on_default_source_and_sink helper

[aubio.git] / examples / aubiopitch.c

diff --git a/examples/aubiopitch.c b/examples/aubiopitch.c
index fcffce6..2d7a012 100644 (file)
--- a/examples/aubiopitch.c
+++ b/examples/aubiopitch.c
@@ -1,5 +1,5 @@
 /*
-  Copyright (C) 2003-2009 Paul Brossier <piem@aubio.org>
+  Copyright (C) 2003-2013 Paul Brossier <piem@aubio.org>
 
   This file is part of aubio.
 
@@ -19,67 +19,73 @@
 */
 
 #include "utils.h"
-
-unsigned int pos = 0; /*frames%dspblocksize*/
+#define PROG_HAS_PITCH 1
+#define PROG_HAS_OUTPUT 1
+#define PROG_HAS_SILENCE 1
+#define PROG_HAS_JACK 1
+#include "parse_args.h"
 
 aubio_pitch_t *o;
 aubio_wavetable_t *wavetable;
 fvec_t *pitch;
 
-static int aubio_process(smpl_t **input, smpl_t **output, int nframes) {
-  unsigned int j;       /*frames*/
-  for (j=0;j<(unsigned)nframes;j++) {
-    if(usejack) {
-      /* write input to datanew */
-      fvec_write_sample(ibuf, input[0][j], pos);
-      /* put synthnew in output */
-      output[0][j] = fvec_read_sample(obuf, pos);
-    }
-    /*time for fft*/
-    if (pos == overlap_size-1) {         
-      /* block loop */
-      aubio_pitch_do (o, ibuf, pitch);
-      smpl_t freq = fvec_read_sample(pitch, 0);
-      aubio_wavetable_set_amp ( wavetable, aubio_level_lin (ibuf) );
-      if (freq != 0.0) {
-        aubio_wavetable_set_freq ( wavetable, freq );
-      } else {
-        aubio_wavetable_set_freq ( wavetable, 0.0 );
-      }
-      aubio_wavetable_do (wavetable, obuf, obuf);
-      /* end of block loop */
-      pos = -1; /* so it will be zero next j loop */
-    }
-    pos++;
-  }
-  return 1;
+void process_block(fvec_t * ibuf, fvec_t * obuf)
+{
+  smpl_t freq;
+  aubio_pitch_do (o, ibuf, pitch);
+  if ( !usejack && ! sink_uri ) return;
+  fvec_zeros(obuf);
+  freq = fvec_get_sample(pitch, 0);
+  aubio_wavetable_set_amp ( wavetable, aubio_level_lin (ibuf) );
+  aubio_wavetable_set_freq ( wavetable, freq );
+  if (mix_input)
+    aubio_wavetable_do (wavetable, ibuf, obuf);
+  else
+    aubio_wavetable_do (wavetable, obuf, obuf);
 }
 
-static void process_print (void) {
-      if (!verbose && usejack) return;
-      smpl_t pitch_found = fvec_read_sample(pitch, 0);
-      outmsg("%f %f\n",(frames) 
-              *overlap_size/(float)samplerate, pitch_found);
+void process_print (void)
+{
+  smpl_t pitch_found = fvec_get_sample(pitch, 0);
+  print_time(blocks * hop_size);
+  outmsg(" %f\n", pitch_found);
 }
 
 int main(int argc, char **argv) {
+  int ret = 0;
+
+  buffer_size = 2048;
+
   examples_common_init(argc,argv);
 
-  o = new_aubio_pitch (pitch_mode, buffer_size, overlap_size, samplerate);
+  verbmsg ("using source: %s at %dHz\n", source_uri, samplerate);
+  verbmsg ("pitch method: %s, ", pitch_method);
+  verbmsg ("pitch unit: %s, ", pitch_unit);
+  verbmsg ("buffer_size: %d, ", buffer_size);
+  verbmsg ("hop_size: %d, ", hop_size);
+  verbmsg ("tolerance: %f\n", pitch_tolerance);
+
+  o = new_aubio_pitch (pitch_method, buffer_size, hop_size, samplerate);
+  if (o == NULL) { ret = 1; goto beach; }
+  if (pitch_tolerance != 0.)
+    aubio_pitch_set_tolerance (o, pitch_tolerance);
+  if (silence_threshold != -90.)
+    aubio_pitch_set_silence (o, silence_threshold);
+  if (pitch_unit != NULL)
+    aubio_pitch_set_unit (o, pitch_unit);
+
   pitch = new_fvec (1);
 
-  wavetable = new_aubio_wavetable (samplerate, overlap_size);
+  wavetable = new_aubio_wavetable (samplerate, hop_size);
   aubio_wavetable_play ( wavetable );
 
-  examples_common_process(aubio_process,process_print);
+  examples_common_process((aubio_process_func_t)process_block,process_print);
 
   del_aubio_pitch (o);
   del_aubio_wavetable (wavetable);
   del_fvec (pitch);
 
+beach:
   examples_common_del();
-  debug("End of program.\n");
-  fflush(stderr);
-  return 0;
+  return ret;
 }
-