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 #ifndef VP9_COMMON_VP9_ENTROPYMODE_H_ 12 #define VP9_COMMON_VP9_ENTROPYMODE_H_ 13 14 #include "vp9/common/vp9_blockd.h" 15 #include "vp9/common/vp9_treecoder.h" 16 17 #define SUBMVREF_COUNT 5 18 #define TX_SIZE_CONTEXTS 2 19 #define MODE_UPDATE_PROB 252 20 #define SWITCHABLE_FILTERS 3 // number of switchable filters 21 22 // #define MODE_STATS 23 24 struct VP9Common; 25 26 struct tx_probs { 27 vp9_prob p32x32[TX_SIZE_CONTEXTS][TX_SIZES - 1]; 28 vp9_prob p16x16[TX_SIZE_CONTEXTS][TX_SIZES - 2]; 29 vp9_prob p8x8[TX_SIZE_CONTEXTS][TX_SIZES - 3]; 30 }; 31 32 struct tx_counts { 33 unsigned int p32x32[TX_SIZE_CONTEXTS][TX_SIZES]; 34 unsigned int p16x16[TX_SIZE_CONTEXTS][TX_SIZES - 1]; 35 unsigned int p8x8[TX_SIZE_CONTEXTS][TX_SIZES - 2]; 36 }; 37 38 extern const vp9_prob vp9_kf_uv_mode_prob[INTRA_MODES][INTRA_MODES - 1]; 39 extern const vp9_prob vp9_kf_y_mode_prob[INTRA_MODES][INTRA_MODES] 40 [INTRA_MODES - 1]; 41 42 extern const vp9_tree_index vp9_intra_mode_tree[]; 43 extern const vp9_tree_index vp9_inter_mode_tree[]; 44 45 extern struct vp9_token vp9_intra_mode_encodings[INTRA_MODES]; 46 extern struct vp9_token vp9_inter_mode_encodings[INTER_MODES]; 47 48 // probability models for partition information 49 extern const vp9_tree_index vp9_partition_tree[]; 50 extern struct vp9_token vp9_partition_encodings[PARTITION_TYPES]; 51 52 extern const vp9_tree_index vp9_switchable_interp_tree 53 [2 * (SWITCHABLE_FILTERS - 1)]; 54 55 extern struct vp9_token vp9_switchable_interp_encodings[SWITCHABLE_FILTERS]; 56 57 void vp9_entropy_mode_init(); 58 59 void vp9_setup_past_independence(struct VP9Common *cm); 60 61 void vp9_init_mbmode_probs(struct VP9Common *cm); 62 63 void vp9_adapt_mode_probs(struct VP9Common *cm); 64 65 void tx_counts_to_branch_counts_32x32(unsigned int *tx_count_32x32p, 66 unsigned int (*ct_32x32p)[2]); 67 void tx_counts_to_branch_counts_16x16(unsigned int *tx_count_16x16p, 68 unsigned int (*ct_16x16p)[2]); 69 void tx_counts_to_branch_counts_8x8(unsigned int *tx_count_8x8p, 70 unsigned int (*ct_8x8p)[2]); 71 72 #endif // VP9_COMMON_VP9_ENTROPYMODE_H_ 73