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 uint_t err = 0;
9 if (argc < 3 || argc >= 6) {
10 PRINT_ERR("wrong number of arguments, running tests\n");
11 err = test_wrong_params();
12 PRINT_MSG("usage: %s <input_path> <output_path> [samplerate] [hop_size]\n",
13 argv[0]);
14 return err;
15 }
16
17 uint_t samplerate = 0;
18 uint_t hop_size = 512;
19 uint_t n_frames = 0, read = 0;
20
21 char_t *source_path = argv[1];
22 char_t *sink_path = argv[2];
23
24 aubio_source_t *src = NULL;
25 aubio_sink_t *snk = NULL;
26
27 if ( argc >= 4 ) samplerate = atoi(argv[3]);
28 if ( argc >= 5 ) hop_size = atoi(argv[4]);
29
30 fvec_t *vec = new_fvec(hop_size);
31 if (!vec) { err = 1; goto failure; }
32
33 src = new_aubio_source(source_path, samplerate, hop_size);
34 if (!src) { err = 1; goto failure; }
35 if (samplerate == 0 ) samplerate = aubio_source_get_samplerate(src);
36
37 snk = new_aubio_sink(sink_path, samplerate);
38 if (!snk) { err = 1; goto failure; }
39
40 do {
41 aubio_source_do(src, vec, &read);
42 aubio_sink_do(snk, vec, read);
43 n_frames += read;
44 } while ( read == hop_size );
45
46 PRINT_MSG("%d frames read at %dHz (%d blocks) from %s and written to %s\n",
47 n_frames, samplerate, n_frames / hop_size,
48 source_path, sink_path);
49
50 // close sink now (optional)
51 aubio_sink_close(snk);
52
53 failure:
54 if (snk)
55 del_aubio_sink(snk);
56 if (src)
57 del_aubio_source(src);
58 if (vec)
59 del_fvec(vec);
60
61 return err;
62 }
63
test_wrong_params(void)64 int test_wrong_params(void)
65 {
66 fvec_t *vec;
67 fmat_t *mat;
68 aubio_sink_t *s;
69 char_t sink_path[PATH_MAX] = "tmp_aubio_XXXXXX";
70 uint_t samplerate = 44100;
71 uint_t hop_size = 256;
72 uint_t oversized_hop_size = 4097;
73 uint_t oversized_samplerate = 192000 * 8 + 1;
74 uint_t channels = 3;
75 uint_t oversized_channels = 1025;
76 // create temp file
77 int fd = create_temp_sink(sink_path);
78
79 if (!fd) return 1;
80
81 if (new_aubio_sink( 0, samplerate)) return 1;
82 if (new_aubio_sink("\0", samplerate)) return 1;
83 if (new_aubio_sink(sink_path, -1)) return 1;
84
85 s = new_aubio_sink(sink_path, 0);
86
87 // check setting wrong parameters fails
88 if (!aubio_sink_preset_samplerate(s, oversized_samplerate)) return 1;
89 if (!aubio_sink_preset_channels(s, oversized_channels)) return 1;
90 if (!aubio_sink_preset_channels(s, -1)) return 1;
91
92 // check setting valid parameters passes
93 if (aubio_sink_preset_samplerate(s, samplerate)) return 1;
94 if (aubio_sink_preset_channels(s, 1)) return 1;
95
96 // check writing a vector with valid length
97 vec = new_fvec(hop_size);
98 aubio_sink_do(s, vec, hop_size);
99 // check writing more than in the input
100 aubio_sink_do(s, vec, hop_size+1);
101 // check write 0 frames
102 aubio_sink_do(s, vec, 0);
103 del_fvec(vec);
104
105 // check writing an oversized vector
106 vec = new_fvec(oversized_hop_size);
107 aubio_sink_do(s, vec, oversized_hop_size);
108 del_fvec(vec);
109
110 // test delete without closing
111 del_aubio_sink(s);
112
113 s = new_aubio_sink(sink_path, 0);
114
115 // preset channels first
116 if (aubio_sink_preset_channels(s, channels)) return 1;
117 if (aubio_sink_preset_samplerate(s, samplerate)) return 1;
118
119 if (aubio_sink_get_samplerate(s) != samplerate) return 1;
120 if (aubio_sink_get_channels(s) != channels) return 1;
121
122 mat = new_fmat(channels, hop_size);
123 // check writing a vector with valid length
124 aubio_sink_do_multi(s, mat, hop_size);
125 // check writing 0 frames
126 aubio_sink_do_multi(s, mat, 0);
127 // check writing more than in the input
128 aubio_sink_do_multi(s, mat, hop_size+1);
129 del_fmat(mat);
130
131 // check writing oversized input
132 mat = new_fmat(channels, oversized_hop_size);
133 aubio_sink_do_multi(s, mat, oversized_hop_size);
134 del_fmat(mat);
135
136 // check writing undersized input
137 mat = new_fmat(channels - 1, hop_size);
138 aubio_sink_do_multi(s, mat, hop_size);
139 del_fmat(mat);
140
141 aubio_sink_close(s);
142 // test closing twice
143 aubio_sink_close(s);
144
145 del_aubio_sink(s);
146
147 // delete temp file
148 close_temp_sink(sink_path, fd);
149
150 // shouldn't crash on null
151 del_aubio_sink(NULL);
152
153 return run_on_default_source_and_sink(main);
154 }
155