mp3dec: adjust IMDCT scaling to avoid overflows

[ffmpeg] / tests / videogen.c

diff --git a/tests/videogen.c b/tests/videogen.c
index f151f3edf84adc3964c3b269b315eb5b1966dae9..94e6e7759830c270841bf10264168c96421459e1 100644 (file)
--- a/tests/videogen.c
+++ b/tests/videogen.c
@@ -145,13 +145,17 @@ int main(int argc, char **argv)
 {
     int w, h, i;
     char buf[1024];
+    int isdir = 0;
 
     if (argc != 2) {
-        printf("usage: %s file\n"
+        printf("usage: %s file|dir\n"
                "generate a test video stream\n", argv[0]);
         exit(1);
     }
 
+    if (!freopen(argv[1], "wb", stdout))
+        isdir = 1;
+
     w = DEFAULT_WIDTH;
     h = DEFAULT_HEIGHT;
 
@@ -161,9 +165,13 @@ int main(int argc, char **argv)
     height  = h;
 
     for (i = 0; i < DEFAULT_NB_PICT; i++) {
-        snprintf(buf, sizeof(buf), "%s%02d.pgm", argv[1], i);
         gen_image(i, w, h);
-        pgmyuv_save(buf, w, h, rgb_tab);
+        if (isdir) {
+            snprintf(buf, sizeof(buf), "%s%02d.pgm", argv[1], i);
+            pgmyuv_save(buf, w, h, rgb_tab);
+        } else {
+            pgmyuv_save(NULL, w, h, rgb_tab);
+        }
     }
 
     free(rgb_tab);