granicus.if.org Git - libvpx/blob - vp9/encoder/vp9_onyx_int.h

   1 /*
   2  *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
   3  *
   4  *  Use of this source code is governed by a BSD-style license
   5  *  that can be found in the LICENSE file in the root of the source
   6  *  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.
   9  */
  10
  11
  12 #ifndef VP9_ENCODER_VP9_ONYX_INT_H_
  13 #define VP9_ENCODER_VP9_ONYX_INT_H_
  14
  15 #include <stdio.h>
  16 #include "./vpx_config.h"
  17 #include "vp9/common/vp9_onyx.h"
  18 #include "vp9/encoder/vp9_treewriter.h"
  19 #include "vp9/encoder/vp9_tokenize.h"
  20 #include "vp9/common/vp9_onyxc_int.h"
  21 #include "vp9/encoder/vp9_variance.h"
  22 #include "vp9/encoder/vp9_encodemb.h"
  23 #include "vp9/encoder/vp9_quantize.h"
  24 #include "vp9/common/vp9_entropy.h"
  25 #include "vp9/common/vp9_entropymode.h"
  26 #include "vpx_ports/mem.h"
  27 #include "vpx/internal/vpx_codec_internal.h"
  28 #include "vp9/encoder/vp9_mcomp.h"
  29 #include "vp9/common/vp9_findnearmv.h"
  30 #include "vp9/encoder/vp9_lookahead.h"
  31
  32 // Experimental rate control switches
  33 #if CONFIG_ONESHOTQ
  34 #define ONE_SHOT_Q_ESTIMATE 0
  35 #define STRICT_ONE_SHOT_Q 0
  36 #define DISABLE_RC_LONG_TERM_MEM 0
  37 #endif
  38
  39 // #define SPEEDSTATS 1
  40 #if CONFIG_MULTIPLE_ARF
  41 // Set MIN_GF_INTERVAL to 1 for the full decomposition.
  42 #define MIN_GF_INTERVAL             2
  43 #else
  44 #define MIN_GF_INTERVAL             4
  45 #endif
  46 #define DEFAULT_GF_INTERVAL         7
  47
  48 #define KEY_FRAME_CONTEXT 5
  49
  50 #define MAX_MODES 36
  51
  52 #define MIN_THRESHMULT  32
  53 #define MAX_THRESHMULT  512
  54
  55 #define GF_ZEROMV_ZBIN_BOOST 0
  56 #define LF_ZEROMV_ZBIN_BOOST 0
  57 #define MV_ZBIN_BOOST        0
  58 #define SPLIT_MV_ZBIN_BOOST  0
  59 #define INTRA_ZBIN_BOOST     0
  60
  61 typedef struct {
  62   nmv_context nmvc;
  63   int nmvjointcost[MV_JOINTS];
  64   int nmvcosts[2][MV_VALS];
  65   int nmvcosts_hp[2][MV_VALS];
  66
  67   vp9_prob segment_pred_probs[PREDICTION_PROBS];
  68   vp9_prob intra_inter_prob[INTRA_INTER_CONTEXTS];
  69   vp9_prob comp_inter_prob[COMP_INTER_CONTEXTS];
  70   vp9_prob single_ref_prob[REF_CONTEXTS][2];
  71   vp9_prob comp_ref_prob[REF_CONTEXTS];
  72
  73   unsigned char *last_frame_seg_map_copy;
  74
  75   // 0 = Intra, Last, GF, ARF
  76   signed char last_ref_lf_deltas[MAX_REF_LF_DELTAS];
  77   // 0 = ZERO_MV, MV
  78   signed char last_mode_lf_deltas[MAX_MODE_LF_DELTAS];
  79
  80   vp9_coeff_probs_model coef_probs[TX_SIZE_MAX_SB][BLOCK_TYPES];
  81
  82   vp9_prob y_mode_prob[4][VP9_INTRA_MODES - 1];
  83   vp9_prob uv_mode_prob[VP9_INTRA_MODES][VP9_INTRA_MODES - 1];
  84   vp9_prob partition_prob[2][NUM_PARTITION_CONTEXTS][PARTITION_TYPES - 1];
  85
  86   vp9_prob switchable_interp_prob[VP9_SWITCHABLE_FILTERS + 1]
  87                                  [VP9_SWITCHABLE_FILTERS - 1];
  88
  89   int inter_mode_counts[INTER_MODE_CONTEXTS][VP9_INTER_MODES - 1][2];
  90   vp9_prob inter_mode_probs[INTER_MODE_CONTEXTS][VP9_INTER_MODES - 1];
  91
  92   vp9_prob tx_probs_8x8p[TX_SIZE_CONTEXTS][TX_SIZE_MAX_SB - 3];
  93   vp9_prob tx_probs_16x16p[TX_SIZE_CONTEXTS][TX_SIZE_MAX_SB - 2];
  94   vp9_prob tx_probs_32x32p[TX_SIZE_CONTEXTS][TX_SIZE_MAX_SB - 1];
  95   vp9_prob mbskip_probs[MBSKIP_CONTEXTS];
  96 } CODING_CONTEXT;
  97
  98 typedef struct {
  99   double frame;
 100   double intra_error;
 101   double coded_error;
 102   double sr_coded_error;
 103   double ssim_weighted_pred_err;
 104   double pcnt_inter;
 105   double pcnt_motion;
 106   double pcnt_second_ref;
 107   double pcnt_neutral;
 108   double MVr;
 109   double mvr_abs;
 110   double MVc;
 111   double mvc_abs;
 112   double MVrv;
 113   double MVcv;
 114   double mv_in_out_count;
 115   double new_mv_count;
 116   double duration;
 117   double count;
 118 } FIRSTPASS_STATS;
 119
 120 typedef struct {
 121   int frames_so_far;
 122   double frame_intra_error;
 123   double frame_coded_error;
 124   double frame_pcnt_inter;
 125   double frame_pcnt_motion;
 126   double frame_mvr;
 127   double frame_mvr_abs;
 128   double frame_mvc;
 129   double frame_mvc_abs;
 130 } ONEPASS_FRAMESTATS;
 131
 132 typedef struct {
 133   struct {
 134     int err;
 135     union {
 136       int_mv mv;
 137       MB_PREDICTION_MODE mode;
 138     } m;
 139   } ref[MAX_REF_FRAMES];
 140 } MBGRAPH_MB_STATS;
 141
 142 typedef struct {
 143   MBGRAPH_MB_STATS *mb_stats;
 144 } MBGRAPH_FRAME_STATS;
 145
 146 typedef enum {
 147   THR_ZEROMV,
 148   THR_DC,
 149
 150   THR_NEARESTMV,
 151   THR_NEARMV,
 152
 153   THR_ZEROG,
 154   THR_NEARESTG,
 155
 156   THR_ZEROA,
 157   THR_NEARESTA,
 158
 159   THR_NEARG,
 160   THR_NEARA,
 161
 162   THR_V_PRED,
 163   THR_H_PRED,
 164   THR_D45_PRED,
 165   THR_D135_PRED,
 166   THR_D117_PRED,
 167   THR_D153_PRED,
 168   THR_D27_PRED,
 169   THR_D63_PRED,
 170   THR_TM,
 171
 172   THR_NEWMV,
 173   THR_NEWG,
 174   THR_NEWA,
 175
 176   THR_SPLITMV,
 177   THR_SPLITG,
 178   THR_SPLITA,
 179
 180   THR_B_PRED,
 181
 182   THR_COMP_ZEROLA,
 183   THR_COMP_NEARESTLA,
 184   THR_COMP_NEARLA,
 185
 186   THR_COMP_ZEROGA,
 187   THR_COMP_NEARESTGA,
 188   THR_COMP_NEARGA,
 189
 190   THR_COMP_NEWLA,
 191   THR_COMP_NEWGA,
 192
 193   THR_COMP_SPLITLA,
 194   THR_COMP_SPLITGA,
 195 } THR_MODES;
 196
 197 typedef enum {
 198   DIAMOND = 0,
 199   NSTEP = 1,
 200   HEX = 2
 201 } SEARCH_METHODS;
 202
 203 typedef enum {
 204   USE_FULL_RD = 0,
 205   USE_LARGESTINTRA,
 206   USE_LARGESTINTRA_MODELINTER,
 207   USE_LARGESTALL
 208 } TX_SIZE_SEARCH_METHOD;
 209
 210 typedef struct {
 211   int RD;
 212   SEARCH_METHODS search_method;
 213   int auto_filter;
 214   int recode_loop;
 215   int iterative_sub_pixel;
 216   int half_pixel_search;
 217   int quarter_pixel_search;
 218   int thresh_mult[MAX_MODES];
 219   int max_step_search_steps;
 220   int reduce_first_step_size;
 221   int auto_mv_step_size;
 222   int optimize_coefficients;
 223   int search_best_filter;
 224   int static_segmentation;
 225   int comp_inter_joint_search_thresh;
 226   int adaptive_rd_thresh;
 227   int skip_encode_sb;
 228   int use_lastframe_partitioning;
 229   TX_SIZE_SEARCH_METHOD tx_size_search_method;
 230   int use_8tap_always;
 231   int use_avoid_tested_higherror;
 232   int skip_lots_of_modes;
 233   int adjust_thresholds_by_speed;
 234   int partition_by_variance;
 235   int use_one_partition_size_always;
 236   int less_rectangular_check;
 237   int use_square_partition_only;
 238   int unused_mode_skip_lvl;
 239   int reference_masking;
 240   BLOCK_SIZE_TYPE always_this_block_size;
 241   int use_partitions_greater_than;
 242   BLOCK_SIZE_TYPE greater_than_block_size;
 243   int use_partitions_less_than;
 244   BLOCK_SIZE_TYPE less_than_block_size;
 245   int adjust_partitioning_from_last_frame;
 246   int last_partitioning_redo_frequency;
 247   int disable_splitmv;
 248   // Search the D27, D63, D117 and D153 modes
 249   // only if the best intra mode so far is one
 250   // of the two directional modes nearest to each.
 251   int conditional_oblique_intramodes;
 252 } SPEED_FEATURES;
 253
 254 enum BlockSize {
 255   BLOCK_4X4,
 256   BLOCK_4X8,
 257   BLOCK_8X4,
 258   BLOCK_8X8,
 259   BLOCK_8X16,
 260   BLOCK_16X8,
 261   BLOCK_16X16,
 262   BLOCK_32X32,
 263   BLOCK_32X16,
 264   BLOCK_16X32,
 265   BLOCK_64X32,
 266   BLOCK_32X64,
 267   BLOCK_64X64,
 268   BLOCK_MAX_SB_SEGMENTS,
 269 };
 270
 271 typedef struct VP9_COMP {
 272
 273   DECLARE_ALIGNED(16, int16_t, y_quant[QINDEX_RANGE][8]);
 274   DECLARE_ALIGNED(16, int16_t, y_quant_shift[QINDEX_RANGE][8]);
 275   DECLARE_ALIGNED(16, int16_t, y_zbin[QINDEX_RANGE][8]);
 276   DECLARE_ALIGNED(16, int16_t, y_round[QINDEX_RANGE][8]);
 277
 278   DECLARE_ALIGNED(16, int16_t, uv_quant[QINDEX_RANGE][8]);
 279   DECLARE_ALIGNED(16, int16_t, uv_quant_shift[QINDEX_RANGE][8]);
 280   DECLARE_ALIGNED(16, int16_t, uv_zbin[QINDEX_RANGE][8]);
 281   DECLARE_ALIGNED(16, int16_t, uv_round[QINDEX_RANGE][8]);
 282
 283 #if CONFIG_ALPHA
 284   DECLARE_ALIGNED(16, int16_t, a_quant[QINDEX_RANGE][8]);
 285   DECLARE_ALIGNED(16, int16_t, a_quant_shift[QINDEX_RANGE][8]);
 286   DECLARE_ALIGNED(16, int16_t, a_zbin[QINDEX_RANGE][8]);
 287   DECLARE_ALIGNED(16, int16_t, a_round[QINDEX_RANGE][8]);
 288 #endif
 289
 290   MACROBLOCK mb;
 291   VP9_COMMON common;
 292   VP9_CONFIG oxcf;
 293
 294   struct lookahead_ctx    *lookahead;
 295   struct lookahead_entry  *source;
 296 #if CONFIG_MULTIPLE_ARF
 297   struct lookahead_entry  *alt_ref_source[NUM_REF_FRAMES];
 298 #else
 299   struct lookahead_entry  *alt_ref_source;
 300 #endif
 301
 302   YV12_BUFFER_CONFIG *Source;
 303   YV12_BUFFER_CONFIG *un_scaled_source;
 304   YV12_BUFFER_CONFIG scaled_source;
 305
 306   int source_alt_ref_pending; // frame in src_buffers has been identified to be encoded as an alt ref
 307   int source_alt_ref_active;  // an alt ref frame has been encoded and is usable
 308
 309   int is_src_frame_alt_ref;   // source of frame to encode is an exact copy of an alt ref frame
 310
 311   int gold_is_last; // golden frame same as last frame ( short circuit gold searches)
 312   int alt_is_last;  // Alt reference frame same as last ( short circuit altref search)
 313   int gold_is_alt;  // don't do both alt and gold search ( just do gold).
 314
 315   int scaled_ref_idx[3];
 316   int lst_fb_idx;
 317   int gld_fb_idx;
 318   int alt_fb_idx;
 319 #if CONFIG_MULTIPLE_ARF
 320   int alt_ref_fb_idx[NUM_REF_FRAMES - 3];
 321 #endif
 322   int refresh_last_frame;
 323   int refresh_golden_frame;
 324   int refresh_alt_ref_frame;
 325   YV12_BUFFER_CONFIG last_frame_uf;
 326
 327   TOKENEXTRA *tok;
 328   unsigned int tok_count[4][1 << 6];
 329
 330
 331   unsigned int frames_since_key;
 332   unsigned int key_frame_frequency;
 333   unsigned int this_key_frame_forced;
 334   unsigned int next_key_frame_forced;
 335 #if CONFIG_MULTIPLE_ARF
 336   // Position within a frame coding order (including any additional ARF frames).
 337   unsigned int sequence_number;
 338   // Next frame in naturally occurring order that has not yet been coded.
 339   int next_frame_in_order;
 340 #endif
 341
 342   // Ambient reconstruction err target for force key frames
 343   int ambient_err;
 344
 345   unsigned int mode_check_freq[MAX_MODES];
 346   unsigned int mode_test_hit_counts[MAX_MODES];
 347   unsigned int mode_chosen_counts[MAX_MODES];
 348   int64_t unused_mode_skip_mask;
 349   int ref_frame_mask;
 350   int set_ref_frame_mask;
 351
 352   int rd_thresh_mult[MAX_MODES];
 353   int rd_baseline_thresh[BLOCK_SIZE_TYPES][MAX_MODES];
 354   int rd_threshes[BLOCK_SIZE_TYPES][MAX_MODES];
 355   int rd_thresh_freq_fact[BLOCK_SIZE_TYPES][MAX_MODES];
 356
 357   int64_t rd_comp_pred_diff[NB_PREDICTION_TYPES];
 358   int rd_prediction_type_threshes[4][NB_PREDICTION_TYPES];
 359   unsigned int intra_inter_count[INTRA_INTER_CONTEXTS][2];
 360   unsigned int comp_inter_count[COMP_INTER_CONTEXTS][2];
 361   unsigned int single_ref_count[REF_CONTEXTS][2][2];
 362   unsigned int comp_ref_count[REF_CONTEXTS][2];
 363
 364   // FIXME contextualize
 365
 366   int64_t rd_tx_select_diff[NB_TXFM_MODES];
 367   int rd_tx_select_threshes[4][NB_TXFM_MODES];
 368
 369   int RDMULT;
 370   int RDDIV;
 371
 372   CODING_CONTEXT coding_context;
 373
 374   // Rate targetting variables
 375   int this_frame_target;
 376   int projected_frame_size;
 377   int last_q[2];                   // Separate values for Intra/Inter
 378   int last_boosted_qindex;         // Last boosted GF/KF/ARF q
 379
 380   double rate_correction_factor;
 381   double key_frame_rate_correction_factor;
 382   double gf_rate_correction_factor;
 383
 384   int frames_till_gf_update_due;      // Count down till next GF
 385
 386   int gf_overspend_bits;            // Total bits overspent becasue of GF boost (cumulative)
 387
 388   int non_gf_bitrate_adjustment;     // Used in the few frames following a GF to recover the extra bits spent in that GF
 389
 390   int kf_overspend_bits;            // Extra bits spent on key frames that need to be recovered on inter frames
 391   int kf_bitrate_adjustment;        // Current number of bit s to try and recover on each inter frame.
 392   int max_gf_interval;
 393   int baseline_gf_interval;
 394   int active_arnr_frames;           // <= cpi->oxcf.arnr_max_frames
 395   int active_arnr_strength;         // <= cpi->oxcf.arnr_max_strength
 396
 397   int64_t key_frame_count;
 398   int prior_key_frame_distance[KEY_FRAME_CONTEXT];
 399   int per_frame_bandwidth;          // Current section per frame bandwidth target
 400   int av_per_frame_bandwidth;        // Average frame size target for clip
 401   int min_frame_bandwidth;          // Minimum allocation that should be used for any frame
 402   int inter_frame_target;
 403   double output_frame_rate;
 404   int64_t last_time_stamp_seen;
 405   int64_t last_end_time_stamp_seen;
 406   int64_t first_time_stamp_ever;
 407
 408   int ni_av_qi;
 409   int ni_tot_qi;
 410   int ni_frames;
 411   int avg_frame_qindex;
 412   double tot_q;
 413   double avg_q;
 414
 415   int zbin_mode_boost;
 416   int zbin_mode_boost_enabled;
 417
 418   int64_t total_byte_count;
 419
 420   int buffered_mode;
 421
 422   int buffer_level;
 423   int bits_off_target;
 424
 425   int rolling_target_bits;
 426   int rolling_actual_bits;
 427
 428   int long_rolling_target_bits;
 429   int long_rolling_actual_bits;
 430
 431   int64_t total_actual_bits;
 432   int total_target_vs_actual;        // debug stats
 433
 434   int worst_quality;
 435   int active_worst_quality;
 436   int best_quality;
 437   int active_best_quality;
 438
 439   int cq_target_quality;
 440
 441   int y_mode_count[4][VP9_INTRA_MODES];
 442   int y_uv_mode_count[VP9_INTRA_MODES][VP9_INTRA_MODES];
 443   unsigned int partition_count[NUM_PARTITION_CONTEXTS][PARTITION_TYPES];
 444
 445   nmv_context_counts NMVcount;
 446
 447   vp9_coeff_count coef_counts[TX_SIZE_MAX_SB][BLOCK_TYPES];
 448   vp9_coeff_probs_model frame_coef_probs[TX_SIZE_MAX_SB][BLOCK_TYPES];
 449   vp9_coeff_stats frame_branch_ct[TX_SIZE_MAX_SB][BLOCK_TYPES];
 450
 451   int gfu_boost;
 452   int last_boost;
 453   int kf_boost;
 454   int kf_zeromotion_pct;
 455
 456   int64_t target_bandwidth;
 457   struct vpx_codec_pkt_list  *output_pkt_list;
 458
 459 #if 0
 460   // Experimental code for lagged and one pass
 461   ONEPASS_FRAMESTATS one_pass_frame_stats[MAX_LAG_BUFFERS];
 462   int one_pass_frame_index;
 463 #endif
 464   MBGRAPH_FRAME_STATS mbgraph_stats[MAX_LAG_BUFFERS];
 465   int mbgraph_n_frames;             // number of frames filled in the above
 466   int static_mb_pct;                // % forced skip mbs by segmentation
 467   int seg0_progress, seg0_idx, seg0_cnt;
 468
 469   int decimation_factor;
 470   int decimation_count;
 471
 472   // for real time encoding
 473   int avg_encode_time;              // microsecond
 474   int avg_pick_mode_time;            // microsecond
 475   int speed;
 476   unsigned int cpu_freq;           // Mhz
 477   int compressor_speed;
 478
 479   int interquantizer;
 480   int goldfreq;
 481   int auto_worst_q;
 482   int cpu_used;
 483   int pass;
 484
 485   vp9_prob last_skip_false_probs[3][MBSKIP_CONTEXTS];
 486   int last_skip_probs_q[3];
 487
 488   int ref_frame_flags;
 489
 490   SPEED_FEATURES sf;
 491   int error_bins[1024];
 492
 493   unsigned int max_mv_magnitude;
 494
 495   // Data used for real time conferencing mode to help determine if it would be good to update the gf
 496   int inter_zz_count;
 497   int gf_bad_count;
 498   int gf_update_recommended;
 499
 500   unsigned char *segmentation_map;
 501
 502   // segment threashold for encode breakout
 503   int  segment_encode_breakout[MAX_MB_SEGMENTS];
 504
 505   unsigned char *active_map;
 506   unsigned int active_map_enabled;
 507
 508   fractional_mv_step_fp *find_fractional_mv_step;
 509   vp9_full_search_fn_t full_search_sad;
 510   vp9_refining_search_fn_t refining_search_sad;
 511   vp9_diamond_search_fn_t diamond_search_sad;
 512   vp9_variance_fn_ptr_t fn_ptr[BLOCK_MAX_SB_SEGMENTS];
 513   uint64_t time_receive_data;
 514   uint64_t time_compress_data;
 515   uint64_t time_pick_lpf;
 516   uint64_t time_encode_mb_row;
 517
 518   struct twopass_rc {
 519     unsigned int section_intra_rating;
 520     unsigned int next_iiratio;
 521     unsigned int this_iiratio;
 522     FIRSTPASS_STATS total_stats;
 523     FIRSTPASS_STATS this_frame_stats;
 524     FIRSTPASS_STATS *stats_in, *stats_in_end, *stats_in_start;
 525     FIRSTPASS_STATS total_left_stats;
 526     int first_pass_done;
 527     int64_t bits_left;
 528     int64_t clip_bits_total;
 529     double avg_iiratio;
 530     double modified_error_total;
 531     double modified_error_used;
 532     double modified_error_left;
 533     double kf_intra_err_min;
 534     double gf_intra_err_min;
 535     int frames_to_key;
 536     int maxq_max_limit;
 537     int maxq_min_limit;
 538     int static_scene_max_gf_interval;
 539     int kf_bits;
 540     // Remaining error from uncoded frames in a gf group. Two pass use only
 541     int64_t gf_group_error_left;
 542
 543     // Projected total bits available for a key frame group of frames
 544     int64_t kf_group_bits;
 545
 546     // Error score of frames still to be coded in kf group
 547     int64_t kf_group_error_left;
 548
 549     // Projected Bits available for a group of frames including 1 GF or ARF
 550     int64_t gf_group_bits;
 551     // Bits for the golden frame or ARF - 2 pass only
 552     int gf_bits;
 553     int alt_extra_bits;
 554
 555     int sr_update_lag;
 556     double est_max_qcorrection_factor;
 557   } twopass;
 558
 559   YV12_BUFFER_CONFIG alt_ref_buffer;
 560   YV12_BUFFER_CONFIG *frames[MAX_LAG_BUFFERS];
 561   int fixed_divide[512];
 562
 563 #if CONFIG_INTERNAL_STATS
 564   int    count;
 565   double total_y;
 566   double total_u;
 567   double total_v;
 568   double total;
 569   double total_sq_error;
 570   double totalp_y;
 571   double totalp_u;
 572   double totalp_v;
 573   double totalp;
 574   double total_sq_error2;
 575   int    bytes;
 576   double summed_quality;
 577   double summed_weights;
 578   double summedp_quality;
 579   double summedp_weights;
 580   unsigned int tot_recode_hits;
 581
 582
 583   double total_ssimg_y;
 584   double total_ssimg_u;
 585   double total_ssimg_v;
 586   double total_ssimg_all;
 587
 588   int b_calculate_ssimg;
 589 #endif
 590   int b_calculate_psnr;
 591
 592   // Per MB activity measurement
 593   unsigned int activity_avg;
 594   unsigned int *mb_activity_map;
 595   int *mb_norm_activity_map;
 596   int output_partition;
 597
 598   /* force next frame to intra when kf_auto says so */
 599   int force_next_frame_intra;
 600
 601   int droppable;
 602
 603   int dummy_packing;    /* flag to indicate if packing is dummy */
 604
 605   unsigned int switchable_interp_count[VP9_SWITCHABLE_FILTERS + 1]
 606                                       [VP9_SWITCHABLE_FILTERS];
 607   unsigned int best_switchable_interp_count[VP9_SWITCHABLE_FILTERS];
 608
 609   unsigned int txfm_stepdown_count[TX_SIZE_MAX_SB];
 610
 611   int initial_width;
 612   int initial_height;
 613
 614 #if CONFIG_MULTIPLE_ARF
 615   // ARF tracking variables.
 616   int multi_arf_enabled;
 617   unsigned int frame_coding_order_period;
 618   unsigned int new_frame_coding_order_period;
 619   int frame_coding_order[MAX_LAG_BUFFERS * 2];
 620   int arf_buffer_idx[MAX_LAG_BUFFERS * 3 / 2];
 621   int arf_weight[MAX_LAG_BUFFERS];
 622   int arf_buffered;
 623   int this_frame_weight;
 624   int max_arf_level;
 625 #endif
 626
 627 #ifdef ENTROPY_STATS
 628   int64_t mv_ref_stats[INTER_MODE_CONTEXTS][VP9_INTER_MODES - 1][2];
 629 #endif
 630 } VP9_COMP;
 631
 632 static int get_ref_frame_idx(VP9_COMP *cpi, MV_REFERENCE_FRAME ref_frame) {
 633   if (ref_frame == LAST_FRAME) {
 634     return cpi->lst_fb_idx;
 635   } else if (ref_frame == GOLDEN_FRAME) {
 636     return cpi->gld_fb_idx;
 637   } else {
 638     return cpi->alt_fb_idx;
 639   }
 640 }
 641
 642 void vp9_encode_frame(VP9_COMP *cpi);
 643
 644 void vp9_pack_bitstream(VP9_COMP *cpi, unsigned char *dest,
 645                         unsigned long *size);
 646
 647 void vp9_activity_masking(VP9_COMP *cpi, MACROBLOCK *x);
 648
 649 void vp9_set_speed_features(VP9_COMP *cpi);
 650
 651 extern int vp9_calc_ss_err(YV12_BUFFER_CONFIG *source,
 652                            YV12_BUFFER_CONFIG *dest);
 653
 654 extern void vp9_alloc_compressor_data(VP9_COMP *cpi);
 655
 656 #endif  // VP9_ENCODER_VP9_ONYX_INT_H_