vp9_cx_iface.c 39.4 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
    case VPX_IMG_FMT_YV12:
    case VPX_IMG_FMT_I420:
214
215
    case VPX_IMG_FMT_I422:
    case VPX_IMG_FMT_I444:
John Koleszar's avatar
John Koleszar committed
216
      break;
John Koleszar's avatar
John Koleszar committed
217
    default:
218
219
      ERROR("Invalid image format. Only YV12, I420, I422, I444 images are "
            "supported.");
John Koleszar's avatar
John Koleszar committed
220
  }
John Koleszar's avatar
John Koleszar committed
221

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

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


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

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

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

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

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

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

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

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

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

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

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

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


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


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

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

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

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

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

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

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

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

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

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

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


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

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

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

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

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

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

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

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

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

490
491
492
493
    // 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
494

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

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

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

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

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

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

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

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

523

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


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


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

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

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

  yv12->uv_width = img->x_chroma_shift == 1 ? (1 + yv12->y_width) / 2
                                            : yv12->y_width;
  yv12->uv_height = img->y_chroma_shift == 1 ? (1 + yv12->y_height) / 2
                                             : yv12->y_height;
John Koleszar's avatar
John Koleszar committed
562
563
564
565
566

  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;
567
  yv12->clrtype = REG_YUV;
John Koleszar's avatar
John Koleszar committed
568
  return res;
John Koleszar's avatar
John Koleszar committed
569
570
571
572
}

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


596
static int write_superframe_index(vpx_codec_alg_priv_t *ctx) {
John Koleszar's avatar
John Koleszar committed
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
629
630
631
632
  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;
  }
633
  return index_sz;
John Koleszar's avatar
John Koleszar committed
634
635
}

John Koleszar's avatar
John Koleszar committed
636
637
638
639
640
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
641
642
                                   unsigned long          deadline) {
  vpx_codec_err_t res = VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
643

John Koleszar's avatar
John Koleszar committed
644
645
  if (img)
    res = validate_img(ctx, img);
John Koleszar's avatar
John Koleszar committed
646

John Koleszar's avatar
John Koleszar committed
647
648
  pick_quickcompress_mode(ctx, duration, deadline);
  vpx_codec_pkt_list_init(&ctx->pkt_list);
John Koleszar's avatar
John Koleszar committed
649

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

John Koleszar's avatar
John Koleszar committed
657
658
659
  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
660

John Koleszar's avatar
John Koleszar committed
661
    if (flags & VP8_EFLAG_NO_REF_LAST)
662
      ref ^= VP9_LAST_FLAG;
John Koleszar's avatar
John Koleszar committed
663

John Koleszar's avatar
John Koleszar committed
664
    if (flags & VP8_EFLAG_NO_REF_GF)
665
      ref ^= VP9_GOLD_FLAG;
John Koleszar's avatar
John Koleszar committed
666

John Koleszar's avatar
John Koleszar committed
667
    if (flags & VP8_EFLAG_NO_REF_ARF)
668
      ref ^= VP9_ALT_FLAG;
John Koleszar's avatar
John Koleszar committed
669

670
    vp9_use_as_reference(ctx->cpi, ref);
John Koleszar's avatar
John Koleszar committed
671
  }
John Koleszar's avatar
John Koleszar committed
672

John Koleszar's avatar
John Koleszar committed
673
674
675
676
  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
677

John Koleszar's avatar
John Koleszar committed
678
    if (flags & VP8_EFLAG_NO_UPD_LAST)
679
      upd ^= VP9_LAST_FLAG;
John Koleszar's avatar
John Koleszar committed
680

John Koleszar's avatar
John Koleszar committed
681
    if (flags & VP8_EFLAG_NO_UPD_GF)
682
      upd ^= VP9_GOLD_FLAG;
John Koleszar's avatar
John Koleszar committed
683

John Koleszar's avatar
John Koleszar committed
684
    if (flags & VP8_EFLAG_NO_UPD_ARF)
685
      upd ^= VP9_ALT_FLAG;
John Koleszar's avatar
John Koleszar committed
686

687
    vp9_update_reference(ctx->cpi, upd);
John Koleszar's avatar
John Koleszar committed
688
  }
John Koleszar's avatar
John Koleszar committed
689

John Koleszar's avatar
John Koleszar committed
690
  if (flags & VP8_EFLAG_NO_UPD_ENTROPY) {
691
    vp9_update_entropy(ctx->cpi, 0);
John Koleszar's avatar
John Koleszar committed
692
  }
John Koleszar's avatar
John Koleszar committed
693

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

John Koleszar's avatar
John Koleszar committed
703
704
705
706
707
708
709
  /* 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
710

John Koleszar's avatar
John Koleszar committed
711
712
    /* Set up internal flags */
    if (ctx->base.init_flags & VPX_CODEC_USE_PSNR)
713
      ((VP9_COMP *)ctx->cpi)->b_calculate_psnr = 1;
John Koleszar's avatar
John Koleszar committed
714

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

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

John Koleszar's avatar
John Koleszar committed
721
722
723
    /* 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
724

John Koleszar's avatar
John Koleszar committed
725
726
    if (img != NULL) {
      res = image2yuvconfig(img, &sd);
John Koleszar's avatar
John Koleszar committed
727

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

735
736
    cx_data = ctx->cx_data;
    cx_data_sz = ctx->cx_data_sz;
John Koleszar's avatar
John Koleszar committed
737
738
    lib_flags = 0;

739
740
741
742
743
744
745
746
747
748
749
750
751
752
    /* 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;
      }
753
754
    }

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

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

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

John Koleszar's avatar
John Koleszar committed
804
805
        if (cpi->droppable)
          pkt.data.frame.flags |= VPX_FRAME_IS_DROPPABLE;
John Koleszar's avatar
John Koleszar committed
806

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

John Koleszar's avatar
John Koleszar committed
855
  return res;
John Koleszar's avatar
John Koleszar committed
856
857
858
859
}


static const vpx_codec_cx_pkt_t *vp8e_get_cxdata(vpx_codec_alg_priv_t  *ctx,
John Koleszar's avatar
John Koleszar committed
860
861
                                                 vpx_codec_iter_t      *iter) {
  return vpx_codec_pkt_list_get(&ctx->pkt_list.head, iter);
John Koleszar's avatar
John Koleszar committed
862
863
864
}

static vpx_codec_err_t vp8e_set_reference(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
865
866
867
                                          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
868

John Koleszar's avatar
John Koleszar committed
869
870
871
  if (data) {
    vpx_ref_frame_t *frame = (vpx_ref_frame_t *)data;
    YV12_BUFFER_CONFIG sd;
John Koleszar's avatar
John Koleszar committed
872

John Koleszar's avatar
John Koleszar committed
873
    image2yuvconfig(&frame->img, &sd);
874
    vp9_set_reference_enc(ctx->cpi, frame->frame_type, &sd);
John Koleszar's avatar
John Koleszar committed
875
876
877
    return VPX_CODEC_OK;
  } else
    return VPX_CODEC_INVALID_PARAM;
John Koleszar's avatar
John Koleszar committed
878
879
880

}

John Koleszar's avatar
John Koleszar committed
881
882
883
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
884

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

John Koleszar's avatar
John Koleszar committed
887
888
889
  if (data) {
    vpx_ref_frame_t *frame = (vpx_ref_frame_t *)data;
    YV12_BUFFER_CONFIG sd;
John Koleszar's avatar
John Koleszar committed
890

John Koleszar's avatar
John Koleszar committed
891
    image2yuvconfig(&frame->img, &sd);
John Koleszar's avatar
John Koleszar committed
892
    vp9_copy_reference_enc(ctx->cpi, frame->frame_type, &sd);
John Koleszar's avatar
John Koleszar committed
893
894
895
    return VPX_CODEC_OK;
  } else
    return VPX_CODEC_INVALID_PARAM;
John Koleszar's avatar
John Koleszar committed
896
897
}

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

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


John Koleszar's avatar
John Koleszar committed
935
static vpx_image_t *vp8e_get_preview(vpx_codec_alg_priv_t *ctx) {
John Koleszar's avatar
John Koleszar committed
936

John Koleszar's avatar
John Koleszar committed
937
  YV12_BUFFER_CONFIG sd;
938
  vp9_ppflags_t flags = {0};
John Koleszar's avatar
John Koleszar committed
939

John Koleszar's avatar
John Koleszar committed
940
941
942
943
944
  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;
  }
945

946
  if (0 == vp9_get_preview_raw_frame(ctx->cpi, &sd, &flags)) {
947
    yuvconfig2image(&ctx->preview_img, &sd, NULL);
John Koleszar's avatar
John Koleszar committed
948
949
950
    return &ctx->preview_img;
  } else
    return NULL;
John Koleszar's avatar
John Koleszar committed
951
952
953
}

static vpx_codec_err_t vp8e_update_entropy(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
954
955
956
                                           int ctr_id,
                                           va_list args) {
  int update = va_arg(args, int);
957
  vp9_update_entropy(ctx->cpi, update);
John Koleszar's avatar
John Koleszar committed
958
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
959
960
961
962

}

static vpx_codec_err_t vp8e_update_reference(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
963
964
965
                                             int ctr_id,
                                             va_list args) {
  int update = va_arg(args, int);
966
  vp9_update_reference(ctx->cpi, update);
John Koleszar's avatar
John Koleszar committed
967
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
968
969
970
}

static vpx_codec_err_t vp8e_use_reference(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
971
972
973
                                          int ctr_id,
                                          va_list args) {
  int reference_flag = va_arg(args, int);
974
  vp9_use_as_reference(ctx->cpi, reference_flag);
John Koleszar's avatar
John Koleszar committed
975
  return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
976
977
978
979
}

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

John Koleszar's avatar
John Koleszar committed
983
984
  if (data) {
    vpx_roi_map_t *roi = (vpx_roi_map_t *)data;
John Koleszar's avatar
John Koleszar committed
985

986
987
    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
988
      return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
989
    else
John Koleszar's avatar
John Koleszar committed
990
991
992
      return VPX_CODEC_INVALID_PARAM;
  } else
    return VPX_CODEC_INVALID_PARAM;
John Koleszar's avatar
John Koleszar committed
993
994
995
996
}


static vpx_codec_err_t vp8e_set_activemap(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
997
998
999
                                          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
1000

John Koleszar's avatar
John Koleszar committed
1001
  if (data) {
John Koleszar's avatar
John Koleszar committed
1002

John Koleszar's avatar
John Koleszar committed
1003
    vpx_active_map_t *map = (vpx_active_map_t *)data;
John Koleszar's avatar
John Koleszar committed
1004

1005
    if (!vp9_set_active_map(ctx->cpi, map->active_map, map->rows, map->cols))
John Koleszar's avatar
John Koleszar committed
1006
      return VPX_CODEC_OK;
John Koleszar's avatar
John Koleszar committed
1007
    else
John Koleszar's avatar
John Koleszar committed
1008
1009
1010
      return VPX_CODEC_INVALID_PARAM;
  } else
    return VPX_CODEC_INVALID_PARAM;
John Koleszar's avatar
John Koleszar committed
1011
1012
1013
}

static vpx_codec_err_t vp8e_set_scalemode(vpx_codec_alg_priv_t *ctx,
John Koleszar's avatar
John Koleszar committed
1014
1015
1016
1017
1018
1019
1020
1021
                                          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;
1022
1023
    res = vp9_set_internal_size(ctx->cpi, scalemode.h_scaling_mode,
                                scalemode.v_scaling_mode);
John Koleszar's avatar
John Koleszar committed
1024
1025
1026
1027
1028
1029
1030

    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
1031
1032
1033
}


John Koleszar's avatar
John Koleszar committed
1034
1035
static vpx_codec_ctrl_fn_map_t vp8e_ctf_maps[] = {
  {VP8_SET_REFERENCE,                 vp8e_set_reference},
John Koleszar's avatar
John Koleszar committed
1036
  {VP8_COPY_REFERENCE,                vp8e_copy_reference},
John Koleszar's avatar
John Koleszar committed
1037
1038
1039
1040
1041
1042
1043
1044
1045
1046
1047
1048
  {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},
1049
  {VP9E_SET_TILE_COLUMNS,             set_param},
Ronald S. Bultje's avatar
Ronald S. Bultje committed
1050
  {VP9E_SET_TILE_ROWS,                set_param},
John Koleszar's avatar
John Koleszar committed
1051
1052
1053
1054
1055
1056
1057
1058
  {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
1059
  {VP9E_SET_LOSSLESS,                 set_param},
John Koleszar's avatar
John Koleszar committed
1060
  {VP9_GET_REFERENCE,                 get_reference},
John Koleszar's avatar
John Koleszar committed
1061
  { -1, NULL},
John Koleszar's avatar
John Koleszar committed
1062
1063
};

John Koleszar's avatar
John Koleszar committed
1064
1065
static vpx_codec_enc_cfg_map_t vp8e_usage_cfg_map[] = {
  {
John Koleszar's avatar
John Koleszar committed
1066
1067
    0,
    {
John Koleszar's avatar
John Koleszar committed
1068
1069
1070
      0,                  /* g_usage */
      0,                  /* g_threads */
      0,                  /* g_profile */
John Koleszar's avatar
John Koleszar committed
1071

John Koleszar's avatar
John Koleszar committed
1072
1073
1074
      320,                /* g_width */
      240,                /* g_height */
      {1, 30},            /* g_timebase */
John Koleszar's avatar
John Koleszar committed
1075

John Koleszar's avatar
John Koleszar committed
1076
      0,                  /* g_error_resilient */
John Koleszar's avatar
John Koleszar committed
1077

John Koleszar's avatar
John Koleszar committed
1078
      VPX_RC_ONE_PASS,    /* g_pass */
John Koleszar's avatar
John Koleszar committed
1079

John Koleszar's avatar
John Koleszar committed
1080
      0,                  /* g_lag_in_frames */
John Koleszar's avatar
John Koleszar committed
1081

John Koleszar's avatar
John Koleszar committed
1082
1083
1084
1085
      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
1086

John Koleszar's avatar
John Koleszar committed
1087
      VPX_VBR,            /* rc_end_usage */
John Koleszar's avatar
John Koleszar committed
1088
#if VPX_ENCODER_ABI_VERSION > (1 + VPX_CODEC_ABI_VERSION)
John Koleszar's avatar
John Koleszar committed
1089
      {0},                /* rc_twopass_stats_in */
John Koleszar's avatar
John Koleszar committed
1090
#endif
John Koleszar's avatar
John Koleszar committed
1091
1092
1093
1094
1095
      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
1096

John Koleszar's avatar
John Koleszar committed
1097
1098
1099
      6000,               /* rc_max_buffer_size */
      4000,               /* rc_buffer_initial_size; */
      5000,               /* rc_buffer_optimal_size; */
John Koleszar's avatar
John Koleszar committed
1100

John Koleszar's avatar
John Koleszar committed
1101
1102