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 TX_SIZE_CONTEXTS 2 18 #define SWITCHABLE_FILTERS 3 // number of switchable filters 19 #define SWITCHABLE_FILTER_CONTEXTS (SWITCHABLE_FILTERS + 1) 20 21 // #define MODE_STATS 22 23 struct VP9Common; 24 25 struct tx_probs { 26 vp9_prob p32x32[TX_SIZE_CONTEXTS][TX_SIZES - 1]; 27 vp9_prob p16x16[TX_SIZE_CONTEXTS][TX_SIZES - 2]; 28 vp9_prob p8x8[TX_SIZE_CONTEXTS][TX_SIZES - 3]; 29 }; 30 31 struct tx_counts { 32 unsigned int p32x32[TX_SIZE_CONTEXTS][TX_SIZES]; 33 unsigned int p16x16[TX_SIZE_CONTEXTS][TX_SIZES - 1]; 34 unsigned int p8x8[TX_SIZE_CONTEXTS][TX_SIZES - 2]; 35 }; 36 37 extern const vp9_prob vp9_kf_uv_mode_prob[INTRA_MODES][INTRA_MODES - 1]; 38 extern const vp9_prob vp9_kf_y_mode_prob[INTRA_MODES][INTRA_MODES] 39 [INTRA_MODES - 1]; 40 41 extern const vp9_prob vp9_kf_partition_probs[PARTITION_CONTEXTS] 42 [PARTITION_TYPES - 1]; 43 44 extern const vp9_tree_index vp9_intra_mode_tree[TREE_SIZE(INTRA_MODES)]; 45 extern struct vp9_token vp9_intra_mode_encodings[INTRA_MODES]; 46 47 extern const vp9_tree_index vp9_inter_mode_tree[TREE_SIZE(INTER_MODES)]; 48 extern struct vp9_token vp9_inter_mode_encodings[INTER_MODES]; 49 50 extern const vp9_tree_index vp9_partition_tree[TREE_SIZE(PARTITION_TYPES)]; 51 extern struct vp9_token vp9_partition_encodings[PARTITION_TYPES]; 52 53 extern const vp9_tree_index vp9_switchable_interp_tree 54 [TREE_SIZE(SWITCHABLE_FILTERS)]; 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(const unsigned int *tx_count_32x32p, 66 unsigned int (*ct_32x32p)[2]); 67 void tx_counts_to_branch_counts_16x16(const unsigned int *tx_count_16x16p, 68 unsigned int (*ct_16x16p)[2]); 69 void tx_counts_to_branch_counts_8x8(const unsigned int *tx_count_8x8p, 70 unsigned int (*ct_8x8p)[2]); 71 72 #endif // VP9_COMMON_VP9_ENTROPYMODE_H_ 73