postproc: Add mode and refrence frame visualizers.
[libvpx.git] / vpxdec.c
blobee84197845ca08599bef39f8bfb1ebf04c32cde7
1 /*
2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved.
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
12 /* This is a simple program that reads ivf files and decodes them
13 * using the new interface. Decoded frames are output as YV12 raw.
15 #include <stdio.h>
16 #include <stdlib.h>
17 #include <stdarg.h>
18 #include <string.h>
19 #include <limits.h>
20 #if defined(_WIN32)
21 #include <io.h>
22 #define snprintf _snprintf
23 #define isatty _isatty
24 #define fileno _fileno
25 #else
26 #include <unistd.h>
27 #endif
28 #define VPX_CODEC_DISABLE_COMPAT 1
29 #include "vpx_config.h"
30 #include "vpx/vpx_decoder.h"
31 #include "vpx_ports/vpx_timer.h"
32 #if CONFIG_VP8_DECODER
33 #include "vpx/vp8dx.h"
34 #endif
35 #if CONFIG_MD5
36 #include "md5_utils.h"
37 #endif
38 #include "nestegg/include/nestegg/nestegg.h"
40 #ifndef PATH_MAX
41 #define PATH_MAX 256
42 #endif
44 static const char *exec_name;
46 #define VP8_FOURCC (0x00385056)
47 static const struct
49 char const *name;
50 const vpx_codec_iface_t *iface;
51 unsigned int fourcc;
52 unsigned int fourcc_mask;
53 } ifaces[] =
55 #if CONFIG_VP8_DECODER
56 {"vp8", &vpx_codec_vp8_dx_algo, VP8_FOURCC, 0x00FFFFFF},
57 #endif
60 #include "args.h"
61 static const arg_def_t codecarg = ARG_DEF(NULL, "codec", 1,
62 "Codec to use");
63 static const arg_def_t use_yv12 = ARG_DEF(NULL, "yv12", 0,
64 "Output raw YV12 frames");
65 static const arg_def_t use_i420 = ARG_DEF(NULL, "i420", 0,
66 "Output raw I420 frames");
67 static const arg_def_t flipuvarg = ARG_DEF(NULL, "flipuv", 0,
68 "Flip the chroma planes in the output");
69 static const arg_def_t noblitarg = ARG_DEF(NULL, "noblit", 0,
70 "Don't process the decoded frames");
71 static const arg_def_t progressarg = ARG_DEF(NULL, "progress", 0,
72 "Show progress after each frame decodes");
73 static const arg_def_t limitarg = ARG_DEF(NULL, "limit", 1,
74 "Stop decoding after n frames");
75 static const arg_def_t postprocarg = ARG_DEF(NULL, "postproc", 0,
76 "Postprocess decoded frames");
77 static const arg_def_t summaryarg = ARG_DEF(NULL, "summary", 0,
78 "Show timing summary");
79 static const arg_def_t outputfile = ARG_DEF("o", "output", 1,
80 "Output file name pattern (see below)");
81 static const arg_def_t threadsarg = ARG_DEF("t", "threads", 1,
82 "Max threads to use");
83 static const arg_def_t verbosearg = ARG_DEF("v", "verbose", 0,
84 "Show version string");
86 #if CONFIG_MD5
87 static const arg_def_t md5arg = ARG_DEF(NULL, "md5", 0,
88 "Compute the MD5 sum of the decoded frame");
89 #endif
90 static const arg_def_t *all_args[] =
92 &codecarg, &use_yv12, &use_i420, &flipuvarg, &noblitarg,
93 &progressarg, &limitarg, &postprocarg, &summaryarg, &outputfile,
94 &threadsarg, &verbosearg,
95 #if CONFIG_MD5
96 &md5arg,
97 #endif
98 NULL
101 #if CONFIG_VP8_DECODER
102 static const arg_def_t addnoise_level = ARG_DEF(NULL, "noise-level", 1,
103 "Enable VP8 postproc add noise");
104 static const arg_def_t deblock = ARG_DEF(NULL, "deblock", 0,
105 "Enable VP8 deblocking");
106 static const arg_def_t demacroblock_level = ARG_DEF(NULL, "demacroblock-level", 1,
107 "Enable VP8 demacroblocking, w/ level");
108 static const arg_def_t pp_debug_info = ARG_DEF(NULL, "pp-debug-info", 1,
109 "Enable VP8 visible debug info");
112 static const arg_def_t *vp8_pp_args[] =
114 &addnoise_level, &deblock, &demacroblock_level, &pp_debug_info,
115 NULL
117 #endif
119 static void usage_exit()
121 int i;
123 fprintf(stderr, "Usage: %s <options> filename\n\n"
124 "Options:\n", exec_name);
125 arg_show_usage(stderr, all_args);
126 #if CONFIG_VP8_DECODER
127 fprintf(stderr, "\nVP8 Postprocessing Options:\n");
128 arg_show_usage(stderr, vp8_pp_args);
129 #endif
130 fprintf(stderr,
131 "\nOutput File Patterns:\n\n"
132 " The -o argument specifies the name of the file(s) to "
133 "write to. If the\n argument does not include any escape "
134 "characters, the output will be\n written to a single file. "
135 "Otherwise, the filename will be calculated by\n expanding "
136 "the following escape characters:\n"
137 "\n\t%%w - Frame width"
138 "\n\t%%h - Frame height"
139 "\n\t%%<n> - Frame number, zero padded to <n> places (1..9)"
140 "\n\n Pattern arguments are only supported in conjunction "
141 "with the --yv12 and\n --i420 options. If the -o option is "
142 "not specified, the output will be\n directed to stdout.\n"
144 fprintf(stderr, "\nIncluded decoders:\n\n");
146 for (i = 0; i < sizeof(ifaces) / sizeof(ifaces[0]); i++)
147 fprintf(stderr, " %-6s - %s\n",
148 ifaces[i].name,
149 vpx_codec_iface_name(ifaces[i].iface));
151 exit(EXIT_FAILURE);
154 void die(const char *fmt, ...)
156 va_list ap;
157 va_start(ap, fmt);
158 vfprintf(stderr, fmt, ap);
159 fprintf(stderr, "\n");
160 usage_exit();
163 static unsigned int mem_get_le16(const void *vmem)
165 unsigned int val;
166 const unsigned char *mem = (const unsigned char *)vmem;
168 val = mem[1] << 8;
169 val |= mem[0];
170 return val;
173 static unsigned int mem_get_le32(const void *vmem)
175 unsigned int val;
176 const unsigned char *mem = (const unsigned char *)vmem;
178 val = mem[3] << 24;
179 val |= mem[2] << 16;
180 val |= mem[1] << 8;
181 val |= mem[0];
182 return val;
185 enum file_kind
187 RAW_FILE,
188 IVF_FILE,
189 WEBM_FILE
192 struct input_ctx
194 enum file_kind kind;
195 FILE *infile;
196 nestegg *nestegg_ctx;
197 nestegg_packet *pkt;
198 unsigned int chunk;
199 unsigned int chunks;
200 unsigned int video_track;
203 #define IVF_FRAME_HDR_SZ (sizeof(uint32_t) + sizeof(uint64_t))
204 #define RAW_FRAME_HDR_SZ (sizeof(uint32_t))
205 static int read_frame(struct input_ctx *input,
206 uint8_t **buf,
207 size_t *buf_sz,
208 size_t *buf_alloc_sz)
210 char raw_hdr[IVF_FRAME_HDR_SZ];
211 size_t new_buf_sz;
212 FILE *infile = input->infile;
213 enum file_kind kind = input->kind;
214 if(kind == WEBM_FILE)
216 if(input->chunk >= input->chunks)
218 unsigned int track;
222 /* End of this packet, get another. */
223 if(input->pkt)
224 nestegg_free_packet(input->pkt);
226 if(nestegg_read_packet(input->nestegg_ctx, &input->pkt) <= 0
227 || nestegg_packet_track(input->pkt, &track))
228 return 1;
230 } while(track != input->video_track);
232 if(nestegg_packet_count(input->pkt, &input->chunks))
233 return 1;
234 input->chunk = 0;
237 if(nestegg_packet_data(input->pkt, input->chunk, buf, buf_sz))
238 return 1;
239 input->chunk++;
241 return 0;
243 /* For both the raw and ivf formats, the frame size is the first 4 bytes
244 * of the frame header. We just need to special case on the header
245 * size.
247 else if (fread(raw_hdr, kind==IVF_FILE
248 ? IVF_FRAME_HDR_SZ : RAW_FRAME_HDR_SZ, 1, infile) != 1)
250 if (!feof(infile))
251 fprintf(stderr, "Failed to read frame size\n");
253 new_buf_sz = 0;
255 else
257 new_buf_sz = mem_get_le32(raw_hdr);
259 if (new_buf_sz > 256 * 1024 * 1024)
261 fprintf(stderr, "Error: Read invalid frame size (%u)\n",
262 (unsigned int)new_buf_sz);
263 new_buf_sz = 0;
266 if (kind == RAW_FILE && new_buf_sz > 256 * 1024)
267 fprintf(stderr, "Warning: Read invalid frame size (%u)"
268 " - not a raw file?\n", (unsigned int)new_buf_sz);
270 if (new_buf_sz > *buf_alloc_sz)
272 uint8_t *new_buf = realloc(*buf, 2 * new_buf_sz);
274 if (new_buf)
276 *buf = new_buf;
277 *buf_alloc_sz = 2 * new_buf_sz;
279 else
281 fprintf(stderr, "Failed to allocate compressed data buffer\n");
282 new_buf_sz = 0;
287 *buf_sz = new_buf_sz;
289 if (*buf_sz)
291 if (fread(*buf, 1, *buf_sz, infile) != *buf_sz)
293 fprintf(stderr, "Failed to read full frame\n");
294 return 1;
297 return 0;
300 return 1;
303 void *out_open(const char *out_fn, int do_md5)
305 void *out = NULL;
307 if (do_md5)
309 #if CONFIG_MD5
310 MD5Context *md5_ctx = out = malloc(sizeof(MD5Context));
311 (void)out_fn;
312 MD5Init(md5_ctx);
313 #endif
315 else
317 FILE *outfile = out = strcmp("-", out_fn) ? fopen(out_fn, "wb") : stdout;
319 if (!outfile)
321 fprintf(stderr, "Failed to output file");
322 exit(EXIT_FAILURE);
326 return out;
329 void out_put(void *out, const uint8_t *buf, unsigned int len, int do_md5)
331 if (do_md5)
333 #if CONFIG_MD5
334 MD5Update(out, buf, len);
335 #endif
337 else
339 fwrite(buf, 1, len, out);
343 void out_close(void *out, const char *out_fn, int do_md5)
345 if (do_md5)
347 #if CONFIG_MD5
348 uint8_t md5[16];
349 int i;
351 MD5Final(md5, out);
352 free(out);
354 for (i = 0; i < 16; i++)
355 printf("%02x", md5[i]);
357 printf(" %s\n", out_fn);
358 #endif
360 else
362 fclose(out);
366 unsigned int file_is_ivf(FILE *infile,
367 unsigned int *fourcc,
368 unsigned int *width,
369 unsigned int *height,
370 unsigned int *fps_den,
371 unsigned int *fps_num)
373 char raw_hdr[32];
374 int is_ivf = 0;
376 if (fread(raw_hdr, 1, 32, infile) == 32)
378 if (raw_hdr[0] == 'D' && raw_hdr[1] == 'K'
379 && raw_hdr[2] == 'I' && raw_hdr[3] == 'F')
381 is_ivf = 1;
383 if (mem_get_le16(raw_hdr + 4) != 0)
384 fprintf(stderr, "Error: Unrecognized IVF version! This file may not"
385 " decode properly.");
387 *fourcc = mem_get_le32(raw_hdr + 8);
388 *width = mem_get_le16(raw_hdr + 12);
389 *height = mem_get_le16(raw_hdr + 14);
390 *fps_num = mem_get_le32(raw_hdr + 16);
391 *fps_den = mem_get_le32(raw_hdr + 20);
393 /* Some versions of vpxenc used 1/(2*fps) for the timebase, so
394 * we can guess the framerate using only the timebase in this
395 * case. Other files would require reading ahead to guess the
396 * timebase, like we do for webm.
398 if(*fps_num < 1000)
400 /* Correct for the factor of 2 applied to the timebase in the
401 * encoder.
403 if(*fps_num&1)*fps_den<<=1;
404 else *fps_num>>=1;
406 else
408 /* Don't know FPS for sure, and don't have readahead code
409 * (yet?), so just default to 30fps.
411 *fps_num = 30;
412 *fps_den = 1;
417 if (!is_ivf)
418 rewind(infile);
420 return is_ivf;
424 unsigned int file_is_raw(FILE *infile,
425 unsigned int *fourcc,
426 unsigned int *width,
427 unsigned int *height,
428 unsigned int *fps_den,
429 unsigned int *fps_num)
431 unsigned char buf[32];
432 int is_raw = 0;
433 vpx_codec_stream_info_t si;
435 if (fread(buf, 1, 32, infile) == 32)
437 int i;
439 if(mem_get_le32(buf) < 256 * 1024 * 1024)
440 for (i = 0; i < sizeof(ifaces) / sizeof(ifaces[0]); i++)
441 if(!vpx_codec_peek_stream_info(ifaces[i].iface,
442 buf + 4, 32 - 4, &si))
444 is_raw = 1;
445 *fourcc = ifaces[i].fourcc;
446 *width = si.w;
447 *height = si.h;
448 *fps_num = 30;
449 *fps_den = 1;
450 break;
454 rewind(infile);
455 return is_raw;
459 static int
460 nestegg_read_cb(void *buffer, size_t length, void *userdata)
462 FILE *f = userdata;
464 fread(buffer, 1, length, f);
465 if (ferror(f))
466 return -1;
467 if (feof(f))
468 return 0;
469 return 1;
473 static int
474 nestegg_seek_cb(int64_t offset, int whence, void * userdata)
476 switch(whence) {
477 case NESTEGG_SEEK_SET: whence = SEEK_SET; break;
478 case NESTEGG_SEEK_CUR: whence = SEEK_CUR; break;
479 case NESTEGG_SEEK_END: whence = SEEK_END; break;
481 return fseek(userdata, offset, whence)? -1 : 0;
485 static int64_t
486 nestegg_tell_cb(void * userdata)
488 return ftell(userdata);
492 static void
493 nestegg_log_cb(nestegg * context, unsigned int severity, char const * format,
494 ...)
496 va_list ap;
498 va_start(ap, format);
499 vfprintf(stderr, format, ap);
500 fprintf(stderr, "\n");
501 va_end(ap);
505 static int
506 webm_guess_framerate(struct input_ctx *input,
507 unsigned int *fps_den,
508 unsigned int *fps_num)
510 unsigned int i;
511 uint64_t tstamp=0;
513 /* Guess the framerate. Read up to 1 second, or 50 video packets,
514 * whichever comes first.
516 for(i=0; tstamp < 1000000000 && i < 50;)
518 nestegg_packet * pkt;
519 unsigned int track;
521 if(nestegg_read_packet(input->nestegg_ctx, &pkt) <= 0)
522 break;
524 nestegg_packet_track(pkt, &track);
525 if(track == input->video_track)
527 nestegg_packet_tstamp(pkt, &tstamp);
528 i++;
531 nestegg_free_packet(pkt);
534 if(nestegg_track_seek(input->nestegg_ctx, input->video_track, 0))
535 goto fail;
537 *fps_num = (i - 1) * 1000000;
538 *fps_den = tstamp / 1000;
539 return 0;
540 fail:
541 input->nestegg_ctx = NULL;
542 rewind(input->infile);
543 return 1;
547 static int
548 file_is_webm(struct input_ctx *input,
549 unsigned int *fourcc,
550 unsigned int *width,
551 unsigned int *height,
552 unsigned int *fps_den,
553 unsigned int *fps_num)
555 unsigned int i, n;
556 int track_type = -1;
557 uint64_t tstamp=0;
559 nestegg_io io = {nestegg_read_cb, nestegg_seek_cb, nestegg_tell_cb,
560 input->infile};
561 nestegg_video_params params;
562 nestegg_packet * pkt;
564 if(nestegg_init(&input->nestegg_ctx, io, NULL))
565 goto fail;
567 if(nestegg_track_count(input->nestegg_ctx, &n))
568 goto fail;
570 for(i=0; i<n; i++)
572 track_type = nestegg_track_type(input->nestegg_ctx, i);
574 if(track_type == NESTEGG_TRACK_VIDEO)
575 break;
576 else if(track_type < 0)
577 goto fail;
580 if(nestegg_track_codec_id(input->nestegg_ctx, i) != NESTEGG_CODEC_VP8)
582 fprintf(stderr, "Not VP8 video, quitting.\n");
583 exit(1);
586 input->video_track = i;
588 if(nestegg_track_video_params(input->nestegg_ctx, i, &params))
589 goto fail;
591 *fps_den = 0;
592 *fps_num = 0;
593 *fourcc = VP8_FOURCC;
594 *width = params.width;
595 *height = params.height;
596 return 1;
597 fail:
598 input->nestegg_ctx = NULL;
599 rewind(input->infile);
600 return 0;
604 void show_progress(int frame_in, int frame_out, unsigned long dx_time)
606 fprintf(stderr, "%d decoded frames/%d showed frames in %lu us (%.2f fps)\r",
607 frame_in, frame_out, dx_time,
608 (float)frame_out * 1000000.0 / (float)dx_time);
612 void generate_filename(const char *pattern, char *out, size_t q_len,
613 unsigned int d_w, unsigned int d_h,
614 unsigned int frame_in)
616 const char *p = pattern;
617 char *q = out;
621 char *next_pat = strchr(p, '%');
623 if(p == next_pat)
625 size_t pat_len;
627 // parse the pattern
628 q[q_len - 1] = '\0';
629 switch(p[1])
631 case 'w': snprintf(q, q_len - 1, "%d", d_w); break;
632 case 'h': snprintf(q, q_len - 1, "%d", d_h); break;
633 case '1': snprintf(q, q_len - 1, "%d", frame_in); break;
634 case '2': snprintf(q, q_len - 1, "%02d", frame_in); break;
635 case '3': snprintf(q, q_len - 1, "%03d", frame_in); break;
636 case '4': snprintf(q, q_len - 1, "%04d", frame_in); break;
637 case '5': snprintf(q, q_len - 1, "%05d", frame_in); break;
638 case '6': snprintf(q, q_len - 1, "%06d", frame_in); break;
639 case '7': snprintf(q, q_len - 1, "%07d", frame_in); break;
640 case '8': snprintf(q, q_len - 1, "%08d", frame_in); break;
641 case '9': snprintf(q, q_len - 1, "%09d", frame_in); break;
642 default:
643 die("Unrecognized pattern %%%c\n", p[1]);
646 pat_len = strlen(q);
647 if(pat_len >= q_len - 1)
648 die("Output filename too long.\n");
649 q += pat_len;
650 p += 2;
651 q_len -= pat_len;
653 else
655 size_t copy_len;
657 // copy the next segment
658 if(!next_pat)
659 copy_len = strlen(p);
660 else
661 copy_len = next_pat - p;
663 if(copy_len >= q_len - 1)
664 die("Output filename too long.\n");
666 memcpy(q, p, copy_len);
667 q[copy_len] = '\0';
668 q += copy_len;
669 p += copy_len;
670 q_len -= copy_len;
672 } while(*p);
676 int main(int argc, const char **argv_)
678 vpx_codec_ctx_t decoder;
679 char *fn = NULL;
680 int i;
681 uint8_t *buf = NULL;
682 size_t buf_sz = 0, buf_alloc_sz = 0;
683 FILE *infile;
684 int frame_in = 0, frame_out = 0, flipuv = 0, noblit = 0, do_md5 = 0, progress = 0;
685 int stop_after = 0, postproc = 0, summary = 0, quiet = 1;
686 vpx_codec_iface_t *iface = NULL;
687 unsigned int fourcc;
688 unsigned long dx_time = 0;
689 struct arg arg;
690 char **argv, **argi, **argj;
691 const char *outfile_pattern = 0;
692 char outfile[PATH_MAX];
693 int single_file;
694 int use_y4m = 1;
695 unsigned int width;
696 unsigned int height;
697 unsigned int fps_den;
698 unsigned int fps_num;
699 void *out = NULL;
700 vpx_codec_dec_cfg_t cfg = {0};
701 #if CONFIG_VP8_DECODER
702 vp8_postproc_cfg_t vp8_pp_cfg = {0};
703 #endif
704 struct input_ctx input = {0};
706 /* Parse command line */
707 exec_name = argv_[0];
708 argv = argv_dup(argc - 1, argv_ + 1);
710 for (argi = argj = argv; (*argj = *argi); argi += arg.argv_step)
712 memset(&arg, 0, sizeof(arg));
713 arg.argv_step = 1;
715 if (arg_match(&arg, &codecarg, argi))
717 int j, k = -1;
719 for (j = 0; j < sizeof(ifaces) / sizeof(ifaces[0]); j++)
720 if (!strcmp(ifaces[j].name, arg.val))
721 k = j;
723 if (k >= 0)
724 iface = ifaces[k].iface;
725 else
726 die("Error: Unrecognized argument (%s) to --codec\n",
727 arg.val);
729 else if (arg_match(&arg, &outputfile, argi))
730 outfile_pattern = arg.val;
731 else if (arg_match(&arg, &use_yv12, argi))
733 use_y4m = 0;
734 flipuv = 1;
736 else if (arg_match(&arg, &use_i420, argi))
738 use_y4m = 0;
739 flipuv = 0;
741 else if (arg_match(&arg, &flipuvarg, argi))
742 flipuv = 1;
743 else if (arg_match(&arg, &noblitarg, argi))
744 noblit = 1;
745 else if (arg_match(&arg, &progressarg, argi))
746 progress = 1;
747 else if (arg_match(&arg, &limitarg, argi))
748 stop_after = arg_parse_uint(&arg);
749 else if (arg_match(&arg, &postprocarg, argi))
750 postproc = 1;
751 else if (arg_match(&arg, &md5arg, argi))
752 do_md5 = 1;
753 else if (arg_match(&arg, &summaryarg, argi))
754 summary = 1;
755 else if (arg_match(&arg, &threadsarg, argi))
756 cfg.threads = arg_parse_uint(&arg);
757 else if (arg_match(&arg, &verbosearg, argi))
758 quiet = 0;
760 #if CONFIG_VP8_DECODER
761 else if (arg_match(&arg, &addnoise_level, argi))
763 postproc = 1;
764 vp8_pp_cfg.post_proc_flag |= VP8_ADDNOISE;
765 vp8_pp_cfg.noise_level = arg_parse_uint(&arg);
767 else if (arg_match(&arg, &demacroblock_level, argi))
769 postproc = 1;
770 vp8_pp_cfg.post_proc_flag |= VP8_DEMACROBLOCK;
771 vp8_pp_cfg.deblocking_level = arg_parse_uint(&arg);
773 else if (arg_match(&arg, &deblock, argi))
775 postproc = 1;
776 vp8_pp_cfg.post_proc_flag |= VP8_DEBLOCK;
778 else if (arg_match(&arg, &pp_debug_info, argi))
780 unsigned int level = arg_parse_uint(&arg);
782 postproc = 1;
783 vp8_pp_cfg.post_proc_flag &= ~0x7;
785 if (level)
786 vp8_pp_cfg.post_proc_flag |= 8 << (level - 1);
789 #endif
790 else
791 argj++;
794 /* Check for unrecognized options */
795 for (argi = argv; *argi; argi++)
796 if (argi[0][0] == '-' && strlen(argi[0]) > 1)
797 die("Error: Unrecognized option %s\n", *argi);
799 /* Handle non-option arguments */
800 fn = argv[0];
802 if (!fn)
803 usage_exit();
805 /* Open file */
806 infile = strcmp(fn, "-") ? fopen(fn, "rb") : stdin;
808 if (!infile)
810 fprintf(stderr, "Failed to open file '%s'",
811 strcmp(fn, "-") ? fn : "stdin");
812 return EXIT_FAILURE;
815 /* Make sure we don't dump to the terminal, unless forced to with -o - */
816 if(!outfile_pattern && isatty(fileno(stdout)) && !do_md5)
818 fprintf(stderr,
819 "Not dumping raw video to your terminal. Use '-o -' to "
820 "override.\n");
821 return EXIT_FAILURE;
824 input.infile = infile;
825 if(file_is_ivf(infile, &fourcc, &width, &height, &fps_den,
826 &fps_num))
827 input.kind = IVF_FILE;
828 else if(file_is_webm(&input, &fourcc, &width, &height, &fps_den, &fps_num))
829 input.kind = WEBM_FILE;
830 else if(file_is_raw(infile, &fourcc, &width, &height, &fps_den, &fps_num))
831 input.kind = RAW_FILE;
832 else
834 fprintf(stderr, "Unrecognized input file type.\n");
835 return EXIT_FAILURE;
838 /* If the output file is not set or doesn't have a sequence number in
839 * it, then we only open it once.
841 outfile_pattern = outfile_pattern ? outfile_pattern : "-";
842 single_file = 1;
844 const char *p = outfile_pattern;
847 p = strchr(p, '%');
848 if(p && p[1] >= '1' && p[1] <= '9')
850 // pattern contains sequence number, so it's not unique.
851 single_file = 0;
852 break;
854 if(p)
855 p++;
856 } while(p);
859 if(single_file && !noblit)
861 generate_filename(outfile_pattern, outfile, sizeof(outfile)-1,
862 width, height, 0);
863 out = out_open(outfile, do_md5);
866 if (use_y4m && !noblit)
868 char buffer[128];
869 if (!single_file)
871 fprintf(stderr, "YUV4MPEG2 not supported with output patterns,"
872 " try --i420 or --yv12.\n");
873 return EXIT_FAILURE;
876 if(input.kind == WEBM_FILE)
877 webm_guess_framerate(&input, &fps_den, &fps_num);
879 /*Note: We can't output an aspect ratio here because IVF doesn't
880 store one, and neither does VP8.
881 That will have to wait until these tools support WebM natively.*/
882 sprintf(buffer, "YUV4MPEG2 C%s W%u H%u F%u:%u I%c\n",
883 "420jpeg", width, height, fps_num, fps_den, 'p');
884 out_put(out, (unsigned char *)buffer, strlen(buffer), do_md5);
887 /* Try to determine the codec from the fourcc. */
888 for (i = 0; i < sizeof(ifaces) / sizeof(ifaces[0]); i++)
889 if ((fourcc & ifaces[i].fourcc_mask) == ifaces[i].fourcc)
891 vpx_codec_iface_t *ivf_iface = ifaces[i].iface;
893 if (iface && iface != ivf_iface)
894 fprintf(stderr, "Notice -- IVF header indicates codec: %s\n",
895 ifaces[i].name);
896 else
897 iface = ivf_iface;
899 break;
902 if (vpx_codec_dec_init(&decoder, iface ? iface : ifaces[0].iface, &cfg,
903 postproc ? VPX_CODEC_USE_POSTPROC : 0))
905 fprintf(stderr, "Failed to initialize decoder: %s\n", vpx_codec_error(&decoder));
906 return EXIT_FAILURE;
909 if (!quiet)
910 fprintf(stderr, "%s\n", decoder.name);
912 #if CONFIG_VP8_DECODER
914 if (vp8_pp_cfg.post_proc_flag
915 && vpx_codec_control(&decoder, VP8_SET_POSTPROC, &vp8_pp_cfg))
917 fprintf(stderr, "Failed to configure postproc: %s\n", vpx_codec_error(&decoder));
918 return EXIT_FAILURE;
921 #endif
923 /* Decode file */
924 while (!read_frame(&input, &buf, &buf_sz, &buf_alloc_sz))
926 vpx_codec_iter_t iter = NULL;
927 vpx_image_t *img;
928 struct vpx_usec_timer timer;
930 vpx_usec_timer_start(&timer);
932 if (vpx_codec_decode(&decoder, buf, buf_sz, NULL, 0))
934 const char *detail = vpx_codec_error_detail(&decoder);
935 fprintf(stderr, "Failed to decode frame: %s\n", vpx_codec_error(&decoder));
937 if (detail)
938 fprintf(stderr, " Additional information: %s\n", detail);
940 goto fail;
943 vpx_usec_timer_mark(&timer);
944 dx_time += vpx_usec_timer_elapsed(&timer);
946 ++frame_in;
948 if ((img = vpx_codec_get_frame(&decoder, &iter)))
949 ++frame_out;
951 if (progress)
952 show_progress(frame_in, frame_out, dx_time);
954 if (!noblit)
956 if (img)
958 unsigned int y;
959 char out_fn[PATH_MAX];
960 uint8_t *buf;
962 if (!single_file)
964 size_t len = sizeof(out_fn)-1;
966 out_fn[len] = '\0';
967 generate_filename(outfile_pattern, out_fn, len-1,
968 img->d_w, img->d_h, frame_in);
969 out = out_open(out_fn, do_md5);
971 else if(use_y4m)
972 out_put(out, (unsigned char *)"FRAME\n", 6, do_md5);
974 buf = img->planes[VPX_PLANE_Y];
976 for (y = 0; y < img->d_h; y++)
978 out_put(out, buf, img->d_w, do_md5);
979 buf += img->stride[VPX_PLANE_Y];
982 buf = img->planes[flipuv?VPX_PLANE_V:VPX_PLANE_U];
984 for (y = 0; y < (1 + img->d_h) / 2; y++)
986 out_put(out, buf, (1 + img->d_w) / 2, do_md5);
987 buf += img->stride[VPX_PLANE_U];
990 buf = img->planes[flipuv?VPX_PLANE_U:VPX_PLANE_V];
992 for (y = 0; y < (1 + img->d_h) / 2; y++)
994 out_put(out, buf, (1 + img->d_w) / 2, do_md5);
995 buf += img->stride[VPX_PLANE_V];
998 if (!single_file)
999 out_close(out, out_fn, do_md5);
1003 if (stop_after && frame_in >= stop_after)
1004 break;
1007 if (summary || progress)
1009 show_progress(frame_in, frame_out, dx_time);
1010 fprintf(stderr, "\n");
1013 fail:
1015 if (vpx_codec_destroy(&decoder))
1017 fprintf(stderr, "Failed to destroy decoder: %s\n", vpx_codec_error(&decoder));
1018 return EXIT_FAILURE;
1021 if (single_file && !noblit)
1022 out_close(out, outfile, do_md5);
1024 if(input.nestegg_ctx)
1025 nestegg_destroy(input.nestegg_ctx);
1026 if(input.kind != WEBM_FILE)
1027 free(buf);
1028 fclose(infile);
1029 free(argv);
1031 return EXIT_SUCCESS;