From: Paul Brossier <piem@piem.org>
Date: Sun, 23 Feb 2014 02:13:26 +0000 (-0300)
Subject: tests/src/io/test-sink*: improve, unify
X-Git-Tag: 0.4.1~39
X-Git-Url: https://git.aubio.org/?a=commitdiff_plain;h=9f68f113664ff064949992ff605896be3857986a;p=aubio.git

tests/src/io/test-sink*: improve, unify
---

diff --git a/tests/src/io/test-sink.c b/tests/src/io/test-sink.c
index f5220f49..2f04177b 100644
--- a/tests/src/io/test-sink.c
+++ b/tests/src/io/test-sink.c
@@ -12,11 +12,13 @@ int main (int argc, char **argv)
     return err;
   }
 
+  uint_t samplerate = 0;
+  uint_t hop_size = 512;
+  uint_t n_frames = 0, read = 0;
+
   char_t *source_path = argv[1];
   char_t *sink_path = argv[2];
 
-  uint_t samplerate = 0;
-  uint_t hop_size = 256;
   if ( argc >= 4 ) samplerate = atoi(argv[3]);
   if ( argc >= 5 ) hop_size = atoi(argv[4]);
   if ( argc >= 6 ) {
@@ -36,8 +38,6 @@ int main (int argc, char **argv)
   aubio_sink_t *o = new_aubio_sink(sink_path, samplerate);
   if (!o) { err = 1; goto beach_sink; }
 
-  uint_t n_frames = 0, read = 0;
-
   do {
     aubio_source_do(i, vec, &read);
     aubio_sink_do(o, vec, read);
diff --git a/tests/src/io/test-sink_apple_audio.c b/tests/src/io/test-sink_apple_audio.c
index ebafbd95..265efd28 100644
--- a/tests/src/io/test-sink_apple_audio.c
+++ b/tests/src/io/test-sink_apple_audio.c
@@ -12,7 +12,7 @@ int main (int argc, char **argv)
   if (argc < 3) {
     err = 2;
     PRINT_ERR("not enough arguments\n");
-    PRINT_MSG("usage: %s <input_path> <output_path> [samplerate]\n", argv[0]);
+    PRINT_MSG("usage: %s <input_path> <output_path> [samplerate] [hop_size]\n", argv[0]);
     return err;
   }
 
@@ -23,14 +23,25 @@ int main (int argc, char **argv)
 
   char_t *source_path = argv[1];
   char_t *sink_path = argv[2];
-  if ( argc == 4 ) samplerate = atoi(argv[3]);
+
+  if ( argc >= 4 ) samplerate = atoi(argv[3]);
+  if ( argc >= 5 ) hop_size = atoi(argv[4]);
+  if ( argc >= 6 ) {
+    err = 2;
+    PRINT_ERR("too many arguments\n");
+    return err;
+  }
 
   fvec_t *vec = new_fvec(hop_size);
+  if (!vec) { err = 1; goto beach_fvec; }
+
   aubio_source_t *i = new_aubio_source(source_path, samplerate, hop_size);
+  if (!i) { err = 1; goto beach_source; }
+
   if (samplerate == 0 ) samplerate = aubio_source_get_samplerate(i);
-  aubio_sink_apple_audio_t *o = new_aubio_sink_apple_audio(sink_path, samplerate);
 
-  if (!i || !o) { err = 1; goto beach; }
+  aubio_sink_apple_audio_t *o = new_aubio_sink_apple_audio(sink_path, samplerate);
+  if (!o) { err = 1; goto beach_sink; }
 
   do {
     aubio_source_do(i, vec, &read);
@@ -38,13 +49,16 @@ int main (int argc, char **argv)
     n_frames += read;
   } while ( read == hop_size );
 
-  PRINT_MSG("%d frames read from %s\n written to %s at %dHz\n",
-      n_frames, source_path, sink_path, samplerate);
+  PRINT_MSG("read %d frames at %dHz (%d blocks) from %s written to %s\n",
+      n_frames, samplerate, n_frames / hop_size,
+      source_path, sink_path);
 
-beach:
-  del_aubio_source(i);
   del_aubio_sink_apple_audio(o);
+beach_sink:
+  del_aubio_source(i);
+beach_source:
   del_fvec(vec);
+beach_fvec:
 #else
   err = 3;
   PRINT_ERR("aubio was not compiled with aubio_source_apple_audio\n");
diff --git a/tests/src/io/test-sink_sndfile.c b/tests/src/io/test-sink_sndfile.c
index bd52e566..78122296 100644
--- a/tests/src/io/test-sink_sndfile.c
+++ b/tests/src/io/test-sink_sndfile.c
@@ -13,7 +13,7 @@ int main (int argc, char **argv)
   if (argc < 3) {
     err = 2;
     PRINT_ERR("not enough arguments\n");
-    PRINT_MSG("usage: %s <input_path> <output_path> [samplerate]\n", argv[0]);
+    PRINT_MSG("usage: %s <input_path> <output_path> [samplerate] [hop_size]\n", argv[0]);
     return err;
   }
 
@@ -24,28 +24,42 @@ int main (int argc, char **argv)
 
   char_t *source_path = argv[1];
   char_t *sink_path = argv[2];
-  if ( argc == 4 ) samplerate = atoi(argv[3]);
+
+  if ( argc >= 4 ) samplerate = atoi(argv[3]);
+  if ( argc >= 5 ) hop_size = atoi(argv[4]);
+  if ( argc >= 6 ) {
+    err = 2;
+    PRINT_ERR("too many arguments\n");
+    return err;
+  }
 
   fvec_t *vec = new_fvec(hop_size);
-  aubio_source_sndfile_t * i = new_aubio_source_sndfile(source_path, samplerate, hop_size);
-  if (samplerate == 0 ) samplerate = aubio_source_sndfile_get_samplerate(i);
-  aubio_sink_sndfile_t *   o = new_aubio_sink_sndfile(sink_path, samplerate);
+  if (!vec) { err = 1; goto beach_fvec; }
+
+  aubio_source_t *i = new_aubio_source(source_path, samplerate, hop_size);
+  if (!i) { err = 1; goto beach_source; }
+
+  if (samplerate == 0 ) samplerate = aubio_source_get_samplerate(i);
 
-  if (!i || !o) { err = 1; goto beach; }
+  aubio_sink_sndfile_t *o = new_aubio_sink_sndfile(sink_path, samplerate);
+  if (!o) { err = 1; goto beach_sink; }
 
   do {
-    aubio_source_sndfile_do(i, vec, &read);
+    aubio_source_do(i, vec, &read);
     aubio_sink_sndfile_do(o, vec, read);
     n_frames += read;
   } while ( read == hop_size );
 
-  PRINT_MSG("%d frames read from %s\n written to %s at %dHz\n",
-      n_frames, source_path, sink_path, samplerate);
+  PRINT_MSG("read %d frames at %dHz (%d blocks) from %s written to %s\n",
+      n_frames, samplerate, n_frames / hop_size,
+      source_path, sink_path);
 
-beach:
-  del_aubio_source_sndfile(i);
   del_aubio_sink_sndfile(o);
+beach_sink:
+  del_aubio_source(i);
+beach_source:
   del_fvec(vec);
+beach_fvec:
 #else
   err = 3;
   PRINT_ERR("aubio was not compiled with aubio_source_sndfile\n");
diff --git a/tests/src/io/test-sink_wavwrite.c b/tests/src/io/test-sink_wavwrite.c
index 0d2994f5..2e758e34 100644
--- a/tests/src/io/test-sink_wavwrite.c
+++ b/tests/src/io/test-sink_wavwrite.c
@@ -12,7 +12,7 @@ int main (int argc, char **argv)
   if (argc < 3) {
     err = 2;
     PRINT_ERR("not enough arguments\n");
-    PRINT_MSG("usage: %s <input_path> <output_path> [samplerate]\n", argv[0]);
+    PRINT_MSG("usage: %s <input_path> <output_path> [samplerate] [hop_size]\n", argv[0]);
     return err;
   }
 
@@ -23,14 +23,25 @@ int main (int argc, char **argv)
 
   char_t *source_path = argv[1];
   char_t *sink_path = argv[2];
-  if ( argc == 4 ) samplerate = atoi(argv[3]);
+
+  if ( argc >= 4 ) samplerate = atoi(argv[3]);
+  if ( argc >= 5 ) hop_size = atoi(argv[4]);
+  if ( argc >= 6 ) {
+    err = 2;
+    PRINT_ERR("too many arguments\n");
+    return err;
+  }
 
   fvec_t *vec = new_fvec(hop_size);
+  if (!vec) { err = 1; goto beach_fvec; }
+
   aubio_source_t *i = new_aubio_source(source_path, samplerate, hop_size);
+  if (!i) { err = 1; goto beach_source; }
+
   if (samplerate == 0 ) samplerate = aubio_source_get_samplerate(i);
-  aubio_sink_wavwrite_t *o = new_aubio_sink_wavwrite(sink_path, samplerate);
 
-  if (!i || !o) { err = 1; goto beach; }
+  aubio_sink_wavwrite_t *o = new_aubio_sink_wavwrite(sink_path, samplerate);
+  if (!o) { err = 1; goto beach_sink; }
 
   do {
     aubio_source_do(i, vec, &read);
@@ -38,13 +49,16 @@ int main (int argc, char **argv)
     n_frames += read;
   } while ( read == hop_size );
 
-  PRINT_MSG("%d frames read from %s\n written to %s at %dHz\n",
-      n_frames, source_path, sink_path, samplerate);
+  PRINT_MSG("read %d frames at %dHz (%d blocks) from %s written to %s\n",
+      n_frames, samplerate, n_frames / hop_size,
+      source_path, sink_path);
 
-beach:
-  del_aubio_source(i);
   del_aubio_sink_wavwrite(o);
+beach_sink:
+  del_aubio_source(i);
+beach_source:
   del_fvec(vec);
+beach_fvec:
 #else
   err = 3;
   PRINT_ERR("aubio was not compiled with aubio_sink_wavwrite\n");