source: tests/src/io/test-sink.c @ 8797138

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

[tests] test del_aubio_sink on null

  • Property mode set to 100644
File size: 4.0 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  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
53failure:
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
64int 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}
Note: See TracBrowser for help on using the repository browser.