aubio: io/test-sink.c

aubio 0.4.9

Loading...
Searching...
No Matches
io/test-sink.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 < 3 || argc >= 6) {
PRINT_ERR("wrong number of arguments, running tests\n");
err = test_wrong_params();
PRINT_MSG("usage: %s <input_path> <output_path> [samplerate] [hop_size]\n",
argv[0]);
return err;
}
uint_t samplerate = 0;
uint_t hop_size = 512;
uint_t n_frames = 0, read = 0;
char_t *source_path = argv[1];
char_t *sink_path = argv[2];
aubio_source_t *src = NULL;
aubio_sink_t *snk = NULL;
if ( argc >= 4 ) samplerate = atoi(argv[3]);
if ( argc >= 5 ) hop_size = atoi(argv[4]);
fvec_t *vec = new_fvec(hop_size);
if (!vec) { err = 1; goto failure; }
src = new_aubio_source(source_path, samplerate, hop_size);
if (!src) { err = 1; goto failure; }
if (samplerate == 0 ) samplerate = aubio_source_get_samplerate(src);
snk = new_aubio_sink(sink_path, samplerate);
if (!snk) { err = 1; goto failure; }
do {
aubio_source_do(src, vec, &read);
aubio_sink_do(snk, vec, read);
n_frames += read;
} while ( read == hop_size );
PRINT_MSG("%d frames read at %dHz (%d blocks) from %s and written to %s\n",
n_frames, samplerate, n_frames / hop_size,
source_path, sink_path);
// close sink now (optional)
aubio_sink_close(snk);
failure:
if (snk)
del_aubio_sink(snk);
if (src)
del_aubio_source(src);
if (vec)
del_fvec(vec);
return err;
}
int test_wrong_params(void)
{
fvec_t *vec;
fmat_t *mat;
aubio_sink_t *s;
char_t sink_path[PATH_MAX] = "tmp_aubio_XXXXXX";
uint_t samplerate = 44100;
uint_t hop_size = 256;
uint_t oversized_hop_size = 4097;
uint_t oversized_samplerate = 192000 * 8 + 1;
uint_t channels = 3;
uint_t oversized_channels = 1025;
// create temp file
int fd = create_temp_sink(sink_path);
if (!fd) return 1;
if (new_aubio_sink( 0, samplerate)) return 1;
if (new_aubio_sink("\0", samplerate)) return 1;
if (new_aubio_sink(sink_path, -1)) return 1;
s = new_aubio_sink(sink_path, 0);
// check setting wrong parameters fails
if (!aubio_sink_preset_samplerate(s, oversized_samplerate)) return 1;
if (!aubio_sink_preset_channels(s, oversized_channels)) return 1;
if (!aubio_sink_preset_channels(s, -1)) return 1;
// check setting valid parameters passes
if (aubio_sink_preset_samplerate(s, samplerate)) return 1;
if (aubio_sink_preset_channels(s, 1)) return 1;
// check writing a vector with valid length
vec = new_fvec(hop_size);
aubio_sink_do(s, vec, hop_size);
// check writing more than in the input
aubio_sink_do(s, vec, hop_size+1);
// check write 0 frames
aubio_sink_do(s, vec, 0);
del_fvec(vec);
// check writing an oversized vector
vec = new_fvec(oversized_hop_size);
aubio_sink_do(s, vec, oversized_hop_size);
del_fvec(vec);
// test delete without closing
del_aubio_sink(s);
s = new_aubio_sink(sink_path, 0);
// preset channels first
if (aubio_sink_preset_channels(s, channels)) return 1;
if (aubio_sink_preset_samplerate(s, samplerate)) return 1;
if (aubio_sink_get_samplerate(s) != samplerate) return 1;
if (aubio_sink_get_channels(s) != channels) return 1;
mat = new_fmat(channels, hop_size);
// check writing a vector with valid length
aubio_sink_do_multi(s, mat, hop_size);
// check writing 0 frames
aubio_sink_do_multi(s, mat, 0);
// check writing more than in the input
aubio_sink_do_multi(s, mat, hop_size+1);
del_fmat(mat);
// check writing oversized input
mat = new_fmat(channels, oversized_hop_size);
aubio_sink_do_multi(s, mat, oversized_hop_size);
del_fmat(mat);
// check writing undersized input
mat = new_fmat(channels - 1, hop_size);
aubio_sink_do_multi(s, mat, hop_size);
del_fmat(mat);
aubio_sink_close(s);
// test closing twice
aubio_sink_close(s);
del_aubio_sink(s);
// delete temp file
close_temp_sink(sink_path, fd);
// shouldn't crash on null
del_aubio_sink(NULL);
return run_on_default_source_and_sink(main);
}
aubio.h
Global aubio include file.
del_fmat
void del_fmat(fmat_t *s)
fmat_t buffer deletion function
new_fmat
fmat_t * new_fmat(uint_t height, uint_t length)
fmat_t buffer creation function
new_fvec
fvec_t * new_fvec(uint_t length)
fvec_t buffer creation function
del_fvec
void del_fvec(fvec_t *s)
fvec_t buffer deletion function
aubio_sink_close
uint_t aubio_sink_close(aubio_sink_t *s)
close sink
aubio_sink_preset_samplerate
uint_t aubio_sink_preset_samplerate(aubio_sink_t *s, uint_t samplerate)
preset sink samplerate
aubio_sink_do_multi
void aubio_sink_do_multi(aubio_sink_t *s, fmat_t *write_data, uint_t write)
write polyphonic vector of length hop_size to sink
aubio_sink_do
void aubio_sink_do(aubio_sink_t *s, fvec_t *write_data, uint_t write)
write monophonic vector of length hop_size to sink
del_aubio_sink
void del_aubio_sink(aubio_sink_t *s)
close sink and cleanup memory
aubio_sink_get_channels
uint_t aubio_sink_get_channels(const aubio_sink_t *s)
get channels of sink object
aubio_sink_t
struct _aubio_sink_t aubio_sink_t
media sink object
Definition: sink.h:61
new_aubio_sink
aubio_sink_t * new_aubio_sink(const char_t *uri, uint_t samplerate)
create new aubio_sink_t
aubio_sink_preset_channels
uint_t aubio_sink_preset_channels(aubio_sink_t *s, uint_t channels)
preset sink channels
aubio_sink_get_samplerate
uint_t aubio_sink_get_samplerate(const aubio_sink_t *s)
get samplerate of sink object
aubio_source_get_samplerate
uint_t aubio_source_get_samplerate(aubio_source_t *s)
get samplerate of source object
aubio_source_t
struct _aubio_source_t aubio_source_t
media source object
Definition: source.h:70
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
del_aubio_source
void del_aubio_source(aubio_source_t *s)
close source and cleanup memory
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
fmat_t
Buffer for real data.
Definition: fmat.h:40
fvec_t
Buffer for real data.
Definition: fvec.h:67
uint_t
unsigned int uint_t
unsigned integer
Definition: types.h:60
char_t
char char_t
character
Definition: types.h:64