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