examples/utils.c: allocate note event once
[aubio.git] / examples / utils.c
index 51a6da6..39adef9 100644 (file)
+/*
+  Copyright (C) 2003-2013 Paul Brossier <piem@aubio.org>
 
-#include "aubio.h"
+  This file is part of aubio.
 
-#ifndef JACK_SUPPORT
-#define JACK_SUPPORT 0
-#endif
+  aubio is free software: you can redistribute it and/or modify
+  it under the terms of the GNU General Public License as published by
+  the Free Software Foundation, either version 3 of the License, or
+  (at your option) any later version.
 
-#include <getopt.h>
-#include <stdlib.h>
-#include <stdio.h>
-#include <string.h>
-#include <math.h> /* for isfinite */
-#include "utils.h"
+  aubio is distributed in the hope that it will be useful,
+  but WITHOUT ANY WARRANTY; without even the implied warranty of
+  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+  GNU General Public License for more details.
 
-#ifdef LASH_SUPPORT
-#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 /* LASH_SUPPORT */
+  You should have received a copy of the GNU General Public License
+  along with aubio.  If not, see <http://www.gnu.org/licenses/>.
 
-/* 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 usejack = 0;
-int usedoubled = 1;
-
-
-/* energy,specdiff,hfc,complexdomain,phase */
-aubio_onsetdetection_type type_onset  = aubio_onset_kl;
-aubio_onsetdetection_type type_onset2 = aubio_onset_complex;
-smpl_t threshold                      = 0.3;
-smpl_t silence                        = -90.;
-uint_t buffer_size                    = 512; //1024;
-uint_t overlap_size                   = 256; //512;
-uint_t channels                       = 1;
-uint_t samplerate                     = 44100;
-
-
-aubio_sndfile_t * file = NULL;
-aubio_sndfile_t * fileout = NULL;
-
-aubio_pvoc_t * pv;
-fvec_t * ibuf;
-fvec_t * obuf;
-cvec_t * fftgrain;
-fvec_t * woodblock;
-aubio_onsetdetection_t *o;
-aubio_onsetdetection_t *o2;
-fvec_t *onset;
-fvec_t *onset2;
-int isonset = 0;
-aubio_pickpeak_t * parms;
-
-/* pitch objects */
-smpl_t pitch               = 0.;
-aubio_pitchdetection_t * pitchdet;
-aubio_pitchdetection_type type_pitch = aubio_pitch_yinfft; // aubio_pitch_mcomb
-aubio_pitchdetection_mode mode_pitch = aubio_pitchm_freq;
-uint_t median         = 6;
-
-fvec_t * note_buffer  = NULL;
-fvec_t * note_buffer2 = NULL;
-smpl_t curlevel       = 0.;
-smpl_t maxonset       = 0.;
-
-/* midi objects */
-aubio_midi_player_t * mplay; 
-aubio_midi_driver_t * mdriver; 
-aubio_midi_event_t  * event;
-
-smpl_t curnote = 0.;
-smpl_t newnote = 0.;
-uint_t isready = 0;
-
-
-
-/* badly redeclare some things */
-aubio_onsetdetection_type type_onset;
-smpl_t threshold;
-smpl_t averaging;
-const char * prog_name;
-
-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"
-                        "       -H      --hopsize       Set hopsize.\n"
-                        "       -a      --averaging     Use averaging.\n"
-               );
-        exit(exit_code);
-}
+*/
 
-int parse_args (int argc, char **argv) {
-        const char *options = "hvjo:i:O:t:s:p: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'},
-                {"hopsize",   1, NULL, 'H'},
-                {NULL       , 0, NULL, 0}
-        };
-#ifdef LASH_SUPPORT
-        lash_args = lash_extract_args(&argc, &argv);
-#endif /* LASH_SUPPORT */
-        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*/
-                                if (strcmp(optarg,"energy") == 0) 
-                                        type_onset = aubio_onset_energy;
-                                else if (strcmp(optarg,"specdiff") == 0) 
-                                        type_onset = aubio_onset_specdiff;
-                                else if (strcmp(optarg,"hfc") == 0) 
-                                        type_onset = aubio_onset_hfc;
-                                else if (strcmp(optarg,"complexdomain") == 0) 
-                                        type_onset = aubio_onset_complex;
-                                else if (strcmp(optarg,"complex") == 0) 
-                                        type_onset = aubio_onset_complex;
-                                else if (strcmp(optarg,"phase") == 0) 
-                                        type_onset = aubio_onset_phase;
-                                else if (strcmp(optarg,"mkl") == 0) 
-                                        type_onset = aubio_onset_mkl;
-                                else if (strcmp(optarg,"kl") == 0) 
-                                        type_onset = aubio_onset_kl;
-                                else {
-                                        errmsg("unknown onset type.\n");
-                                        abort();
-                                }
-                                usedoubled = 0;
-                                break;
-                        case 's':   /* threshold value for onset */
-                                silence = (smpl_t)atof(optarg);
-                                break;
-                        case 't':   /* threshold value for onset */
-                                threshold = (smpl_t)atof(optarg);
-                                /*
-                                   if (!isfinite(threshold)) {
-                                   debug("could not get threshold.\n");
-                                   abort();
-                                   }
-                                   */
-                                break;
-                        case 'p':
-                                if (strcmp(optarg,"mcomb") == 0) 
-                                        type_pitch = aubio_pitch_mcomb;
-                                else if (strcmp(optarg,"yinfft") == 0) 
-                                        type_pitch = aubio_pitch_yin;
-                                else if (strcmp(optarg,"yin") == 0) 
-                                        type_pitch = aubio_pitch_yin;
-                                else if (strcmp(optarg,"schmitt") == 0) 
-                                        type_pitch = aubio_pitch_schmitt;
-                                else if (strcmp(optarg,"fcomb") == 0) 
-                                        type_pitch = aubio_pitch_fcomb;
-                                else {
-                                        errmsg("unknown pitch type.\n");
-                                        abort();
-                                }
-                                break;
-                        case 'a':
-                                averaging = 1;
-                                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 */
-                                abort ();
-                }
-        }
-        while (next_option != -1);
-
-        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 (JACK_SUPPORT)
-                {
-                        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);
-                }
-        }
-
-        return 0;
-}
+/**
 
-void examples_common_init(int argc,char ** argv) {
+  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
+  program instead.
 
+*/
 
-  aubio_sndfile_t * onsetfile = NULL;
-  /* parse command line arguments */
-  parse_args(argc, argv);
-
-  woodblock = new_fvec(buffer_size,1);
-  if (output_filename || usejack) {
-          /* dummy assignement to keep egcs happy */
-          isonset = (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 (onsetfile) {
-          /* read the output sound once */
-          aubio_sndfile_read(onsetfile, overlap_size, woodblock);
-  }
-
-  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);
-      exit(1);
-    }
-    if (verbose) aubio_sndfile_info(file);
-    channels = aubio_sndfile_channels(file);
-    samplerate = aubio_sndfile_samplerate(file);
-    if (output_filename != NULL)
-      fileout = new_aubio_sndfile_wo(file, output_filename);
-  }
-#ifdef LASH_SUPPORT
-  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);
-    }
-  }
-#endif /* LASH_SUPPORT */
+#include "utils.h"
+#ifdef HAVE_JACK
+#include "jackio.h"
+#endif /* HAVE_JACK */
 
-  ibuf      = new_fvec(overlap_size, channels);
-  obuf      = new_fvec(overlap_size, channels);
-  fftgrain  = new_cvec(buffer_size, channels);
+int verbose = 0;
+int usejack = 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_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.;
+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 *ibuf;
+fvec_t *obuf;
+
+smpl_t miditap_note = 69.;
+smpl_t miditap_velo = 65.;
 
-  
-  if (usepitch) {
-    pitchdet = new_aubio_pitchdetection(buffer_size*4, 
-        overlap_size, channels, samplerate, type_pitch, mode_pitch);
-    aubio_pitchdetection_set_yinthresh(pitchdet, 0.7);
+/* settings */
+int blocks = 0;
 
-    if (median) {
-      note_buffer = new_fvec(median, 1);
-      note_buffer2= new_fvec(median, 1);
-    }
-  }
-  /* phase vocoder */
-  pv = new_aubio_pvoc(buffer_size, overlap_size, channels);
-  
-  /* onsets */
-  parms = new_aubio_peakpicker(threshold);
-  o = new_aubio_onsetdetection(type_onset,buffer_size,channels);
-  onset = new_fvec(1, channels);
-  if (usedoubled)    {
-    o2 = new_aubio_onsetdetection(type_onset2,buffer_size,channels);
-    onset2 = new_fvec(1 , channels);
-  }
+extern void usage (FILE * stream, int exit_code);
+extern int parse_args (int argc, char **argv);
 
-}
+#if HAVE_JACK
+aubio_jack_t *jack_setup;
+jack_midi_event_t ev;
+#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_del(void){
-  if (usepitch) {
-          send_noteon(curnote,0);
-          del_aubio_pitchdetection(pitchdet);
-          if (median) {
-                  del_fvec(note_buffer);
-                  del_fvec(note_buffer2);
-          }
-  }
-  if (usedoubled)    {
-    del_aubio_onsetdetection(o2);
-    del_fvec(onset2);
-  }
-  del_aubio_onsetdetection(o);
-  del_aubio_peakpicker(parms);
-  del_aubio_pvoc(pv);
-  del_fvec(obuf);
-  del_fvec(ibuf);
-  del_cvec(fftgrain);
-  del_fvec(onset);
-  del_fvec(woodblock);
-  
-  aubio_cleanup();
-}
+void examples_common_init (int argc, char **argv)
+{
 
-void examples_common_process(aubio_process_func_t process_func, aubio_print_func_t print ){
-  if(usejack) {
-#if JACK_SUPPORT
-    aubio_jack_t * jack_setup;
-    debug("Jack init ...\n");
-    jack_setup = new_aubio_jack(channels, channels,
-          (aubio_process_func_t)process_func);
-    if (usepitch) {
-            debug("Midi init ...\n");
-            mplay = new_aubio_midi_player();
-            mdriver = new_aubio_midi_driver("alsa_seq",
-                            (handle_midi_event_func_t)aubio_midi_send_event, mplay);
-            event = new_aubio_midi_event();
+  /* parse command line arguments */
+  parse_args (argc, argv);
+
+  if (!usejack) {
+    debug ("Opening files ...\n");
+    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);
     }
-    debug("Jack activation ...\n");
-    aubio_jack_activate(jack_setup);
-    debug("Processing (Ctrl+C to quit) ...\n");
-    pause();
-    aubio_jack_close(jack_setup);
-    if (usepitch) {
-            send_noteon(curnote,0);
-            del_aubio_midi_driver(mdriver);
+    if (samplerate == 0) {
+      samplerate = aubio_source_get_samplerate(this_source);
     }
-#else
-    usage(stderr, 1);
-    outmsg("Compiled without jack output, exiting.\n");
-#endif
-
-  } else {
-    /* phasevoc */
-    debug("Processing ...\n");
-
-    frames = 0;
-
-    while ((signed)overlap_size == aubio_sndfile_read(file, overlap_size, ibuf))
-    {
-      isonset=0;
-      process_func(ibuf->data, obuf->data, overlap_size);
-      print(); 
-      if (output_filename != NULL) {
-        aubio_sndfile_write(fileout,overlap_size,obuf);
+    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);
       }
-      frames++;
     }
-
-    debug("Processed %d frames of %d samples.\n", frames, buffer_size);
-    del_aubio_sndfile(file);
-
-    if (output_filename != NULL)
-      del_aubio_sndfile(fileout);
-
+#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 */
   }
-}
-
+  ibuf = new_fvec (hop_size);
+  obuf = new_fvec (hop_size);
 
+}
 
-void send_noteon(int pitch, int velo)
+void examples_common_del (void)
 {
-    smpl_t mpitch = floor(aubio_freqtomidi(pitch)+.5);
-    /* we should check if we use midi here, not jack */
-#if ALSA_SUPPORT
-    if (usejack) {
-        if (velo==0) {
-            aubio_midi_event_set_type(event,NOTE_OFF);
-        } else {
-            aubio_midi_event_set_type(event,NOTE_ON);
-        }
-        aubio_midi_event_set_channel(event,0);
-        aubio_midi_event_set_pitch(event,mpitch);
-        aubio_midi_event_set_velocity(event,velo);
-        aubio_midi_direct_output(mdriver,event);
-    } else 
+#ifdef HAVE_JACK
+  if (ev.buffer) free(ev.buffer);
 #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);
-        }
-    }
+  del_fvec (ibuf);
+  del_fvec (obuf);
+  aubio_cleanup ();
+  fflush(stderr);
+  fflush(stdout);
 }
 
+void examples_common_process (aubio_process_func_t process_func,
+    aubio_print_func_t print)
+{
 
-void note_append(fvec_t * note_buffer, smpl_t curnote) {
-  uint_t i = 0;
-  for (i = 0; i < note_buffer->length - 1; i++) { 
-      note_buffer->data[0][i] = note_buffer->data[0][i+1];
-  }
-  note_buffer->data[0][note_buffer->length - 1] = curnote;
-  return;
-}
+  uint_t read = 0;
+  if (usejack) {
+
+#ifdef HAVE_JACK
+    ev.size = 3;
+    ev.buffer = malloc (3 * sizeof (jack_midi_data_t));
+    ev.time = 0; // send it now
+    debug ("Jack activation ...\n");
+    aubio_jack_activate (jack_setup, process_func);
+    debug ("Processing (Ctrl+C to quit) ...\n");
+    pause ();
+    aubio_jack_close (jack_setup);
+#else /* HAVE_JACK */
+    usage (stderr, 1);
+    outmsg ("Compiled without jack output, exiting.\n");
+#endif /* HAVE_JACK */
 
-uint_t get_note(fvec_t *note_buffer, fvec_t *note_buffer2){
-  uint_t i = 0;
-  for (i = 0; i < note_buffer->length; i++) { 
-      note_buffer2->data[0][i] = note_buffer->data[0][i];
-  }
-  return vec_median(note_buffer2);
-}
+  } else {
 
-#if LASH_SUPPORT
+    uint_t total_read = 0;
+    blocks = 0;
 
-void * lash_thread_main(void *data __attribute__((unused)))
-{
-       printf("LASH thread running\n");
+    do {
+      aubio_source_do (this_source, ibuf, &read);
+      process_func (ibuf, obuf);
+      // print to console if verbose or no output given
+      if (verbose || sink_uri == NULL) {
+        print();
+      }
+      if (this_sink) {
+        aubio_sink_do (this_sink, obuf, hop_size);
+      }
+      blocks++;
+      total_read += read;
+    } while (read == hop_size);
 
-       while (!lash_main())
-               usleep(1000);
+    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);
 
-       printf("LASH thread finished\n");
-       return NULL;
-}
+    del_aubio_source (this_source);
+    del_aubio_sink   (this_sink);
 
-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);
-       }
-
-       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
+send_noteon (smpl_t pitch, smpl_t velo)
+{
+#ifdef HAVE_JACK
+  if (usejack) {
+    ev.buffer[2] = velo;
+    ev.buffer[1] = pitch;
+    if (velo == 0) {
+      ev.buffer[0] = 0x80;      /* note off */
+    } else {
+      ev.buffer[0] = 0x90;      /* note on */
+    }
+    aubio_jack_midi_event_write (jack_setup, (jack_midi_event_t *) & ev);
+  } else
+#endif
+  if (velo == 0) {
+    print_time (blocks * hop_size);
+    outmsg ("\n");
+  } else {
+    outmsg ("%f\t", pitch);
+    print_time (blocks * hop_size);
+    outmsg ("\t");
+  }
 }
 
-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 /* LASH_SUPPORT */
-