aomdec.c 35.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

#include "./tools_common.h"
47
#if CONFIG_WEBM_IO
48
#include "./webmdec.h"
49
#endif
50
#include "./y4menc.h"
John Koleszar's avatar
John Koleszar committed
51 52 53

static const char *exec_name;

Yaowu Xu's avatar
Yaowu Xu committed
54 55
struct AvxDecInputContext {
  struct AvxInputContext *aom_input_ctx;
56 57 58
  struct WebmInputContext *webm_ctx;
};

James Zern's avatar
James Zern committed
59 60
static const arg_def_t help =
    ARG_DEF(NULL, "help", 0, "Show usage options and exit");
clang-format's avatar
clang-format committed
61 62 63 64 65 66 67 68 69 70 71
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");
72 73 74 75
#if CONFIG_MONO_VIDEO
static const arg_def_t monochrome = ARG_DEF(
    NULL, "monochrome", 0, "Force monochrome output (constant chroma planes)");
#endif
clang-format's avatar
clang-format committed
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
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");
104 105
static const arg_def_t framestatsarg =
    ARG_DEF(NULL, "framestats", 1, "Output per-frame stats (.csv format)");
106
#if CONFIG_HIGHBITDEPTH
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
#endif
110
#if CONFIG_EXT_TILE
clang-format's avatar
clang-format committed
111 112 113 114 115 116
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)");
117
#endif  // CONFIG_EXT_TILE
Dmitry Kovalev's avatar
Dmitry Kovalev committed
118

James Zern's avatar
James Zern committed
119 120
static const arg_def_t *all_args[] = { &help,
                                       &codecarg,
clang-format's avatar
clang-format committed
121 122 123 124
                                       &use_yv12,
                                       &use_i420,
                                       &flipuvarg,
                                       &rawvideo,
125 126 127
#if CONFIG_MONO_VIDEO
                                       &monochrome,
#endif
clang-format's avatar
clang-format committed
128 129 130 131 132 133 134 135 136 137 138 139 140
                                       &noblitarg,
                                       &progressarg,
                                       &limitarg,
                                       &skiparg,
                                       &postprocarg,
                                       &summaryarg,
                                       &outputfile,
                                       &threadsarg,
                                       &frameparallelarg,
                                       &verbosearg,
                                       &scalearg,
                                       &fb_arg,
                                       &md5arg,
141
                                       &framestatsarg,
clang-format's avatar
clang-format committed
142
                                       &continuearg,
143
#if CONFIG_HIGHBITDEPTH
clang-format's avatar
clang-format committed
144
                                       &outbitdeptharg,
145
#endif
146
#if CONFIG_EXT_TILE
clang-format's avatar
clang-format committed
147 148
                                       &tiler,
                                       &tilec,
149
#endif  // CONFIG_EXT_TILE
clang-format's avatar
clang-format committed
150
                                       NULL };
John Koleszar's avatar
John Koleszar committed
151

152
#if CONFIG_LIBYUV
Yaowu Xu's avatar
Yaowu Xu committed
153
static INLINE int libyuv_scale(aom_image_t *src, aom_image_t *dst,
clang-format's avatar
clang-format committed
154
                               FilterModeEnum mode) {
155
#if CONFIG_HIGHBITDEPTH
Yaowu Xu's avatar
Yaowu Xu committed
156 157
  if (src->fmt == AOM_IMG_FMT_I42016) {
    assert(dst->fmt == AOM_IMG_FMT_I42016);
clang-format's avatar
clang-format committed
158
    return I420Scale_16(
Yaowu Xu's avatar
Yaowu Xu committed
159 160 161 162 163 164 165
        (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);
166 167
  }
#endif
Yaowu Xu's avatar
Yaowu Xu committed
168 169 170 171 172 173 174 175
  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
176
                   dst->d_h, mode);
177
}
178
#endif
179

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

James Zern's avatar
James Zern committed
183 184 185 186 187 188 189 190
  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
191 192 193 194 195
          "\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
196
          "the following escape characters:\n");
James Zern's avatar
James Zern committed
197
  fprintf(fout,
John Koleszar's avatar
John Koleszar committed
198 199 200 201 202
          "\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
203
          "not specified, the output will be\n  directed to stdout.\n");
James Zern's avatar
James Zern committed
204
  fprintf(fout, "\nIncluded decoders:\n\n");
John Koleszar's avatar
John Koleszar committed
205

James Zern's avatar
James Zern committed
206
  for (int i = 0; i < get_aom_decoder_count(); ++i) {
Yaowu Xu's avatar
Yaowu Xu committed
207
    const AvxInterface *const decoder = get_aom_decoder_by_index(i);
James Zern's avatar
James Zern committed
208
    fprintf(fout, "    %-6s - %s\n", decoder->name,
Yaowu Xu's avatar
Yaowu Xu committed
209
            aom_codec_iface_name(decoder->codec_interface()));
210
  }
James Zern's avatar
James Zern committed
211
}
John Koleszar's avatar
John Koleszar committed
212

James Zern's avatar
James Zern committed
213 214
void usage_exit(void) {
  show_help(stderr, 1);
John Koleszar's avatar
John Koleszar committed
215
  exit(EXIT_FAILURE);
John Koleszar's avatar
John Koleszar committed
216 217
}

clang-format's avatar
clang-format committed
218 219
static int raw_read_frame(FILE *infile, uint8_t **buffer, size_t *bytes_read,
                          size_t *buffer_size) {
220
  char raw_hdr[RAW_FRAME_HDR_SZ];
221
  size_t frame_size = 0;
John Koleszar's avatar
John Koleszar committed
222

223
  if (fread(raw_hdr, RAW_FRAME_HDR_SZ, 1, infile) != 1) {
clang-format's avatar
clang-format committed
224
    if (!feof(infile)) warn("Failed to read RAW frame size\n");
225
  } else {
226 227
    const size_t kCorruptFrameThreshold = 256 * 1024 * 1024;
    const size_t kFrameTooSmallThreshold = 256 * 1024;
228
    frame_size = mem_get_le32(raw_hdr);
John Koleszar's avatar
John Koleszar committed
229

230 231 232 233
    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
234

235 236 237
    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
238
    }
John Koleszar's avatar
John Koleszar committed
239

240 241 242 243 244 245 246 247
    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;
248
      }
John Koleszar's avatar
John Koleszar committed
249
    }
250
  }
John Koleszar's avatar
John Koleszar committed
251

252 253 254 255 256 257
  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
258 259
  }

260 261 262
  return 0;
}

Yaowu Xu's avatar
Yaowu Xu committed
263
static int read_frame(struct AvxDecInputContext *input, uint8_t **buf,
264
                      size_t *bytes_in_buffer, size_t *buffer_size) {
Yaowu Xu's avatar
Yaowu Xu committed
265
  switch (input->aom_input_ctx->file_type) {
266
#if CONFIG_WEBM_IO
267
    case FILE_TYPE_WEBM:
268
      return webm_read_frame(input->webm_ctx, buf, bytes_in_buffer);
269
#endif
270
    case FILE_TYPE_RAW:
Yaowu Xu's avatar
Yaowu Xu committed
271
      return raw_read_frame(input->aom_input_ctx->file, buf, bytes_in_buffer,
clang-format's avatar
clang-format committed
272
                            buffer_size);
273
    case FILE_TYPE_IVF:
Yaowu Xu's avatar
Yaowu Xu committed
274
      return ivf_read_frame(input->aom_input_ctx->file, buf, bytes_in_buffer,
clang-format's avatar
clang-format committed
275 276
                            buffer_size);
    default: return 1;
277
  }
John Koleszar's avatar
John Koleszar committed
278 279
}

Yaowu Xu's avatar
Yaowu Xu committed
280
static void update_image_md5(const aom_image_t *img, const int planes[3],
281 282 283 284 285 286 287
                             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
288 289 290
    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);
291 292 293 294 295 296 297 298

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

299 300
static void write_image_file(const aom_image_t *img, const int *planes,
                             const int num_planes, FILE *file) {
301
  int i, y;
302
#if CONFIG_HIGHBITDEPTH
Yaowu Xu's avatar
Yaowu Xu committed
303
  const int bytes_per_sample = ((img->fmt & AOM_IMG_FMT_HIGHBITDEPTH) ? 2 : 1);
304 305 306
#else
  const int bytes_per_sample = 1;
#endif
307

308
  for (i = 0; i < num_planes; ++i) {
309 310 311
    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
312 313
    const int w = aom_img_plane_width(img, plane);
    const int h = aom_img_plane_height(img, plane);
314 315

    for (y = 0; y < h; ++y) {
316
      fwrite(buf, bytes_per_sample, w, file);
317 318
      buf += stride;
    }
John Koleszar's avatar
John Koleszar committed
319
  }
John Koleszar's avatar
John Koleszar committed
320 321
}

Yaowu Xu's avatar
Yaowu Xu committed
322
static int file_is_raw(struct AvxInputContext *input) {
323
  uint8_t buf[32];
John Koleszar's avatar
John Koleszar committed
324
  int is_raw = 0;
Yaowu Xu's avatar
Yaowu Xu committed
325
  aom_codec_stream_info_t si;
326

327
  if (fread(buf, 1, 32, input->file) == 32) {
John Koleszar's avatar
John Koleszar committed
328 329
    int i;

330
    if (mem_get_le32(buf) < 256 * 1024 * 1024) {
Yaowu Xu's avatar
Yaowu Xu committed
331 332 333
      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
334
                                        32 - 4, &si)) {
John Koleszar's avatar
John Koleszar committed
335
          is_raw = 1;
336
          input->fourcc = decoder->fourcc;
337 338 339 340
          input->width = si.w;
          input->height = si.h;
          input->framerate.numerator = 30;
          input->framerate.denominator = 1;
John Koleszar's avatar
John Koleszar committed
341 342
          break;
        }
343 344
      }
    }
John Koleszar's avatar
John Koleszar committed
345 346
  }

347
  rewind(input->file);
John Koleszar's avatar
John Koleszar committed
348
  return is_raw;
349 350
}

351
static void show_progress(int frame_in, int frame_out, uint64_t dx_time) {
352
  fprintf(stderr,
clang-format's avatar
clang-format committed
353
          "%d decoded frames/%d showed frames in %" PRId64 " us (%.2f fps)\r",
John Koleszar's avatar
John Koleszar committed
354
          frame_in, frame_out, dx_time,
355
          (double)frame_out * 1000000.0 / (double)dx_time);
356 357
}

358
struct ExternalFrameBuffer {
clang-format's avatar
clang-format committed
359
  uint8_t *data;
360 361 362 363 364 365 366 367 368
  size_t size;
  int in_use;
};

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

369
// Callback used by libaom to request an external frame buffer. |cb_priv|
370 371 372
// 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
373 374
static int get_av1_frame_buffer(void *cb_priv, size_t min_size,
                                aom_codec_frame_buffer_t *fb) {
375 376 377
  int i;
  struct ExternalFrameBufferList *const ext_fb_list =
      (struct ExternalFrameBufferList *)cb_priv;
clang-format's avatar
clang-format committed
378
  if (ext_fb_list == NULL) return -1;
379 380 381

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

clang-format's avatar
clang-format committed
385
  if (i == ext_fb_list->num_external_frame_buffers) return -1;
386 387 388

  if (ext_fb_list->ext_fb[i].size < min_size) {
    free(ext_fb_list->ext_fb[i].data);
389
    ext_fb_list->ext_fb[i].data = (uint8_t *)calloc(min_size, sizeof(uint8_t));
clang-format's avatar
clang-format committed
390
    if (!ext_fb_list->ext_fb[i].data) return -1;
391 392 393 394 395 396 397 398 399 400 401 402 403

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

404
// Callback used by libaom when there are no references to the frame buffer.
405 406
// |cb_priv| user private data passed into the set function. |fb| pointer
// to the frame buffer.
Yaowu Xu's avatar
Yaowu Xu committed
407 408
static int release_av1_frame_buffer(void *cb_priv,
                                    aom_codec_frame_buffer_t *fb) {
409 410 411 412 413 414 415
  struct ExternalFrameBuffer *const ext_fb =
      (struct ExternalFrameBuffer *)fb->priv;
  (void)cb_priv;
  ext_fb->in_use = 0;
  return 0;
}

416 417 418
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
419 420 421 422 423 424 425 426 427
  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
428
      /* parse the pattern */
John Koleszar's avatar
John Koleszar committed
429 430
      q[q_len - 1] = '\0';
      switch (p[1]) {
clang-format's avatar
clang-format committed
431 432 433 434 435 436 437 438 439 440 441 442
        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
443 444 445
      }

      pat_len = strlen(q);
clang-format's avatar
clang-format committed
446
      if (pat_len >= q_len - 1) die("Output filename too long.\n");
John Koleszar's avatar
John Koleszar committed
447 448 449 450 451 452
      q += pat_len;
      p += 2;
      q_len -= pat_len;
    } else {
      size_t copy_len;

John Koleszar's avatar
John Koleszar committed
453
      /* copy the next segment */
John Koleszar's avatar
John Koleszar committed
454 455 456 457 458
      if (!next_pat)
        copy_len = strlen(p);
      else
        copy_len = next_pat - p;

clang-format's avatar
clang-format committed
459
      if (copy_len >= q_len - 1) die("Output filename too long.\n");
John Koleszar's avatar
John Koleszar committed
460 461 462 463 464 465 466 467

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

470 471 472 473 474 475 476
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
477
    if (p) p++;
478 479 480 481 482 483 484 485
  } while (p);

  return 1;
}

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

clang-format's avatar
clang-format committed
486
  for (i = 0; i < 16; ++i) printf("%02x", digest[i]);
487 488 489 490 491 492 493 494 495
  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
496
    if (!file) fatal("Failed to open output file '%s'", name);
497 498 499 500
    return file;
  }
}

501
#if CONFIG_HIGHBITDEPTH
Yaowu Xu's avatar
Yaowu Xu committed
502 503 504
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
505
  return img->d_w != shifted->d_w || img->d_h != shifted->d_h ||
506 507
         required_fmt != shifted->fmt;
}
508 509
#endif

510
static int main_loop(int argc, const char **argv_) {
Yaowu Xu's avatar
Yaowu Xu committed
511
  aom_codec_ctx_t decoder;
clang-format's avatar
clang-format committed
512 513
  char *fn = NULL;
  int i;
James Zern's avatar
James Zern committed
514
  int ret = EXIT_FAILURE;
clang-format's avatar
clang-format committed
515 516 517 518 519 520 521 522
  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
523 524
  const AvxInterface *interface = NULL;
  const AvxInterface *fourcc_interface = NULL;
525
  uint64_t dx_time = 0;
clang-format's avatar
clang-format committed
526 527 528 529 530 531 532
  struct arg arg;
  char **argv, **argi, **argj;

  int single_file;
  int use_y4m = 1;
  int opt_yv12 = 0;
  int opt_i420 = 0;
533 534 535 536
#if CONFIG_MONO_VIDEO
  int opt_rawvideo = 0;
#endif
  aom_codec_dec_cfg_t cfg = { 0, 0, 0, CONFIG_LOWBITDEPTH, 0 };
537
#if CONFIG_HIGHBITDEPTH
clang-format's avatar
clang-format committed
538
  unsigned int output_bit_depth = 0;
539
#endif
540
#if CONFIG_EXT_TILE
clang-format's avatar
clang-format committed
541 542
  int tile_row = -1;
  int tile_col = -1;
543
#endif  // CONFIG_EXT_TILE
clang-format's avatar
clang-format committed
544 545 546
  int frames_corrupted = 0;
  int dec_flags = 0;
  int do_scale = 0;
Yaowu Xu's avatar
Yaowu Xu committed
547
  aom_image_t *scaled_img = NULL;
548
#if CONFIG_HIGHBITDEPTH
Yaowu Xu's avatar
Yaowu Xu committed
549
  aom_image_t *img_shifted = NULL;
550
#endif
clang-format's avatar
clang-format committed
551 552 553
  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
554

555
  const char *outfile_pattern = NULL;
clang-format's avatar
clang-format committed
556
  char outfile_name[PATH_MAX] = { 0 };
557 558
  FILE *outfile = NULL;

559 560
  FILE *framestats_file = NULL;

561 562 563
  MD5Context md5_ctx;
  unsigned char md5_digest[16];

Yaowu Xu's avatar
Yaowu Xu committed
564 565
  struct AvxDecInputContext input = { NULL, NULL };
  struct AvxInputContext aom_input_ctx;
566
#if CONFIG_WEBM_IO
567 568
  struct WebmInputContext webm_ctx;
  memset(&(webm_ctx), 0, sizeof(webm_ctx));
569
  input.webm_ctx = &webm_ctx;
570
#endif
Yaowu Xu's avatar
Yaowu Xu committed
571
  input.aom_input_ctx = &aom_input_ctx;
572

John Koleszar's avatar
John Koleszar committed
573 574 575 576 577 578 579 580
  /* 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
581 582 583 584
    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
585
      interface = get_aom_decoder_by_name(arg.val);
586 587
      if (!interface)
        die("Error: Unrecognized argument (%s) to --codec\n", arg.val);
John Koleszar's avatar
John Koleszar committed
588 589
    } else if (arg_match(&arg, &looparg, argi)) {
      // no-op
590
    } else if (arg_match(&arg, &outputfile, argi)) {
John Koleszar's avatar
John Koleszar committed
591
      outfile_pattern = arg.val;
592
    } else if (arg_match(&arg, &use_yv12, argi)) {
John Koleszar's avatar
John Koleszar committed
593 594
      use_y4m = 0;
      flipuv = 1;
595
      opt_yv12 = 1;
John Koleszar's avatar
John Koleszar committed
596 597 598
    } else if (arg_match(&arg, &use_i420, argi)) {
      use_y4m = 0;
      flipuv = 0;
599 600 601
      opt_i420 = 1;
    } else if (arg_match(&arg, &rawvideo, argi)) {
      use_y4m = 0;
602 603 604 605 606
#if CONFIG_MONO_VIDEO
      opt_rawvideo = 1;
    } else if (arg_match(&arg, &monochrome, argi)) {
      cfg.monochrome = 1;
#endif
607
    } else if (arg_match(&arg, &flipuvarg, argi)) {
John Koleszar's avatar
John Koleszar committed
608
      flipuv = 1;
609
    } else if (arg_match(&arg, &noblitarg, argi)) {
John Koleszar's avatar
John Koleszar committed
610
      noblit = 1;
611
    } else if (arg_match(&arg, &progressarg, argi)) {
John Koleszar's avatar
John Koleszar committed
612
      progress = 1;
613
    } else if (arg_match(&arg, &limitarg, argi)) {
John Koleszar's avatar
John Koleszar committed
614
      stop_after = arg_parse_uint(&arg);
615
    } else if (arg_match(&arg, &skiparg, argi)) {
616
      arg_skip = arg_parse_uint(&arg);
617
    } else if (arg_match(&arg, &postprocarg, argi)) {
John Koleszar's avatar
John Koleszar committed
618
      postproc = 1;
619
    } else if (arg_match(&arg, &md5arg, argi)) {
John Koleszar's avatar
John Koleszar committed
620
      do_md5 = 1;
621 622 623 624 625 626 627
    } 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
628
      summary = 1;
629
    } else if (arg_match(&arg, &threadsarg, argi)) {
John Koleszar's avatar
John Koleszar committed
630
      cfg.threads = arg_parse_uint(&arg);
631
    }
Yaowu Xu's avatar
Yaowu Xu committed
632
#if CONFIG_AV1_DECODER
633 634 635
    else if (arg_match(&arg, &frameparallelarg, argi))
      frame_parallel = 1;
#endif
John Koleszar's avatar
John Koleszar committed
636 637
    else if (arg_match(&arg, &verbosearg, argi))
      quiet = 0;
John Koleszar's avatar
John Koleszar committed
638 639
    else if (arg_match(&arg, &scalearg, argi))
      do_scale = 1;
640 641
    else if (arg_match(&arg, &fb_arg, argi))
      num_external_frame_buffers = arg_parse_uint(&arg);
642 643
    else if (arg_match(&arg, &continuearg, argi))
      keep_going = 1;
644
#if CONFIG_HIGHBITDEPTH
645 646 647 648
    else if (arg_match(&arg, &outbitdeptharg, argi)) {
      output_bit_depth = arg_parse_uint(&arg);
    }
#endif
649 650 651 652 653 654
#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
655 656 657
    else
      argj++;
  }
John Koleszar's avatar
John Koleszar committed
658

John Koleszar's avatar
John Koleszar committed
659 660 661 662
  /* 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
663

John Koleszar's avatar
John Koleszar committed
664 665
  /* Handle non-option arguments */
  fn = argv[0];
John Koleszar's avatar
John Koleszar committed
666

hanno's avatar
hanno committed
667 668
  if (!fn) {
    free(argv);
James Zern's avatar
James Zern committed
669
    fprintf(stderr, "No input file specified!\n");
John Koleszar's avatar
John Koleszar committed
670
    usage_exit();
hanno's avatar
hanno committed
671
  }
John Koleszar's avatar
John Koleszar committed
672 673
  /* Open file */
  infile = strcmp(fn, "-") ? fopen(fn, "rb") : set_binary_mode(stdin);
John Koleszar's avatar
John Koleszar committed
674

John Koleszar's avatar
John Koleszar committed
675
  if (!infile) {
676
    fatal("Failed to open input file '%s'", strcmp(fn, "-") ? fn : "stdin");
John Koleszar's avatar
John Koleszar committed
677
  }
678
#if CONFIG_OS_SUPPORT
John Koleszar's avatar
John Koleszar committed
679
  /* Make sure we don't dump to the terminal, unless forced to with -o - */
James Zern's avatar
James Zern committed
680
  if (!outfile_pattern && isatty(STDOUT_FILENO) && !do_md5 && !noblit) {
John Koleszar's avatar
John Koleszar committed
681 682 683 684 685
    fprintf(stderr,
            "Not dumping raw video to your terminal. Use '-o -' to "
            "override.\n");
    return EXIT_FAILURE;
  }
686
#endif
Yaowu Xu's avatar
Yaowu Xu committed
687 688 689
  input.aom_input_ctx->file = infile;
  if (file_is_ivf(input.aom_input_ctx))
    input.aom_input_ctx->file_type = FILE_TYPE_IVF;
690
#if CONFIG_WEBM_IO
Yaowu Xu's avatar
Yaowu Xu committed
691 692
  else if (file_is_webm(input.webm_ctx, input.aom_input_ctx))
    input.aom_input_ctx->file_type = FILE_TYPE_WEBM;
693
#endif
Yaowu Xu's avatar
Yaowu Xu committed
694 695
  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
696 697
  else {
    fprintf(stderr, "Unrecognized input file type.\n");
698
#if !CONFIG_WEBM_IO
Yaowu Xu's avatar
Yaowu Xu committed
699
    fprintf(stderr, "aomdec was built without WebM container support.\n");
700
#endif
John Koleszar's avatar
John Koleszar committed
701 702 703 704
    return EXIT_FAILURE;
  }

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

707 708
  if (!noblit && single_file) {
    generate_filename(outfile_pattern, outfile_name, PATH_MAX,
Yaowu Xu's avatar
Yaowu Xu committed
709
                      aom_input_ctx.width, aom_input_ctx.height, 0);
710 711 712 713
    if (do_md5)
      MD5Init(&md5_ctx);
    else
      outfile = open_outfile(outfile_name);
John Koleszar's avatar
John Koleszar committed
714 715 716 717
  }

  if (use_y4m && !noblit) {
    if (!single_file) {
clang-format's avatar
clang-format committed
718 719
      fprintf(stderr,
              "YUV4MPEG2 not supported with output patterns,"
Deb Mukherjee's avatar
Deb Mukherjee committed
720
              " try --i420 or --yv12 or --rawvideo.\n");
John Koleszar's avatar
John Koleszar committed
721
      return EXIT_FAILURE;
722
    }
John Koleszar's avatar
John Koleszar committed
723

724
#if CONFIG_WEBM_IO
Yaowu Xu's avatar
Yaowu Xu committed
725 726
    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
727 728
        fprintf(stderr,
                "Failed to guess framerate -- error parsing "
John Koleszar's avatar
John Koleszar committed
729 730 731
                "webm file?\n");
        return EXIT_FAILURE;
      }
732
    }
733
#endif
John Koleszar's avatar
John Koleszar committed
734 735
  }

Yaowu Xu's avatar
Yaowu Xu committed
736
  fourcc_interface = get_aom_decoder_by_fourcc(aom_input_ctx.fourcc);
737 738 739 740
  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
741

Yaowu Xu's avatar
Yaowu Xu committed
742
  if (!interface) interface = get_aom_decoder_by_index(0);
743

Yaowu Xu's avatar
Yaowu Xu committed
744 745 746
  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
747
                         dec_flags)) {
748
    fprintf(stderr, "Failed to initialize decoder: %s\n",
Yaowu Xu's avatar
Yaowu Xu committed
749
            aom_codec_error(&decoder));
James Zern's avatar
James Zern committed
750
    goto fail2;
John Koleszar's avatar
John Koleszar committed
751
  }
John Koleszar's avatar
John Koleszar committed
752

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

Yaowu Xu's avatar
Yaowu Xu committed
755
#if CONFIG_AV1_DECODER && CONFIG_EXT_TILE
756 757 758 759 760
  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;
  }
761

762 763 764 765
  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;
766 767
  }
#endif
768

clang-format's avatar
clang-format committed
769
  if (arg_skip) fprintf(stderr, "Skipping first %d frames.\n", arg_skip);
770
  while (arg_skip) {
clang-format's avatar
clang-format committed
771
    if (read_frame(&input, &buf, &bytes_in_buffer, &buffer_size)) break;
772 773 774
    arg_skip--;
  }

775 776 777 778
  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
779 780
    if (aom_codec_set_frame_buffer_functions(&decoder, get_av1_frame_buffer,
                                             release_av1_frame_buffer,
clang-format's avatar
clang-format committed
781
                                             &ext_fb_list)) {
782
      fprintf(stderr, "Failed to configure external frame buffers: %s\n",
Yaowu Xu's avatar
Yaowu Xu committed
783
              aom_codec_error(&decoder));
James Zern's avatar
James Zern committed
784
      goto fail;
785 786 787
    }
  }

Scott LaVarnway's avatar
Scott LaVarnway committed
788 789 790
  frame_avail = 1;
  got_data = 0;

791 792
  if (framestats_file) fprintf(framestats_file, "bytes,qp\r\n");

John Koleszar's avatar
John Koleszar committed
793
  /* Decode file */
Scott LaVarnway's avatar
Scott LaVarnway committed
794
  while (frame_avail || got_data) {
Yaowu Xu's avatar
Yaowu Xu committed
795 796 797
    aom_codec_iter_t iter = NULL;
    aom_image_t *img;
    struct aom_usec_timer timer;
clang-format's avatar
clang-format committed
798
    int corrupted = 0;
John Koleszar's avatar
John Koleszar committed
799

Scott LaVarnway's avatar
Scott LaVarnway committed
800 801
    frame_avail = 0;
    if (!stop_after || frame_in < stop_after) {
802
      if (!read_frame(&input, &buf, &bytes_in_buffer, &buffer_size)) {
Scott LaVarnway's avatar
Scott LaVarnway committed
803 804
        frame_avail = 1;
        frame_in++;
John Koleszar's avatar
John Koleszar committed
805

Yaowu Xu's avatar
Yaowu Xu committed
806
        aom_usec_timer_start(&timer);
John Koleszar's avatar
John Koleszar committed
807

Yaowu Xu's avatar
Yaowu Xu committed
808
        if (aom_codec_decode(&decoder, buf, (unsigned int)bytes_in_buffer, NULL,
clang-format's avatar
clang-format committed
809
                             0)) {
Yaowu Xu's avatar
Yaowu Xu committed
810
          const char *detail = aom_codec_error_detail(&decoder);
clang-format's avatar
clang-format committed
811
          warn("Failed to decode frame %d: %s", frame_in,
Yaowu Xu's avatar
Yaowu Xu committed
812
               aom_codec_error(&decoder));
813

clang-format's avatar
clang-format committed
814 815
          if (detail) warn("Additional information: %s", detail);
          if (!keep_going) goto fail;
Scott LaVarnway's avatar
Scott LaVarnway committed
816 817
        }

818 819 820 821 822 823 824 825 826 827
        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
828 829
        aom_usec_timer_mark(&timer);
        dx_time += aom_usec_timer_elapsed(&timer);
830 831
      } else {
        flush_decoder = 1;
Scott LaVarnway's avatar
Scott LaVarnway committed
832
      }
833 834
    } else {
      flush_decoder = 1;
Scott LaVarnway's avatar
Scott LaVarnway committed
835 836
    }

Yaowu Xu's avatar
Yaowu Xu committed
837
    aom_usec_timer_start(&timer);
Scott LaVarnway's avatar
Scott LaVarnway committed
838

839 840
    if (flush_decoder) {
      // Flush the decoder in frame parallel decode.
Yaowu Xu's avatar
Yaowu Xu committed
841 842
      if (aom_codec_decode(&decoder, NULL, 0, NULL, 0)) {
        warn("Failed to flush decoder: %s", aom_codec_error(&decoder));
843 844 845
      }
    }

Scott LaVarnway's avatar
Scott LaVarnway committed
846
    got_data = 0;
Yaowu Xu's avatar
Yaowu Xu committed
847
    if ((img = aom_codec_get_frame(&decoder, &iter))) {
Scott LaVarnway's avatar
Scott LaVarnway committed
848 849
      ++frame_out;
      got_data = 1;
850 851
    }

Yaowu Xu's avatar
Yaowu Xu committed
852 853
    aom_usec_timer_mark(&timer);
    dx_time += (unsigned int)aom_usec_timer_elapsed(&timer);
854

855
    if (!frame_parallel &&
Yaowu Xu's avatar
Yaowu Xu committed
856 857
        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
858
      if (!keep_going) goto fail;
859
    }
John Koleszar's avatar
John Koleszar committed
860
    frames_corrupted += corrupted;
John Koleszar's avatar
John Koleszar committed
861

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

864
    if (!noblit && img) {
Yaowu Xu's avatar
Yaowu Xu committed
865 866
      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 };
867
      const int *planes = flipuv ? PLANES_YVU : PLANES_YUV;
868

869
      if (do_scale) {
870 871 872 873
        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
874 875
          // header. If that is unavailable, use the raw decoded size of the
          // first decoded frame.
Yaowu Xu's avatar
Yaowu Xu committed
876 877
          int render_width = aom_input_ctx.width;
          int render_height = aom_input_ctx.height;
878 879
          if (!render_width || !render_height) {
            int render_size[2];
Yaowu Xu's avatar
Yaowu Xu committed
880
            if (aom_codec_control(&decoder, AV1D_GET_DISPLAY_SIZE,
881
                                  render_size)) {
882
              // As last resort use size of first frame as display size.
883 884
              render_width = img->d_w;
              render_height = img->d_h;
885
            } else {
886 887
              render_width = render_size[0];
              render_height = render_size[1];
888
            }
889
          }
clang-format's avatar
clang-format committed
890
          scaled_img =
Yaowu Xu's avatar
Yaowu Xu committed
891
              aom_img_alloc(NULL, img->fmt, render_width, render_height, 16);
892
          scaled_img->bit_depth = img->bit_depth;
John Koleszar's avatar
John Koleszar committed
893
        }
894

895
        if (img->d_w != scaled_img->d_w || img->d_h != scaled_img->d_h) {
Deb Mukherjee's avatar
Deb Mukherjee committed
896
#if CONFIG_LIBYUV
897
          libyuv_scale(img, scaled_img, kFilterBox);
John Koleszar's avatar
John Koleszar committed
898
          img = scaled_img;
Deb Mukherjee's avatar
Deb Mukherjee committed
899
#else
clang-format's avatar
clang-format committed
900 901
          fprintf(stderr,
                  "Failed  to scale output frame: %s.\n"
Deb Mukherjee's avatar
Deb Mukherjee committed
902 903
                  "Scaling is disabled in this configuration. "
                  "To enable scaling, configure with --enable-libyuv\n",
Yaowu Xu's avatar
Yaowu Xu committed
904
                  aom_codec_error(&decoder));
James Zern's avatar
James Zern committed
905
          goto fail;
Deb Mukherjee's avatar
Deb Mukherjee committed
906
#endif
John Koleszar's avatar
John Koleszar committed
907 908
        }
      }
909
#if CONFIG_HIGHBITDEPTH
910
      // Default to codec bit depth if output bit depth not set
911
      if (!output_bit_depth && single_file && !do_md5) {
912 913 914
        output_bit_depth = img->bit_depth;
      }
      // Shift up or down if necessary
Sebastien Alaiwan's avatar
Sebastien Alaiwan committed
915
      if (output_bit_depth != 0) {
Yaowu Xu's avatar
Yaowu Xu committed
916
        const aom_img_fmt_t shifted_fmt =
clang-format's avatar
clang-format committed
917
            output_bit_depth == 8
Yaowu Xu's avatar
Yaowu Xu committed
918 919
                ? img->fmt ^ (img->fmt & AOM_IMG_FMT_HIGHBITDEPTH)
                : img->fmt | AOM_IMG_FMT_HIGHBITDEPTH;
Sebastien Alaiwan's avatar
Sebastien Alaiwan committed
920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939

        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;
940 941 942
        }
      }
#endif
943

944
#if CONFIG_EXT_TILE
Yaowu Xu's avatar
Yaowu Xu committed
945 946
      aom_input_ctx.width = img->d_w;
      aom_input_ctx.height = img->d_h;
947 948
#endif  // CONFIG_EXT_TILE

949 950 951 952 953 954
#if CONFIG_MONO_VIDEO
      int num_planes = (opt_rawvideo && cfg.monochrome) ? 1 : 3;
#else
      int num_planes = 3;
#endif

955
      if (single_file) {
956
        if (use_y4m) {
957
          char y4m_buf[Y4M_BUFFER_SIZE] = { 0 };
958
          size_t len = 0;
Yaowu Xu's avatar
Yaowu Xu committed
959
          if (img->fmt == AOM_IMG_FMT_I440 || img->fmt == AOM_IMG_FMT_I44016) {
Deb Mukherjee's avatar
Deb Mukherjee committed
960 961 962
            fprintf(stderr, "Cannot produce y4m output for 440 sampling.\n");
            goto fail;
          }
963 964
          if (frame_out == 1) {
            // Y4M file header
clang-format's avatar
clang-format committed
965
            len = y4m_write_file_header(
966 967 968
                y4m_buf, sizeof(y4m_buf), aom_input_ctx.width,
                aom_input_ctx.height, &aom_input_ctx.framerate, img->fmt,
                img->bit_depth);
969
            if (do_md5) {
970
              MD5Update(&md5_ctx, (md5byte *)y4m_buf, (unsigned int)len);
971
            } else {
972
              fputs(y4m_buf, outfile);
973 974 975 976
            }
          }

          // Y4M frame header
977
          len = y4m_write_frame_header(y4m_buf, sizeof(y4m_buf));