[examples] add quiet mode for performance measurements
[aubio.git] / examples / utils.c
index dd6f91e..01c308d 100644 (file)
@@ -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.
 
@@ -18,7 +18,7 @@
 
 */
 
-/** 
+/**
 
   This file includes some tools common to all examples. Code specific to the
   algorithm performed by each program should go in the source file of that
 */
 
 #include "utils.h"
+#ifdef HAVE_JACK
+#include "jackio.h"
+#endif /* HAVE_JACK */
 
-#ifdef HAVE_LASH
-#include <lash/lash.h>
-#include <pthread.h>
-lash_client_t *aubio_lash_client;
-lash_args_t *lash_args;
-void *lash_thread_main (void *data);
-int lash_main (void);
-void save_data (void);
-void restore_data (lash_config_t * lash_config);
-pthread_t lash_thread;
-#endif /* HAVE_LASH */
-
-/* settings */
-const char *output_filename = NULL;
-const char *input_filename = NULL;
-const char *onset_filename =
-    AUBIO_PREFIX "/share/sounds/" PACKAGE "/woodblock.aiff";
-int frames = 0;
 int verbose = 0;
+int quiet = 0;
 int usejack = 0;
-int frames_delay = 0;
-
-
+// input / output
+char_t *sink_uri = NULL;
+char_t *source_uri = NULL;
+// general stuff
+uint_t samplerate = 0;
+uint_t buffer_size = 512;
+uint_t hop_size = 256;
+// onset stuff
+char_t * onset_method = "default";
+smpl_t onset_threshold = 0.0; // will be set if != 0.
+smpl_t onset_minioi = 0.0; // will be set if != 0.
+// pitch stuff
 char_t * pitch_unit = "default";
-char_t * pitch_mode = "default";
-
-/* energy,specdiff,hfc,complexdomain,phase */
-char_t * onset_mode = "default";
-smpl_t threshold = 0.0;         // leave unset, only set as asked 
-smpl_t silence = -90.;
-uint_t buffer_size = 512;       //1024;
-uint_t overlap_size = 256;      //512;
-uint_t samplerate = 44100;
-
+char_t * pitch_method = "default";
+smpl_t pitch_tolerance = 0.0; // will be set if != 0.
+// time stuff
+uint_t time_format = 0; // for "seconds", 1 for "ms", 2 for "samples"
+// tempo stuff
+char_t * tempo_method = "default";
+// more general stuff
+smpl_t silence_threshold = -90.;
+smpl_t release_drop = 10.;
+uint_t mix_input = 0;
+
+uint_t force_overwrite = 0;
+
+//
+// internal memory stuff
+aubio_source_t *this_source = NULL;
+aubio_sink_t *this_sink = NULL;
+fvec_t *input_buffer;
+fvec_t *output_buffer;
+
+smpl_t miditap_note = 69.;
+smpl_t miditap_velo = 65.;
 
-aubio_sndfile_t *file = NULL;
-aubio_sndfile_t *fileout = NULL;
-
-fvec_t *ibuf;
-fvec_t *obuf;
-fvec_t *woodblock;
-
-/* badly redeclare some things */
-smpl_t threshold;
-smpl_t averaging;
-const char *prog_name;
-
-void flush_process (aubio_process_func_t process_func,
-    aubio_print_func_t print);
-
-void
-usage (FILE * stream, int exit_code)
-{
-  fprintf (stream, "usage: %s [ options ] \n", prog_name);
-  fprintf (stream,
-      "       -h      --help          Display this message.\n"
-      "       -v      --verbose       Be verbose.\n"
-      "       -j      --jack          Use Jack.\n"
-      "       -o      --output        Output type.\n"
-      "       -i      --input         Input type.\n"
-      "       -O      --onset         Select onset detection algorithm.\n"
-      "       -t      --threshold     Set onset detection threshold.\n"
-      "       -s      --silence       Select silence threshold.\n"
-      "       -p      --pitch         Select pitch detection algorithm.\n"
-      "       -B      --bufsize       Set buffer size.\n"
-      "       -H      --hopsize       Set hopsize.\n"
-      "       -a      --averaging     Use averaging.\n");
-  exit (exit_code);
-}
+/* settings */
+int blocks = 0;
 
-int
-parse_args (int argc, char **argv)
-{
-  const char *options = "hvjo:i:O:t:s:p:B:H:a";
-  int next_option;
-  struct option long_options[] = {
-    {"help", 0, NULL, 'h'},
-    {"verbose", 0, NULL, 'v'},
-    {"jack", 0, NULL, 'j'},
-    {"output", 1, NULL, 'o'},
-    {"input", 1, NULL, 'i'},
-    {"onset", 1, NULL, 'O'},
-    {"threshold", 1, NULL, 't'},
-    {"silence", 1, NULL, 's'},
-    {"pitch", 1, NULL, 'p'},
-    {"averaging", 0, NULL, 'a'},
-    {"bufsize", 1, NULL, 'B'},
-    {"hopsize", 1, NULL, 'H'},
-    {NULL, 0, NULL, 0}
-  };
-#ifdef HAVE_LASH
-  lash_args = lash_extract_args (&argc, &argv);
-#endif /* HAVE_LASH */
-  prog_name = argv[0];
-  if (argc < 1) {
-    usage (stderr, 1);
-    return -1;
-  }
-  do {
-    next_option = getopt_long (argc, argv, options, long_options, NULL);
-    switch (next_option) {
-      case 'o':
-        output_filename = optarg;
-        break;
-      case 'i':
-        input_filename = optarg;
-        break;
-      case 'h':                /* help */
-        usage (stdout, 0);
-        return -1;
-      case 'v':                /* verbose */
-        verbose = 1;
-        break;
-      case 'j':
-        usejack = 1;
-        break;
-      case 'O':                /*onset type */
-        onset_mode = optarg;
-        break;
-      case 's':                /* silence value for onset */
-        silence = (smpl_t) atof (optarg);
-        break;
-      case 't':                /* threshold value for onset */
-        threshold = (smpl_t) atof (optarg);
-        break;
-      case 'p':
-        pitch_mode = optarg;
-        break;
-      case 'a':
-        averaging = 1;
-        break;
-      case 'B':
-        buffer_size = atoi (optarg);
-        break;
-      case 'H':
-        overlap_size = atoi (optarg);
-        break;
-      case '?':                /* unknown options */
-        usage (stderr, 1);
-        break;
-      case -1:                 /* done with options */
-        break;
-      default:                 /*something else unexpected */
-        fprintf (stderr, "Error parsing option '%c'\n", next_option);
-        abort ();
-    }
-  }
-  while (next_option != -1);
+extern void usage (FILE * stream, int exit_code);
+extern int parse_args (int argc, char **argv);
 
-  if (input_filename != NULL) {
-    debug ("Input file : %s\n", input_filename);
-  } else if (input_filename != NULL && output_filename != NULL) {
-    debug ("Input file : %s\n", input_filename);
-    debug ("Output file : %s\n", output_filename);
-  } else {
 #if HAVE_JACK
-    debug ("Jack input output\n");
-    usejack = 1;
-#else
-    debug
-        ("Error: Could not switch to jack mode\n   aubio was compiled without jack support\n");
-    exit (1);
-#endif
-  }
-
-  return 0;
-}
+#define MAX_MIDI_EVENTS 128
+#define MAX_MIDI_EVENT_SIZE 3
+aubio_jack_t *jack_setup;
+jack_midi_event_t ev;
+jack_midi_data_t midi_data[MAX_MIDI_EVENTS * MAX_MIDI_EVENT_SIZE];
+size_t midi_event_count = 0;
+#endif /* HAVE_JACK */
+
+void examples_common_init (int argc, char **argv);
+void examples_common_del (void);
+void examples_common_process (aubio_process_func_t process_func,
+    aubio_print_func_t print);
 
-void
-examples_common_init (int argc, char **argv)
+void examples_common_init (int argc, char **argv)
 {
 
-  uint_t found_wood = 0;
-
-  aubio_sndfile_t *onsetfile = NULL;
   /* parse command line arguments */
   parse_args (argc, argv);
 
   if (!usejack) {
     debug ("Opening files ...\n");
-    file = new_aubio_sndfile_ro (input_filename);
-    if (file == NULL) {
-      outmsg ("Could not open input file %s.\n", input_filename);
+    this_source = new_aubio_source ((char_t*)source_uri, samplerate, hop_size);
+    if (this_source == NULL) {
+      errmsg ("Error: could not open input file %s\n", source_uri);
       exit (1);
     }
-    if (verbose)
-      aubio_sndfile_info (file);
-    samplerate = aubio_sndfile_samplerate (file);
-    if (output_filename != NULL)
-      fileout = new_aubio_sndfile_wo (file, output_filename);
-  }
-#ifdef HAVE_LASH
-  else {
-    aubio_lash_client = lash_init (lash_args, argv[0],
-        LASH_Config_Data_Set | LASH_Terminal, LASH_PROTOCOL (2, 0));
-    if (!aubio_lash_client) {
-      fprintf (stderr, "%s: could not initialise lash\n", __FUNCTION__);
-    }
-    /* tell the lash server our client id */
-    if (lash_enabled (aubio_lash_client)) {
-      lash_event_t *event =
-          (lash_event_t *) lash_event_new_with_type (LASH_Client_Name);
-      lash_event_set_string (event, "aubio");
-      lash_send_event (aubio_lash_client, event);
-      pthread_create (&lash_thread, NULL, lash_thread_main, NULL);
+    if (samplerate == 0) {
+      samplerate = aubio_source_get_samplerate(this_source);
     }
-  }
-#endif /* HAVE_LASH */
-
-  woodblock = new_fvec (overlap_size);
-  if (output_filename || usejack) {
-    /* dummy assignement to keep egcs happy */
-    found_wood = (onsetfile = new_aubio_sndfile_ro (onset_filename)) ||
-        (onsetfile = new_aubio_sndfile_ro ("sounds/woodblock.aiff")) ||
-        (onsetfile = new_aubio_sndfile_ro ("../sounds/woodblock.aiff"));
-    if (onsetfile == NULL) {
-      outmsg ("Could not find woodblock.aiff\n");
-      exit (1);
+    if (sink_uri != NULL) {
+      uint_t sink_exists = (access(sink_uri, F_OK) == 0 );
+      if (!force_overwrite && sink_exists) {
+        errmsg ("Error: output file %s already exists, use -f to overwrite.\n",
+            sink_uri);
+        exit (1);
+      }
+      this_sink = new_aubio_sink ((char_t*)sink_uri, samplerate);
+      if (this_sink == NULL) {
+        errmsg ("Error: could not create output file %s\n", sink_uri);
+        exit (1);
+      }
     }
+#ifdef HAVE_JACK
+  } else {
+    debug ("Jack init ...\n");
+    jack_setup = new_aubio_jack (hop_size, 1, 1, 0, 1);
+    samplerate = aubio_jack_get_samplerate (jack_setup);
+    source_uri = "jack";
+#endif /* HAVE_JACK */
   }
-  if (onsetfile) {
-    /* read the output sound once */
-    aubio_sndfile_read_mono (onsetfile, overlap_size, woodblock);
-  }
-
-  ibuf = new_fvec (overlap_size);
-  obuf = new_fvec (overlap_size);
+  input_buffer = new_fvec (hop_size);
+  output_buffer = new_fvec (hop_size);
 
 }
 
-
-void
-examples_common_del (void)
+void examples_common_del (void)
 {
-  uint_t i;
-  del_fvec (ibuf);
-  del_fvec (obuf);
-  del_fvec (woodblock);
+  del_fvec (input_buffer);
+  del_fvec (output_buffer);
   aubio_cleanup ();
+  fflush(stderr);
+  fflush(stdout);
 }
 
-#if HAVE_JACK
-aubio_jack_t *jack_setup;
-#endif
-
-void
-examples_common_process (aubio_process_func_t process_func,
+void examples_common_process (aubio_process_func_t process_func,
     aubio_print_func_t print)
 {
+
+  uint_t read = 0;
   if (usejack) {
 
-#if HAVE_JACK
-    debug ("Jack init ...\n");
-    jack_setup = new_aubio_jack (1, 1,
-        0, 1, (aubio_process_func_t) process_func);
+#ifdef HAVE_JACK
+    ev.size = MAX_MIDI_EVENT_SIZE;
+    ev.time = 0; // send it now
     debug ("Jack activation ...\n");
-    aubio_jack_activate (jack_setup);
+    aubio_jack_activate (jack_setup, process_func);
     debug ("Processing (Ctrl+C to quit) ...\n");
     pause ();
     aubio_jack_close (jack_setup);
-#else
+#else /* HAVE_JACK */
     usage (stderr, 1);
     outmsg ("Compiled without jack output, exiting.\n");
-#endif
+#endif /* HAVE_JACK */
 
   } else {
-    /* phasevoc */
-    debug ("Processing ...\n");
 
-    frames = 0;
+    uint_t total_read = 0;
+    blocks = 0;
 
-    while ((signed) overlap_size ==
-        aubio_sndfile_read_mono (file, overlap_size, ibuf)) {
-      process_func (&ibuf->data, &obuf->data, overlap_size);
-      print ();
-      if (output_filename != NULL) {
-        aubio_sndfile_write (fileout, overlap_size, &obuf);
+    do {
+      aubio_source_do (this_source, input_buffer, &read);
+      process_func (input_buffer, output_buffer);
+      // print to console if verbose or no output given
+      if ((verbose || sink_uri == NULL) && !quiet) {
+        print();
       }
-      frames++;
-    }
-
-    debug ("Processed %d frames of %d samples.\n", frames, buffer_size);
-
-    flush_process (process_func, print);
-    del_aubio_sndfile (file);
+      if (this_sink) {
+        aubio_sink_do (this_sink, output_buffer, hop_size);
+      }
+      blocks++;
+      total_read += read;
+    } while (read == hop_size);
 
-    if (output_filename != NULL)
-      del_aubio_sndfile (fileout);
+    verbmsg ("read %.2fs (%d samples in %d blocks of %d) from %s at %dHz\n",
+        total_read * 1. / samplerate,
+        total_read, blocks, hop_size, source_uri, samplerate);
 
-  }
-}
+    del_aubio_source (this_source);
+    if (this_sink)
+      del_aubio_sink   (this_sink);
 
-void
-flush_process (aubio_process_func_t process_func, aubio_print_func_t print)
-{
-  uint_t i;
-  fvec_zeros(obuf);
-  for (i = 0; (signed) i < frames_delay; i++) {
-    process_func (&ibuf->data, &obuf->data, overlap_size);
-    print ();
   }
 }
 
 void
-send_noteon (int pitch, int velo)
+send_noteon (smpl_t pitch, smpl_t velo)
 {
-  smpl_t mpitch = floor (aubio_freqtomidi (pitch) + .5);
-#if HAVE_JACK
-  jack_midi_event_t ev;
-  ev.size = 3;
-  ev.buffer = malloc (3 * sizeof (jack_midi_data_t)); // FIXME
-  ev.time = 0;
+#ifdef HAVE_JACK
   if (usejack) {
+    ev.buffer = midi_data + midi_event_count++ * MAX_MIDI_EVENT_SIZE;
+    if (midi_event_count >= MAX_MIDI_EVENTS) {
+      midi_event_count = 0;
+    }
     ev.buffer[2] = velo;
-    ev.buffer[1] = mpitch;
+    ev.buffer[1] = pitch;
     if (velo == 0) {
       ev.buffer[0] = 0x80;      /* note off */
     } else {
@@ -351,90 +210,23 @@ send_noteon (int pitch, int velo)
     aubio_jack_midi_event_write (jack_setup, (jack_midi_event_t *) & ev);
   } else
 #endif
-  if (!verbose) {
-    if (velo == 0) {
-      outmsg ("%f\n", frames * overlap_size / (float) samplerate);
-    } else {
-      outmsg ("%f\t%f\t", mpitch, frames * overlap_size / (float) samplerate);
-    }
-  }
-}
-
-
-#if HAVE_LASH
-
-void *
-lash_thread_main (void *data __attribute__ ((unused)))
-{
-  printf ("LASH thread running\n");
-
-  while (!lash_main ())
-    usleep (1000);
-
-  printf ("LASH thread finished\n");
-  return NULL;
-}
-
-int
-lash_main (void)
-{
-  lash_event_t *lash_event;
-  lash_config_t *lash_config;
-
-  while ((lash_event = lash_get_event (aubio_lash_client))) {
-    switch (lash_event_get_type (lash_event)) {
-      case LASH_Quit:
-        lash_event_destroy (lash_event);
-        exit (1);
-        return 1;
-      case LASH_Restore_Data_Set:
-        lash_send_event (aubio_lash_client, lash_event);
-        break;
-      case LASH_Save_Data_Set:
-        save_data ();
-        lash_send_event (aubio_lash_client, lash_event);
-        break;
-      case LASH_Server_Lost:
-        return 1;
-      default:
-        printf ("%s: received unknown LASH event of type %d",
-            __FUNCTION__, lash_event_get_type (lash_event));
-        lash_event_destroy (lash_event);
-        break;
-    }
-  }
-
-  while ((lash_config = lash_get_config (aubio_lash_client))) {
-    restore_data (lash_config);
-    lash_config_destroy (lash_config);
+  if (velo == 0) {
+    print_time (blocks * hop_size);
+    outmsg ("\n");
+  } else {
+    outmsg ("%f\t", pitch);
+    print_time (blocks * hop_size);
+    outmsg ("\t");
   }
-
-  return 0;
-}
-
-void
-save_data ()
-{
-  lash_config_t *lash_config;
-
-  lash_config = lash_config_new_with_key ("threshold");
-  lash_config_set_value_double (lash_config, threshold);
-  lash_send_config (aubio_lash_client, lash_config);
-
 }
 
-void
-restore_data (lash_config_t * lash_config)
-{
-  const char *lash_key;
-
-  lash_key = lash_config_get_key (lash_config);
-
-  if (strcmp (lash_key, "threshold") == 0) {
-    threshold = lash_config_get_value_double (lash_config);
-    return;
+void print_time (uint_t time_in_samples) {
+  /* output times in selected format */
+  if (time_format == 2) {
+    outmsg ("%d", time_in_samples);
+  } else if (time_format == 1) {
+    outmsg ("%f", 1000. * time_in_samples / (float) samplerate);
+  } else {
+    outmsg ("%f", time_in_samples / (float) samplerate);
   }
-
 }
-
-#endif /* HAVE_LASH */