1 // this should be included *after* custom functions have been defined
3 #ifndef aubio_sink_custom
4 #define aubio_sink_custom "undefined"
5 #endif /* aubio_sink_custom */
7 #ifdef HAVE_AUBIO_SINK_CUSTOM
8 int test_wrong_params(void);
10 int base_main(int argc, char **argv)
13 if (argc < 3 || argc >= 6) {
14 PRINT_ERR("wrong number of arguments, running tests\n");
15 err = test_wrong_params();
16 PRINT_MSG("usage: %s <input_path> <output_path> [samplerate] [hop_size]\n",
21 uint_t samplerate = 0;
22 uint_t hop_size = 512;
23 uint_t n_frames = 0, read = 0;
25 char_t *source_path = argv[1];
26 char_t *sink_path = argv[2];
28 if ( argc >= 4 ) samplerate = atoi(argv[3]);
29 if ( argc >= 5 ) hop_size = atoi(argv[4]);
31 fvec_t *vec = new_fvec(hop_size);
33 aubio_source_t *i = new_aubio_source(source_path, samplerate, hop_size);
34 if (samplerate == 0 ) samplerate = aubio_source_get_samplerate(i);
36 aubio_sink_custom_t *o = new_aubio_sink_custom(sink_path, samplerate);
38 if (!vec || !i || !o) { err = 1; goto failure; }
41 aubio_source_do(i, vec, &read);
42 aubio_sink_custom_do(o, vec, read);
44 } while ( read == hop_size );
46 PRINT_MSG("%d frames at %dHz (%d blocks) read from %s, wrote to %s\n",
47 n_frames, samplerate, n_frames / hop_size,
48 source_path, sink_path);
50 // close sink now (optional)
51 aubio_sink_custom_close(o);
55 del_aubio_sink_custom(o);
64 int test_wrong_params(void)
68 aubio_sink_custom_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;
75 uint_t oversized_channels = 1025;
77 int fd = create_temp_sink(sink_path);
81 if (new_aubio_sink_custom( 0, samplerate)) return 1;
82 if (new_aubio_sink_custom("\0", samplerate)) return 1;
83 if (new_aubio_sink_custom(sink_path, -1)) return 1;
85 s = new_aubio_sink_custom(sink_path, 0);
87 // check setting wrong parameters fails
88 if (!aubio_sink_custom_preset_samplerate(s, oversized_samplerate)) return 1;
89 if (!aubio_sink_custom_preset_channels(s, oversized_channels)) return 1;
90 if (!aubio_sink_custom_preset_channels(s, -1)) return 1;
92 // check setting valid parameters passes
93 if (aubio_sink_custom_preset_samplerate(s, samplerate)) return 1;
94 if (aubio_sink_custom_preset_channels(s, 1)) return 1;
96 // check writing a vector with valid length
97 vec = new_fvec(hop_size);
98 aubio_sink_custom_do(s, vec, hop_size);
99 // check writing more than in the input
100 aubio_sink_custom_do(s, vec, hop_size+1);
101 // check write 0 frames
102 aubio_sink_custom_do(s, vec, 0);
105 // check writing an oversized vector
106 vec = new_fvec(oversized_hop_size);
107 aubio_sink_custom_do(s, vec, oversized_hop_size);
110 // test delete without closing
111 del_aubio_sink_custom(s);
113 s = new_aubio_sink_custom(sink_path, 0);
115 // preset channels first
116 if (aubio_sink_custom_preset_channels(s, channels)) return 1;
117 if (aubio_sink_custom_preset_samplerate(s, samplerate)) return 1;
119 if (aubio_sink_custom_get_samplerate(s) != samplerate) return 1;
120 if (aubio_sink_custom_get_channels(s) != channels) return 1;
122 mat = new_fmat(channels, hop_size);
123 // check writing a vector with valid length
124 aubio_sink_custom_do_multi(s, mat, hop_size);
125 // check writing 0 frames
126 aubio_sink_custom_do_multi(s, mat, 0);
127 // check writing more than in the input
128 aubio_sink_custom_do_multi(s, mat, hop_size+1);
131 // check writing oversized input
132 mat = new_fmat(channels, oversized_hop_size);
133 aubio_sink_custom_do_multi(s, mat, oversized_hop_size);
136 // check writing undersized input
137 mat = new_fmat(channels - 1, hop_size);
138 aubio_sink_custom_do_multi(s, mat, hop_size);
141 aubio_sink_custom_close(s);
142 // test closing twice
143 aubio_sink_custom_close(s);
145 del_aubio_sink_custom(s);
148 close_temp_sink(sink_path, fd);
150 return run_on_default_source_and_sink(base_main);
153 #else /* HAVE_AUBIO_SINK_CUSTOM */
155 int base_main(int argc, char** argv)
157 PRINT_ERR("aubio was not compiled with aubio_sink_"
158 aubio_sink_custom ", failed running %s with %d args\n",
163 #endif /* HAVE_AUBIO_SINK_CUSTOM */