ref: aa175814750ff59d3dc11040ef9929b614860a45
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" /* not supported yet */ #ifdef LADCCA_SUPPORT #include <ladcca/ladcca.h> cca_client_t * aubio_cca_client; #endif /* LADCCA_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; /* 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 threshold2 = -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_schmitt; // 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" " -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} }; 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 */ threshold2 = (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,"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) { /* 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 (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); } 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); 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); } } del_aubio_pvoc(pv); del_fvec(obuf); del_fvec(ibuf); del_cvec(fftgrain); del_fvec(onset); } 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 (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); del_aubio_sndfile(file); if (output_filename != NULL) del_aubio_sndfile(fileout); } } 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); }