ref: 954585ac18e2791c05adc6546b3481952aabbb5c
dir: /examples/utils.c/
#include "aubio.h"
#ifndef JACK_SUPPORT
#define JACK_SUPPORT 0
#endif
#include <getopt.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <math.h> /* for isfinite */
#include "utils.h"
#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 */
/* 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;
int frames_delay = 0;
/* 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 if (strcmp(optarg,"specflux") == 0)
type_onset = aubio_onset_specflux;
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) {
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 */
ibuf = new_fvec(overlap_size, channels);
obuf = new_fvec(overlap_size, channels);
fftgrain = new_cvec(buffer_size, channels);
if (usepitch) {
pitchdet = new_aubio_pitchdetection(buffer_size*4,
overlap_size, channels, samplerate, type_pitch, mode_pitch);
aubio_pitchdetection_set_yinthresh(pitchdet, 0.7);
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);
}
}
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_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();
}
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);
}
#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);
}
frames++;
}
debug("Processed %d frames of %d samples.\n", frames, buffer_size);
flush_process(process_func, print);
del_aubio_sndfile(file);
if (output_filename != NULL)
del_aubio_sndfile(fileout);
}
}
void flush_process(aubio_process_func_t process_func, aubio_print_func_t print){
uint_t i,j;
for (i = 0; i < channels; i++) {
for (j = 0; j < obuf->length; j++) {
fvec_write_sample(obuf,0.,i,j);
}
}
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);
/* 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
#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);
}
}
}
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 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);
}
#if LASH_SUPPORT
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 /* LASH_SUPPORT */