ref: 90e25cc760d412b373a8830e3df8ce4ce6e59a2b
parent: 04ebe31d4ec7fbc950c7fc57f0f8da559d0a616c
author: Paul Brossier <[email protected]>
date: Fri Mar 10 19:22:38 EST 2017
tests/src/spectral/tests-awhitening.c: add whitening example
--- /dev/null
+++ b/tests/src/spectral/test-awhitening.c
@@ -1,0 +1,84 @@
+#include <aubio.h>
+#include "utils_tests.h"
+
+int main (int argc, char **argv)
+{+ sint_t err = 0;
+
+ if (argc < 3) {+ err = 2;
+ PRINT_ERR("not enough arguments\n");+ 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]);
+ if ( argc >= 6 ) {+ err = 2;
+ PRINT_ERR("too many arguments\n");+ return err;
+ }
+
+ 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);
+
+ aubio_spectral_whitening_t *awhitening =
+ new_aubio_spectral_whitening (win_size, hop_size, samplerate);
+
+ 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_sink(o);
+beach_sink:
+ del_aubio_source(i);
+beach_source:
+ del_fvec(vec);
+beach_fvec:
+ return err;
+}
+