ref: 6ff6d18ce7790aa8adc8889923f13eafc8414ab6
dir: /examples/utils.c/
/* Copyright (C) 2003-2009 Paul Brossier <piem@aubio.org> This file is part of aubio. 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. 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. You should have received a copy of the GNU General Public License along with aubio. If not, see <http://www.gnu.org/licenses/>. */ /** 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. */ #include "utils.h" #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 *sink_uri = NULL; const char *source_uri = NULL; int frames = 0; int verbose = 0; int usejack = 0; int frames_delay = 0; 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; aubio_source_t *this_source = NULL; aubio_sink_t *this_sink = 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); } 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': sink_uri = optarg; break; case 'i': source_uri = 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); if ( source_uri == NULL ) { if (argc - optind == 1) { source_uri = argv[optind]; } else if ( argc - optind > 1 ) { errmsg ("Error: too many non-option arguments `%s'\n", argv[argc - 1]); usage ( stderr, 1 ); } } else if ( argc - optind > 0 ) { errmsg ("Error: extra non-option argument %s\n", argv[optind]); usage ( stderr, 1 ); } if (source_uri != NULL) { debug ("Input file : %s\n", source_uri); } else if (source_uri != NULL && sink_uri != NULL) { debug ("Input file : %s\n", source_uri); debug ("Output file : %s\n", sink_uri); } else { #if HAVE_JACK debug ("Jack input output\n"); usejack = 1; #else errmsg("Error: no arguments given (and no available audio input)\n"); usage ( stderr, 1 ); #endif } return 0; } void examples_common_init (int argc, char **argv) { /* parse command line arguments */ parse_args (argc, argv); if (!usejack) { debug ("Opening files ...\n"); this_source = new_aubio_source ((char_t*)source_uri, 0, overlap_size); if (this_source == NULL) { outmsg ("Could not open input file %s.\n", source_uri); exit (1); } samplerate = aubio_source_get_samplerate(this_source); if (sink_uri != NULL) { this_sink = new_aubio_sink ((char_t*)sink_uri, samplerate); if (this_sink == NULL) { outmsg ("Could not open output file %s.\n", sink_uri); exit (1); } } } #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); } } #endif /* HAVE_LASH */ woodblock = new_fvec (overlap_size); //TODO create woodblock sound ibuf = new_fvec (overlap_size); obuf = new_fvec (overlap_size); } void examples_common_del (void) { del_fvec (ibuf); del_fvec (obuf); del_fvec (woodblock); aubio_cleanup (); } #if HAVE_JACK aubio_jack_t *jack_setup; #endif 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); debug ("Jack activation ...\n"); aubio_jack_activate (jack_setup); debug ("Processing (Ctrl+C to quit) ...\n"); pause (); aubio_jack_close (jack_setup); #else usage (stderr, 1); outmsg ("Compiled without jack output, exiting.\n"); #endif } else { /* phasevoc */ debug ("Processing ...\n"); frames = 0; do { aubio_source_do (this_source, ibuf, &read); process_func (&ibuf->data, &obuf->data, overlap_size); print (); if (this_sink) { aubio_sink_do (this_sink, obuf, overlap_size); } frames++; } while (read == overlap_size); debug ("Processed %d frames of %d samples.\n", frames, buffer_size); flush_process (process_func, print); del_aubio_source (this_source); 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) { 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; if (usejack) { ev.buffer[2] = velo; ev.buffer[1] = mpitch; 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 (!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); } 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; } } #endif /* HAVE_LASH */