media/libvpx/vp9/common/vp9_entropymv.c@ac0c01689b40
media/libvpx/vp9/common/vp9_entropymv.c
Thu, 15 Jan 2015 15:59:08 +0100
- author
- Michael Schloh von Bennewitz <michael@schloh.com>
- date
- Thu, 15 Jan 2015 15:59:08 +0100
- branch
- TOR_BUG_9701
- changeset 10
- ac0c01689b40
- permissions
- -rw-r--r--
Implement a real Private Browsing Mode condition by changing the API/ABI;
This solves Tor bug #9701, complying with disk avoidance documented in
https://www.torproject.org/projects/torbrowser/design/#disk-avoidance.
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 */
12 #include "vp9/common/vp9_onyxc_int.h"
13 #include "vp9/common/vp9_entropymv.h"
15 #define MV_COUNT_SAT 20
16 #define MV_MAX_UPDATE_FACTOR 128
18 /* Integer pel reference mv threshold for use of high-precision 1/8 mv */
19 #define COMPANDED_MVREF_THRESH 8
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];
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];
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];
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];
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 };
80 #define mv_class_base(c) ((c) ? (CLASS0_SIZE << (c + 2)) : 0)
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 };
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];
135 if (offset)
136 *offset = z - mv_class_base(c);
137 return c;
138 }
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 }
145 int vp9_get_mv_mag(MV_CLASS_TYPE c, int offset) {
146 return mv_class_base(c) + offset;
147 }
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 */
157 c = vp9_get_mv_class(z, &o);
158 comp_counts->classes[c] += incr;
160 d = (o >> 3); /* int mv data */
161 f = (o >> 1) & 3; /* fractional pel mv data */
162 e = (o & 1); /* high precision mv data */
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 }
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];
183 if (mv_joint_vertical(j)) {
184 inc_mv_component(mv->row, &counts->comps[0], 1, 1);
185 }
187 if (mv_joint_horizontal(j)) {
188 inc_mv_component(mv->col, &counts->comps[1], 1, 1);
189 }
190 }
191 }
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 }
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 }
203 void vp9_adapt_mv_probs(VP9_COMMON *cm, int allow_hp) {
204 int i, j;
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;
210 adapt_probs(vp9_mv_joint_tree, pre_fc->joints, counts->joints, fc->joints);
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];
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);
222 for (j = 0; j < MV_OFFSET_BITS; ++j)
223 comp->bits[j] = adapt_prob(pre_comp->bits[j], c->bits[j]);
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]);
229 adapt_probs(vp9_mv_fp_tree, pre_comp->fp, c->fp, comp->fp);
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 }
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 }
245 void vp9_init_mv_probs(VP9_COMMON *cm) {
246 cm->fc.nmvc = default_nmv_context;
247 }
