|  | /* | 
|  | * Copyright (c) 2017, Alliance for Open Media. All rights reserved | 
|  | * | 
|  | * This source code is subject to the terms of the BSD 2 Clause License and | 
|  | * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License | 
|  | * was not distributed with this source code in the LICENSE file, you can | 
|  | * obtain it at www.aomedia.org/license/software. If the Alliance for Open | 
|  | * Media Patent License 1.0 was not distributed with this source code in the | 
|  | * PATENTS file, you can obtain it at www.aomedia.org/license/patent. | 
|  | */ | 
|  |  | 
|  | #ifndef AV1_COMMON_TXB_COMMON_H_ | 
|  | #define AV1_COMMON_TXB_COMMON_H_ | 
|  |  | 
|  | #define REDUCE_CONTEXT_DEPENDENCY 0 | 
|  | #define MIN_SCAN_IDX_REDUCE_CONTEXT_DEPENDENCY 0 | 
|  |  | 
|  | extern const int16_t k_eob_group_start[12]; | 
|  | extern const int16_t k_eob_offset_bits[12]; | 
|  | int16_t get_eob_pos_token(int eob, int16_t *extra); | 
|  | int av1_get_eob_pos_ctx(TX_TYPE tx_type, int eob_token); | 
|  |  | 
|  | extern const int16_t av1_coeff_band_4x4[16]; | 
|  |  | 
|  | extern const int16_t av1_coeff_band_8x8[64]; | 
|  |  | 
|  | extern const int16_t av1_coeff_band_16x16[256]; | 
|  |  | 
|  | extern const int16_t av1_coeff_band_32x32[1024]; | 
|  |  | 
|  | typedef struct txb_ctx { | 
|  | int txb_skip_ctx; | 
|  | int dc_sign_ctx; | 
|  | } TXB_CTX; | 
|  |  | 
|  | static INLINE TX_SIZE get_txsize_context(TX_SIZE tx_size) { | 
|  | return txsize_sqr_up_map[tx_size]; | 
|  | } | 
|  |  | 
|  | static const int base_level_count_to_index[13] = { | 
|  | 0, 0, 1, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3, | 
|  | }; | 
|  |  | 
|  | static const int base_ref_offset[BASE_CONTEXT_POSITION_NUM][2] = { | 
|  | /* clang-format off*/ | 
|  | { -2, 0 }, { -1, -1 }, { -1, 0 }, { -1, 1 }, { 0, -2 }, { 0, -1 }, { 0, 1 }, | 
|  | { 0, 2 },  { 1, -1 },  { 1, 0 },  { 1, 1 },  { 2, 0 } | 
|  | /* clang-format on*/ | 
|  | }; | 
|  |  | 
|  | // TODO(linfengz): Some functions have coeff_is_byte_flag to handle different | 
|  | // types of input coefficients. If possible, unify types to uint8_t* later. | 
|  |  | 
|  | static INLINE void get_base_count_mag(int *mag, int *count, | 
|  | const tran_low_t *tcoeffs, int bwl, | 
|  | int height, int row, int col) { | 
|  | mag[0] = 0; | 
|  | mag[1] = 0; | 
|  | for (int i = 0; i < NUM_BASE_LEVELS; ++i) count[i] = 0; | 
|  | for (int idx = 0; idx < BASE_CONTEXT_POSITION_NUM; ++idx) { | 
|  | const int ref_row = row + base_ref_offset[idx][0]; | 
|  | const int ref_col = col + base_ref_offset[idx][1]; | 
|  | if (ref_row < 0 || ref_col < 0 || ref_row >= height || | 
|  | ref_col >= (1 << bwl)) | 
|  | continue; | 
|  | const int pos = (ref_row << bwl) + ref_col; | 
|  | tran_low_t abs_coeff = abs(tcoeffs[pos]); | 
|  | // count | 
|  | for (int i = 0; i < NUM_BASE_LEVELS; ++i) { | 
|  | count[i] += abs_coeff > i; | 
|  | } | 
|  | // mag | 
|  | if (base_ref_offset[idx][0] >= 0 && base_ref_offset[idx][1] >= 0) { | 
|  | if (abs_coeff > mag[0]) { | 
|  | mag[0] = abs_coeff; | 
|  | mag[1] = 1; | 
|  | } else if (abs_coeff == mag[0]) { | 
|  | ++mag[1]; | 
|  | } | 
|  | } | 
|  | } | 
|  | } | 
|  |  | 
|  | static INLINE int get_level_count_mag( | 
|  | int *const mag, const uint8_t *const levels, const int bwl, | 
|  | const int height, const int row, const int col, const int level, | 
|  | const int (*nb_offset)[2], const int nb_num) { | 
|  | const int stride = 1 << bwl; | 
|  | int count = 0; | 
|  |  | 
|  | for (int idx = 0; idx < nb_num; ++idx) { | 
|  | const int ref_row = row + nb_offset[idx][0]; | 
|  | const int ref_col = col + nb_offset[idx][1]; | 
|  | if (ref_row < 0 || ref_col < 0 || ref_row >= height || ref_col >= stride) | 
|  | continue; | 
|  | const int pos = (ref_row << bwl) + ref_col; | 
|  | count += levels[pos] > level; | 
|  | if (nb_offset[idx][0] == 0 && nb_offset[idx][1] == 1) mag[0] = levels[pos]; | 
|  | if (nb_offset[idx][0] == 1 && nb_offset[idx][1] == 0) mag[1] = levels[pos]; | 
|  | if (nb_offset[idx][0] == 1 && nb_offset[idx][1] == 1) mag[2] = levels[pos]; | 
|  | } | 
|  | return count; | 
|  | } | 
|  |  | 
|  | static INLINE int get_level_count_mag_coeff( | 
|  | int *const mag, const tran_low_t *const tcoeffs, const int bwl, | 
|  | const int height, const int row, const int col, const int level, | 
|  | const int (*nb_offset)[2], const int nb_num) { | 
|  | const int stride = 1 << bwl; | 
|  | int count = 0; | 
|  | for (int idx = 0; idx < nb_num; ++idx) { | 
|  | const int ref_row = row + nb_offset[idx][0]; | 
|  | const int ref_col = col + nb_offset[idx][1]; | 
|  | if (ref_row < 0 || ref_col < 0 || ref_row >= height || ref_col >= stride) | 
|  | continue; | 
|  | const int pos = (ref_row << bwl) + ref_col; | 
|  | tran_low_t abs_coeff = abs(tcoeffs[pos]); | 
|  | count += abs_coeff > level; | 
|  |  | 
|  | if (nb_offset[idx][0] == 0 && nb_offset[idx][1] == 1) mag[0] = abs_coeff; | 
|  | if (nb_offset[idx][0] == 1 && nb_offset[idx][1] == 0) mag[1] = abs_coeff; | 
|  | if (nb_offset[idx][0] == 1 && nb_offset[idx][1] == 1) mag[2] = abs_coeff; | 
|  | } | 
|  | return count; | 
|  | } | 
|  |  | 
|  | static INLINE int get_base_ctx_from_count_mag(int row, int col, int count, | 
|  | int sig_mag) { | 
|  | const int ctx = base_level_count_to_index[count]; | 
|  | int ctx_idx = -1; | 
|  |  | 
|  | if (row == 0 && col == 0) { | 
|  | if (sig_mag >= 2) return ctx_idx = 0; | 
|  | if (sig_mag == 1) { | 
|  | if (count >= 2) | 
|  | ctx_idx = 1; | 
|  | else | 
|  | ctx_idx = 2; | 
|  |  | 
|  | return ctx_idx; | 
|  | } | 
|  |  | 
|  | ctx_idx = 3 + ctx; | 
|  | assert(ctx_idx <= 6); | 
|  | return ctx_idx; | 
|  | } else if (row == 0) { | 
|  | if (sig_mag >= 2) return ctx_idx = 6; | 
|  | if (sig_mag == 1) { | 
|  | if (count >= 2) | 
|  | ctx_idx = 7; | 
|  | else | 
|  | ctx_idx = 8; | 
|  | return ctx_idx; | 
|  | } | 
|  |  | 
|  | ctx_idx = 9 + ctx; | 
|  | assert(ctx_idx <= 11); | 
|  | return ctx_idx; | 
|  | } else if (col == 0) { | 
|  | if (sig_mag >= 2) return ctx_idx = 12; | 
|  | if (sig_mag == 1) { | 
|  | if (count >= 2) | 
|  | ctx_idx = 13; | 
|  | else | 
|  | ctx_idx = 14; | 
|  |  | 
|  | return ctx_idx; | 
|  | } | 
|  |  | 
|  | ctx_idx = 15 + ctx; | 
|  | assert(ctx_idx <= 17); | 
|  | // TODO(angiebird): turn this on once the optimization is finalized | 
|  | // assert(ctx_idx < 28); | 
|  | } else { | 
|  | if (sig_mag >= 2) return ctx_idx = 18; | 
|  | if (sig_mag == 1) { | 
|  | if (count >= 2) | 
|  | ctx_idx = 19; | 
|  | else | 
|  | ctx_idx = 20; | 
|  | return ctx_idx; | 
|  | } | 
|  |  | 
|  | ctx_idx = 21 + ctx; | 
|  |  | 
|  | assert(ctx_idx <= 24); | 
|  | } | 
|  | return ctx_idx; | 
|  | } | 
|  |  | 
|  | static INLINE int get_base_ctx(const uint8_t *const levels, | 
|  | const int c,  // raster order | 
|  | const int bwl, const int height, | 
|  | const int level) { | 
|  | const int row = c >> bwl; | 
|  | const int col = c - (row << bwl); | 
|  | const int level_minus_1 = level - 1; | 
|  | int mag_count = 0; | 
|  | int nb_mag[3] = { 0 }; | 
|  | const int count = | 
|  | get_level_count_mag(nb_mag, levels, bwl, height, row, col, level_minus_1, | 
|  | base_ref_offset, BASE_CONTEXT_POSITION_NUM); | 
|  |  | 
|  | for (int idx = 0; idx < 3; ++idx) mag_count += nb_mag[idx] > level; | 
|  | const int ctx_idx = | 
|  | get_base_ctx_from_count_mag(row, col, count, AOMMIN(2, mag_count)); | 
|  | return ctx_idx; | 
|  | } | 
|  |  | 
|  | #define BR_CONTEXT_POSITION_NUM 8  // Base range coefficient context | 
|  | static const int br_ref_offset[BR_CONTEXT_POSITION_NUM][2] = { | 
|  | /* clang-format off*/ | 
|  | { -1, -1 }, { -1, 0 }, { -1, 1 }, { 0, -1 }, | 
|  | { 0, 1 },   { 1, -1 }, { 1, 0 },  { 1, 1 }, | 
|  | /* clang-format on*/ | 
|  | }; | 
|  |  | 
|  | static const int br_level_map[9] = { | 
|  | 0, 0, 1, 1, 2, 2, 3, 3, 3, | 
|  | }; | 
|  |  | 
|  | static const int coeff_to_br_index[COEFF_BASE_RANGE] = { | 
|  | 0, 0, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2, | 
|  | }; | 
|  |  | 
|  | static const int br_index_to_coeff[BASE_RANGE_SETS] = { | 
|  | 0, 2, 6, | 
|  | }; | 
|  |  | 
|  | static const int br_extra_bits[BASE_RANGE_SETS] = { | 
|  | 1, 2, 3, | 
|  | }; | 
|  |  | 
|  | #define BR_MAG_OFFSET 1 | 
|  | // TODO(angiebird): optimize this function by using a table to map from | 
|  | // count/mag to ctx | 
|  |  | 
|  | static INLINE int get_br_count_mag(int *mag, const tran_low_t *tcoeffs, int bwl, | 
|  | int height, int row, int col, int level) { | 
|  | mag[0] = 0; | 
|  | mag[1] = 0; | 
|  | int count = 0; | 
|  | for (int idx = 0; idx < BR_CONTEXT_POSITION_NUM; ++idx) { | 
|  | const int ref_row = row + br_ref_offset[idx][0]; | 
|  | const int ref_col = col + br_ref_offset[idx][1]; | 
|  | if (ref_row < 0 || ref_col < 0 || ref_row >= height || | 
|  | ref_col >= (1 << bwl)) | 
|  | continue; | 
|  | const int pos = (ref_row << bwl) + ref_col; | 
|  | tran_low_t abs_coeff = abs(tcoeffs[pos]); | 
|  | count += abs_coeff > level; | 
|  | if (br_ref_offset[idx][0] >= 0 && br_ref_offset[idx][1] >= 0) { | 
|  | if (abs_coeff > mag[0]) { | 
|  | mag[0] = abs_coeff; | 
|  | mag[1] = 1; | 
|  | } else if (abs_coeff == mag[0]) { | 
|  | ++mag[1]; | 
|  | } | 
|  | } | 
|  | } | 
|  | return count; | 
|  | } | 
|  |  | 
|  | static INLINE int get_br_ctx_from_count_mag(int row, int col, int count, | 
|  | int mag) { | 
|  | int offset = 0; | 
|  | if (mag <= BR_MAG_OFFSET) | 
|  | offset = 0; | 
|  | else if (mag <= 3) | 
|  | offset = 1; | 
|  | else if (mag <= 5) | 
|  | offset = 2; | 
|  | else | 
|  | offset = 3; | 
|  |  | 
|  | int ctx = br_level_map[count]; | 
|  | ctx += offset * BR_TMP_OFFSET; | 
|  |  | 
|  | // DC: 0 - 1 | 
|  | if (row == 0 && col == 0) return ctx; | 
|  |  | 
|  | // Top row: 2 - 4 | 
|  | if (row == 0) return 2 + ctx; | 
|  |  | 
|  | // Left column: 5 - 7 | 
|  | if (col == 0) return 5 + ctx; | 
|  |  | 
|  | // others: 8 - 11 | 
|  | return 8 + ctx; | 
|  | } | 
|  |  | 
|  | static INLINE int get_br_ctx(const uint8_t *const levels, | 
|  | const int c,  // raster order | 
|  | const int bwl, const int height) { | 
|  | const int row = c >> bwl; | 
|  | const int col = c - (row << bwl); | 
|  | const int level_minus_1 = NUM_BASE_LEVELS; | 
|  | int mag = 0; | 
|  | int nb_mag[3] = { 0 }; | 
|  | const int count = | 
|  | get_level_count_mag(nb_mag, levels, bwl, height, row, col, level_minus_1, | 
|  | br_ref_offset, BR_CONTEXT_POSITION_NUM); | 
|  | for (int idx = 0; idx < 3; ++idx) mag = AOMMAX(mag, nb_mag[idx]); | 
|  | const int ctx = get_br_ctx_from_count_mag(row, col, count, mag); | 
|  | return ctx; | 
|  | } | 
|  |  | 
|  | static INLINE int get_br_ctx_coeff(const tran_low_t *const tcoeffs, | 
|  | const int c,  // raster order | 
|  | const int bwl, const int height) { | 
|  | const int row = c >> bwl; | 
|  | const int col = c - (row << bwl); | 
|  | const int level_minus_1 = NUM_BASE_LEVELS; | 
|  | int mag = 0; | 
|  | int nb_mag[3] = { 0 }; | 
|  | const int count = get_level_count_mag_coeff(nb_mag, tcoeffs, bwl, height, row, | 
|  | col, level_minus_1, br_ref_offset, | 
|  | BR_CONTEXT_POSITION_NUM); | 
|  | for (int idx = 0; idx < 3; ++idx) mag = AOMMAX(mag, nb_mag[idx]); | 
|  | const int ctx = get_br_ctx_from_count_mag(row, col, count, mag); | 
|  | return ctx; | 
|  | } | 
|  |  | 
|  | #define SIG_REF_OFFSET_NUM 7 | 
|  |  | 
|  | static const int sig_ref_offset[SIG_REF_OFFSET_NUM][2] = { | 
|  | { 0, 1 }, { 1, 0 }, { 1, 1 }, { 0, 2 }, { 2, 0 }, { 1, 2 }, { 2, 1 }, | 
|  | }; | 
|  |  | 
|  | static const int sig_ref_offset_vert[SIG_REF_OFFSET_NUM][2] = { | 
|  | { 1, 0 }, { 2, 0 }, { 0, 1 }, { 3, 0 }, { 4, 0 }, { 1, 1 }, { 2, 1 }, | 
|  | }; | 
|  |  | 
|  | static const int sig_ref_offset_horiz[SIG_REF_OFFSET_NUM][2] = { | 
|  | { 0, 1 }, { 0, 2 }, { 1, 0 }, { 0, 3 }, { 0, 4 }, { 1, 1 }, { 1, 2 }, | 
|  | }; | 
|  |  | 
|  | #if USE_CAUSAL_BASE_CTX | 
|  | static INLINE int get_nz_count_mag(const tran_low_t *tcoeffs, int bwl, | 
|  | int height, int row, int col, | 
|  | TX_CLASS tx_class, | 
|  | const int coeff_is_byte_flag, int *mag) { | 
|  | int count = 0; | 
|  | *mag = 0; | 
|  | for (int idx = 0; idx < SIG_REF_OFFSET_NUM; ++idx) { | 
|  | const int row_offset = | 
|  | ((tx_class == TX_CLASS_2D) | 
|  | ? sig_ref_offset[idx][0] | 
|  | : ((tx_class == TX_CLASS_VERT) ? sig_ref_offset_vert[idx][0] | 
|  | : sig_ref_offset_horiz[idx][0])); | 
|  | const int col_offset = | 
|  | ((tx_class == TX_CLASS_2D) | 
|  | ? sig_ref_offset[idx][1] | 
|  | : ((tx_class == TX_CLASS_VERT) ? sig_ref_offset_vert[idx][1] | 
|  | : sig_ref_offset_horiz[idx][1])); | 
|  | const int ref_row = row + row_offset; | 
|  | const int ref_col = col + col_offset; | 
|  | if (ref_row < 0 || ref_col < 0 || ref_row >= height || | 
|  | ref_col >= (1 << bwl)) | 
|  | continue; | 
|  | const int nb_pos = (ref_row << bwl) + ref_col; | 
|  | int level = abs(coeff_is_byte_flag ? ((const uint8_t *)tcoeffs)[nb_pos] | 
|  | : ((const tran_low_t *)tcoeffs)[nb_pos]); | 
|  | count += (level != 0); | 
|  | #if 1 | 
|  | if (idx < 5) { | 
|  | *mag += AOMMIN(level, 3); | 
|  | } | 
|  | #endif | 
|  | } | 
|  | return count; | 
|  | } | 
|  | #endif | 
|  | static INLINE int get_nz_count(const tran_low_t *tcoeffs, int bwl, int height, | 
|  | int row, int col, TX_CLASS tx_class, | 
|  | const int coeff_is_byte_flag) { | 
|  | int count = 0; | 
|  | for (int idx = 0; idx < SIG_REF_OFFSET_NUM; ++idx) { | 
|  | const int ref_row = row + ((tx_class == TX_CLASS_2D) | 
|  | ? sig_ref_offset[idx][0] | 
|  | : ((tx_class == TX_CLASS_VERT) | 
|  | ? sig_ref_offset_vert[idx][0] | 
|  | : sig_ref_offset_horiz[idx][0])); | 
|  | const int ref_col = col + ((tx_class == TX_CLASS_2D) | 
|  | ? sig_ref_offset[idx][1] | 
|  | : ((tx_class == TX_CLASS_VERT) | 
|  | ? sig_ref_offset_vert[idx][1] | 
|  | : sig_ref_offset_horiz[idx][1])); | 
|  | if (ref_row < 0 || ref_col < 0 || ref_row >= height || | 
|  | ref_col >= (1 << bwl)) | 
|  | continue; | 
|  | const int nb_pos = (ref_row << bwl) + ref_col; | 
|  | count += | 
|  | ((coeff_is_byte_flag ? ((const uint8_t *)tcoeffs)[nb_pos] | 
|  | : ((const tran_low_t *)tcoeffs)[nb_pos]) != 0); | 
|  | } | 
|  | return count; | 
|  | } | 
|  |  | 
|  | static INLINE TX_CLASS get_tx_class(TX_TYPE tx_type) { | 
|  | switch (tx_type) { | 
|  | case V_DCT: | 
|  | case V_ADST: | 
|  | case V_FLIPADST: return TX_CLASS_VERT; | 
|  | case H_DCT: | 
|  | case H_ADST: | 
|  | case H_FLIPADST: return TX_CLASS_HORIZ; | 
|  | default: return TX_CLASS_2D; | 
|  | } | 
|  | } | 
|  |  | 
|  | // TODO(angiebird): optimize this function by generate a table that maps from | 
|  | // count to ctx | 
|  | static INLINE int get_nz_map_ctx_from_count(int count, | 
|  | int coeff_idx,  // raster order | 
|  | int bwl, int height, | 
|  | #if USE_CAUSAL_BASE_CTX | 
|  | const int mag, | 
|  | #endif | 
|  | TX_TYPE tx_type) { | 
|  | (void)tx_type; | 
|  | const int row = coeff_idx >> bwl; | 
|  | const int col = coeff_idx - (row << bwl); | 
|  | const int width = 1 << bwl; | 
|  |  | 
|  | int ctx = 0; | 
|  | int tx_class = get_tx_class(tx_type); | 
|  | int offset; | 
|  | if (tx_class == TX_CLASS_2D) | 
|  | offset = 0; | 
|  | else if (tx_class == TX_CLASS_VERT) | 
|  | offset = SIG_COEF_CONTEXTS_2D; | 
|  | else | 
|  | #if USE_CAUSAL_BASE_CTX | 
|  | offset = SIG_COEF_CONTEXTS_2D; | 
|  | #else | 
|  | offset = SIG_COEF_CONTEXTS_2D + SIG_COEF_CONTEXTS_1D; | 
|  | #endif | 
|  |  | 
|  | #if USE_CAUSAL_BASE_CTX | 
|  | (void)count; | 
|  | ctx = AOMMIN((mag + 1) >> 1, 4); | 
|  | #else | 
|  | ctx = (count + 1) >> 1; | 
|  | #endif | 
|  |  | 
|  | if (tx_class == TX_CLASS_2D) { | 
|  | { | 
|  | if (row == 0 && col == 0) return offset + 0; | 
|  |  | 
|  | if (width < height) | 
|  | if (row < 2) return offset + 11 + ctx; | 
|  |  | 
|  | if (width > height) | 
|  | if (col < 2) return offset + 16 + ctx; | 
|  |  | 
|  | if (row + col < 2) return offset + ctx + 1; | 
|  | if (row + col < 4) return offset + 5 + ctx + 1; | 
|  |  | 
|  | return offset + 21 + AOMMIN(ctx, 4); | 
|  | } | 
|  | } else { | 
|  | if (tx_class == TX_CLASS_VERT) { | 
|  | if (row == 0) return offset + ctx; | 
|  | if (row < 2) return offset + 5 + ctx; | 
|  | return offset + 10 + ctx; | 
|  | } else { | 
|  | if (col == 0) return offset + ctx; | 
|  | if (col < 2) return offset + 5 + ctx; | 
|  | return offset + 10 + ctx; | 
|  | } | 
|  | } | 
|  | } | 
|  |  | 
|  | static INLINE int get_nz_map_ctx(const void *const tcoeffs, const int scan_idx, | 
|  | const int16_t *const scan, const int bwl, | 
|  | const int height, const TX_TYPE tx_type, | 
|  | const int coeff_is_byte_flag) { | 
|  | const int coeff_idx = scan[scan_idx]; | 
|  | const int row = coeff_idx >> bwl; | 
|  | const int col = coeff_idx - (row << bwl); | 
|  |  | 
|  | int tx_class = get_tx_class(tx_type); | 
|  | #if USE_CAUSAL_BASE_CTX | 
|  | int mag = 0; | 
|  | int count = get_nz_count_mag(tcoeffs, bwl, height, row, col, tx_class, | 
|  | coeff_is_byte_flag, &mag); | 
|  | return get_nz_map_ctx_from_count(count, coeff_idx, bwl, height, mag, tx_type); | 
|  | #else | 
|  | int count = get_nz_count(tcoeffs, bwl, height, row, col, tx_class, | 
|  | coeff_is_byte_flag); | 
|  | return get_nz_map_ctx_from_count(count, coeff_idx, bwl, height, tx_type); | 
|  | #endif | 
|  | } | 
|  |  | 
|  | static INLINE int get_eob_ctx(const int coeff_idx,  // raster order | 
|  | const TX_SIZE txs_ctx, const TX_TYPE tx_type) { | 
|  | int offset = 0; | 
|  | #if CONFIG_CTX1D | 
|  | const TX_CLASS tx_class = get_tx_class(tx_type); | 
|  | if (tx_class == TX_CLASS_VERT) | 
|  | offset = EOB_COEF_CONTEXTS_2D; | 
|  | else if (tx_class == TX_CLASS_HORIZ) | 
|  | offset = EOB_COEF_CONTEXTS_2D + EOB_COEF_CONTEXTS_1D; | 
|  | #else | 
|  | (void)tx_type; | 
|  | #endif | 
|  |  | 
|  | if (txs_ctx == TX_4X4) return offset + av1_coeff_band_4x4[coeff_idx]; | 
|  | if (txs_ctx == TX_8X8) return offset + av1_coeff_band_8x8[coeff_idx]; | 
|  | if (txs_ctx == TX_16X16) return offset + av1_coeff_band_16x16[coeff_idx]; | 
|  | if (txs_ctx == TX_32X32) return offset + av1_coeff_band_32x32[coeff_idx]; | 
|  |  | 
|  | assert(0); | 
|  | return 0; | 
|  | } | 
|  |  | 
|  | static INLINE void set_dc_sign(int *cul_level, tran_low_t v) { | 
|  | if (v < 0) | 
|  | *cul_level |= 1 << COEFF_CONTEXT_BITS; | 
|  | else if (v > 0) | 
|  | *cul_level += 2 << COEFF_CONTEXT_BITS; | 
|  | } | 
|  |  | 
|  | static INLINE int get_dc_sign_ctx(int dc_sign) { | 
|  | int dc_sign_ctx = 0; | 
|  | if (dc_sign < 0) | 
|  | dc_sign_ctx = 1; | 
|  | else if (dc_sign > 0) | 
|  | dc_sign_ctx = 2; | 
|  |  | 
|  | return dc_sign_ctx; | 
|  | } | 
|  |  | 
|  | static INLINE void get_txb_ctx(BLOCK_SIZE plane_bsize, TX_SIZE tx_size, | 
|  | int plane, const ENTROPY_CONTEXT *a, | 
|  | const ENTROPY_CONTEXT *l, TXB_CTX *txb_ctx) { | 
|  | const int txb_w_unit = tx_size_wide_unit[tx_size]; | 
|  | const int txb_h_unit = tx_size_high_unit[tx_size]; | 
|  | int ctx_offset = (plane == 0) ? 0 : 7; | 
|  |  | 
|  | if (plane_bsize > txsize_to_bsize[tx_size]) ctx_offset += 3; | 
|  |  | 
|  | int dc_sign = 0; | 
|  | for (int k = 0; k < txb_w_unit; ++k) { | 
|  | int sign = ((uint8_t)a[k]) >> COEFF_CONTEXT_BITS; | 
|  | if (sign == 1) | 
|  | --dc_sign; | 
|  | else if (sign == 2) | 
|  | ++dc_sign; | 
|  | else if (sign != 0) | 
|  | assert(0); | 
|  | } | 
|  |  | 
|  | for (int k = 0; k < txb_h_unit; ++k) { | 
|  | int sign = ((uint8_t)l[k]) >> COEFF_CONTEXT_BITS; | 
|  | if (sign == 1) | 
|  | --dc_sign; | 
|  | else if (sign == 2) | 
|  | ++dc_sign; | 
|  | else if (sign != 0) | 
|  | assert(0); | 
|  | } | 
|  |  | 
|  | txb_ctx->dc_sign_ctx = get_dc_sign_ctx(dc_sign); | 
|  |  | 
|  | if (plane == 0) { | 
|  | int top = 0; | 
|  | int left = 0; | 
|  |  | 
|  | for (int k = 0; k < txb_w_unit; ++k) { | 
|  | top = AOMMAX(top, ((uint8_t)a[k] & COEFF_CONTEXT_MASK)); | 
|  | } | 
|  |  | 
|  | for (int k = 0; k < txb_h_unit; ++k) { | 
|  | left = AOMMAX(left, ((uint8_t)l[k] & COEFF_CONTEXT_MASK)); | 
|  | } | 
|  |  | 
|  | top = AOMMIN(top, 255); | 
|  | left = AOMMIN(left, 255); | 
|  |  | 
|  | if (plane_bsize == txsize_to_bsize[tx_size]) | 
|  | txb_ctx->txb_skip_ctx = 0; | 
|  | else if (top == 0 && left == 0) | 
|  | txb_ctx->txb_skip_ctx = 1; | 
|  | else if (top == 0 || left == 0) | 
|  | txb_ctx->txb_skip_ctx = 2 + (AOMMAX(top, left) > 3); | 
|  | else if (AOMMAX(top, left) <= 3) | 
|  | txb_ctx->txb_skip_ctx = 4; | 
|  | else if (AOMMIN(top, left) <= 3) | 
|  | txb_ctx->txb_skip_ctx = 5; | 
|  | else | 
|  | txb_ctx->txb_skip_ctx = 6; | 
|  | } else { | 
|  | int ctx_base = get_entropy_context(tx_size, a, l); | 
|  | txb_ctx->txb_skip_ctx = ctx_offset + ctx_base; | 
|  | } | 
|  | } | 
|  |  | 
|  | void av1_init_txb_probs(FRAME_CONTEXT *fc); | 
|  |  | 
|  | void av1_init_lv_map(AV1_COMMON *cm); | 
|  |  | 
|  | #endif  // AV1_COMMON_TXB_COMMON_H_ |