vp9_cx_iface.c 40.3 KB
Newer Older
John Koleszar's avatar
John Koleszar committed
1
/*
2
 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
John Koleszar's avatar
John Koleszar committed
3
 *
4
 *  Use of this source code is governed by a BSD-style license
5 6
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
7
 *  in the file PATENTS.  All contributing project authors may
8
 *  be found in the AUTHORS file in the root of the source tree.
John Koleszar's avatar
John Koleszar committed
9 10
 */

11 12
#include <stdlib.h>
#include <string.h>
John Koleszar's avatar
John Koleszar committed
13

14 15
#include "vpx/vpx_codec.h"
#include "vpx/internal/vpx_codec_internal.h"
16
#include "./vpx_version.h"
17
#include "vp9/encoder/vp9_onyx_int.h"
John Koleszar's avatar
John Koleszar committed
18
#include "vpx/vp8cx.h"
19 20
#include "vp9/encoder/vp9_firstpass.h"
#include "vp9/common/vp9_onyx.h"
John Koleszar's avatar
John Koleszar committed
21
#include "vp9/vp9_iface_common.h"
John Koleszar's avatar
John Koleszar committed
22

23
struct vp9_extracfg {
John Koleszar's avatar
John Koleszar committed
24
  struct vpx_codec_pkt_list *pkt_list;
25 26
  int                         cpu_used;  /* available cpu percentage in 1/16 */
  unsigned int                enable_auto_alt_ref;
John Koleszar's avatar
John Koleszar committed
27 28 29
  unsigned int                noise_sensitivity;
  unsigned int                Sharpness;
  unsigned int                static_thresh;
30
  unsigned int                tile_columns;
Ronald S. Bultje's avatar
Ronald S. Bultje committed
31
  unsigned int                tile_rows;
32 33 34
  unsigned int                arnr_max_frames;
  unsigned int                arnr_strength;
  unsigned int                arnr_type;
John Koleszar's avatar
John Koleszar committed
35 36 37 38
  unsigned int                experimental;
  vp8e_tuning                 tuning;
  unsigned int                cq_level;         /* constrained quality level */
  unsigned int                rc_max_intra_bitrate_pct;
John Koleszar's avatar
John Koleszar committed
39
  unsigned int                lossless;
40
  unsigned int                frame_parallel_decoding_mode;
John Koleszar's avatar
John Koleszar committed
41 42
};

John Koleszar's avatar
John Koleszar committed
43 44
struct extraconfig_map {
  int                 usage;
45
  struct vp9_extracfg cfg;
John Koleszar's avatar
John Koleszar committed
46 47
};

John Koleszar's avatar
John Koleszar committed
48 49 50
static const struct extraconfig_map extracfg_map[] = {
  {
    0,
51
    { // NOLINT
John Koleszar's avatar
John Koleszar committed
52 53
      NULL,
      0,                          /* cpu_used      */
54
      1,                          /* enable_auto_alt_ref */
John Koleszar's avatar
John Koleszar committed
55 56 57
      0,                          /* noise_sensitivity */
      0,                          /* Sharpness */
      0,                          /* static_thresh */
58
      0,                          /* tile_columns */
Ronald S. Bultje's avatar
Ronald S. Bultje committed
59
      0,                          /* tile_rows */
60 61
      7,                          /* arnr_max_frames */
      5,                          /* arnr_strength */
John Koleszar's avatar
John Koleszar committed
62 63 64 65 66
      3,                          /* arnr_type*/
      0,                          /* experimental mode */
      0,                          /* tuning*/
      10,                         /* cq_level */
      0,                          /* rc_max_intra_bitrate_pct */
John Koleszar's avatar
John Koleszar committed
67
      0,                          /* lossless */
68
      0,                          /* frame_parallel_decoding_mode */
John Koleszar's avatar
John Koleszar committed
69
    }
John Koleszar's avatar
John Koleszar committed
70
  }
John Koleszar's avatar
John Koleszar committed
71 72
};

John Koleszar's avatar
John Koleszar committed
73 74 75
struct vpx_codec_alg_priv {
  vpx_codec_priv_t        base;
  vpx_codec_enc_cfg_t     cfg;
76
  struct vp9_extracfg     vp8_cfg;
77
  VP9_CONFIG              oxcf;
78
  VP9_PTR             cpi;
John Koleszar's avatar
John Koleszar committed
79 80
  unsigned char          *cx_data;
  unsigned int            cx_data_sz;
81 82
  unsigned char          *pending_cx_data;
  unsigned int            pending_cx_data_sz;
John Koleszar's avatar
John Koleszar committed
83 84 85
  int                     pending_frame_count;
  uint32_t                pending_frame_sizes[8];
  uint32_t                pending_frame_magnitude;
John Koleszar's avatar
John Koleszar committed
86 87
  vpx_image_t             preview_img;
  vp8_postproc_cfg_t      preview_ppcfg;
88
  vpx_codec_pkt_list_decl(64) pkt_list;
John Koleszar's avatar
John Koleszar committed
89
  unsigned int                fixed_kf_cntr;
John Koleszar's avatar
John Koleszar committed
90 91
};

92
static VP9_REFFRAME ref_frame_to_vp9_reframe(vpx_ref_frame_type_t frame) {
93 94 95 96 97 98 99 100 101 102 103
  switch (frame) {
    case VP8_LAST_FRAME:
      return VP9_LAST_FLAG;
    case VP8_GOLD_FRAME:
      return VP9_GOLD_FLAG;
    case VP8_ALTR_FRAME:
      return VP9_ALT_FLAG;
  }
  assert(!"Invalid Reference Frame");
  return VP9_LAST_FLAG;
}
John Koleszar's avatar
John Koleszar committed
104 105 106

static vpx_codec_err_t
update_error_state(vpx_codec_alg_priv_t                 *ctx,
John Koleszar's avatar
John Koleszar committed
107 108
                   const struct vpx_internal_error_info *error) {
  vpx_codec_err_t res;
John Koleszar's avatar
John Koleszar committed
109

John Koleszar's avatar
John Koleszar committed
110 111 112 113
  if ((res = error->error_code))
    ctx->base.err_detail = error->has_detail
                           ? error->detail
                           : NULL;
John Koleszar's avatar
John Koleszar committed
114

John Koleszar's avatar
John Koleszar committed
115
  return res;
John Koleszar's avatar
John Koleszar committed
116 117 118
}


119
#undef ERROR
John Koleszar's avatar
John Koleszar committed
120
#define ERROR(str) do {\
John Koleszar's avatar
John Koleszar committed
121 122
    ctx->base.err_detail = str;\
    return VPX_CODEC_INVALID_PARAM;\
123
  } while (0)
John Koleszar's avatar
John Koleszar committed
124

125 126
#define RANGE_CHECK(p, memb, lo, hi) do {\
    if (!(((p)->memb == lo || (p)->memb > (lo)) && (p)->memb <= hi)) \
John Koleszar's avatar
John Koleszar committed
127
      ERROR(#memb " out of range ["#lo".."#hi"]");\
128
  } while (0)
John Koleszar's avatar
John Koleszar committed
129

130 131
#define RANGE_CHECK_HI(p, memb, hi) do {\
    if (!((p)->memb <= (hi))) \
John Koleszar's avatar
John Koleszar committed
132
      ERROR(#memb " out of range [.."#hi"]");\
133
  } while (0)
134

135 136
#define RANGE_CHECK_LO(p, memb, lo) do {\
    if (!((p)->memb >= (lo))) \
John Koleszar's avatar
John Koleszar committed
137
      ERROR(#memb " out of range ["#lo"..]");\
138
  } while (0)
John Koleszar's avatar
John Koleszar committed
139

140 141 142
#define RANGE_CHECK_BOOL(p, memb) do {\
    if (!!((p)->memb) != (p)->memb) ERROR(#memb " expected boolean");\
  } while (0)
John Koleszar's avatar
John Koleszar committed
143 144 145

static vpx_codec_err_t validate_config(vpx_codec_alg_priv_t      *ctx,
                                       const vpx_codec_enc_cfg_t *cfg,
146
                                       const struct vp9_extracfg *vp8_cfg) {
Jingning Han's avatar
Jingning Han committed
147 148
  RANGE_CHECK(cfg, g_w,                   1, 65535); /* 16 bits available */
  RANGE_CHECK(cfg, g_h,                   1, 65535); /* 16 bits available */
John Koleszar's avatar
John Koleszar committed
149 150 151
  RANGE_CHECK(cfg, g_timebase.den,        1, 1000000000);
  RANGE_CHECK(cfg, g_timebase.num,        1, cfg->g_timebase.den);
  RANGE_CHECK_HI(cfg, g_profile,          3);
John Koleszar's avatar
John Koleszar committed
152

John Koleszar's avatar
John Koleszar committed
153 154
  RANGE_CHECK_HI(cfg, rc_max_quantizer,   63);
  RANGE_CHECK_HI(cfg, rc_min_quantizer,   cfg->rc_max_quantizer);
John Koleszar's avatar
John Koleszar committed
155 156 157 158 159 160
  RANGE_CHECK_BOOL(vp8_cfg, lossless);
  if (vp8_cfg->lossless) {
    RANGE_CHECK_HI(cfg, rc_max_quantizer, 0);
    RANGE_CHECK_HI(cfg, rc_min_quantizer, 0);
  }

John Koleszar's avatar
John Koleszar committed
161 162
  RANGE_CHECK_HI(cfg, g_threads,          64);
  RANGE_CHECK_HI(cfg, g_lag_in_frames,    MAX_LAG_BUFFERS);
163
  RANGE_CHECK(cfg, rc_end_usage,          VPX_VBR, VPX_Q);
John Koleszar's avatar
John Koleszar committed
164 165 166 167 168 169 170 171 172 173 174
  RANGE_CHECK_HI(cfg, rc_undershoot_pct,  1000);
  RANGE_CHECK_HI(cfg, rc_overshoot_pct,   1000);
  RANGE_CHECK_HI(cfg, rc_2pass_vbr_bias_pct, 100);
  RANGE_CHECK(cfg, kf_mode,               VPX_KF_DISABLED, VPX_KF_AUTO);
  // RANGE_CHECK_BOOL(cfg,                 g_delete_firstpassfile);
  RANGE_CHECK_BOOL(cfg,                   rc_resize_allowed);
  RANGE_CHECK_HI(cfg, rc_dropframe_thresh,   100);
  RANGE_CHECK_HI(cfg, rc_resize_up_thresh,   100);
  RANGE_CHECK_HI(cfg, rc_resize_down_thresh, 100);
  RANGE_CHECK(cfg,        g_pass,         VPX_RC_ONE_PASS, VPX_RC_LAST_PASS);

175 176
  RANGE_CHECK(cfg, ss_number_layers,      1,
              VPX_SS_MAX_LAYERS); /*Spatial layers max */
John Koleszar's avatar
John Koleszar committed
177 178 179 180 181 182 183 184 185 186 187 188 189
  /* VP8 does not support a lower bound on the keyframe interval in
   * automatic keyframe placement mode.
   */
  if (cfg->kf_mode != VPX_KF_DISABLED && cfg->kf_min_dist != cfg->kf_max_dist
      && cfg->kf_min_dist > 0)
    ERROR("kf_min_dist not supported in auto mode, use 0 "
          "or kf_max_dist instead.");

  RANGE_CHECK_BOOL(vp8_cfg,               enable_auto_alt_ref);
  RANGE_CHECK(vp8_cfg, cpu_used,           -16, 16);

  RANGE_CHECK_HI(vp8_cfg, noise_sensitivity,  6);

190
  RANGE_CHECK(vp8_cfg, tile_columns, 0, 6);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
191
  RANGE_CHECK(vp8_cfg, tile_rows, 0, 2);
John Koleszar's avatar
John Koleszar committed
192 193 194 195 196 197 198 199
  RANGE_CHECK_HI(vp8_cfg, Sharpness,       7);
  RANGE_CHECK(vp8_cfg, arnr_max_frames, 0, 15);
  RANGE_CHECK_HI(vp8_cfg, arnr_strength,   6);
  RANGE_CHECK(vp8_cfg, arnr_type,       1, 3);
  RANGE_CHECK(vp8_cfg, cq_level, 0, 63);

  if (cfg->g_pass == VPX_RC_LAST_PASS) {
    size_t           packet_sz = sizeof(FIRSTPASS_STATS);
200
    int              n_packets = (int)(cfg->rc_twopass_stats_in.sz / packet_sz);
John Koleszar's avatar
John Koleszar committed
201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219
    FIRSTPASS_STATS *stats;

    if (!cfg->rc_twopass_stats_in.buf)
      ERROR("rc_twopass_stats_in.buf not set.");

    if (cfg->rc_twopass_stats_in.sz % packet_sz)
      ERROR("rc_twopass_stats_in.sz indicates truncated packet.");

    if (cfg->rc_twopass_stats_in.sz < 2 * packet_sz)
      ERROR("rc_twopass_stats_in requires at least two packets.");

    stats = (void *)((char *)cfg->rc_twopass_stats_in.buf
                     + (n_packets - 1) * packet_sz);

    if ((int)(stats->count + 0.5) != n_packets - 1)
      ERROR("rc_twopass_stats_in missing EOS stats packet");
  }

  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
220 221 222 223
}


static vpx_codec_err_t validate_img(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
224 225
                                    const vpx_image_t    *img) {
  switch (img->fmt) {
James Zern's avatar
James Zern committed
226 227
    case VPX_IMG_FMT_YV12:
    case VPX_IMG_FMT_I420:
228 229
    case VPX_IMG_FMT_I422:
    case VPX_IMG_FMT_I444:
John Koleszar's avatar
John Koleszar committed
230
      break;
John Koleszar's avatar
John Koleszar committed
231
    default:
232 233
      ERROR("Invalid image format. Only YV12, I420, I422, I444 images are "
            "supported.");
John Koleszar's avatar
John Koleszar committed
234
  }
John Koleszar's avatar
John Koleszar committed
235

John Koleszar's avatar
John Koleszar committed
236 237
  if ((img->d_w != ctx->cfg.g_w) || (img->d_h != ctx->cfg.g_h))
    ERROR("Image size must match encoder init configuration size");
John Koleszar's avatar
John Koleszar committed
238

John Koleszar's avatar
John Koleszar committed
239
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
240 241 242
}


243
static vpx_codec_err_t set_vp9e_config(VP9_CONFIG *oxcf,
John Koleszar's avatar
John Koleszar committed
244
                                       vpx_codec_enc_cfg_t cfg,
245
                                       struct vp9_extracfg vp8_cfg) {
246 247 248
  oxcf->version = cfg.g_profile | (vp8_cfg.experimental ? 0x4 : 0);
  oxcf->width   = cfg.g_w;
  oxcf->height  = cfg.g_h;
John Koleszar's avatar
John Koleszar committed
249
  /* guess a frame rate if out of whack, use 30 */
250 251
  oxcf->framerate = (double)(cfg.g_timebase.den)
                    / (double)(cfg.g_timebase.num);
John Koleszar's avatar
John Koleszar committed
252

James Zern's avatar
James Zern committed
253 254
  if (oxcf->framerate > 180) {
    oxcf->framerate = 30;
John Koleszar's avatar
John Koleszar committed
255
  }
John Koleszar's avatar
John Koleszar committed
256

John Koleszar's avatar
John Koleszar committed
257
  switch (cfg.g_pass) {
John Koleszar's avatar
John Koleszar committed
258
    case VPX_RC_ONE_PASS:
Marco Paniconi's avatar
Marco Paniconi committed
259
      oxcf->Mode = MODE_GOODQUALITY;
John Koleszar's avatar
John Koleszar committed
260
      break;
John Koleszar's avatar
John Koleszar committed
261
    case VPX_RC_FIRST_PASS:
John Koleszar's avatar
John Koleszar committed
262 263
      oxcf->Mode = MODE_FIRSTPASS;
      break;
John Koleszar's avatar
John Koleszar committed
264
    case VPX_RC_LAST_PASS:
John Koleszar's avatar
John Koleszar committed
265 266 267 268 269
      oxcf->Mode = MODE_SECONDPASS_BEST;
      break;
  }

  if (cfg.g_pass == VPX_RC_FIRST_PASS) {
Jingning Han's avatar
Jingning Han committed
270 271
    oxcf->allow_lag     = 0;
    oxcf->lag_in_frames = 0;
John Koleszar's avatar
John Koleszar committed
272
  } else {
Jingning Han's avatar
Jingning Han committed
273 274
    oxcf->allow_lag     = (cfg.g_lag_in_frames) > 0;
    oxcf->lag_in_frames = cfg.g_lag_in_frames;
John Koleszar's avatar
John Koleszar committed
275 276 277 278 279
  }

  // VBR only supported for now.
  // CBR code has been deprectated for experimental phase.
  // CQ mode not yet tested
280 281 282
  oxcf->end_usage        = USAGE_LOCAL_FILE_PLAYBACK;
  if (cfg.rc_end_usage == VPX_CQ)
    oxcf->end_usage      = USAGE_CONSTRAINED_QUALITY;
283
  else if (cfg.rc_end_usage == VPX_Q)
284
    oxcf->end_usage      = USAGE_CONSTANT_QUALITY;
Marco Paniconi's avatar
Marco Paniconi committed
285 286
  else if (cfg.rc_end_usage == VPX_CBR)
    oxcf->end_usage = USAGE_STREAM_FROM_SERVER;
287

Jingning Han's avatar
Jingning Han committed
288
  oxcf->target_bandwidth         = cfg.rc_target_bitrate;
John Koleszar's avatar
John Koleszar committed
289 290 291 292 293 294 295 296 297 298 299 300 301 302
  oxcf->rc_max_intra_bitrate_pct = vp8_cfg.rc_max_intra_bitrate_pct;

  oxcf->best_allowed_q          = cfg.rc_min_quantizer;
  oxcf->worst_allowed_q         = cfg.rc_max_quantizer;
  oxcf->cq_level                = vp8_cfg.cq_level;
  oxcf->fixed_q = -1;

  oxcf->under_shoot_pct         = cfg.rc_undershoot_pct;
  oxcf->over_shoot_pct          = cfg.rc_overshoot_pct;

  oxcf->maximum_buffer_size     = cfg.rc_buf_sz;
  oxcf->starting_buffer_level   = cfg.rc_buf_initial_sz;
  oxcf->optimal_buffer_level    = cfg.rc_buf_optimal_sz;

Jingning Han's avatar
Jingning Han committed
303
  oxcf->two_pass_vbrbias         = cfg.rc_2pass_vbr_bias_pct;
John Koleszar's avatar
John Koleszar committed
304 305 306 307 308 309 310 311 312 313 314 315 316 317 318
  oxcf->two_pass_vbrmin_section  = cfg.rc_2pass_vbr_minsection_pct;
  oxcf->two_pass_vbrmax_section  = cfg.rc_2pass_vbr_maxsection_pct;

  oxcf->auto_key               = cfg.kf_mode == VPX_KF_AUTO
                                 && cfg.kf_min_dist != cfg.kf_max_dist;
  // oxcf->kf_min_dist         = cfg.kf_min_dis;
  oxcf->key_freq               = cfg.kf_max_dist;

  // oxcf->delete_first_pass_file = cfg.g_delete_firstpassfile;
  // strcpy(oxcf->first_pass_file, cfg.g_firstpass_file);

  oxcf->cpu_used               =  vp8_cfg.cpu_used;
  oxcf->encode_breakout        =  vp8_cfg.static_thresh;
  oxcf->play_alternate         =  vp8_cfg.enable_auto_alt_ref;
  oxcf->noise_sensitivity      =  vp8_cfg.noise_sensitivity;
Jingning Han's avatar
Jingning Han committed
319
  oxcf->Sharpness              =  vp8_cfg.Sharpness;
John Koleszar's avatar
John Koleszar committed
320

Jingning Han's avatar
Jingning Han committed
321 322
  oxcf->two_pass_stats_in      =  cfg.rc_twopass_stats_in;
  oxcf->output_pkt_list        =  vp8_cfg.pkt_list;
John Koleszar's avatar
John Koleszar committed
323 324

  oxcf->arnr_max_frames = vp8_cfg.arnr_max_frames;
Jingning Han's avatar
Jingning Han committed
325 326
  oxcf->arnr_strength   = vp8_cfg.arnr_strength;
  oxcf->arnr_type       = vp8_cfg.arnr_type;
John Koleszar's avatar
John Koleszar committed
327 328

  oxcf->tuning = vp8_cfg.tuning;
John Koleszar's avatar
John Koleszar committed
329

330
  oxcf->tile_columns = vp8_cfg.tile_columns;
Jingning Han's avatar
Jingning Han committed
331
  oxcf->tile_rows    = vp8_cfg.tile_rows;
332

John Koleszar's avatar
John Koleszar committed
333
  oxcf->lossless = vp8_cfg.lossless;
Hui Su's avatar
Hui Su committed
334

Jingning Han's avatar
Jingning Han committed
335
  oxcf->error_resilient_mode         = cfg.g_error_resilient;
336
  oxcf->frame_parallel_decoding_mode = vp8_cfg.frame_parallel_decoding_mode;
337 338

  oxcf->ss_number_layers = cfg.ss_number_layers;
John Koleszar's avatar
John Koleszar committed
339
  /*
340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366
  printf("Current VP9 Settings: \n");
  printf("target_bandwidth: %d\n", oxcf->target_bandwidth);
  printf("noise_sensitivity: %d\n", oxcf->noise_sensitivity);
  printf("Sharpness: %d\n",    oxcf->Sharpness);
  printf("cpu_used: %d\n",  oxcf->cpu_used);
  printf("Mode: %d\n",     oxcf->Mode);
  // printf("delete_first_pass_file: %d\n",  oxcf->delete_first_pass_file);
  printf("auto_key: %d\n",  oxcf->auto_key);
  printf("key_freq: %d\n", oxcf->key_freq);
  printf("end_usage: %d\n", oxcf->end_usage);
  printf("under_shoot_pct: %d\n", oxcf->under_shoot_pct);
  printf("over_shoot_pct: %d\n", oxcf->over_shoot_pct);
  printf("starting_buffer_level: %d\n", oxcf->starting_buffer_level);
  printf("optimal_buffer_level: %d\n",  oxcf->optimal_buffer_level);
  printf("maximum_buffer_size: %d\n", oxcf->maximum_buffer_size);
  printf("fixed_q: %d\n",  oxcf->fixed_q);
  printf("worst_allowed_q: %d\n", oxcf->worst_allowed_q);
  printf("best_allowed_q: %d\n", oxcf->best_allowed_q);
  printf("two_pass_vbrbias: %d\n",  oxcf->two_pass_vbrbias);
  printf("two_pass_vbrmin_section: %d\n", oxcf->two_pass_vbrmin_section);
  printf("two_pass_vbrmax_section: %d\n", oxcf->two_pass_vbrmax_section);
  printf("allow_lag: %d\n", oxcf->allow_lag);
  printf("lag_in_frames: %d\n", oxcf->lag_in_frames);
  printf("play_alternate: %d\n", oxcf->play_alternate);
  printf("Version: %d\n", oxcf->Version);
  printf("encode_breakout: %d\n", oxcf->encode_breakout);
  printf("error resilient: %d\n", oxcf->error_resilient_mode);
367 368
  printf("frame parallel detokenization: %d\n",
         oxcf->frame_parallel_decoding_mode);
John Koleszar's avatar
John Koleszar committed
369 370
  */
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
371 372
}

373
static vpx_codec_err_t vp9e_set_config(vpx_codec_alg_priv_t       *ctx,
John Koleszar's avatar
John Koleszar committed
374 375
                                       const vpx_codec_enc_cfg_t  *cfg) {
  vpx_codec_err_t res;
John Koleszar's avatar
John Koleszar committed
376

John Koleszar's avatar
John Koleszar committed
377 378
  if ((cfg->g_w != ctx->cfg.g_w) || (cfg->g_h != ctx->cfg.g_h))
    ERROR("Cannot change width or height after initialization");
John Koleszar's avatar
John Koleszar committed
379

John Koleszar's avatar
John Koleszar committed
380 381 382 383 384 385 386
  /* Prevent increasing lag_in_frames. This check is stricter than it needs
   * to be -- the limit is not increasing past the first lag_in_frames
   * value, but we don't track the initial config, only the last successful
   * config.
   */
  if ((cfg->g_lag_in_frames > ctx->cfg.g_lag_in_frames))
    ERROR("Cannot increase lag_in_frames");
John Koleszar's avatar
John Koleszar committed
387

John Koleszar's avatar
John Koleszar committed
388
  res = validate_config(ctx, cfg, &ctx->vp8_cfg);
John Koleszar's avatar
John Koleszar committed
389

John Koleszar's avatar
John Koleszar committed
390 391
  if (!res) {
    ctx->cfg = *cfg;
392
    set_vp9e_config(&ctx->oxcf, ctx->cfg, ctx->vp8_cfg);
393
    vp9_change_config(ctx->cpi, &ctx->oxcf);
John Koleszar's avatar
John Koleszar committed
394
  }
John Koleszar's avatar
John Koleszar committed
395

John Koleszar's avatar
John Koleszar committed
396
  return res;
John Koleszar's avatar
John Koleszar committed
397 398 399
}


400
int vp9_reverse_trans(int q);
John Koleszar's avatar
John Koleszar committed
401 402 403 404


static vpx_codec_err_t get_param(vpx_codec_alg_priv_t *ctx,
                                 int                   ctrl_id,
John Koleszar's avatar
John Koleszar committed
405 406
                                 va_list               args) {
  void *arg = va_arg(args, void *);
John Koleszar's avatar
John Koleszar committed
407 408 409

#define MAP(id, var) case id: *(RECAST(id, arg)) = var; break

John Koleszar's avatar
John Koleszar committed
410 411
  if (!arg)
    return VPX_CODEC_INVALID_PARAM;
John Koleszar's avatar
John Koleszar committed
412

John Koleszar's avatar
John Koleszar committed
413
  switch (ctrl_id) {
414
      MAP(VP8E_GET_LAST_QUANTIZER, vp9_get_quantizer(ctx->cpi));
415 416
      MAP(VP8E_GET_LAST_QUANTIZER_64,
          vp9_reverse_trans(vp9_get_quantizer(ctx->cpi)));
John Koleszar's avatar
John Koleszar committed
417
  }
John Koleszar's avatar
John Koleszar committed
418

John Koleszar's avatar
John Koleszar committed
419
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
420 421 422 423 424 425
#undef MAP
}


static vpx_codec_err_t set_param(vpx_codec_alg_priv_t *ctx,
                                 int                   ctrl_id,
John Koleszar's avatar
John Koleszar committed
426 427
                                 va_list               args) {
  vpx_codec_err_t     res  = VPX_CODEC_OK;
428
  struct vp9_extracfg xcfg = ctx->vp8_cfg;
John Koleszar's avatar
John Koleszar committed
429 430 431

#define MAP(id, var) case id: var = CAST(id, args); break;

John Koleszar's avatar
John Koleszar committed
432
  switch (ctrl_id) {
433 434 435 436 437 438 439 440 441 442 443 444 445 446
      MAP(VP8E_SET_CPUUSED,                 xcfg.cpu_used);
      MAP(VP8E_SET_ENABLEAUTOALTREF,        xcfg.enable_auto_alt_ref);
      MAP(VP8E_SET_NOISE_SENSITIVITY,       xcfg.noise_sensitivity);
      MAP(VP8E_SET_SHARPNESS,               xcfg.Sharpness);
      MAP(VP8E_SET_STATIC_THRESHOLD,        xcfg.static_thresh);
      MAP(VP9E_SET_TILE_COLUMNS,            xcfg.tile_columns);
      MAP(VP9E_SET_TILE_ROWS,               xcfg.tile_rows);
      MAP(VP8E_SET_ARNR_MAXFRAMES,          xcfg.arnr_max_frames);
      MAP(VP8E_SET_ARNR_STRENGTH,           xcfg.arnr_strength);
      MAP(VP8E_SET_ARNR_TYPE,               xcfg.arnr_type);
      MAP(VP8E_SET_TUNING,                  xcfg.tuning);
      MAP(VP8E_SET_CQ_LEVEL,                xcfg.cq_level);
      MAP(VP8E_SET_MAX_INTRA_BITRATE_PCT,   xcfg.rc_max_intra_bitrate_pct);
      MAP(VP9E_SET_LOSSLESS,                xcfg.lossless);
447
      MAP(VP9E_SET_FRAME_PARALLEL_DECODING, xcfg.frame_parallel_decoding_mode);
John Koleszar's avatar
John Koleszar committed
448 449 450 451 452 453
  }

  res = validate_config(ctx, &ctx->cfg, &xcfg);

  if (!res) {
    ctx->vp8_cfg = xcfg;
454
    set_vp9e_config(&ctx->oxcf, ctx->cfg, ctx->vp8_cfg);
455
    vp9_change_config(ctx->cpi, &ctx->oxcf);
John Koleszar's avatar
John Koleszar committed
456 457 458
  }

  return res;
John Koleszar's avatar
John Koleszar committed
459 460
#undef MAP
}
461 462


463
static vpx_codec_err_t vp9e_common_init(vpx_codec_ctx_t *ctx,
John Koleszar's avatar
John Koleszar committed
464
                                        int              experimental) {
John Koleszar's avatar
John Koleszar committed
465
  vpx_codec_err_t            res = VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
466 467 468
  struct vpx_codec_alg_priv *priv;
  vpx_codec_enc_cfg_t       *cfg;
  unsigned int               i;
John Koleszar's avatar
John Koleszar committed
469

470
  VP9_PTR optr;
John Koleszar's avatar
John Koleszar committed
471

John Koleszar's avatar
John Koleszar committed
472 473
  if (!ctx->priv) {
    priv = calloc(1, sizeof(struct vpx_codec_alg_priv));
John Koleszar's avatar
John Koleszar committed
474

John Koleszar's avatar
John Koleszar committed
475 476 477
    if (!priv) {
      return VPX_CODEC_MEM_ERROR;
    }
478

John Koleszar's avatar
John Koleszar committed
479 480 481 482 483
    ctx->priv = &priv->base;
    ctx->priv->sz = sizeof(*ctx->priv);
    ctx->priv->iface = ctx->iface;
    ctx->priv->alg_priv = priv;
    ctx->priv->init_flags = ctx->init_flags;
John Koleszar's avatar
John Koleszar committed
484
    ctx->priv->enc.total_encoders = 1;
John Koleszar's avatar
John Koleszar committed
485 486 487 488 489 490 491 492

    if (ctx->config.enc) {
      /* Update the reference to the config structure to an
       * internal copy.
       */
      ctx->priv->alg_priv->cfg = *ctx->config.enc;
      ctx->config.enc = &ctx->priv->alg_priv->cfg;
    }
493

John Koleszar's avatar
John Koleszar committed
494
    cfg =  &ctx->priv->alg_priv->cfg;
John Koleszar's avatar
John Koleszar committed
495

John Koleszar's avatar
John Koleszar committed
496 497 498 499 500 501
    /* Select the extra vp6 configuration table based on the current
     * usage value. If the current usage value isn't found, use the
     * values for usage case 0.
     */
    for (i = 0;
         extracfg_map[i].usage && extracfg_map[i].usage != cfg->g_usage;
502
         i++) {}
John Koleszar's avatar
John Koleszar committed
503

John Koleszar's avatar
John Koleszar committed
504 505 506
    priv->vp8_cfg = extracfg_map[i].cfg;
    priv->vp8_cfg.pkt_list = &priv->pkt_list.head;
    priv->vp8_cfg.experimental = experimental;
John Koleszar's avatar
John Koleszar committed
507

508
    // TODO(agrange) Check the limits set on this buffer, or the check that is
509
    // applied in vp9e_encode.
510 511
    priv->cx_data_sz = priv->cfg.g_w * priv->cfg.g_h * 3 / 2 * 8;
//    priv->cx_data_sz = priv->cfg.g_w * priv->cfg.g_h * 3 / 2 * 2;
John Koleszar's avatar
John Koleszar committed
512

John Koleszar's avatar
John Koleszar committed
513
    if (priv->cx_data_sz < 4096) priv->cx_data_sz = 4096;
John Koleszar's avatar
John Koleszar committed
514

John Koleszar's avatar
John Koleszar committed
515
    priv->cx_data = malloc(priv->cx_data_sz);
John Koleszar's avatar
John Koleszar committed
516

John Koleszar's avatar
John Koleszar committed
517 518 519
    if (!priv->cx_data) {
      return VPX_CODEC_MEM_ERROR;
    }
John Koleszar's avatar
John Koleszar committed
520

521
    vp9_initialize_enc();
John Koleszar's avatar
John Koleszar committed
522

John Koleszar's avatar
John Koleszar committed
523
    res = validate_config(priv, &priv->cfg, &priv->vp8_cfg);
John Koleszar's avatar
John Koleszar committed
524

John Koleszar's avatar
John Koleszar committed
525
    if (!res) {
526
      set_vp9e_config(&ctx->priv->alg_priv->oxcf,
John Koleszar's avatar
John Koleszar committed
527 528
                      ctx->priv->alg_priv->cfg,
                      ctx->priv->alg_priv->vp8_cfg);
529
      optr = vp9_create_compressor(&ctx->priv->alg_priv->oxcf);
530

John Koleszar's avatar
John Koleszar committed
531 532 533 534
      if (!optr)
        res = VPX_CODEC_MEM_ERROR;
      else
        ctx->priv->alg_priv->cpi = optr;
John Koleszar's avatar
John Koleszar committed
535
    }
John Koleszar's avatar
John Koleszar committed
536
  }
John Koleszar's avatar
John Koleszar committed
537

John Koleszar's avatar
John Koleszar committed
538
  return res;
John Koleszar's avatar
John Koleszar committed
539 540
}

541

542
static vpx_codec_err_t vp9e_init(vpx_codec_ctx_t *ctx,
John Koleszar's avatar
John Koleszar committed
543
                                 vpx_codec_priv_enc_mr_cfg_t *data) {
544
  return vp9e_common_init(ctx, 0);
545 546 547 548
}


#if CONFIG_EXPERIMENTAL
549
static vpx_codec_err_t vp9e_exp_init(vpx_codec_ctx_t *ctx,
John Koleszar's avatar
John Koleszar committed
550
                                     vpx_codec_priv_enc_mr_cfg_t *data) {
551
  return vp9e_common_init(ctx, 1);
552 553 554 555
}
#endif


556
static vpx_codec_err_t vp9e_destroy(vpx_codec_alg_priv_t *ctx) {
John Koleszar's avatar
John Koleszar committed
557
  free(ctx->cx_data);
558
  vp9_remove_compressor(&ctx->cpi);
John Koleszar's avatar
John Koleszar committed
559 560
  free(ctx);
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
561 562 563 564
}

static void pick_quickcompress_mode(vpx_codec_alg_priv_t  *ctx,
                                    unsigned long          duration,
John Koleszar's avatar
John Koleszar committed
565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582
                                    unsigned long          deadline) {
  unsigned int new_qc;

  /* Use best quality mode if no deadline is given. */
  if (deadline)
    new_qc = MODE_GOODQUALITY;
  else
    new_qc = MODE_BESTQUALITY;

  if (ctx->cfg.g_pass == VPX_RC_FIRST_PASS)
    new_qc = MODE_FIRSTPASS;
  else if (ctx->cfg.g_pass == VPX_RC_LAST_PASS)
    new_qc = (new_qc == MODE_BESTQUALITY)
             ? MODE_SECONDPASS_BEST
             : MODE_SECONDPASS;

  if (ctx->oxcf.Mode != new_qc) {
    ctx->oxcf.Mode = new_qc;
583
    vp9_change_config(ctx->cpi, &ctx->oxcf);
John Koleszar's avatar
John Koleszar committed
584
  }
John Koleszar's avatar
John Koleszar committed
585 586 587
}


588
static int write_superframe_index(vpx_codec_alg_priv_t *ctx) {
John Koleszar's avatar
John Koleszar committed
589
  uint8_t marker = 0xc0;
Yaowu Xu's avatar
Yaowu Xu committed
590 591
  unsigned int mask;
  int mag, index_sz;
John Koleszar's avatar
John Koleszar committed
592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625

  assert(ctx->pending_frame_count);
  assert(ctx->pending_frame_count <= 8);

  /* Add the number of frames to the marker byte */
  marker |= ctx->pending_frame_count - 1;

  /* Choose the magnitude */
  for (mag = 0, mask = 0xff; mag < 4; mag++) {
    if (ctx->pending_frame_magnitude < mask)
      break;
    mask <<= 8;
    mask |= 0xff;
  }
  marker |= mag << 3;

  /* Write the index */
  index_sz = 2 + (mag + 1) * ctx->pending_frame_count;
  if (ctx->pending_cx_data_sz + index_sz < ctx->cx_data_sz) {
    uint8_t *x = ctx->pending_cx_data + ctx->pending_cx_data_sz;
    int i, j;

    *x++ = marker;
    for (i = 0; i < ctx->pending_frame_count; i++) {
      int this_sz = ctx->pending_frame_sizes[i];

      for (j = 0; j <= mag; j++) {
        *x++ = this_sz & 0xff;
        this_sz >>= 8;
      }
    }
    *x++ = marker;
    ctx->pending_cx_data_sz += index_sz;
  }
626
  return index_sz;
John Koleszar's avatar
John Koleszar committed
627 628
}

629
static vpx_codec_err_t vp9e_encode(vpx_codec_alg_priv_t  *ctx,
John Koleszar's avatar
John Koleszar committed
630 631 632 633
                                   const vpx_image_t     *img,
                                   vpx_codec_pts_t        pts,
                                   unsigned long          duration,
                                   vpx_enc_frame_flags_t  flags,
John Koleszar's avatar
John Koleszar committed
634 635
                                   unsigned long          deadline) {
  vpx_codec_err_t res = VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
636

John Koleszar's avatar
John Koleszar committed
637 638
  if (img)
    res = validate_img(ctx, img);
John Koleszar's avatar
John Koleszar committed
639

John Koleszar's avatar
John Koleszar committed
640 641
  pick_quickcompress_mode(ctx, duration, deadline);
  vpx_codec_pkt_list_init(&ctx->pkt_list);
John Koleszar's avatar
John Koleszar committed
642

John Koleszar's avatar
John Koleszar committed
643 644 645 646 647 648
  /* Handle Flags */
  if (((flags & VP8_EFLAG_NO_UPD_GF) && (flags & VP8_EFLAG_FORCE_GF))
      || ((flags & VP8_EFLAG_NO_UPD_ARF) && (flags & VP8_EFLAG_FORCE_ARF))) {
    ctx->base.err_detail = "Conflicting flags.";
    return VPX_CODEC_INVALID_PARAM;
  }
John Koleszar's avatar
John Koleszar committed
649

John Koleszar's avatar
John Koleszar committed
650 651 652
  if (flags & (VP8_EFLAG_NO_REF_LAST | VP8_EFLAG_NO_REF_GF
               | VP8_EFLAG_NO_REF_ARF)) {
    int ref = 7;
John Koleszar's avatar
John Koleszar committed
653

John Koleszar's avatar
John Koleszar committed
654
    if (flags & VP8_EFLAG_NO_REF_LAST)
655
      ref ^= VP9_LAST_FLAG;
John Koleszar's avatar
John Koleszar committed
656

John Koleszar's avatar
John Koleszar committed
657
    if (flags & VP8_EFLAG_NO_REF_GF)
658
      ref ^= VP9_GOLD_FLAG;
John Koleszar's avatar
John Koleszar committed
659

John Koleszar's avatar
John Koleszar committed
660
    if (flags & VP8_EFLAG_NO_REF_ARF)
661
      ref ^= VP9_ALT_FLAG;
John Koleszar's avatar
John Koleszar committed
662

663
    vp9_use_as_reference(ctx->cpi, ref);
John Koleszar's avatar
John Koleszar committed
664
  }
John Koleszar's avatar
John Koleszar committed
665

John Koleszar's avatar
John Koleszar committed
666 667 668 669
  if (flags & (VP8_EFLAG_NO_UPD_LAST | VP8_EFLAG_NO_UPD_GF
               | VP8_EFLAG_NO_UPD_ARF | VP8_EFLAG_FORCE_GF
               | VP8_EFLAG_FORCE_ARF)) {
    int upd = 7;
John Koleszar's avatar
John Koleszar committed
670

John Koleszar's avatar
John Koleszar committed
671
    if (flags & VP8_EFLAG_NO_UPD_LAST)
672
      upd ^= VP9_LAST_FLAG;
John Koleszar's avatar
John Koleszar committed
673

John Koleszar's avatar
John Koleszar committed
674
    if (flags & VP8_EFLAG_NO_UPD_GF)
675
      upd ^= VP9_GOLD_FLAG;
John Koleszar's avatar
John Koleszar committed
676

John Koleszar's avatar
John Koleszar committed
677
    if (flags & VP8_EFLAG_NO_UPD_ARF)
678
      upd ^= VP9_ALT_FLAG;
John Koleszar's avatar
John Koleszar committed
679

680
    vp9_update_reference(ctx->cpi, upd);
John Koleszar's avatar
John Koleszar committed
681
  }
John Koleszar's avatar
John Koleszar committed
682

John Koleszar's avatar
John Koleszar committed
683
  if (flags & VP8_EFLAG_NO_UPD_ENTROPY) {
684
    vp9_update_entropy(ctx->cpi, 0);
John Koleszar's avatar
John Koleszar committed
685
  }
John Koleszar's avatar
John Koleszar committed
686

John Koleszar's avatar
John Koleszar committed
687 688 689 690 691 692
  /* Handle fixed keyframe intervals */
  if (ctx->cfg.kf_mode == VPX_KF_AUTO
      && ctx->cfg.kf_min_dist == ctx->cfg.kf_max_dist) {
    if (++ctx->fixed_kf_cntr > ctx->cfg.kf_min_dist) {
      flags |= VPX_EFLAG_FORCE_KF;
      ctx->fixed_kf_cntr = 1;
John Koleszar's avatar
John Koleszar committed
693
    }
John Koleszar's avatar
John Koleszar committed
694
  }
John Koleszar's avatar
John Koleszar committed
695

John Koleszar's avatar
John Koleszar committed
696 697 698 699 700 701 702
  /* Initialize the encoder instance on the first frame*/
  if (!res && ctx->cpi) {
    unsigned int lib_flags;
    YV12_BUFFER_CONFIG sd;
    int64_t dst_time_stamp, dst_end_time_stamp;
    unsigned long size, cx_data_sz;
    unsigned char *cx_data;
John Koleszar's avatar
John Koleszar committed
703

John Koleszar's avatar
John Koleszar committed
704 705
    /* Set up internal flags */
    if (ctx->base.init_flags & VPX_CODEC_USE_PSNR)
706
      ((VP9_COMP *)ctx->cpi)->b_calculate_psnr = 1;
John Koleszar's avatar
John Koleszar committed
707

John Koleszar's avatar
John Koleszar committed
708
    // if (ctx->base.init_flags & VPX_CODEC_USE_OUTPUT_PARTITION)
709
    //    ((VP9_COMP *)ctx->cpi)->output_partition = 1;
710

John Koleszar's avatar
John Koleszar committed
711 712
    /* Convert API flags to internal codec lib flags */
    lib_flags = (flags & VPX_EFLAG_FORCE_KF) ? FRAMEFLAGS_KEY : 0;
John Koleszar's avatar
John Koleszar committed
713

John Koleszar's avatar
John Koleszar committed
714
    /* vp8 use 10,000,000 ticks/second as time stamp */
715 716 717 718
    dst_time_stamp = pts * 10000000 * ctx->cfg.g_timebase.num
                     / ctx->cfg.g_timebase.den;
    dst_end_time_stamp = (pts + duration) * 10000000 * ctx->cfg.g_timebase.num /
                         ctx->cfg.g_timebase.den;
John Koleszar's avatar
John Koleszar committed
719

John Koleszar's avatar
John Koleszar committed
720 721
    if (img != NULL) {
      res = image2yuvconfig(img, &sd);
John Koleszar's avatar
John Koleszar committed
722

723
      if (vp9_receive_raw_frame(ctx->cpi, lib_flags,
John Koleszar's avatar
John Koleszar committed
724
                                &sd, dst_time_stamp, dst_end_time_stamp)) {
725
        VP9_COMP *cpi = (VP9_COMP *)ctx->cpi;
John Koleszar's avatar
John Koleszar committed
726 727 728
        res = update_error_state(ctx, &cpi->common.error);
      }
    }
John Koleszar's avatar
John Koleszar committed
729

730 731
    cx_data = ctx->cx_data;
    cx_data_sz = ctx->cx_data_sz;
John Koleszar's avatar
John Koleszar committed
732 733
    lib_flags = 0;

734 735 736 737 738 739 740 741 742 743 744 745 746 747
    /* Any pending invisible frames? */
    if (ctx->pending_cx_data) {
      memmove(cx_data, ctx->pending_cx_data, ctx->pending_cx_data_sz);
      ctx->pending_cx_data = cx_data;
      cx_data += ctx->pending_cx_data_sz;
      cx_data_sz -= ctx->pending_cx_data_sz;

      /* TODO: this is a minimal check, the underlying codec doesn't respect
       * the buffer size anyway.
       */
      if (cx_data_sz < ctx->cx_data_sz / 2) {
        ctx->base.err_detail = "Compressed data buffer too small";
        return VPX_CODEC_ERROR;
      }
748 749
    }

750 751 752 753
    while (cx_data_sz >= ctx->cx_data_sz / 2 &&
           -1 != vp9_get_compressed_data(ctx->cpi, &lib_flags, &size,
                                         cx_data, &dst_time_stamp,
                                         &dst_end_time_stamp, !img)) {
John Koleszar's avatar
John Koleszar committed
754 755 756
      if (size) {
        vpx_codec_pts_t    round, delta;
        vpx_codec_cx_pkt_t pkt;
757
        VP9_COMP *cpi = (VP9_COMP *)ctx->cpi;
John Koleszar's avatar
John Koleszar committed
758

759
        /* Pack invisible frames with the next visible frame */
760
        if (!cpi->common.show_frame) {
761 762 763
          if (!ctx->pending_cx_data)
            ctx->pending_cx_data = cx_data;
          ctx->pending_cx_data_sz += size;
John Koleszar's avatar
John Koleszar committed
764 765
          ctx->pending_frame_sizes[ctx->pending_frame_count++] = size;
          ctx->pending_frame_magnitude |= size;
766 767 768 769 770
          cx_data += size;
          cx_data_sz -= size;
          continue;
        }

John Koleszar's avatar
John Koleszar committed
771
        /* Add the frame packet to the list of returned packets. */
Yaowu Xu's avatar
Yaowu Xu committed
772
        round = (vpx_codec_pts_t)1000000 * ctx->cfg.g_timebase.num / 2 - 1;
John Koleszar's avatar
John Koleszar committed
773 774 775 776 777
        delta = (dst_end_time_stamp - dst_time_stamp);
        pkt.kind = VPX_CODEC_CX_FRAME_PKT;
        pkt.data.frame.pts =
          (dst_time_stamp * ctx->cfg.g_timebase.den + round)
          / ctx->cfg.g_timebase.num / 10000000;
778 779 780
        pkt.data.frame.duration = (unsigned long)
          ((delta * ctx->cfg.g_timebase.den + round)
          / ctx->cfg.g_timebase.num / 10000000);
John Koleszar's avatar
John Koleszar committed
781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796
        pkt.data.frame.flags = lib_flags << 16;

        if (lib_flags & FRAMEFLAGS_KEY)
          pkt.data.frame.flags |= VPX_FRAME_IS_KEY;

        if (!cpi->common.show_frame) {
          pkt.data.frame.flags |= VPX_FRAME_IS_INVISIBLE;

          // This timestamp should be as close as possible to the
          // prior PTS so that if a decoder uses pts to schedule when
          // to do this, we start right after last frame was decoded.
          // Invisible frames have no duration.
          pkt.data.frame.pts = ((cpi->last_time_stamp_seen
                                 * ctx->cfg.g_timebase.den + round)
                                / ctx->cfg.g_timebase.num / 10000000) + 1;
          pkt.data.frame.duration = 0;
John Koleszar's avatar
John Koleszar committed
797 798
        }

John Koleszar's avatar
John Koleszar committed
799 800
        if (cpi->droppable)
          pkt.data.frame.flags |= VPX_FRAME_IS_DROPPABLE;
John Koleszar's avatar
John Koleszar committed
801

John Koleszar's avatar
John Koleszar committed
802
        /*if (cpi->output_partition)
John Koleszar's avatar
John Koleszar committed
803
        {
John Koleszar's avatar
John Koleszar committed
804 805 806 807 808 809
            int i;
            const int num_partitions = 1;

            pkt.data.frame.flags |= VPX_FRAME_IS_FRAGMENT;

            for (i = 0; i < num_partitions; ++i)
John Koleszar's avatar
John Koleszar committed
810
            {
John Koleszar's avatar
John Koleszar committed
811 812 813 814 815 816 817 818 819
                pkt.data.frame.buf = cx_data;
                pkt.data.frame.sz = cpi->partition_sz[i];
                pkt.data.frame.partition_id = i;
                // don't set the fragment bit for the last partition
                if (i == (num_partitions - 1))
                    pkt.data.frame.flags &= ~VPX_FRAME_IS_FRAGMENT;
                vpx_codec_pkt_list_add(&ctx->pkt_list.head, &pkt);
                cx_data += cpi->partition_sz[i];
                cx_data_sz -= cpi->partition_sz[i];
John Koleszar's avatar
John Koleszar committed
820 821
            }
        }
John Koleszar's avatar
John Koleszar committed
822 823
        else*/
        {
824
          if (ctx->pending_cx_data) {
John Koleszar's avatar
John Koleszar committed
825 826 827
            ctx->pending_frame_sizes[ctx->pending_frame_count++] = size;
            ctx->pending_frame_magnitude |= size;
            ctx->pending_cx_data_sz += size;
828
            size += write_superframe_index(ctx);
829
            pkt.data.frame.buf = ctx->pending_cx_data;
John Koleszar's avatar
John Koleszar committed
830
            pkt.data.frame.sz  = ctx->pending_cx_data_sz;
831 832
            ctx->pending_cx_data = NULL;
            ctx->pending_cx_data_sz = 0;
John Koleszar's avatar
John Koleszar committed
833 834
            ctx->pending_frame_count = 0;
            ctx->pending_frame_magnitude = 0;
835 836
          } else {
            pkt.data.frame.buf = cx_data;
837
            pkt.data.frame.sz  = size;
838
          }
John Koleszar's avatar
John Koleszar committed
839 840 841 842 843 844
          pkt.data.frame.partition_id = -1;
          vpx_codec_pkt_list_add(&ctx->pkt_list.head, &pkt);
          cx_data += size;
          cx_data_sz -= size;
        }
      }
John Koleszar's avatar
John Koleszar committed
845
    }
John Koleszar's avatar
John Koleszar committed
846
  }
John Koleszar's avatar
John Koleszar committed
847

John Koleszar's avatar
John Koleszar committed
848
  return res;
John Koleszar's avatar
John Koleszar committed
849 850 851
}


852
static const vpx_codec_cx_pkt_t *vp9e_get_cxdata(vpx_codec_alg_priv_t  *ctx,
John Koleszar's avatar
John Koleszar committed
853 854
                                                 vpx_codec_iter_t      *iter) {
  return vpx_codec_pkt_list_get(&ctx->pkt_list.head, iter);
John Koleszar's avatar
John Koleszar committed
855 856
}

857
static vpx_codec_err_t vp9e_set_reference(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
858 859 860
                                          int ctr_id,
                                          va_list args) {
  vpx_ref_frame_t *data = va_arg(args, vpx_ref_frame_t *);
John Koleszar's avatar
John Koleszar committed
861

John Koleszar's avatar
John Koleszar committed
862 863 864
  if (data) {
    vpx_ref_frame_t *frame = (vpx_ref_frame_t *)data;
    YV12_BUFFER_CONFIG sd;
John Koleszar's avatar
John Koleszar committed
865

John Koleszar's avatar
John Koleszar committed
866
    image2yuvconfig(&frame->img, &sd);
867 868
    vp9_set_reference_enc(ctx->cpi, ref_frame_to_vp9_reframe(frame->frame_type),
                          &sd);
John Koleszar's avatar
John Koleszar committed
869
    return VPX_CODEC_OK;
870
  } else {
John Koleszar's avatar
John Koleszar committed
871
    return VPX_CODEC_INVALID_PARAM;
872
  }
John Koleszar's avatar
John Koleszar committed
873 874
}

875
static vpx_codec_err_t vp9e_copy_reference(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
876 877
                                           int ctr_id,
                                           va_list args) {
John Koleszar's avatar
John Koleszar committed
878
  vpx_ref_frame_t *data = va_arg(args, vpx_ref_frame_t *);
John Koleszar's avatar
John Koleszar committed
879

John Koleszar's avatar
John Koleszar committed
880 881 882
  if (data) {
    vpx_ref_frame_t *frame = (vpx_ref_frame_t *)data;
    YV12_BUFFER_CONFIG sd;
John Koleszar's avatar
John Koleszar committed
883

John Koleszar's avatar
John Koleszar committed
884
    image2yuvconfig(&frame->img, &sd);
885 886
    vp9_copy_reference_enc(ctx->cpi,
                           ref_frame_to_vp9_reframe(frame->frame_type), &sd);
John Koleszar's avatar
John Koleszar committed
887
    return VPX_CODEC_OK;
888
  } else {
John Koleszar's avatar
John Koleszar committed
889
    return VPX_CODEC_INVALID_PARAM;