[tests] improve test-mfcc (closes #219)
[aubio.git] / tests / src / spectral / test-mfcc.c
1 #include <aubio.h>
2 #include "utils_tests.h"
3
4 int test_wrong_params(void);
5
6 int 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
63 failure:
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
81 int 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 }