OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved. | 2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #ifndef VP9_COMMON_VP9_ENTROPYMODE_H_ | 11 #ifndef VP9_COMMON_VP9_ENTROPYMODE_H_ |
12 #define VP9_COMMON_VP9_ENTROPYMODE_H_ | 12 #define VP9_COMMON_VP9_ENTROPYMODE_H_ |
13 | 13 |
14 #include "vp9/common/vp9_blockd.h" | 14 #include "vp9/common/vp9_blockd.h" |
15 #include "vp9/common/vp9_treecoder.h" | 15 #include "vp9/common/vp9_treecoder.h" |
16 | 16 |
17 #define SUBMVREF_COUNT 5 | |
18 #define TX_SIZE_CONTEXTS 2 | 17 #define TX_SIZE_CONTEXTS 2 |
19 #define MODE_UPDATE_PROB 252 | |
20 #define SWITCHABLE_FILTERS 3 // number of switchable filters | 18 #define SWITCHABLE_FILTERS 3 // number of switchable filters |
| 19 #define SWITCHABLE_FILTER_CONTEXTS (SWITCHABLE_FILTERS + 1) |
21 | 20 |
22 // #define MODE_STATS | 21 // #define MODE_STATS |
23 | 22 |
24 struct VP9Common; | 23 struct VP9Common; |
25 | 24 |
26 struct tx_probs { | 25 struct tx_probs { |
27 vp9_prob p32x32[TX_SIZE_CONTEXTS][TX_SIZES - 1]; | 26 vp9_prob p32x32[TX_SIZE_CONTEXTS][TX_SIZES - 1]; |
28 vp9_prob p16x16[TX_SIZE_CONTEXTS][TX_SIZES - 2]; | 27 vp9_prob p16x16[TX_SIZE_CONTEXTS][TX_SIZES - 2]; |
29 vp9_prob p8x8[TX_SIZE_CONTEXTS][TX_SIZES - 3]; | 28 vp9_prob p8x8[TX_SIZE_CONTEXTS][TX_SIZES - 3]; |
30 }; | 29 }; |
31 | 30 |
32 struct tx_counts { | 31 struct tx_counts { |
33 unsigned int p32x32[TX_SIZE_CONTEXTS][TX_SIZES]; | 32 unsigned int p32x32[TX_SIZE_CONTEXTS][TX_SIZES]; |
34 unsigned int p16x16[TX_SIZE_CONTEXTS][TX_SIZES - 1]; | 33 unsigned int p16x16[TX_SIZE_CONTEXTS][TX_SIZES - 1]; |
35 unsigned int p8x8[TX_SIZE_CONTEXTS][TX_SIZES - 2]; | 34 unsigned int p8x8[TX_SIZE_CONTEXTS][TX_SIZES - 2]; |
36 }; | 35 }; |
37 | 36 |
38 extern const vp9_prob vp9_kf_uv_mode_prob[INTRA_MODES][INTRA_MODES - 1]; | 37 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] | 38 extern const vp9_prob vp9_kf_y_mode_prob[INTRA_MODES][INTRA_MODES] |
40 [INTRA_MODES - 1]; | 39 [INTRA_MODES - 1]; |
41 | 40 |
42 extern const vp9_tree_index vp9_intra_mode_tree[]; | 41 extern const vp9_tree_index vp9_intra_mode_tree[TREE_SIZE(INTRA_MODES)]; |
43 extern const vp9_tree_index vp9_inter_mode_tree[]; | 42 extern struct vp9_token vp9_intra_mode_encodings[INTRA_MODES]; |
44 | 43 |
45 extern struct vp9_token vp9_intra_mode_encodings[INTRA_MODES]; | 44 extern const vp9_tree_index vp9_inter_mode_tree[TREE_SIZE(INTER_MODES)]; |
46 extern struct vp9_token vp9_inter_mode_encodings[INTER_MODES]; | 45 extern struct vp9_token vp9_inter_mode_encodings[INTER_MODES]; |
47 | 46 |
48 // probability models for partition information | 47 extern const vp9_tree_index vp9_partition_tree[TREE_SIZE(PARTITION_TYPES)]; |
49 extern const vp9_tree_index vp9_partition_tree[]; | |
50 extern struct vp9_token vp9_partition_encodings[PARTITION_TYPES]; | 48 extern struct vp9_token vp9_partition_encodings[PARTITION_TYPES]; |
51 | 49 |
52 extern const vp9_tree_index vp9_switchable_interp_tree | 50 extern const vp9_tree_index vp9_switchable_interp_tree |
53 [2 * (SWITCHABLE_FILTERS - 1)]; | 51 [TREE_SIZE(SWITCHABLE_FILTERS)]; |
54 | |
55 extern struct vp9_token vp9_switchable_interp_encodings[SWITCHABLE_FILTERS]; | 52 extern struct vp9_token vp9_switchable_interp_encodings[SWITCHABLE_FILTERS]; |
56 | 53 |
57 void vp9_entropy_mode_init(); | 54 void vp9_entropy_mode_init(); |
58 | 55 |
59 void vp9_setup_past_independence(struct VP9Common *cm); | 56 void vp9_setup_past_independence(struct VP9Common *cm); |
60 | 57 |
61 void vp9_init_mbmode_probs(struct VP9Common *cm); | 58 void vp9_init_mbmode_probs(struct VP9Common *cm); |
62 | 59 |
63 void vp9_adapt_mode_probs(struct VP9Common *cm); | 60 void vp9_adapt_mode_probs(struct VP9Common *cm); |
64 | 61 |
65 void tx_counts_to_branch_counts_32x32(unsigned int *tx_count_32x32p, | 62 void tx_counts_to_branch_counts_32x32(const unsigned int *tx_count_32x32p, |
66 unsigned int (*ct_32x32p)[2]); | 63 unsigned int (*ct_32x32p)[2]); |
67 void tx_counts_to_branch_counts_16x16(unsigned int *tx_count_16x16p, | 64 void tx_counts_to_branch_counts_16x16(const unsigned int *tx_count_16x16p, |
68 unsigned int (*ct_16x16p)[2]); | 65 unsigned int (*ct_16x16p)[2]); |
69 void tx_counts_to_branch_counts_8x8(unsigned int *tx_count_8x8p, | 66 void tx_counts_to_branch_counts_8x8(const unsigned int *tx_count_8x8p, |
70 unsigned int (*ct_8x8p)[2]); | 67 unsigned int (*ct_8x8p)[2]); |
71 | 68 |
72 #endif // VP9_COMMON_VP9_ENTROPYMODE_H_ | 69 #endif // VP9_COMMON_VP9_ENTROPYMODE_H_ |
OLD | NEW |