vp9_cx_iface.c 37.8 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
20
21
#include <stdlib.h>
#include <string.h>

John Koleszar's avatar
John Koleszar committed
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
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;
  unsigned int                token_partitions;
  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
37
38
39
#if CONFIG_LOSSLESS
  unsigned int                lossless;
#endif
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
57
58
59
60
61
62
63
64
      NULL,
      0,                          /* cpu_used      */
      0,                          /* enable_auto_alt_ref */
      0,                          /* noise_sensitivity */
      0,                          /* Sharpness */
      0,                          /* static_thresh */
      VP8_ONE_TOKENPARTITION,     /* token_partitions */
      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
65
66
67
#if CONFIG_LOSSLESS
      0,                          /* lossless */
#endif
John Koleszar's avatar
John Koleszar committed
68
    }
John Koleszar's avatar
John Koleszar committed
69
  }
John Koleszar's avatar
John Koleszar committed
70
71
};

John Koleszar's avatar
John Koleszar committed
72
73
74
75
struct vpx_codec_alg_priv {
  vpx_codec_priv_t        base;
  vpx_codec_enc_cfg_t     cfg;
  struct vp8_extracfg     vp8_cfg;
76
  VP9_CONFIG              oxcf;
77
  VP9_PTR             cpi;
John Koleszar's avatar
John Koleszar committed
78
79
  unsigned char          *cx_data;
  unsigned int            cx_data_sz;
80
81
  unsigned char          *pending_cx_data;
  unsigned int            pending_cx_data_sz;
John Koleszar's avatar
John Koleszar committed
82
83
84
85
  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
86
87
88
89
90
};


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

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

John Koleszar's avatar
John Koleszar committed
99
  return res;
John Koleszar's avatar
John Koleszar committed
100
101
102
}


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

#define RANGE_CHECK(p,memb,lo,hi) do {\
John Koleszar's avatar
John Koleszar committed
110
111
112
    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
113

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

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

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

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
130
131
132
133
134
135
                                       const struct vp8_extracfg *vp8_cfg) {
  RANGE_CHECK(cfg, g_w,                   1, 16383); /* 14 bits available */
  RANGE_CHECK(cfg, g_h,                   1, 16383); /* 14 bits available */
  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
136

John Koleszar's avatar
John Koleszar committed
137
138
  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
139
140
141
142
143
144
145
146
#if CONFIG_LOSSLESS
  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);
  }
#endif

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
174
175
176
177
178
179
180
181
182
  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);

  RANGE_CHECK(vp8_cfg, token_partitions,   VP8_ONE_TOKENPARTITION, VP8_EIGHT_TOKENPARTITION);
  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);
183
    int              n_packets = (int)(cfg->rc_twopass_stats_in.sz / packet_sz);
John Koleszar's avatar
John Koleszar committed
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
    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
203
204
205
206
}


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

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

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


225
static vpx_codec_err_t set_vp8e_config(VP9_CONFIG *oxcf,
John Koleszar's avatar
John Koleszar committed
226
                                       vpx_codec_enc_cfg_t cfg,
John Koleszar's avatar
John Koleszar committed
227
228
229
                                       struct vp8_extracfg vp8_cfg) {
  oxcf->Version               = cfg.g_profile;
  oxcf->Version              |= vp8_cfg.experimental ? 0x4 : 0;
John Koleszar's avatar
John Koleszar committed
230

John Koleszar's avatar
John Koleszar committed
231
232
233
234
  oxcf->Width                 = cfg.g_w;
  oxcf->Height                = cfg.g_h;
  /* 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
235

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

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

Hui Su's avatar
Hui Su committed
311
#if CONFIG_LOSSLESS
John Koleszar's avatar
John Koleszar committed
312
  oxcf->lossless = vp8_cfg.lossless;
Hui Su's avatar
Hui Su committed
313
314
#endif

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
416
417
418
419
420
421
422
423
  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);
      MAP(VP8E_SET_TOKEN_PARTITIONS,      xcfg.token_partitions);

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

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

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

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


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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

517

John Koleszar's avatar
John Koleszar committed
518
519
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
520
  return vp8e_common_init(ctx, 0);
521
522
523
524
}


#if CONFIG_EXPERIMENTAL
John Koleszar's avatar
John Koleszar committed
525
526
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
527
  return vp8e_common_init(ctx, 1);
528
529
530
531
}
#endif


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

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

static vpx_codec_err_t image2yuvconfig(const vpx_image_t   *img,
John Koleszar's avatar
John Koleszar committed
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
                                       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];

  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
558
559
560
561
}

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


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
590
591
                                   unsigned long          deadline) {
  vpx_codec_err_t res = VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
592

John Koleszar's avatar
John Koleszar committed
593
594
  if (img)
    res = validate_img(ctx, img);
John Koleszar's avatar
John Koleszar committed
595

John Koleszar's avatar
John Koleszar committed
596
597
  pick_quickcompress_mode(ctx, duration, deadline);
  vpx_codec_pkt_list_init(&ctx->pkt_list);
John Koleszar's avatar
John Koleszar committed
598

John Koleszar's avatar
John Koleszar committed
599
600
601
602
603
604
  /* 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
605

John Koleszar's avatar
John Koleszar committed
606
607
608
  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
609

John Koleszar's avatar
John Koleszar committed
610
    if (flags & VP8_EFLAG_NO_REF_LAST)
611
      ref ^= VP9_LAST_FLAG;
John Koleszar's avatar
John Koleszar committed
612

John Koleszar's avatar
John Koleszar committed
613
    if (flags & VP8_EFLAG_NO_REF_GF)
614
      ref ^= VP9_GOLD_FLAG;
John Koleszar's avatar
John Koleszar committed
615

John Koleszar's avatar
John Koleszar committed
616
    if (flags & VP8_EFLAG_NO_REF_ARF)
617
      ref ^= VP9_ALT_FLAG;
John Koleszar's avatar
John Koleszar committed
618

619
    vp9_use_as_reference(ctx->cpi, ref);
John Koleszar's avatar
John Koleszar committed
620
  }
John Koleszar's avatar
John Koleszar committed
621

John Koleszar's avatar
John Koleszar committed
622
623
624
625
  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
626

John Koleszar's avatar
John Koleszar committed
627
    if (flags & VP8_EFLAG_NO_UPD_LAST)
628
      upd ^= VP9_LAST_FLAG;
John Koleszar's avatar
John Koleszar committed
629

John Koleszar's avatar
John Koleszar committed
630
    if (flags & VP8_EFLAG_NO_UPD_GF)
631
      upd ^= VP9_GOLD_FLAG;
John Koleszar's avatar
John Koleszar committed
632

John Koleszar's avatar
John Koleszar committed
633
    if (flags & VP8_EFLAG_NO_UPD_ARF)
634
      upd ^= VP9_ALT_FLAG;
John Koleszar's avatar
John Koleszar committed
635

636
    vp9_update_reference(ctx->cpi, upd);
John Koleszar's avatar
John Koleszar committed
637
  }
John Koleszar's avatar
John Koleszar committed
638

John Koleszar's avatar
John Koleszar committed
639
  if (flags & VP8_EFLAG_NO_UPD_ENTROPY) {
640
    vp9_update_entropy(ctx->cpi, 0);
John Koleszar's avatar
John Koleszar committed
641
  }
John Koleszar's avatar
John Koleszar committed
642

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

John Koleszar's avatar
John Koleszar committed
652
653
654
655
656
657
658
  /* 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
659

John Koleszar's avatar
John Koleszar committed
660
661
    /* Set up internal flags */
    if (ctx->base.init_flags & VPX_CODEC_USE_PSNR)
662
      ((VP9_COMP *)ctx->cpi)->b_calculate_psnr = 1;
John Koleszar's avatar
John Koleszar committed
663

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

John Koleszar's avatar
John Koleszar committed
667
668
    /* 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
669

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

John Koleszar's avatar
John Koleszar committed
674
675
    if (img != NULL) {
      res = image2yuvconfig(img, &sd);
John Koleszar's avatar
John Koleszar committed
676

677
      if (vp9_receive_raw_frame(ctx->cpi, lib_flags,
John Koleszar's avatar
John Koleszar committed
678
                                &sd, dst_time_stamp, dst_end_time_stamp)) {
679
        VP9_COMP *cpi = (VP9_COMP *)ctx->cpi;
John Koleszar's avatar
John Koleszar committed
680
681
682
        res = update_error_state(ctx, &cpi->common.error);
      }
    }
John Koleszar's avatar
John Koleszar committed
683

684
685
    cx_data = ctx->cx_data;
    cx_data_sz = ctx->cx_data_sz;
John Koleszar's avatar
John Koleszar committed
686
687
    lib_flags = 0;

688
689
690
691
692
693
694
695
696
697
698
699
700
701
    /* 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;
      }
702
703
    }

704
705
706
707
    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
708
709
710
      if (size) {
        vpx_codec_pts_t    round, delta;
        vpx_codec_cx_pkt_t pkt;
711
        VP9_COMP *cpi = (VP9_COMP *)ctx->cpi;
John Koleszar's avatar
John Koleszar committed
712

713
        /* Pack invisible frames with the next visisble frame */
714
        if (!cpi->common.show_frame) {
715
716
717
          if (!ctx->pending_cx_data)
            ctx->pending_cx_data = cx_data;
          ctx->pending_cx_data_sz += size;
718
719
720
721
722
          cx_data += size;
          cx_data_sz -= size;
          continue;
        }

John Koleszar's avatar
John Koleszar committed
723
724
725
726
727
728
729
        /* 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;
730
731
732
        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
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
        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
749
750
        }

John Koleszar's avatar
John Koleszar committed
751
752
        if (cpi->droppable)
          pkt.data.frame.flags |= VPX_FRAME_IS_DROPPABLE;
John Koleszar's avatar
John Koleszar committed
753

John Koleszar's avatar
John Koleszar committed
754
        /*if (cpi->output_partition)
John Koleszar's avatar
John Koleszar committed
755
        {
John Koleszar's avatar
John Koleszar committed
756
757
758
759
760
761
            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
762
            {
John Koleszar's avatar
John Koleszar committed
763
764
765
766
767
768
769
770
771
                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
772
773
            }
        }
John Koleszar's avatar
John Koleszar committed
774
775
        else*/
        {
776
777
778
779
780
          if (ctx->pending_cx_data) {
            pkt.data.frame.buf = ctx->pending_cx_data;
            pkt.data.frame.sz  = ctx->pending_cx_data_sz + size;
            ctx->pending_cx_data = NULL;
            ctx->pending_cx_data_sz = 0;
781
782
          } else {
            pkt.data.frame.buf = cx_data;
783
            pkt.data.frame.sz  = size;
784
          }
John Koleszar's avatar
John Koleszar committed
785
786
787
788
789
790
791
792
          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
793
    }
John Koleszar's avatar
John Koleszar committed
794
  }
John Koleszar's avatar
John Koleszar committed
795

John Koleszar's avatar
John Koleszar committed
796
  return res;
John Koleszar's avatar
John Koleszar committed
797
798
799
800
}


static const vpx_codec_cx_pkt_t *vp8e_get_cxdata(vpx_codec_alg_priv_t  *ctx,
John Koleszar's avatar
John Koleszar committed
801
802
                                                 vpx_codec_iter_t      *iter) {
  return vpx_codec_pkt_list_get(&ctx->pkt_list.head, iter);
John Koleszar's avatar
John Koleszar committed
803
804
805
}

static vpx_codec_err_t vp8e_set_reference(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
806
807
808
                                          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
809

John Koleszar's avatar
John Koleszar committed
810
811
812
  if (data) {
    vpx_ref_frame_t *frame = (vpx_ref_frame_t *)data;
    YV12_BUFFER_CONFIG sd;
John Koleszar's avatar
John Koleszar committed
813

John Koleszar's avatar
John Koleszar committed
814
    image2yuvconfig(&frame->img, &sd);
815
    vp9_set_reference_enc(ctx->cpi, frame->frame_type, &sd);
John Koleszar's avatar
John Koleszar committed
816
817
818
    return VPX_CODEC_OK;
  } else
    return VPX_CODEC_INVALID_PARAM;
John Koleszar's avatar
John Koleszar committed
819
820
821
822

}

static vpx_codec_err_t vp8e_get_reference(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
823
824
                                          int ctr_id,
                                          va_list args) {
John Koleszar's avatar
John Koleszar committed
825

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

John Koleszar's avatar
John Koleszar committed
828
829
830
  if (data) {
    vpx_ref_frame_t *frame = (vpx_ref_frame_t *)data;
    YV12_BUFFER_CONFIG sd;
John Koleszar's avatar
John Koleszar committed
831

John Koleszar's avatar
John Koleszar committed
832
    image2yuvconfig(&frame->img, &sd);
833
    vp9_get_reference_enc(ctx->cpi, frame->frame_type, &sd);
John Koleszar's avatar
John Koleszar committed
834
835
836
    return VPX_CODEC_OK;
  } else
    return VPX_CODEC_INVALID_PARAM;
John Koleszar's avatar
John Koleszar committed
837
838
839
}

static vpx_codec_err_t vp8e_set_previewpp(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
840
841
                                          int ctr_id,
                                          va_list args) {
842
#if CONFIG_POSTPROC
John Koleszar's avatar
John Koleszar committed
843
844
  vp8_postproc_cfg_t *data = va_arg(args, vp8_postproc_cfg_t *);
  (void)ctr_id;
John Koleszar's avatar
John Koleszar committed
845

John Koleszar's avatar
John Koleszar committed
846
847
848
849
850
  if (data) {
    ctx->preview_ppcfg = *((vp8_postproc_cfg_t *)data);
    return VPX_CODEC_OK;
  } else
    return VPX_CODEC_INVALID_PARAM;
851
#else
John Koleszar's avatar
John Koleszar committed
852
853
854
855
  (void)ctx;
  (void)ctr_id;
  (void)args;
  return VPX_CODEC_INCAPABLE;
856
#endif
John Koleszar's avatar
John Koleszar committed
857
858
859
}


John Koleszar's avatar
John Koleszar committed
860
static vpx_image_t *vp8e_get_preview(vpx_codec_alg_priv_t *ctx) {
John Koleszar's avatar
John Koleszar committed
861

John Koleszar's avatar
John Koleszar committed
862
  YV12_BUFFER_CONFIG sd;
863
  vp9_ppflags_t flags = {0};
John Koleszar's avatar
John Koleszar committed
864

John Koleszar's avatar
John Koleszar committed
865
866
867
868
869
  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;
  }
870

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

John Koleszar's avatar
John Koleszar committed
873
874
    /*
    vpx_img_wrap(&ctx->preview_img, VPX_IMG_FMT_YV12,
John Koleszar's avatar
John Koleszar committed
875
876
        sd.y_width + 2*VP9BORDERINPIXELS,
        sd.y_height + 2*VP9BORDERINPIXELS,
John Koleszar's avatar
John Koleszar committed
877
878
879
        1,
        sd.buffer_alloc);
    vpx_img_set_rect(&ctx->preview_img,
John Koleszar's avatar
John Koleszar committed
880
        VP9BORDERINPIXELS, VP9BORDERINPIXELS,
John Koleszar's avatar
John Koleszar committed
881
882
883
884
885
886
887
888
889
890
        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
891
    else
John Koleszar's avatar
John Koleszar committed
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
      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
908
909
910
}

static vpx_codec_err_t vp8e_update_entropy(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
911
912
913
                                           int ctr_id,
                                           va_list args) {
  int update = va_arg(args, int);
914
  vp9_update_entropy(ctx->cpi, update);
John Koleszar's avatar
John Koleszar committed
915
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
916
917
918
919

}

static vpx_codec_err_t vp8e_update_reference(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
920
921
922
                                             int ctr_id,
                                             va_list args) {
  int update = va_arg(args, int);
923
  vp9_update_reference(ctx->cpi, update);
John Koleszar's avatar
John Koleszar committed
924
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
925
926
927
}

static vpx_codec_err_t vp8e_use_reference(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
928
929
930
                                          int ctr_id,
                                          va_list args) {
  int reference_flag = va_arg(args, int);
931
  vp9_use_as_reference(ctx->cpi, reference_flag);
John Koleszar's avatar
John Koleszar committed
932
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
933
934
935
936
}

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
937
938
                                        va_list args) {
  vpx_roi_map_t *data = va_arg(args, vpx_roi_map_t *);
John Koleszar's avatar
John Koleszar committed
939

John Koleszar's avatar
John Koleszar committed
940
941
  if (data) {
    vpx_roi_map_t *roi = (vpx_roi_map_t *)data;
John Koleszar's avatar
John Koleszar committed
942

943
944
    if (!vp9_set_roimap(ctx->cpi, roi->roi_map, roi->rows, roi->cols,
                        roi->delta_q, roi->delta_lf, roi->static_threshold))
John Koleszar's avatar
John Koleszar committed
945
      return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
946
    else
John Koleszar's avatar
John Koleszar committed
947
948
949
      return VPX_CODEC_INVALID_PARAM;
  } else
    return VPX_CODEC_INVALID_PARAM;
John Koleszar's avatar
John Koleszar committed
950
951
952
953
}


static vpx_codec_err_t vp8e_set_activemap(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
954
955
956
                                          int ctr_id,
                                          va_list args) {
  vpx_active_map_t *data = va_arg(args, vpx_active_map_t *);
John Koleszar's avatar
John Koleszar committed
957

John Koleszar's avatar
John Koleszar committed
958
  if (data) {
John Koleszar's avatar
John Koleszar committed
959

John Koleszar's avatar
John Koleszar committed
960
    vpx_active_map_t *map = (vpx_active_map_t *)data;
John Koleszar's avatar
John Koleszar committed
961

962
    if (!vp9_set_active_map(ctx->cpi, map->active_map, map->rows, map->cols))
John Koleszar's avatar
John Koleszar committed
963
      return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
964
    else
John Koleszar's avatar
John Koleszar committed
965
966
967
      return VPX_CODEC_INVALID_PARAM;
  } else
    return VPX_CODEC_INVALID_PARAM;
John Koleszar's avatar
John Koleszar committed
968
969
970
}

static vpx_codec_err_t vp8e_set_scalemode(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
971
972
973
974
975
976
977
978
                                          int ctr_id,
                                          va_list args) {

  vpx_scaling_mode_t *data =  va_arg(args, vpx_scaling_mode_t *);

  if (data) {
    int res;
    vpx_scaling_mode_t scalemode = *(vpx_scaling_mode_t *)data;
979
980
    res = vp9_set_internal_size(ctx->cpi, scalemode.h_scaling_mode,
                                scalemode.v_scaling_mode);
John Koleszar's avatar
John Koleszar committed
981
982
983
984
985
986
987

    if (!res) {
      return VPX_CODEC_OK;
    } else
      return VPX_CODEC_INVALID_PARAM;
  } else
    return VPX_CODEC_INVALID_PARAM;
John Koleszar's avatar
John Koleszar committed
988
989
990
}


John Koleszar's avatar
John Koleszar committed
991
992
993
994
995
996
997
998
999
1000
1001
1002
1003
1004
1005
1006
1007
1008
1009
1010
1011
1012
1013
1014
static vpx_codec_ctrl_fn_map_t vp8e_ctf_maps[] = {
  {VP8_SET_REFERENCE,                 vp8e_set_reference},
  {VP8_COPY_REFERENCE,                vp8e_get_reference},
  {VP8_SET_POSTPROC,                  vp8e_set_previewpp},
  {VP8E_UPD_ENTROPY,                  vp8e_update_entropy},
  {VP8E_UPD_REFERENCE,                vp8e_update_reference},
  {VP8E_USE_REFERENCE,                vp8e_use_reference},
  {VP8E_SET_ROI_MAP,                  vp8e_set_roi_map},
  {VP8E_SET_ACTIVEMAP,                vp8e_set_activemap},
  {VP8E_SET_SCALEMODE,                vp8e_set_scalemode},
  {VP8E_SET_CPUUSED,                  set_param},
  {VP8E_SET_NOISE_SENSITIVITY,        set_param},
  {VP8E_SET_ENABLEAUTOALTREF,         set_param},
  {VP8E_SET_SHARPNESS,                set_param},
  {VP8E_SET_STATIC_THRESHOLD,         set_param},
  {VP8E_SET_TOKEN_PARTITIONS,         set_param},
  {VP8E_GET_LAST_QUANTIZER,           get_param},
  {VP8E_GET_LAST_QUANTIZER_64,        get_param},
  {VP8E_SET_ARNR_MAXFRAMES,           set_param},
  {VP8E_SET_ARNR_STRENGTH,           set_param},
  {VP8E_SET_ARNR_TYPE,           set_param},
  {VP8E_SET_TUNING,                   set_param},
  {VP8E_SET_CQ_LEVEL,                 set_param},
  {VP8E_SET_MAX_INTRA_BITRATE_PCT,    set_param},
John Koleszar's avatar
John Koleszar committed
1015
1016
1017
#if CONFIG_LOSSLESS
  {VP9E_SET_LOSSLESS,                 set_param},
#endif
John Koleszar's avatar
John Koleszar committed
1018
  { -1, NULL},
John Koleszar's avatar
John Koleszar committed
1019
1020
};

John Koleszar's avatar
John Koleszar committed
1021
1022
static vpx_codec_enc_cfg_map_t vp8e_usage_cfg_map[] = {
  {
John Koleszar's avatar
John Koleszar committed
1023
1024
    0,
    {
John Koleszar's avatar
John Koleszar committed
1025
1026
1027
      0,                  /* g_usage */
      0,                  /* g_threads */
      0,                  /* g_profile */
John Koleszar's avatar
John Koleszar committed
1028

John Koleszar's avatar
John Koleszar committed
1029
1030
1031
      320,                /* g_width */
      240,                /* g_height */
      {1, 30},            /* g_timebase */
John Koleszar's avatar
John Koleszar committed
1032

John Koleszar's avatar
John Koleszar committed
1033
      0,                  /* g_error_resilient */
1034
      0,                  /* g_frame_parallel_decoding */
John Koleszar's avatar
John Koleszar committed
1035

John Koleszar's avatar
John Koleszar committed
1036
      VPX_RC_ONE_PASS,    /* g_pass */
John Koleszar's avatar
John Koleszar committed
1037

John Koleszar's avatar
John Koleszar committed
1038
      0,                  /* g_lag_in_frames */
John Koleszar's avatar
John Koleszar committed
1039

John Koleszar's avatar
John Koleszar committed
1040
1041
1042
1043
      0,                  /* rc_dropframe_thresh */
      0,                  /* rc_resize_allowed */
      60,                 /* rc_resize_down_thresold */
      30,                 /* rc_resize_up_thresold */
John Koleszar's avatar
John Koleszar committed
1044

John Koleszar's avatar
John Koleszar committed
1045
      VPX_VBR,            /* rc_end_usage */
John Koleszar's avatar
John Koleszar committed
1046
#if VPX_ENCODER_ABI_VERSION > (1 + VPX_CODEC_ABI_VERSION)
John Koleszar's avatar
John Koleszar committed
1047
      {0},                /* rc_twopass_stats_in */
John Koleszar's avatar
John Koleszar committed
1048
#endif
John Koleszar's avatar
John Koleszar committed
1049
1050
1051
1052
1053
      256,                /* rc_target_bandwidth */
      4,                  /* rc_min_quantizer */
      63,                 /* rc_max_quantizer */
      100,                /* rc_undershoot_pct */
      100,                /* rc_overshoot_pct */
John Koleszar's avatar
John Koleszar committed
1054

John Koleszar's avatar
John Koleszar committed
1055
1056
1057
      6000,               /* rc_max_buffer_size */
      4000,               /* rc_buffer_initial_size; */
      5000,               /* rc_buffer_optimal_size; */
John Koleszar's avatar
John Koleszar committed
1058

John Koleszar's avatar
John Koleszar committed
1059
1060
1061
      50,                 /* rc_two_pass_vbrbias  */
      0,                  /* rc_two_pass_vbrmin_section */
      400,                /* rc_two_pass_vbrmax_section */
John Koleszar's avatar
John Koleszar committed
1062

John Koleszar's avatar
John Koleszar committed
1063
1064
1065
1066
      /* keyframing settings (kf) */
      VPX_KF_AUTO,        /* g_kfmode*/
      0,                  /* kf_min_dist */
      9999,               /* kf_max_dist */
John Koleszar's avatar
John Koleszar committed
1067
1068

#if VPX_ENCODER_ABI_VERSION == (1 + VPX_CODEC_ABI_VERSION)
John Koleszar's avatar
John Koleszar committed
1069
1070
      1,                  /* g_delete_first_pass_file */
      "vp8.fpf"           /* first pass filename */
John Koleszar's avatar
John Koleszar committed
1071
#endif
John Koleszar's avatar
John Koleszar committed
1072
1073
1074
    }
  },
  { -1, {NOT_IMPLEMENTED}}
John Koleszar's avatar
John Koleszar committed
1075
1076
1077
1078
1079
1080
};


#ifndef VERSION_STRING
#define VERSION_STRING
#endif
1081
1082
CODEC_INTERFACE(vpx_codec_vp9_cx) = {
  "WebM Project VP9 Encoder" VERSION_STRING,
John Koleszar's avatar
John Koleszar committed
1083
1084
1085
1086
1087
1088
1089
1090
1091
1092
1093
1094
1095
1096
1097
1098
1099
1100
1101
1102
1103
1104
1105
  VPX_CODEC_INTERNAL_ABI_VERSION,
  VPX_CODEC_CAP_ENCODER | VPX_CODEC_CAP_PSNR |
  VPX_CODEC_CAP_OUTPUT_PARTITION,
  /* vpx_codec_caps_t          caps; */
  vp8e_init,          /* vpx_codec_init_fn_t       init; */
  vp8e_destroy,       /* vpx_codec_destroy_fn_t    destroy; */
  vp8e_ctf_maps,      /* vpx_codec_ctrl_fn_map_t  *ctrl_maps; */
  NOT_IMPLEMENTED,    /* vpx_codec_get_mmap_fn_t   get_mmap; */
  NOT_IMPLEMENTED,    /* vpx_codec_set_mmap_fn_t   set_mmap; */
  {
    NOT_IMPLEMENTED,    /* vpx_codec_peek_si_fn_t    peek_si; */
    NOT_IMPLEMENTED,    /* vpx_codec_get_si_fn_t     get_si; */
    NOT_IMPLEMENTED,    /* vpx_codec_decode_fn_t     decode; */
    NOT_IMPLEMENTED,    /* vpx_codec_frame_get_fn_t  frame_get; */
  },
  {
    vp8e_usage_cfg_map, /* vpx_codec_enc_cfg_map_t    peek_si; */
    vp8e_encode,        /* vpx_codec_encode_fn_t      encode; */
    vp8e_get_cxdata,    /* vpx_codec_get_cx_data_fn_t   frame_get; */
    vp8e_set_config,
    NOT_IMPLEMENTED,
    vp8e_get_preview,
  } /* encoder functions */
John Koleszar's avatar
John Koleszar committed
1106
1107
1108
};


1109
#if CONFIG_EXPERIMENTAL
Jim Bankoski's avatar
Jim Bankoski committed
1110

1111
CODEC_INTERFACE(vpx_codec_vp9x_cx) = {
John Koleszar's avatar
John Koleszar committed
1112
1113
1114
1115
1116
1117
1118
1119
1120
1121
1122
1123
1124
1125
1126
1127
1128
1129
1130
1131
1132
1133
1134
  "VP8 Experimental Encoder" VERSION_STRING,
  VPX_CODEC_INTERNAL_ABI_VERSION,
  VPX_CODEC_CAP_ENCODER | VPX_CODEC_CAP_PSNR,
  /* vpx_codec_caps_t          caps; */
  vp8e_exp_init,      /* vpx_codec_init_fn_t       init; */
  vp8e_destroy,       /* vpx_codec_destroy_fn_t    destroy; */
  vp8e_ctf_maps,      /* vpx_codec_ctrl_fn_map_t  *ctrl_maps; */
  NOT_IMPLEMENTED,    /* vpx_codec_get_mmap_fn_t   get_mmap; */
  NOT_IMPLEMENTED,    /* vpx_codec_set_mmap_fn_t   set_mmap; */
  {
    NOT_IMPLEMENTED,    /* vpx_codec_peek_si_fn_t    peek_si; */
    NOT_IMPLEMENTED,    /* vpx_codec_get_si_fn_t     get_si; */
    NOT_IMPLEMENTED,    /* vpx_codec_decode_fn_t     decode; */
    NOT_IMPLEMENTED,    /* vpx_codec_frame_get_fn_t  frame_get; */
  },
  {
    vp8e_usage_cfg_map, /* vpx_codec_enc_cfg_map_t    peek_si; */
    vp8e_encode,        /* vpx_codec_encode_fn_t      encode; */
    vp8e_get_cxdata,    /* vpx_codec_get_cx_data_fn_t   frame_get; */
    vp8e_set_config,
    NOT_IMPLEMENTED,
    vp8e_get_preview,
  } /* encoder functions */
1135
1136
};
#endif