aomdec.c 33.3 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 21

#if CONFIG_LIBYUV
22
#include "third_party/libyuv/include/libyuv/scale.h"
23
#endif
24 25

#include "./args.h"
26 27
#include "./ivfdec.h"

Yaowu Xu's avatar
Yaowu Xu committed
28
#include "aom/aom_decoder.h"
29
#include "aom_ports/mem_ops.h"
Yaowu Xu's avatar
Yaowu Xu committed
30
#include "aom_ports/aom_timer.h"
31

Yaowu Xu's avatar
Yaowu Xu committed
32 33
#if CONFIG_AV1_DECODER
#include "aom/aomdx.h"
John Koleszar's avatar
John Koleszar committed
34
#endif
35

36
#include "./md5_utils.h"
37 38

#include "./tools_common.h"
39
#if CONFIG_WEBM_IO
40
#include "./webmdec.h"
41
#endif
42
#include "./y4menc.h"
John Koleszar's avatar
John Koleszar committed
43 44 45

static const char *exec_name;

Yaowu Xu's avatar
Yaowu Xu committed
46 47
struct AvxDecInputContext {
  struct AvxInputContext *aom_input_ctx;
48 49 50
  struct WebmInputContext *webm_ctx;
};

clang-format's avatar
clang-format committed
51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 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
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 error_concealment =
    ARG_DEF(NULL, "error-concealment", 0, "Enable decoder error-concealment");
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");
Yaowu Xu's avatar
Yaowu Xu committed
92
#if CONFIG_AOM_HIGHBITDEPTH
clang-format's avatar
clang-format committed
93 94
static const arg_def_t outbitdeptharg =
    ARG_DEF(NULL, "output-bit-depth", 1, "Output bit-depth for decoded frames");
95
#endif
96
#if CONFIG_EXT_TILE
clang-format's avatar
clang-format committed
97 98 99 100 101 102
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)");
103
#endif  // CONFIG_EXT_TILE
Dmitry Kovalev's avatar
Dmitry Kovalev committed
104

clang-format's avatar
clang-format committed
105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124
static const arg_def_t *all_args[] = { &codecarg,
                                       &use_yv12,
                                       &use_i420,
                                       &flipuvarg,
                                       &rawvideo,
                                       &noblitarg,
                                       &progressarg,
                                       &limitarg,
                                       &skiparg,
                                       &postprocarg,
                                       &summaryarg,
                                       &outputfile,
                                       &threadsarg,
                                       &frameparallelarg,
                                       &verbosearg,
                                       &scalearg,
                                       &fb_arg,
                                       &md5arg,
                                       &error_concealment,
                                       &continuearg,
Yaowu Xu's avatar
Yaowu Xu committed
125
#if CONFIG_AOM_HIGHBITDEPTH
clang-format's avatar
clang-format committed
126
                                       &outbitdeptharg,
127
#endif
128
#if CONFIG_EXT_TILE
clang-format's avatar
clang-format committed
129 130
                                       &tiler,
                                       &tilec,
131
#endif  // CONFIG_EXT_TILE
clang-format's avatar
clang-format committed
132
                                       NULL };
John Koleszar's avatar
John Koleszar committed
133

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

162
void usage_exit(void) {
John Koleszar's avatar
John Koleszar committed
163
  int i;
John Koleszar's avatar
John Koleszar committed
164

clang-format's avatar
clang-format committed
165 166 167 168
  fprintf(stderr,
          "Usage: %s <options> filename\n\n"
          "Options:\n",
          exec_name);
John Koleszar's avatar
John Koleszar committed
169 170 171 172 173 174 175
  arg_show_usage(stderr, all_args);
  fprintf(stderr,
          "\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
176 177
          "the following escape characters:\n");
  fprintf(stderr,
John Koleszar's avatar
John Koleszar committed
178 179 180 181 182
          "\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
183
          "not specified, the output will be\n  directed to stdout.\n");
John Koleszar's avatar
John Koleszar committed
184 185
  fprintf(stderr, "\nIncluded decoders:\n\n");

Yaowu Xu's avatar
Yaowu Xu committed
186 187
  for (i = 0; i < get_aom_decoder_count(); ++i) {
    const AvxInterface *const decoder = get_aom_decoder_by_index(i);
clang-format's avatar
clang-format committed
188
    fprintf(stderr, "    %-6s - %s\n", decoder->name,
Yaowu Xu's avatar
Yaowu Xu committed
189
            aom_codec_iface_name(decoder->codec_interface()));
190
  }
John Koleszar's avatar
John Koleszar committed
191 192

  exit(EXIT_FAILURE);
John Koleszar's avatar
John Koleszar committed
193 194
}

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

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

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

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

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

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

237 238 239
  return 0;
}

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

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

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

Yaowu Xu's avatar
Yaowu Xu committed
276
static void write_image_file(const aom_image_t *img, const int planes[3],
277 278
                             FILE *file) {
  int i, y;
Yaowu Xu's avatar
Yaowu Xu committed
279 280
#if CONFIG_AOM_HIGHBITDEPTH
  const int bytes_per_sample = ((img->fmt & AOM_IMG_FMT_HIGHBITDEPTH) ? 2 : 1);
281 282 283
#else
  const int bytes_per_sample = 1;
#endif
284 285 286 287 288

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

John Koleszar's avatar
John Koleszar committed
304 305
  si.sz = sizeof(si);

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

  return 1;
}

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

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

Yaowu Xu's avatar
Yaowu Xu committed
480 481 482 483
#if CONFIG_AOM_HIGHBITDEPTH
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
484
  return img->d_w != shifted->d_w || img->d_h != shifted->d_h ||
485 486
         required_fmt != shifted->fmt;
}
487 488
#endif

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

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

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

  MD5Context md5_ctx;
  unsigned char md5_digest[16];

Yaowu Xu's avatar
Yaowu Xu committed
538 539
  struct AvxDecInputContext input = { NULL, NULL };
  struct AvxInputContext aom_input_ctx;
540
#if CONFIG_WEBM_IO
541 542
  struct WebmInputContext webm_ctx;
  memset(&(webm_ctx), 0, sizeof(webm_ctx));
543
  input.webm_ctx = &webm_ctx;
544
#endif
Yaowu Xu's avatar
Yaowu Xu committed
545
  input.aom_input_ctx = &aom_input_ctx;
546

John Koleszar's avatar
John Koleszar committed
547 548 549 550 551 552 553 554 555
  /* 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;

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

John Koleszar's avatar
John Koleszar committed
618 619 620 621
  /* 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
622

John Koleszar's avatar
John Koleszar committed
623 624
  /* Handle non-option arguments */
  fn = argv[0];
John Koleszar's avatar
John Koleszar committed
625

hanno's avatar
hanno committed
626 627
  if (!fn) {
    free(argv);
John Koleszar's avatar
John Koleszar committed
628
    usage_exit();
hanno's avatar
hanno committed
629
  }
John Koleszar's avatar
John Koleszar committed
630 631
  /* Open file */
  infile = strcmp(fn, "-") ? fopen(fn, "rb") : set_binary_mode(stdin);
John Koleszar's avatar
John Koleszar committed
632

John Koleszar's avatar
John Koleszar committed
633
  if (!infile) {
634
    fatal("Failed to open input file '%s'", strcmp(fn, "-") ? fn : "stdin");
John Koleszar's avatar
John Koleszar committed
635
  }
636
#if CONFIG_OS_SUPPORT
John Koleszar's avatar
John Koleszar committed
637 638 639 640 641 642 643
  /* Make sure we don't dump to the terminal, unless forced to with -o - */
  if (!outfile_pattern && isatty(fileno(stdout)) && !do_md5 && !noblit) {
    fprintf(stderr,
            "Not dumping raw video to your terminal. Use '-o -' to "
            "override.\n");
    return EXIT_FAILURE;
  }
644
#endif
Yaowu Xu's avatar
Yaowu Xu committed
645 646 647
  input.aom_input_ctx->file = infile;
  if (file_is_ivf(input.aom_input_ctx))
    input.aom_input_ctx->file_type = FILE_TYPE_IVF;
648
#if CONFIG_WEBM_IO
Yaowu Xu's avatar
Yaowu Xu committed
649 650
  else if (file_is_webm(input.webm_ctx, input.aom_input_ctx))
    input.aom_input_ctx->file_type = FILE_TYPE_WEBM;
651
#endif
Yaowu Xu's avatar
Yaowu Xu committed
652 653
  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
654 655
  else {
    fprintf(stderr, "Unrecognized input file type.\n");
656
#if !CONFIG_WEBM_IO
Yaowu Xu's avatar
Yaowu Xu committed
657
    fprintf(stderr, "aomdec was built without WebM container support.\n");
658
#endif
John Koleszar's avatar
John Koleszar committed
659 660 661 662
    return EXIT_FAILURE;
  }

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

665 666
  if (!noblit && single_file) {
    generate_filename(outfile_pattern, outfile_name, PATH_MAX,
Yaowu Xu's avatar
Yaowu Xu committed
667
                      aom_input_ctx.width, aom_input_ctx.height, 0);
668 669 670 671
    if (do_md5)
      MD5Init(&md5_ctx);
    else
      outfile = open_outfile(outfile_name);
John Koleszar's avatar
John Koleszar committed
672 673 674 675
  }

  if (use_y4m && !noblit) {
    if (!single_file) {
clang-format's avatar
clang-format committed
676 677
      fprintf(stderr,
              "YUV4MPEG2 not supported with output patterns,"
Deb Mukherjee's avatar
Deb Mukherjee committed
678
              " try --i420 or --yv12 or --rawvideo.\n");
John Koleszar's avatar
John Koleszar committed
679
      return EXIT_FAILURE;
680
    }
John Koleszar's avatar
John Koleszar committed
681

682
#if CONFIG_WEBM_IO
Yaowu Xu's avatar
Yaowu Xu committed
683 684
    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
685 686
        fprintf(stderr,
                "Failed to guess framerate -- error parsing "
John Koleszar's avatar
John Koleszar committed
687 688 689
                "webm file?\n");
        return EXIT_FAILURE;
      }
690
    }
691
#endif
John Koleszar's avatar
John Koleszar committed
692 693
  }

Yaowu Xu's avatar
Yaowu Xu committed
694
  fourcc_interface = get_aom_decoder_by_fourcc(aom_input_ctx.fourcc);
695 696 697 698
  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
699

Yaowu Xu's avatar
Yaowu Xu committed
700
  if (!interface) interface = get_aom_decoder_by_index(0);
701

Yaowu Xu's avatar
Yaowu Xu committed
702 703 704 705
  dec_flags = (postproc ? AOM_CODEC_USE_POSTPROC : 0) |
              (ec_enabled ? AOM_CODEC_USE_ERROR_CONCEALMENT : 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
706
                         dec_flags)) {
707
    fprintf(stderr, "Failed to initialize decoder: %s\n",
Yaowu Xu's avatar
Yaowu Xu committed
708
            aom_codec_error(&decoder));
John Koleszar's avatar
John Koleszar committed
709 710
    return EXIT_FAILURE;
  }
John Koleszar's avatar
John Koleszar committed
711

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

Yaowu Xu's avatar
Yaowu Xu committed
714 715 716
#if CONFIG_AV1_DECODER && CONFIG_EXT_TILE
  if (strncmp(decoder.name, "WebM Project AV1", 17) == 0) {
    if (aom_codec_control(&decoder, AV1_SET_DECODE_TILE_ROW, tile_row)) {
717
      fprintf(stderr, "Failed to set decode_tile_row: %s\n",
Yaowu Xu's avatar
Yaowu Xu committed
718
              aom_codec_error(&decoder));
719 720 721
      return EXIT_FAILURE;
    }

Yaowu Xu's avatar
Yaowu Xu committed
722
    if (aom_codec_control(&decoder, AV1_SET_DECODE_TILE_COL, tile_col)) {
723
      fprintf(stderr, "Failed to set decode_tile_col: %s\n",
Yaowu Xu's avatar
Yaowu Xu committed
724
              aom_codec_error(&decoder));
725 726 727 728
      return EXIT_FAILURE;
    }
  }
#endif
729

clang-format's avatar
clang-format committed
730
  if (arg_skip) fprintf(stderr, "Skipping first %d frames.\n", arg_skip);
731
  while (arg_skip) {
clang-format's avatar
clang-format committed
732
    if (read_frame(&input, &buf, &bytes_in_buffer, &buffer_size)) break;
733 734 735
    arg_skip--;
  }

736 737 738 739
  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
740 741
    if (aom_codec_set_frame_buffer_functions(&decoder, get_av1_frame_buffer,
                                             release_av1_frame_buffer,
clang-format's avatar
clang-format committed
742
                                             &ext_fb_list)) {
743
      fprintf(stderr, "Failed to configure external frame buffers: %s\n",
Yaowu Xu's avatar
Yaowu Xu committed
744
              aom_codec_error(&decoder));
745 746 747 748
      return EXIT_FAILURE;
    }
  }

Scott LaVarnway's avatar
Scott LaVarnway committed
749 750 751
  frame_avail = 1;
  got_data = 0;

John Koleszar's avatar
John Koleszar committed
752
  /* Decode file */
Scott LaVarnway's avatar
Scott LaVarnway committed
753
  while (frame_avail || got_data) {
Yaowu Xu's avatar
Yaowu Xu committed
754 755 756
    aom_codec_iter_t iter = NULL;
    aom_image_t *img;
    struct aom_usec_timer timer;
clang-format's avatar
clang-format committed
757
    int corrupted = 0;
John Koleszar's avatar
John Koleszar committed
758

Scott LaVarnway's avatar
Scott LaVarnway committed
759 760
    frame_avail = 0;
    if (!stop_after || frame_in < stop_after) {
761
      if (!read_frame(&input, &buf, &bytes_in_buffer, &buffer_size)) {
Scott LaVarnway's avatar
Scott LaVarnway committed
762 763
        frame_avail = 1;
        frame_in++;
John Koleszar's avatar
John Koleszar committed
764

Yaowu Xu's avatar
Yaowu Xu committed
765
        aom_usec_timer_start(&timer);
John Koleszar's avatar
John Koleszar committed
766

Yaowu Xu's avatar
Yaowu Xu committed
767
        if (aom_codec_decode(&decoder, buf, (unsigned int)bytes_in_buffer, NULL,
clang-format's avatar
clang-format committed
768
                             0)) {
Yaowu Xu's avatar
Yaowu Xu committed
769
          const char *detail = aom_codec_error_detail(&decoder);
clang-format's avatar
clang-format committed
770
          warn("Failed to decode frame %d: %s", frame_in,
Yaowu Xu's avatar
Yaowu Xu committed
771
               aom_codec_error(&decoder));
772

clang-format's avatar
clang-format committed
773 774
          if (detail) warn("Additional information: %s", detail);
          if (!keep_going) goto fail;
Scott LaVarnway's avatar
Scott LaVarnway committed
775 776
        }

Yaowu Xu's avatar
Yaowu Xu committed
777 778
        aom_usec_timer_mark(&timer);
        dx_time += aom_usec_timer_elapsed(&timer);
779 780
      } else {
        flush_decoder = 1;
Scott LaVarnway's avatar
Scott LaVarnway committed
781
      }
782 783
    } else {
      flush_decoder = 1;
Scott LaVarnway's avatar
Scott LaVarnway committed
784 785
    }

Yaowu Xu's avatar
Yaowu Xu committed
786
    aom_usec_timer_start(&timer);
Scott LaVarnway's avatar
Scott LaVarnway committed
787

788 789
    if (flush_decoder) {
      // Flush the decoder in frame parallel decode.
Yaowu Xu's avatar
Yaowu Xu committed
790 791
      if (aom_codec_decode(&decoder, NULL, 0, NULL, 0)) {
        warn("Failed to flush decoder: %s", aom_codec_error(&decoder));
792 793 794
      }
    }

Scott LaVarnway's avatar
Scott LaVarnway committed
795
    got_data = 0;
Yaowu Xu's avatar
Yaowu Xu committed
796
    if ((img = aom_codec_get_frame(&decoder, &iter))) {
Scott LaVarnway's avatar
Scott LaVarnway committed
797 798
      ++frame_out;
      got_data = 1;
799 800
    }

Yaowu Xu's avatar
Yaowu Xu committed
801 802
    aom_usec_timer_mark(&timer);
    dx_time += (unsigned int)aom_usec_timer_elapsed(&timer);
803

804
    if (!frame_parallel &&
Yaowu Xu's avatar
Yaowu Xu committed
805 806
        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
807
      if (!keep_going) goto fail;
808
    }
John Koleszar's avatar
John Koleszar committed
809
    frames_corrupted += corrupted;
John Koleszar's avatar
John Koleszar committed
810

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

813
    if (!noblit && img) {
Yaowu Xu's avatar
Yaowu Xu committed
814 815
      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 };
816
      const int *planes = flipuv ? PLANES_YVU : PLANES_YUV;
817

818
      if (do_scale) {
819 820 821 822
        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
823 824
          // header. If that is unavailable, use the raw decoded size of the
          // first decoded frame.
Yaowu Xu's avatar
Yaowu Xu committed
825 826
          int render_width = aom_input_ctx.width;
          int render_height = aom_input_ctx.height;
827 828
          if (!render_width || !render_height) {
            int render_size[2];
Yaowu Xu's avatar
Yaowu Xu committed
829
            if (aom_codec_control(&decoder, AV1D_GET_DISPLAY_SIZE,
830
                                  render_size)) {
831
              // As last resort use size of first frame as display size.
832 833
              render_width = img->d_w;
              render_height = img->d_h;
834
            } else {
835 836
              render_width = render_size[0];
              render_height = render_size[1];
837
            }
838
          }
clang-format's avatar
clang-format committed
839
          scaled_img =
Yaowu Xu's avatar
Yaowu Xu committed
840
              aom_img_alloc(NULL, img->fmt, render_width, render_height, 16);
841
          scaled_img->bit_depth = img->bit_depth;
John Koleszar's avatar
John Koleszar committed
842
        }
843

844
        if (img->d_w != scaled_img->d_w || img->d_h != scaled_img->d_h) {
Deb Mukherjee's avatar
Deb Mukherjee committed
845
#if CONFIG_LIBYUV
846
          libyuv_scale(img, scaled_img, kFilterBox);
John Koleszar's avatar
John Koleszar committed
847
          img = scaled_img;
Deb Mukherjee's avatar
Deb Mukherjee committed
848
#else
clang-format's avatar
clang-format committed
849 850
          fprintf(stderr,
                  "Failed  to scale output frame: %s.\n"
Deb Mukherjee's avatar
Deb Mukherjee committed
851 852
                  "Scaling is disabled in this configuration. "
                  "To enable scaling, configure with --enable-libyuv\n",
Yaowu Xu's avatar
Yaowu Xu committed
853
                  aom_codec_error(&decoder));
Deb Mukherjee's avatar
Deb Mukherjee committed
854 855
          return EXIT_FAILURE;
#endif
John Koleszar's avatar
John Koleszar committed
856 857
        }
      }
Yaowu Xu's avatar
Yaowu Xu committed
858
#if CONFIG_AOM_HIGHBITDEPTH
859
      // Default to codec bit depth if output bit depth not set
860
      if (!output_bit_depth && single_file && !do_md5) {
861 862 863
        output_bit_depth = img->bit_depth;
      }
      // Shift up or down if necessary
864
      if (output_bit_depth != 0 && output_bit_depth != img->bit_depth) {
Yaowu Xu's avatar
Yaowu Xu committed
865
        const aom_img_fmt_t shifted_fmt =
clang-format's avatar
clang-format committed
866
            output_bit_depth == 8
Yaowu Xu's avatar
Yaowu Xu committed
867 868
                ? img->fmt ^ (img->fmt & AOM_IMG_FMT_HIGHBITDEPTH)
                : img->fmt | AOM_IMG_FMT_HIGHBITDEPTH;
869 870
        if (img_shifted &&
            img_shifted_realloc_required(img, img_shifted, shifted_fmt)) {
Yaowu Xu's avatar
Yaowu Xu committed
871
          aom_img_free(img_shifted);
872 873
          img_shifted = NULL;
        }
874
        if (!img_shifted) {
clang-format's avatar
clang-format committed
875
          img_shifted =
Yaowu Xu's avatar
Yaowu Xu committed
876
              aom_img_alloc(NULL, shifted_fmt, img->d_w, img->d_h, 16);
877 878 879
          img_shifted->bit_depth = output_bit_depth;
        }
        if (output_bit_depth > img->bit_depth) {
Yaowu Xu's avatar
Yaowu Xu committed
880
          aom_img_upshift(img_shifted, img, output_bit_depth - img->bit_depth);
881
        } else {
Yaowu Xu's avatar
Yaowu Xu committed
882
          aom_img_downshift(img_shifted, img,
883
                            img->bit_depth - output_bit_depth);
884 885 886 887
        }
        img = img_shifted;
      }
#endif
888

889
#if CONFIG_EXT_TILE
Yaowu Xu's avatar
Yaowu Xu committed
890 891
      aom_input_ctx.width = img->d_w;
      aom_input_ctx.height = img->d_h;
892 893
#endif  // CONFIG_EXT_TILE

894
      if (single_file) {
895
        if (use_y4m) {
clang-format's avatar
clang-format committed
896
          char buf[Y4M_BUFFER_SIZE] = { 0 };
897
          size_t len = 0;
Yaowu Xu's avatar
Yaowu Xu committed
898
          if (img->fmt == AOM_IMG_FMT_I440 || img->fmt == AOM_IMG_FMT_I44016) {
Deb Mukherjee's avatar
Deb Mukherjee committed
899 900 901
            fprintf(stderr, "Cannot produce y4m output for 440 sampling.\n");
            goto fail;
          }