The Tor Browser / file comparison

comparison: media/libvpx/vp9/common/vp9_entropymv.c

media/libvpx/vp9/common/vp9_entropymv.c

changeset 0
6474c204b198
equal deleted inserted replaced
-1:000000000000 0:5c91effe50d8
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 "vp9/common/vp9_onyxc_int.h"
13 #include "vp9/common/vp9_entropymv.h"
14
15 #define MV_COUNT_SAT 20
16 #define MV_MAX_UPDATE_FACTOR 128
17
18 /* Integer pel reference mv threshold for use of high-precision 1/8 mv */
19 #define COMPANDED_MVREF_THRESH 8
20
21 const vp9_tree_index vp9_mv_joint_tree[TREE_SIZE(MV_JOINTS)] = {
22 -MV_JOINT_ZERO, 2,
23 -MV_JOINT_HNZVZ, 4,
24 -MV_JOINT_HZVNZ, -MV_JOINT_HNZVNZ
25 };
26 struct vp9_token vp9_mv_joint_encodings[MV_JOINTS];
27
28 const vp9_tree_index vp9_mv_class_tree[TREE_SIZE(MV_CLASSES)] = {
29 -MV_CLASS_0, 2,
30 -MV_CLASS_1, 4,
31 6, 8,
32 -MV_CLASS_2, -MV_CLASS_3,
33 10, 12,
34 -MV_CLASS_4, -MV_CLASS_5,
35 -MV_CLASS_6, 14,
36 16, 18,
37 -MV_CLASS_7, -MV_CLASS_8,
38 -MV_CLASS_9, -MV_CLASS_10,
39 };
40 struct vp9_token vp9_mv_class_encodings[MV_CLASSES];
41
42 const vp9_tree_index vp9_mv_class0_tree[TREE_SIZE(CLASS0_SIZE)] = {
43 -0, -1,
44 };
45 struct vp9_token vp9_mv_class0_encodings[CLASS0_SIZE];
46
47 const vp9_tree_index vp9_mv_fp_tree[TREE_SIZE(4)] = {
48 -0, 2,
49 -1, 4,
50 -2, -3
51 };
52 struct vp9_token vp9_mv_fp_encodings[4];
53
54 static const nmv_context default_nmv_context = {
55 {32, 64, 96},
56 { // NOLINT
57 { /* vert component */ // NOLINT
58 128, /* sign */
59 {224, 144, 192, 168, 192, 176, 192, 198, 198, 245}, /* class */
60 {216}, /* class0 */
61 {136, 140, 148, 160, 176, 192, 224, 234, 234, 240}, /* bits */
62 {{128, 128, 64}, {96, 112, 64}}, /* class0_fp */
63 {64, 96, 64}, /* fp */
64 160, /* class0_hp bit */
65 128, /* hp */
66 },
67 { /* hor component */ // NOLINT
68 128, /* sign */
69 {216, 128, 176, 160, 176, 176, 192, 198, 198, 208}, /* class */
70 {208}, /* class0 */
71 {136, 140, 148, 160, 176, 192, 224, 234, 234, 240}, /* bits */
72 {{128, 128, 64}, {96, 112, 64}}, /* class0_fp */
73 {64, 96, 64}, /* fp */
74 160, /* class0_hp bit */
75 128, /* hp */
76 }
77 },
78 };
79
80 #define mv_class_base(c) ((c) ? (CLASS0_SIZE << (c + 2)) : 0)
81
82 static const uint8_t log_in_base_2[] = {
83 0, 0, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3, 3, 4, 4, 4, 4, 4, 4, 4, 4,
84 4, 4, 4, 4, 4, 4, 4, 4, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
85 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 6, 6, 6, 6, 6, 6, 6, 6,
86 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
87 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
88 6, 6, 6, 6, 6, 6, 6, 6, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
89 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
90 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
91 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
92 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
93 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 8, 8, 8, 8, 8, 8, 8, 8,
94 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
95 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
96 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
97 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
98 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
99 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
100 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
101 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
102 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
103 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
104 8, 8, 8, 8, 8, 8, 8, 8, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9,
105 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9,
106 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9,
107 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9,
108 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9,
109 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9,
110 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9,
111 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9,
112 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9,
113 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9,
114 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9,
115 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9,
116 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9,
117 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9,
118 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9,
119 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9,
120 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9,
121 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9,
122 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9,
123 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9,
124 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9,
125 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 10
126 };
127
128 MV_CLASS_TYPE vp9_get_mv_class(int z, int *offset) {
129 MV_CLASS_TYPE c = MV_CLASS_0;
130 if (z >= CLASS0_SIZE * 4096)
131 c = MV_CLASS_10;
132 else
133 c = log_in_base_2[z >> 3];
134
135 if (offset)
136 *offset = z - mv_class_base(c);
137 return c;
138 }
139
140 int vp9_use_mv_hp(const MV *ref) {
141 return (abs(ref->row) >> 3) < COMPANDED_MVREF_THRESH &&
142 (abs(ref->col) >> 3) < COMPANDED_MVREF_THRESH;
143 }
144
145 int vp9_get_mv_mag(MV_CLASS_TYPE c, int offset) {
146 return mv_class_base(c) + offset;
147 }
148
149 static void inc_mv_component(int v, nmv_component_counts *comp_counts,
150 int incr, int usehp) {
151 int s, z, c, o, d, e, f;
152 assert(v != 0); /* should not be zero */
153 s = v < 0;
154 comp_counts->sign[s] += incr;
155 z = (s ? -v : v) - 1; /* magnitude - 1 */
156
157 c = vp9_get_mv_class(z, &o);
158 comp_counts->classes[c] += incr;
159
160 d = (o >> 3); /* int mv data */
161 f = (o >> 1) & 3; /* fractional pel mv data */
162 e = (o & 1); /* high precision mv data */
163
164 if (c == MV_CLASS_0) {
165 comp_counts->class0[d] += incr;
166 comp_counts->class0_fp[d][f] += incr;
167 comp_counts->class0_hp[e] += usehp * incr;
168 } else {
169 int i;
170 int b = c + CLASS0_BITS - 1; // number of bits
171 for (i = 0; i < b; ++i)
172 comp_counts->bits[i][((d >> i) & 1)] += incr;
173 comp_counts->fp[f] += incr;
174 comp_counts->hp[e] += usehp * incr;
175 }
176 }
177
178 void vp9_inc_mv(const MV *mv, nmv_context_counts *counts) {
179 if (counts != NULL) {
180 const MV_JOINT_TYPE j = vp9_get_mv_joint(mv);
181 ++counts->joints[j];
182
183 if (mv_joint_vertical(j)) {
184 inc_mv_component(mv->row, &counts->comps[0], 1, 1);
185 }
186
187 if (mv_joint_horizontal(j)) {
188 inc_mv_component(mv->col, &counts->comps[1], 1, 1);
189 }
190 }
191 }
192
193 static vp9_prob adapt_prob(vp9_prob prep, const unsigned int ct[2]) {
194 return merge_probs(prep, ct, MV_COUNT_SAT, MV_MAX_UPDATE_FACTOR);
195 }
196
197 static void adapt_probs(const vp9_tree_index *tree, const vp9_prob *pre_probs,
198 const unsigned int *counts, vp9_prob *probs) {
199 tree_merge_probs(tree, pre_probs, counts, MV_COUNT_SAT, MV_MAX_UPDATE_FACTOR,
200 probs);
201 }
202
203 void vp9_adapt_mv_probs(VP9_COMMON *cm, int allow_hp) {
204 int i, j;
205
206 nmv_context *fc = &cm->fc.nmvc;
207 const nmv_context *pre_fc = &cm->frame_contexts[cm->frame_context_idx].nmvc;
208 const nmv_context_counts *counts = &cm->counts.mv;
209
210 adapt_probs(vp9_mv_joint_tree, pre_fc->joints, counts->joints, fc->joints);
211
212 for (i = 0; i < 2; ++i) {
213 nmv_component *comp = &fc->comps[i];
214 const nmv_component *pre_comp = &pre_fc->comps[i];
215 const nmv_component_counts *c = &counts->comps[i];
216
217 comp->sign = adapt_prob(pre_comp->sign, c->sign);
218 adapt_probs(vp9_mv_class_tree, pre_comp->classes, c->classes,
219 comp->classes);
220 adapt_probs(vp9_mv_class0_tree, pre_comp->class0, c->class0, comp->class0);
221
222 for (j = 0; j < MV_OFFSET_BITS; ++j)
223 comp->bits[j] = adapt_prob(pre_comp->bits[j], c->bits[j]);
224
225 for (j = 0; j < CLASS0_SIZE; ++j)
226 adapt_probs(vp9_mv_fp_tree, pre_comp->class0_fp[j], c->class0_fp[j],
227 comp->class0_fp[j]);
228
229 adapt_probs(vp9_mv_fp_tree, pre_comp->fp, c->fp, comp->fp);
230
231 if (allow_hp) {
232 comp->class0_hp = adapt_prob(pre_comp->class0_hp, c->class0_hp);
233 comp->hp = adapt_prob(pre_comp->hp, c->hp);
234 }
235 }
236 }
237
238 void vp9_entropy_mv_init() {
239 vp9_tokens_from_tree(vp9_mv_joint_encodings, vp9_mv_joint_tree);
240 vp9_tokens_from_tree(vp9_mv_class_encodings, vp9_mv_class_tree);
241 vp9_tokens_from_tree(vp9_mv_class0_encodings, vp9_mv_class0_tree);
242 vp9_tokens_from_tree(vp9_mv_fp_encodings, vp9_mv_fp_tree);
243 }
244
245 void vp9_init_mv_probs(VP9_COMMON *cm) {
246 cm->fc.nmvc = default_nmv_context;
247 }

Mercurial Repository: The Tor Browser

Europalab SCM Repository Server

mercurial