ref: 69cd58c36e949af602d19ab9bcaa3711fd80d594
dir: /tests/src/spectral/test-awhitening.c/
#include <aubio.h> #include "utils_tests.h" int test_wrong_params(void); int main (int argc, char **argv) { sint_t err = 0; if (argc < 3) { err = 2; PRINT_WRN("no 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 win_size = 1024; uint_t hop_size = 512; uint_t n_frames = 0, read = 0; char_t *source_path = argv[1]; char_t *sink_path = argv[2]; if ( argc >= 4 ) samplerate = atoi(argv[3]); if ( argc >= 5 ) hop_size = atoi(argv[4]); fvec_t *vec = new_fvec(hop_size); fvec_t *out = new_fvec(hop_size); // output buffer fvec_t *scale = new_fvec(hop_size); cvec_t *fftgrain = new_cvec(win_size); // fft norm and phase if (!vec) { err = 1; goto beach_fvec; } aubio_source_t *i = new_aubio_source(source_path, samplerate, hop_size); if (!i) { err = 1; goto beach_source; } if (samplerate == 0 ) samplerate = aubio_source_get_samplerate(i); aubio_sink_t *o = new_aubio_sink(sink_path, samplerate); if (!o) { err = 1; goto beach_sink; } aubio_pvoc_t *pv = new_aubio_pvoc(win_size, hop_size); if (!pv) { err = 1; goto beach_pvoc; } aubio_spectral_whitening_t *awhitening = new_aubio_spectral_whitening (win_size, hop_size, samplerate); if (!awhitening) { err = 1; goto beach_awhitening; } aubio_spectral_whitening_set_relax_time(awhitening, 20.); fvec_set_all(scale, 3.); PRINT_MSG("spectral whitening relaxation time is %f\n", aubio_spectral_whitening_get_relax_time(awhitening)); do { aubio_source_do(i, vec, &read); aubio_pvoc_do(pv, vec, fftgrain); // apply spectral whitening aubio_spectral_whitening_do(awhitening, fftgrain); // rebuild the signal aubio_pvoc_rdo(pv, fftgrain, out); // make louder fvec_weight(out, scale); // make sure we dont saturate fvec_clamp(out, 1.); // write output aubio_sink_do(o, out, read); n_frames += read; } while ( read == hop_size ); PRINT_MSG("read %d frames at %dHz (%d blocks) from %s written to %s\n", n_frames, samplerate, n_frames / hop_size, source_path, sink_path); del_aubio_spectral_whitening(awhitening); beach_awhitening: del_aubio_pvoc(pv); beach_pvoc: del_aubio_sink(o); beach_sink: del_aubio_source(i); beach_source: del_fvec(vec); del_fvec(out); del_fvec(scale); del_cvec(fftgrain); beach_fvec: return err; } int test_wrong_params(void) { uint_t buf_size = 512; uint_t hop_size = 256; uint_t samplerate = 44100; aubio_spectral_whitening_t *o; if (new_aubio_spectral_whitening( 0, hop_size, samplerate)) return 1; if (new_aubio_spectral_whitening(buf_size, 0, samplerate)) return 1; if (new_aubio_spectral_whitening(buf_size, hop_size, 0)) return 1; o = new_aubio_spectral_whitening(buf_size, hop_size, samplerate); aubio_spectral_whitening_get_relax_time(o); aubio_spectral_whitening_get_floor(o); del_aubio_spectral_whitening(o); return run_on_default_source_and_sink(main); }