[0028ea7] | 1 | #include <aubio.h> |
---|
[423a427] | 2 | #include "utils_tests.h" |
---|
[0028ea7] | 3 | |
---|
[423a427] | 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 | |
---|
[ef8a1e4] | 36 | fvec_t *in = new_fvec (hop_s); // phase vocoder input |
---|
| 37 | cvec_t *fftgrain = new_cvec (win_s); // pvoc output / mfcc input |
---|
| 38 | fvec_t *out = new_fvec (n_coeffs); // mfcc output |
---|
[423a427] | 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() |
---|
[0028ea7] | 82 | { |
---|
[6938a20] | 83 | uint_t win_s = 512; // fft size |
---|
| 84 | uint_t n_filters = 40; // number of filters |
---|
[b174319] | 85 | uint_t n_coeffs = 13; // number of coefficients |
---|
[6938a20] | 86 | smpl_t samplerate = 16000.; // samplerate |
---|
[b174319] | 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; |
---|
[0028ea7] | 92 | |
---|
[423a427] | 93 | return run_on_default_source(main); |
---|
[0028ea7] | 94 | } |
---|