aomdec.c 34.1 KB
Newer Older
John Koleszar's avatar
John Koleszar committed
1
/*
Yaowu Xu's avatar
Yaowu Xu committed
2
 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
John Koleszar's avatar
John Koleszar committed
3
 *
Yaowu Xu's avatar
Yaowu Xu committed
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

Yaowu Xu's avatar
Yaowu Xu committed
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"

Yaowu Xu's avatar
Yaowu Xu committed
36
#include "aom/aom_decoder.h"
37
#include "aom_ports/mem_ops.h"
Yaowu Xu's avatar
Yaowu Xu committed
38
#include "aom_ports/aom_timer.h"
39

Yaowu Xu's avatar
Yaowu Xu committed
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;

Yaowu Xu's avatar
Yaowu Xu committed
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
clang-format's avatar
clang-format committed
110 111 112 113 114 115
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)");
116
#endif  // CONFIG_EXT_TILE
Dmitry Kovalev's avatar
Dmitry Kovalev committed
117

118 119 120 121 122 123
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,
124
#if CONFIG_EXT_TILE
125
  &tiler,       &tilec,
126
#endif  // CONFIG_EXT_TILE
127 128
  NULL
};
John Koleszar's avatar
John Koleszar committed
129

130
#if CONFIG_LIBYUV
Yaowu Xu's avatar
Yaowu Xu committed
131
static INLINE int libyuv_scale(aom_image_t *src, aom_image_t *dst,
clang-format's avatar
clang-format committed
132
                               FilterModeEnum mode) {
Yaowu Xu's avatar
Yaowu Xu committed
133 134
  if (src->fmt == AOM_IMG_FMT_I42016) {
    assert(dst->fmt == AOM_IMG_FMT_I42016);
clang-format's avatar
clang-format committed
135
    return I420Scale_16(
Yaowu Xu's avatar
Yaowu Xu committed
136 137 138 139 140 141 142
        (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);
143
  }
Yaowu Xu's avatar
Yaowu Xu committed
144 145 146 147 148 149 150 151
  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
152
                   dst->d_h, mode);
153
}
154
#endif
155

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

James Zern's avatar
James Zern committed
159 160 161 162 163 164 165 166
  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
167 168 169 170 171
          "\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
172
          "the following escape characters:\n");
James Zern's avatar
James Zern committed
173
  fprintf(fout,
John Koleszar's avatar
John Koleszar committed
174 175 176 177 178
          "\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
179
          "not specified, the output will be\n  directed to stdout.\n");
James Zern's avatar
James Zern committed
180
  fprintf(fout, "\nIncluded decoders:\n\n");
John Koleszar's avatar
John Koleszar committed
181

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

James Zern's avatar
James Zern committed
189 190
void usage_exit(void) {
  show_help(stderr, 1);
John Koleszar's avatar
John Koleszar committed
191
  exit(EXIT_FAILURE);
John Koleszar's avatar
John Koleszar committed
192 193
}

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

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

206 207 208 209
    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
210

211 212 213
    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
214
    }
John Koleszar's avatar
John Koleszar committed
215

216 217 218 219 220 221 222 223
    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;
224
      }
John Koleszar's avatar
John Koleszar committed
225
    }
226
  }
John Koleszar's avatar
John Koleszar committed
227

228 229 230 231 232 233
  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
234 235
  }

236 237 238
  return 0;
}

Yaowu Xu's avatar
Yaowu Xu committed
239
static int read_frame(struct AvxDecInputContext *input, uint8_t **buf,
240
                      size_t *bytes_in_buffer, size_t *buffer_size) {
Yaowu Xu's avatar
Yaowu Xu committed
241
  switch (input->aom_input_ctx->file_type) {
242
#if CONFIG_WEBM_IO
243
    case FILE_TYPE_WEBM:
244
      return webm_read_frame(input->webm_ctx, buf, bytes_in_buffer);
245
#endif
246
    case FILE_TYPE_RAW:
Yaowu Xu's avatar
Yaowu Xu committed
247
      return raw_read_frame(input->aom_input_ctx->file, buf, bytes_in_buffer,
clang-format's avatar
clang-format committed
248
                            buffer_size);
249
    case FILE_TYPE_IVF:
Yaowu Xu's avatar
Yaowu Xu committed
250
      return ivf_read_frame(input->aom_input_ctx->file, buf, bytes_in_buffer,
clang-format's avatar
clang-format committed
251
                            buffer_size);
252 253 254 255 256
#if CONFIG_OBU_NO_IVF
    case FILE_TYPE_OBU:
      return obu_read_temporal_unit(input->aom_input_ctx->file, buf,
                                    bytes_in_buffer, buffer_size);
#endif
clang-format's avatar
clang-format committed
257
    default: return 1;
258
  }
John Koleszar's avatar
John Koleszar committed
259 260
}

Yaowu Xu's avatar
Yaowu Xu committed
261
static void update_image_md5(const aom_image_t *img, const int planes[3],
262 263 264 265 266 267 268
                             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];
Yaowu Xu's avatar
Yaowu Xu committed
269 270 271
    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);
272 273 274 275 276 277 278 279

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

280 281
static void write_image_file(const aom_image_t *img, const int *planes,
                             const int num_planes, FILE *file) {
282
  int i, y;
Yaowu Xu's avatar
Yaowu Xu committed
283
  const int bytes_per_sample = ((img->fmt & AOM_IMG_FMT_HIGHBITDEPTH) ? 2 : 1);
284

285
  for (i = 0; i < num_planes; ++i) {
286 287 288
    const int plane = planes[i];
    const unsigned char *buf = img->planes[plane];
    const int stride = img->stride[plane];
Yaowu Xu's avatar
Yaowu Xu committed
289 290
    const int w = aom_img_plane_width(img, plane);
    const int h = aom_img_plane_height(img, plane);
291 292

    for (y = 0; y < h; ++y) {
293
      fwrite(buf, bytes_per_sample, w, file);
294 295
      buf += stride;
    }
John Koleszar's avatar
John Koleszar committed
296
  }
John Koleszar's avatar
John Koleszar committed
297 298
}

Yaowu Xu's avatar
Yaowu Xu committed
299
static int file_is_raw(struct AvxInputContext *input) {
300
  uint8_t buf[32];
John Koleszar's avatar
John Koleszar committed
301
  int is_raw = 0;
Yaowu Xu's avatar
Yaowu Xu committed
302
  aom_codec_stream_info_t si;
303

304
  if (fread(buf, 1, 32, input->file) == 32) {
John Koleszar's avatar
John Koleszar committed
305 306
    int i;

307
    if (mem_get_le32(buf) < 256 * 1024 * 1024) {
Yaowu Xu's avatar
Yaowu Xu committed
308 309 310
      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
311
                                        32 - 4, &si)) {
John Koleszar's avatar
John Koleszar committed
312
          is_raw = 1;
313
          input->fourcc = decoder->fourcc;
314 315 316 317
          input->width = si.w;
          input->height = si.h;
          input->framerate.numerator = 30;
          input->framerate.denominator = 1;
John Koleszar's avatar
John Koleszar committed
318 319
          break;
        }
320 321
      }
    }
John Koleszar's avatar
John Koleszar committed
322 323
  }

324
  rewind(input->file);
John Koleszar's avatar
John Koleszar committed
325
  return is_raw;
326 327
}

328
static void show_progress(int frame_in, int frame_out, uint64_t dx_time) {
329
  fprintf(stderr,
clang-format's avatar
clang-format committed
330
          "%d decoded frames/%d showed frames in %" PRId64 " us (%.2f fps)\r",
John Koleszar's avatar
John Koleszar committed
331
          frame_in, frame_out, dx_time,
332
          (double)frame_out * 1000000.0 / (double)dx_time);
333 334
}

335
struct ExternalFrameBuffer {
clang-format's avatar
clang-format committed
336
  uint8_t *data;
337 338 339 340 341 342 343 344 345
  size_t size;
  int in_use;
};

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

346
// Callback used by libaom to request an external frame buffer. |cb_priv|
347 348 349
// 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.
Yaowu Xu's avatar
Yaowu Xu committed
350 351
static int get_av1_frame_buffer(void *cb_priv, size_t min_size,
                                aom_codec_frame_buffer_t *fb) {
352 353 354
  int i;
  struct ExternalFrameBufferList *const ext_fb_list =
      (struct ExternalFrameBufferList *)cb_priv;
clang-format's avatar
clang-format committed
355
  if (ext_fb_list == NULL) return -1;
356 357 358

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

clang-format's avatar
clang-format committed
362
  if (i == ext_fb_list->num_external_frame_buffers) return -1;
363 364 365

  if (ext_fb_list->ext_fb[i].size < min_size) {
    free(ext_fb_list->ext_fb[i].data);
366
    ext_fb_list->ext_fb[i].data = (uint8_t *)calloc(min_size, sizeof(uint8_t));
clang-format's avatar
clang-format committed
367
    if (!ext_fb_list->ext_fb[i].data) return -1;
368 369 370 371 372 373 374 375 376 377 378 379 380

    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;
}

381
// Callback used by libaom when there are no references to the frame buffer.
382 383
// |cb_priv| user private data passed into the set function. |fb| pointer
// to the frame buffer.
Yaowu Xu's avatar
Yaowu Xu committed
384 385
static int release_av1_frame_buffer(void *cb_priv,
                                    aom_codec_frame_buffer_t *fb) {
386 387 388 389 390 391 392
  struct ExternalFrameBuffer *const ext_fb =
      (struct ExternalFrameBuffer *)fb->priv;
  (void)cb_priv;
  ext_fb->in_use = 0;
  return 0;
}

393 394 395
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
396 397 398 399 400 401 402 403 404
  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
405
      /* parse the pattern */
John Koleszar's avatar
John Koleszar committed
406 407
      q[q_len - 1] = '\0';
      switch (p[1]) {
clang-format's avatar
clang-format committed
408 409 410 411 412 413 414 415 416 417 418 419
        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
420 421 422
      }

      pat_len = strlen(q);
clang-format's avatar
clang-format committed
423
      if (pat_len >= q_len - 1) die("Output filename too long.\n");
John Koleszar's avatar
John Koleszar committed
424 425 426 427 428 429
      q += pat_len;
      p += 2;
      q_len -= pat_len;
    } else {
      size_t copy_len;

John Koleszar's avatar
John Koleszar committed
430
      /* copy the next segment */
John Koleszar's avatar
John Koleszar committed
431 432 433 434 435
      if (!next_pat)
        copy_len = strlen(p);
      else
        copy_len = next_pat - p;

clang-format's avatar
clang-format committed
436
      if (copy_len >= q_len - 1) die("Output filename too long.\n");
John Koleszar's avatar
John Koleszar committed
437 438 439 440 441 442 443 444

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

447 448 449 450 451 452 453
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
454
    if (p) p++;
455 456 457 458 459 460 461 462
  } while (p);

  return 1;
}

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

clang-format's avatar
clang-format committed
463
  for (i = 0; i < 16; ++i) printf("%02x", digest[i]);
464 465 466 467 468 469 470 471 472
  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
473
    if (!file) fatal("Failed to open output file '%s'", name);
474 475 476 477
    return file;
  }
}

Yaowu Xu's avatar
Yaowu Xu committed
478 479 480
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
481
  return img->d_w != shifted->d_w || img->d_h != shifted->d_h ||
482 483
         required_fmt != shifted->fmt;
}
484

485
static int main_loop(int argc, const char **argv_) {
Yaowu Xu's avatar
Yaowu Xu committed
486
  aom_codec_ctx_t decoder;
clang-format's avatar
clang-format committed
487 488
  char *fn = NULL;
  int i;
James Zern's avatar
James Zern committed
489
  int ret = EXIT_FAILURE;
clang-format's avatar
clang-format committed
490 491 492 493 494 495 496 497
  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;
Yaowu Xu's avatar
Yaowu Xu committed
498 499
  const AvxInterface *interface = NULL;
  const AvxInterface *fourcc_interface = NULL;
500
  uint64_t dx_time = 0;
clang-format's avatar
clang-format committed
501 502 503 504 505 506 507
  struct arg arg;
  char **argv, **argi, **argj;

  int single_file;
  int use_y4m = 1;
  int opt_yv12 = 0;
  int opt_i420 = 0;
508
  int opt_raw = 0;
509
  aom_codec_dec_cfg_t cfg = { 0, 0, 0, CONFIG_LOWBITDEPTH };
clang-format's avatar
clang-format committed
510
  unsigned int output_bit_depth = 0;
511
#if CONFIG_EXT_TILE
clang-format's avatar
clang-format committed
512 513
  int tile_row = -1;
  int tile_col = -1;
514
#endif  // CONFIG_EXT_TILE
clang-format's avatar
clang-format committed
515 516 517
  int frames_corrupted = 0;
  int dec_flags = 0;
  int do_scale = 0;
Yaowu Xu's avatar
Yaowu Xu committed
518 519
  aom_image_t *scaled_img = NULL;
  aom_image_t *img_shifted = NULL;
clang-format's avatar
clang-format committed
520 521 522
  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
523

524
  const char *outfile_pattern = NULL;
clang-format's avatar
clang-format committed
525
  char outfile_name[PATH_MAX] = { 0 };
526 527
  FILE *outfile = NULL;

528 529
  FILE *framestats_file = NULL;

530 531 532
  MD5Context md5_ctx;
  unsigned char md5_digest[16];

Yaowu Xu's avatar
Yaowu Xu committed
533 534
  struct AvxDecInputContext input = { NULL, NULL };
  struct AvxInputContext aom_input_ctx;
535
#if CONFIG_WEBM_IO
536 537
  struct WebmInputContext webm_ctx;
  memset(&(webm_ctx), 0, sizeof(webm_ctx));
538
  input.webm_ctx = &webm_ctx;
539
#endif
Yaowu Xu's avatar
Yaowu Xu committed
540
  input.aom_input_ctx = &aom_input_ctx;
541

John Koleszar's avatar
John Koleszar committed
542 543 544 545 546 547 548 549
  /* 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
550 551 552 553
    if (arg_match(&arg, &help, argi)) {
      show_help(stdout, 0);
      exit(EXIT_SUCCESS);
    } else if (arg_match(&arg, &codecarg, argi)) {
Yaowu Xu's avatar
Yaowu Xu committed
554
      interface = get_aom_decoder_by_name(arg.val);
555 556
      if (!interface)
        die("Error: Unrecognized argument (%s) to --codec\n", arg.val);
John Koleszar's avatar
John Koleszar committed
557 558
    } else if (arg_match(&arg, &looparg, argi)) {
      // no-op
559
    } else if (arg_match(&arg, &outputfile, argi)) {
John Koleszar's avatar
John Koleszar committed
560
      outfile_pattern = arg.val;
561
    } else if (arg_match(&arg, &use_yv12, argi)) {
John Koleszar's avatar
John Koleszar committed
562 563
      use_y4m = 0;
      flipuv = 1;
564
      opt_yv12 = 1;
565 566
      opt_i420 = 0;
      opt_raw = 0;
John Koleszar's avatar
John Koleszar committed
567 568 569
    } else if (arg_match(&arg, &use_i420, argi)) {
      use_y4m = 0;
      flipuv = 0;
570
      opt_yv12 = 0;
571
      opt_i420 = 1;
572
      opt_raw = 0;
573 574
    } else if (arg_match(&arg, &rawvideo, argi)) {
      use_y4m = 0;
575 576 577
      opt_yv12 = 0;
      opt_i420 = 0;
      opt_raw = 1;
578
    } else if (arg_match(&arg, &flipuvarg, argi)) {
John Koleszar's avatar
John Koleszar committed
579
      flipuv = 1;
580
    } else if (arg_match(&arg, &noblitarg, argi)) {
John Koleszar's avatar
John Koleszar committed
581
      noblit = 1;
582
    } else if (arg_match(&arg, &progressarg, argi)) {
John Koleszar's avatar
John Koleszar committed
583
      progress = 1;
584
    } else if (arg_match(&arg, &limitarg, argi)) {
John Koleszar's avatar
John Koleszar committed
585
      stop_after = arg_parse_uint(&arg);
586
    } else if (arg_match(&arg, &skiparg, argi)) {
587
      arg_skip = arg_parse_uint(&arg);
588
    } else if (arg_match(&arg, &postprocarg, argi)) {
John Koleszar's avatar
John Koleszar committed
589
      postproc = 1;
590
    } else if (arg_match(&arg, &md5arg, argi)) {
John Koleszar's avatar
John Koleszar committed
591
      do_md5 = 1;
592 593 594 595 596 597 598
    } 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
599
      summary = 1;
600
    } else if (arg_match(&arg, &threadsarg, argi)) {
John Koleszar's avatar
John Koleszar committed
601
      cfg.threads = arg_parse_uint(&arg);
602
    }
Yaowu Xu's avatar
Yaowu Xu committed
603
#if CONFIG_AV1_DECODER
604 605 606
    else if (arg_match(&arg, &frameparallelarg, argi))
      frame_parallel = 1;
#endif
John Koleszar's avatar
John Koleszar committed
607 608
    else if (arg_match(&arg, &verbosearg, argi))
      quiet = 0;
John Koleszar's avatar
John Koleszar committed
609 610
    else if (arg_match(&arg, &scalearg, argi))
      do_scale = 1;
611 612
    else if (arg_match(&arg, &fb_arg, argi))
      num_external_frame_buffers = arg_parse_uint(&arg);
613 614
    else if (arg_match(&arg, &continuearg, argi))
      keep_going = 1;
615 616 617
    else if (arg_match(&arg, &outbitdeptharg, argi)) {
      output_bit_depth = arg_parse_uint(&arg);
    }
618 619 620 621 622 623
#if CONFIG_EXT_TILE
    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
624 625 626
    else
      argj++;
  }
John Koleszar's avatar
John Koleszar committed
627

John Koleszar's avatar
John Koleszar committed
628 629 630 631
  /* 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
632

John Koleszar's avatar
John Koleszar committed
633 634
  /* Handle non-option arguments */
  fn = argv[0];
John Koleszar's avatar
John Koleszar committed
635

hanno's avatar
hanno committed
636 637
  if (!fn) {
    free(argv);
James Zern's avatar
James Zern committed
638
    fprintf(stderr, "No input file specified!\n");
John Koleszar's avatar
John Koleszar committed
639
    usage_exit();
hanno's avatar
hanno committed
640
  }
John Koleszar's avatar
John Koleszar committed
641 642
  /* Open file */
  infile = strcmp(fn, "-") ? fopen(fn, "rb") : set_binary_mode(stdin);
John Koleszar's avatar
John Koleszar committed
643

John Koleszar's avatar
John Koleszar committed
644
  if (!infile) {
645
    fatal("Failed to open input file '%s'", strcmp(fn, "-") ? fn : "stdin");
John Koleszar's avatar
John Koleszar committed
646
  }
647
#if CONFIG_OS_SUPPORT
John Koleszar's avatar
John Koleszar committed
648
  /* Make sure we don't dump to the terminal, unless forced to with -o - */
James Zern's avatar
James Zern committed
649
  if (!outfile_pattern && isatty(STDOUT_FILENO) && !do_md5 && !noblit) {
John Koleszar's avatar
John Koleszar committed
650 651 652 653 654
    fprintf(stderr,
            "Not dumping raw video to your terminal. Use '-o -' to "
            "override.\n");
    return EXIT_FAILURE;
  }
655
#endif
Yaowu Xu's avatar
Yaowu Xu committed
656 657 658
  input.aom_input_ctx->file = infile;
  if (file_is_ivf(input.aom_input_ctx))
    input.aom_input_ctx->file_type = FILE_TYPE_IVF;
659
#if CONFIG_WEBM_IO
Yaowu Xu's avatar
Yaowu Xu committed
660 661
  else if (file_is_webm(input.webm_ctx, input.aom_input_ctx))
    input.aom_input_ctx->file_type = FILE_TYPE_WEBM;
662 663 664 665
#endif
#if CONFIG_OBU_NO_IVF
  else if (file_is_obu(input.aom_input_ctx))
    input.aom_input_ctx->file_type = FILE_TYPE_OBU;
666
#endif
Yaowu Xu's avatar
Yaowu Xu committed
667 668
  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
669 670
  else {
    fprintf(stderr, "Unrecognized input file type.\n");
671
#if !CONFIG_WEBM_IO
Yaowu Xu's avatar
Yaowu Xu committed
672
    fprintf(stderr, "aomdec was built without WebM container support.\n");
673
#endif
John Koleszar's avatar
John Koleszar committed
674 675 676 677
    return EXIT_FAILURE;
  }

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

680 681
  if (!noblit && single_file) {
    generate_filename(outfile_pattern, outfile_name, PATH_MAX,
Yaowu Xu's avatar
Yaowu Xu committed
682
                      aom_input_ctx.width, aom_input_ctx.height, 0);
683 684 685 686
    if (do_md5)
      MD5Init(&md5_ctx);
    else
      outfile = open_outfile(outfile_name);
John Koleszar's avatar
John Koleszar committed
687 688 689 690
  }

  if (use_y4m && !noblit) {
    if (!single_file) {
clang-format's avatar
clang-format committed
691 692
      fprintf(stderr,
              "YUV4MPEG2 not supported with output patterns,"
Deb Mukherjee's avatar
Deb Mukherjee committed
693
              " try --i420 or --yv12 or --rawvideo.\n");
John Koleszar's avatar
John Koleszar committed
694
      return EXIT_FAILURE;
695
    }
John Koleszar's avatar
John Koleszar committed
696

697
#if CONFIG_WEBM_IO
Yaowu Xu's avatar
Yaowu Xu committed
698 699
    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
700 701
        fprintf(stderr,
                "Failed to guess framerate -- error parsing "
John Koleszar's avatar
John Koleszar committed
702 703 704
                "webm file?\n");
        return EXIT_FAILURE;
      }
705
    }
706
#endif
John Koleszar's avatar
John Koleszar committed
707 708
  }

Yaowu Xu's avatar
Yaowu Xu committed
709
  fourcc_interface = get_aom_decoder_by_fourcc(aom_input_ctx.fourcc);
710 711 712 713
  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
714

Yaowu Xu's avatar
Yaowu Xu committed
715
  if (!interface) interface = get_aom_decoder_by_index(0);
716

Yaowu Xu's avatar
Yaowu Xu committed
717 718 719
  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
720
                         dec_flags)) {
721
    fprintf(stderr, "Failed to initialize decoder: %s\n",
Yaowu Xu's avatar
Yaowu Xu committed
722
            aom_codec_error(&decoder));
James Zern's avatar
James Zern committed
723
    goto fail2;
John Koleszar's avatar
John Koleszar committed
724
  }
John Koleszar's avatar
John Koleszar committed
725

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

Yaowu Xu's avatar
Yaowu Xu committed
728
#if CONFIG_AV1_DECODER && CONFIG_EXT_TILE
729 730 731 732 733
  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;
  }
734

735 736 737 738
  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;
739 740
  }
#endif
741

clang-format's avatar
clang-format committed
742
  if (arg_skip) fprintf(stderr, "Skipping first %d frames.\n", arg_skip);
743
  while (arg_skip) {
clang-format's avatar
clang-format committed
744
    if (read_frame(&input, &buf, &bytes_in_buffer, &buffer_size)) break;
745 746 747
    arg_skip--;
  }

748 749 750 751
  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));
Yaowu Xu's avatar
Yaowu Xu committed
752 753
    if (aom_codec_set_frame_buffer_functions(&decoder, get_av1_frame_buffer,
                                             release_av1_frame_buffer,
clang-format's avatar
clang-format committed
754
                                             &ext_fb_list)) {
755
      fprintf(stderr, "Failed to configure external frame buffers: %s\n",
Yaowu Xu's avatar
Yaowu Xu committed
756
              aom_codec_error(&decoder));
James Zern's avatar
James Zern committed
757
      goto fail;
758 759 760
    }
  }

Scott LaVarnway's avatar
Scott LaVarnway committed
761 762 763
  frame_avail = 1;
  got_data = 0;

764 765
  if (framestats_file) fprintf(framestats_file, "bytes,qp\r\n");

John Koleszar's avatar
John Koleszar committed
766
  /* Decode file */
Scott LaVarnway's avatar
Scott LaVarnway committed
767
  while (frame_avail || got_data) {
Yaowu Xu's avatar
Yaowu Xu committed
768 769 770
    aom_codec_iter_t iter = NULL;
    aom_image_t *img;
    struct aom_usec_timer timer;
clang-format's avatar
clang-format committed
771
    int corrupted = 0;
John Koleszar's avatar
John Koleszar committed
772

Scott LaVarnway's avatar
Scott LaVarnway committed
773 774
    frame_avail = 0;
    if (!stop_after || frame_in < stop_after) {
775
      if (!read_frame(&input, &buf, &bytes_in_buffer, &buffer_size)) {
Scott LaVarnway's avatar
Scott LaVarnway committed
776 777
        frame_avail = 1;
        frame_in++;
John Koleszar's avatar
John Koleszar committed
778

Yaowu Xu's avatar
Yaowu Xu committed
779
        aom_usec_timer_start(&timer);
John Koleszar's avatar
John Koleszar committed
780

Yaowu Xu's avatar
Yaowu Xu committed
781
        if (aom_codec_decode(&decoder, buf, (unsigned int)bytes_in_buffer, NULL,
clang-format's avatar
clang-format committed
782
                             0)) {
Yaowu Xu's avatar
Yaowu Xu committed
783
          const char *detail = aom_codec_error_detail(&decoder);
clang-format's avatar
clang-format committed
784
          warn("Failed to decode frame %d: %s", frame_in,
Yaowu Xu's avatar
Yaowu Xu committed
785
               aom_codec_error(&decoder));
786

clang-format's avatar
clang-format committed
787 788
          if (detail) warn("Additional information: %s", detail);
          if (!keep_going) goto fail;
Scott LaVarnway's avatar
Scott LaVarnway committed
789 790
        }

791 792 793 794 795 796 797 798 799 800
        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);
        }

Yaowu Xu's avatar
Yaowu Xu committed
801 802
        aom_usec_timer_mark(&timer);
        dx_time += aom_usec_timer_elapsed(&timer);
803 804
      } else {
        flush_decoder = 1;
Scott LaVarnway's avatar
Scott LaVarnway committed
805
      }
806 807
    } else {
      flush_decoder = 1;
Scott LaVarnway's avatar
Scott LaVarnway committed
808 809
    }

Yaowu Xu's avatar
Yaowu Xu committed
810
    aom_usec_timer_start(&timer);
Scott LaVarnway's avatar
Scott LaVarnway committed
811

812 813
    if (flush_decoder) {
      // Flush the decoder in frame parallel decode.
Yaowu Xu's avatar
Yaowu Xu committed
814 815
      if (aom_codec_decode(&decoder, NULL, 0, NULL, 0)) {
        warn("Failed to flush decoder: %s", aom_codec_error(&decoder));
816 817 818
      }
    }

Scott LaVarnway's avatar
Scott LaVarnway committed
819
    got_data = 0;
Yaowu Xu's avatar
Yaowu Xu committed
820
    if ((img = aom_codec_get_frame(&decoder, &iter))) {
Scott LaVarnway's avatar
Scott LaVarnway committed
821 822
      ++frame_out;
      got_data = 1;
823 824
    }

Yaowu Xu's avatar
Yaowu Xu committed
825 826
    aom_usec_timer_mark(&timer);
    dx_time += (unsigned int)aom_usec_timer_elapsed(&timer);
827

828
    if (!frame_parallel &&
Yaowu Xu's avatar
Yaowu Xu committed
829 830
        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
831
      if (!keep_going) goto fail;
832
    }
John Koleszar's avatar
John Koleszar committed
833
    frames_corrupted += corrupted;
John Koleszar's avatar
John Koleszar committed
834

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

837
    if (!noblit && img) {
Yaowu Xu's avatar
Yaowu Xu committed
838 839
      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 };
840
      const int *planes = flipuv ? PLANES_YVU : PLANES_YUV;
841

842
      if (do_scale) {
843 844 845 846
        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
847 848
          // header. If that is unavailable, use the raw decoded size of the
          // first decoded frame.
Yaowu Xu's avatar
Yaowu Xu committed
849 850
          int render_width = aom_input_ctx.width;
          int render_height = aom_input_ctx.height;
851 852
          if (!render_width || !render_height) {
            int render_size[2];
Yaowu Xu's avatar
Yaowu Xu committed
853
            if (aom_codec_control(&decoder, AV1D_GET_DISPLAY_SIZE,
854
                                  render_size)) {
855
              // As last resort use size of first frame as display size.
856 857
              render_width = img->d_w;
              render_height = img->d_h;
858
            } else {
859 860
              render_width = render_size[0];
              render_height = render_size[1];
861
            }
862
          }
clang-format's avatar
clang-format committed
863
          scaled_img =
Yaowu Xu's avatar
Yaowu Xu committed
864
              aom_img_alloc(NULL, img->fmt, render_width, render_height, 16);
865
          scaled_img->bit_depth = img->bit_depth;
John Koleszar's avatar
John Koleszar committed
866
        }
867

868
        if (img->d_w != scaled_img->d_w || img->d_h != scaled_img->d_h) {
Deb Mukherjee's avatar
Deb Mukherjee committed
869
#if CONFIG_LIBYUV
870
          libyuv_scale(img, scaled_img, kFilterBox);
John Koleszar's avatar
John Koleszar committed
871
          img = scaled_img;
Deb Mukherjee's avatar
Deb Mukherjee committed
872
#else
clang-format's avatar
clang-format committed
873
          fprintf(stderr,
Johann's avatar
Johann committed
874 875 876
                  "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",
Yaowu Xu's avatar
Yaowu Xu committed
877
                  aom_codec_error(&decoder));
James Zern's avatar
James Zern committed
878
          goto fail;
Deb Mukherjee's avatar
Deb Mukherjee committed
879
#endif
John Koleszar's avatar
John Koleszar committed
880 881
        }
      }
882
      // Default to codec bit depth if output bit depth not set
883
      if (!output_bit_depth && single_file && !do_md5) {
884 885 886
        output_bit_depth = img->bit_depth;
      }
      // Shift up or down if necessary
Sebastien Alaiwan's avatar
Sebastien Alaiwan committed
887
      if (output_bit_depth != 0) {
Yaowu Xu's avatar
Yaowu Xu committed
888
        const aom_img_fmt_t shifted_fmt =
clang-format's avatar
clang-format committed
889
            output_bit_depth == 8
Yaowu Xu's avatar
Yaowu Xu committed
890 891
                ? img->fmt ^ (img->fmt & AOM_IMG_FMT_HIGHBITDEPTH)
                : img->fmt | AOM_IMG_FMT_HIGHBITDEPTH;
Sebastien Alaiwan's avatar
Sebastien Alaiwan committed
892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911

        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;
912 913
        }
      }
914

915
#if CONFIG_EXT_TILE
Yaowu Xu's avatar
Yaowu Xu committed
916 917
      aom_input_ctx.width = img->d_w;
      aom_input_ctx.height = img->d_h;
918 919
#endif  // CONFIG_EXT_TILE

920 921
#if CONFIG_MONO_VIDEO
      int num_planes = (!use_y4m && opt_raw && img->monochrome) ? 1 : 3;
922 923
#else
      int num_planes = 3;
924
      (void)opt_raw;
925 926
#endif

927
      if (single_file) {
928
        if (use_y4m) {
929
          char y4m_buf[Y4M_BUFFER_SIZE] = { 0 };
930
          size_t len = 0;
Yaowu Xu's avatar
Yaowu Xu committed
931
          if (img->fmt == AOM_IMG_FMT_I440 || img->fmt == AOM_IMG_FMT_I44016) {
Deb Mukherjee's avatar
Deb Mukherjee committed
932 933 934
            fprintf(stderr, "Cannot produce y4m output for 440 sampling.\n");
            goto fail;
          }
935 936
          if (frame_out == 1) {
            // Y4M file header
clang-format's avatar
clang-format committed
937
            len = y4m_write_file_header(
938 939 940
                y4m_buf, sizeof(y4m_buf), aom_input_ctx.width,
                aom_input_ctx.height, &aom_input_ctx.framerate, img->fmt,
                img->bit_depth);
941
            if (do_md5) {
942
              MD5Update(&md5_ctx, (md5byte *)y4m_buf, (unsigned int)len);
943
            } else {
944
              fputs(y4m_buf, outfile);
945 946 947 948
            }
          }

          // Y4M frame header
949
          len = y4m_write_frame_header(y4m_buf, sizeof(y4m_buf));
950
          if (do_md5) {
951
            MD5Update(&md5_ctx, (md5byte *)y4m_buf, (unsigned int)len);
952
          } else {
953
            fputs(y4m_buf, outfile);
954
          }
955 956 957 958 959
        } else {
          if (frame_out == 1) {
            // Check if --yv12 or --i420 options are consistent with the
            // bit-stream decoded
            if (opt_i420) {
Yaowu Xu's avatar
Yaowu Xu committed
960 961
              if (img->fmt != AOM_IMG_FMT_I420 &&
                  img->fmt != AOM_IMG_FMT_I42016) {
962 963 964 965 966
                fprintf(stderr, "Cannot produce i420 output for bit-stream.\n");
                goto fail;
              }
            }
            if (opt_yv12) {
Yaowu Xu's avatar
Yaowu Xu committed
967 968
              if ((img->fmt != AOM_IMG_FMT_I420 &&
                   img->fmt != AOM_IMG_FMT_YV12) ||
clang-format's avatar