1 #include <aubio.h>
2 #include "utils_tests.h"
3
4 int test_wrong_params(void);
5
main(int argc,char ** argv)6 int main (int argc, char **argv)
7 {
8 sint_t err = 0;
9
10 if (argc < 3) {
11 err = 2;
12 PRINT_WRN("no arguments, running tests\n");
13 err = test_wrong_params();
14 PRINT_MSG("usage: %s <input_path> <output_path> [samplerate] [hop_size]\n", argv[0]);
15 return err;
16 }
17
18 uint_t samplerate = 0;
19 uint_t win_size = 1024;
20 uint_t hop_size = 512;
21 uint_t n_frames = 0, read = 0;
22
23 char_t *source_path = argv[1];
24 char_t *sink_path = argv[2];
25
26 if ( argc >= 4 ) samplerate = atoi(argv[3]);
27 if ( argc >= 5 ) hop_size = atoi(argv[4]);
28
29 fvec_t *vec = new_fvec(hop_size);
30 fvec_t *out = new_fvec(hop_size); // output buffer
31 fvec_t *scale = new_fvec(hop_size);
32 cvec_t *fftgrain = new_cvec(win_size); // fft norm and phase
33 if (!vec) { err = 1; goto beach_fvec; }
34
35 aubio_source_t *i = new_aubio_source(source_path, samplerate, hop_size);
36 if (!i) { err = 1; goto beach_source; }
37
38 if (samplerate == 0 ) samplerate = aubio_source_get_samplerate(i);
39
40 aubio_sink_t *o = new_aubio_sink(sink_path, samplerate);
41 if (!o) { err = 1; goto beach_sink; }
42
43 aubio_pvoc_t *pv = new_aubio_pvoc(win_size, hop_size);
44 if (!pv) { err = 1; goto beach_pvoc; }
45
46 aubio_spectral_whitening_t *awhitening =
47 new_aubio_spectral_whitening (win_size, hop_size, samplerate);
48 if (!awhitening) { err = 1; goto beach_awhitening; }
49
50 aubio_spectral_whitening_set_relax_time(awhitening, 20.);
51 fvec_set_all(scale, 3.);
52
53 PRINT_MSG("spectral whitening relaxation time is %f\n",
54 aubio_spectral_whitening_get_relax_time(awhitening));
55
56 do {
57 aubio_source_do(i, vec, &read);
58 aubio_pvoc_do(pv, vec, fftgrain);
59 // apply spectral whitening
60 aubio_spectral_whitening_do(awhitening, fftgrain);
61 // rebuild the signal
62 aubio_pvoc_rdo(pv, fftgrain, out);
63 // make louder
64 fvec_weight(out, scale);
65 // make sure we dont saturate
66 fvec_clamp(out, 1.);
67 // write output
68 aubio_sink_do(o, out, read);
69 n_frames += read;
70 } while ( read == hop_size );
71
72 PRINT_MSG("read %d frames at %dHz (%d blocks) from %s written to %s\n",
73 n_frames, samplerate, n_frames / hop_size,
74 source_path, sink_path);
75
76 del_aubio_spectral_whitening(awhitening);
77 beach_awhitening:
78 del_aubio_pvoc(pv);
79 beach_pvoc:
80 del_aubio_sink(o);
81 beach_sink:
82 del_aubio_source(i);
83 beach_source:
84 del_fvec(vec);
85 del_fvec(out);
86 del_fvec(scale);
87 del_cvec(fftgrain);
88 beach_fvec:
89 return err;
90 }
91
test_wrong_params(void)92 int test_wrong_params(void)
93 {
94 uint_t buf_size = 512;
95 uint_t hop_size = 256;
96 uint_t samplerate = 44100;
97 aubio_spectral_whitening_t *o;
98
99 if (new_aubio_spectral_whitening( 0, hop_size, samplerate)) return 1;
100 if (new_aubio_spectral_whitening(buf_size, 0, samplerate)) return 1;
101 if (new_aubio_spectral_whitening(buf_size, hop_size, 0)) return 1;
102
103 o = new_aubio_spectral_whitening(buf_size, hop_size, samplerate);
104
105 aubio_spectral_whitening_get_relax_time(o);
106 aubio_spectral_whitening_get_floor(o);
107
108 del_aubio_spectral_whitening(o);
109
110 return run_on_default_source_and_sink(main);
111 }
112