WebM VP8 Codec SDK: vpxdec

WebM VP8 Codec SDK

- Main Page
- Related Pages
- Modules
- Data Structures
- Files

vpxdec
1 /*
2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved.
3 *
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 */
10
11
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.
14 */
15 #include <assert.h>
16 #include <stdio.h>
17 #include <stdlib.h>
18 #include <stdarg.h>
19 #include <string.h>
20 #include <limits.h>
21
22 #define VPX_CODEC_DISABLE_COMPAT 1
23 #include "vpx_config.h"
24 #include "vpx/vpx_decoder.h"
25 #include "vpx_ports/vpx_timer.h"
26 #if CONFIG_VP8_DECODER || CONFIG_VP9_DECODER
27 #include "vpx/vp8dx.h"
28 #endif
29 #if CONFIG_MD5
30 #include "md5_utils.h"
31 #endif
32 #include "tools_common.h"
33 #include "nestegg/include/nestegg/nestegg.h"
34 #include "third_party/libyuv/include/libyuv/scale.h"
35
36 static const char *exec_name;
37
38 static const struct {
39 char const *name;
40 const vpx_codec_iface_t *(*iface)(void);
41 unsigned int fourcc;
42 unsigned int fourcc_mask;
43 } ifaces[] = {
44 #if CONFIG_VP8_DECODER
45 {"vp8", vpx_codec_vp8_dx, VP8_FOURCC_MASK, 0x00FFFFFF},
46 #endif
47 #if CONFIG_VP9_DECODER
48 {"vp9", vpx_codec_vp9_dx, VP9_FOURCC_MASK, 0x00FFFFFF},
49 #endif
50 };
51
52 #include "args.h"
53 static const arg_def_t looparg = ARG_DEF(NULL, "loops", 1,
54 "Number of times to decode the file");
55 static const arg_def_t codecarg = ARG_DEF(NULL, "codec", 1,
56 "Codec to use");
57 static const arg_def_t use_yv12 = ARG_DEF(NULL, "yv12", 0,
58 "Output raw YV12 frames");
59 static const arg_def_t use_i420 = ARG_DEF(NULL, "i420", 0,
60 "Output raw I420 frames");
61 static const arg_def_t flipuvarg = ARG_DEF(NULL, "flipuv", 0,
62 "Flip the chroma planes in the output");
63 static const arg_def_t noblitarg = ARG_DEF(NULL, "noblit", 0,
64 "Don't process the decoded frames");
65 static const arg_def_t progressarg = ARG_DEF(NULL, "progress", 0,
66 "Show progress after each frame decodes");
67 static const arg_def_t limitarg = ARG_DEF(NULL, "limit", 1,
68 "Stop decoding after n frames");
69 static const arg_def_t skiparg = ARG_DEF(NULL, "skip", 1,
70 "Skip the first n input frames");
71 static const arg_def_t postprocarg = ARG_DEF(NULL, "postproc", 0,
72 "Postprocess decoded frames");
73 static const arg_def_t summaryarg = ARG_DEF(NULL, "summary", 0,
74 "Show timing summary");
75 static const arg_def_t outputfile = ARG_DEF("o", "output", 1,
76 "Output file name pattern (see below)");
77 static const arg_def_t threadsarg = ARG_DEF("t", "threads", 1,
78 "Max threads to use");
79 static const arg_def_t verbosearg = ARG_DEF("v", "verbose", 0,
80 "Show version string");
81 static const arg_def_t error_concealment = ARG_DEF(NULL, "error-concealment", 0,
82 "Enable decoder error-concealment");
83 static const arg_def_t scalearg = ARG_DEF("S", "scale", 0,
84 "Scale output frames uniformly");
85
86
87 #if CONFIG_MD5
88 static const arg_def_t md5arg = ARG_DEF(NULL, "md5", 0,
89 "Compute the MD5 sum of the decoded frame");
90 #endif
91 static const arg_def_t *all_args[] = {
92 &codecarg, &use_yv12, &use_i420, &flipuvarg, &noblitarg,
93 &progressarg, &limitarg, &skiparg, &postprocarg, &summaryarg, &outputfile,
94 &threadsarg, &verbosearg, &scalearg,
95 #if CONFIG_MD5
96 &md5arg,
97 #endif
98 &error_concealment,
99 NULL
100 };
101
102 #if CONFIG_VP8_DECODER
103 static const arg_def_t addnoise_level = ARG_DEF(NULL, "noise-level", 1,
104 "Enable VP8 postproc add noise");
105 static const arg_def_t deblock = ARG_DEF(NULL, "deblock", 0,
106 "Enable VP8 deblocking");
107 static const arg_def_t demacroblock_level = ARG_DEF(NULL, "demacroblock-level", 1,
108 "Enable VP8 demacroblocking, w/ level");
109 static const arg_def_t pp_debug_info = ARG_DEF(NULL, "pp-debug-info", 1,
110 "Enable VP8 visible debug info");
111 static const arg_def_t pp_disp_ref_frame = ARG_DEF(NULL, "pp-dbg-ref-frame", 1,
112 "Display only selected reference frame per macro block");
113 static const arg_def_t pp_disp_mb_modes = ARG_DEF(NULL, "pp-dbg-mb-modes", 1,
114 "Display only selected macro block modes");
115 static const arg_def_t pp_disp_b_modes = ARG_DEF(NULL, "pp-dbg-b-modes", 1,
116 "Display only selected block modes");
117 static const arg_def_t pp_disp_mvs = ARG_DEF(NULL, "pp-dbg-mvs", 1,
118 "Draw only selected motion vectors");
119 static const arg_def_t mfqe = ARG_DEF(NULL, "mfqe", 0,
120 "Enable multiframe quality enhancement");
121
122 static const arg_def_t *vp8_pp_args[] = {
123 &addnoise_level, &deblock, &demacroblock_level, &pp_debug_info,
124 &pp_disp_ref_frame, &pp_disp_mb_modes, &pp_disp_b_modes, &pp_disp_mvs, &mfqe,
125 NULL
126 };
127 #endif
128
129 void usage_exit() {
130 int i;
131
132 fprintf(stderr, "Usage: %s <options> filename\n\n"
133 "Options:\n", exec_name);
134 arg_show_usage(stderr, all_args);
135 #if CONFIG_VP8_DECODER
136 fprintf(stderr, "\nVP8 Postprocessing Options:\n");
137 arg_show_usage(stderr, vp8_pp_args);
138 #endif
139 fprintf(stderr,
140 "\nOutput File Patterns:\n\n"
141 " The -o argument specifies the name of the file(s) to "
142 "write to. If the\n argument does not include any escape "
143 "characters, the output will be\n written to a single file. "
144 "Otherwise, the filename will be calculated by\n expanding "
145 "the following escape characters:\n");
146 fprintf(stderr,
147 "\n\t%%w - Frame width"
148 "\n\t%%h - Frame height"
149 "\n\t%%<n> - Frame number, zero padded to <n> places (1..9)"
150 "\n\n Pattern arguments are only supported in conjunction "
151 "with the --yv12 and\n --i420 options. If the -o option is "
152 "not specified, the output will be\n directed to stdout.\n"
153 );
154 fprintf(stderr, "\nIncluded decoders:\n\n");
155
156 for (i = 0; i < sizeof(ifaces) / sizeof(ifaces[0]); i++)
157 fprintf(stderr, " %-6s - %s\n",
158 ifaces[i].name,
159 vpx_codec_iface_name(ifaces[i].iface()));
160
161 exit(EXIT_FAILURE);
162 }
163
164 static unsigned int mem_get_le16(const void *vmem) {
165 unsigned int val;
166 const unsigned char *mem = (const unsigned char *)vmem;
167
168 val = mem[1] << 8;
169 val |= mem[0];
170 return val;
171 }
172
173 static unsigned int mem_get_le32(const void *vmem) {
174 unsigned int val;
175 const unsigned char *mem = (const unsigned char *)vmem;
176
177 val = mem[3] << 24;
178 val |= mem[2] << 16;
179 val |= mem[1] << 8;
180 val |= mem[0];
181 return val;
182 }
183
184 enum file_kind {
185 RAW_FILE,
186 IVF_FILE,
187 WEBM_FILE
188 };
189
190 struct input_ctx {
191 enum file_kind kind;
192 FILE *infile;
193 nestegg *nestegg_ctx;
194 nestegg_packet *pkt;
195 unsigned int chunk;
196 unsigned int chunks;
197 unsigned int video_track;
198 };
199
200 #define IVF_FRAME_HDR_SZ (sizeof(uint32_t) + sizeof(uint64_t))
201 #define RAW_FRAME_HDR_SZ (sizeof(uint32_t))
202 static int read_frame(struct input_ctx *input,
203 uint8_t **buf,
204 size_t *buf_sz,
205 size_t *buf_alloc_sz) {
206 char raw_hdr[IVF_FRAME_HDR_SZ];
207 size_t new_buf_sz;
208 FILE *infile = input->infile;
209 enum file_kind kind = input->kind;
210 if (kind == WEBM_FILE) {
211 if (input->chunk >= input->chunks) {
212 unsigned int track;
213
214 do {
215 /* End of this packet, get another. */
216 if (input->pkt)
217 nestegg_free_packet(input->pkt);
218
219 if (nestegg_read_packet(input->nestegg_ctx, &input->pkt) <= 0
220 || nestegg_packet_track(input->pkt, &track))
221 return 1;
222
223 } while (track != input->video_track);
224
225 if (nestegg_packet_count(input->pkt, &input->chunks))
226 return 1;
227 input->chunk = 0;
228 }
229
230 if (nestegg_packet_data(input->pkt, input->chunk, buf, buf_sz))
231 return 1;
232 input->chunk++;
233
234 return 0;
235 }
236 /* For both the raw and ivf formats, the frame size is the first 4 bytes
237 * of the frame header. We just need to special case on the header
238 * size.
239 */
240 else if (fread(raw_hdr, kind == IVF_FILE
241 ? IVF_FRAME_HDR_SZ : RAW_FRAME_HDR_SZ, 1, infile) != 1) {
242 if (!feof(infile))
243 fprintf(stderr, "Failed to read frame size\n");
244
245 new_buf_sz = 0;
246 } else {
247 new_buf_sz = mem_get_le32(raw_hdr);
248
249 if (new_buf_sz > 256 * 1024 * 1024) {
250 fprintf(stderr, "Error: Read invalid frame size (%u)\n",
251 (unsigned int)new_buf_sz);
252 new_buf_sz = 0;
253 }
254
255 if (kind == RAW_FILE && new_buf_sz > 256 * 1024)
256 fprintf(stderr, "Warning: Read invalid frame size (%u)"
257 " - not a raw file?\n", (unsigned int)new_buf_sz);
258
259 if (new_buf_sz > *buf_alloc_sz) {
260 uint8_t *new_buf = realloc(*buf, 2 * new_buf_sz);
261
262 if (new_buf) {
263 *buf = new_buf;
264 *buf_alloc_sz = 2 * new_buf_sz;
265 } else {
266 fprintf(stderr, "Failed to allocate compressed data buffer\n");
267 new_buf_sz = 0;
268 }
269 }
270 }
271
272 *buf_sz = new_buf_sz;
273
274 if (!feof(infile)) {
275 if (fread(*buf, 1, *buf_sz, infile) != *buf_sz) {
276 fprintf(stderr, "Failed to read full frame\n");
277 return 1;
278 }
279
280 return 0;
281 }
282
283 return 1;
284 }
285
286 void *out_open(const char *out_fn, int do_md5) {
287 void *out = NULL;
288
289 if (do_md5) {
290 #if CONFIG_MD5
291 MD5Context *md5_ctx = out = malloc(sizeof(MD5Context));
292 (void)out_fn;
293 MD5Init(md5_ctx);
294 #endif
295 } else {
296 FILE *outfile = out = strcmp("-", out_fn) ? fopen(out_fn, "wb")
297 : set_binary_mode(stdout);
298
299 if (!outfile) {
300 fprintf(stderr, "Failed to output file");
301 exit(EXIT_FAILURE);
302 }
303 }
304
305 return out;
306 }
307
308 void out_put(void *out, const uint8_t *buf, unsigned int len, int do_md5) {
309 if (do_md5) {
310 #if CONFIG_MD5
311 MD5Update(out, buf, len);
312 #endif
313 } else {
314 (void) fwrite(buf, 1, len, out);
315 }
316 }
317
318 void out_close(void *out, const char *out_fn, int do_md5) {
319 if (do_md5) {
320 #if CONFIG_MD5
321 uint8_t md5[16];
322 int i;
323
324 MD5Final(md5, out);
325 free(out);
326
327 for (i = 0; i < 16; i++)
328 printf("%02x", md5[i]);
329
330 printf(" %s\n", out_fn);
331 #endif
332 } else {
333 fclose(out);
334 }
335 }
336
337 unsigned int file_is_ivf(FILE *infile,
338 unsigned int *fourcc,
339 unsigned int *width,
340 unsigned int *height,
341 unsigned int *fps_den,
342 unsigned int *fps_num) {
343 char raw_hdr[32];
344 int is_ivf = 0;
345
346 if (fread(raw_hdr, 1, 32, infile) == 32) {
347 if (raw_hdr[0] == 'D' && raw_hdr[1] == 'K'
348 && raw_hdr[2] == 'I' && raw_hdr[3] == 'F') {
349 is_ivf = 1;
350
351 if (mem_get_le16(raw_hdr + 4) != 0)
352 fprintf(stderr, "Error: Unrecognized IVF version! This file may not"
353 " decode properly.");
354
355 *fourcc = mem_get_le32(raw_hdr + 8);
356 *width = mem_get_le16(raw_hdr + 12);
357 *height = mem_get_le16(raw_hdr + 14);
358 *fps_num = mem_get_le32(raw_hdr + 16);
359 *fps_den = mem_get_le32(raw_hdr + 20);
360
361 /* Some versions of vpxenc used 1/(2*fps) for the timebase, so
362 * we can guess the framerate using only the timebase in this
363 * case. Other files would require reading ahead to guess the
364 * timebase, like we do for webm.
365 */
366 if (*fps_num < 1000) {
367 /* Correct for the factor of 2 applied to the timebase in the
368 * encoder.
369 */
370 if (*fps_num & 1)*fps_den <<= 1;
371 else *fps_num >>= 1;
372 } else {
373 /* Don't know FPS for sure, and don't have readahead code
374 * (yet?), so just default to 30fps.
375 */
376 *fps_num = 30;
377 *fps_den = 1;
378 }
379 }
380 }
381
382 if (!is_ivf)
383 rewind(infile);
384
385 return is_ivf;
386 }
387
388
389 unsigned int file_is_raw(FILE *infile,
390 unsigned int *fourcc,
391 unsigned int *width,
392 unsigned int *height,
393 unsigned int *fps_den,
394 unsigned int *fps_num) {
395 unsigned char buf[32];
396 int is_raw = 0;
397 vpx_codec_stream_info_t si;
398
399 si.sz = sizeof(si);
400
401 if (fread(buf, 1, 32, infile) == 32) {
402 int i;
403
404 if (mem_get_le32(buf) < 256 * 1024 * 1024)
405 for (i = 0; i < sizeof(ifaces) / sizeof(ifaces[0]); i++)
406 if (!vpx_codec_peek_stream_info(ifaces[i].iface(),
407 buf + 4, 32 - 4, &si)) {
408 is_raw = 1;
409 *fourcc = ifaces[i].fourcc;
410 *width = si.w;
411 *height = si.h;
412 *fps_num = 30;
413 *fps_den = 1;
414 break;
415 }
416 }
417
418 rewind(infile);
419 return is_raw;
420 }
421
422
423 static int
424 nestegg_read_cb(void *buffer, size_t length, void *userdata) {
425 FILE *f = userdata;
426
427 if (fread(buffer, 1, length, f) < length) {
428 if (ferror(f))
429 return -1;
430 if (feof(f))
431 return 0;
432 }
433 return 1;
434 }
435
436
437 static int
438 nestegg_seek_cb(int64_t offset, int whence, void *userdata) {
439 switch (whence) {
440 case NESTEGG_SEEK_SET:
441 whence = SEEK_SET;
442 break;
443 case NESTEGG_SEEK_CUR:
444 whence = SEEK_CUR;
445 break;
446 case NESTEGG_SEEK_END:
447 whence = SEEK_END;
448 break;
449 };
450 return fseek(userdata, (long)offset, whence) ? -1 : 0;
451 }
452
453
454 static int64_t
455 nestegg_tell_cb(void *userdata) {
456 return ftell(userdata);
457 }
458
459
460 static void
461 nestegg_log_cb(nestegg *context, unsigned int severity, char const *format,
462 ...) {
463 va_list ap;
464
465 va_start(ap, format);
466 vfprintf(stderr, format, ap);
467 fprintf(stderr, "\n");
468 va_end(ap);
469 }
470
471
472 static int
473 webm_guess_framerate(struct input_ctx *input,
474 unsigned int *fps_den,
475 unsigned int *fps_num) {
476 unsigned int i;
477 uint64_t tstamp = 0;
478
479 /* Check to see if we can seek before we parse any data. */
480 if (nestegg_track_seek(input->nestegg_ctx, input->video_track, 0)) {
481 fprintf(stderr,
482 "WARNING: Failed to guess framerate (no Cues), set to 30fps.\n");
483 *fps_num = 30;
484 *fps_den = 1;
485 return 0;
486 }
487
488 /* Guess the framerate. Read up to 1 second, or 50 video packets,
489 * whichever comes first.
490 */
491 for (i = 0; tstamp < 1000000000 && i < 50;) {
492 nestegg_packet *pkt;
493 unsigned int track;
494
495 if (nestegg_read_packet(input->nestegg_ctx, &pkt) <= 0)
496 break;
497
498 nestegg_packet_track(pkt, &track);
499 if (track == input->video_track) {
500 nestegg_packet_tstamp(pkt, &tstamp);
501 i++;
502 }
503
504 nestegg_free_packet(pkt);
505 }
506
507 if (nestegg_track_seek(input->nestegg_ctx, input->video_track, 0))
508 goto fail;
509
510 *fps_num = (i - 1) * 1000000;
511 *fps_den = (unsigned int)(tstamp / 1000);
512 return 0;
513 fail:
514 nestegg_destroy(input->nestegg_ctx);
515 input->nestegg_ctx = NULL;
516 rewind(input->infile);
517 return 1;
518 }
519
520
521 static int
522 file_is_webm(struct input_ctx *input,
523 unsigned int *fourcc,
524 unsigned int *width,
525 unsigned int *height,
526 unsigned int *fps_den,
527 unsigned int *fps_num) {
528 unsigned int i, n;
529 int track_type = -1;
530 int codec_id;
531
532 nestegg_io io = {nestegg_read_cb, nestegg_seek_cb, nestegg_tell_cb, 0};
533 nestegg_video_params params;
534
535 io.userdata = input->infile;
536 if (nestegg_init(&input->nestegg_ctx, io, NULL))
537 goto fail;
538
539 if (nestegg_track_count(input->nestegg_ctx, &n))
540 goto fail;
541
542 for (i = 0; i < n; i++) {
543 track_type = nestegg_track_type(input->nestegg_ctx, i);
544
545 if (track_type == NESTEGG_TRACK_VIDEO)
546 break;
547 else if (track_type < 0)
548 goto fail;
549 }
550
551 codec_id = nestegg_track_codec_id(input->nestegg_ctx, i);
552 if (codec_id == NESTEGG_CODEC_VP8) {
553 *fourcc = VP8_FOURCC_MASK;
554 } else if (codec_id == NESTEGG_CODEC_VP9) {
555 *fourcc = VP9_FOURCC_MASK;
556 } else {
557 fprintf(stderr, "Not VPx video, quitting.\n");
558 exit(1);
559 }
560
561 input->video_track = i;
562
563 if (nestegg_track_video_params(input->nestegg_ctx, i, &params))
564 goto fail;
565
566 *fps_den = 0;
567 *fps_num = 0;
568 *width = params.width;
569 *height = params.height;
570 return 1;
571 fail:
572 input->nestegg_ctx = NULL;
573 rewind(input->infile);
574 return 0;
575 }
576
577
578 void show_progress(int frame_in, int frame_out, unsigned long dx_time) {
579 fprintf(stderr, "%d decoded frames/%d showed frames in %lu us (%.2f fps)\r",
580 frame_in, frame_out, dx_time,
581 (float)frame_out * 1000000.0 / (float)dx_time);
582 }
583
584
585 void generate_filename(const char *pattern, char *out, size_t q_len,
586 unsigned int d_w, unsigned int d_h,
587 unsigned int frame_in) {
588 const char *p = pattern;
589 char *q = out;
590
591 do {
592 char *next_pat = strchr(p, '%');
593
594 if (p == next_pat) {
595 size_t pat_len;
596
597 /* parse the pattern */
598 q[q_len - 1] = '\0';
599 switch (p[1]) {
600 case 'w':
601 snprintf(q, q_len - 1, "%d", d_w);
602 break;
603 case 'h':
604 snprintf(q, q_len - 1, "%d", d_h);
605 break;
606 case '1':
607 snprintf(q, q_len - 1, "%d", frame_in);
608 break;
609 case '2':
610 snprintf(q, q_len - 1, "%02d", frame_in);
611 break;
612 case '3':
613 snprintf(q, q_len - 1, "%03d", frame_in);
614 break;
615 case '4':
616 snprintf(q, q_len - 1, "%04d", frame_in);
617 break;
618 case '5':
619 snprintf(q, q_len - 1, "%05d", frame_in);
620 break;
621 case '6':
622 snprintf(q, q_len - 1, "%06d", frame_in);
623 break;
624 case '7':
625 snprintf(q, q_len - 1, "%07d", frame_in);
626 break;
627 case '8':
628 snprintf(q, q_len - 1, "%08d", frame_in);
629 break;
630 case '9':
631 snprintf(q, q_len - 1, "%09d", frame_in);
632 break;
633 default:
634 die("Unrecognized pattern %%%c\n", p[1]);
635 }
636
637 pat_len = strlen(q);
638 if (pat_len >= q_len - 1)
639 die("Output filename too long.\n");
640 q += pat_len;
641 p += 2;
642 q_len -= pat_len;
643 } else {
644 size_t copy_len;
645
646 /* copy the next segment */
647 if (!next_pat)
648 copy_len = strlen(p);
649 else
650 copy_len = next_pat - p;
651
652 if (copy_len >= q_len - 1)
653 die("Output filename too long.\n");
654
655 memcpy(q, p, copy_len);
656 q[copy_len] = '\0';
657 q += copy_len;
658 p += copy_len;
659 q_len -= copy_len;
660 }
661 } while (*p);
662 }
663
664
665 int main_loop(int argc, const char **argv_) {
666 vpx_codec_ctx_t decoder;
667 char *fn = NULL;
668 int i;
669 uint8_t *buf = NULL;
670 size_t buf_sz = 0, buf_alloc_sz = 0;
671 FILE *infile;
672 int frame_in = 0, frame_out = 0, flipuv = 0, noblit = 0, do_md5 = 0, progress = 0;
673 int stop_after = 0, postproc = 0, summary = 0, quiet = 1;
674 int arg_skip = 0;
675 int ec_enabled = 0;
676 vpx_codec_iface_t *iface = NULL;
677 unsigned int fourcc;
678 unsigned long dx_time = 0;
679 struct arg arg;
680 char **argv, **argi, **argj;
681 const char *outfile_pattern = 0;
682 char outfile[PATH_MAX];
683 int single_file;
684 int use_y4m = 1;
685 unsigned int width;
686 unsigned int height;
687 unsigned int fps_den;
688 unsigned int fps_num;
689 void *out = NULL;
690 vpx_codec_dec_cfg_t cfg = {0};
691 #if CONFIG_VP8_DECODER
692 vp8_postproc_cfg_t vp8_pp_cfg = {0};
693 int vp8_dbg_color_ref_frame = 0;
694 int vp8_dbg_color_mb_modes = 0;
695 int vp8_dbg_color_b_modes = 0;
696 int vp8_dbg_display_mv = 0;
697 #endif
698 struct input_ctx input = {0};
699 int frames_corrupted = 0;
700 int dec_flags = 0;
701 int do_scale = 0;
702 int stream_w = 0, stream_h = 0;
703 vpx_image_t *scaled_img = NULL;
704 int frame_avail, got_data;
705
706 /* Parse command line */
707 exec_name = argv_[0];
708 argv = argv_dup(argc - 1, argv_ + 1);
709
710 for (argi = argj = argv; (*argj = *argi); argi += arg.argv_step) {
711 memset(&arg, 0, sizeof(arg));
712 arg.argv_step = 1;
713
714 if (arg_match(&arg, &codecarg, argi)) {
715 int j, k = -1;
716
717 for (j = 0; j < sizeof(ifaces) / sizeof(ifaces[0]); j++)
718 if (!strcmp(ifaces[j].name, arg.val))
719 k = j;
720
721 if (k >= 0)
722 iface = ifaces[k].iface();
723 else
724 die("Error: Unrecognized argument (%s) to --codec\n",
725 arg.val);
726 } else if (arg_match(&arg, &looparg, argi)) {
727 // no-op
728 } else if (arg_match(&arg, &outputfile, argi))
729 outfile_pattern = arg.val;
730 else if (arg_match(&arg, &use_yv12, argi)) {
731 use_y4m = 0;
732 flipuv = 1;
733 } else if (arg_match(&arg, &use_i420, argi)) {
734 use_y4m = 0;
735 flipuv = 0;
736 } else if (arg_match(&arg, &flipuvarg, argi))
737 flipuv = 1;
738 else if (arg_match(&arg, &noblitarg, argi))
739 noblit = 1;
740 else if (arg_match(&arg, &progressarg, argi))
741 progress = 1;
742 else if (arg_match(&arg, &limitarg, argi))
743 stop_after = arg_parse_uint(&arg);
744 else if (arg_match(&arg, &skiparg, argi))
745 arg_skip = arg_parse_uint(&arg);
746 else if (arg_match(&arg, &postprocarg, argi))
747 postproc = 1;
748 else if (arg_match(&arg, &md5arg, argi))
749 do_md5 = 1;
750 else if (arg_match(&arg, &summaryarg, argi))
751 summary = 1;
752 else if (arg_match(&arg, &threadsarg, argi))
753 cfg.threads = arg_parse_uint(&arg);
754 else if (arg_match(&arg, &verbosearg, argi))
755 quiet = 0;
756 else if (arg_match(&arg, &scalearg, argi))
757 do_scale = 1;
758
759 #if CONFIG_VP8_DECODER
760 else if (arg_match(&arg, &addnoise_level, argi)) {
761 postproc = 1;
762 vp8_pp_cfg.post_proc_flag |= VP8_ADDNOISE;
763 vp8_pp_cfg.noise_level = arg_parse_uint(&arg);
764 } else if (arg_match(&arg, &demacroblock_level, argi)) {
765 postproc = 1;
766 vp8_pp_cfg.post_proc_flag |= VP8_DEMACROBLOCK;
767 vp8_pp_cfg.deblocking_level = arg_parse_uint(&arg);
768 } else if (arg_match(&arg, &deblock, argi)) {
769 postproc = 1;
770 vp8_pp_cfg.post_proc_flag |= VP8_DEBLOCK;
771 } else if (arg_match(&arg, &mfqe, argi)) {
772 postproc = 1;
773 vp8_pp_cfg.post_proc_flag |= VP8_MFQE;
774 } else if (arg_match(&arg, &pp_debug_info, argi)) {
775 unsigned int level = arg_parse_uint(&arg);
776
777 postproc = 1;
778 vp8_pp_cfg.post_proc_flag &= ~0x7;
779
780 if (level)
781 vp8_pp_cfg.post_proc_flag |= level;
782 } else if (arg_match(&arg, &pp_disp_ref_frame, argi)) {
783 unsigned int flags = arg_parse_int(&arg);
784 if (flags) {
785 postproc = 1;
786 vp8_dbg_color_ref_frame = flags;
787 }
788 } else if (arg_match(&arg, &pp_disp_mb_modes, argi)) {
789 unsigned int flags = arg_parse_int(&arg);
790 if (flags) {
791 postproc = 1;
792 vp8_dbg_color_mb_modes = flags;
793 }
794 } else if (arg_match(&arg, &pp_disp_b_modes, argi)) {
795 unsigned int flags = arg_parse_int(&arg);
796 if (flags) {
797 postproc = 1;
798 vp8_dbg_color_b_modes = flags;
799 }
800 } else if (arg_match(&arg, &pp_disp_mvs, argi)) {
801 unsigned int flags = arg_parse_int(&arg);
802 if (flags) {
803 postproc = 1;
804 vp8_dbg_display_mv = flags;
805 }
806 } else if (arg_match(&arg, &error_concealment, argi)) {
807 ec_enabled = 1;
808 }
809
810 #endif
811 else
812 argj++;
813 }
814
815 /* Check for unrecognized options */
816 for (argi = argv; *argi; argi++)
817 if (argi[0][0] == '-' && strlen(argi[0]) > 1)
818 die("Error: Unrecognized option %s\n", *argi);
819
820 /* Handle non-option arguments */
821 fn = argv[0];
822
823 if (!fn)
824 usage_exit();
825
826 /* Open file */
827 infile = strcmp(fn, "-") ? fopen(fn, "rb") : set_binary_mode(stdin);
828
829 if (!infile) {
830 fprintf(stderr, "Failed to open file '%s'",
831 strcmp(fn, "-") ? fn : "stdin");
832 return EXIT_FAILURE;
833 }
834 #if CONFIG_OS_SUPPORT
835 /* Make sure we don't dump to the terminal, unless forced to with -o - */
836 if (!outfile_pattern && isatty(fileno(stdout)) && !do_md5 && !noblit) {
837 fprintf(stderr,
838 "Not dumping raw video to your terminal. Use '-o -' to "
839 "override.\n");
840 return EXIT_FAILURE;
841 }
842 #endif
843 input.infile = infile;
844 if (file_is_ivf(infile, &fourcc, &width, &height, &fps_den,
845 &fps_num))
846 input.kind = IVF_FILE;
847 else if (file_is_webm(&input, &fourcc, &width, &height, &fps_den, &fps_num))
848 input.kind = WEBM_FILE;
849 else if (file_is_raw(infile, &fourcc, &width, &height, &fps_den, &fps_num))
850 input.kind = RAW_FILE;
851 else {
852 fprintf(stderr, "Unrecognized input file type.\n");
853 return EXIT_FAILURE;
854 }
855
856 /* If the output file is not set or doesn't have a sequence number in
857 * it, then we only open it once.
858 */
859 outfile_pattern = outfile_pattern ? outfile_pattern : "-";
860 single_file = 1;
861 {
862 const char *p = outfile_pattern;
863 do {
864 p = strchr(p, '%');
865 if (p && p[1] >= '1' && p[1] <= '9') {
866 /* pattern contains sequence number, so it's not unique. */
867 single_file = 0;
868 break;
869 }
870 if (p)
871 p++;
872 } while (p);
873 }
874
875 if (single_file && !noblit) {
876 generate_filename(outfile_pattern, outfile, sizeof(outfile) - 1,
877 width, height, 0);
878 out = out_open(outfile, do_md5);
879 }
880
881 if (use_y4m && !noblit) {
882 char buffer[128];
883
884 if (!single_file) {
885 fprintf(stderr, "YUV4MPEG2 not supported with output patterns,"
886 " try --i420 or --yv12.\n");
887 return EXIT_FAILURE;
888 }
889
890 if (input.kind == WEBM_FILE)
891 if (webm_guess_framerate(&input, &fps_den, &fps_num)) {
892 fprintf(stderr, "Failed to guess framerate -- error parsing "
893 "webm file?\n");
894 return EXIT_FAILURE;
895 }
896
897
898 /*Note: We can't output an aspect ratio here because IVF doesn't
899 store one, and neither does VP8.
900 That will have to wait until these tools support WebM natively.*/
901 snprintf(buffer, sizeof(buffer), "YUV4MPEG2 W%u H%u F%u:%u I%c ",
902 width, height, fps_num, fps_den, 'p');
903 out_put(out, (unsigned char *)buffer,
904 (unsigned int)strlen(buffer), do_md5);
905 }
906
907 /* Try to determine the codec from the fourcc. */
908 for (i = 0; i < sizeof(ifaces) / sizeof(ifaces[0]); i++)
909 if ((fourcc & ifaces[i].fourcc_mask) == ifaces[i].fourcc) {
910 vpx_codec_iface_t *ivf_iface = ifaces[i].iface();
911
912 if (iface && iface != ivf_iface)
913 fprintf(stderr, "Notice -- IVF header indicates codec: %s\n",
914 ifaces[i].name);
915 else
916 iface = ivf_iface;
917
918 break;
919 }
920
921 dec_flags = (postproc ? VPX_CODEC_USE_POSTPROC : 0) |
922 (ec_enabled ? VPX_CODEC_USE_ERROR_CONCEALMENT : 0);
923 if (vpx_codec_dec_init(&decoder, iface ? iface : ifaces[0].iface(), &cfg,
924 dec_flags)) {
925 fprintf(stderr, "Failed to initialize decoder: %s\n", vpx_codec_error(&decoder));
926 return EXIT_FAILURE;
927 }
928
929 if (!quiet)
930 fprintf(stderr, "%s\n", decoder.name);
931
932 #if CONFIG_VP8_DECODER
933
934 if (vp8_pp_cfg.post_proc_flag
935 && vpx_codec_control(&decoder, VP8_SET_POSTPROC, &vp8_pp_cfg)) {
936 fprintf(stderr, "Failed to configure postproc: %s\n", vpx_codec_error(&decoder));
937 return EXIT_FAILURE;
938 }
939
940 if (vp8_dbg_color_ref_frame
941 && vpx_codec_control(&decoder, VP8_SET_DBG_COLOR_REF_FRAME, vp8_dbg_color_ref_frame)) {
942 fprintf(stderr, "Failed to configure reference block visualizer: %s\n", vpx_codec_error(&decoder));
943 return EXIT_FAILURE;
944 }
945
946 if (vp8_dbg_color_mb_modes
947 && vpx_codec_control(&decoder, VP8_SET_DBG_COLOR_MB_MODES, vp8_dbg_color_mb_modes)) {
948 fprintf(stderr, "Failed to configure macro block visualizer: %s\n", vpx_codec_error(&decoder));
949 return EXIT_FAILURE;
950 }
951
952 if (vp8_dbg_color_b_modes
953 && vpx_codec_control(&decoder, VP8_SET_DBG_COLOR_B_MODES, vp8_dbg_color_b_modes)) {
954 fprintf(stderr, "Failed to configure block visualizer: %s\n", vpx_codec_error(&decoder));
955 return EXIT_FAILURE;
956 }
957
958 if (vp8_dbg_display_mv
959 && vpx_codec_control(&decoder, VP8_SET_DBG_DISPLAY_MV, vp8_dbg_display_mv)) {
960 fprintf(stderr, "Failed to configure motion vector visualizer: %s\n", vpx_codec_error(&decoder));
961 return EXIT_FAILURE;
962 }
963 #endif
964
965
966 if(arg_skip)
967 fprintf(stderr, "Skiping first %d frames.\n", arg_skip);
968 while (arg_skip) {
969 if (read_frame(&input, &buf, &buf_sz, &buf_alloc_sz))
970 break;
971 arg_skip--;
972 }
973
974 frame_avail = 1;
975 got_data = 0;
976
977 /* Decode file */
978 while (frame_avail || got_data) {
979 vpx_codec_iter_t iter = NULL;
980 vpx_image_t *img;
981 struct vpx_usec_timer timer;
982 int corrupted;
983
984 frame_avail = 0;
985 if (!stop_after || frame_in < stop_after) {
986 if(!read_frame(&input, &buf, &buf_sz, &buf_alloc_sz)) {
987 frame_avail = 1;
988 frame_in++;
989
990 vpx_usec_timer_start(&timer);
991
992 if (vpx_codec_decode(&decoder, buf, (unsigned int)buf_sz, NULL, 0)) {
993 const char *detail = vpx_codec_error_detail(&decoder);
994 fprintf(stderr, "Failed to decode frame: %s\n",
995 vpx_codec_error(&decoder));
996
997 if (detail)
998 fprintf(stderr, " Additional information: %s\n", detail);
999 goto fail;
1000 }
1001
1002 vpx_usec_timer_mark(&timer);
1003 dx_time += (unsigned int)vpx_usec_timer_elapsed(&timer);
1004 }
1005 }
1006
1007 vpx_usec_timer_start(&timer);
1008
1009 got_data = 0;
1010 if ((img = vpx_codec_get_frame(&decoder, &iter))) {
1011 ++frame_out;
1012 got_data = 1;
1013 }
1014
1015 vpx_usec_timer_mark(&timer);
1016 dx_time += (unsigned int)vpx_usec_timer_elapsed(&timer);
1017
1018 if (vpx_codec_control(&decoder, VP8D_GET_FRAME_CORRUPTED, &corrupted)) {
1019 fprintf(stderr, "Failed VP8_GET_FRAME_CORRUPTED: %s\n",
1020 vpx_codec_error(&decoder));
1021 goto fail;
1022 }
1023 frames_corrupted += corrupted;
1024
1025 if (progress)
1026 show_progress(frame_in, frame_out, dx_time);
1027
1028 if (!noblit) {
1029 if (frame_out == 1 && img && use_y4m) {
1030 /* Write out the color format to terminate the header line */
1031 const char *color =
1032 img->fmt == VPX_IMG_FMT_444A ? "C444alpha\n" :
1033 img->fmt == VPX_IMG_FMT_I444 ? "C444\n" :
1034 img->fmt == VPX_IMG_FMT_I422 ? "C422\n" :
1035 "C420jpeg\n";
1036
1037 out_put(out, (const unsigned char*)color, strlen(color), do_md5);
1038 }
1039
1040 if (do_scale) {
1041 if (img && frame_out == 1) {
1042 stream_w = img->d_w;
1043 stream_h = img->d_h;
1044 scaled_img = vpx_img_alloc(NULL, VPX_IMG_FMT_I420,
1045 stream_w, stream_h, 16);
1046 }
1047 if (img && (img->d_w != stream_w || img->d_h != stream_h)) {
1048 assert(img->fmt == VPX_IMG_FMT_I420);
1049 I420Scale(img->planes[VPX_PLANE_Y], img->stride[VPX_PLANE_Y],
1050 img->planes[VPX_PLANE_U], img->stride[VPX_PLANE_U],
1051 img->planes[VPX_PLANE_V], img->stride[VPX_PLANE_V],
1052 img->d_w, img->d_h,
1053 scaled_img->planes[VPX_PLANE_Y],
1054 scaled_img->stride[VPX_PLANE_Y],
1055 scaled_img->planes[VPX_PLANE_U],
1056 scaled_img->stride[VPX_PLANE_U],
1057 scaled_img->planes[VPX_PLANE_V],
1058 scaled_img->stride[VPX_PLANE_V],
1059 stream_w, stream_h,
1060 kFilterBox);
1061 img = scaled_img;
1062 }
1063 }
1064
1065 if (img) {
1066 unsigned int y;
1067 char out_fn[PATH_MAX];
1068 uint8_t *buf;
1069 unsigned int c_w =
1070 img->x_chroma_shift ? (1 + img->d_w) >> img->x_chroma_shift
1071 : img->d_w;
1072 unsigned int c_h =
1073 img->y_chroma_shift ? (1 + img->d_h) >> img->y_chroma_shift
1074 : img->d_h;
1075
1076 if (!single_file) {
1077 size_t len = sizeof(out_fn) - 1;
1078
1079 out_fn[len] = '\0';
1080 generate_filename(outfile_pattern, out_fn, len - 1,
1081 img->d_w, img->d_h, frame_in);
1082 out = out_open(out_fn, do_md5);
1083 } else if (use_y4m)
1084 out_put(out, (unsigned char *)"FRAME\n", 6, do_md5);
1085
1086 buf = img->planes[VPX_PLANE_Y];
1087
1088 for (y = 0; y < img->d_h; y++) {
1089 out_put(out, buf, img->d_w, do_md5);
1090 buf += img->stride[VPX_PLANE_Y];
1091 }
1092
1093 buf = img->planes[flipuv ? VPX_PLANE_V : VPX_PLANE_U];
1094
1095 for (y = 0; y < c_h; y++) {
1096 out_put(out, buf, c_w, do_md5);
1097 buf += img->stride[VPX_PLANE_U];
1098 }
1099
1100 buf = img->planes[flipuv ? VPX_PLANE_U : VPX_PLANE_V];
1101
1102 for (y = 0; y < c_h; y++) {
1103 out_put(out, buf, c_w, do_md5);
1104 buf += img->stride[VPX_PLANE_V];
1105 }
1106
1107 if (!single_file)
1108 out_close(out, out_fn, do_md5);
1109 }
1110 }
1111
1112 if (stop_after && frame_in >= stop_after)
1113 break;
1114 }
1115
1116 if (summary || progress) {
1117 show_progress(frame_in, frame_out, dx_time);
1118 fprintf(stderr, "\n");
1119 }
1120
1121 if (frames_corrupted)
1122 fprintf(stderr, "WARNING: %d frames corrupted.\n", frames_corrupted);
1123
1124 fail:
1125
1126 if (vpx_codec_destroy(&decoder)) {
1127 fprintf(stderr, "Failed to destroy decoder: %s\n",
1128 vpx_codec_error(&decoder));
1129 return EXIT_FAILURE;
1130 }
1131
1132 if (single_file && !noblit)
1133 out_close(out, outfile, do_md5);
1134
1135 if (input.nestegg_ctx)
1136 nestegg_destroy(input.nestegg_ctx);
1137 if (input.kind != WEBM_FILE)
1138 free(buf);
1139
1140 if (scaled_img) vpx_img_free(scaled_img);
1141
1142 fclose(infile);
1143 free(argv);
1144
1145 return frames_corrupted ? EXIT_FAILURE : EXIT_SUCCESS;
1146 }
1147
1148 int main(int argc, const char **argv_) {
1149 unsigned int loops = 1, i;
1150 char **argv, **argi, **argj;
1151 struct arg arg;
1152 int error = 0;
1153
1154 argv = argv_dup(argc - 1, argv_ + 1);
1155 for (argi = argj = argv; (*argj = *argi); argi += arg.argv_step) {
1156 memset(&arg, 0, sizeof(arg));
1157 arg.argv_step = 1;
1158
1159 if (arg_match(&arg, &looparg, argi)) {
1160 loops = arg_parse_uint(&arg);
1161 break;
1162 }
1163 }
1164 free(argv);
1165 for (i = 0; !error && i < loops; i++)
1166 error = main_loop(argc, argv_);
1167 return error;
1168 }
vpx_image
Image Descriptor.
Definition: vpx_image.h:99
vpx_decoder.h
Describes the decoder algorithm interface to applications.
VPX_IMG_FMT_444A
Definition: vpx_image.h:61
vpx_codec_iface_name
const char * vpx_codec_iface_name(vpx_codec_iface_t *iface)
Return the name for a given interface.
VP8_SET_DBG_COLOR_REF_FRAME
Definition: vp8.h:48
vpx_codec_dec_cfg::threads
unsigned int threads
Definition: vpx_decoder.h:112
vpx_codec_stream_info::sz
unsigned int sz
Definition: vpx_decoder.h:93
vpx_codec_stream_info
Stream properties.
Definition: vpx_decoder.h:92
vp8_postproc_cfg::noise_level
int noise_level
Definition: vp8.h:88
vp8dx.h
Provides definitions for using the VP8 algorithm within the vpx Decoder interface.
vpx_codec_peek_stream_info
vpx_codec_err_t vpx_codec_peek_stream_info(vpx_codec_iface_t *iface, const uint8_t *data, unsigned int data_sz, vpx_codec_stream_info_t *si)
Parse stream info from a buffer.
VPX_PLANE_Y
#define VPX_PLANE_Y
Definition: vpx_image.h:116
vpx_codec_ctx::name
const char * name
Definition: vpx_codec.h:202
VPX_PLANE_V
#define VPX_PLANE_V
Definition: vpx_image.h:118
vpx_image::y_chroma_shift
unsigned int y_chroma_shift
Definition: vpx_image.h:112
vpx_image::x_chroma_shift
unsigned int x_chroma_shift
Definition: vpx_image.h:111
vpx_img_alloc
vpx_image_t * vpx_img_alloc(vpx_image_t *img, vpx_img_fmt_t fmt, unsigned int d_w, unsigned int d_h, unsigned int align)
Open a descriptor, allocating storage for the underlying image.
VPX_IMG_FMT_I420
Definition: vpx_image.h:56
vpx_image::d_w
unsigned int d_w
Definition: vpx_image.h:107
vpx_codec_dec_init
#define vpx_codec_dec_init(ctx, iface, cfg, flags)
Convenience macro for vpx_codec_dec_init_ver()
Definition: vpx_decoder.h:154
vpx_codec_decode
vpx_codec_err_t vpx_codec_decode(vpx_codec_ctx_t *ctx, const uint8_t *data, unsigned int data_sz, void *user_priv, long deadline)
Decode data.
vpx_image::stride
int stride[4]
Definition: vpx_image.h:128
vpx_img_free
void vpx_img_free(vpx_image_t *img)
Close an image descriptor.
vpx_image::fmt
vpx_img_fmt_t fmt
Definition: vpx_image.h:100
VPX_IMG_FMT_I422
Definition: vpx_image.h:59
vpx_image::planes
unsigned char * planes[4]
Definition: vpx_image.h:127
VP8_SET_POSTPROC
Definition: vp8.h:47
VPX_CODEC_USE_POSTPROC
#define VPX_CODEC_USE_POSTPROC
Definition: vpx_decoder.h:73
VP8D_GET_FRAME_CORRUPTED
Definition: vp8dx.h:63
vpx_codec_iface_t
const struct vpx_codec_iface vpx_codec_iface_t
Codec interface structure.
Definition: vpx_codec.h:175
vpx_codec_error_detail
const char * vpx_codec_error_detail(vpx_codec_ctx_t *ctx)
Retrieve detailed error information for codec context.
VPX_PLANE_U
#define VPX_PLANE_U
Definition: vpx_image.h:117
vp8_postproc_cfg::deblocking_level
int deblocking_level
Definition: vp8.h:87
VP8_SET_DBG_DISPLAY_MV
Definition: vp8.h:51
VP8_SET_DBG_COLOR_MB_MODES
Definition: vp8.h:49
vpx_codec_stream_info::w
unsigned int w
Definition: vpx_decoder.h:94
VPX_IMG_FMT_I444
Definition: vpx_image.h:60
VP8_SET_DBG_COLOR_B_MODES
Definition: vp8.h:50
vpx_codec_control
#define vpx_codec_control(ctx, id, data)
vpx_codec_control wrapper macro
Definition: vpx_codec.h:398
vpx_codec_destroy
vpx_codec_err_t vpx_codec_destroy(vpx_codec_ctx_t *ctx)
Destroy a codec instance.
vpx_image::d_h
unsigned int d_h
Definition: vpx_image.h:108
vp8_postproc_cfg
post process flags
Definition: vp8.h:85
vpx_codec_stream_info::h
unsigned int h
Definition: vpx_decoder.h:95
vpx_codec_error
const char * vpx_codec_error(vpx_codec_ctx_t *ctx)
Retrieve error synopsis for codec context.
vpx_codec_dec_cfg
Initialization Configurations.
Definition: vpx_decoder.h:111
vpx_codec_iter_t
const void * vpx_codec_iter_t
Iterator.
Definition: vpx_codec.h:190
vpx_codec_get_frame
vpx_image_t * vpx_codec_get_frame(vpx_codec_ctx_t *ctx, vpx_codec_iter_t *iter)
Decoded frames iterator.
vp8_postproc_cfg::post_proc_flag
int post_proc_flag
Definition: vp8.h:86
vpx_codec_ctx
Codec context structure.
Definition: vpx_codec.h:201