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
12 #include "entropymode.h"
13 #include "entropy.h"
14 #include "vpx_mem/vpx_mem.h"
15
16 static const unsigned int kf_y_mode_cts[VP8_YMODES] = { 1607, 915, 812, 811, 5455};
17 static const unsigned int y_mode_cts [VP8_YMODES] = { 8080, 1908, 1582, 1007, 5874};
18
19 static const unsigned int uv_mode_cts [VP8_UV_MODES] = { 59483, 13605, 16492, 4230};
20 static const unsigned int kf_uv_mode_cts[VP8_UV_MODES] = { 5319, 1904, 1703, 674};
21
22 static const unsigned int bmode_cts[VP8_BINTRAMODES] =
23 {
24 43891, 17694, 10036, 3920, 3363, 2546, 5119, 3221, 2471, 1723
25 };
26
27 typedef enum
28 {
29 SUBMVREF_NORMAL,
30 SUBMVREF_LEFT_ZED,
31 SUBMVREF_ABOVE_ZED,
32 SUBMVREF_LEFT_ABOVE_SAME,
33 SUBMVREF_LEFT_ABOVE_ZED
34 } sumvfref_t;
35
vp8_mv_cont(const MV * l,const MV * a)36 int vp8_mv_cont(const MV *l, const MV *a)
37 {
38 int lez = (l->row == 0 && l->col == 0);
39 int aez = (a->row == 0 && a->col == 0);
40 int lea = (l->row == a->row && l->col == a->col);
41
42 if (lea && lez)
43 return SUBMVREF_LEFT_ABOVE_ZED;
44
45 if (lea)
46 return SUBMVREF_LEFT_ABOVE_SAME;
47
48 if (aez)
49 return SUBMVREF_ABOVE_ZED;
50
51 if (lez)
52 return SUBMVREF_LEFT_ZED;
53
54 return SUBMVREF_NORMAL;
55 }
56
57 static const vp8_prob sub_mv_ref_prob [VP8_SUBMVREFS-1] = { 180, 162, 25};
58
59 const vp8_prob vp8_sub_mv_ref_prob2 [SUBMVREF_COUNT][VP8_SUBMVREFS-1] =
60 {
61 { 147, 136, 18 },
62 { 106, 145, 1 },
63 { 179, 121, 1 },
64 { 223, 1 , 34 },
65 { 208, 1 , 1 }
66 };
67
68
69
70 vp8_mbsplit vp8_mbsplits [VP8_NUMMBSPLITS] =
71 {
72 {
73 0, 0, 0, 0,
74 0, 0, 0, 0,
75 1, 1, 1, 1,
76 1, 1, 1, 1,
77 },
78 {
79 0, 0, 1, 1,
80 0, 0, 1, 1,
81 0, 0, 1, 1,
82 0, 0, 1, 1,
83 },
84 {
85 0, 0, 1, 1,
86 0, 0, 1, 1,
87 2, 2, 3, 3,
88 2, 2, 3, 3,
89 },
90 {
91 0, 1, 2, 3,
92 4, 5, 6, 7,
93 8, 9, 10, 11,
94 12, 13, 14, 15,
95 },
96 };
97
98 const int vp8_mbsplit_count [VP8_NUMMBSPLITS] = { 2, 2, 4, 16};
99
100 const vp8_prob vp8_mbsplit_probs [VP8_NUMMBSPLITS-1] = { 110, 111, 150};
101
102
103 /* Array indices are identical to previously-existing INTRAMODECONTEXTNODES. */
104
105 const vp8_tree_index vp8_bmode_tree[18] = /* INTRAMODECONTEXTNODE value */
106 {
107 -B_DC_PRED, 2, /* 0 = DC_NODE */
108 -B_TM_PRED, 4, /* 1 = TM_NODE */
109 -B_VE_PRED, 6, /* 2 = VE_NODE */
110 8, 12, /* 3 = COM_NODE */
111 -B_HE_PRED, 10, /* 4 = HE_NODE */
112 -B_RD_PRED, -B_VR_PRED, /* 5 = RD_NODE */
113 -B_LD_PRED, 14, /* 6 = LD_NODE */
114 -B_VL_PRED, 16, /* 7 = VL_NODE */
115 -B_HD_PRED, -B_HU_PRED /* 8 = HD_NODE */
116 };
117
118 /* Again, these trees use the same probability indices as their
119 explicitly-programmed predecessors. */
120
121 const vp8_tree_index vp8_ymode_tree[8] =
122 {
123 -DC_PRED, 2,
124 4, 6,
125 -V_PRED, -H_PRED,
126 -TM_PRED, -B_PRED
127 };
128
129 const vp8_tree_index vp8_kf_ymode_tree[8] =
130 {
131 -B_PRED, 2,
132 4, 6,
133 -DC_PRED, -V_PRED,
134 -H_PRED, -TM_PRED
135 };
136
137 const vp8_tree_index vp8_uv_mode_tree[6] =
138 {
139 -DC_PRED, 2,
140 -V_PRED, 4,
141 -H_PRED, -TM_PRED
142 };
143
144 const vp8_tree_index vp8_mbsplit_tree[6] =
145 {
146 -3, 2,
147 -2, 4,
148 -0, -1
149 };
150
151 const vp8_tree_index vp8_mv_ref_tree[8] =
152 {
153 -ZEROMV, 2,
154 -NEARESTMV, 4,
155 -NEARMV, 6,
156 -NEWMV, -SPLITMV
157 };
158
159 const vp8_tree_index vp8_sub_mv_ref_tree[6] =
160 {
161 -LEFT4X4, 2,
162 -ABOVE4X4, 4,
163 -ZERO4X4, -NEW4X4
164 };
165
166
167 struct vp8_token_struct vp8_bmode_encodings [VP8_BINTRAMODES];
168 struct vp8_token_struct vp8_ymode_encodings [VP8_YMODES];
169 struct vp8_token_struct vp8_kf_ymode_encodings [VP8_YMODES];
170 struct vp8_token_struct vp8_uv_mode_encodings [VP8_UV_MODES];
171 struct vp8_token_struct vp8_mbsplit_encodings [VP8_NUMMBSPLITS];
172
173 struct vp8_token_struct vp8_mv_ref_encoding_array [VP8_MVREFS];
174 struct vp8_token_struct vp8_sub_mv_ref_encoding_array [VP8_SUBMVREFS];
175
176
177 const vp8_tree_index vp8_small_mvtree [14] =
178 {
179 2, 8,
180 4, 6,
181 -0, -1,
182 -2, -3,
183 10, 12,
184 -4, -5,
185 -6, -7
186 };
187
188 struct vp8_token_struct vp8_small_mvencodings [8];
189
vp8_init_mbmode_probs(VP8_COMMON * x)190 void vp8_init_mbmode_probs(VP8_COMMON *x)
191 {
192 unsigned int bct [VP8_YMODES] [2]; /* num Ymodes > num UV modes */
193
194 vp8_tree_probs_from_distribution(
195 VP8_YMODES, vp8_ymode_encodings, vp8_ymode_tree,
196 x->fc.ymode_prob, bct, y_mode_cts,
197 256, 1
198 );
199 vp8_tree_probs_from_distribution(
200 VP8_YMODES, vp8_kf_ymode_encodings, vp8_kf_ymode_tree,
201 x->kf_ymode_prob, bct, kf_y_mode_cts,
202 256, 1
203 );
204 vp8_tree_probs_from_distribution(
205 VP8_UV_MODES, vp8_uv_mode_encodings, vp8_uv_mode_tree,
206 x->fc.uv_mode_prob, bct, uv_mode_cts,
207 256, 1
208 );
209 vp8_tree_probs_from_distribution(
210 VP8_UV_MODES, vp8_uv_mode_encodings, vp8_uv_mode_tree,
211 x->kf_uv_mode_prob, bct, kf_uv_mode_cts,
212 256, 1
213 );
214 vpx_memcpy(x->fc.sub_mv_ref_prob, sub_mv_ref_prob, sizeof(sub_mv_ref_prob));
215 }
216
217
intra_bmode_probs_from_distribution(vp8_prob p[VP8_BINTRAMODES-1],unsigned int branch_ct[VP8_BINTRAMODES-1][2],const unsigned int events[VP8_BINTRAMODES])218 static void intra_bmode_probs_from_distribution(
219 vp8_prob p [VP8_BINTRAMODES-1],
220 unsigned int branch_ct [VP8_BINTRAMODES-1] [2],
221 const unsigned int events [VP8_BINTRAMODES]
222 )
223 {
224 vp8_tree_probs_from_distribution(
225 VP8_BINTRAMODES, vp8_bmode_encodings, vp8_bmode_tree,
226 p, branch_ct, events,
227 256, 1
228 );
229 }
230
vp8_default_bmode_probs(vp8_prob p[VP8_BINTRAMODES-1])231 void vp8_default_bmode_probs(vp8_prob p [VP8_BINTRAMODES-1])
232 {
233 unsigned int branch_ct [VP8_BINTRAMODES-1] [2];
234 intra_bmode_probs_from_distribution(p, branch_ct, bmode_cts);
235 }
236
vp8_kf_default_bmode_probs(vp8_prob p[VP8_BINTRAMODES][VP8_BINTRAMODES][VP8_BINTRAMODES-1])237 void vp8_kf_default_bmode_probs(vp8_prob p [VP8_BINTRAMODES] [VP8_BINTRAMODES] [VP8_BINTRAMODES-1])
238 {
239 unsigned int branch_ct [VP8_BINTRAMODES-1] [2];
240
241 int i = 0;
242
243 do
244 {
245 int j = 0;
246
247 do
248 {
249 intra_bmode_probs_from_distribution(
250 p[i][j], branch_ct, vp8_kf_default_bmode_counts[i][j]);
251
252 }
253 while (++j < VP8_BINTRAMODES);
254 }
255 while (++i < VP8_BINTRAMODES);
256 }
257
258
vp8_entropy_mode_init()259 void vp8_entropy_mode_init()
260 {
261 vp8_tokens_from_tree(vp8_bmode_encodings, vp8_bmode_tree);
262 vp8_tokens_from_tree(vp8_ymode_encodings, vp8_ymode_tree);
263 vp8_tokens_from_tree(vp8_kf_ymode_encodings, vp8_kf_ymode_tree);
264 vp8_tokens_from_tree(vp8_uv_mode_encodings, vp8_uv_mode_tree);
265 vp8_tokens_from_tree(vp8_mbsplit_encodings, vp8_mbsplit_tree);
266
267 vp8_tokens_from_tree_offset(vp8_mv_ref_encoding_array,
268 vp8_mv_ref_tree, NEARESTMV);
269 vp8_tokens_from_tree_offset(vp8_sub_mv_ref_encoding_array,
270 vp8_sub_mv_ref_tree, LEFT4X4);
271
272 vp8_tokens_from_tree(vp8_small_mvencodings, vp8_small_mvtree);
273 }
274