diff options
Diffstat (limited to 'tests/utils.c')
-rw-r--r-- | tests/utils.c | 33 |
1 files changed, 25 insertions, 8 deletions
diff --git a/tests/utils.c b/tests/utils.c index 5310a114f5..2fdc491f49 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); |