yading@11: /*
yading@11:  * originally by Andreas Öman (andoma)
yading@11:  * some changes by Alexander Strange
yading@11:  */
yading@11: 
yading@11: #include <string.h>
yading@11: #include <stdlib.h>
yading@11: #include <inttypes.h>
yading@11: #include <stdio.h>
yading@11: #include <sys/stat.h>
yading@11: #include <fcntl.h>
yading@11: #include <unistd.h>
yading@11: 
yading@11: 
yading@11: int
yading@11: main(int argc, char **argv)
yading@11: {
yading@11:     int fd[2];
yading@11:     int print_pixels = 0;
yading@11:     int dump_blocks = 0;
yading@11: 
yading@11:     int width;
yading@11:     int height;
yading@11:     int to_skip = 0;
yading@11: 
yading@11:     if (argc < 6) {
yading@11:         fprintf(stderr, "%s [YUV file 1] [YUV file 2] width height pixelcmp|blockdump (# to skip)\n", argv[0]);
yading@11:         return 1;
yading@11:     }
yading@11: 
yading@11:     width  = atoi(argv[3]);
yading@11:     height = atoi(argv[4]);
yading@11:     if (argc > 6)
yading@11:         to_skip = atoi(argv[6]);
yading@11: 
yading@11:     uint8_t *Y[2], *C[2][2];
yading@11:     int i, v, c, p;
yading@11:     int lsiz = width * height;
yading@11:     int csiz = width * height / 4;
yading@11:     int x, y;
yading@11:     int cwidth = width / 2;
yading@11:     int fr = to_skip;
yading@11:     int mb;
yading@11:     char *mberrors;
yading@11:     int mb_x, mb_y;
yading@11:     uint8_t *a;
yading@11:     uint8_t *b;
yading@11:     int die = 0;
yading@11: 
yading@11:     print_pixels = strstr(argv[5], "pixelcmp") ? 1 : 0;
yading@11:     dump_blocks  = strstr(argv[5], "blockdump") ? 1 : 0;
yading@11: 
yading@11:     for(i = 0; i < 2; i++) {
yading@11:         Y[i] = malloc(lsiz);
yading@11:         C[0][i] = malloc(csiz);
yading@11:         C[1][i] = malloc(csiz);
yading@11: 
yading@11:         fd[i] = open(argv[1 + i], O_RDONLY);
yading@11:         if(fd[i] == -1) {
yading@11:             perror("open");
yading@11:             exit(1);
yading@11:         }
yading@11:         fcntl(fd[i], F_NOCACHE, 1);
yading@11: 
yading@11:         if (to_skip)
yading@11:             lseek(fd[i], to_skip * (lsiz + 2*csiz), SEEK_SET);
yading@11:     }
yading@11: 
yading@11:     mb_x = width / 16;
yading@11:     mb_y = height / 16;
yading@11: 
yading@11:     mberrors = malloc(mb_x * mb_y);
yading@11: 
yading@11:     while(!die) {
yading@11:         memset(mberrors, 0, mb_x * mb_y);
yading@11: 
yading@11:         printf("Loading frame %d\n", ++fr);
yading@11: 
yading@11:         for(i = 0; i < 2; i++) {
yading@11:             v = read(fd[i], Y[i], lsiz);
yading@11:             if(v != lsiz) {
yading@11:                 fprintf(stderr, "Unable to read Y from file %d, exiting\n", i + 1);
yading@11:                 return 1;
yading@11:             }
yading@11:         }
yading@11: 
yading@11: 
yading@11:         for(c = 0; c < lsiz; c++) {
yading@11:             if(Y[0][c] != Y[1][c]) {
yading@11:                 x = c % width;
yading@11:                 y = c / width;
yading@11: 
yading@11:                 mb = x / 16 + (y / 16) * mb_x;
yading@11: 
yading@11:                 if(print_pixels)
yading@11:                     printf("Luma diff 0x%02x != 0x%02x at pixel (%4d,%-4d) mb(%d,%d) #%d\n",
yading@11:                            Y[0][c],
yading@11:                            Y[1][c],
yading@11:                            x, y,
yading@11:                            x / 16,
yading@11:                            y / 16,
yading@11:                            mb);
yading@11: 
yading@11:                 mberrors[mb] |= 1;
yading@11:             }
yading@11:         }
yading@11: 
yading@11:         /* Chroma planes */
yading@11: 
yading@11:         for(p = 0; p < 2; p++) {
yading@11: 
yading@11:             for(i = 0; i < 2; i++) {
yading@11:                 v = read(fd[i], C[p][i], csiz);
yading@11:                 if(v != csiz) {
yading@11:                     fprintf(stderr, "Unable to read %c from file %d, exiting\n",
yading@11:                             "UV"[p], i + 1);
yading@11:                     return 1;
yading@11:                 }
yading@11:             }
yading@11: 
yading@11:             for(c = 0; c < csiz; c++) {
yading@11:                 if(C[p][0][c] != C[p][1][c]) {
yading@11:                     x = c % cwidth;
yading@11:                     y = c / cwidth;
yading@11: 
yading@11:                     mb = x / 8 + (y / 8) * mb_x;
yading@11: 
yading@11:                     mberrors[mb] |= 2 << p;
yading@11: 
yading@11:                     if(print_pixels)
yading@11: 
yading@11:                         printf("c%c diff 0x%02x != 0x%02x at pixel (%4d,%-4d) "
yading@11:                                "mb(%3d,%-3d) #%d\n",
yading@11:                                p ? 'r' : 'b',
yading@11:                                C[p][0][c],
yading@11:                                C[p][1][c],
yading@11: 
yading@11:                                x, y,
yading@11:                                x / 8,
yading@11:                                y / 8,
yading@11:                                x / 8 + y / 8 * cwidth / 8);
yading@11:                 }
yading@11:             }
yading@11:         }
yading@11: 
yading@11:         for(i = 0; i < mb_x * mb_y; i++) {
yading@11:             x = i % mb_x;
yading@11:             y = i / mb_x;
yading@11: 
yading@11:             if(mberrors[i]) {
yading@11:                 die = 1;
yading@11: 
yading@11:                 printf("MB (%3d,%-3d) %4d %d %c%c%c damaged\n",
yading@11:                        x, y, i, mberrors[i],
yading@11:                        mberrors[i] & 1 ? 'Y' : ' ',
yading@11:                        mberrors[i] & 2 ? 'U' : ' ',
yading@11:                        mberrors[i] & 4 ? 'V' : ' ');
yading@11: 
yading@11:                 if(dump_blocks) {
yading@11:                     a = Y[0] + x * 16 + y * 16 * width;
yading@11:                     b = Y[1] + x * 16 + y * 16 * width;
yading@11: 
yading@11:                     for(y = 0; y < 16; y++) {
yading@11:                         printf("%c ", "TB"[y&1]);
yading@11:                         for(x = 0; x < 16; x++)
yading@11:                             printf("%02x%c", a[x + y * width],
yading@11:                                    a[x + y * width] != b[x + y * width] ? '<' : ' ');
yading@11: 
yading@11:                         printf("| ");
yading@11:                         for(x = 0; x < 16; x++)
yading@11:                             printf("%02x%c", b[x + y * width],
yading@11:                                    a[x + y * width] != b[x + y * width] ? '<' : ' ');
yading@11: 
yading@11:                         printf("\n");
yading@11:                     }
yading@11:                 }
yading@11:             }
yading@11:         }
yading@11:     }
yading@11: 
yading@11:     return 0;
yading@11: }