1e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org/* 2e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org * Copyright (c) 2010 The WebM project authors. All Rights Reserved. 3e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org * 4e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org * Use of this source code is governed by a BSD-style license 5e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org * that can be found in the LICENSE file in the root of the source 6e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org * tree. An additional intellectual property rights grant can be found 7e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org * in the file PATENTS. All contributing project authors may 8e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org * be found in the AUTHORS file in the root of the source tree. 9e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org */ 10e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 11e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include <assert.h> 12e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include <math.h> 13e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include <stdio.h> 14e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 15e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "./vp9_rtcd.h" 16e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 17e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "vpx_mem/vpx_mem.h" 18e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 19e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "vp9/common/vp9_common.h" 20e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "vp9/common/vp9_entropy.h" 21e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "vp9/common/vp9_entropymode.h" 22e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "vp9/common/vp9_mvref_common.h" 23e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "vp9/common/vp9_pred_common.h" 24e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "vp9/common/vp9_quant_common.h" 25e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "vp9/common/vp9_reconinter.h" 26e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "vp9/common/vp9_reconintra.h" 27e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "vp9/common/vp9_seg_common.h" 28e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "vp9/common/vp9_systemdependent.h" 29e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 30e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "vp9/encoder/vp9_cost.h" 31e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "vp9/encoder/vp9_encodemb.h" 32e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "vp9/encoder/vp9_encodemv.h" 33e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "vp9/encoder/vp9_encoder.h" 34e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "vp9/encoder/vp9_mcomp.h" 35e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "vp9/encoder/vp9_quantize.h" 36e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "vp9/encoder/vp9_ratectrl.h" 37e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "vp9/encoder/vp9_rd.h" 38e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "vp9/encoder/vp9_tokenize.h" 39e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#include "vp9/encoder/vp9_variance.h" 40e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 41e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#define RD_THRESH_POW 1.25 42e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#define RD_MULT_EPB_RATIO 64 43e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 44e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org// Factor to weigh the rate for switchable interp filters. 45e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org#define SWITCHABLE_INTERP_RATE_FACTOR 1 46e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 47e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org// The baseline rd thresholds for breaking out of the rd loop for 48e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org// certain modes are assumed to be based on 8x8 blocks. 49e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org// This table is used to correct for block size. 50e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org// The factors here are << 2 (2 = x0.5, 32 = x8 etc). 51e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.orgstatic const uint8_t rd_thresh_block_size_factor[BLOCK_SIZES] = { 52e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 2, 3, 3, 4, 6, 6, 8, 12, 12, 16, 24, 24, 32 53e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org}; 54e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 55e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.orgstatic void fill_mode_costs(VP9_COMP *cpi) { 56e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const FRAME_CONTEXT *const fc = &cpi->common.fc; 57e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int i, j; 58e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 59e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (i = 0; i < INTRA_MODES; ++i) 60e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (j = 0; j < INTRA_MODES; ++j) 61e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp9_cost_tokens(cpi->y_mode_costs[i][j], vp9_kf_y_mode_prob[i][j], 62e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp9_intra_mode_tree); 63e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 64e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp9_cost_tokens(cpi->mbmode_cost, fc->y_mode_prob[1], vp9_intra_mode_tree); 65e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp9_cost_tokens(cpi->intra_uv_mode_cost[KEY_FRAME], 66e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp9_kf_uv_mode_prob[TM_PRED], vp9_intra_mode_tree); 67e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp9_cost_tokens(cpi->intra_uv_mode_cost[INTER_FRAME], 68e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org fc->uv_mode_prob[TM_PRED], vp9_intra_mode_tree); 69e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 70e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (i = 0; i < SWITCHABLE_FILTER_CONTEXTS; ++i) 71e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp9_cost_tokens(cpi->switchable_interp_costs[i], 72e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org fc->switchable_interp_prob[i], vp9_switchable_interp_tree); 73e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org} 74e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 75e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.orgstatic void fill_token_costs(vp9_coeff_cost *c, 76e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp9_coeff_probs_model (*p)[PLANE_TYPES]) { 77e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int i, j, k, l; 78e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org TX_SIZE t; 79e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (t = TX_4X4; t <= TX_32X32; ++t) 80e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (i = 0; i < PLANE_TYPES; ++i) 81e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (j = 0; j < REF_TYPES; ++j) 82e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (k = 0; k < COEF_BANDS; ++k) 83e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (l = 0; l < BAND_COEFF_CONTEXTS(k); ++l) { 84e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp9_prob probs[ENTROPY_NODES]; 85e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp9_model_to_full_probs(p[t][i][j][k][l], probs); 86e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp9_cost_tokens((int *)c[t][i][j][k][0][l], probs, 87e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp9_coef_tree); 88e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp9_cost_tokens_skip((int *)c[t][i][j][k][1][l], probs, 89e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp9_coef_tree); 90e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org assert(c[t][i][j][k][0][l][EOB_TOKEN] == 91e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org c[t][i][j][k][1][l][EOB_TOKEN]); 92e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 93e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org} 94e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 95e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org// Values are now correlated to quantizer. 9687997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.orgstatic int sad_per_bit16lut_8[QINDEX_RANGE]; 9787997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.orgstatic int sad_per_bit4lut_8[QINDEX_RANGE]; 9887997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org 9987997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org#if CONFIG_VP9_HIGHBITDEPTH 10087997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.orgstatic int sad_per_bit16lut_10[QINDEX_RANGE]; 10187997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.orgstatic int sad_per_bit4lut_10[QINDEX_RANGE]; 10287997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.orgstatic int sad_per_bit16lut_12[QINDEX_RANGE]; 10387997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.orgstatic int sad_per_bit4lut_12[QINDEX_RANGE]; 10487997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org#endif 10587997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org 10687997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.orgstatic void init_me_luts_bd(int *bit16lut, int *bit4lut, int range, 10787997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org vpx_bit_depth_t bit_depth) { 108e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int i; 109e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Initialize the sad lut tables using a formulaic calculation for now. 110e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // This is to make it easier to resolve the impact of experimental changes 111e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // to the quantizer tables. 11287997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org for (i = 0; i < range; i++) { 11387997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org const double q = vp9_convert_qindex_to_q(i, bit_depth); 11487997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org bit16lut[i] = (int)(0.0418 * q + 2.4107); 11587997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org bit4lut[i] = (int)(0.063 * q + 2.742); 116e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 117e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org} 118e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 11987997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.orgvoid vp9_init_me_luts() { 12087997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org init_me_luts_bd(sad_per_bit16lut_8, sad_per_bit4lut_8, QINDEX_RANGE, 12187997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org VPX_BITS_8); 12287997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org#if CONFIG_VP9_HIGHBITDEPTH 12387997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org init_me_luts_bd(sad_per_bit16lut_10, sad_per_bit4lut_10, QINDEX_RANGE, 12487997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org VPX_BITS_10); 12587997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org init_me_luts_bd(sad_per_bit16lut_12, sad_per_bit4lut_12, QINDEX_RANGE, 12687997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org VPX_BITS_12); 12787997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org#endif 12887997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org} 12987997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org 130ac4e313c19203132648a2a271703b6ee76fe4284johannkoenig@chromium.orgstatic const int rd_boost_factor[16] = { 131ac4e313c19203132648a2a271703b6ee76fe4284johannkoenig@chromium.org 64, 32, 32, 32, 24, 16, 12, 12, 132ac4e313c19203132648a2a271703b6ee76fe4284johannkoenig@chromium.org 8, 8, 4, 4, 2, 2, 1, 0 133ac4e313c19203132648a2a271703b6ee76fe4284johannkoenig@chromium.org}; 134ac4e313c19203132648a2a271703b6ee76fe4284johannkoenig@chromium.orgstatic const int rd_frame_type_factor[FRAME_UPDATE_TYPES] = { 13587997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org 128, 144, 128, 128, 144 136ac4e313c19203132648a2a271703b6ee76fe4284johannkoenig@chromium.org}; 137ac4e313c19203132648a2a271703b6ee76fe4284johannkoenig@chromium.org 138e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.orgint vp9_compute_rd_mult(const VP9_COMP *cpi, int qindex) { 13987997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org const int q = vp9_dc_quant(qindex, 0, cpi->common.bit_depth); 14087997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org#if CONFIG_VP9_HIGHBITDEPTH 14187997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org int rdmult = 0; 14287997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org switch (cpi->common.bit_depth) { 14387997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org case VPX_BITS_8: 14487997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org rdmult = 88 * q * q / 24; 14587997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org break; 14687997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org case VPX_BITS_10: 14787997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org rdmult = ROUND_POWER_OF_TWO(88 * q * q / 24, 4); 14887997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org break; 14987997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org case VPX_BITS_12: 15087997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org rdmult = ROUND_POWER_OF_TWO(88 * q * q / 24, 8); 15187997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org break; 15287997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org default: 15387997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org assert(0 && "bit_depth should be VPX_BITS_8, VPX_BITS_10 or VPX_BITS_12"); 15487997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org return -1; 15587997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org } 15687997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org#else 157ac4e313c19203132648a2a271703b6ee76fe4284johannkoenig@chromium.org int rdmult = 88 * q * q / 24; 15887997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org#endif 15941294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org if (cpi->oxcf.pass == 2 && (cpi->common.frame_type != KEY_FRAME)) { 160ac4e313c19203132648a2a271703b6ee76fe4284johannkoenig@chromium.org const GF_GROUP *const gf_group = &cpi->twopass.gf_group; 161ac4e313c19203132648a2a271703b6ee76fe4284johannkoenig@chromium.org const FRAME_UPDATE_TYPE frame_type = gf_group->update_type[gf_group->index]; 162ac4e313c19203132648a2a271703b6ee76fe4284johannkoenig@chromium.org const int boost_index = MIN(15, (cpi->rc.gfu_boost / 100)); 163ac4e313c19203132648a2a271703b6ee76fe4284johannkoenig@chromium.org 164ac4e313c19203132648a2a271703b6ee76fe4284johannkoenig@chromium.org rdmult = (rdmult * rd_frame_type_factor[frame_type]) >> 7; 165ac4e313c19203132648a2a271703b6ee76fe4284johannkoenig@chromium.org rdmult += ((rdmult * rd_boost_factor[boost_index]) >> 7); 166e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 167e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org return rdmult; 168e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org} 169e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 17087997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.orgstatic int compute_rd_thresh_factor(int qindex, vpx_bit_depth_t bit_depth) { 17187997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org double q; 17287997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org#if CONFIG_VP9_HIGHBITDEPTH 17387997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org switch (bit_depth) { 17487997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org case VPX_BITS_8: 17587997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org q = vp9_dc_quant(qindex, 0, VPX_BITS_8) / 4.0; 17687997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org break; 17787997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org case VPX_BITS_10: 17887997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org q = vp9_dc_quant(qindex, 0, VPX_BITS_10) / 16.0; 17987997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org break; 18087997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org case VPX_BITS_12: 18187997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org q = vp9_dc_quant(qindex, 0, VPX_BITS_12) / 64.0; 18287997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org break; 18387997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org default: 18487997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org assert(0 && "bit_depth should be VPX_BITS_8, VPX_BITS_10 or VPX_BITS_12"); 18587997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org return -1; 18687997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org } 18787997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org#else 18887997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org (void) bit_depth; 18987997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org q = vp9_dc_quant(qindex, 0, VPX_BITS_8) / 4.0; 19087997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org#endif 191e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // TODO(debargha): Adjust the function below. 19287997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org return MAX((int)(pow(q, RD_THRESH_POW) * 5.12), 8); 193e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org} 194e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 195e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.orgvoid vp9_initialize_me_consts(VP9_COMP *cpi, int qindex) { 19687997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org#if CONFIG_VP9_HIGHBITDEPTH 19787997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org switch (cpi->common.bit_depth) { 19887997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org case VPX_BITS_8: 19987997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org cpi->mb.sadperbit16 = sad_per_bit16lut_8[qindex]; 20087997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org cpi->mb.sadperbit4 = sad_per_bit4lut_8[qindex]; 20187997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org break; 20287997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org case VPX_BITS_10: 20387997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org cpi->mb.sadperbit16 = sad_per_bit16lut_10[qindex]; 20487997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org cpi->mb.sadperbit4 = sad_per_bit4lut_10[qindex]; 20587997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org break; 20687997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org case VPX_BITS_12: 20787997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org cpi->mb.sadperbit16 = sad_per_bit16lut_12[qindex]; 20887997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org cpi->mb.sadperbit4 = sad_per_bit4lut_12[qindex]; 20987997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org break; 21087997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org default: 21187997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org assert(0 && "bit_depth should be VPX_BITS_8, VPX_BITS_10 or VPX_BITS_12"); 21287997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org } 21387997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org#else 21487997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org cpi->mb.sadperbit16 = sad_per_bit16lut_8[qindex]; 21587997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org cpi->mb.sadperbit4 = sad_per_bit4lut_8[qindex]; 21687997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org#endif 217e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org} 218e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 219e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.orgstatic void set_block_thresholds(const VP9_COMMON *cm, RD_OPT *rd) { 220e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int i, bsize, segment_id; 221e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 222e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (segment_id = 0; segment_id < MAX_SEGMENTS; ++segment_id) { 223e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const int qindex = 224e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org clamp(vp9_get_qindex(&cm->seg, segment_id, cm->base_qindex) + 22587997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org cm->y_dc_delta_q, 0, MAXQ); 22687997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org const int q = compute_rd_thresh_factor(qindex, cm->bit_depth); 227e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 228e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (bsize = 0; bsize < BLOCK_SIZES; ++bsize) { 229e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Threshold here seems unnecessarily harsh but fine given actual 230e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // range of values used for cpi->sf.thresh_mult[]. 231e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const int t = q * rd_thresh_block_size_factor[bsize]; 232e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const int thresh_max = INT_MAX / t; 233e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 234e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (bsize >= BLOCK_8X8) { 235e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (i = 0; i < MAX_MODES; ++i) 236e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->threshes[segment_id][bsize][i] = 237e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[i] < thresh_max 238e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org ? rd->thresh_mult[i] * t / 4 239e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org : INT_MAX; 240e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } else { 241e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (i = 0; i < MAX_REFS; ++i) 242e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->threshes[segment_id][bsize][i] = 243e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult_sub8x8[i] < thresh_max 244e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org ? rd->thresh_mult_sub8x8[i] * t / 4 245e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org : INT_MAX; 246e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 247e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 248e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 249e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org} 250e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 251e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.orgvoid vp9_initialize_rd_consts(VP9_COMP *cpi) { 252e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org VP9_COMMON *const cm = &cpi->common; 253e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org MACROBLOCK *const x = &cpi->mb; 254e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org RD_OPT *const rd = &cpi->rd; 255e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int i; 256e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 257e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp9_clear_system_state(); 258e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 259e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->RDDIV = RDDIV_BITS; // In bits (to multiply D by 128). 260e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->RDMULT = vp9_compute_rd_mult(cpi, cm->base_qindex + cm->y_dc_delta_q); 261e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 262e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org x->errorperbit = rd->RDMULT / RD_MULT_EPB_RATIO; 263e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org x->errorperbit += (x->errorperbit == 0); 264e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 265e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org x->select_tx_size = (cpi->sf.tx_size_search_method == USE_LARGESTALL && 266e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org cm->frame_type != KEY_FRAME) ? 0 : 1; 267e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 268e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org set_block_thresholds(cm, rd); 269e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 270e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (!cpi->sf.use_nonrd_pick_mode || cm->frame_type == KEY_FRAME) { 271e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org fill_token_costs(x->token_costs, cm->fc.coef_probs); 272e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 273e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (i = 0; i < PARTITION_CONTEXTS; ++i) 274e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp9_cost_tokens(cpi->partition_cost[i], get_partition_probs(cm, i), 275e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp9_partition_tree); 276e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 277e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 278e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (!cpi->sf.use_nonrd_pick_mode || (cm->current_video_frame & 0x07) == 1 || 279e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org cm->frame_type == KEY_FRAME) { 280e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org fill_mode_costs(cpi); 281e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 282e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (!frame_is_intra_only(cm)) { 283e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp9_build_nmv_cost_table(x->nmvjointcost, 284e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org cm->allow_high_precision_mv ? x->nmvcost_hp 285e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org : x->nmvcost, 286e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org &cm->fc.nmvc, cm->allow_high_precision_mv); 287e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 288e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (i = 0; i < INTER_MODE_CONTEXTS; ++i) 289e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp9_cost_tokens((int *)cpi->inter_mode_cost[i], 290e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org cm->fc.inter_mode_probs[i], vp9_inter_mode_tree); 291e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 292e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 293e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org} 294e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 295e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.orgstatic void model_rd_norm(int xsq_q10, int *r_q10, int *d_q10) { 296e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // NOTE: The tables below must be of the same size. 297e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 298e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // The functions described below are sampled at the four most significant 299e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // bits of x^2 + 8 / 256. 300e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 301e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Normalized rate: 302e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // This table models the rate for a Laplacian source with given variance 303e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // when quantized with a uniform quantizer with given stepsize. The 304e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // closed form expression is: 305e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Rn(x) = H(sqrt(r)) + sqrt(r)*[1 + H(r)/(1 - r)], 306e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // where r = exp(-sqrt(2) * x) and x = qpstep / sqrt(variance), 307e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // and H(x) is the binary entropy function. 308e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org static const int rate_tab_q10[] = { 309e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 65536, 6086, 5574, 5275, 5063, 4899, 4764, 4651, 310e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 4553, 4389, 4255, 4142, 4044, 3958, 3881, 3811, 311e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 3748, 3635, 3538, 3453, 3376, 3307, 3244, 3186, 312e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 3133, 3037, 2952, 2877, 2809, 2747, 2690, 2638, 313e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 2589, 2501, 2423, 2353, 2290, 2232, 2179, 2130, 314e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 2084, 2001, 1928, 1862, 1802, 1748, 1698, 1651, 315e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 1608, 1530, 1460, 1398, 1342, 1290, 1243, 1199, 316e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 1159, 1086, 1021, 963, 911, 864, 821, 781, 317e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 745, 680, 623, 574, 530, 490, 455, 424, 318e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 395, 345, 304, 269, 239, 213, 190, 171, 319e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 154, 126, 104, 87, 73, 61, 52, 44, 320e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 38, 28, 21, 16, 12, 10, 8, 6, 321e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 5, 3, 2, 1, 1, 1, 0, 0, 322e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org }; 323e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Normalized distortion: 324e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // This table models the normalized distortion for a Laplacian source 325e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // with given variance when quantized with a uniform quantizer 326e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // with given stepsize. The closed form expression is: 327e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Dn(x) = 1 - 1/sqrt(2) * x / sinh(x/sqrt(2)) 328e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // where x = qpstep / sqrt(variance). 329e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Note the actual distortion is Dn * variance. 330e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org static const int dist_tab_q10[] = { 331e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 0, 0, 1, 1, 1, 2, 2, 2, 332e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 3, 3, 4, 5, 5, 6, 7, 7, 333e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 8, 9, 11, 12, 13, 15, 16, 17, 334e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 18, 21, 24, 26, 29, 31, 34, 36, 335e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 39, 44, 49, 54, 59, 64, 69, 73, 336e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 78, 88, 97, 106, 115, 124, 133, 142, 337e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 151, 167, 184, 200, 215, 231, 245, 260, 338e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 274, 301, 327, 351, 375, 397, 418, 439, 339e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 458, 495, 528, 559, 587, 613, 637, 659, 340e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 680, 717, 749, 777, 801, 823, 842, 859, 341e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 874, 899, 919, 936, 949, 960, 969, 977, 342e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 983, 994, 1001, 1006, 1010, 1013, 1015, 1017, 343e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 1018, 1020, 1022, 1022, 1023, 1023, 1023, 1024, 344e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org }; 345e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org static const int xsq_iq_q10[] = { 346e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 0, 4, 8, 12, 16, 20, 24, 28, 347e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 32, 40, 48, 56, 64, 72, 80, 88, 348e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 96, 112, 128, 144, 160, 176, 192, 208, 349e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 224, 256, 288, 320, 352, 384, 416, 448, 350e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 480, 544, 608, 672, 736, 800, 864, 928, 351e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 992, 1120, 1248, 1376, 1504, 1632, 1760, 1888, 352e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 2016, 2272, 2528, 2784, 3040, 3296, 3552, 3808, 353e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 4064, 4576, 5088, 5600, 6112, 6624, 7136, 7648, 354e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 8160, 9184, 10208, 11232, 12256, 13280, 14304, 15328, 355e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 16352, 18400, 20448, 22496, 24544, 26592, 28640, 30688, 356e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 32736, 36832, 40928, 45024, 49120, 53216, 57312, 61408, 357e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 65504, 73696, 81888, 90080, 98272, 106464, 114656, 122848, 358e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 131040, 147424, 163808, 180192, 196576, 212960, 229344, 245728, 359e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org }; 360e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const int tmp = (xsq_q10 >> 2) + 8; 361e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const int k = get_msb(tmp) - 3; 362e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const int xq = (k << 3) + ((tmp >> k) & 0x7); 363e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const int one_q10 = 1 << 10; 364e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const int a_q10 = ((xsq_q10 - xsq_iq_q10[xq]) << 10) >> (2 + k); 365e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const int b_q10 = one_q10 - a_q10; 366e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org *r_q10 = (rate_tab_q10[xq] * b_q10 + rate_tab_q10[xq + 1] * a_q10) >> 10; 367e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org *d_q10 = (dist_tab_q10[xq] * b_q10 + dist_tab_q10[xq + 1] * a_q10) >> 10; 368e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org} 369e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 370e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.orgvoid vp9_model_rd_from_var_lapndz(unsigned int var, unsigned int n, 371e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org unsigned int qstep, int *rate, 372e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int64_t *dist) { 373e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // This function models the rate and distortion for a Laplacian 374e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // source with given variance when quantized with a uniform quantizer 375e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // with given stepsize. The closed form expressions are in: 376e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Hang and Chen, "Source Model for transform video coder and its 377e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // application - Part I: Fundamental Theory", IEEE Trans. Circ. 378e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Sys. for Video Tech., April 1997. 379e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (var == 0) { 380e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org *rate = 0; 381e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org *dist = 0; 382e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } else { 383e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int d_q10, r_q10; 38441294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org static const uint32_t MAX_XSQ_Q10 = 245727; 385e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const uint64_t xsq_q10_64 = 386e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org ((((uint64_t)qstep * qstep * n) << 10) + (var >> 1)) / var; 38741294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org const int xsq_q10 = (int)MIN(xsq_q10_64, MAX_XSQ_Q10); 388e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org model_rd_norm(xsq_q10, &r_q10, &d_q10); 389e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org *rate = (n * r_q10 + 2) >> 2; 390e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org *dist = (var * (int64_t)d_q10 + 512) >> 10; 391e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 392e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org} 393e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 394e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.orgvoid vp9_get_entropy_contexts(BLOCK_SIZE bsize, TX_SIZE tx_size, 395e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const struct macroblockd_plane *pd, 396e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org ENTROPY_CONTEXT t_above[16], 397e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org ENTROPY_CONTEXT t_left[16]) { 398e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const BLOCK_SIZE plane_bsize = get_plane_block_size(bsize, pd); 399e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const int num_4x4_w = num_4x4_blocks_wide_lookup[plane_bsize]; 400e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const int num_4x4_h = num_4x4_blocks_high_lookup[plane_bsize]; 401e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const ENTROPY_CONTEXT *const above = pd->above_context; 402e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const ENTROPY_CONTEXT *const left = pd->left_context; 403e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 404e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int i; 405e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org switch (tx_size) { 406e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org case TX_4X4: 407e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vpx_memcpy(t_above, above, sizeof(ENTROPY_CONTEXT) * num_4x4_w); 408e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vpx_memcpy(t_left, left, sizeof(ENTROPY_CONTEXT) * num_4x4_h); 409e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org break; 410e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org case TX_8X8: 411e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (i = 0; i < num_4x4_w; i += 2) 412e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org t_above[i] = !!*(const uint16_t *)&above[i]; 413e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (i = 0; i < num_4x4_h; i += 2) 414e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org t_left[i] = !!*(const uint16_t *)&left[i]; 415e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org break; 416e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org case TX_16X16: 417e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (i = 0; i < num_4x4_w; i += 4) 418e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org t_above[i] = !!*(const uint32_t *)&above[i]; 419e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (i = 0; i < num_4x4_h; i += 4) 420e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org t_left[i] = !!*(const uint32_t *)&left[i]; 421e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org break; 422e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org case TX_32X32: 423e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (i = 0; i < num_4x4_w; i += 8) 424e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org t_above[i] = !!*(const uint64_t *)&above[i]; 425e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (i = 0; i < num_4x4_h; i += 8) 426e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org t_left[i] = !!*(const uint64_t *)&left[i]; 427e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org break; 428e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org default: 429e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org assert(0 && "Invalid transform size."); 43041294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org break; 431e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 432e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org} 433e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 434e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.orgvoid vp9_mv_pred(VP9_COMP *cpi, MACROBLOCK *x, 435e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org uint8_t *ref_y_buffer, int ref_y_stride, 436e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int ref_frame, BLOCK_SIZE block_size) { 437e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org MACROBLOCKD *xd = &x->e_mbd; 43887997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org MB_MODE_INFO *mbmi = &xd->mi[0].src_mi->mbmi; 439e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int i; 440e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int zero_seen = 0; 441e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int best_index = 0; 442e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int best_sad = INT_MAX; 443e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int this_sad = INT_MAX; 444e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int max_mv = 0; 445e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org uint8_t *src_y_ptr = x->plane[0].src.buf; 446e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org uint8_t *ref_y_ptr; 447d95585fb0ec024f6abd96f7b02e0df58019d46afjohannkoenig@chromium.org const int num_mv_refs = MAX_MV_REF_CANDIDATES + 448e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org (cpi->sf.adaptive_motion_search && 449e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org block_size < cpi->sf.max_partition_size); 450e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 451e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org MV pred_mv[3]; 452e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org pred_mv[0] = mbmi->ref_mvs[ref_frame][0].as_mv; 453e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org pred_mv[1] = mbmi->ref_mvs[ref_frame][1].as_mv; 454e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org pred_mv[2] = x->pred_mv[ref_frame]; 455e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 456e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Get the sad for each candidate reference mv. 457e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (i = 0; i < num_mv_refs; ++i) { 458d95585fb0ec024f6abd96f7b02e0df58019d46afjohannkoenig@chromium.org const MV *this_mv = &pred_mv[i]; 459e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 460d95585fb0ec024f6abd96f7b02e0df58019d46afjohannkoenig@chromium.org max_mv = MAX(max_mv, MAX(abs(this_mv->row), abs(this_mv->col)) >> 3); 461d95585fb0ec024f6abd96f7b02e0df58019d46afjohannkoenig@chromium.org if (is_zero_mv(this_mv) && zero_seen) 462e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org continue; 463e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 464d95585fb0ec024f6abd96f7b02e0df58019d46afjohannkoenig@chromium.org zero_seen |= is_zero_mv(this_mv); 465e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 466d95585fb0ec024f6abd96f7b02e0df58019d46afjohannkoenig@chromium.org ref_y_ptr = 467d95585fb0ec024f6abd96f7b02e0df58019d46afjohannkoenig@chromium.org &ref_y_buffer[ref_y_stride * (this_mv->row >> 3) + (this_mv->col >> 3)]; 468e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 469e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Find sad for current vector. 470e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org this_sad = cpi->fn_ptr[block_size].sdf(src_y_ptr, x->plane[0].src.stride, 471e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org ref_y_ptr, ref_y_stride); 472e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 473e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Note if it is the best so far. 474e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (this_sad < best_sad) { 475e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org best_sad = this_sad; 476e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org best_index = i; 477e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 478e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 479e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 480e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Note the index of the mv that worked best in the reference list. 481e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org x->mv_best_ref_index[ref_frame] = best_index; 482e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org x->max_mv_context[ref_frame] = max_mv; 483e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org x->pred_mv_sad[ref_frame] = best_sad; 484e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org} 485e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 486e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.orgvoid vp9_setup_pred_block(const MACROBLOCKD *xd, 487e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org struct buf_2d dst[MAX_MB_PLANE], 488e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const YV12_BUFFER_CONFIG *src, 489e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int mi_row, int mi_col, 490e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const struct scale_factors *scale, 491e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const struct scale_factors *scale_uv) { 492e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int i; 493e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 494e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org dst[0].buf = src->y_buffer; 495e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org dst[0].stride = src->y_stride; 496e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org dst[1].buf = src->u_buffer; 497e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org dst[2].buf = src->v_buffer; 498e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org dst[1].stride = dst[2].stride = src->uv_stride; 499e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 500e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (i = 0; i < MAX_MB_PLANE; ++i) { 501e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org setup_pred_plane(dst + i, dst[i].buf, dst[i].stride, mi_row, mi_col, 502e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org i ? scale_uv : scale, 503e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org xd->plane[i].subsampling_x, xd->plane[i].subsampling_y); 504e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 505e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org} 506e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 507e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.orgconst YV12_BUFFER_CONFIG *vp9_get_scaled_ref_frame(const VP9_COMP *cpi, 508e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int ref_frame) { 509e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const VP9_COMMON *const cm = &cpi->common; 510e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const int ref_idx = cm->ref_frame_map[get_ref_frame_idx(cpi, ref_frame)]; 511e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const int scaled_idx = cpi->scaled_ref_idx[ref_frame - 1]; 512e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org return (scaled_idx != ref_idx) ? &cm->frame_bufs[scaled_idx].buf : NULL; 513e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org} 514e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 515e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.orgint vp9_get_switchable_rate(const VP9_COMP *cpi) { 516e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const MACROBLOCKD *const xd = &cpi->mb.e_mbd; 51787997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org const MB_MODE_INFO *const mbmi = &xd->mi[0].src_mi->mbmi; 518e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const int ctx = vp9_get_pred_context_switchable_interp(xd); 519e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org return SWITCHABLE_INTERP_RATE_FACTOR * 520e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org cpi->switchable_interp_costs[ctx][mbmi->interp_filter]; 521e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org} 522e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 523e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.orgvoid vp9_set_rd_speed_thresholds(VP9_COMP *cpi) { 524e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int i; 525e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org RD_OPT *const rd = &cpi->rd; 526e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org SPEED_FEATURES *const sf = &cpi->sf; 527e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 528e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Set baseline threshold values. 529e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (i = 0; i < MAX_MODES; ++i) 530d95585fb0ec024f6abd96f7b02e0df58019d46afjohannkoenig@chromium.org rd->thresh_mult[i] = cpi->oxcf.mode == BEST ? -500 : 0; 531e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 53287997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org if (sf->adaptive_rd_thresh) { 53387997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org rd->thresh_mult[THR_NEARESTMV] = 300; 53487997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org rd->thresh_mult[THR_NEARESTG] = 300; 53587997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org rd->thresh_mult[THR_NEARESTA] = 300; 53687997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org } else { 53787997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org rd->thresh_mult[THR_NEARESTMV] = 0; 53887997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org rd->thresh_mult[THR_NEARESTG] = 0; 53987997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org rd->thresh_mult[THR_NEARESTA] = 0; 54087997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org } 541e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 542e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_DC] += 1000; 543e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 544e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_NEWMV] += 1000; 545e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_NEWA] += 1000; 546e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_NEWG] += 1000; 547e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 548e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Adjust threshold only in real time mode, which only uses last 549e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // reference frame. 550e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_NEWMV] += sf->elevate_newmv_thresh; 551e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 552e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_NEARMV] += 1000; 553e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_NEARA] += 1000; 554e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_COMP_NEARESTLA] += 1000; 555e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_COMP_NEARESTGA] += 1000; 556e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 557e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_TM] += 1000; 558e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 559e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_COMP_NEARLA] += 1500; 560e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_COMP_NEWLA] += 2000; 561e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_NEARG] += 1000; 562e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_COMP_NEARGA] += 1500; 563e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_COMP_NEWGA] += 2000; 564e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 565e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_ZEROMV] += 2000; 566e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_ZEROG] += 2000; 567e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_ZEROA] += 2000; 568e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_COMP_ZEROLA] += 2500; 569e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_COMP_ZEROGA] += 2500; 570e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 571e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_H_PRED] += 2000; 572e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_V_PRED] += 2000; 573e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_D45_PRED ] += 2500; 574e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_D135_PRED] += 2500; 575e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_D117_PRED] += 2500; 576e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_D153_PRED] += 2500; 577e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_D207_PRED] += 2500; 578e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult[THR_D63_PRED] += 2500; 579e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org} 580e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 581e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.orgvoid vp9_set_rd_speed_thresholds_sub8x8(VP9_COMP *cpi) { 582e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const SPEED_FEATURES *const sf = &cpi->sf; 583e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org RD_OPT *const rd = &cpi->rd; 584e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int i; 585e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 586e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (i = 0; i < MAX_REFS; ++i) 587d95585fb0ec024f6abd96f7b02e0df58019d46afjohannkoenig@chromium.org rd->thresh_mult_sub8x8[i] = cpi->oxcf.mode == BEST ? -500 : 0; 588e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 589e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult_sub8x8[THR_LAST] += 2500; 590e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult_sub8x8[THR_GOLD] += 2500; 591e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult_sub8x8[THR_ALTR] += 2500; 592e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult_sub8x8[THR_INTRA] += 2500; 593e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult_sub8x8[THR_COMP_LA] += 4500; 594e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult_sub8x8[THR_COMP_GA] += 4500; 595e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 596e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Check for masked out split cases. 597e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (i = 0; i < MAX_REFS; ++i) 598e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (sf->disable_split_mask & (1 << i)) 599e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org rd->thresh_mult_sub8x8[i] = INT_MAX; 600e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org} 601