diff options
Diffstat (limited to 'ffmpeg/tests/videogen.c')
| -rw-r--r-- | ffmpeg/tests/videogen.c | 179 |
1 files changed, 179 insertions, 0 deletions
diff --git a/ffmpeg/tests/videogen.c b/ffmpeg/tests/videogen.c new file mode 100644 index 0000000..94e6e77 --- /dev/null +++ b/ffmpeg/tests/videogen.c @@ -0,0 +1,179 @@ +/* + * Generate a synthetic YUV video sequence suitable for codec testing. + * NOTE: No floats are used to guarantee bitexact output. + * + * Copyright (c) 2002 Fabrice Bellard + * + * This file is part of FFmpeg. + * + * FFmpeg is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * FFmpeg is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with FFmpeg; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +#include <stdlib.h> +#include <stdint.h> +#include <stdio.h> + +#include "utils.c" + +static unsigned int myrnd(unsigned int *seed_ptr, int n) +{ + unsigned int seed, val; + + seed = *seed_ptr; + seed = (seed * 314159) + 1; + if (n == 256) { + val = seed >> 24; + } else { + val = seed % n; + } + *seed_ptr = seed; + return val; +} + +#define NOISE_X 10 +#define NOISE_Y 30 +#define NOISE_W 26 + +#define FRAC_BITS 8 +#define FRAC_ONE (1 << FRAC_BITS) + +/* cosine approximate with 1-x^2 */ +static int int_cos(int a) +{ + int v, neg; + a = a & (FRAC_ONE - 1); + if (a >= (FRAC_ONE / 2)) + a = FRAC_ONE - a; + neg = 0; + if (a > (FRAC_ONE / 4)) { + neg = -1; + a = (FRAC_ONE / 2) - a; + } + v = FRAC_ONE - ((a * a) >> 4); + v = (v ^ neg) - neg; + return v; +} + +#define NB_OBJS 10 + +typedef struct VObj { + int x, y, w, h; + int r, g, b; +} VObj; + +static VObj objs[NB_OBJS]; + +static unsigned int seed = 1; + +static void gen_image(int num, int w, int h) +{ + int r, g, b, x, y, i, dx, dy, x1, y1; + unsigned int seed1; + + if (num == 0) { + for (i = 0; i < NB_OBJS; i++) { + objs[i].x = myrnd(&seed, w); + objs[i].y = myrnd(&seed, h); + objs[i].w = myrnd(&seed, w / 4) + 10; + objs[i].h = myrnd(&seed, h / 4) + 10; + objs[i].r = myrnd(&seed, 256); + objs[i].g = myrnd(&seed, 256); + objs[i].b = myrnd(&seed, 256); + } + } + + /* first a moving background with gradients */ + /* test motion estimation */ + dx = int_cos(num * FRAC_ONE / 50) * 35; + dy = int_cos(num * FRAC_ONE / 50 + FRAC_ONE / 10) * 30; + for (y = 0; y < h; y++) { + for (x = 0; x < w; x++) { + x1 = (x << FRAC_BITS) + dx; + y1 = (y << FRAC_BITS) + dy; + r = ((y1 * 7) >> FRAC_BITS) & 0xff; + g = (((x1 + y1) * 9) >> FRAC_BITS) & 0xff; + b = ((x1 * 5) >> FRAC_BITS) & 0xff; + put_pixel(x, y, r, g, b); + } + } + + /* then some noise with very high intensity to test saturation */ + seed1 = num; + for (y = 0; y < NOISE_W; y++) { + for (x = 0; x < NOISE_W; x++) { + r = myrnd(&seed1, 256); + g = myrnd(&seed1, 256); + b = myrnd(&seed1, 256); + put_pixel(x + NOISE_X, y + NOISE_Y, r, g, b); + } + } + + /* then moving objects */ + for (i = 0; i < NB_OBJS; i++) { + VObj *p = &objs[i]; + seed1 = i; + for (y = 0; y < p->h; y++) { + for (x = 0; x < p->w; x++) { + r = p->r; + g = p->g; + b = p->b; + /* add a per object noise */ + r += myrnd(&seed1, 50); + g += myrnd(&seed1, 50); + b += myrnd(&seed1, 50); + put_pixel(x + p->x, y + p->y, r, g, b); + } + } + p->x += myrnd(&seed, 21) - 10; + p->y += myrnd(&seed, 21) - 10; + } +} + +int main(int argc, char **argv) +{ + int w, h, i; + char buf[1024]; + int isdir = 0; + + if (argc != 2) { + 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; + + rgb_tab = malloc(w * h * 3); + wrap = w * 3; + width = w; + height = h; + + for (i = 0; i < DEFAULT_NB_PICT; i++) { + gen_image(i, w, h); + 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); + return 0; +} |