vp9_cx_iface.c 40.2 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;
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 45
struct extraconfig_map {
  int                 usage;
  struct vp8_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,
John Koleszar's avatar
John Koleszar committed
51
    {
John Koleszar's avatar
John Koleszar committed
52 53 54 55 56 57
      NULL,
      0,                          /* cpu_used      */
      0,                          /* enable_auto_alt_ref */
      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 */
John Koleszar's avatar
John Koleszar committed
60 61 62 63 64 65 66
      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
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 76
struct vpx_codec_alg_priv {
  vpx_codec_priv_t        base;
  vpx_codec_enc_cfg_t     cfg;
  struct vp8_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 88 89
  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
90 91 92 93 94
};


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

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

John Koleszar's avatar
John Koleszar committed
103
  return res;
John Koleszar's avatar
John Koleszar committed
104 105 106
}


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

#define RANGE_CHECK(p,memb,lo,hi) do {\
John Koleszar's avatar
John Koleszar committed
114 115 116
    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
117

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

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

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

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
134
                                       const struct vp8_extracfg *vp8_cfg) {
Jingning Han's avatar
Jingning Han committed
135 136
  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
137 138 139
  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
140

John Koleszar's avatar
John Koleszar committed
141 142
  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
143 144 145 146 147 148
  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
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 174 175
  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);

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


static vpx_codec_err_t validate_img(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
210 211
                                    const vpx_image_t    *img) {
  switch (img->fmt) {
James Zern's avatar
James Zern committed
212 213 214 215
    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
216
      break;
John Koleszar's avatar
John Koleszar committed
217
    default:
John Koleszar's avatar
John Koleszar committed
218 219
      ERROR("Invalid image format. Only YV12 and I420 images are supported");
  }
John Koleszar's avatar
John Koleszar committed
220

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

John Koleszar's avatar
John Koleszar committed
224
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
225 226 227
}


228
static vpx_codec_err_t set_vp8e_config(VP9_CONFIG *oxcf,
John Koleszar's avatar
John Koleszar committed
229
                                       vpx_codec_enc_cfg_t cfg,
John Koleszar's avatar
John Koleszar committed
230
                                       struct vp8_extracfg vp8_cfg) {
231 232 233
  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
234 235
  /* 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
236

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

John Koleszar's avatar
John Koleszar committed
241
  switch (cfg.g_pass) {
John Koleszar's avatar
John Koleszar committed
242
    case VPX_RC_ONE_PASS:
John Koleszar's avatar
John Koleszar committed
243 244
      oxcf->Mode = MODE_BESTQUALITY;
      break;
John Koleszar's avatar
John Koleszar committed
245
    case VPX_RC_FIRST_PASS:
John Koleszar's avatar
John Koleszar committed
246 247
      oxcf->Mode = MODE_FIRSTPASS;
      break;
John Koleszar's avatar
John Koleszar committed
248
    case VPX_RC_LAST_PASS:
John Koleszar's avatar
John Koleszar committed
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 309 310
      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
311

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

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

317
  oxcf->error_resilient_mode = cfg.g_error_resilient;
318
  oxcf->frame_parallel_decoding_mode = vp8_cfg.frame_parallel_decoding_mode;
John Koleszar's avatar
John Koleszar committed
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 345 346
  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);
347 348
  printf("frame parallel detokenization: %d\n",
         oxcf->frame_parallel_decoding_mode);
John Koleszar's avatar
John Koleszar committed
349 350
  */
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
351 352 353
}

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

John Koleszar's avatar
John Koleszar committed
357 358
  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
359

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

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

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

John Koleszar's avatar
John Koleszar committed
376
  return res;
John Koleszar's avatar
John Koleszar committed
377 378 379
}


380
int vp9_reverse_trans(int q);
John Koleszar's avatar
John Koleszar committed
381 382 383 384


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

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

John Koleszar's avatar
John Koleszar committed
390 391
  if (!arg)
    return VPX_CODEC_INVALID_PARAM;
John Koleszar's avatar
John Koleszar committed
392

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

John Koleszar's avatar
John Koleszar committed
399
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
400 401 402 403 404 405
#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
406 407 408
                                 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
409 410 411

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

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

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

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

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

  return res;
John Koleszar's avatar
John Koleszar committed
440 441
#undef MAP
}
442 443 444


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

451
  VP9_PTR optr;
John Koleszar's avatar
John Koleszar committed
452

John Koleszar's avatar
John Koleszar committed
453 454
  if (!ctx->priv) {
    priv = calloc(1, sizeof(struct vpx_codec_alg_priv));
John Koleszar's avatar
John Koleszar committed
455

John Koleszar's avatar
John Koleszar committed
456 457 458
    if (!priv) {
      return VPX_CODEC_MEM_ERROR;
    }
459

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

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

John Koleszar's avatar
John Koleszar committed
475
    cfg =  &ctx->priv->alg_priv->cfg;
John Koleszar's avatar
John Koleszar committed
476

John Koleszar's avatar
John Koleszar committed
477 478 479 480 481 482 483
    /* 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
484

John Koleszar's avatar
John Koleszar committed
485 486 487
    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
488

489 490 491 492
    // TODO(agrange) Check the limits set on this buffer, or the check that is
    // applied in vp8e_encode.
    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
493

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

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

John Koleszar's avatar
John Koleszar committed
498 499 500
    if (!priv->cx_data) {
      return VPX_CODEC_MEM_ERROR;
    }
John Koleszar's avatar
John Koleszar committed
501

502
    vp9_initialize_enc();
John Koleszar's avatar
John Koleszar committed
503

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

John Koleszar's avatar
John Koleszar committed
506 507 508 509
    if (!res) {
      set_vp8e_config(&ctx->priv->alg_priv->oxcf,
                      ctx->priv->alg_priv->cfg,
                      ctx->priv->alg_priv->vp8_cfg);
510
      optr = vp9_create_compressor(&ctx->priv->alg_priv->oxcf);
511

John Koleszar's avatar
John Koleszar committed
512 513 514 515
      if (!optr)
        res = VPX_CODEC_MEM_ERROR;
      else
        ctx->priv->alg_priv->cpi = optr;
John Koleszar's avatar
John Koleszar committed
516
    }
John Koleszar's avatar
John Koleszar committed
517
  }
John Koleszar's avatar
John Koleszar committed
518

John Koleszar's avatar
John Koleszar committed
519
  return res;
John Koleszar's avatar
John Koleszar committed
520 521
}

522

John Koleszar's avatar
John Koleszar committed
523 524
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
525
  return vp8e_common_init(ctx, 0);
526 527 528 529
}


#if CONFIG_EXPERIMENTAL
John Koleszar's avatar
John Koleszar committed
530 531
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
532
  return vp8e_common_init(ctx, 1);
533 534 535 536
}
#endif


John Koleszar's avatar
John Koleszar committed
537
static vpx_codec_err_t vp8e_destroy(vpx_codec_alg_priv_t *ctx) {
John Koleszar's avatar
John Koleszar committed
538

John Koleszar's avatar
John Koleszar committed
539
  free(ctx->cx_data);
540
  vp9_remove_compressor(&ctx->cpi);
John Koleszar's avatar
John Koleszar committed
541 542
  free(ctx);
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
543 544 545
}

static vpx_codec_err_t image2yuvconfig(const vpx_image_t   *img,
John Koleszar's avatar
John Koleszar committed
546 547 548 549 550 551
                                       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
552 553
  yv12->y_crop_width  = img->d_w;
  yv12->y_crop_height = img->d_h;
John Koleszar's avatar
John Koleszar committed
554 555 556 557 558 559 560 561 562 563 564
  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
565 566 567 568
}

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


592
static int write_superframe_index(vpx_codec_alg_priv_t *ctx) {
John Koleszar's avatar
John Koleszar committed
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 626 627 628
  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;
  }
629
  return index_sz;
John Koleszar's avatar
John Koleszar committed
630 631
}

John Koleszar's avatar
John Koleszar committed
632 633 634 635 636
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
637 638
                                   unsigned long          deadline) {
  vpx_codec_err_t res = VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
639

John Koleszar's avatar
John Koleszar committed
640 641
  if (img)
    res = validate_img(ctx, img);
John Koleszar's avatar
John Koleszar committed
642

John Koleszar's avatar
John Koleszar committed
643 644
  pick_quickcompress_mode(ctx, duration, deadline);
  vpx_codec_pkt_list_init(&ctx->pkt_list);
John Koleszar's avatar
John Koleszar committed
645

John Koleszar's avatar
John Koleszar committed
646 647 648 649 650 651
  /* 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
652

John Koleszar's avatar
John Koleszar committed
653 654 655
  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
656

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

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

John Koleszar's avatar
John Koleszar committed
663
    if (flags & VP8_EFLAG_NO_REF_ARF)
664
      ref ^= VP9_ALT_FLAG;
John Koleszar's avatar
John Koleszar committed
665

666
    vp9_use_as_reference(ctx->cpi, ref);
John Koleszar's avatar
John Koleszar committed
667
  }
John Koleszar's avatar
John Koleszar committed
668

John Koleszar's avatar
John Koleszar committed
669 670 671 672
  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
673

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

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

John Koleszar's avatar
John Koleszar committed
680
    if (flags & VP8_EFLAG_NO_UPD_ARF)
681
      upd ^= VP9_ALT_FLAG;
John Koleszar's avatar
John Koleszar committed
682

683
    vp9_update_reference(ctx->cpi, upd);
John Koleszar's avatar
John Koleszar committed
684
  }
John Koleszar's avatar
John Koleszar committed
685

John Koleszar's avatar
John Koleszar committed
686
  if (flags & VP8_EFLAG_NO_UPD_ENTROPY) {
687
    vp9_update_entropy(ctx->cpi, 0);
John Koleszar's avatar
John Koleszar committed
688
  }
John Koleszar's avatar
John Koleszar committed
689

John Koleszar's avatar
John Koleszar committed
690 691 692 693 694 695
  /* 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
696
    }
John Koleszar's avatar
John Koleszar committed
697
  }
John Koleszar's avatar
John Koleszar committed
698

John Koleszar's avatar
John Koleszar committed
699 700 701 702 703 704 705
  /* 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
706

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

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

John Koleszar's avatar
John Koleszar committed
714 715
    /* 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
716

John Koleszar's avatar
John Koleszar committed
717 718 719
    /* 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
720

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

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

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

735 736 737 738 739 740 741 742 743 744 745 746 747 748
    /* 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;
      }
749 750
    }

751 752 753 754
    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
755 756 757
      if (size) {
        vpx_codec_pts_t    round, delta;
        vpx_codec_cx_pkt_t pkt;
758
        VP9_COMP *cpi = (VP9_COMP *)ctx->cpi;
John Koleszar's avatar
John Koleszar committed
759

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

John Koleszar's avatar
John Koleszar committed
772 773 774 775 776 777 778
        /* 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;
779 780 781
        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
782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797
        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
798 799
        }

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

John Koleszar's avatar
John Koleszar committed
803
        /*if (cpi->output_partition)
John Koleszar's avatar
John Koleszar committed
804
        {
John Koleszar's avatar
John Koleszar committed
805 806 807 808 809 810
            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
811
            {
John Koleszar's avatar
John Koleszar committed
812 813 814 815 816 817 818 819 820
                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
821 822
            }
        }
John Koleszar's avatar
John Koleszar committed
823 824
        else*/
        {
825
          if (ctx->pending_cx_data) {
John Koleszar's avatar
John Koleszar committed
826 827 828
            ctx->pending_frame_sizes[ctx->pending_frame_count++] = size;
            ctx->pending_frame_magnitude |= size;
            ctx->pending_cx_data_sz += size;
829
            size += write_superframe_index(ctx);
830
            pkt.data.frame.buf = ctx->pending_cx_data;
John Koleszar's avatar
John Koleszar committed
831
            pkt.data.frame.sz  = ctx->pending_cx_data_sz;
832 833
            ctx->pending_cx_data = NULL;
            ctx->pending_cx_data_sz = 0;
John Koleszar's avatar
John Koleszar committed
834 835
            ctx->pending_frame_count = 0;
            ctx->pending_frame_magnitude = 0;
836 837
          } else {
            pkt.data.frame.buf = cx_data;
838
            pkt.data.frame.sz  = size;
839
          }
John Koleszar's avatar
John Koleszar committed
840 841 842 843 844 845 846 847
          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
848
    }
John Koleszar's avatar
John Koleszar committed
849
  }
John Koleszar's avatar
John Koleszar committed
850

John Koleszar's avatar
John Koleszar committed
851
  return res;
John Koleszar's avatar
John Koleszar committed
852 853 854 855
}


static const vpx_codec_cx_pkt_t *vp8e_get_cxdata(vpx_codec_alg_priv_t  *ctx,
John Koleszar's avatar
John Koleszar committed
856 857
                                                 vpx_codec_iter_t      *iter) {
  return vpx_codec_pkt_list_get(&ctx->pkt_list.head, iter);
John Koleszar's avatar
John Koleszar committed
858 859 860
}

static vpx_codec_err_t vp8e_set_reference(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
861 862 863
                                          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
864

John Koleszar's avatar
John Koleszar committed
865 866 867
  if (data) {
    vpx_ref_frame_t *frame = (vpx_ref_frame_t *)data;
    YV12_BUFFER_CONFIG sd;
John Koleszar's avatar
John Koleszar committed
868

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

}

John Koleszar's avatar
John Koleszar committed
877 878 879
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
880

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

John Koleszar's avatar
John Koleszar committed
883 884 885
  if (data) {
    vpx_ref_frame_t *frame = (vpx_ref_frame_t *)data;
    YV12_BUFFER_CONFIG sd;
John Koleszar's avatar
John Koleszar committed
886

John Koleszar's avatar
John Koleszar committed
887
    image2yuvconfig(&frame->img, &sd);
John Koleszar's avatar
John Koleszar committed
888
    vp9_copy_reference_enc(ctx->cpi, frame->frame_type, &sd);
John Koleszar's avatar
John Koleszar committed
889 890 891
    return VPX_CODEC_OK;
  } else
    return VPX_CODEC_INVALID_PARAM;
John Koleszar's avatar
John Koleszar committed
892 893
}

John Koleszar's avatar
John Koleszar committed
894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909
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
910
static vpx_codec_err_t vp8e_set_previewpp(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
911 912
                                          int ctr_id,
                                          va_list args) {
913
#if CONFIG_POSTPROC
John Koleszar's avatar
John Koleszar committed
914 915
  vp8_postproc_cfg_t *data = va_arg(args, vp8_postproc_cfg_t *);
  (void)ctr_id;
John Koleszar's avatar
John Koleszar committed
916

John Koleszar's avatar
John Koleszar committed
917 918 919 920 921
  if (data) {
    ctx->preview_ppcfg = *((vp8_postproc_cfg_t *)data);
    return VPX_CODEC_OK;
  } else
    return VPX_CODEC_INVALID_PARAM;
922
#else
John Koleszar's avatar
John Koleszar committed
923 924 925 926
  (void)ctx;
  (void)ctr_id;
  (void)args;
  return VPX_CODEC_INCAPABLE;
927
#endif
John Koleszar's avatar
John Koleszar committed
928 929 930
}


John Koleszar's avatar
John Koleszar committed
931
static vpx_image_t *vp8e_get_preview(vpx_codec_alg_priv_t *ctx) {
John Koleszar's avatar
John Koleszar committed
932

John Koleszar's avatar
John Koleszar committed
933
  YV12_BUFFER_CONFIG sd;
934
  vp9_ppflags_t flags = {0};
John Koleszar's avatar
John Koleszar committed
935

John Koleszar's avatar
John Koleszar committed
936 937 938 939 940
  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;
  }
941

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

John Koleszar's avatar
John Koleszar committed
944 945
    /*
    vpx_img_wrap(&ctx->preview_img, VPX_IMG_FMT_YV12,
John Koleszar's avatar
John Koleszar committed
946 947
        sd.y_width + 2*VP9BORDERINPIXELS,
        sd.y_height + 2*VP9BORDERINPIXELS,
John Koleszar's avatar
John Koleszar committed
948 949 950
        1,
        sd.buffer_alloc);
    vpx_img_set_rect(&ctx->preview_img,
John Koleszar's avatar
John Koleszar committed
951
        VP9BORDERINPIXELS, VP9BORDERINPIXELS,
John Koleszar's avatar
John Koleszar committed
952 953 954 955 956 957 958 959 960 961
        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
962
    else
John Koleszar's avatar
John Koleszar committed
963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978
      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
979 980 981
}

static vpx_codec_err_t vp8e_update_entropy(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
982 983 984
                                           int ctr_id,
                                           va_list args) {
  int update = va_arg(args, int);
985
  vp9_update_entropy(ctx->cpi, update);
John Koleszar's avatar
John Koleszar committed
986
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
987 988 989 990

}

static vpx_codec_err_t vp8e_update_reference(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
991 992 993
                                             int ctr_id,
                                             va_list args) {
  int update = va_arg(args, int);
994
  vp9_update_reference(ctx->cpi, update);
John Koleszar's avatar
John Koleszar committed
995
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
996 997 998
}

static vpx_codec_err_t vp8e_use_reference(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
999 1000 1001
                                          int ctr_id,
                                          va_list args) {
  int reference_flag = va_arg(args, int);
1002
  vp9_use_as_reference(ctx->cpi, reference_flag);
John Koleszar's avatar
John Koleszar committed
1003
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
1004 1005 1006 1007
}

static vpx_codec_err_t vp8e_set_roi_map(vpx_codec_alg_priv_t *ctx,
                                        int ctr_id,