examples/utils.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 /**
  22
  23   This file includes some tools common to all examples. Code specific to the
  24   algorithm performed by each program should go in the source file of that
  25   program instead.
  26
  27 */
  28
  29 #include "utils.h"
  30 #ifdef HAVE_JACK
  31 #include "jackio.h"
  32 #endif /* HAVE_JACK */
  33
  34 int verbose = 0;
  35 int usejack = 0;
  36 // input / output
  37 char_t *sink_uri = NULL;
  38 char_t *source_uri = NULL;
  39 // general stuff
  40 uint_t samplerate = 0;
  41 uint_t buffer_size = 512;
  42 uint_t hop_size = 256;
  43 // onset stuff
  44 char_t * onset_method = "default";
  45 smpl_t onset_threshold = 0.0; // will be set if != 0.
  46 smpl_t onset_minioi = 0.0; // will be set if != 0.
  47 // pitch stuff
  48 char_t * pitch_unit = "default";
  49 char_t * pitch_method = "default";
  50 smpl_t pitch_tolerance = 0.0; // will be set if != 0.
  51 // time stuff
  52 uint_t time_format = 0; // for "seconds", 1 for "ms", 2 for "samples"
  53 // tempo stuff
  54 char_t * tempo_method = "default";
  55 // more general stuff
  56 smpl_t silence_threshold = -90.;
  57 uint_t mix_input = 0;
  58
  59 uint_t force_overwrite = 0;
  60
  61 //
  62 // internal memory stuff
  63 aubio_source_t *this_source = NULL;
  64 aubio_sink_t *this_sink = NULL;
  65 fvec_t *ibuf;
  66 fvec_t *obuf;
  67
  68 smpl_t miditap_note = 69.;
  69 smpl_t miditap_velo = 65.;
  70
  71 /* settings */
  72 int blocks = 0;
  73
  74 extern void usage (FILE * stream, int exit_code);
  75 extern int parse_args (int argc, char **argv);
  76
  77 #if HAVE_JACK
  78 aubio_jack_t *jack_setup;
  79 jack_midi_event_t ev;
  80 #endif /* HAVE_JACK */
  81
  82 void examples_common_init (int argc, char **argv);
  83 void examples_common_del (void);
  84 void examples_common_process (aubio_process_func_t process_func,
  85     aubio_print_func_t print);
  86
  87 void examples_common_init (int argc, char **argv)
  88 {
  89
  90   /* parse command line arguments */
  91   parse_args (argc, argv);
  92
  93   if (!usejack) {
  94     debug ("Opening files ...\n");
  95     this_source = new_aubio_source ((char_t*)source_uri, samplerate, hop_size);
  96     if (this_source == NULL) {
  97       errmsg ("Error: could not open input file %s\n", source_uri);
  98       exit (1);
  99     }
 100     if (samplerate == 0) {
 101       samplerate = aubio_source_get_samplerate(this_source);
 102     }
 103     if (sink_uri != NULL) {
 104       uint_t sink_exists = (access(sink_uri, F_OK) == 0 );
 105       if (!force_overwrite && sink_exists) {
 106         errmsg ("Error: output file %s already exists, use -f to overwrite.\n",
 107             sink_uri);
 108         exit (1);
 109       }
 110       this_sink = new_aubio_sink ((char_t*)sink_uri, samplerate);
 111       if (this_sink == NULL) {
 112         errmsg ("Error: could not create output file %s\n", sink_uri);
 113         exit (1);
 114       }
 115     }
 116 #ifdef HAVE_JACK
 117   } else {
 118     debug ("Jack init ...\n");
 119     jack_setup = new_aubio_jack (hop_size, 1, 1, 0, 1);
 120     samplerate = aubio_jack_get_samplerate (jack_setup);
 121     source_uri = "jack";
 122 #endif /* HAVE_JACK */
 123   }
 124   ibuf = new_fvec (hop_size);
 125   obuf = new_fvec (hop_size);
 126
 127 }
 128
 129 void examples_common_del (void)
 130 {
 131 #ifdef HAVE_JACK
 132   if (ev.buffer) free(ev.buffer);
 133 #endif
 134   del_fvec (ibuf);
 135   del_fvec (obuf);
 136   aubio_cleanup ();
 137   fflush(stderr);
 138   fflush(stdout);
 139 }
 140
 141 void examples_common_process (aubio_process_func_t process_func,
 142     aubio_print_func_t print)
 143 {
 144
 145   uint_t read = 0;
 146   if (usejack) {
 147
 148 #ifdef HAVE_JACK
 149     ev.size = 3;
 150     ev.buffer = malloc (3 * sizeof (jack_midi_data_t));
 151     ev.time = 0; // send it now
 152     debug ("Jack activation ...\n");
 153     aubio_jack_activate (jack_setup, process_func);
 154     debug ("Processing (Ctrl+C to quit) ...\n");
 155     pause ();
 156     aubio_jack_close (jack_setup);
 157 #else /* HAVE_JACK */
 158     usage (stderr, 1);
 159     outmsg ("Compiled without jack output, exiting.\n");
 160 #endif /* HAVE_JACK */
 161
 162   } else {
 163
 164     uint_t total_read = 0;
 165     blocks = 0;
 166
 167     do {
 168       aubio_source_do (this_source, ibuf, &read);
 169       process_func (ibuf, obuf);
 170       // print to console if verbose or no output given
 171       if (verbose || sink_uri == NULL) {
 172         print();
 173       }
 174       if (this_sink) {
 175         aubio_sink_do (this_sink, obuf, hop_size);
 176       }
 177       blocks++;
 178       total_read += read;
 179     } while (read == hop_size);
 180
 181     verbmsg ("read %.2fs (%d samples in %d blocks of %d) from %s at %dHz\n",
 182         total_read * 1. / samplerate,
 183         total_read, blocks, hop_size, source_uri, samplerate);
 184
 185     del_aubio_source (this_source);
 186     del_aubio_sink   (this_sink);
 187
 188   }
 189 }
 190
 191 void
 192 send_noteon (smpl_t pitch, smpl_t velo)
 193 {
 194 #ifdef HAVE_JACK
 195   if (usejack) {
 196     ev.buffer[2] = velo;
 197     ev.buffer[1] = pitch;
 198     if (velo == 0) {
 199       ev.buffer[0] = 0x80;      /* note off */
 200     } else {
 201       ev.buffer[0] = 0x90;      /* note on */
 202     }
 203     aubio_jack_midi_event_write (jack_setup, (jack_midi_event_t *) & ev);
 204   } else
 205 #endif
 206   if (velo == 0) {
 207     print_time (blocks * hop_size);
 208     outmsg ("\n");
 209   } else {
 210     outmsg ("%f\t", pitch);
 211     print_time (blocks * hop_size);
 212     outmsg ("\t");
 213   }
 214 }
 215
 216 void print_time (uint_t time_in_samples) {
 217   /* output times in selected format */
 218   if (time_format == 2) {
 219     outmsg ("%d", time_in_samples);
 220   } else if (time_format == 1) {
 221     outmsg ("%f", 1000. * time_in_samples / (float) samplerate);
 222   } else {
 223     outmsg ("%f", time_in_samples / (float) samplerate);
 224   }
 225 }