21 #define VPX_CODEC_DISABLE_COMPAT 1
22 #include "vpx_config.h"
24 #include "vpx_ports/vpx_timer.h"
25 #if CONFIG_VP8_DECODER
29 #include "md5_utils.h"
31 #include "tools_common.h"
32 #include "nestegg/include/nestegg/nestegg.h"
37 #define snprintf _snprintf
38 #define isatty _isatty
39 #define fileno _fileno
49 static const char *exec_name;
51 #define VP8_FOURCC (0x00385056)
57 unsigned int fourcc_mask;
60 #if CONFIG_VP8_DECODER
61 {
"vp8", &vpx_codec_vp8_dx_algo, VP8_FOURCC, 0x00FFFFFF},
66 static const arg_def_t codecarg = ARG_DEF(NULL,
"codec", 1,
68 static const arg_def_t use_yv12 = ARG_DEF(NULL,
"yv12", 0,
69 "Output raw YV12 frames");
70 static const arg_def_t use_i420 = ARG_DEF(NULL,
"i420", 0,
71 "Output raw I420 frames");
72 static const arg_def_t flipuvarg = ARG_DEF(NULL,
"flipuv", 0,
73 "Flip the chroma planes in the output");
74 static const arg_def_t noblitarg = ARG_DEF(NULL,
"noblit", 0,
75 "Don't process the decoded frames");
76 static const arg_def_t progressarg = ARG_DEF(NULL,
"progress", 0,
77 "Show progress after each frame decodes");
78 static const arg_def_t limitarg = ARG_DEF(NULL,
"limit", 1,
79 "Stop decoding after n frames");
80 static const arg_def_t postprocarg = ARG_DEF(NULL,
"postproc", 0,
81 "Postprocess decoded frames");
82 static const arg_def_t summaryarg = ARG_DEF(NULL,
"summary", 0,
83 "Show timing summary");
84 static const arg_def_t outputfile = ARG_DEF(
"o",
"output", 1,
85 "Output file name pattern (see below)");
86 static const arg_def_t threadsarg = ARG_DEF(
"t",
"threads", 1,
87 "Max threads to use");
88 static const arg_def_t verbosearg = ARG_DEF(
"v",
"verbose", 0,
89 "Show version string");
90 static const arg_def_t error_concealment = ARG_DEF(NULL,
"error-concealment", 0,
91 "Enable decoder error-concealment");
95 static const arg_def_t md5arg = ARG_DEF(NULL,
"md5", 0,
96 "Compute the MD5 sum of the decoded frame");
98 static const arg_def_t *all_args[] =
100 &codecarg, &use_yv12, &use_i420, &flipuvarg, &noblitarg,
101 &progressarg, &limitarg, &postprocarg, &summaryarg, &outputfile,
102 &threadsarg, &verbosearg,
110 #if CONFIG_VP8_DECODER
111 static const arg_def_t addnoise_level = ARG_DEF(NULL,
"noise-level", 1,
112 "Enable VP8 postproc add noise");
113 static const arg_def_t deblock = ARG_DEF(NULL,
"deblock", 0,
114 "Enable VP8 deblocking");
115 static const arg_def_t demacroblock_level = ARG_DEF(NULL,
"demacroblock-level", 1,
116 "Enable VP8 demacroblocking, w/ level");
117 static const arg_def_t pp_debug_info = ARG_DEF(NULL,
"pp-debug-info", 1,
118 "Enable VP8 visible debug info");
119 static const arg_def_t pp_disp_ref_frame = ARG_DEF(NULL,
"pp-dbg-ref-frame", 1,
120 "Display only selected reference frame per macro block");
121 static const arg_def_t pp_disp_mb_modes = ARG_DEF(NULL,
"pp-dbg-mb-modes", 1,
122 "Display only selected macro block modes");
123 static const arg_def_t pp_disp_b_modes = ARG_DEF(NULL,
"pp-dbg-b-modes", 1,
124 "Display only selected block modes");
125 static const arg_def_t pp_disp_mvs = ARG_DEF(NULL,
"pp-dbg-mvs", 1,
126 "Draw only selected motion vectors");
127 static const arg_def_t mfqe = ARG_DEF(NULL,
"mfqe", 0,
128 "Enable multiframe quality enhancement");
130 static const arg_def_t *vp8_pp_args[] =
132 &addnoise_level, &deblock, &demacroblock_level, &pp_debug_info,
133 &pp_disp_ref_frame, &pp_disp_mb_modes, &pp_disp_b_modes, &pp_disp_mvs, &mfqe,
138 static void usage_exit()
142 fprintf(stderr,
"Usage: %s <options> filename\n\n"
143 "Options:\n", exec_name);
144 arg_show_usage(stderr, all_args);
145 #if CONFIG_VP8_DECODER
146 fprintf(stderr,
"\nVP8 Postprocessing Options:\n");
147 arg_show_usage(stderr, vp8_pp_args);
150 "\nOutput File Patterns:\n\n"
151 " The -o argument specifies the name of the file(s) to "
152 "write to. If the\n argument does not include any escape "
153 "characters, the output will be\n written to a single file. "
154 "Otherwise, the filename will be calculated by\n expanding "
155 "the following escape characters:\n");
157 "\n\t%%w - Frame width"
158 "\n\t%%h - Frame height"
159 "\n\t%%<n> - Frame number, zero padded to <n> places (1..9)"
160 "\n\n Pattern arguments are only supported in conjunction "
161 "with the --yv12 and\n --i420 options. If the -o option is "
162 "not specified, the output will be\n directed to stdout.\n"
164 fprintf(stderr,
"\nIncluded decoders:\n\n");
166 for (i = 0; i <
sizeof(ifaces) /
sizeof(ifaces[0]); i++)
167 fprintf(stderr,
" %-6s - %s\n",
174 void die(
const char *fmt, ...)
178 vfprintf(stderr, fmt, ap);
179 fprintf(stderr,
"\n");
183 static unsigned int mem_get_le16(
const void *vmem)
186 const unsigned char *mem = (
const unsigned char *)vmem;
193 static unsigned int mem_get_le32(
const void *vmem)
196 const unsigned char *mem = (
const unsigned char *)vmem;
216 nestegg *nestegg_ctx;
220 unsigned int video_track;
223 #define IVF_FRAME_HDR_SZ (sizeof(uint32_t) + sizeof(uint64_t))
224 #define RAW_FRAME_HDR_SZ (sizeof(uint32_t))
225 static int read_frame(
struct input_ctx *input,
228 size_t *buf_alloc_sz)
230 char raw_hdr[IVF_FRAME_HDR_SZ];
232 FILE *infile = input->infile;
233 enum file_kind kind = input->kind;
234 if(kind == WEBM_FILE)
236 if(input->chunk >= input->chunks)
244 nestegg_free_packet(input->pkt);
246 if(nestegg_read_packet(input->nestegg_ctx, &input->pkt) <= 0
247 || nestegg_packet_track(input->pkt, &track))
250 }
while(track != input->video_track);
252 if(nestegg_packet_count(input->pkt, &input->chunks))
257 if(nestegg_packet_data(input->pkt, input->chunk, buf, buf_sz))
267 else if (fread(raw_hdr, kind==IVF_FILE
268 ? IVF_FRAME_HDR_SZ : RAW_FRAME_HDR_SZ, 1, infile) != 1)
271 fprintf(stderr,
"Failed to read frame size\n");
277 new_buf_sz = mem_get_le32(raw_hdr);
279 if (new_buf_sz > 256 * 1024 * 1024)
281 fprintf(stderr,
"Error: Read invalid frame size (%u)\n",
282 (
unsigned int)new_buf_sz);
286 if (kind == RAW_FILE && new_buf_sz > 256 * 1024)
287 fprintf(stderr,
"Warning: Read invalid frame size (%u)"
288 " - not a raw file?\n", (
unsigned int)new_buf_sz);
290 if (new_buf_sz > *buf_alloc_sz)
292 uint8_t *new_buf = realloc(*buf, 2 * new_buf_sz);
297 *buf_alloc_sz = 2 * new_buf_sz;
301 fprintf(stderr,
"Failed to allocate compressed data buffer\n");
307 *buf_sz = new_buf_sz;
311 if (fread(*buf, 1, *buf_sz, infile) != *buf_sz)
313 fprintf(stderr,
"Failed to read full frame\n");
323 void *out_open(
const char *out_fn,
int do_md5)
330 MD5Context *md5_ctx = out = malloc(
sizeof(MD5Context));
337 FILE *outfile = out = strcmp(
"-", out_fn) ? fopen(out_fn,
"wb")
338 : set_binary_mode(stdout);
342 fprintf(stderr,
"Failed to output file");
350 void out_put(
void *out,
const uint8_t *buf,
unsigned int len,
int do_md5)
355 MD5Update(out, buf, len);
360 (void) fwrite(buf, 1, len, out);
364 void out_close(
void *out,
const char *out_fn,
int do_md5)
375 for (i = 0; i < 16; i++)
376 printf(
"%02x", md5[i]);
378 printf(
" %s\n", out_fn);
387 unsigned int file_is_ivf(FILE *infile,
388 unsigned int *fourcc,
390 unsigned int *height,
391 unsigned int *fps_den,
392 unsigned int *fps_num)
397 if (fread(raw_hdr, 1, 32, infile) == 32)
399 if (raw_hdr[0] ==
'D' && raw_hdr[1] ==
'K'
400 && raw_hdr[2] ==
'I' && raw_hdr[3] ==
'F')
404 if (mem_get_le16(raw_hdr + 4) != 0)
405 fprintf(stderr,
"Error: Unrecognized IVF version! This file may not"
406 " decode properly.");
408 *fourcc = mem_get_le32(raw_hdr + 8);
409 *width = mem_get_le16(raw_hdr + 12);
410 *height = mem_get_le16(raw_hdr + 14);
411 *fps_num = mem_get_le32(raw_hdr + 16);
412 *fps_den = mem_get_le32(raw_hdr + 20);
424 if(*fps_num&1)*fps_den<<=1;
445 unsigned int file_is_raw(FILE *infile,
446 unsigned int *fourcc,
448 unsigned int *height,
449 unsigned int *fps_den,
450 unsigned int *fps_num)
452 unsigned char buf[32];
458 if (fread(buf, 1, 32, infile) == 32)
462 if(mem_get_le32(buf) < 256 * 1024 * 1024)
463 for (i = 0; i <
sizeof(ifaces) /
sizeof(ifaces[0]); i++)
465 buf + 4, 32 - 4, &si))
468 *fourcc = ifaces[i].fourcc;
483 nestegg_read_cb(
void *buffer,
size_t length,
void *userdata)
487 if(fread(buffer, 1, length, f) < length)
499 nestegg_seek_cb(int64_t offset,
int whence,
void * userdata)
502 case NESTEGG_SEEK_SET: whence = SEEK_SET;
break;
503 case NESTEGG_SEEK_CUR: whence = SEEK_CUR;
break;
504 case NESTEGG_SEEK_END: whence = SEEK_END;
break;
506 return fseek(userdata, (
long)offset, whence)? -1 : 0;
511 nestegg_tell_cb(
void * userdata)
513 return ftell(userdata);
518 nestegg_log_cb(nestegg * context,
unsigned int severity,
char const * format,
523 va_start(ap, format);
524 vfprintf(stderr, format, ap);
525 fprintf(stderr,
"\n");
531 webm_guess_framerate(
struct input_ctx *input,
532 unsigned int *fps_den,
533 unsigned int *fps_num)
541 for(i=0; tstamp < 1000000000 && i < 50;)
543 nestegg_packet * pkt;
546 if(nestegg_read_packet(input->nestegg_ctx, &pkt) <= 0)
549 nestegg_packet_track(pkt, &track);
550 if(track == input->video_track)
552 nestegg_packet_tstamp(pkt, &tstamp);
556 nestegg_free_packet(pkt);
559 if(nestegg_track_seek(input->nestegg_ctx, input->video_track, 0))
562 *fps_num = (i - 1) * 1000000;
563 *fps_den = (
unsigned int)(tstamp / 1000);
566 nestegg_destroy(input->nestegg_ctx);
567 input->nestegg_ctx = NULL;
568 rewind(input->infile);
574 file_is_webm(
struct input_ctx *input,
575 unsigned int *fourcc,
577 unsigned int *height,
578 unsigned int *fps_den,
579 unsigned int *fps_num)
584 nestegg_io io = {nestegg_read_cb, nestegg_seek_cb, nestegg_tell_cb, 0};
585 nestegg_video_params params;
587 io.userdata = input->infile;
588 if(nestegg_init(&input->nestegg_ctx, io, NULL))
591 if(nestegg_track_count(input->nestegg_ctx, &n))
596 track_type = nestegg_track_type(input->nestegg_ctx, i);
598 if(track_type == NESTEGG_TRACK_VIDEO)
600 else if(track_type < 0)
604 if(nestegg_track_codec_id(input->nestegg_ctx, i) != NESTEGG_CODEC_VP8)
606 fprintf(stderr,
"Not VP8 video, quitting.\n");
610 input->video_track = i;
612 if(nestegg_track_video_params(input->nestegg_ctx, i, ¶ms))
617 *fourcc = VP8_FOURCC;
618 *width = params.width;
619 *height = params.height;
622 input->nestegg_ctx = NULL;
623 rewind(input->infile);
628 void show_progress(
int frame_in,
int frame_out,
unsigned long dx_time)
630 fprintf(stderr,
"%d decoded frames/%d showed frames in %lu us (%.2f fps)\r",
631 frame_in, frame_out, dx_time,
632 (
float)frame_out * 1000000.0 / (
float)dx_time);
636 void generate_filename(
const char *pattern,
char *out,
size_t q_len,
637 unsigned int d_w,
unsigned int d_h,
638 unsigned int frame_in)
640 const char *p = pattern;
645 char *next_pat = strchr(p,
'%');
655 case 'w': snprintf(q, q_len - 1,
"%d", d_w);
break;
656 case 'h': snprintf(q, q_len - 1,
"%d", d_h);
break;
657 case '1': snprintf(q, q_len - 1,
"%d", frame_in);
break;
658 case '2': snprintf(q, q_len - 1,
"%02d", frame_in);
break;
659 case '3': snprintf(q, q_len - 1,
"%03d", frame_in);
break;
660 case '4': snprintf(q, q_len - 1,
"%04d", frame_in);
break;
661 case '5': snprintf(q, q_len - 1,
"%05d", frame_in);
break;
662 case '6': snprintf(q, q_len - 1,
"%06d", frame_in);
break;
663 case '7': snprintf(q, q_len - 1,
"%07d", frame_in);
break;
664 case '8': snprintf(q, q_len - 1,
"%08d", frame_in);
break;
665 case '9': snprintf(q, q_len - 1,
"%09d", frame_in);
break;
667 die(
"Unrecognized pattern %%%c\n", p[1]);
671 if(pat_len >= q_len - 1)
672 die(
"Output filename too long.\n");
683 copy_len = strlen(p);
685 copy_len = next_pat - p;
687 if(copy_len >= q_len - 1)
688 die(
"Output filename too long.\n");
690 memcpy(q, p, copy_len);
700 int main(
int argc,
const char **argv_)
706 size_t buf_sz = 0, buf_alloc_sz = 0;
708 int frame_in = 0, frame_out = 0, flipuv = 0, noblit = 0, do_md5 = 0, progress = 0;
709 int stop_after = 0, postproc = 0, summary = 0, quiet = 1;
713 unsigned long dx_time = 0;
715 char **argv, **argi, **argj;
716 const char *outfile_pattern = 0;
717 char outfile[PATH_MAX];
722 unsigned int fps_den;
723 unsigned int fps_num;
726 #if CONFIG_VP8_DECODER
728 int vp8_dbg_color_ref_frame = 0;
729 int vp8_dbg_color_mb_modes = 0;
730 int vp8_dbg_color_b_modes = 0;
731 int vp8_dbg_display_mv = 0;
733 struct input_ctx input = {0};
734 int frames_corrupted = 0;
738 exec_name = argv_[0];
739 argv = argv_dup(argc - 1, argv_ + 1);
741 for (argi = argj = argv; (*argj = *argi); argi += arg.argv_step)
743 memset(&arg, 0,
sizeof(arg));
746 if (arg_match(&arg, &codecarg, argi))
750 for (j = 0; j <
sizeof(ifaces) /
sizeof(ifaces[0]); j++)
751 if (!strcmp(ifaces[j].name, arg.val))
755 iface = ifaces[k].iface;
757 die(
"Error: Unrecognized argument (%s) to --codec\n",
760 else if (arg_match(&arg, &outputfile, argi))
761 outfile_pattern = arg.val;
762 else if (arg_match(&arg, &use_yv12, argi))
767 else if (arg_match(&arg, &use_i420, argi))
772 else if (arg_match(&arg, &flipuvarg, argi))
774 else if (arg_match(&arg, &noblitarg, argi))
776 else if (arg_match(&arg, &progressarg, argi))
778 else if (arg_match(&arg, &limitarg, argi))
779 stop_after = arg_parse_uint(&arg);
780 else if (arg_match(&arg, &postprocarg, argi))
782 else if (arg_match(&arg, &md5arg, argi))
784 else if (arg_match(&arg, &summaryarg, argi))
786 else if (arg_match(&arg, &threadsarg, argi))
787 cfg.
threads = arg_parse_uint(&arg);
788 else if (arg_match(&arg, &verbosearg, argi))
791 #if CONFIG_VP8_DECODER
792 else if (arg_match(&arg, &addnoise_level, argi))
798 else if (arg_match(&arg, &demacroblock_level, argi))
804 else if (arg_match(&arg, &deblock, argi))
809 else if (arg_match(&arg, &mfqe, argi))
814 else if (arg_match(&arg, &pp_debug_info, argi))
816 unsigned int level = arg_parse_uint(&arg);
824 else if (arg_match(&arg, &pp_disp_ref_frame, argi))
826 unsigned int flags = arg_parse_int(&arg);
830 vp8_dbg_color_ref_frame = flags;
833 else if (arg_match(&arg, &pp_disp_mb_modes, argi))
835 unsigned int flags = arg_parse_int(&arg);
839 vp8_dbg_color_mb_modes = flags;
842 else if (arg_match(&arg, &pp_disp_b_modes, argi))
844 unsigned int flags = arg_parse_int(&arg);
848 vp8_dbg_color_b_modes = flags;
851 else if (arg_match(&arg, &pp_disp_mvs, argi))
853 unsigned int flags = arg_parse_int(&arg);
857 vp8_dbg_display_mv = flags;
860 else if (arg_match(&arg, &error_concealment, argi))
871 for (argi = argv; *argi; argi++)
872 if (argi[0][0] ==
'-' && strlen(argi[0]) > 1)
873 die(
"Error: Unrecognized option %s\n", *argi);
882 infile = strcmp(fn,
"-") ? fopen(fn,
"rb") : set_binary_mode(stdin);
886 fprintf(stderr,
"Failed to open file '%s'",
887 strcmp(fn,
"-") ? fn :
"stdin");
890 #if CONFIG_OS_SUPPORT
892 if(!outfile_pattern && isatty(fileno(stdout)) && !do_md5 && !noblit)
895 "Not dumping raw video to your terminal. Use '-o -' to "
900 input.infile = infile;
901 if(file_is_ivf(infile, &fourcc, &width, &height, &fps_den,
903 input.kind = IVF_FILE;
904 else if(file_is_webm(&input, &fourcc, &width, &height, &fps_den, &fps_num))
905 input.kind = WEBM_FILE;
906 else if(file_is_raw(infile, &fourcc, &width, &height, &fps_den, &fps_num))
907 input.kind = RAW_FILE;
910 fprintf(stderr,
"Unrecognized input file type.\n");
917 outfile_pattern = outfile_pattern ? outfile_pattern :
"-";
920 const char *p = outfile_pattern;
924 if(p && p[1] >=
'1' && p[1] <=
'9')
935 if(single_file && !noblit)
937 generate_filename(outfile_pattern, outfile,
sizeof(outfile)-1,
939 out = out_open(outfile, do_md5);
942 if (use_y4m && !noblit)
947 fprintf(stderr,
"YUV4MPEG2 not supported with output patterns,"
948 " try --i420 or --yv12.\n");
952 if(input.kind == WEBM_FILE)
953 if(webm_guess_framerate(&input, &fps_den, &fps_num))
955 fprintf(stderr,
"Failed to guess framerate -- error parsing "
964 sprintf(buffer,
"YUV4MPEG2 C%s W%u H%u F%u:%u I%c\n",
965 "420jpeg", width, height, fps_num, fps_den,
'p');
966 out_put(out, (
unsigned char *)buffer,
967 (
unsigned int)strlen(buffer), do_md5);
971 for (i = 0; i <
sizeof(ifaces) /
sizeof(ifaces[0]); i++)
972 if ((fourcc & ifaces[i].fourcc_mask) == ifaces[i].fourcc)
976 if (iface && iface != ivf_iface)
977 fprintf(stderr,
"Notice -- IVF header indicates codec: %s\n",
986 (ec_enabled ? VPX_CODEC_USE_ERROR_CONCEALMENT : 0);
990 fprintf(stderr,
"Failed to initialize decoder: %s\n",
vpx_codec_error(&decoder));
995 fprintf(stderr,
"%s\n", decoder.
name);
997 #if CONFIG_VP8_DECODER
1002 fprintf(stderr,
"Failed to configure postproc: %s\n",
vpx_codec_error(&decoder));
1003 return EXIT_FAILURE;
1006 if (vp8_dbg_color_ref_frame
1009 fprintf(stderr,
"Failed to configure reference block visualizer: %s\n",
vpx_codec_error(&decoder));
1010 return EXIT_FAILURE;
1013 if (vp8_dbg_color_mb_modes
1016 fprintf(stderr,
"Failed to configure macro block visualizer: %s\n",
vpx_codec_error(&decoder));
1017 return EXIT_FAILURE;
1020 if (vp8_dbg_color_b_modes
1023 fprintf(stderr,
"Failed to configure block visualizer: %s\n",
vpx_codec_error(&decoder));
1024 return EXIT_FAILURE;
1027 if (vp8_dbg_display_mv
1030 fprintf(stderr,
"Failed to configure motion vector visualizer: %s\n",
vpx_codec_error(&decoder));
1031 return EXIT_FAILURE;
1036 while (!read_frame(&input, &buf, &buf_sz, &buf_alloc_sz))
1040 struct vpx_usec_timer timer;
1043 vpx_usec_timer_start(&timer);
1048 fprintf(stderr,
"Failed to decode frame: %s\n",
vpx_codec_error(&decoder));
1051 fprintf(stderr,
" Additional information: %s\n", detail);
1056 vpx_usec_timer_mark(&timer);
1057 dx_time += (
unsigned int)vpx_usec_timer_elapsed(&timer);
1063 fprintf(stderr,
"Failed VP8_GET_FRAME_CORRUPTED: %s\n",
1067 frames_corrupted += corrupted;
1069 vpx_usec_timer_start(&timer);
1074 vpx_usec_timer_mark(&timer);
1075 dx_time += (
unsigned int)vpx_usec_timer_elapsed(&timer);
1078 show_progress(frame_in, frame_out, dx_time);
1085 char out_fn[PATH_MAX];
1090 size_t len =
sizeof(out_fn)-1;
1093 generate_filename(outfile_pattern, out_fn, len-1,
1094 img->
d_w, img->
d_h, frame_in);
1095 out = out_open(out_fn, do_md5);
1098 out_put(out, (
unsigned char *)
"FRAME\n", 6, do_md5);
1102 for (y = 0; y < img->
d_h; y++)
1104 out_put(out, buf, img->
d_w, do_md5);
1110 for (y = 0; y < (1 + img->
d_h) / 2; y++)
1112 out_put(out, buf, (1 + img->
d_w) / 2, do_md5);
1118 for (y = 0; y < (1 + img->
d_h) / 2; y++)
1120 out_put(out, buf, (1 + img->
d_w) / 2, do_md5);
1125 out_close(out, out_fn, do_md5);
1129 if (stop_after && frame_in >= stop_after)
1133 if (summary || progress)
1135 show_progress(frame_in, frame_out, dx_time);
1136 fprintf(stderr,
"\n");
1139 if (frames_corrupted)
1140 fprintf(stderr,
"WARNING: %d frames corrupted.\n",frames_corrupted);
1146 fprintf(stderr,
"Failed to destroy decoder: %s\n",
vpx_codec_error(&decoder));
1147 return EXIT_FAILURE;
1150 if (single_file && !noblit)
1151 out_close(out, outfile, do_md5);
1153 if(input.nestegg_ctx)
1154 nestegg_destroy(input.nestegg_ctx);
1155 if(input.kind != WEBM_FILE)
1160 return frames_corrupted ? EXIT_FAILURE : EXIT_SUCCESS;