source: tests/src/spectral/test-mfcc.c @ 6b84d81

feature/autosinkfeature/cnnfeature/cnn_orgfeature/constantqfeature/crepefeature/crepe_orgfeature/pitchshiftfeature/timestretchfix/ffmpeg5
Last change on this file since 6b84d81 was 423a427, checked in by Paul Brossier <piem@piem.org>, 5 years ago

[tests] improve test-mfcc (closes #219)

  • Property mode set to 100644
File size: 2.3 KB
Line 
1#include <aubio.h>
2#include "utils_tests.h"
3
4int test_wrong_params(void);
5
6int main (int argc, char** argv)
7{
8  sint_t err = 0;
9
10  if (argc < 2) {
11    err = 2;
12    PRINT_WRN("no arguments, running tests\n");
13    err = test_wrong_params();
14    PRINT_MSG("usage: %s <input_path> [samplerate] [hop_size]\n", argv[0]);
15    return err;
16  }
17
18  uint_t win_s; // fft size
19  uint_t hop_s = 256; // block size
20  uint_t samplerate = 0; // samplerate
21  uint_t n_filters = 40; // number of filters
22  uint_t n_coeffs = 13; // number of coefficients
23  uint_t read = 0;
24
25  char_t *source_path = argv[1];
26
27  if ( argc >= 3 ) samplerate = atoi(argv[2]);
28  if ( argc >= 4 ) hop_s = atoi(argv[3]);
29
30  win_s = 2 * hop_s;
31
32  aubio_source_t *source = 0;
33  aubio_pvoc_t *pv = 0;
34  aubio_mfcc_t *mfcc = 0;
35
36  fvec_t *in = new_fvec (win_s); // input buffer
37  cvec_t *fftgrain = new_cvec (win_s); // input buffer
38  fvec_t *out = new_fvec (n_coeffs); // output coefficients
39
40  if (!in || !fftgrain || !out) { err = 1; goto failure; }
41
42  // source
43  source = new_aubio_source(source_path, samplerate, hop_s);
44  if (!source) { err = 1; goto failure; }
45  if (samplerate == 0) samplerate = aubio_source_get_samplerate(source);
46
47  // phase vocoder
48  pv = new_aubio_pvoc(win_s, hop_s);
49  if (!pv) { err = 1; goto failure; }
50
51  // mfcc object
52  mfcc = new_aubio_mfcc (win_s, n_filters, n_coeffs, samplerate);
53  if (!mfcc) { err = 1; goto failure; }
54
55  // processing loop
56  do {
57    aubio_source_do(source, in, &read);
58    aubio_pvoc_do(pv, in, fftgrain);
59    aubio_mfcc_do(mfcc, fftgrain, out);
60    fvec_print(out);
61  } while (read == hop_s);
62
63failure:
64
65  if (mfcc)
66    del_aubio_mfcc(mfcc);
67  if (pv)
68    del_aubio_pvoc(pv);
69  if (source)
70    del_aubio_source(source);
71  if (in)
72    del_fvec(in);
73  if (fftgrain)
74    del_cvec(fftgrain);
75  if (out)
76    del_fvec(out);
77  aubio_cleanup();
78  return err;
79}
80
81int test_wrong_params()
82{
83  uint_t win_s = 512; // fft size
84  uint_t n_filters = 40; // number of filters
85  uint_t n_coeffs = 13; // number of coefficients
86  smpl_t samplerate = 16000.; // samplerate
87
88  if (new_aubio_mfcc(    0, n_filters, n_coeffs, samplerate)) return 1;
89  if (new_aubio_mfcc(win_s,         0, n_coeffs, samplerate)) return 1;
90  if (new_aubio_mfcc(win_s, n_filters,        0, samplerate)) return 1;
91  if (new_aubio_mfcc(win_s, n_filters, n_coeffs,          0)) return 1;
92
93  return run_on_default_source(main);
94}
Note: See TracBrowser for help on using the repository browser.