source: tests/src/spectral/test-awhitening.c @ 90e25cc

feature/autosinkfeature/cnnfeature/cnn_orgfeature/constantqfeature/crepefeature/crepe_orgfeature/pitchshiftfeature/pydocstringsfeature/timestretchfix/ffmpeg5sampler
Last change on this file since 90e25cc was 90e25cc, checked in by Paul Brossier <piem@piem.org>, 7 years ago

tests/src/spectral/tests-awhitening.c: add whitening example

  • Property mode set to 100644
File size: 2.2 KB
Line 
1#include <aubio.h>
2#include "utils_tests.h"
3
4int main (int argc, char **argv)
5{
6  sint_t err = 0;
7
8  if (argc < 3) {
9    err = 2;
10    PRINT_ERR("not enough arguments\n");
11    PRINT_MSG("usage: %s <input_path> <output_path> [samplerate] [hop_size]\n", argv[0]);
12    return err;
13  }
14
15  uint_t samplerate = 0;
16  uint_t win_size = 1024;
17  uint_t hop_size = 512;
18  uint_t n_frames = 0, read = 0;
19
20  char_t *source_path = argv[1];
21  char_t *sink_path = argv[2];
22
23  if ( argc >= 4 ) samplerate = atoi(argv[3]);
24  if ( argc >= 5 ) hop_size = atoi(argv[4]);
25  if ( argc >= 6 ) {
26    err = 2;
27    PRINT_ERR("too many arguments\n");
28    return err;
29  }
30
31  fvec_t *vec = new_fvec(hop_size);
32  fvec_t *out = new_fvec(hop_size); // output buffer
33  fvec_t *scale = new_fvec(hop_size);
34  cvec_t *fftgrain = new_cvec(win_size); // fft norm and phase
35  if (!vec) { err = 1; goto beach_fvec; }
36
37  aubio_source_t *i = new_aubio_source(source_path, samplerate, hop_size);
38  if (!i) { err = 1; goto beach_source; }
39
40  if (samplerate == 0 ) samplerate = aubio_source_get_samplerate(i);
41
42  aubio_sink_t *o = new_aubio_sink(sink_path, samplerate);
43  if (!o) { err = 1; goto beach_sink; }
44
45  aubio_pvoc_t *pv = new_aubio_pvoc(win_size, hop_size);
46
47  aubio_spectral_whitening_t *awhitening =
48    new_aubio_spectral_whitening (win_size, hop_size, samplerate);
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_sink(o);
77beach_sink:
78  del_aubio_source(i);
79beach_source:
80  del_fvec(vec);
81beach_fvec:
82  return err;
83}
84
Note: See TracBrowser for help on using the repository browser.