vp9_cx_iface.c 39.9 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 13
#include "vpx/vpx_codec.h"
#include "vpx/internal/vpx_codec_internal.h"
John Koleszar's avatar
John Koleszar committed
14
#include "vpx_version.h"
15
#include "vp9/encoder/vp9_onyx_int.h"
John Koleszar's avatar
John Koleszar committed
16
#include "vpx/vp8cx.h"
17 18
#include "vp9/encoder/vp9_firstpass.h"
#include "vp9/common/vp9_onyx.h"
John Koleszar's avatar
John Koleszar committed
19
#include "vp9/vp9_iface_common.h"
John Koleszar's avatar
John Koleszar committed
20 21 22
#include <stdlib.h>
#include <string.h>

John Koleszar's avatar
John Koleszar committed
23 24 25 26 27 28 29
struct vp8_extracfg {
  struct vpx_codec_pkt_list *pkt_list;
  int                         cpu_used;                    /** available cpu percentage in 1/16*/
  unsigned int                enable_auto_alt_ref;           /** if encoder decides to uses alternate reference frame */
  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;
John Koleszar's avatar
John Koleszar committed
32 33 34 35 36 37 38
  unsigned int                arnr_max_frames;    /* alt_ref Noise Reduction Max Frame Count */
  unsigned int                arnr_strength;    /* alt_ref Noise Reduction Strength */
  unsigned int                arnr_type;        /* alt_ref filter type */
  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;
John Koleszar's avatar
John Koleszar committed
40 41
};

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

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

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


static vpx_codec_err_t
update_error_state(vpx_codec_alg_priv_t                 *ctx,
John Koleszar's avatar
John Koleszar committed
93 94
                   const struct vpx_internal_error_info *error) {
  vpx_codec_err_t res;
John Koleszar's avatar
John Koleszar committed
95

John Koleszar's avatar
John Koleszar committed
96 97 98 99
  if ((res = error->error_code))
    ctx->base.err_detail = error->has_detail
                           ? error->detail
                           : NULL;
John Koleszar's avatar
John Koleszar committed
100

John Koleszar's avatar
John Koleszar committed
101
  return res;
John Koleszar's avatar
John Koleszar committed
102 103 104
}


105
#undef ERROR
John Koleszar's avatar
John Koleszar committed
106
#define ERROR(str) do {\
John Koleszar's avatar
John Koleszar committed
107 108 109
    ctx->base.err_detail = str;\
    return VPX_CODEC_INVALID_PARAM;\
  } while(0)
John Koleszar's avatar
John Koleszar committed
110 111

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

116
#define RANGE_CHECK_HI(p,memb,hi) do {\
John Koleszar's avatar
John Koleszar committed
117 118 119
    if(!((p)->memb <= (hi))) \
      ERROR(#memb " out of range [.."#hi"]");\
  } while(0)
120

John Koleszar's avatar
John Koleszar committed
121
#define RANGE_CHECK_LO(p,memb,lo) do {\
John Koleszar's avatar
John Koleszar committed
122 123 124
    if(!((p)->memb >= (lo))) \
      ERROR(#memb " out of range ["#lo"..]");\
  } while(0)
John Koleszar's avatar
John Koleszar committed
125 126

#define RANGE_CHECK_BOOL(p,memb) do {\
John Koleszar's avatar
John Koleszar committed
127 128
    if(!!((p)->memb) != (p)->memb) ERROR(#memb " expected boolean");\
  } while(0)
John Koleszar's avatar
John Koleszar committed
129 130 131

static vpx_codec_err_t validate_config(vpx_codec_alg_priv_t      *ctx,
                                       const vpx_codec_enc_cfg_t *cfg,
John Koleszar's avatar
John Koleszar committed
132
                                       const struct vp8_extracfg *vp8_cfg) {
Jingning Han's avatar
Jingning Han committed
133 134
  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
135 136 137
  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
138

John Koleszar's avatar
John Koleszar committed
139 140
  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
141 142 143 144 145 146
  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
147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173
  RANGE_CHECK_HI(cfg, g_threads,          64);
  RANGE_CHECK_HI(cfg, g_lag_in_frames,    MAX_LAG_BUFFERS);
  RANGE_CHECK(cfg, rc_end_usage,          VPX_VBR, VPX_CQ);
  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);

  /* 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);

174
  RANGE_CHECK(vp8_cfg, tile_columns, 0, 6);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
175
  RANGE_CHECK(vp8_cfg, tile_rows, 0, 2);
John Koleszar's avatar
John Koleszar committed
176 177 178 179 180 181 182 183
  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);
184
    int              n_packets = (int)(cfg->rc_twopass_stats_in.sz / packet_sz);
John Koleszar's avatar
John Koleszar committed
185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203
    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
204 205 206 207
}


static vpx_codec_err_t validate_img(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
208 209
                                    const vpx_image_t    *img) {
  switch (img->fmt) {
James Zern's avatar
James Zern committed
210 211 212 213
    case VPX_IMG_FMT_YV12:
    case VPX_IMG_FMT_I420:
    case VPX_IMG_FMT_VPXI420:
    case VPX_IMG_FMT_VPXYV12:
John Koleszar's avatar
John Koleszar committed
214
      break;
John Koleszar's avatar
John Koleszar committed
215
    default:
John Koleszar's avatar
John Koleszar committed
216 217
      ERROR("Invalid image format. Only YV12 and I420 images are supported");
  }
John Koleszar's avatar
John Koleszar committed
218

John Koleszar's avatar
John Koleszar committed
219 220
  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
221

John Koleszar's avatar
John Koleszar committed
222
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
223 224 225
}


226
static vpx_codec_err_t set_vp8e_config(VP9_CONFIG *oxcf,
John Koleszar's avatar
John Koleszar committed
227
                                       vpx_codec_enc_cfg_t cfg,
John Koleszar's avatar
John Koleszar committed
228
                                       struct vp8_extracfg vp8_cfg) {
229 230 231
  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
232 233
  /* guess a frame rate if out of whack, use 30 */
  oxcf->frame_rate             = (double)(cfg.g_timebase.den) / (double)(cfg.g_timebase.num);
John Koleszar's avatar
John Koleszar committed
234

John Koleszar's avatar
John Koleszar committed
235 236 237
  if (oxcf->frame_rate > 180) {
    oxcf->frame_rate = 30;
  }
John Koleszar's avatar
John Koleszar committed
238

John Koleszar's avatar
John Koleszar committed
239
  switch (cfg.g_pass) {
John Koleszar's avatar
John Koleszar committed
240
    case VPX_RC_ONE_PASS:
John Koleszar's avatar
John Koleszar committed
241 242
      oxcf->Mode = MODE_BESTQUALITY;
      break;
John Koleszar's avatar
John Koleszar committed
243
    case VPX_RC_FIRST_PASS:
John Koleszar's avatar
John Koleszar committed
244 245
      oxcf->Mode = MODE_FIRSTPASS;
      break;
John Koleszar's avatar
John Koleszar committed
246
    case VPX_RC_LAST_PASS:
John Koleszar's avatar
John Koleszar committed
247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308
      oxcf->Mode = MODE_SECONDPASS_BEST;
      break;
  }

  if (cfg.g_pass == VPX_RC_FIRST_PASS) {
    oxcf->allow_lag              = 0;
    oxcf->lag_in_frames           = 0;
  } else {
    oxcf->allow_lag              = (cfg.g_lag_in_frames) > 0;
    oxcf->lag_in_frames           = cfg.g_lag_in_frames;
  }

  // VBR only supported for now.
  // CBR code has been deprectated for experimental phase.
  // CQ mode not yet tested
  oxcf->end_usage          = USAGE_LOCAL_FILE_PLAYBACK;
  /*if (cfg.rc_end_usage == VPX_CQ)
      oxcf->end_usage      = USAGE_CONSTRAINED_QUALITY;
  else
      oxcf->end_usage      = USAGE_LOCAL_FILE_PLAYBACK;*/

  oxcf->target_bandwidth       = cfg.rc_target_bitrate;
  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;

  oxcf->two_pass_vbrbias        = cfg.rc_2pass_vbr_bias_pct;
  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;
  oxcf->Sharpness             =  vp8_cfg.Sharpness;

  oxcf->two_pass_stats_in        =  cfg.rc_twopass_stats_in;
  oxcf->output_pkt_list         =  vp8_cfg.pkt_list;

  oxcf->arnr_max_frames = vp8_cfg.arnr_max_frames;
  oxcf->arnr_strength =  vp8_cfg.arnr_strength;
  oxcf->arnr_type =      vp8_cfg.arnr_type;

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

310
  oxcf->tile_columns = vp8_cfg.tile_columns;
Ronald S. Bultje's avatar
Ronald S. Bultje committed
311
  oxcf->tile_rows = vp8_cfg.tile_rows;
312

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

315
  oxcf->error_resilient_mode = cfg.g_error_resilient;
316
  oxcf->frame_parallel_decoding_mode = cfg.g_frame_parallel_decoding;
John Koleszar's avatar
John Koleszar committed
317
  /*
318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344
  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);
345 346
  printf("frame parallel detokenization: %d\n",
         oxcf->frame_parallel_decoding_mode);
John Koleszar's avatar
John Koleszar committed
347 348
  */
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
349 350 351
}

static vpx_codec_err_t vp8e_set_config(vpx_codec_alg_priv_t       *ctx,
John Koleszar's avatar
John Koleszar committed
352 353
                                       const vpx_codec_enc_cfg_t  *cfg) {
  vpx_codec_err_t res;
John Koleszar's avatar
John Koleszar committed
354

John Koleszar's avatar
John Koleszar committed
355 356
  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
357

John Koleszar's avatar
John Koleszar committed
358 359 360 361 362 363 364
  /* 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
365

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

John Koleszar's avatar
John Koleszar committed
368 369 370
  if (!res) {
    ctx->cfg = *cfg;
    set_vp8e_config(&ctx->oxcf, ctx->cfg, ctx->vp8_cfg);
371
    vp9_change_config(ctx->cpi, &ctx->oxcf);
John Koleszar's avatar
John Koleszar committed
372
  }
John Koleszar's avatar
John Koleszar committed
373

John Koleszar's avatar
John Koleszar committed
374
  return res;
John Koleszar's avatar
John Koleszar committed
375 376 377
}


378
int vp9_reverse_trans(int q);
John Koleszar's avatar
John Koleszar committed
379 380 381 382


static vpx_codec_err_t get_param(vpx_codec_alg_priv_t *ctx,
                                 int                   ctrl_id,
John Koleszar's avatar
John Koleszar committed
383 384
                                 va_list               args) {
  void *arg = va_arg(args, void *);
John Koleszar's avatar
John Koleszar committed
385 386 387

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

John Koleszar's avatar
John Koleszar committed
388 389
  if (!arg)
    return VPX_CODEC_INVALID_PARAM;
John Koleszar's avatar
John Koleszar committed
390

John Koleszar's avatar
John Koleszar committed
391
  switch (ctrl_id) {
392
      MAP(VP8E_GET_LAST_QUANTIZER, vp9_get_quantizer(ctx->cpi));
393 394
      MAP(VP8E_GET_LAST_QUANTIZER_64,
          vp9_reverse_trans(vp9_get_quantizer(ctx->cpi)));
John Koleszar's avatar
John Koleszar committed
395
  }
John Koleszar's avatar
John Koleszar committed
396

John Koleszar's avatar
John Koleszar committed
397
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
398 399 400 401 402 403
#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
404 405 406
                                 va_list               args) {
  vpx_codec_err_t     res  = VPX_CODEC_OK;
  struct vp8_extracfg xcfg = ctx->vp8_cfg;
John Koleszar's avatar
John Koleszar committed
407 408 409

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

John Koleszar's avatar
John Koleszar committed
410 411 412 413 414 415
  switch (ctrl_id) {
      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);
416
      MAP(VP9E_SET_TILE_COLUMNS,          xcfg.tile_columns);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
417
      MAP(VP9E_SET_TILE_ROWS,             xcfg.tile_rows);
John Koleszar's avatar
John Koleszar committed
418 419 420 421 422 423 424

      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);
John Koleszar's avatar
John Koleszar committed
425
      MAP(VP9E_SET_LOSSLESS,              xcfg.lossless);
John Koleszar's avatar
John Koleszar committed
426 427 428 429 430 431 432
  }

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

  if (!res) {
    ctx->vp8_cfg = xcfg;
    set_vp8e_config(&ctx->oxcf, ctx->cfg, ctx->vp8_cfg);
433
    vp9_change_config(ctx->cpi, &ctx->oxcf);
John Koleszar's avatar
John Koleszar committed
434 435 436
  }

  return res;
John Koleszar's avatar
John Koleszar committed
437 438
#undef MAP
}
439 440 441


static vpx_codec_err_t vp8e_common_init(vpx_codec_ctx_t *ctx,
John Koleszar's avatar
John Koleszar committed
442
                                        int              experimental) {
John Koleszar's avatar
John Koleszar committed
443
  vpx_codec_err_t            res = VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
444 445 446
  struct vpx_codec_alg_priv *priv;
  vpx_codec_enc_cfg_t       *cfg;
  unsigned int               i;
John Koleszar's avatar
John Koleszar committed
447

448
  VP9_PTR optr;
John Koleszar's avatar
John Koleszar committed
449

John Koleszar's avatar
John Koleszar committed
450 451
  if (!ctx->priv) {
    priv = calloc(1, sizeof(struct vpx_codec_alg_priv));
John Koleszar's avatar
John Koleszar committed
452

John Koleszar's avatar
John Koleszar committed
453 454 455
    if (!priv) {
      return VPX_CODEC_MEM_ERROR;
    }
456

John Koleszar's avatar
John Koleszar committed
457 458 459 460 461
    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
462
    ctx->priv->enc.total_encoders = 1;
John Koleszar's avatar
John Koleszar committed
463 464 465 466 467 468 469 470

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

John Koleszar's avatar
John Koleszar committed
472
    cfg =  &ctx->priv->alg_priv->cfg;
John Koleszar's avatar
John Koleszar committed
473

John Koleszar's avatar
John Koleszar committed
474 475 476 477 478 479 480
    /* 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;
         i++);
John Koleszar's avatar
John Koleszar committed
481

John Koleszar's avatar
John Koleszar committed
482 483 484
    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
485

John Koleszar's avatar
John Koleszar committed
486
    priv->cx_data_sz = priv->cfg.g_w * priv->cfg.g_h * 3 / 2 * 2;
John Koleszar's avatar
John Koleszar committed
487

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

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

John Koleszar's avatar
John Koleszar committed
492 493 494
    if (!priv->cx_data) {
      return VPX_CODEC_MEM_ERROR;
    }
John Koleszar's avatar
John Koleszar committed
495

496
    vp9_initialize_enc();
John Koleszar's avatar
John Koleszar committed
497

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

John Koleszar's avatar
John Koleszar committed
500 501 502 503
    if (!res) {
      set_vp8e_config(&ctx->priv->alg_priv->oxcf,
                      ctx->priv->alg_priv->cfg,
                      ctx->priv->alg_priv->vp8_cfg);
504
      optr = vp9_create_compressor(&ctx->priv->alg_priv->oxcf);
505

John Koleszar's avatar
John Koleszar committed
506 507 508 509
      if (!optr)
        res = VPX_CODEC_MEM_ERROR;
      else
        ctx->priv->alg_priv->cpi = optr;
John Koleszar's avatar
John Koleszar committed
510
    }
John Koleszar's avatar
John Koleszar committed
511
  }
John Koleszar's avatar
John Koleszar committed
512

John Koleszar's avatar
John Koleszar committed
513
  return res;
John Koleszar's avatar
John Koleszar committed
514 515
}

516

John Koleszar's avatar
John Koleszar committed
517 518
static vpx_codec_err_t vp8e_init(vpx_codec_ctx_t *ctx,
                                 vpx_codec_priv_enc_mr_cfg_t *data) {
John Koleszar's avatar
John Koleszar committed
519
  return vp8e_common_init(ctx, 0);
520 521 522 523
}


#if CONFIG_EXPERIMENTAL
John Koleszar's avatar
John Koleszar committed
524 525
static vpx_codec_err_t vp8e_exp_init(vpx_codec_ctx_t *ctx,
                                     vpx_codec_priv_enc_mr_cfg_t *data) {
John Koleszar's avatar
John Koleszar committed
526
  return vp8e_common_init(ctx, 1);
527 528 529 530
}
#endif


John Koleszar's avatar
John Koleszar committed
531
static vpx_codec_err_t vp8e_destroy(vpx_codec_alg_priv_t *ctx) {
John Koleszar's avatar
John Koleszar committed
532

John Koleszar's avatar
John Koleszar committed
533
  free(ctx->cx_data);
534
  vp9_remove_compressor(&ctx->cpi);
John Koleszar's avatar
John Koleszar committed
535 536
  free(ctx);
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
537 538 539
}

static vpx_codec_err_t image2yuvconfig(const vpx_image_t   *img,
John Koleszar's avatar
John Koleszar committed
540 541 542 543 544 545
                                       YV12_BUFFER_CONFIG  *yv12) {
  vpx_codec_err_t        res = VPX_CODEC_OK;
  yv12->y_buffer = img->planes[VPX_PLANE_Y];
  yv12->u_buffer = img->planes[VPX_PLANE_U];
  yv12->v_buffer = img->planes[VPX_PLANE_V];

John Koleszar's avatar
John Koleszar committed
546 547
  yv12->y_crop_width  = img->d_w;
  yv12->y_crop_height = img->d_h;
John Koleszar's avatar
John Koleszar committed
548 549 550 551 552 553 554 555 556 557 558
  yv12->y_width  = img->d_w;
  yv12->y_height = img->d_h;
  yv12->uv_width = (1 + yv12->y_width) / 2;
  yv12->uv_height = (1 + yv12->y_height) / 2;

  yv12->y_stride = img->stride[VPX_PLANE_Y];
  yv12->uv_stride = img->stride[VPX_PLANE_U];

  yv12->border  = (img->stride[VPX_PLANE_Y] - img->w) / 2;
  yv12->clrtype = (img->fmt == VPX_IMG_FMT_VPXI420 || img->fmt == VPX_IMG_FMT_VPXYV12); // REG_YUV = 0
  return res;
John Koleszar's avatar
John Koleszar committed
559 560 561 562
}

static void pick_quickcompress_mode(vpx_codec_alg_priv_t  *ctx,
                                    unsigned long          duration,
John Koleszar's avatar
John Koleszar committed
563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580
                                    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;
581
    vp9_change_config(ctx->cpi, &ctx->oxcf);
John Koleszar's avatar
John Koleszar committed
582
  }
John Koleszar's avatar
John Koleszar committed
583 584 585
}


586
static int write_superframe_index(vpx_codec_alg_priv_t *ctx) {
John Koleszar's avatar
John Koleszar committed
587 588 589 590 591 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
  uint8_t marker = 0xc0;
  int mag, mask, index_sz;

  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;
  }
623
  return index_sz;
John Koleszar's avatar
John Koleszar committed
624 625
}

John Koleszar's avatar
John Koleszar committed
626 627 628 629 630
static vpx_codec_err_t vp8e_encode(vpx_codec_alg_priv_t  *ctx,
                                   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
631 632
                                   unsigned long          deadline) {
  vpx_codec_err_t res = VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
633

John Koleszar's avatar
John Koleszar committed
634 635
  if (img)
    res = validate_img(ctx, img);
John Koleszar's avatar
John Koleszar committed
636

John Koleszar's avatar
John Koleszar committed
637 638
  pick_quickcompress_mode(ctx, duration, deadline);
  vpx_codec_pkt_list_init(&ctx->pkt_list);
John Koleszar's avatar
John Koleszar committed
639

John Koleszar's avatar
John Koleszar committed
640 641 642 643 644 645
  /* 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
646

John Koleszar's avatar
John Koleszar committed
647 648 649
  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
650

John Koleszar's avatar
John Koleszar committed
651
    if (flags & VP8_EFLAG_NO_REF_LAST)
652
      ref ^= VP9_LAST_FLAG;
John Koleszar's avatar
John Koleszar committed
653

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

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

660
    vp9_use_as_reference(ctx->cpi, ref);
John Koleszar's avatar
John Koleszar committed
661
  }
John Koleszar's avatar
John Koleszar committed
662

John Koleszar's avatar
John Koleszar committed
663 664 665 666
  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
667

John Koleszar's avatar
John Koleszar committed
668
    if (flags & VP8_EFLAG_NO_UPD_LAST)
669
      upd ^= VP9_LAST_FLAG;
John Koleszar's avatar
John Koleszar committed
670

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

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

677
    vp9_update_reference(ctx->cpi, upd);
John Koleszar's avatar
John Koleszar committed
678
  }
John Koleszar's avatar
John Koleszar committed
679

John Koleszar's avatar
John Koleszar committed
680
  if (flags & VP8_EFLAG_NO_UPD_ENTROPY) {
681
    vp9_update_entropy(ctx->cpi, 0);
John Koleszar's avatar
John Koleszar committed
682
  }
John Koleszar's avatar
John Koleszar committed
683

John Koleszar's avatar
John Koleszar committed
684 685 686 687 688 689
  /* 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
690
    }
John Koleszar's avatar
John Koleszar committed
691
  }
John Koleszar's avatar
John Koleszar committed
692

John Koleszar's avatar
John Koleszar committed
693 694 695 696 697 698 699
  /* 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
700

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

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

John Koleszar's avatar
John Koleszar committed
708 709
    /* 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
710

John Koleszar's avatar
John Koleszar committed
711 712 713
    /* vp8 use 10,000,000 ticks/second as time stamp */
    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
714

John Koleszar's avatar
John Koleszar committed
715 716
    if (img != NULL) {
      res = image2yuvconfig(img, &sd);
John Koleszar's avatar
John Koleszar committed
717

718
      if (vp9_receive_raw_frame(ctx->cpi, lib_flags,
John Koleszar's avatar
John Koleszar committed
719
                                &sd, dst_time_stamp, dst_end_time_stamp)) {
720
        VP9_COMP *cpi = (VP9_COMP *)ctx->cpi;
John Koleszar's avatar
John Koleszar committed
721 722 723
        res = update_error_state(ctx, &cpi->common.error);
      }
    }
John Koleszar's avatar
John Koleszar committed
724

725 726
    cx_data = ctx->cx_data;
    cx_data_sz = ctx->cx_data_sz;
John Koleszar's avatar
John Koleszar committed
727 728
    lib_flags = 0;

729 730 731 732 733 734 735 736 737 738 739 740 741 742
    /* 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;
      }
743 744
    }

745 746 747 748
    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
749 750 751
      if (size) {
        vpx_codec_pts_t    round, delta;
        vpx_codec_cx_pkt_t pkt;
752
        VP9_COMP *cpi = (VP9_COMP *)ctx->cpi;
John Koleszar's avatar
John Koleszar committed
753

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

John Koleszar's avatar
John Koleszar committed
766 767 768 769 770 771 772
        /* Add the frame packet to the list of returned packets. */
        round = 1000000 * ctx->cfg.g_timebase.num / 2 - 1;
        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;
773 774 775
        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
776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791
        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
792 793
        }

John Koleszar's avatar
John Koleszar committed
794 795
        if (cpi->droppable)
          pkt.data.frame.flags |= VPX_FRAME_IS_DROPPABLE;
John Koleszar's avatar
John Koleszar committed
796

John Koleszar's avatar
John Koleszar committed
797
        /*if (cpi->output_partition)
John Koleszar's avatar
John Koleszar committed
798
        {
John Koleszar's avatar
John Koleszar committed
799 800 801 802 803 804
            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
805
            {
John Koleszar's avatar
John Koleszar committed
806 807 808 809 810 811 812 813 814
                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
815 816
            }
        }
John Koleszar's avatar
John Koleszar committed
817 818
        else*/
        {
819
          if (ctx->pending_cx_data) {
John Koleszar's avatar
John Koleszar committed
820 821 822
            ctx->pending_frame_sizes[ctx->pending_frame_count++] = size;
            ctx->pending_frame_magnitude |= size;
            ctx->pending_cx_data_sz += size;
823
            size += write_superframe_index(ctx);
824
            pkt.data.frame.buf = ctx->pending_cx_data;
John Koleszar's avatar
John Koleszar committed
825
            pkt.data.frame.sz  = ctx->pending_cx_data_sz;
826 827
            ctx->pending_cx_data = NULL;
            ctx->pending_cx_data_sz = 0;
John Koleszar's avatar
John Koleszar committed
828 829
            ctx->pending_frame_count = 0;
            ctx->pending_frame_magnitude = 0;
830 831
          } else {
            pkt.data.frame.buf = cx_data;
832
            pkt.data.frame.sz  = size;
833
          }
John Koleszar's avatar
John Koleszar committed
834 835 836 837 838 839 840 841
          pkt.data.frame.partition_id = -1;
          vpx_codec_pkt_list_add(&ctx->pkt_list.head, &pkt);
          cx_data += size;
          cx_data_sz -= size;
        }

        // printf("timestamp: %lld, duration: %d\n", pkt->data.frame.pts, pkt->data.frame.duration);
      }
John Koleszar's avatar
John Koleszar committed
842
    }
John Koleszar's avatar
John Koleszar committed
843
  }
John Koleszar's avatar
John Koleszar committed
844

John Koleszar's avatar
John Koleszar committed
845
  return res;
John Koleszar's avatar
John Koleszar committed
846 847 848 849
}


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

static vpx_codec_err_t vp8e_set_reference(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
855 856 857
                                          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
858

John Koleszar's avatar
John Koleszar committed
859 860 861
  if (data) {
    vpx_ref_frame_t *frame = (vpx_ref_frame_t *)data;
    YV12_BUFFER_CONFIG sd;
John Koleszar's avatar
John Koleszar committed
862

John Koleszar's avatar
John Koleszar committed
863
    image2yuvconfig(&frame->img, &sd);
864
    vp9_set_reference_enc(ctx->cpi, frame->frame_type, &sd);
John Koleszar's avatar
John Koleszar committed
865 866 867
    return VPX_CODEC_OK;
  } else
    return VPX_CODEC_INVALID_PARAM;
John Koleszar's avatar
John Koleszar committed
868 869 870

}

John Koleszar's avatar
John Koleszar committed
871 872 873
static vpx_codec_err_t vp8e_copy_reference(vpx_codec_alg_priv_t *ctx,
                                           int ctr_id,
                                           va_list args) {
John Koleszar's avatar
John Koleszar committed
874

John Koleszar's avatar
John Koleszar committed
875
  vpx_ref_frame_t *data = va_arg(args, vpx_ref_frame_t *);
John Koleszar's avatar
John Koleszar committed
876

John Koleszar's avatar
John Koleszar committed
877 878 879
  if (data) {
    vpx_ref_frame_t *frame = (vpx_ref_frame_t *)data;
    YV12_BUFFER_CONFIG sd;
John Koleszar's avatar
John Koleszar committed
880

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

John Koleszar's avatar
John Koleszar committed
888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903
static vpx_codec_err_t get_reference(vpx_codec_alg_priv_t *ctx,
                                     int ctr_id,
                                     va_list args) {
  vp9_ref_frame_t *data = va_arg(args, vp9_ref_frame_t *);

  if (data) {
    YV12_BUFFER_CONFIG* fb;

    vp9_get_reference_enc(ctx->cpi, data->idx, &fb);
    yuvconfig2image(&data->img, fb, NULL);
    return VPX_CODEC_OK;
  } else {
    return VPX_CODEC_INVALID_PARAM;
  }
}

John Koleszar's avatar
John Koleszar committed
904
static vpx_codec_err_t vp8e_set_previewpp(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
905 906
                                          int ctr_id,
                                          va_list args) {
907
#if CONFIG_POSTPROC
John Koleszar's avatar
John Koleszar committed
908 909
  vp8_postproc_cfg_t *data = va_arg(args, vp8_postproc_cfg_t *);
  (void)ctr_id;
John Koleszar's avatar
John Koleszar committed
910

John Koleszar's avatar
John Koleszar committed
911 912 913 914 915
  if (data) {
    ctx->preview_ppcfg = *((vp8_postproc_cfg_t *)data);
    return VPX_CODEC_OK;
  } else
    return VPX_CODEC_INVALID_PARAM;
916
#else
John Koleszar's avatar
John Koleszar committed
917 918 919 920
  (void)ctx;
  (void)ctr_id;
  (void)args;
  return VPX_CODEC_INCAPABLE;
921
#endif
John Koleszar's avatar
John Koleszar committed
922 923 924
}


John Koleszar's avatar
John Koleszar committed
925
static vpx_image_t *vp8e_get_preview(vpx_codec_alg_priv_t *ctx) {
John Koleszar's avatar
John Koleszar committed
926

John Koleszar's avatar
John Koleszar committed
927
  YV12_BUFFER_CONFIG sd;
928
  vp9_ppflags_t flags = {0};
John Koleszar's avatar
John Koleszar committed
929

John Koleszar's avatar
John Koleszar committed
930 931 932 933 934
  if (ctx->preview_ppcfg.post_proc_flag) {
    flags.post_proc_flag        = ctx->preview_ppcfg.post_proc_flag;
    flags.deblocking_level      = ctx->preview_ppcfg.deblocking_level;
    flags.noise_level           = ctx->preview_ppcfg.noise_level;
  }
935

936
  if (0 == vp9_get_preview_raw_frame(ctx->cpi, &sd, &flags)) {
John Koleszar's avatar
John Koleszar committed
937

John Koleszar's avatar
John Koleszar committed
938 939
    /*
    vpx_img_wrap(&ctx->preview_img, VPX_IMG_FMT_YV12,
John Koleszar's avatar
John Koleszar committed
940 941
        sd.y_width + 2*VP9BORDERINPIXELS,
        sd.y_height + 2*VP9BORDERINPIXELS,
John Koleszar's avatar
John Koleszar committed
942 943 944
        1,
        sd.buffer_alloc);
    vpx_img_set_rect(&ctx->preview_img,
John Koleszar's avatar
John Koleszar committed
945
        VP9BORDERINPIXELS, VP9BORDERINPIXELS,
John Koleszar's avatar
John Koleszar committed
946 947 948 949 950 951 952 953 954 955
        sd.y_width, sd.y_height);
        */

    ctx->preview_img.bps = 12;
    ctx->preview_img.planes[VPX_PLANE_Y] = sd.y_buffer;
    ctx->preview_img.planes[VPX_PLANE_U] = sd.u_buffer;
    ctx->preview_img.planes[VPX_PLANE_V] = sd.v_buffer;

    if (sd.clrtype == REG_YUV)
      ctx->preview_img.fmt = VPX_IMG_FMT_I420;
John Koleszar's avatar
John Koleszar committed
956
    else
John Koleszar's avatar
John Koleszar committed
957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972
      ctx->preview_img.fmt = VPX_IMG_FMT_VPXI420;

    ctx->preview_img.x_chroma_shift = 1;
    ctx->preview_img.y_chroma_shift = 1;

    ctx->preview_img.d_w = sd.y_width;
    ctx->preview_img.d_h = sd.y_height;
    ctx->preview_img.stride[VPX_PLANE_Y] = sd.y_stride;
    ctx->preview_img.stride[VPX_PLANE_U] = sd.uv_stride;
    ctx->preview_img.stride[VPX_PLANE_V] = sd.uv_stride;
    ctx->preview_img.w   = sd.y_width;
    ctx->preview_img.h   = sd.y_height;

    return &ctx->preview_img;
  } else
    return NULL;
John Koleszar's avatar
John Koleszar committed
973 974 975
}

static vpx_codec_err_t vp8e_update_entropy(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
976 977 978
                                           int ctr_id,
                                           va_list args) {
  int update = va_arg(args, int);
979
  vp9_update_entropy(ctx->cpi, update);
John Koleszar's avatar
John Koleszar committed
980
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
981 982 983 984

}

static vpx_codec_err_t vp8e_update_reference(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
985 986 987
                                             int ctr_id,
                                             va_list args) {
  int update = va_arg(args, int);
988
  vp9_update_reference(ctx->cpi, update);
John Koleszar's avatar
John Koleszar committed
989
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
990 991 992
}

static vpx_codec_err_t vp8e_use_reference(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
993 994 995
                                          int ctr_id,
                                          va_list args) {
  int reference_flag = va_arg(args, int);
996
  vp9_use_as_reference(ctx->cpi, reference_flag);
John Koleszar's avatar
John Koleszar committed
997
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
998 999 1000 1001
}

static vpx_codec_err_t vp8e_set_roi_map(vpx_codec_alg_priv_t *ctx,
                                        int ctr_id,
John Koleszar's avatar
John Koleszar committed
1002 1003
                                        va_list args) {
  vpx_roi_map_t *data = va_arg(args, vpx_roi_map_t *);
John Koleszar's avatar
John Koleszar committed
1004

John Koleszar's avatar
John Koleszar committed
1005 1006
  if (data) {
    vpx_roi_map_t *roi = (vpx_roi_map_t *)data;
John Koleszar's avatar
John Koleszar committed
1007