michael@0: /* michael@0: * Copyright (c) 2010 The WebM project authors. All Rights Reserved. michael@0: * michael@0: * Use of this source code is governed by a BSD-style license michael@0: * that can be found in the LICENSE file in the root of the source michael@0: * tree. An additional intellectual property rights grant can be found michael@0: * in the file PATENTS. All contributing project authors may michael@0: * be found in the AUTHORS file in the root of the source tree. michael@0: */ michael@0: michael@0: #ifndef VP9_COMMON_VP9_ENTROPY_H_ michael@0: #define VP9_COMMON_VP9_ENTROPY_H_ michael@0: michael@0: #include "vpx/vpx_integer.h" michael@0: michael@0: #include "vp9/common/vp9_blockd.h" michael@0: #include "vp9/common/vp9_common.h" michael@0: #include "vp9/common/vp9_scan.h" michael@0: #include "vp9/common/vp9_treecoder.h" michael@0: michael@0: #define DIFF_UPDATE_PROB 252 michael@0: michael@0: /* Coefficient token alphabet */ michael@0: michael@0: #define ZERO_TOKEN 0 /* 0 Extra Bits 0+0 */ michael@0: #define ONE_TOKEN 1 /* 1 Extra Bits 0+1 */ michael@0: #define TWO_TOKEN 2 /* 2 Extra Bits 0+1 */ michael@0: #define THREE_TOKEN 3 /* 3 Extra Bits 0+1 */ michael@0: #define FOUR_TOKEN 4 /* 4 Extra Bits 0+1 */ michael@0: #define DCT_VAL_CATEGORY1 5 /* 5-6 Extra Bits 1+1 */ michael@0: #define DCT_VAL_CATEGORY2 6 /* 7-10 Extra Bits 2+1 */ michael@0: #define DCT_VAL_CATEGORY3 7 /* 11-18 Extra Bits 3+1 */ michael@0: #define DCT_VAL_CATEGORY4 8 /* 19-34 Extra Bits 4+1 */ michael@0: #define DCT_VAL_CATEGORY5 9 /* 35-66 Extra Bits 5+1 */ michael@0: #define DCT_VAL_CATEGORY6 10 /* 67+ Extra Bits 14+1 */ michael@0: #define DCT_EOB_TOKEN 11 /* EOB Extra Bits 0+0 */ michael@0: #define MAX_ENTROPY_TOKENS 12 michael@0: #define ENTROPY_NODES 11 michael@0: #define EOSB_TOKEN 127 /* Not signalled, encoder only */ michael@0: michael@0: #define INTER_MODE_CONTEXTS 7 michael@0: michael@0: extern DECLARE_ALIGNED(16, const uint8_t, michael@0: vp9_pt_energy_class[MAX_ENTROPY_TOKENS]); michael@0: michael@0: extern const vp9_tree_index vp9_coef_tree[TREE_SIZE(MAX_ENTROPY_TOKENS)]; michael@0: michael@0: #define DCT_EOB_MODEL_TOKEN 3 /* EOB Extra Bits 0+0 */ michael@0: extern const vp9_tree_index vp9_coefmodel_tree[]; michael@0: michael@0: extern struct vp9_token vp9_coef_encodings[MAX_ENTROPY_TOKENS]; michael@0: michael@0: typedef struct { michael@0: const vp9_tree_index *tree; michael@0: const vp9_prob *prob; michael@0: int len; michael@0: int base_val; michael@0: } vp9_extra_bit; michael@0: michael@0: // indexed by token value michael@0: extern const vp9_extra_bit vp9_extra_bits[MAX_ENTROPY_TOKENS]; michael@0: michael@0: #define MAX_PROB 255 michael@0: #define DCT_MAX_VALUE 16384 michael@0: michael@0: /* Coefficients are predicted via a 3-dimensional probability table. */ michael@0: michael@0: /* Outside dimension. 0 = Y with DC, 1 = UV */ michael@0: #define BLOCK_TYPES 2 michael@0: #define REF_TYPES 2 // intra=0, inter=1 michael@0: michael@0: /* Middle dimension reflects the coefficient position within the transform. */ michael@0: #define COEF_BANDS 6 michael@0: michael@0: /* Inside dimension is measure of nearby complexity, that reflects the energy michael@0: of nearby coefficients are nonzero. For the first coefficient (DC, unless michael@0: block type is 0), we look at the (already encoded) blocks above and to the michael@0: left of the current block. The context index is then the number (0,1,or 2) michael@0: of these blocks having nonzero coefficients. michael@0: After decoding a coefficient, the measure is determined by the size of the michael@0: most recently decoded coefficient. michael@0: Note that the intuitive meaning of this measure changes as coefficients michael@0: are decoded, e.g., prior to the first token, a zero means that my neighbors michael@0: are empty while, after the first token, because of the use of end-of-block, michael@0: a zero means we just decoded a zero and hence guarantees that a non-zero michael@0: coefficient will appear later in this block. However, this shift michael@0: in meaning is perfectly OK because our context depends also on the michael@0: coefficient band (and since zigzag positions 0, 1, and 2 are in michael@0: distinct bands). */ michael@0: michael@0: #define PREV_COEF_CONTEXTS 6 michael@0: michael@0: // #define ENTROPY_STATS michael@0: michael@0: typedef unsigned int vp9_coeff_count[REF_TYPES][COEF_BANDS][PREV_COEF_CONTEXTS] michael@0: [MAX_ENTROPY_TOKENS]; michael@0: typedef unsigned int vp9_coeff_stats[REF_TYPES][COEF_BANDS][PREV_COEF_CONTEXTS] michael@0: [ENTROPY_NODES][2]; michael@0: michael@0: #define SUBEXP_PARAM 4 /* Subexponential code parameter */ michael@0: #define MODULUS_PARAM 13 /* Modulus parameter */ michael@0: michael@0: struct VP9Common; michael@0: void vp9_default_coef_probs(struct VP9Common *cm); michael@0: michael@0: void vp9_coef_tree_initialize(); michael@0: void vp9_adapt_coef_probs(struct VP9Common *cm); michael@0: michael@0: static INLINE void reset_skip_context(MACROBLOCKD *xd, BLOCK_SIZE bsize) { michael@0: int i; michael@0: for (i = 0; i < MAX_MB_PLANE; i++) { michael@0: struct macroblockd_plane *const pd = &xd->plane[i]; michael@0: const BLOCK_SIZE plane_bsize = get_plane_block_size(bsize, pd); michael@0: vpx_memset(pd->above_context, 0, sizeof(ENTROPY_CONTEXT) * michael@0: num_4x4_blocks_wide_lookup[plane_bsize]); michael@0: vpx_memset(pd->left_context, 0, sizeof(ENTROPY_CONTEXT) * michael@0: num_4x4_blocks_high_lookup[plane_bsize]); michael@0: } michael@0: } michael@0: michael@0: // This is the index in the scan order beyond which all coefficients for michael@0: // 8x8 transform and above are in the top band. michael@0: // This macro is currently unused but may be used by certain implementations michael@0: #define MAXBAND_INDEX 21 michael@0: michael@0: extern const uint8_t vp9_coefband_trans_8x8plus[1024]; michael@0: extern const uint8_t vp9_coefband_trans_4x4[16]; michael@0: michael@0: static const uint8_t *get_band_translate(TX_SIZE tx_size) { michael@0: return tx_size == TX_4X4 ? vp9_coefband_trans_4x4 michael@0: : vp9_coefband_trans_8x8plus; michael@0: } michael@0: michael@0: // 128 lists of probabilities are stored for the following ONE node probs: michael@0: // 1, 3, 5, 7, ..., 253, 255 michael@0: // In between probabilities are interpolated linearly michael@0: michael@0: #define COEFPROB_MODELS 128 michael@0: michael@0: #define UNCONSTRAINED_NODES 3 michael@0: michael@0: #define PIVOT_NODE 2 // which node is pivot michael@0: michael@0: typedef vp9_prob vp9_coeff_probs_model[REF_TYPES][COEF_BANDS] michael@0: [PREV_COEF_CONTEXTS] michael@0: [UNCONSTRAINED_NODES]; michael@0: michael@0: typedef unsigned int vp9_coeff_count_model[REF_TYPES][COEF_BANDS] michael@0: [PREV_COEF_CONTEXTS] michael@0: [UNCONSTRAINED_NODES + 1]; michael@0: michael@0: void vp9_model_to_full_probs(const vp9_prob *model, vp9_prob *full); michael@0: michael@0: static int get_entropy_context(TX_SIZE tx_size, const ENTROPY_CONTEXT *a, michael@0: const ENTROPY_CONTEXT *l) { michael@0: ENTROPY_CONTEXT above_ec = 0, left_ec = 0; michael@0: michael@0: switch (tx_size) { michael@0: case TX_4X4: michael@0: above_ec = a[0] != 0; michael@0: left_ec = l[0] != 0; michael@0: break; michael@0: case TX_8X8: michael@0: above_ec = !!*(const uint16_t *)a; michael@0: left_ec = !!*(const uint16_t *)l; michael@0: break; michael@0: case TX_16X16: michael@0: above_ec = !!*(const uint32_t *)a; michael@0: left_ec = !!*(const uint32_t *)l; michael@0: break; michael@0: case TX_32X32: michael@0: above_ec = !!*(const uint64_t *)a; michael@0: left_ec = !!*(const uint64_t *)l; michael@0: break; michael@0: default: michael@0: assert(!"Invalid transform size."); michael@0: } michael@0: michael@0: return combine_entropy_contexts(above_ec, left_ec); michael@0: } michael@0: michael@0: static void get_scan(const MACROBLOCKD *xd, TX_SIZE tx_size, michael@0: PLANE_TYPE type, int block_idx, michael@0: const int16_t **scan, const int16_t **scan_nb) { michael@0: switch (tx_size) { michael@0: case TX_4X4: michael@0: get_scan_nb_4x4(get_tx_type_4x4(type, xd, block_idx), scan, scan_nb); michael@0: break; michael@0: case TX_8X8: michael@0: get_scan_nb_8x8(get_tx_type_8x8(type, xd), scan, scan_nb); michael@0: break; michael@0: case TX_16X16: michael@0: get_scan_nb_16x16(get_tx_type_16x16(type, xd), scan, scan_nb); michael@0: break; michael@0: case TX_32X32: michael@0: *scan = vp9_default_scan_32x32; michael@0: *scan_nb = vp9_default_scan_32x32_neighbors; michael@0: break; michael@0: default: michael@0: assert(!"Invalid transform size."); michael@0: } michael@0: } michael@0: michael@0: #endif // VP9_COMMON_VP9_ENTROPY_H_