videogen.c 4.54 KB
Newer Older
1
/*
2 3
 * Generate a synthetic YUV video sequence suitable for codec testing.
 * NOTE: No floats are used to guarantee bitexact output.
4 5 6
 *
 * Copyright (c) 2002 Fabrice Bellard
 *
7
 * This file is part of Libav.
8
 *
9
 * Libav is free software; you can redistribute it and/or
10 11 12 13
 * 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.
 *
14
 * Libav is distributed in the hope that it will be useful,
15 16 17 18 19
 * 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
20
 * License along with Libav; if not, write to the Free Software
21
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22
 */
23

24
#include <stdlib.h>
25
#include <stdint.h>
26 27
#include <stdio.h>

28
#include "utils.c"
29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52

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 */
53
static int int_cos(int a)
54 55 56 57 58 59 60 61
{
    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;
62
        a   = (FRAC_ONE / 2) - a;
63 64 65 66 67 68 69 70 71 72 73 74 75
    }
    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;

76
static VObj objs[NB_OBJS];
77

78
static unsigned int seed = 1;
79

80
static void gen_image(int num, int w, int h)
81 82 83 84 85
{
    int r, g, b, x, y, i, dx, dy, x1, y1;
    unsigned int seed1;

    if (num == 0) {
86
        for (i = 0; i < NB_OBJS; i++) {
87 88 89 90 91 92 93 94 95 96 97 98 99 100
            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;
101 102
    for (y = 0; y < h; y++) {
        for (x = 0; x < w; x++) {
103
            x1 = (x << FRAC_BITS) + dx;
104
            y1 = (y << FRAC_BITS) + dy;
105 106 107
            r  =       ((y1  * 7) >> FRAC_BITS) & 0xff;
            g  = (((x1 + y1) * 9) >> FRAC_BITS) & 0xff;
            b  =  ((x1       * 5) >> FRAC_BITS) & 0xff;
108 109 110 111 112 113
            put_pixel(x, y, r, g, b);
        }
    }

    /* then some noise with very high intensity to test saturation */
    seed1 = num;
114 115
    for (y = 0; y < NOISE_W; y++) {
        for (x = 0; x < NOISE_W; x++) {
116 117 118 119 120 121
            r = myrnd(&seed1, 256);
            g = myrnd(&seed1, 256);
            b = myrnd(&seed1, 256);
            put_pixel(x + NOISE_X, y + NOISE_Y, r, g, b);
        }
    }
122

123
    /* then moving objects */
124
    for (i = 0; i < NB_OBJS; i++) {
125 126
        VObj *p = &objs[i];
        seed1 = i;
127 128
        for (y = 0; y < p->h; y++) {
            for (x = 0; x < p->w; x++) {
129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147
                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];
148
    int isdir = 0;
149

150 151
    if (argc != 2) {
        printf("usage: %s file|dir\n"
152 153 154 155
               "generate a test video stream\n", argv[0]);
        exit(1);
    }

156 157
    if (!freopen(argv[1], "wb", stdout))
        isdir = 1;
158

159 160 161 162
    w = DEFAULT_WIDTH;
    h = DEFAULT_HEIGHT;

    rgb_tab = malloc(w * h * 3);
163 164 165
    wrap    = w * 3;
    width   = w;
    height  = h;
166

167
    for (i = 0; i < DEFAULT_NB_PICT; i++) {
168
        gen_image(i, w, h);
169
        if (isdir) {
170 171 172 173 174
            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);
        }
175
    }
176

177 178 179
    free(rgb_tab);
    return 0;
}