[FFmpeg-cvslog] fate: teach videogen/rotozoom to output a single raw video stream

[Mans Rullgard]

ffmpeg | branch: master | Mans Rullgard <mans at mansr.com> | Thu May 17 15:55:14 2012 +0100| [7e5880e0cb041d0f1362bddd75130471ffc811ce] | committer: Mans Rullgard

fate: teach videogen/rotozoom to output a single raw video stream

This makes videogen/rotozoom output a raw video stream on stdout
if no output directory is specified.

Signed-off-by: Mans Rullgard <mans at mansr.com>

> http://git.videolan.org/gitweb.cgi/ffmpeg.git/?a=commit;h=7e5880e0cb041d0f1362bddd75130471ffc811ce
---

 tests/Makefile   |    2 +-
 tests/rotozoom.c |   17 ++++++++++++-----
 tests/utils.c    |   33 +++++++++++++++++++++++++--------
 tests/videogen.c |   15 +++++++++++----
 4 files changed, 49 insertions(+), 18 deletions(-)

diff --git a/tests/Makefile b/tests/Makefile
index 79316b8..3509916 100644
--- a/tests/Makefile
+++ b/tests/Makefile
@@ -14,7 +14,7 @@ tests/vsynth1/00.pgm: tests/videogen$(HOSTEXESUF) | tests/vsynth1
 	$(M)./$< 'tests/vsynth1/'
 
 tests/vsynth2/00.pgm: tests/rotozoom$(HOSTEXESUF) | tests/vsynth2
-	$(M)./$< 'tests/vsynth2/' $(SRC_PATH)/tests/lena.pnm
+	$(M)./$< $(SRC_PATH)/tests/lena.pnm 'tests/vsynth2/'
 
 tests/data/asynth1.sw: tests/audiogen$(HOSTEXESUF) | tests/data
 	$(M)./$< $@
diff --git a/tests/rotozoom.c b/tests/rotozoom.c
index 48c06b0..683e070 100644
--- a/tests/rotozoom.c
+++ b/tests/rotozoom.c
@@ -159,12 +159,15 @@ int main(int argc, char **argv)
     int w, h, i;
     char buf[1024];
 
-    if (argc != 3) {
-        printf("usage: %s directory/ image.pnm\n"
+    if (argc > 3) {
+        printf("usage: %s image.pnm [directory/]\n"
                "generate a test video stream\n", argv[0]);
         return 1;
     }
 
+    if (argc < 3)
+        err_if(!freopen(NULL, "wb", stdout));
+
     w = DEFAULT_WIDTH;
     h = DEFAULT_HEIGHT;
 
@@ -173,13 +176,17 @@ int main(int argc, char **argv)
     width   = w;
     height  = h;
 
-    if (init_demo(argv[2]))
+    if (init_demo(argv[1]))
         return 1;
 
     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 (argc > 2) {
+            snprintf(buf, sizeof(buf), "%s%02d.pgm", argv[2], i);
+            pgmyuv_save(buf, w, h, rgb_tab);
+        } else {
+            pgmyuv_save(NULL, w, h, rgb_tab);
+        }
     }
 
     free(rgb_tab);
diff --git a/tests/utils.c b/tests/utils.c
index 5310a11..2fdc491 100644
--- a/tests/utils.c
+++ b/tests/utils.c
@@ -115,20 +115,37 @@ static void pgmyuv_save(const char *filename, int w, int h,
 
     rgb24_to_yuv420p(lum_tab, cb_tab, cr_tab, rgb_tab, w, h);
 
-    f = fopen(filename, "wb");
-    fprintf(f, "P5\n%d %d\n%d\n", w, h * 3 / 2, 255);
+    if (filename) {
+        f = fopen(filename, "wb");
+        fprintf(f, "P5\n%d %d\n%d\n", w, h * 3 / 2, 255);
+    } else {
+        f = stdout;
+    }
+
     err_if(fwrite(lum_tab, 1, w * h, f) != w * h);
     h2 = h / 2;
     w2 = w / 2;
     cb = cb_tab;
     cr = cr_tab;
-    for (i = 0; i < h2; i++) {
-        err_if(fwrite(cb, 1, w2, f) != w2);
-        err_if(fwrite(cr, 1, w2, f) != w2);
-        cb += w2;
-        cr += w2;
+
+    if (filename) {
+        for (i = 0; i < h2; i++) {
+            err_if(fwrite(cb, 1, w2, f) != w2);
+            err_if(fwrite(cr, 1, w2, f) != w2);
+            cb += w2;
+            cr += w2;
+        }
+        fclose(f);
+    } else {
+        for (i = 0; i < h2; i++) {
+            err_if(fwrite(cb, 1, w2, f) != w2);
+            cb += w2;
+        }
+        for (i = 0; i < h2; i++) {
+            err_if(fwrite(cr, 1, w2, f) != w2);
+            cr += w2;
+        }
     }
-    fclose(f);
 
     free(lum_tab);
     free(cb_tab);
diff --git a/tests/videogen.c b/tests/videogen.c
index 8c3d539..7228bd5 100644
--- a/tests/videogen.c
+++ b/tests/videogen.c
@@ -146,12 +146,15 @@ int main(int argc, char **argv)
     int w, h, i;
     char buf[1024];
 
-    if (argc != 2) {
-        printf("usage: %s file\n"
+    if (argc > 2) {
+        printf("usage: %s [file]\n"
                "generate a test video stream\n", argv[0]);
         exit(1);
     }
 
+    if (argc < 2)
+        err_if(!freopen(NULL, "wb", stdout));
+
     w = DEFAULT_WIDTH;
     h = DEFAULT_HEIGHT;
 
@@ -161,9 +164,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 (argc > 1) {
+            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);