aomdec.c 33.4 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;
};

clang-format's avatar
clang-format committed
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 92 93 94 95 96 97 98 99
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
100
#if CONFIG_AOM_HIGHBITDEPTH
clang-format's avatar
clang-format committed
101 102
static const arg_def_t outbitdeptharg =
    ARG_DEF(NULL, "output-bit-depth", 1, "Output bit-depth for decoded frames");
103
#endif
104
#if CONFIG_EXT_TILE
clang-format's avatar
clang-format committed
105 106 107 108 109 110
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)");
111
#endif  // CONFIG_EXT_TILE
Dmitry Kovalev's avatar
Dmitry Kovalev committed
112

clang-format's avatar
clang-format committed
113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132
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
133
#if CONFIG_AOM_HIGHBITDEPTH
clang-format's avatar
clang-format committed
134
                                       &outbitdeptharg,
135
#endif
136
#if CONFIG_EXT_TILE
clang-format's avatar
clang-format committed
137 138
                                       &tiler,
                                       &tilec,
139
#endif  // CONFIG_EXT_TILE
clang-format's avatar
clang-format committed
140
                                       NULL };
John Koleszar's avatar
John Koleszar committed
141

142
#if CONFIG_LIBYUV
Yaowu Xu's avatar
Yaowu Xu committed
143
static INLINE int libyuv_scale(aom_image_t *src, aom_image_t *dst,
clang-format's avatar
clang-format committed
144
                               FilterModeEnum mode) {
Yaowu Xu's avatar
Yaowu Xu committed
145 146 147
#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
148
    return I420Scale_16(
Yaowu Xu's avatar
Yaowu Xu committed
149 150 151 152 153 154 155
        (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);
156 157
  }
#endif
Yaowu Xu's avatar
Yaowu Xu committed
158 159 160 161 162 163 164 165
  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
166
                   dst->d_h, mode);
167
}
168
#endif
169

170
void usage_exit(void) {
John Koleszar's avatar
John Koleszar committed
171
  int i;
John Koleszar's avatar
John Koleszar committed
172

clang-format's avatar
clang-format committed
173 174 175 176
  fprintf(stderr,
          "Usage: %s <options> filename\n\n"
          "Options:\n",
          exec_name);
John Koleszar's avatar
John Koleszar committed
177 178 179 180 181 182 183
  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
184 185
          "the following escape characters:\n");
  fprintf(stderr,
John Koleszar's avatar
John Koleszar committed
186 187 188 189 190
          "\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
191
          "not specified, the output will be\n  directed to stdout.\n");
John Koleszar's avatar
John Koleszar committed
192 193
  fprintf(stderr, "\nIncluded decoders:\n\n");

Yaowu Xu's avatar
Yaowu Xu committed
194 195
  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
196
    fprintf(stderr, "    %-6s - %s\n", decoder->name,
Yaowu Xu's avatar
Yaowu Xu committed
197
            aom_codec_iface_name(decoder->codec_interface()));
198
  }
John Koleszar's avatar
John Koleszar committed
199 200

  exit(EXIT_FAILURE);
John Koleszar's avatar
John Koleszar committed
201 202
}

clang-format's avatar
clang-format committed
203 204
static int raw_read_frame(FILE *infile, uint8_t **buffer, size_t *bytes_read,
                          size_t *buffer_size) {
205
  char raw_hdr[RAW_FRAME_HDR_SZ];
206
  size_t frame_size = 0;
John Koleszar's avatar
John Koleszar committed
207

208
  if (fread(raw_hdr, RAW_FRAME_HDR_SZ, 1, infile) != 1) {
clang-format's avatar
clang-format committed
209
    if (!feof(infile)) warn("Failed to read RAW frame size\n");
210
  } else {
211 212
    const size_t kCorruptFrameThreshold = 256 * 1024 * 1024;
    const size_t kFrameTooSmallThreshold = 256 * 1024;
213
    frame_size = mem_get_le32(raw_hdr);
John Koleszar's avatar
John Koleszar committed
214

215 216 217 218
    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
219

220 221 222
    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
223
    }
John Koleszar's avatar
John Koleszar committed
224

225 226 227 228 229 230 231 232
    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;
233
      }
John Koleszar's avatar
John Koleszar committed
234
    }
235
  }
John Koleszar's avatar
John Koleszar committed
236

237 238 239 240 241 242
  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
243 244
  }

245 246 247
  return 0;
}

Yaowu Xu's avatar
Yaowu Xu committed
248
static int read_frame(struct AvxDecInputContext *input, uint8_t **buf,
249
                      size_t *bytes_in_buffer, size_t *buffer_size) {
Yaowu Xu's avatar
Yaowu Xu committed
250
  switch (input->aom_input_ctx->file_type) {
251
#if CONFIG_WEBM_IO
252
    case FILE_TYPE_WEBM:
253
      return webm_read_frame(input->webm_ctx, buf, bytes_in_buffer);
254
#endif
255
    case FILE_TYPE_RAW:
Yaowu Xu's avatar
Yaowu Xu committed
256
      return raw_read_frame(input->aom_input_ctx->file, buf, bytes_in_buffer,
clang-format's avatar
clang-format committed
257
                            buffer_size);
258
    case FILE_TYPE_IVF:
Yaowu Xu's avatar
Yaowu Xu committed
259
      return ivf_read_frame(input->aom_input_ctx->file, buf, bytes_in_buffer,
clang-format's avatar
clang-format committed
260 261
                            buffer_size);
    default: return 1;
262
  }
John Koleszar's avatar
John Koleszar committed
263 264
}

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

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

Yaowu Xu's avatar
Yaowu Xu committed
284
static void write_image_file(const aom_image_t *img, const int planes[3],
285 286
                             FILE *file) {
  int i, y;
Yaowu Xu's avatar
Yaowu Xu committed
287 288
#if CONFIG_AOM_HIGHBITDEPTH
  const int bytes_per_sample = ((img->fmt & AOM_IMG_FMT_HIGHBITDEPTH) ? 2 : 1);
289 290 291
#else
  const int bytes_per_sample = 1;
#endif
292 293 294 295 296

  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
297 298
    const int w = aom_img_plane_width(img, plane);
    const int h = aom_img_plane_height(img, plane);
299 300

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

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

John Koleszar's avatar
John Koleszar committed
312 313
  si.sz = sizeof(si);

314
  if (fread(buf, 1, 32, input->file) == 32) {
John Koleszar's avatar
John Koleszar committed
315 316
    int i;

317
    if (mem_get_le32(buf) < 256 * 1024 * 1024) {
Yaowu Xu's avatar
Yaowu Xu committed
318 319 320
      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
321
                                        32 - 4, &si)) {
John Koleszar's avatar
John Koleszar committed
322
          is_raw = 1;
323
          input->fourcc = decoder->fourcc;
324 325 326 327
          input->width = si.w;
          input->height = si.h;
          input->framerate.numerator = 30;
          input->framerate.denominator = 1;
John Koleszar's avatar
John Koleszar committed
328 329
          break;
        }
330 331
      }
    }
John Koleszar's avatar
John Koleszar committed
332 333
  }

334
  rewind(input->file);
John Koleszar's avatar
John Koleszar committed
335
  return is_raw;
336 337
}

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

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

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

356
// Callback used by libaom to request an external frame buffer. |cb_priv|
357 358 359
// 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
360 361
static int get_av1_frame_buffer(void *cb_priv, size_t min_size,
                                aom_codec_frame_buffer_t *fb) {
362 363 364
  int i;
  struct ExternalFrameBufferList *const ext_fb_list =
      (struct ExternalFrameBufferList *)cb_priv;
clang-format's avatar
clang-format committed
365
  if (ext_fb_list == NULL) return -1;
366 367 368

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

clang-format's avatar
clang-format committed
372
  if (i == ext_fb_list->num_external_frame_buffers) return -1;
373 374 375

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

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

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

403 404 405
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
406 407 408 409 410 411 412 413 414
  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
415
      /* parse the pattern */
John Koleszar's avatar
John Koleszar committed
416 417
      q[q_len - 1] = '\0';
      switch (p[1]) {
clang-format's avatar
clang-format committed
418 419 420 421 422 423 424 425 426 427 428 429
        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
430 431 432
      }

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

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

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

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

457 458 459 460 461 462 463
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
464
    if (p) p++;
465 466 467 468 469 470 471 472
  } while (p);

  return 1;
}

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

clang-format's avatar
clang-format committed
473
  for (i = 0; i < 16; ++i) printf("%02x", digest[i]);
474 475 476 477 478 479 480 481 482
  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
483
    if (!file) fatal("Failed to open output file '%s'", name);
484 485 486 487
    return file;
  }
}

Yaowu Xu's avatar
Yaowu Xu committed
488 489 490 491
#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
492
  return img->d_w != shifted->d_w || img->d_h != shifted->d_h ||
493 494
         required_fmt != shifted->fmt;
}
495 496
#endif

497
static int main_loop(int argc, const char **argv_) {
Yaowu Xu's avatar
Yaowu Xu committed
498
  aom_codec_ctx_t decoder;
clang-format's avatar
clang-format committed
499 500
  char *fn = NULL;
  int i;
James Zern's avatar
James Zern committed
501
  int ret = EXIT_FAILURE;
clang-format's avatar
clang-format committed
502 503 504 505 506 507 508 509 510
  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
511 512
  const AvxInterface *interface = NULL;
  const AvxInterface *fourcc_interface = NULL;
513
  uint64_t dx_time = 0;
clang-format's avatar
clang-format committed
514 515 516 517 518 519 520
  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
521 522
  aom_codec_dec_cfg_t cfg = { 0, 0, 0 };
#if CONFIG_AOM_HIGHBITDEPTH
clang-format's avatar
clang-format committed
523
  unsigned int output_bit_depth = 0;
524
#endif
525
#if CONFIG_EXT_TILE
clang-format's avatar
clang-format committed
526 527
  int tile_row = -1;
  int tile_col = -1;
528
#endif  // CONFIG_EXT_TILE
clang-format's avatar
clang-format committed
529 530 531
  int frames_corrupted = 0;
  int dec_flags = 0;
  int do_scale = 0;
Yaowu Xu's avatar
Yaowu Xu committed
532 533 534
  aom_image_t *scaled_img = NULL;
#if CONFIG_AOM_HIGHBITDEPTH
  aom_image_t *img_shifted = NULL;
535
#endif
clang-format's avatar
clang-format committed
536 537 538
  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
539

540
  const char *outfile_pattern = NULL;
clang-format's avatar
clang-format committed
541
  char outfile_name[PATH_MAX] = { 0 };
542 543 544 545 546
  FILE *outfile = NULL;

  MD5Context md5_ctx;
  unsigned char md5_digest[16];

Yaowu Xu's avatar
Yaowu Xu committed
547 548
  struct AvxDecInputContext input = { NULL, NULL };
  struct AvxInputContext aom_input_ctx;
549
#if CONFIG_WEBM_IO
550 551
  struct WebmInputContext webm_ctx;
  memset(&(webm_ctx), 0, sizeof(webm_ctx));
552
  input.webm_ctx = &webm_ctx;
553
#endif
Yaowu Xu's avatar
Yaowu Xu committed
554
  input.aom_input_ctx = &aom_input_ctx;
555

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

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

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

hanno's avatar
hanno committed
635 636
  if (!fn) {
    free(argv);
John Koleszar's avatar
John Koleszar committed
637
    usage_exit();
hanno's avatar
hanno committed
638
  }
John Koleszar's avatar
John Koleszar committed
639 640
  /* Open file */
  infile = strcmp(fn, "-") ? fopen(fn, "rb") : set_binary_mode(stdin);
John Koleszar's avatar
John Koleszar committed
641

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

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

674 675
  if (!noblit && single_file) {
    generate_filename(outfile_pattern, outfile_name, PATH_MAX,
Yaowu Xu's avatar
Yaowu Xu committed
676
                      aom_input_ctx.width, aom_input_ctx.height, 0);
677 678 679 680
    if (do_md5)
      MD5Init(&md5_ctx);
    else
      outfile = open_outfile(outfile_name);
John Koleszar's avatar
John Koleszar committed
681 682 683 684
  }

  if (use_y4m && !noblit) {
    if (!single_file) {
clang-format's avatar
clang-format committed
685 686
      fprintf(stderr,
              "YUV4MPEG2 not supported with output patterns,"
Deb Mukherjee's avatar
Deb Mukherjee committed
687
              " try --i420 or --yv12 or --rawvideo.\n");
John Koleszar's avatar
John Koleszar committed
688
      return EXIT_FAILURE;
689
    }
John Koleszar's avatar
John Koleszar committed
690

691
#if CONFIG_WEBM_IO
Yaowu Xu's avatar
Yaowu Xu committed
692 693
    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
694 695
        fprintf(stderr,
                "Failed to guess framerate -- error parsing "
John Koleszar's avatar
John Koleszar committed
696 697 698
                "webm file?\n");
        return EXIT_FAILURE;
      }
699
    }
700
#endif
John Koleszar's avatar
John Koleszar committed
701 702
  }

Yaowu Xu's avatar
Yaowu Xu committed
703
  fourcc_interface = get_aom_decoder_by_fourcc(aom_input_ctx.fourcc);
704 705 706 707
  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
708

Yaowu Xu's avatar
Yaowu Xu committed
709
  if (!interface) interface = get_aom_decoder_by_index(0);
710

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

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

Yaowu Xu's avatar
Yaowu Xu committed
723 724 725
#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)) {
726
      fprintf(stderr, "Failed to set decode_tile_row: %s\n",
Yaowu Xu's avatar
Yaowu Xu committed
727
              aom_codec_error(&decoder));
James Zern's avatar
James Zern committed
728
      goto fail;
729 730
    }

Yaowu Xu's avatar
Yaowu Xu committed
731
    if (aom_codec_control(&decoder, AV1_SET_DECODE_TILE_COL, tile_col)) {
732
      fprintf(stderr, "Failed to set decode_tile_col: %s\n",
Yaowu Xu's avatar
Yaowu Xu committed
733
              aom_codec_error(&decoder));
James Zern's avatar
James Zern committed
734
      goto fail;
735 736 737
    }
  }
#endif
738

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

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

Scott LaVarnway's avatar
Scott LaVarnway committed
758 759 760
  frame_avail = 1;
  got_data = 0;

John Koleszar's avatar
John Koleszar committed
761
  /* Decode file */
Scott LaVarnway's avatar
Scott LaVarnway committed
762
  while (frame_avail || got_data) {
Yaowu Xu's avatar
Yaowu Xu committed
763 764 765
    aom_codec_iter_t iter = NULL;
    aom_image_t *img;
    struct aom_usec_timer timer;
clang-format's avatar
clang-format committed
766
    int corrupted = 0;
John Koleszar's avatar
John Koleszar committed
767

Scott LaVarnway's avatar
Scott LaVarnway committed
768 769
    frame_avail = 0;
    if (!stop_after || frame_in < stop_after) {
770
      if (!read_frame(&input, &buf, &bytes_in_buffer, &buffer_size)) {
Scott LaVarnway's avatar
Scott LaVarnway committed
771 772
        frame_avail = 1;
        frame_in++;
John Koleszar's avatar
John Koleszar committed
773

Yaowu Xu's avatar
Yaowu Xu committed
774
        aom_usec_timer_start(&timer);
John Koleszar's avatar
John Koleszar committed
775

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

clang-format's avatar
clang-format committed
782 783
          if (detail) warn("Additional information: %s", detail);
          if (!keep_going) goto fail;
Scott LaVarnway's avatar
Scott LaVarnway committed
784 785
        }

Yaowu Xu's avatar
Yaowu Xu committed
786 787
        aom_usec_timer_mark(&timer);
        dx_time += aom_usec_timer_elapsed(&timer);
788 789
      } else {
        flush_decoder = 1;
Scott LaVarnway's avatar
Scott LaVarnway committed
790
      }
791 792
    } else {
      flush_decoder = 1;
Scott LaVarnway's avatar
Scott LaVarnway committed
793 794
    }

Yaowu Xu's avatar
Yaowu Xu committed
795
    aom_usec_timer_start(&timer);
Scott LaVarnway's avatar
Scott LaVarnway committed
796

797 798
    if (flush_decoder) {
      // Flush the decoder in frame parallel decode.
Yaowu Xu's avatar
Yaowu Xu committed
799 800
      if (aom_codec_decode(&decoder, NULL, 0, NULL, 0)) {
        warn("Failed to flush decoder: %s", aom_codec_error(&decoder));
801 802 803
      }
    }

Scott LaVarnway's avatar
Scott LaVarnway committed
804
    got_data = 0;
Yaowu Xu's avatar
Yaowu Xu committed
805
    if ((img = aom_codec_get_frame(&decoder, &iter))) {
Scott LaVarnway's avatar
Scott LaVarnway committed
806 807
      ++frame_out;
      got_data = 1;
808 809
    }

Yaowu Xu's avatar
Yaowu Xu committed
810 811
    aom_usec_timer_mark(&timer);
    dx_time += (unsigned int)aom_usec_timer_elapsed(&timer);
812

813
    if (!frame_parallel &&
Yaowu Xu's avatar
Yaowu Xu committed
814 815
        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
816
      if (!keep_going) goto fail;
817
    }
John Koleszar's avatar
John Koleszar committed
818
    frames_corrupted += corrupted;
John Koleszar's avatar
John Koleszar committed
819

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

822
    if (!noblit && img) {
Yaowu Xu's avatar
Yaowu Xu committed
823 824
      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 };
825
      const int *planes = flipuv ? PLANES_YVU : PLANES_YUV;
826

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

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