qt-faststart: dont allocate a bigger buffer than needed
[ffmpeg.git] / tools / yuvcmp.c
1 /*
2  * originally by Andreas Ă–man (andoma)
3  * some changes by Alexander Strange
4  */
5
6 #include <string.h>
7 #include <stdlib.h>
8 #include <inttypes.h>
9 #include <stdio.h>
10 #include <sys/stat.h>
11 #include <fcntl.h>
12 #include <unistd.h>
13
14
15 int
16 main(int argc, char **argv)
17 {
18     int fd[2];
19     int print_pixels = 0;
20     int dump_blocks = 0;
21
22     int width;
23     int height;
24     int to_skip = 0;
25
26     if (argc < 6) {
27         fprintf(stderr, "%s [YUV file 1] [YUV file 2] width height pixelcmp|blockdump (# to skip)\n", argv[0]);
28         return 1;
29     }
30
31     width  = atoi(argv[3]);
32     height = atoi(argv[4]);
33     if (argc > 6)
34         to_skip = atoi(argv[6]);
35
36     uint8_t *Y[2], *C[2][2];
37     int i, v, c, p;
38     int lsiz = width * height;
39     int csiz = width * height / 4;
40     int x, y;
41     int cwidth = width / 2;
42     int fr = to_skip;
43     int mb;
44     char *mberrors;
45     int mb_x, mb_y;
46     uint8_t *a;
47     uint8_t *b;
48     int die = 0;
49
50     print_pixels = strstr(argv[5], "pixelcmp") ? 1 : 0;
51     dump_blocks  = strstr(argv[5], "blockdump") ? 1 : 0;
52
53     for(i = 0; i < 2; i++) {
54         Y[i] = malloc(lsiz);
55         C[0][i] = malloc(csiz);
56         C[1][i] = malloc(csiz);
57
58         fd[i] = open(argv[1 + i], O_RDONLY);
59         if(fd[i] == -1) {
60             perror("open");
61             exit(1);
62         }
63         fcntl(fd[i], F_NOCACHE, 1);
64
65         if (to_skip)
66             lseek(fd[i], to_skip * (lsiz + 2*csiz), SEEK_SET);
67     }
68
69     mb_x = width / 16;
70     mb_y = height / 16;
71
72     mberrors = malloc(mb_x * mb_y);
73
74     while(!die) {
75         memset(mberrors, 0, mb_x * mb_y);
76
77         printf("Loading frame %d\n", ++fr);
78
79         for(i = 0; i < 2; i++) {
80             v = read(fd[i], Y[i], lsiz);
81             if(v != lsiz) {
82                 fprintf(stderr, "Unable to read Y from file %d, exiting\n", i + 1);
83                 return 1;
84             }
85         }
86
87
88         for(c = 0; c < lsiz; c++) {
89             if(Y[0][c] != Y[1][c]) {
90                 x = c % width;
91                 y = c / width;
92
93                 mb = x / 16 + (y / 16) * mb_x;
94
95                 if(print_pixels)
96                     printf("Luma diff 0x%02x != 0x%02x at pixel (%4d,%-4d) mb(%d,%d) #%d\n",
97                            Y[0][c],
98                            Y[1][c],
99                            x, y,
100                            x / 16,
101                            y / 16,
102                            mb);
103
104                 mberrors[mb] |= 1;
105             }
106         }
107
108         /* Chroma planes */
109
110         for(p = 0; p < 2; p++) {
111
112             for(i = 0; i < 2; i++) {
113                 v = read(fd[i], C[p][i], csiz);
114                 if(v != csiz) {
115                     fprintf(stderr, "Unable to read %c from file %d, exiting\n",
116                             "UV"[p], i + 1);
117                     return 1;
118                 }
119             }
120
121             for(c = 0; c < csiz; c++) {
122                 if(C[p][0][c] != C[p][1][c]) {
123                     x = c % cwidth;
124                     y = c / cwidth;
125
126                     mb = x / 8 + (y / 8) * mb_x;
127
128                     mberrors[mb] |= 2 << p;
129
130                     if(print_pixels)
131
132                         printf("c%c diff 0x%02x != 0x%02x at pixel (%4d,%-4d) "
133                                "mb(%3d,%-3d) #%d\n",
134                                p ? 'r' : 'b',
135                                C[p][0][c],
136                                C[p][1][c],
137
138                                x, y,
139                                x / 8,
140                                y / 8,
141                                x / 8 + y / 8 * cwidth / 8);
142                 }
143             }
144         }
145
146         for(i = 0; i < mb_x * mb_y; i++) {
147             x = i % mb_x;
148             y = i / mb_x;
149
150             if(mberrors[i]) {
151                 die = 1;
152
153                 printf("MB (%3d,%-3d) %4d %d %c%c%c damaged\n",
154                        x, y, i, mberrors[i],
155                        mberrors[i] & 1 ? 'Y' : ' ',
156                        mberrors[i] & 2 ? 'U' : ' ',
157                        mberrors[i] & 4 ? 'V' : ' ');
158
159                 if(dump_blocks) {
160                     a = Y[0] + x * 16 + y * 16 * width;
161                     b = Y[1] + x * 16 + y * 16 * width;
162
163                     for(y = 0; y < 16; y++) {
164                         printf("%c ", "TB"[y&1]);
165                         for(x = 0; x < 16; x++)
166                             printf("%02x%c", a[x + y * width],
167                                    a[x + y * width] != b[x + y * width] ? '<' : ' ');
168
169                         printf("| ");
170                         for(x = 0; x < 16; x++)
171                             printf("%02x%c", b[x + y * width],
172                                    a[x + y * width] != b[x + y * width] ? '<' : ' ');
173
174                         printf("\n");
175                     }
176                 }
177             }
178         }
179     }
180
181     return 0;
182 }