aubio: tempo/test-tempo.c

aubio 0.4.9

tempo/test-tempo.c
#include <aubio.h>
#include "utils_tests.h"
int test_wrong_params(void);
int main (int argc, char **argv)
{
uint_t err = 0;
if (argc < 2) {
PRINT_WRN("no arguments, running tests\n");
err = test_wrong_params();
PRINT_MSG("usage: %s <source_path> [samplerate] [win_size] [hop_size]\n",
argv[0]);
return err;
}
uint_t samplerate = 0;
if ( argc >= 3 ) samplerate = atoi(argv[2]);
uint_t win_size = 1024; // window size
if ( argc >= 4 ) win_size = atoi(argv[3]);
uint_t hop_size = win_size / 4;
if ( argc >= 5 ) hop_size = atoi(argv[4]);
uint_t n_frames = 0, read = 0;
char_t *source_path = argv[1];
aubio_source_t * source = new_aubio_source(source_path, samplerate,
hop_size);
if (!source) { err = 1; goto beach; }
if (samplerate == 0 ) samplerate = aubio_source_get_samplerate(source);
// create some vectors
fvec_t * in = new_fvec (hop_size); // input audio buffer
fvec_t * out = new_fvec (1); // output position
// create tempo object
aubio_tempo_t * o = new_aubio_tempo("default", win_size, hop_size,
samplerate);
if (!o) { err = 1; goto beach_tempo; }
do {
// put some fresh data in input vector
aubio_source_do(source, in, &read);
// execute tempo
aubio_tempo_do(o,in,out);
// do something with the beats
if (out->data[0] != 0) {
PRINT_MSG("beat at %.3fms, %.3fs, frame %d, %.2f bpm "
"with confidence %.2f\n",
aubio_tempo_get_last_ms(o), aubio_tempo_get_last_s(o),
aubio_tempo_get_last(o), aubio_tempo_get_bpm(o),
aubio_tempo_get_confidence(o));
}
n_frames += read;
} while ( read == hop_size );
PRINT_MSG("read %.2fs, %d frames at %dHz (%d blocks) from %s\n",
n_frames * 1. / samplerate,
n_frames, samplerate,
n_frames / hop_size, source_path);
// clean up memory
del_aubio_tempo(o);
beach_tempo:
del_fvec(in);
del_fvec(out);
del_aubio_source(source);
beach:
aubio_cleanup();
return err;
}
int test_wrong_params(void)
{
uint_t win_size = 1024;
uint_t hop_size = 256;
uint_t samplerate = 44100;
aubio_tempo_t *t;
fvec_t* in, *out;
uint_t i;
// test wrong method fails
if (new_aubio_tempo("undefined", win_size, hop_size, samplerate)) return 1;
// test hop > win fails
if (new_aubio_tempo("default", hop_size, win_size, samplerate)) return 1;
// test null hop_size fails
if (new_aubio_tempo("default", win_size, 0, samplerate)) return 1;
// test 1 buf_size fails
if (new_aubio_tempo("default", 1, 1, samplerate)) return 1;
// test null samplerate fails
if (new_aubio_tempo("default", win_size, hop_size, 0)) return 1;
// test short sizes workaround
t = new_aubio_tempo("default", 2048, 2048, 500);
if (!t) return 1;
del_aubio_tempo(t);
t = new_aubio_tempo("default", win_size, hop_size, samplerate);
if (!t) return 1;
in = new_fvec(hop_size);
out = new_fvec(1);
// up to step = (next_power_of_two(5.8 * samplerate / hop_size ) / 4 )
for (i = 0; i < 256 + 1; i++)
{
aubio_tempo_do(t,in,out);
PRINT_MSG("beat at %.3fms, %.3fs, frame %d, %.2f bpm "
"with confidence %.2f, was tatum %d\n",
aubio_tempo_get_last_ms(t), aubio_tempo_get_last_s(t),
aubio_tempo_get_last(t), aubio_tempo_get_bpm(t),
aubio_tempo_get_confidence(t), aubio_tempo_was_tatum(t));
}
del_aubio_tempo(t);
del_fvec(in);
del_fvec(out);
return run_on_default_source(main);
}
new_aubio_tempo
aubio_tempo_t * new_aubio_tempo(const char_t *method, uint_t buf_size, uint_t hop_size, uint_t samplerate)
create tempo detection object
char_t
char char_t
character
Definition: types.h:64
aubio_source_t
struct _aubio_source_t aubio_source_t
media source object
Definition: source.h:70
fvec_t::data
smpl_t * data
data vector of length fvec_t.length
Definition: fvec.h:69
aubio_tempo_get_last_ms
smpl_t aubio_tempo_get_last_ms(aubio_tempo_t *o)
get the time of the latest beat detected, in milliseconds
aubio_source_get_samplerate
uint_t aubio_source_get_samplerate(aubio_source_t *s)
get samplerate of source object
del_aubio_tempo
void del_aubio_tempo(aubio_tempo_t *o)
delete tempo detection object
aubio_tempo_get_confidence
smpl_t aubio_tempo_get_confidence(aubio_tempo_t *o)
get current tempo confidence
aubio_tempo_was_tatum
uint_t aubio_tempo_was_tatum(aubio_tempo_t *o)
check whether a tatum was detected in the current frame
aubio_cleanup
void aubio_cleanup(void)
clean up cached memory at the end of program
aubio_tempo_get_bpm
smpl_t aubio_tempo_get_bpm(aubio_tempo_t *o)
get current tempo
aubio_tempo_t
struct _aubio_tempo_t aubio_tempo_t
tempo detection structure
Definition: tempo.h:41
new_fvec
fvec_t * new_fvec(uint_t length)
fvec_t buffer creation function
aubio.h
fvec_t
Buffer for real data.
Definition: fvec.h:67
aubio_tempo_get_last
uint_t aubio_tempo_get_last(aubio_tempo_t *o)
get the time of the latest beat detected, in samples
del_aubio_source
void del_aubio_source(aubio_source_t *s)
close source and cleanup memory
del_fvec
void del_fvec(fvec_t *s)
fvec_t buffer deletion function
aubio_tempo_do
void aubio_tempo_do(aubio_tempo_t *o, const fvec_t *input, fvec_t *tempo)
execute tempo detection
new_aubio_source
aubio_source_t * new_aubio_source(const char_t *uri, uint_t samplerate, uint_t hop_size)
create new aubio_source_t
aubio_tempo_get_last_s
smpl_t aubio_tempo_get_last_s(aubio_tempo_t *o)
get the time of the latest beat detected, in seconds
uint_t
unsigned int uint_t
unsigned integer
Definition: types.h:60
aubio_source_do
void aubio_source_do(aubio_source_t *s, fvec_t *read_to, uint_t *read)
read monophonic vector of length hop_size from source object