]> git.sesse.net Git - ffmpeg/blob - tests/rotozoom.c
Merge commit 'e744281c49496b0e0a357e9f84c37fbf99215e20'
[ffmpeg] / tests / rotozoom.c
1 /*
2  * Generate a synthetic YUV video sequence suitable for codec testing.
3  *
4  * copyright (c) Sebastien Bechet <s.bechet@av7.net>
5  *
6  * This file is part of FFmpeg.
7  *
8  * FFmpeg is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * FFmpeg is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with FFmpeg; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22
23 #include <stdlib.h>
24 #include <stdio.h>
25 #include <inttypes.h>
26
27 #include "utils.c"
28
29 #define FIXP (1 << 16)
30 #define MY_PI 205887 // (M_PI * FIX)
31
32 static int64_t int_pow(int64_t a, int p)
33 {
34     int64_t v = FIXP;
35
36     for (; p; p--) {
37         v *= a;
38         v /= FIXP;
39     }
40
41     return v;
42 }
43
44 static int64_t int_sin(int64_t a)
45 {
46     if (a < 0)
47         a = MY_PI - a;  // 0..inf
48     a %= 2 * MY_PI;     // 0..2PI
49
50     if (a >= MY_PI * 3 / 2)
51         a -= 2 * MY_PI; // -PI / 2 .. 3PI / 2
52     if (a >= MY_PI / 2)
53         a = MY_PI - a;  // -PI / 2 ..  PI / 2
54
55     return a - int_pow(a, 3) / 6 + int_pow(a, 5) / 120 - int_pow(a, 7) / 5040;
56 }
57
58 static unsigned char tab_r[256 * 256];
59 static unsigned char tab_g[256 * 256];
60 static unsigned char tab_b[256 * 256];
61
62 static int h_cos[360];
63 static int h_sin[360];
64
65 static int ipol(uint8_t *src, int x, int y)
66 {
67     int int_x  = x >> 16;
68     int int_y  = y >> 16;
69     int frac_x = x & 0xFFFF;
70     int frac_y = y & 0xFFFF;
71     int s00    = src[( int_x      & 255) + 256 * ( int_y      & 255)];
72     int s01    = src[((int_x + 1) & 255) + 256 * ( int_y      & 255)];
73     int s10    = src[( int_x      & 255) + 256 * ((int_y + 1) & 255)];
74     int s11    = src[((int_x + 1) & 255) + 256 * ((int_y + 1) & 255)];
75     int s0     = (((1 << 16) - frac_x) * s00 + frac_x * s01) >> 8;
76     int s1     = (((1 << 16) - frac_x) * s10 + frac_x * s11) >> 8;
77
78     return (((1 << 16) - frac_y) * s0 + frac_y * s1) >> 24;
79 }
80
81 static void gen_image(int num, int w, int h)
82 {
83     const int c = h_cos[num % 360];
84     const int s = h_sin[num % 360];
85
86     const int xi = -(w / 2) * c;
87     const int yi =  (w / 2) * s;
88
89     const int xj = -(h / 2) * s;
90     const int yj = -(h / 2) * c;
91     int i, j;
92
93     int x, y;
94     int xprime = xj;
95     int yprime = yj;
96
97     for (j = 0; j < h; j++) {
98         x       = xprime + xi + FIXP * w / 2;
99         xprime += s;
100
101         y       = yprime + yi + FIXP * h / 2;
102         yprime += c;
103
104         for (i = 0; i < w; i++) {
105             x += c;
106             y -= s;
107             put_pixel(i, j,
108                       ipol(tab_r, x, y),
109                       ipol(tab_g, x, y),
110                       ipol(tab_b, x, y));
111         }
112     }
113 }
114
115 #define W 256
116 #define H 256
117
118 static int init_demo(const char *filename)
119 {
120     int i, j;
121     int h;
122     int radian;
123     char line[3 * W];
124
125     FILE *input_file;
126
127     input_file = fopen(filename, "rb");
128     if (!input_file) {
129         perror(filename);
130         return 1;
131     }
132
133     if (fread(line, 1, 15, input_file) != 15)
134         return 1;
135     for (i = 0; i < H; i++) {
136         if (fread(line, 1, 3 * W, input_file) != 3 * W)
137             return 1;
138         for (j = 0; j < W; j++) {
139             tab_r[W * i + j] = line[3 * j    ];
140             tab_g[W * i + j] = line[3 * j + 1];
141             tab_b[W * i + j] = line[3 * j + 2];
142         }
143     }
144     fclose(input_file);
145
146     /* tables sin/cos */
147     for (i = 0; i < 360; i++) {
148         radian   = 2 * i * MY_PI / 360;
149         h        = 2 * FIXP + int_sin(radian);
150         h_cos[i] = h * int_sin(radian + MY_PI / 2) / 2 / FIXP;
151         h_sin[i] = h * int_sin(radian)             / 2 / FIXP;
152     }
153
154     return 0;
155 }
156
157 int main(int argc, char **argv)
158 {
159     int w, h, i;
160     char buf[1024];
161     int isdir = 0;
162
163     if (argc != 3) {
164         printf("usage: %s image.pnm file|dir\n"
165                "generate a test video stream\n", argv[0]);
166         return 1;
167     }
168
169     if (!freopen(argv[2], "wb", stdout))
170         isdir = 1;
171
172     w = DEFAULT_WIDTH;
173     h = DEFAULT_HEIGHT;
174
175     rgb_tab = malloc(w * h * 3);
176     wrap    = w * 3;
177     width   = w;
178     height  = h;
179
180     if (init_demo(argv[1]))
181         return 1;
182
183     for (i = 0; i < DEFAULT_NB_PICT; i++) {
184         gen_image(i, w, h);
185         if (isdir) {
186             snprintf(buf, sizeof(buf), "%s%02d.pgm", argv[2], i);
187             pgmyuv_save(buf, w, h, rgb_tab);
188         } else {
189             pgmyuv_save(NULL, w, h, rgb_tab);
190         }
191     }
192
193     free(rgb_tab);
194     return 0;
195 }