aomdec.c 34.7 KB
Newer Older
John Koleszar's avatar
John Koleszar committed
1
/*
2
 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
John Koleszar's avatar
John Koleszar committed
3
 *
4 5 6 7 8 9
 * This source code is subject to the terms of the BSD 2 Clause License and
 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
 * was not distributed with this source code in the LICENSE file, you can
 * obtain it at www.aomedia.org/license/software. If the Alliance for Open
 * Media Patent License 1.0 was not distributed with this source code in the
 * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
John Koleszar's avatar
John Koleszar committed
10 11
 */

12
#include <assert.h>
John Koleszar's avatar
John Koleszar committed
13 14 15 16
#include <stdio.h>
#include <stdlib.h>
#include <stdarg.h>
#include <string.h>
17
#include <limits.h>
18

19
#include "./aom_config.h"
20

James Zern's avatar
James Zern committed
21 22 23 24 25 26 27 28
#if CONFIG_OS_SUPPORT
#if HAVE_UNISTD_H
#include <unistd.h>  // NOLINT
#elif !defined(STDOUT_FILENO)
#define STDOUT_FILENO 1
#endif
#endif

29
#if CONFIG_LIBYUV
30
#include "third_party/libyuv/include/libyuv/scale.h"
31
#endif
32 33

#include "./args.h"
34 35
#include "./ivfdec.h"

36
#include "aom/aom_decoder.h"
37
#include "aom_ports/mem_ops.h"
38
#include "aom_ports/aom_timer.h"
39

40 41
#if CONFIG_AV1_DECODER
#include "aom/aomdx.h"
John Koleszar's avatar
John Koleszar committed
42
#endif
43

44
#include "./md5_utils.h"
45

46 47 48 49
#if CONFIG_OBU_NO_IVF
#include "./obudec.h"
#endif

50
#include "./tools_common.h"
51
#if CONFIG_WEBM_IO
52
#include "./webmdec.h"
53
#endif
54

55
#include "./y4menc.h"
John Koleszar's avatar
John Koleszar committed
56 57 58

static const char *exec_name;

59 60
struct AvxDecInputContext {
  struct AvxInputContext *aom_input_ctx;
61 62 63
  struct WebmInputContext *webm_ctx;
};

James Zern's avatar
James Zern committed
64 65
static const arg_def_t help =
    ARG_DEF(NULL, "help", 0, "Show usage options and exit");
clang-format's avatar
clang-format committed
66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104
static const arg_def_t looparg =
    ARG_DEF(NULL, "loops", 1, "Number of times to decode the file");
static const arg_def_t codecarg = ARG_DEF(NULL, "codec", 1, "Codec to use");
static const arg_def_t use_yv12 =
    ARG_DEF(NULL, "yv12", 0, "Output raw YV12 frames");
static const arg_def_t use_i420 =
    ARG_DEF(NULL, "i420", 0, "Output raw I420 frames");
static const arg_def_t flipuvarg =
    ARG_DEF(NULL, "flipuv", 0, "Flip the chroma planes in the output");
static const arg_def_t rawvideo =
    ARG_DEF(NULL, "rawvideo", 0, "Output raw YUV frames");
static const arg_def_t noblitarg =
    ARG_DEF(NULL, "noblit", 0, "Don't process the decoded frames");
static const arg_def_t progressarg =
    ARG_DEF(NULL, "progress", 0, "Show progress after each frame decodes");
static const arg_def_t limitarg =
    ARG_DEF(NULL, "limit", 1, "Stop decoding after n frames");
static const arg_def_t skiparg =
    ARG_DEF(NULL, "skip", 1, "Skip the first n input frames");
static const arg_def_t postprocarg =
    ARG_DEF(NULL, "postproc", 0, "Postprocess decoded frames");
static const arg_def_t summaryarg =
    ARG_DEF(NULL, "summary", 0, "Show timing summary");
static const arg_def_t outputfile =
    ARG_DEF("o", "output", 1, "Output file name pattern (see below)");
static const arg_def_t threadsarg =
    ARG_DEF("t", "threads", 1, "Max threads to use");
static const arg_def_t frameparallelarg =
    ARG_DEF(NULL, "frame-parallel", 0, "Frame parallel decode");
static const arg_def_t verbosearg =
    ARG_DEF("v", "verbose", 0, "Show version string");
static const arg_def_t scalearg =
    ARG_DEF("S", "scale", 0, "Scale output frames uniformly");
static const arg_def_t continuearg =
    ARG_DEF("k", "keep-going", 0, "(debug) Continue decoding after error");
static const arg_def_t fb_arg =
    ARG_DEF(NULL, "frame-buffers", 1, "Number of frame buffers to use");
static const arg_def_t md5arg =
    ARG_DEF(NULL, "md5", 0, "Compute the MD5 sum of the decoded frame");
105 106
static const arg_def_t framestatsarg =
    ARG_DEF(NULL, "framestats", 1, "Output per-frame stats (.csv format)");
clang-format's avatar
clang-format committed
107 108
static const arg_def_t outbitdeptharg =
    ARG_DEF(NULL, "output-bit-depth", 1, "Output bit-depth for decoded frames");
109
#if CONFIG_EXT_TILE
110 111 112
static const arg_def_t tilem = ARG_DEF(NULL, "tile-mode", 1,
                                       "Tile coding mode "
                                       "(0 for normal tile coding mode)");
clang-format's avatar
clang-format committed
113 114 115 116 117 118
static const arg_def_t tiler = ARG_DEF(NULL, "tile-row", 1,
                                       "Row index of tile to decode "
                                       "(-1 for all rows)");
static const arg_def_t tilec = ARG_DEF(NULL, "tile-column", 1,
                                       "Column index of tile to decode "
                                       "(-1 for all columns)");
119
#endif  // CONFIG_EXT_TILE
Dmitry Kovalev's avatar
Dmitry Kovalev committed
120

121 122 123 124 125 126
static const arg_def_t *all_args[] = {
  &help,        &codecarg,       &use_yv12,    &use_i420,   &flipuvarg,
  &rawvideo,    &noblitarg,      &progressarg, &limitarg,   &skiparg,
  &postprocarg, &summaryarg,     &outputfile,  &threadsarg, &frameparallelarg,
  &verbosearg,  &scalearg,       &fb_arg,      &md5arg,     &framestatsarg,
  &continuearg, &outbitdeptharg,
127
#if CONFIG_EXT_TILE
128
  &tilem,       &tiler,          &tilec,
129
#endif  // CONFIG_EXT_TILE
130 131
  NULL
};
John Koleszar's avatar
John Koleszar committed
132

133
#if CONFIG_LIBYUV
134
static INLINE int libyuv_scale(aom_image_t *src, aom_image_t *dst,
clang-format's avatar
clang-format committed
135
                               FilterModeEnum mode) {
136 137
  if (src->fmt == AOM_IMG_FMT_I42016) {
    assert(dst->fmt == AOM_IMG_FMT_I42016);
clang-format's avatar
clang-format committed
138
    return I420Scale_16(
139 140 141 142 143 144 145
        (uint16_t *)src->planes[AOM_PLANE_Y], src->stride[AOM_PLANE_Y] / 2,
        (uint16_t *)src->planes[AOM_PLANE_U], src->stride[AOM_PLANE_U] / 2,
        (uint16_t *)src->planes[AOM_PLANE_V], src->stride[AOM_PLANE_V] / 2,
        src->d_w, src->d_h, (uint16_t *)dst->planes[AOM_PLANE_Y],
        dst->stride[AOM_PLANE_Y] / 2, (uint16_t *)dst->planes[AOM_PLANE_U],
        dst->stride[AOM_PLANE_U] / 2, (uint16_t *)dst->planes[AOM_PLANE_V],
        dst->stride[AOM_PLANE_V] / 2, dst->d_w, dst->d_h, mode);
146
  }
147 148 149 150 151 152 153 154
  assert(src->fmt == AOM_IMG_FMT_I420);
  assert(dst->fmt == AOM_IMG_FMT_I420);
  return I420Scale(src->planes[AOM_PLANE_Y], src->stride[AOM_PLANE_Y],
                   src->planes[AOM_PLANE_U], src->stride[AOM_PLANE_U],
                   src->planes[AOM_PLANE_V], src->stride[AOM_PLANE_V], src->d_w,
                   src->d_h, dst->planes[AOM_PLANE_Y], dst->stride[AOM_PLANE_Y],
                   dst->planes[AOM_PLANE_U], dst->stride[AOM_PLANE_U],
                   dst->planes[AOM_PLANE_V], dst->stride[AOM_PLANE_V], dst->d_w,
clang-format's avatar
clang-format committed
155
                   dst->d_h, mode);
156
}
157
#endif
158

James Zern's avatar
James Zern committed
159 160
void show_help(FILE *fout, int shorthelp) {
  fprintf(fout, "Usage: %s <options> filename\n\n", exec_name);
John Koleszar's avatar
John Koleszar committed
161

James Zern's avatar
James Zern committed
162 163 164 165 166 167 168 169
  if (shorthelp) {
    fprintf(fout, "Use --help to see the full list of options.\n");
    return;
  }

  fprintf(fout, "Options:\n");
  arg_show_usage(fout, all_args);
  fprintf(fout,
John Koleszar's avatar
John Koleszar committed
170 171 172 173 174
          "\nOutput File Patterns:\n\n"
          "  The -o argument specifies the name of the file(s) to "
          "write to. If the\n  argument does not include any escape "
          "characters, the output will be\n  written to a single file. "
          "Otherwise, the filename will be calculated by\n  expanding "
John Koleszar's avatar
John Koleszar committed
175
          "the following escape characters:\n");
James Zern's avatar
James Zern committed
176
  fprintf(fout,
John Koleszar's avatar
John Koleszar committed
177 178 179 180 181
          "\n\t%%w   - Frame width"
          "\n\t%%h   - Frame height"
          "\n\t%%<n> - Frame number, zero padded to <n> places (1..9)"
          "\n\n  Pattern arguments are only supported in conjunction "
          "with the --yv12 and\n  --i420 options. If the -o option is "
clang-format's avatar
clang-format committed
182
          "not specified, the output will be\n  directed to stdout.\n");
James Zern's avatar
James Zern committed
183
  fprintf(fout, "\nIncluded decoders:\n\n");
John Koleszar's avatar
John Koleszar committed
184

James Zern's avatar
James Zern committed
185
  for (int i = 0; i < get_aom_decoder_count(); ++i) {
186
    const AvxInterface *const decoder = get_aom_decoder_by_index(i);
James Zern's avatar
James Zern committed
187
    fprintf(fout, "    %-6s - %s\n", decoder->name,
188
            aom_codec_iface_name(decoder->codec_interface()));
189
  }
James Zern's avatar
James Zern committed
190
}
John Koleszar's avatar
John Koleszar committed
191

James Zern's avatar
James Zern committed
192 193
void usage_exit(void) {
  show_help(stderr, 1);
John Koleszar's avatar
John Koleszar committed
194
  exit(EXIT_FAILURE);
John Koleszar's avatar
John Koleszar committed
195 196
}

clang-format's avatar
clang-format committed
197 198
static int raw_read_frame(FILE *infile, uint8_t **buffer, size_t *bytes_read,
                          size_t *buffer_size) {
199
  char raw_hdr[RAW_FRAME_HDR_SZ];
200
  size_t frame_size = 0;
John Koleszar's avatar
John Koleszar committed
201

202
  if (fread(raw_hdr, RAW_FRAME_HDR_SZ, 1, infile) != 1) {
clang-format's avatar
clang-format committed
203
    if (!feof(infile)) warn("Failed to read RAW frame size\n");
204
  } else {
205 206
    const size_t kCorruptFrameThreshold = 256 * 1024 * 1024;
    const size_t kFrameTooSmallThreshold = 256 * 1024;
207
    frame_size = mem_get_le32(raw_hdr);
John Koleszar's avatar
John Koleszar committed
208

209 210 211 212
    if (frame_size > kCorruptFrameThreshold) {
      warn("Read invalid frame size (%u)\n", (unsigned int)frame_size);
      frame_size = 0;
    }
John Koleszar's avatar
John Koleszar committed
213

214 215 216
    if (frame_size < kFrameTooSmallThreshold) {
      warn("Warning: Read invalid frame size (%u) - not a raw file?\n",
           (unsigned int)frame_size);
John Koleszar's avatar
John Koleszar committed
217
    }
John Koleszar's avatar
John Koleszar committed
218

219 220 221 222 223 224 225 226
    if (frame_size > *buffer_size) {
      uint8_t *new_buf = realloc(*buffer, 2 * frame_size);
      if (new_buf) {
        *buffer = new_buf;
        *buffer_size = 2 * frame_size;
      } else {
        warn("Failed to allocate compressed data buffer\n");
        frame_size = 0;
227
      }
John Koleszar's avatar
John Koleszar committed
228
    }
229
  }
John Koleszar's avatar
John Koleszar committed
230

231 232 233 234 235 236
  if (!feof(infile)) {
    if (fread(*buffer, 1, frame_size, infile) != frame_size) {
      warn("Failed to read full frame\n");
      return 1;
    }
    *bytes_read = frame_size;
John Koleszar's avatar
John Koleszar committed
237 238
  }

239 240 241
  return 0;
}

242
static int read_frame(struct AvxDecInputContext *input, uint8_t **buf,
243
                      size_t *bytes_in_buffer, size_t *buffer_size) {
244
  switch (input->aom_input_ctx->file_type) {
245
#if CONFIG_WEBM_IO
246
    case FILE_TYPE_WEBM:
247
      return webm_read_frame(input->webm_ctx, buf, bytes_in_buffer);
248
#endif
249
    case FILE_TYPE_RAW:
250
      return raw_read_frame(input->aom_input_ctx->file, buf, bytes_in_buffer,
clang-format's avatar
clang-format committed
251
                            buffer_size);
252
    case FILE_TYPE_IVF:
253
      return ivf_read_frame(input->aom_input_ctx->file, buf, bytes_in_buffer,
clang-format's avatar
clang-format committed
254
                            buffer_size);
255 256 257
#if CONFIG_OBU_NO_IVF
    case FILE_TYPE_OBU:
      return obu_read_temporal_unit(input->aom_input_ctx->file, buf,
Soo-Chul Han's avatar
Soo-Chul Han committed
258 259 260
#if CONFIG_SCALABILITY
                                    bytes_in_buffer, buffer_size, 0);
#else
261
                                    bytes_in_buffer, buffer_size);
Soo-Chul Han's avatar
Soo-Chul Han committed
262
#endif
263
#endif
clang-format's avatar
clang-format committed
264
    default: return 1;
265
  }
John Koleszar's avatar
John Koleszar committed
266 267
}

268
static void update_image_md5(const aom_image_t *img, const int planes[3],
269 270 271 272 273 274 275
                             MD5Context *md5) {
  int i, y;

  for (i = 0; i < 3; ++i) {
    const int plane = planes[i];
    const unsigned char *buf = img->planes[plane];
    const int stride = img->stride[plane];
276 277 278
    const int w = aom_img_plane_width(img, plane) *
                  ((img->fmt & AOM_IMG_FMT_HIGHBITDEPTH) ? 2 : 1);
    const int h = aom_img_plane_height(img, plane);
279 280 281 282 283 284 285 286

    for (y = 0; y < h; ++y) {
      MD5Update(md5, buf, w);
      buf += stride;
    }
  }
}

287 288
static void write_image_file(const aom_image_t *img, const int *planes,
                             const int num_planes, FILE *file) {
289
  int i, y;
290
  const int bytes_per_sample = ((img->fmt & AOM_IMG_FMT_HIGHBITDEPTH) ? 2 : 1);
291

292
  for (i = 0; i < num_planes; ++i) {
293 294 295
    const int plane = planes[i];
    const unsigned char *buf = img->planes[plane];
    const int stride = img->stride[plane];
296 297
    const int w = aom_img_plane_width(img, plane);
    const int h = aom_img_plane_height(img, plane);
298 299

    for (y = 0; y < h; ++y) {
300
      fwrite(buf, bytes_per_sample, w, file);
301 302
      buf += stride;
    }
John Koleszar's avatar
John Koleszar committed
303
  }
John Koleszar's avatar
John Koleszar committed
304 305
}

306
static int file_is_raw(struct AvxInputContext *input) {
307
  uint8_t buf[32];
John Koleszar's avatar
John Koleszar committed
308
  int is_raw = 0;
309
  aom_codec_stream_info_t si;
310

311
  if (fread(buf, 1, 32, input->file) == 32) {
John Koleszar's avatar
John Koleszar committed
312 313
    int i;

314
    if (mem_get_le32(buf) < 256 * 1024 * 1024) {
315 316 317
      for (i = 0; i < get_aom_decoder_count(); ++i) {
        const AvxInterface *const decoder = get_aom_decoder_by_index(i);
        if (!aom_codec_peek_stream_info(decoder->codec_interface(), buf + 4,
clang-format's avatar
clang-format committed
318
                                        32 - 4, &si)) {
John Koleszar's avatar
John Koleszar committed
319
          is_raw = 1;
320
          input->fourcc = decoder->fourcc;
321 322 323 324
          input->width = si.w;
          input->height = si.h;
          input->framerate.numerator = 30;
          input->framerate.denominator = 1;
John Koleszar's avatar
John Koleszar committed
325 326
          break;
        }
327 328
      }
    }
John Koleszar's avatar
John Koleszar committed
329 330
  }

331
  rewind(input->file);
John Koleszar's avatar
John Koleszar committed
332
  return is_raw;
333 334
}

335
static void show_progress(int frame_in, int frame_out, uint64_t dx_time) {
336
  fprintf(stderr,
clang-format's avatar
clang-format committed
337
          "%d decoded frames/%d showed frames in %" PRId64 " us (%.2f fps)\r",
John Koleszar's avatar
John Koleszar committed
338
          frame_in, frame_out, dx_time,
339
          (double)frame_out * 1000000.0 / (double)dx_time);
340 341
}

342
struct ExternalFrameBuffer {
clang-format's avatar
clang-format committed
343
  uint8_t *data;
344 345 346 347 348 349 350 351 352
  size_t size;
  int in_use;
};

struct ExternalFrameBufferList {
  int num_external_frame_buffers;
  struct ExternalFrameBuffer *ext_fb;
};

353
// Callback used by libaom to request an external frame buffer. |cb_priv|
354 355 356
// Application private data passed into the set function. |min_size| is the
// minimum size in bytes needed to decode the next frame. |fb| pointer to the
// frame buffer.
357 358
static int get_av1_frame_buffer(void *cb_priv, size_t min_size,
                                aom_codec_frame_buffer_t *fb) {
359 360 361
  int i;
  struct ExternalFrameBufferList *const ext_fb_list =
      (struct ExternalFrameBufferList *)cb_priv;
clang-format's avatar
clang-format committed
362
  if (ext_fb_list == NULL) return -1;
363 364 365

  // Find a free frame buffer.
  for (i = 0; i < ext_fb_list->num_external_frame_buffers; ++i) {
clang-format's avatar
clang-format committed
366
    if (!ext_fb_list->ext_fb[i].in_use) break;
367 368
  }

clang-format's avatar
clang-format committed
369
  if (i == ext_fb_list->num_external_frame_buffers) return -1;
370 371 372

  if (ext_fb_list->ext_fb[i].size < min_size) {
    free(ext_fb_list->ext_fb[i].data);
373
    ext_fb_list->ext_fb[i].data = (uint8_t *)calloc(min_size, sizeof(uint8_t));
clang-format's avatar
clang-format committed
374
    if (!ext_fb_list->ext_fb[i].data) return -1;
375 376 377 378 379 380 381 382 383 384 385 386 387

    ext_fb_list->ext_fb[i].size = min_size;
  }

  fb->data = ext_fb_list->ext_fb[i].data;
  fb->size = ext_fb_list->ext_fb[i].size;
  ext_fb_list->ext_fb[i].in_use = 1;

  // Set the frame buffer's private data to point at the external frame buffer.
  fb->priv = &ext_fb_list->ext_fb[i];
  return 0;
}

388
// Callback used by libaom when there are no references to the frame buffer.
389 390
// |cb_priv| user private data passed into the set function. |fb| pointer
// to the frame buffer.
391 392
static int release_av1_frame_buffer(void *cb_priv,
                                    aom_codec_frame_buffer_t *fb) {
393 394 395 396 397 398 399
  struct ExternalFrameBuffer *const ext_fb =
      (struct ExternalFrameBuffer *)fb->priv;
  (void)cb_priv;
  ext_fb->in_use = 0;
  return 0;
}

400 401 402
static void generate_filename(const char *pattern, char *out, size_t q_len,
                              unsigned int d_w, unsigned int d_h,
                              unsigned int frame_in) {
John Koleszar's avatar
John Koleszar committed
403 404 405 406 407 408 409 410 411
  const char *p = pattern;
  char *q = out;

  do {
    char *next_pat = strchr(p, '%');

    if (p == next_pat) {
      size_t pat_len;

John Koleszar's avatar
John Koleszar committed
412
      /* parse the pattern */
John Koleszar's avatar
John Koleszar committed
413 414
      q[q_len - 1] = '\0';
      switch (p[1]) {
clang-format's avatar
clang-format committed
415 416 417 418 419 420 421 422 423 424 425 426
        case 'w': snprintf(q, q_len - 1, "%d", d_w); break;
        case 'h': snprintf(q, q_len - 1, "%d", d_h); break;
        case '1': snprintf(q, q_len - 1, "%d", frame_in); break;
        case '2': snprintf(q, q_len - 1, "%02d", frame_in); break;
        case '3': snprintf(q, q_len - 1, "%03d", frame_in); break;
        case '4': snprintf(q, q_len - 1, "%04d", frame_in); break;
        case '5': snprintf(q, q_len - 1, "%05d", frame_in); break;
        case '6': snprintf(q, q_len - 1, "%06d", frame_in); break;
        case '7': snprintf(q, q_len - 1, "%07d", frame_in); break;
        case '8': snprintf(q, q_len - 1, "%08d", frame_in); break;
        case '9': snprintf(q, q_len - 1, "%09d", frame_in); break;
        default: die("Unrecognized pattern %%%c\n", p[1]); break;
John Koleszar's avatar
John Koleszar committed
427 428 429
      }

      pat_len = strlen(q);
clang-format's avatar
clang-format committed
430
      if (pat_len >= q_len - 1) die("Output filename too long.\n");
John Koleszar's avatar
John Koleszar committed
431 432 433 434 435 436
      q += pat_len;
      p += 2;
      q_len -= pat_len;
    } else {
      size_t copy_len;

John Koleszar's avatar
John Koleszar committed
437
      /* copy the next segment */
John Koleszar's avatar
John Koleszar committed
438 439 440 441 442
      if (!next_pat)
        copy_len = strlen(p);
      else
        copy_len = next_pat - p;

clang-format's avatar
clang-format committed
443
      if (copy_len >= q_len - 1) die("Output filename too long.\n");
John Koleszar's avatar
John Koleszar committed
444 445 446 447 448 449 450 451

      memcpy(q, p, copy_len);
      q[copy_len] = '\0';
      q += copy_len;
      p += copy_len;
      q_len -= copy_len;
    }
  } while (*p);
452 453
}

454 455 456 457 458 459 460
static int is_single_file(const char *outfile_pattern) {
  const char *p = outfile_pattern;

  do {
    p = strchr(p, '%');
    if (p && p[1] >= '1' && p[1] <= '9')
      return 0;  // pattern contains sequence number, so it's not unique
clang-format's avatar
clang-format committed
461
    if (p) p++;
462 463 464 465 466 467 468 469
  } while (p);

  return 1;
}

static void print_md5(unsigned char digest[16], const char *filename) {
  int i;

clang-format's avatar
clang-format committed
470
  for (i = 0; i < 16; ++i) printf("%02x", digest[i]);
471 472 473 474 475 476 477 478 479
  printf("  %s\n", filename);
}

static FILE *open_outfile(const char *name) {
  if (strcmp("-", name) == 0) {
    set_binary_mode(stdout);
    return stdout;
  } else {
    FILE *file = fopen(name, "wb");
clang-format's avatar
clang-format committed
480
    if (!file) fatal("Failed to open output file '%s'", name);
481 482 483 484
    return file;
  }
}

485 486 487
static int img_shifted_realloc_required(const aom_image_t *img,
                                        const aom_image_t *shifted,
                                        aom_img_fmt_t required_fmt) {
clang-format's avatar
clang-format committed
488
  return img->d_w != shifted->d_w || img->d_h != shifted->d_h ||
489 490
         required_fmt != shifted->fmt;
}
491

492
static int main_loop(int argc, const char **argv_) {
493
  aom_codec_ctx_t decoder;
clang-format's avatar
clang-format committed
494 495
  char *fn = NULL;
  int i;
James Zern's avatar
James Zern committed
496
  int ret = EXIT_FAILURE;
clang-format's avatar
clang-format committed
497 498 499 500 501 502 503 504
  uint8_t *buf = NULL;
  size_t bytes_in_buffer = 0, buffer_size = 0;
  FILE *infile;
  int frame_in = 0, frame_out = 0, flipuv = 0, noblit = 0;
  int do_md5 = 0, progress = 0, frame_parallel = 0;
  int stop_after = 0, postproc = 0, summary = 0, quiet = 1;
  int arg_skip = 0;
  int keep_going = 0;
505 506
  const AvxInterface *interface = NULL;
  const AvxInterface *fourcc_interface = NULL;
507
  uint64_t dx_time = 0;
clang-format's avatar
clang-format committed
508 509 510 511 512 513 514
  struct arg arg;
  char **argv, **argi, **argj;

  int single_file;
  int use_y4m = 1;
  int opt_yv12 = 0;
  int opt_i420 = 0;
515
  int opt_raw = 0;
516 517 518
  aom_codec_dec_cfg_t cfg = {
    0, 0, 0, CONFIG_LOWBITDEPTH, { CONFIG_EXT_PARTITION }
  };
clang-format's avatar
clang-format committed
519
  unsigned int output_bit_depth = 0;
520
#if CONFIG_EXT_TILE
521
  unsigned int tile_mode = 0;
clang-format's avatar
clang-format committed
522 523
  int tile_row = -1;
  int tile_col = -1;
524
#endif  // CONFIG_EXT_TILE
clang-format's avatar
clang-format committed
525 526 527
  int frames_corrupted = 0;
  int dec_flags = 0;
  int do_scale = 0;
528 529
  aom_image_t *scaled_img = NULL;
  aom_image_t *img_shifted = NULL;
clang-format's avatar
clang-format committed
530 531 532
  int frame_avail, got_data, flush_decoder = 0;
  int num_external_frame_buffers = 0;
  struct ExternalFrameBufferList ext_fb_list = { 0, NULL };
John Koleszar's avatar
John Koleszar committed
533

534
  const char *outfile_pattern = NULL;
clang-format's avatar
clang-format committed
535
  char outfile_name[PATH_MAX] = { 0 };
536 537
  FILE *outfile = NULL;

538 539
  FILE *framestats_file = NULL;

540 541 542
  MD5Context md5_ctx;
  unsigned char md5_digest[16];

543 544
  struct AvxDecInputContext input = { NULL, NULL };
  struct AvxInputContext aom_input_ctx;
545
#if CONFIG_WEBM_IO
546 547
  struct WebmInputContext webm_ctx;
  memset(&(webm_ctx), 0, sizeof(webm_ctx));
548
  input.webm_ctx = &webm_ctx;
549
#endif
550
  input.aom_input_ctx = &aom_input_ctx;
551

John Koleszar's avatar
John Koleszar committed
552 553 554 555 556 557 558 559
  /* Parse command line */
  exec_name = argv_[0];
  argv = argv_dup(argc - 1, argv_ + 1);

  for (argi = argj = argv; (*argj = *argi); argi += arg.argv_step) {
    memset(&arg, 0, sizeof(arg));
    arg.argv_step = 1;

James Zern's avatar
James Zern committed
560 561 562 563
    if (arg_match(&arg, &help, argi)) {
      show_help(stdout, 0);
      exit(EXIT_SUCCESS);
    } else if (arg_match(&arg, &codecarg, argi)) {
564
      interface = get_aom_decoder_by_name(arg.val);
565 566
      if (!interface)
        die("Error: Unrecognized argument (%s) to --codec\n", arg.val);
John Koleszar's avatar
John Koleszar committed
567 568
    } else if (arg_match(&arg, &looparg, argi)) {
      // no-op
569
    } else if (arg_match(&arg, &outputfile, argi)) {
John Koleszar's avatar
John Koleszar committed
570
      outfile_pattern = arg.val;
571
    } else if (arg_match(&arg, &use_yv12, argi)) {
John Koleszar's avatar
John Koleszar committed
572 573
      use_y4m = 0;
      flipuv = 1;
574
      opt_yv12 = 1;
575 576
      opt_i420 = 0;
      opt_raw = 0;
John Koleszar's avatar
John Koleszar committed
577 578 579
    } else if (arg_match(&arg, &use_i420, argi)) {
      use_y4m = 0;
      flipuv = 0;
580
      opt_yv12 = 0;
581
      opt_i420 = 1;
582
      opt_raw = 0;
583 584
    } else if (arg_match(&arg, &rawvideo, argi)) {
      use_y4m = 0;
585 586 587
      opt_yv12 = 0;
      opt_i420 = 0;
      opt_raw = 1;
588
    } else if (arg_match(&arg, &flipuvarg, argi)) {
John Koleszar's avatar
John Koleszar committed
589
      flipuv = 1;
590
    } else if (arg_match(&arg, &noblitarg, argi)) {
John Koleszar's avatar
John Koleszar committed
591
      noblit = 1;
592
    } else if (arg_match(&arg, &progressarg, argi)) {
John Koleszar's avatar
John Koleszar committed
593
      progress = 1;
594
    } else if (arg_match(&arg, &limitarg, argi)) {
John Koleszar's avatar
John Koleszar committed
595
      stop_after = arg_parse_uint(&arg);
596
    } else if (arg_match(&arg, &skiparg, argi)) {
597
      arg_skip = arg_parse_uint(&arg);
598
    } else if (arg_match(&arg, &postprocarg, argi)) {
John Koleszar's avatar
John Koleszar committed
599
      postproc = 1;
600
    } else if (arg_match(&arg, &md5arg, argi)) {
John Koleszar's avatar
John Koleszar committed
601
      do_md5 = 1;
602 603 604 605 606 607 608
    } else if (arg_match(&arg, &framestatsarg, argi)) {
      framestats_file = fopen(arg.val, "w");
      if (!framestats_file) {
        die("Error: Could not open --framestats file (%s) for writing.\n",
            arg.val);
      }
    } else if (arg_match(&arg, &summaryarg, argi)) {
John Koleszar's avatar
John Koleszar committed
609
      summary = 1;
610
    } else if (arg_match(&arg, &threadsarg, argi)) {
John Koleszar's avatar
John Koleszar committed
611
      cfg.threads = arg_parse_uint(&arg);
612
    }
613
#if CONFIG_AV1_DECODER
614 615 616
    else if (arg_match(&arg, &frameparallelarg, argi))
      frame_parallel = 1;
#endif
John Koleszar's avatar
John Koleszar committed
617 618
    else if (arg_match(&arg, &verbosearg, argi))
      quiet = 0;
619 620
    else if (arg_match(&arg, &scalearg, argi))
      do_scale = 1;
621 622
    else if (arg_match(&arg, &fb_arg, argi))
      num_external_frame_buffers = arg_parse_uint(&arg);
623 624
    else if (arg_match(&arg, &continuearg, argi))
      keep_going = 1;
625 626 627
    else if (arg_match(&arg, &outbitdeptharg, argi)) {
      output_bit_depth = arg_parse_uint(&arg);
    }
628
#if CONFIG_EXT_TILE
629 630
    else if (arg_match(&arg, &tilem, argi))
      tile_mode = arg_parse_int(&arg);
631 632 633 634 635
    else if (arg_match(&arg, &tiler, argi))
      tile_row = arg_parse_int(&arg);
    else if (arg_match(&arg, &tilec, argi))
      tile_col = arg_parse_int(&arg);
#endif  // CONFIG_EXT_TILE
John Koleszar's avatar
John Koleszar committed
636 637 638
    else
      argj++;
  }
John Koleszar's avatar
John Koleszar committed
639

John Koleszar's avatar
John Koleszar committed
640 641 642 643
  /* Check for unrecognized options */
  for (argi = argv; *argi; argi++)
    if (argi[0][0] == '-' && strlen(argi[0]) > 1)
      die("Error: Unrecognized option %s\n", *argi);
John Koleszar's avatar
John Koleszar committed
644

John Koleszar's avatar
John Koleszar committed
645 646
  /* Handle non-option arguments */
  fn = argv[0];
John Koleszar's avatar
John Koleszar committed
647

hanno's avatar
hanno committed
648 649
  if (!fn) {
    free(argv);
James Zern's avatar
James Zern committed
650
    fprintf(stderr, "No input file specified!\n");
John Koleszar's avatar
John Koleszar committed
651
    usage_exit();
hanno's avatar
hanno committed
652
  }
John Koleszar's avatar
John Koleszar committed
653 654
  /* Open file */
  infile = strcmp(fn, "-") ? fopen(fn, "rb") : set_binary_mode(stdin);
John Koleszar's avatar
John Koleszar committed
655

John Koleszar's avatar
John Koleszar committed
656
  if (!infile) {
657
    fatal("Failed to open input file '%s'", strcmp(fn, "-") ? fn : "stdin");
John Koleszar's avatar
John Koleszar committed
658
  }
659
#if CONFIG_OS_SUPPORT
John Koleszar's avatar
John Koleszar committed
660
  /* Make sure we don't dump to the terminal, unless forced to with -o - */
James Zern's avatar
James Zern committed
661
  if (!outfile_pattern && isatty(STDOUT_FILENO) && !do_md5 && !noblit) {
John Koleszar's avatar
John Koleszar committed
662 663 664 665 666
    fprintf(stderr,
            "Not dumping raw video to your terminal. Use '-o -' to "
            "override.\n");
    return EXIT_FAILURE;
  }
667
#endif
668 669 670
  input.aom_input_ctx->file = infile;
  if (file_is_ivf(input.aom_input_ctx))
    input.aom_input_ctx->file_type = FILE_TYPE_IVF;
671
#if CONFIG_WEBM_IO
672 673
  else if (file_is_webm(input.webm_ctx, input.aom_input_ctx))
    input.aom_input_ctx->file_type = FILE_TYPE_WEBM;
674 675 676 677
#endif
#if CONFIG_OBU_NO_IVF
  else if (file_is_obu(input.aom_input_ctx))
    input.aom_input_ctx->file_type = FILE_TYPE_OBU;
678
#endif
679 680
  else if (file_is_raw(input.aom_input_ctx))
    input.aom_input_ctx->file_type = FILE_TYPE_RAW;
John Koleszar's avatar
John Koleszar committed
681 682
  else {
    fprintf(stderr, "Unrecognized input file type.\n");
683
#if !CONFIG_WEBM_IO
684
    fprintf(stderr, "aomdec was built without WebM container support.\n");
685
#endif
John Koleszar's avatar
John Koleszar committed
686 687 688 689
    return EXIT_FAILURE;
  }

  outfile_pattern = outfile_pattern ? outfile_pattern : "-";
690
  single_file = is_single_file(outfile_pattern);
John Koleszar's avatar
John Koleszar committed
691

692 693
  if (!noblit && single_file) {
    generate_filename(outfile_pattern, outfile_name, PATH_MAX,
694
                      aom_input_ctx.width, aom_input_ctx.height, 0);
695 696 697 698
    if (do_md5)
      MD5Init(&md5_ctx);
    else
      outfile = open_outfile(outfile_name);
John Koleszar's avatar
John Koleszar committed
699 700 701 702
  }

  if (use_y4m && !noblit) {
    if (!single_file) {
clang-format's avatar
clang-format committed
703 704
      fprintf(stderr,
              "YUV4MPEG2 not supported with output patterns,"
Deb Mukherjee's avatar
Deb Mukherjee committed
705
              " try --i420 or --yv12 or --rawvideo.\n");
John Koleszar's avatar
John Koleszar committed
706
      return EXIT_FAILURE;
707
    }
John Koleszar's avatar
John Koleszar committed
708

709
#if CONFIG_WEBM_IO
710 711
    if (aom_input_ctx.file_type == FILE_TYPE_WEBM) {
      if (webm_guess_framerate(input.webm_ctx, input.aom_input_ctx)) {
clang-format's avatar
clang-format committed
712 713
        fprintf(stderr,
                "Failed to guess framerate -- error parsing "
John Koleszar's avatar
John Koleszar committed
714 715 716
                "webm file?\n");
        return EXIT_FAILURE;
      }
717
    }
718
#endif
John Koleszar's avatar
John Koleszar committed
719 720
  }

721
  fourcc_interface = get_aom_decoder_by_fourcc(aom_input_ctx.fourcc);
722 723 724 725
  if (interface && fourcc_interface && interface != fourcc_interface)
    warn("Header indicates codec: %s\n", fourcc_interface->name);
  else
    interface = fourcc_interface;
John Koleszar's avatar
John Koleszar committed
726

727
  if (!interface) interface = get_aom_decoder_by_index(0);
728

729 730 731
  dec_flags = (postproc ? AOM_CODEC_USE_POSTPROC : 0) |
              (frame_parallel ? AOM_CODEC_USE_FRAME_THREADING : 0);
  if (aom_codec_dec_init(&decoder, interface->codec_interface(), &cfg,
clang-format's avatar
clang-format committed
732
                         dec_flags)) {
733
    fprintf(stderr, "Failed to initialize decoder: %s\n",
734
            aom_codec_error(&decoder));
James Zern's avatar
James Zern committed
735
    goto fail2;
John Koleszar's avatar
John Koleszar committed
736
  }
John Koleszar's avatar
John Koleszar committed
737

clang-format's avatar
clang-format committed
738
  if (!quiet) fprintf(stderr, "%s\n", decoder.name);
John Koleszar's avatar
John Koleszar committed
739

740
#if CONFIG_AV1_DECODER && CONFIG_EXT_TILE
741 742 743 744 745 746
  if (aom_codec_control(&decoder, AV1_SET_TILE_MODE, tile_mode)) {
    fprintf(stderr, "Failed to set decode_tile_mode: %s\n",
            aom_codec_error(&decoder));
    goto fail;
  }

747 748 749 750 751
  if (aom_codec_control(&decoder, AV1_SET_DECODE_TILE_ROW, tile_row)) {
    fprintf(stderr, "Failed to set decode_tile_row: %s\n",
            aom_codec_error(&decoder));
    goto fail;
  }
752

753 754 755 756
  if (aom_codec_control(&decoder, AV1_SET_DECODE_TILE_COL, tile_col)) {
    fprintf(stderr, "Failed to set decode_tile_col: %s\n",
            aom_codec_error(&decoder));
    goto fail;
757 758
  }
#endif
759

clang-format's avatar
clang-format committed
760
  if (arg_skip) fprintf(stderr, "Skipping first %d frames.\n", arg_skip);
761
  while (arg_skip) {
clang-format's avatar
clang-format committed
762
    if (read_frame(&input, &buf, &bytes_in_buffer, &buffer_size)) break;
763 764 765
    arg_skip--;
  }

766 767 768 769
  if (num_external_frame_buffers > 0) {
    ext_fb_list.num_external_frame_buffers = num_external_frame_buffers;
    ext_fb_list.ext_fb = (struct ExternalFrameBuffer *)calloc(
        num_external_frame_buffers, sizeof(*ext_fb_list.ext_fb));
770 771
    if (aom_codec_set_frame_buffer_functions(&decoder, get_av1_frame_buffer,
                                             release_av1_frame_buffer,
clang-format's avatar
clang-format committed
772
                                             &ext_fb_list)) {
773
      fprintf(stderr, "Failed to configure external frame buffers: %s\n",
774
              aom_codec_error(&decoder));
James Zern's avatar
James Zern committed
775
      goto fail;
776 777 778
    }
  }

Scott LaVarnway's avatar
Scott LaVarnway committed
779 780 781
  frame_avail = 1;
  got_data = 0;

782 783
  if (framestats_file) fprintf(framestats_file, "bytes,qp\r\n");

John Koleszar's avatar
John Koleszar committed
784
  /* Decode file */
Scott LaVarnway's avatar
Scott LaVarnway committed
785
  while (frame_avail || got_data) {
786 787 788
    aom_codec_iter_t iter = NULL;
    aom_image_t *img;
    struct aom_usec_timer timer;
clang-format's avatar
clang-format committed
789
    int corrupted = 0;
John Koleszar's avatar
John Koleszar committed
790

Scott LaVarnway's avatar
Scott LaVarnway committed
791 792
    frame_avail = 0;
    if (!stop_after || frame_in < stop_after) {
793
      if (!read_frame(&input, &buf, &bytes_in_buffer, &buffer_size)) {
Scott LaVarnway's avatar
Scott LaVarnway committed
794 795
        frame_avail = 1;
        frame_in++;
John Koleszar's avatar
John Koleszar committed
796

797
        aom_usec_timer_start(&timer);
John Koleszar's avatar
John Koleszar committed
798

Sean DuBois's avatar
Sean DuBois committed
799 800
        if (aom_codec_decode(&decoder, buf, (unsigned int)bytes_in_buffer,
                             NULL)) {
801
          const char *detail = aom_codec_error_detail(&decoder);
clang-format's avatar
clang-format committed
802
          warn("Failed to decode frame %d: %s", frame_in,
803
               aom_codec_error(&decoder));
804

clang-format's avatar
clang-format committed
805 806
          if (detail) warn("Additional information: %s", detail);
          if (!keep_going) goto fail;
Scott LaVarnway's avatar
Scott LaVarnway committed
807 808
        }

809 810 811 812 813 814 815 816 817 818
        if (framestats_file) {
          int qp;
          if (aom_codec_control(&decoder, AOMD_GET_LAST_QUANTIZER, &qp)) {
            warn("Failed AOMD_GET_LAST_QUANTIZER: %s",
                 aom_codec_error(&decoder));
            if (!keep_going) goto fail;
          }
          fprintf(framestats_file, "%d,%d\r\n", (int)bytes_in_buffer, qp);
        }

819 820
        aom_usec_timer_mark(&timer);
        dx_time += aom_usec_timer_elapsed(&timer);
821 822
      } else {
        flush_decoder = 1;
Scott LaVarnway's avatar
Scott LaVarnway committed
823
      }
824 825
    } else {
      flush_decoder = 1;
Scott LaVarnway's avatar
Scott LaVarnway committed
826 827
    }

828
    aom_usec_timer_start(&timer);
Scott LaVarnway's avatar
Scott LaVarnway committed
829

830 831
    if (flush_decoder) {
      // Flush the decoder in frame parallel decode.
Sean DuBois's avatar
Sean DuBois committed
832
      if (aom_codec_decode(&decoder, NULL, 0, NULL)) {
833
        warn("Failed to flush decoder: %s", aom_codec_error(&decoder));
834 835 836
      }
    }

Scott LaVarnway's avatar
Scott LaVarnway committed
837
    got_data = 0;
838
    if ((img = aom_codec_get_frame(&decoder, &iter))) {
Scott LaVarnway's avatar
Scott LaVarnway committed
839 840
      ++frame_out;
      got_data = 1;
841 842
    }

843 844
    aom_usec_timer_mark(&timer);
    dx_time += (unsigned int)aom_usec_timer_elapsed(&timer);
845

846
    if (!frame_parallel &&
847 848
        aom_codec_control(&decoder, AOMD_GET_FRAME_CORRUPTED, &corrupted)) {
      warn("Failed AOM_GET_FRAME_CORRUPTED: %s", aom_codec_error(&decoder));
clang-format's avatar
clang-format committed
849
      if (!keep_going) goto fail;
850
    }
John Koleszar's avatar
John Koleszar committed
851
    frames_corrupted += corrupted;
John Koleszar's avatar
John Koleszar committed
852

clang-format's avatar
clang-format committed
853
    if (progress) show_progress(frame_in, frame_out, dx_time);
John Koleszar's avatar
John Koleszar committed
854

855
    if (!noblit && img) {
856 857
      const int PLANES_YUV[] = { AOM_PLANE_Y, AOM_PLANE_U, AOM_PLANE_V };
      const int PLANES_YVU[] = { AOM_PLANE_Y, AOM_PLANE_V, AOM_PLANE_U };
858
      const int *planes = flipuv ? PLANES_YVU : PLANES_YUV;
859

860
      if (do_scale) {
861 862 863 864
        if (frame_out == 1) {
          // If the output frames are to be scaled to a fixed display size then
          // use the width and height specified in the container. If either of
          // these is set to 0, use the display size set in the first frame
865 866
          // header. If that is unavailable, use the raw decoded size of the
          // first decoded frame.
867 868
          int render_width = aom_input_ctx.width;
          int render_height = aom_input_ctx.height;
869 870
          if (!render_width || !render_height) {
            int render_size[2];
871
            if (aom_codec_control(&decoder, AV1D_GET_DISPLAY_SIZE,
872
                                  render_size)) {
873
              // As last resort use size of first frame as display size.
874 875
              render_width = img->d_w;
              render_height = img->d_h;
876
            } else {
877 878
              render_width = render_size[0];
              render_height = render_size[1];
879
            }
880
          }
clang-format's avatar
clang-format committed
881
          scaled_img =
882
              aom_img_alloc(NULL, img->fmt, render_width, render_height, 16);
883
          scaled_img->bit_depth = img->bit_depth;
884
        }
885

886
        if (img->d_w != scaled_img->d_w || img->d_h != scaled_img->d_h) {
887
#if CONFIG_LIBYUV
888
          libyuv_scale(img, scaled_img, kFilterBox);
889
          img = scaled_img;
890
#else
clang-format's avatar
clang-format committed
891
          fprintf(stderr,
Johann's avatar
Johann committed
892 893 894
                  "Failed to scale output frame: %s.\n"
                  "libyuv is required for scaling but is currently disabled.\n"
                  "Be sure to specify -DCONFIG_LIBYUV=1 when running cmake.\n",
895
                  aom_codec_error(&decoder));
James Zern's avatar
James Zern committed
896
          goto fail;
897
#endif
898 899
        }
      }
900
      // Default to codec bit depth if output bit depth not set
901
      if (!output_bit_depth && single_file && !do_md5) {
902 903 904
        output_bit_depth = img->bit_depth;
      }
      // Shift up or down if necessary
Sebastien Alaiwan's avatar
Sebastien Alaiwan committed
905
      if (output_bit_depth != 0) {
906
        const aom_img_fmt_t shifted_fmt =
clang-format's avatar
clang-format committed
907
            output_bit_depth == 8
908 909
                ? img->fmt ^ (img->fmt & AOM_IMG_FMT_HIGHBITDEPTH)
                : img->fmt | AOM_IMG_FMT_HIGHBITDEPTH;
Sebastien Alaiwan's avatar
Sebastien Alaiwan committed
910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929

        if (shifted_fmt != img->fmt || output_bit_depth != img->bit_depth) {
          if (img_shifted &&
              img_shifted_realloc_required(img, img_shifted, shifted_fmt)) {
            aom_img_free(img_shifted);
            img_shifted = NULL;
          }
          if (!img_shifted) {
            img_shifted =
                aom_img_alloc(NULL, shifted_fmt, img->d_w, img->d_h, 16);
            img_shifted->bit_depth = output_bit_depth;
          }
          if (output_bit_depth > img->bit_depth) {
            aom_img_upshift(img_shifted, img,
                            output_bit_depth - img->bit_depth);
          } else {
            aom_img_downshift(img_shifted, img,
                              img->bit_depth - output_bit_depth);
          }
          img = img_shifted;
930 931
        }
      }
932

933
#if CONFIG_EXT_TILE
934 935
      aom_input_ctx.width = img->d_w;
      aom_input_ctx.height = img->d_h;
936 937
#endif  // CONFIG_EXT_TILE

938 939
#if CONFIG_MONO_VIDEO
      int num_planes = (!use_y4m && opt_raw && img->monochrome) ? 1 : 3;
940 941
#else
      int num_planes = 3;
942
      (void)opt_raw;
943 944
#endif

945
      if (single_file) {
946
        if (use_y4m) {
947
          char y4m_buf[Y4M_BUFFER_SIZE] = { 0 };
948
          size_t len = 0;
949
          if (img->fmt == AOM_IMG_FMT_I440 || img->fmt == AOM_IMG_FMT_I44016) {
Deb Mukherjee's avatar
Deb Mukherjee committed
950 951 952
            fprintf(stderr, "Cannot produce y4m output for 440 sampling.\n");
            goto fail;
          }
953 954
          if (frame_out == 1) {
            // Y4M file header
clang-format's avatar
clang-format committed
955
            len = y4m_write_file_header(
956 957 958
                y4m_buf, sizeof(y4m_buf), aom_input_ctx.width,
                aom_input_ctx.height, &aom_input_ctx.framerate, img->fmt,
                img->bit_depth);
959
            if (do_md5) {
960
              MD5Update(&md5_ctx, (md5byte *)y4m_buf, (unsigned int)len);
961
            } else {
962
              fputs(y4m_buf, outfile);
963 964 965 966
            }
          }

          // Y4M frame header
967
          len = y4m_write_frame_header(y4m_buf, sizeof(y4m_buf));
968
          if (do_md5) {