ref: abd326c73a577a832b2f51106e80098e4b11034b
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 */