xref: /aosp_15_r20/external/webp/src/enc/tree_enc.c (revision b2055c353e87c8814eb2b6b1b11112a1562253bd)
1*b2055c35SXin Li // Copyright 2011 Google Inc. All Rights Reserved.
2*b2055c35SXin Li //
3*b2055c35SXin Li // Use of this source code is governed by a BSD-style license
4*b2055c35SXin Li // that can be found in the COPYING file in the root of the source
5*b2055c35SXin Li // tree. An additional intellectual property rights grant can be found
6*b2055c35SXin Li // in the file PATENTS. All contributing project authors may
7*b2055c35SXin Li // be found in the AUTHORS file in the root of the source tree.
8*b2055c35SXin Li // -----------------------------------------------------------------------------
9*b2055c35SXin Li //
10*b2055c35SXin Li // Coding of token probabilities, intra modes and segments.
11*b2055c35SXin Li //
12*b2055c35SXin Li // Author: Skal ([email protected])
13*b2055c35SXin Li 
14*b2055c35SXin Li #include "src/enc/vp8i_enc.h"
15*b2055c35SXin Li 
16*b2055c35SXin Li //------------------------------------------------------------------------------
17*b2055c35SXin Li // Default probabilities
18*b2055c35SXin Li 
19*b2055c35SXin Li // Paragraph 13.5
20*b2055c35SXin Li const uint8_t
21*b2055c35SXin Li   VP8CoeffsProba0[NUM_TYPES][NUM_BANDS][NUM_CTX][NUM_PROBAS] = {
22*b2055c35SXin Li   { { { 128, 128, 128, 128, 128, 128, 128, 128, 128, 128, 128 },
23*b2055c35SXin Li       { 128, 128, 128, 128, 128, 128, 128, 128, 128, 128, 128 },
24*b2055c35SXin Li       { 128, 128, 128, 128, 128, 128, 128, 128, 128, 128, 128 }
25*b2055c35SXin Li     },
26*b2055c35SXin Li     { { 253, 136, 254, 255, 228, 219, 128, 128, 128, 128, 128 },
27*b2055c35SXin Li       { 189, 129, 242, 255, 227, 213, 255, 219, 128, 128, 128 },
28*b2055c35SXin Li       { 106, 126, 227, 252, 214, 209, 255, 255, 128, 128, 128 }
29*b2055c35SXin Li     },
30*b2055c35SXin Li     { { 1, 98, 248, 255, 236, 226, 255, 255, 128, 128, 128 },
31*b2055c35SXin Li       { 181, 133, 238, 254, 221, 234, 255, 154, 128, 128, 128 },
32*b2055c35SXin Li       { 78, 134, 202, 247, 198, 180, 255, 219, 128, 128, 128 },
33*b2055c35SXin Li     },
34*b2055c35SXin Li     { { 1, 185, 249, 255, 243, 255, 128, 128, 128, 128, 128 },
35*b2055c35SXin Li       { 184, 150, 247, 255, 236, 224, 128, 128, 128, 128, 128 },
36*b2055c35SXin Li       { 77, 110, 216, 255, 236, 230, 128, 128, 128, 128, 128 },
37*b2055c35SXin Li     },
38*b2055c35SXin Li     { { 1, 101, 251, 255, 241, 255, 128, 128, 128, 128, 128 },
39*b2055c35SXin Li       { 170, 139, 241, 252, 236, 209, 255, 255, 128, 128, 128 },
40*b2055c35SXin Li       { 37, 116, 196, 243, 228, 255, 255, 255, 128, 128, 128 }
41*b2055c35SXin Li     },
42*b2055c35SXin Li     { { 1, 204, 254, 255, 245, 255, 128, 128, 128, 128, 128 },
43*b2055c35SXin Li       { 207, 160, 250, 255, 238, 128, 128, 128, 128, 128, 128 },
44*b2055c35SXin Li       { 102, 103, 231, 255, 211, 171, 128, 128, 128, 128, 128 }
45*b2055c35SXin Li     },
46*b2055c35SXin Li     { { 1, 152, 252, 255, 240, 255, 128, 128, 128, 128, 128 },
47*b2055c35SXin Li       { 177, 135, 243, 255, 234, 225, 128, 128, 128, 128, 128 },
48*b2055c35SXin Li       { 80, 129, 211, 255, 194, 224, 128, 128, 128, 128, 128 }
49*b2055c35SXin Li     },
50*b2055c35SXin Li     { { 1, 1, 255, 128, 128, 128, 128, 128, 128, 128, 128 },
51*b2055c35SXin Li       { 246, 1, 255, 128, 128, 128, 128, 128, 128, 128, 128 },
52*b2055c35SXin Li       { 255, 128, 128, 128, 128, 128, 128, 128, 128, 128, 128 }
53*b2055c35SXin Li     }
54*b2055c35SXin Li   },
55*b2055c35SXin Li   { { { 198, 35, 237, 223, 193, 187, 162, 160, 145, 155, 62 },
56*b2055c35SXin Li       { 131, 45, 198, 221, 172, 176, 220, 157, 252, 221, 1 },
57*b2055c35SXin Li       { 68, 47, 146, 208, 149, 167, 221, 162, 255, 223, 128 }
58*b2055c35SXin Li     },
59*b2055c35SXin Li     { { 1, 149, 241, 255, 221, 224, 255, 255, 128, 128, 128 },
60*b2055c35SXin Li       { 184, 141, 234, 253, 222, 220, 255, 199, 128, 128, 128 },
61*b2055c35SXin Li       { 81, 99, 181, 242, 176, 190, 249, 202, 255, 255, 128 }
62*b2055c35SXin Li     },
63*b2055c35SXin Li     { { 1, 129, 232, 253, 214, 197, 242, 196, 255, 255, 128 },
64*b2055c35SXin Li       { 99, 121, 210, 250, 201, 198, 255, 202, 128, 128, 128 },
65*b2055c35SXin Li       { 23, 91, 163, 242, 170, 187, 247, 210, 255, 255, 128 }
66*b2055c35SXin Li     },
67*b2055c35SXin Li     { { 1, 200, 246, 255, 234, 255, 128, 128, 128, 128, 128 },
68*b2055c35SXin Li       { 109, 178, 241, 255, 231, 245, 255, 255, 128, 128, 128 },
69*b2055c35SXin Li       { 44, 130, 201, 253, 205, 192, 255, 255, 128, 128, 128 }
70*b2055c35SXin Li     },
71*b2055c35SXin Li     { { 1, 132, 239, 251, 219, 209, 255, 165, 128, 128, 128 },
72*b2055c35SXin Li       { 94, 136, 225, 251, 218, 190, 255, 255, 128, 128, 128 },
73*b2055c35SXin Li       { 22, 100, 174, 245, 186, 161, 255, 199, 128, 128, 128 }
74*b2055c35SXin Li     },
75*b2055c35SXin Li     { { 1, 182, 249, 255, 232, 235, 128, 128, 128, 128, 128 },
76*b2055c35SXin Li       { 124, 143, 241, 255, 227, 234, 128, 128, 128, 128, 128 },
77*b2055c35SXin Li       { 35, 77, 181, 251, 193, 211, 255, 205, 128, 128, 128 }
78*b2055c35SXin Li     },
79*b2055c35SXin Li     { { 1, 157, 247, 255, 236, 231, 255, 255, 128, 128, 128 },
80*b2055c35SXin Li       { 121, 141, 235, 255, 225, 227, 255, 255, 128, 128, 128 },
81*b2055c35SXin Li       { 45, 99, 188, 251, 195, 217, 255, 224, 128, 128, 128 }
82*b2055c35SXin Li     },
83*b2055c35SXin Li     { { 1, 1, 251, 255, 213, 255, 128, 128, 128, 128, 128 },
84*b2055c35SXin Li       { 203, 1, 248, 255, 255, 128, 128, 128, 128, 128, 128 },
85*b2055c35SXin Li       { 137, 1, 177, 255, 224, 255, 128, 128, 128, 128, 128 }
86*b2055c35SXin Li     }
87*b2055c35SXin Li   },
88*b2055c35SXin Li   { { { 253, 9, 248, 251, 207, 208, 255, 192, 128, 128, 128 },
89*b2055c35SXin Li       { 175, 13, 224, 243, 193, 185, 249, 198, 255, 255, 128 },
90*b2055c35SXin Li       { 73, 17, 171, 221, 161, 179, 236, 167, 255, 234, 128 }
91*b2055c35SXin Li     },
92*b2055c35SXin Li     { { 1, 95, 247, 253, 212, 183, 255, 255, 128, 128, 128 },
93*b2055c35SXin Li       { 239, 90, 244, 250, 211, 209, 255, 255, 128, 128, 128 },
94*b2055c35SXin Li       { 155, 77, 195, 248, 188, 195, 255, 255, 128, 128, 128 }
95*b2055c35SXin Li     },
96*b2055c35SXin Li     { { 1, 24, 239, 251, 218, 219, 255, 205, 128, 128, 128 },
97*b2055c35SXin Li       { 201, 51, 219, 255, 196, 186, 128, 128, 128, 128, 128 },
98*b2055c35SXin Li       { 69, 46, 190, 239, 201, 218, 255, 228, 128, 128, 128 }
99*b2055c35SXin Li     },
100*b2055c35SXin Li     { { 1, 191, 251, 255, 255, 128, 128, 128, 128, 128, 128 },
101*b2055c35SXin Li       { 223, 165, 249, 255, 213, 255, 128, 128, 128, 128, 128 },
102*b2055c35SXin Li       { 141, 124, 248, 255, 255, 128, 128, 128, 128, 128, 128 }
103*b2055c35SXin Li     },
104*b2055c35SXin Li     { { 1, 16, 248, 255, 255, 128, 128, 128, 128, 128, 128 },
105*b2055c35SXin Li       { 190, 36, 230, 255, 236, 255, 128, 128, 128, 128, 128 },
106*b2055c35SXin Li       { 149, 1, 255, 128, 128, 128, 128, 128, 128, 128, 128 }
107*b2055c35SXin Li     },
108*b2055c35SXin Li     { { 1, 226, 255, 128, 128, 128, 128, 128, 128, 128, 128 },
109*b2055c35SXin Li       { 247, 192, 255, 128, 128, 128, 128, 128, 128, 128, 128 },
110*b2055c35SXin Li       { 240, 128, 255, 128, 128, 128, 128, 128, 128, 128, 128 }
111*b2055c35SXin Li     },
112*b2055c35SXin Li     { { 1, 134, 252, 255, 255, 128, 128, 128, 128, 128, 128 },
113*b2055c35SXin Li       { 213, 62, 250, 255, 255, 128, 128, 128, 128, 128, 128 },
114*b2055c35SXin Li       { 55, 93, 255, 128, 128, 128, 128, 128, 128, 128, 128 }
115*b2055c35SXin Li     },
116*b2055c35SXin Li     { { 128, 128, 128, 128, 128, 128, 128, 128, 128, 128, 128 },
117*b2055c35SXin Li       { 128, 128, 128, 128, 128, 128, 128, 128, 128, 128, 128 },
118*b2055c35SXin Li       { 128, 128, 128, 128, 128, 128, 128, 128, 128, 128, 128 }
119*b2055c35SXin Li     }
120*b2055c35SXin Li   },
121*b2055c35SXin Li   { { { 202, 24, 213, 235, 186, 191, 220, 160, 240, 175, 255 },
122*b2055c35SXin Li       { 126, 38, 182, 232, 169, 184, 228, 174, 255, 187, 128 },
123*b2055c35SXin Li       { 61, 46, 138, 219, 151, 178, 240, 170, 255, 216, 128 }
124*b2055c35SXin Li     },
125*b2055c35SXin Li     { { 1, 112, 230, 250, 199, 191, 247, 159, 255, 255, 128 },
126*b2055c35SXin Li       { 166, 109, 228, 252, 211, 215, 255, 174, 128, 128, 128 },
127*b2055c35SXin Li       { 39, 77, 162, 232, 172, 180, 245, 178, 255, 255, 128 }
128*b2055c35SXin Li     },
129*b2055c35SXin Li     { { 1, 52, 220, 246, 198, 199, 249, 220, 255, 255, 128 },
130*b2055c35SXin Li       { 124, 74, 191, 243, 183, 193, 250, 221, 255, 255, 128 },
131*b2055c35SXin Li       { 24, 71, 130, 219, 154, 170, 243, 182, 255, 255, 128 }
132*b2055c35SXin Li     },
133*b2055c35SXin Li     { { 1, 182, 225, 249, 219, 240, 255, 224, 128, 128, 128 },
134*b2055c35SXin Li       { 149, 150, 226, 252, 216, 205, 255, 171, 128, 128, 128 },
135*b2055c35SXin Li       { 28, 108, 170, 242, 183, 194, 254, 223, 255, 255, 128 }
136*b2055c35SXin Li     },
137*b2055c35SXin Li     { { 1, 81, 230, 252, 204, 203, 255, 192, 128, 128, 128 },
138*b2055c35SXin Li       { 123, 102, 209, 247, 188, 196, 255, 233, 128, 128, 128 },
139*b2055c35SXin Li       { 20, 95, 153, 243, 164, 173, 255, 203, 128, 128, 128 }
140*b2055c35SXin Li     },
141*b2055c35SXin Li     { { 1, 222, 248, 255, 216, 213, 128, 128, 128, 128, 128 },
142*b2055c35SXin Li       { 168, 175, 246, 252, 235, 205, 255, 255, 128, 128, 128 },
143*b2055c35SXin Li       { 47, 116, 215, 255, 211, 212, 255, 255, 128, 128, 128 }
144*b2055c35SXin Li     },
145*b2055c35SXin Li     { { 1, 121, 236, 253, 212, 214, 255, 255, 128, 128, 128 },
146*b2055c35SXin Li       { 141, 84, 213, 252, 201, 202, 255, 219, 128, 128, 128 },
147*b2055c35SXin Li       { 42, 80, 160, 240, 162, 185, 255, 205, 128, 128, 128 }
148*b2055c35SXin Li     },
149*b2055c35SXin Li     { { 1, 1, 255, 128, 128, 128, 128, 128, 128, 128, 128 },
150*b2055c35SXin Li       { 244, 1, 255, 128, 128, 128, 128, 128, 128, 128, 128 },
151*b2055c35SXin Li       { 238, 1, 255, 128, 128, 128, 128, 128, 128, 128, 128 }
152*b2055c35SXin Li     }
153*b2055c35SXin Li   }
154*b2055c35SXin Li };
155*b2055c35SXin Li 
VP8DefaultProbas(VP8Encoder * const enc)156*b2055c35SXin Li void VP8DefaultProbas(VP8Encoder* const enc) {
157*b2055c35SXin Li   VP8EncProba* const probas = &enc->proba_;
158*b2055c35SXin Li   probas->use_skip_proba_ = 0;
159*b2055c35SXin Li   memset(probas->segments_, 255u, sizeof(probas->segments_));
160*b2055c35SXin Li   memcpy(probas->coeffs_, VP8CoeffsProba0, sizeof(VP8CoeffsProba0));
161*b2055c35SXin Li   // Note: we could hard-code the level_costs_ corresponding to VP8CoeffsProba0,
162*b2055c35SXin Li   // but that's ~11k of static data. Better call VP8CalculateLevelCosts() later.
163*b2055c35SXin Li   probas->dirty_ = 1;
164*b2055c35SXin Li }
165*b2055c35SXin Li 
166*b2055c35SXin Li // Paragraph 11.5.  900bytes.
167*b2055c35SXin Li static const uint8_t kBModesProba[NUM_BMODES][NUM_BMODES][NUM_BMODES - 1] = {
168*b2055c35SXin Li   { { 231, 120, 48, 89, 115, 113, 120, 152, 112 },
169*b2055c35SXin Li     { 152, 179, 64, 126, 170, 118, 46, 70, 95 },
170*b2055c35SXin Li     { 175, 69, 143, 80, 85, 82, 72, 155, 103 },
171*b2055c35SXin Li     { 56, 58, 10, 171, 218, 189, 17, 13, 152 },
172*b2055c35SXin Li     { 114, 26, 17, 163, 44, 195, 21, 10, 173 },
173*b2055c35SXin Li     { 121, 24, 80, 195, 26, 62, 44, 64, 85 },
174*b2055c35SXin Li     { 144, 71, 10, 38, 171, 213, 144, 34, 26 },
175*b2055c35SXin Li     { 170, 46, 55, 19, 136, 160, 33, 206, 71 },
176*b2055c35SXin Li     { 63, 20, 8, 114, 114, 208, 12, 9, 226 },
177*b2055c35SXin Li     { 81, 40, 11, 96, 182, 84, 29, 16, 36 } },
178*b2055c35SXin Li   { { 134, 183, 89, 137, 98, 101, 106, 165, 148 },
179*b2055c35SXin Li     { 72, 187, 100, 130, 157, 111, 32, 75, 80 },
180*b2055c35SXin Li     { 66, 102, 167, 99, 74, 62, 40, 234, 128 },
181*b2055c35SXin Li     { 41, 53, 9, 178, 241, 141, 26, 8, 107 },
182*b2055c35SXin Li     { 74, 43, 26, 146, 73, 166, 49, 23, 157 },
183*b2055c35SXin Li     { 65, 38, 105, 160, 51, 52, 31, 115, 128 },
184*b2055c35SXin Li     { 104, 79, 12, 27, 217, 255, 87, 17, 7 },
185*b2055c35SXin Li     { 87, 68, 71, 44, 114, 51, 15, 186, 23 },
186*b2055c35SXin Li     { 47, 41, 14, 110, 182, 183, 21, 17, 194 },
187*b2055c35SXin Li     { 66, 45, 25, 102, 197, 189, 23, 18, 22 } },
188*b2055c35SXin Li   { { 88, 88, 147, 150, 42, 46, 45, 196, 205 },
189*b2055c35SXin Li     { 43, 97, 183, 117, 85, 38, 35, 179, 61 },
190*b2055c35SXin Li     { 39, 53, 200, 87, 26, 21, 43, 232, 171 },
191*b2055c35SXin Li     { 56, 34, 51, 104, 114, 102, 29, 93, 77 },
192*b2055c35SXin Li     { 39, 28, 85, 171, 58, 165, 90, 98, 64 },
193*b2055c35SXin Li     { 34, 22, 116, 206, 23, 34, 43, 166, 73 },
194*b2055c35SXin Li     { 107, 54, 32, 26, 51, 1, 81, 43, 31 },
195*b2055c35SXin Li     { 68, 25, 106, 22, 64, 171, 36, 225, 114 },
196*b2055c35SXin Li     { 34, 19, 21, 102, 132, 188, 16, 76, 124 },
197*b2055c35SXin Li     { 62, 18, 78, 95, 85, 57, 50, 48, 51 } },
198*b2055c35SXin Li   { { 193, 101, 35, 159, 215, 111, 89, 46, 111 },
199*b2055c35SXin Li     { 60, 148, 31, 172, 219, 228, 21, 18, 111 },
200*b2055c35SXin Li     { 112, 113, 77, 85, 179, 255, 38, 120, 114 },
201*b2055c35SXin Li     { 40, 42, 1, 196, 245, 209, 10, 25, 109 },
202*b2055c35SXin Li     { 88, 43, 29, 140, 166, 213, 37, 43, 154 },
203*b2055c35SXin Li     { 61, 63, 30, 155, 67, 45, 68, 1, 209 },
204*b2055c35SXin Li     { 100, 80, 8, 43, 154, 1, 51, 26, 71 },
205*b2055c35SXin Li     { 142, 78, 78, 16, 255, 128, 34, 197, 171 },
206*b2055c35SXin Li     { 41, 40, 5, 102, 211, 183, 4, 1, 221 },
207*b2055c35SXin Li     { 51, 50, 17, 168, 209, 192, 23, 25, 82 } },
208*b2055c35SXin Li   { { 138, 31, 36, 171, 27, 166, 38, 44, 229 },
209*b2055c35SXin Li     { 67, 87, 58, 169, 82, 115, 26, 59, 179 },
210*b2055c35SXin Li     { 63, 59, 90, 180, 59, 166, 93, 73, 154 },
211*b2055c35SXin Li     { 40, 40, 21, 116, 143, 209, 34, 39, 175 },
212*b2055c35SXin Li     { 47, 15, 16, 183, 34, 223, 49, 45, 183 },
213*b2055c35SXin Li     { 46, 17, 33, 183, 6, 98, 15, 32, 183 },
214*b2055c35SXin Li     { 57, 46, 22, 24, 128, 1, 54, 17, 37 },
215*b2055c35SXin Li     { 65, 32, 73, 115, 28, 128, 23, 128, 205 },
216*b2055c35SXin Li     { 40, 3, 9, 115, 51, 192, 18, 6, 223 },
217*b2055c35SXin Li     { 87, 37, 9, 115, 59, 77, 64, 21, 47 } },
218*b2055c35SXin Li   { { 104, 55, 44, 218, 9, 54, 53, 130, 226 },
219*b2055c35SXin Li     { 64, 90, 70, 205, 40, 41, 23, 26, 57 },
220*b2055c35SXin Li     { 54, 57, 112, 184, 5, 41, 38, 166, 213 },
221*b2055c35SXin Li     { 30, 34, 26, 133, 152, 116, 10, 32, 134 },
222*b2055c35SXin Li     { 39, 19, 53, 221, 26, 114, 32, 73, 255 },
223*b2055c35SXin Li     { 31, 9, 65, 234, 2, 15, 1, 118, 73 },
224*b2055c35SXin Li     { 75, 32, 12, 51, 192, 255, 160, 43, 51 },
225*b2055c35SXin Li     { 88, 31, 35, 67, 102, 85, 55, 186, 85 },
226*b2055c35SXin Li     { 56, 21, 23, 111, 59, 205, 45, 37, 192 },
227*b2055c35SXin Li     { 55, 38, 70, 124, 73, 102, 1, 34, 98 } },
228*b2055c35SXin Li   { { 125, 98, 42, 88, 104, 85, 117, 175, 82 },
229*b2055c35SXin Li     { 95, 84, 53, 89, 128, 100, 113, 101, 45 },
230*b2055c35SXin Li     { 75, 79, 123, 47, 51, 128, 81, 171, 1 },
231*b2055c35SXin Li     { 57, 17, 5, 71, 102, 57, 53, 41, 49 },
232*b2055c35SXin Li     { 38, 33, 13, 121, 57, 73, 26, 1, 85 },
233*b2055c35SXin Li     { 41, 10, 67, 138, 77, 110, 90, 47, 114 },
234*b2055c35SXin Li     { 115, 21, 2, 10, 102, 255, 166, 23, 6 },
235*b2055c35SXin Li     { 101, 29, 16, 10, 85, 128, 101, 196, 26 },
236*b2055c35SXin Li     { 57, 18, 10, 102, 102, 213, 34, 20, 43 },
237*b2055c35SXin Li     { 117, 20, 15, 36, 163, 128, 68, 1, 26 } },
238*b2055c35SXin Li   { { 102, 61, 71, 37, 34, 53, 31, 243, 192 },
239*b2055c35SXin Li     { 69, 60, 71, 38, 73, 119, 28, 222, 37 },
240*b2055c35SXin Li     { 68, 45, 128, 34, 1, 47, 11, 245, 171 },
241*b2055c35SXin Li     { 62, 17, 19, 70, 146, 85, 55, 62, 70 },
242*b2055c35SXin Li     { 37, 43, 37, 154, 100, 163, 85, 160, 1 },
243*b2055c35SXin Li     { 63, 9, 92, 136, 28, 64, 32, 201, 85 },
244*b2055c35SXin Li     { 75, 15, 9, 9, 64, 255, 184, 119, 16 },
245*b2055c35SXin Li     { 86, 6, 28, 5, 64, 255, 25, 248, 1 },
246*b2055c35SXin Li     { 56, 8, 17, 132, 137, 255, 55, 116, 128 },
247*b2055c35SXin Li     { 58, 15, 20, 82, 135, 57, 26, 121, 40 } },
248*b2055c35SXin Li   { { 164, 50, 31, 137, 154, 133, 25, 35, 218 },
249*b2055c35SXin Li     { 51, 103, 44, 131, 131, 123, 31, 6, 158 },
250*b2055c35SXin Li     { 86, 40, 64, 135, 148, 224, 45, 183, 128 },
251*b2055c35SXin Li     { 22, 26, 17, 131, 240, 154, 14, 1, 209 },
252*b2055c35SXin Li     { 45, 16, 21, 91, 64, 222, 7, 1, 197 },
253*b2055c35SXin Li     { 56, 21, 39, 155, 60, 138, 23, 102, 213 },
254*b2055c35SXin Li     { 83, 12, 13, 54, 192, 255, 68, 47, 28 },
255*b2055c35SXin Li     { 85, 26, 85, 85, 128, 128, 32, 146, 171 },
256*b2055c35SXin Li     { 18, 11, 7, 63, 144, 171, 4, 4, 246 },
257*b2055c35SXin Li     { 35, 27, 10, 146, 174, 171, 12, 26, 128 } },
258*b2055c35SXin Li   { { 190, 80, 35, 99, 180, 80, 126, 54, 45 },
259*b2055c35SXin Li     { 85, 126, 47, 87, 176, 51, 41, 20, 32 },
260*b2055c35SXin Li     { 101, 75, 128, 139, 118, 146, 116, 128, 85 },
261*b2055c35SXin Li     { 56, 41, 15, 176, 236, 85, 37, 9, 62 },
262*b2055c35SXin Li     { 71, 30, 17, 119, 118, 255, 17, 18, 138 },
263*b2055c35SXin Li     { 101, 38, 60, 138, 55, 70, 43, 26, 142 },
264*b2055c35SXin Li     { 146, 36, 19, 30, 171, 255, 97, 27, 20 },
265*b2055c35SXin Li     { 138, 45, 61, 62, 219, 1, 81, 188, 64 },
266*b2055c35SXin Li     { 32, 41, 20, 117, 151, 142, 20, 21, 163 },
267*b2055c35SXin Li     { 112, 19, 12, 61, 195, 128, 48, 4, 24 } }
268*b2055c35SXin Li };
269*b2055c35SXin Li 
PutI4Mode(VP8BitWriter * const bw,int mode,const uint8_t * const prob)270*b2055c35SXin Li static int PutI4Mode(VP8BitWriter* const bw, int mode,
271*b2055c35SXin Li                      const uint8_t* const prob) {
272*b2055c35SXin Li   if (VP8PutBit(bw, mode != B_DC_PRED, prob[0])) {
273*b2055c35SXin Li     if (VP8PutBit(bw, mode != B_TM_PRED, prob[1])) {
274*b2055c35SXin Li       if (VP8PutBit(bw, mode != B_VE_PRED, prob[2])) {
275*b2055c35SXin Li         if (!VP8PutBit(bw, mode >= B_LD_PRED, prob[3])) {
276*b2055c35SXin Li           if (VP8PutBit(bw, mode != B_HE_PRED, prob[4])) {
277*b2055c35SXin Li             VP8PutBit(bw, mode != B_RD_PRED, prob[5]);
278*b2055c35SXin Li           }
279*b2055c35SXin Li         } else {
280*b2055c35SXin Li           if (VP8PutBit(bw, mode != B_LD_PRED, prob[6])) {
281*b2055c35SXin Li             if (VP8PutBit(bw, mode != B_VL_PRED, prob[7])) {
282*b2055c35SXin Li               VP8PutBit(bw, mode != B_HD_PRED, prob[8]);
283*b2055c35SXin Li             }
284*b2055c35SXin Li           }
285*b2055c35SXin Li         }
286*b2055c35SXin Li       }
287*b2055c35SXin Li     }
288*b2055c35SXin Li   }
289*b2055c35SXin Li   return mode;
290*b2055c35SXin Li }
291*b2055c35SXin Li 
PutI16Mode(VP8BitWriter * const bw,int mode)292*b2055c35SXin Li static void PutI16Mode(VP8BitWriter* const bw, int mode) {
293*b2055c35SXin Li   if (VP8PutBit(bw, (mode == TM_PRED || mode == H_PRED), 156)) {
294*b2055c35SXin Li     VP8PutBit(bw, mode == TM_PRED, 128);    // TM or HE
295*b2055c35SXin Li   } else {
296*b2055c35SXin Li     VP8PutBit(bw, mode == V_PRED, 163);     // VE or DC
297*b2055c35SXin Li   }
298*b2055c35SXin Li }
299*b2055c35SXin Li 
PutUVMode(VP8BitWriter * const bw,int uv_mode)300*b2055c35SXin Li static void PutUVMode(VP8BitWriter* const bw, int uv_mode) {
301*b2055c35SXin Li   if (VP8PutBit(bw, uv_mode != DC_PRED, 142)) {
302*b2055c35SXin Li     if (VP8PutBit(bw, uv_mode != V_PRED, 114)) {
303*b2055c35SXin Li       VP8PutBit(bw, uv_mode != H_PRED, 183);    // else: TM_PRED
304*b2055c35SXin Li     }
305*b2055c35SXin Li   }
306*b2055c35SXin Li }
307*b2055c35SXin Li 
PutSegment(VP8BitWriter * const bw,int s,const uint8_t * p)308*b2055c35SXin Li static void PutSegment(VP8BitWriter* const bw, int s, const uint8_t* p) {
309*b2055c35SXin Li   if (VP8PutBit(bw, s >= 2, p[0])) p += 1;
310*b2055c35SXin Li   VP8PutBit(bw, s & 1, p[1]);
311*b2055c35SXin Li }
312*b2055c35SXin Li 
VP8CodeIntraModes(VP8Encoder * const enc)313*b2055c35SXin Li void VP8CodeIntraModes(VP8Encoder* const enc) {
314*b2055c35SXin Li   VP8BitWriter* const bw = &enc->bw_;
315*b2055c35SXin Li   VP8EncIterator it;
316*b2055c35SXin Li   VP8IteratorInit(enc, &it);
317*b2055c35SXin Li   do {
318*b2055c35SXin Li     const VP8MBInfo* const mb = it.mb_;
319*b2055c35SXin Li     const uint8_t* preds = it.preds_;
320*b2055c35SXin Li     if (enc->segment_hdr_.update_map_) {
321*b2055c35SXin Li       PutSegment(bw, mb->segment_, enc->proba_.segments_);
322*b2055c35SXin Li     }
323*b2055c35SXin Li     if (enc->proba_.use_skip_proba_) {
324*b2055c35SXin Li       VP8PutBit(bw, mb->skip_, enc->proba_.skip_proba_);
325*b2055c35SXin Li     }
326*b2055c35SXin Li     if (VP8PutBit(bw, (mb->type_ != 0), 145)) {  // i16x16
327*b2055c35SXin Li       PutI16Mode(bw, preds[0]);
328*b2055c35SXin Li     } else {
329*b2055c35SXin Li       const int preds_w = enc->preds_w_;
330*b2055c35SXin Li       const uint8_t* top_pred = preds - preds_w;
331*b2055c35SXin Li       int x, y;
332*b2055c35SXin Li       for (y = 0; y < 4; ++y) {
333*b2055c35SXin Li         int left = preds[-1];
334*b2055c35SXin Li         for (x = 0; x < 4; ++x) {
335*b2055c35SXin Li           const uint8_t* const probas = kBModesProba[top_pred[x]][left];
336*b2055c35SXin Li           left = PutI4Mode(bw, preds[x], probas);
337*b2055c35SXin Li         }
338*b2055c35SXin Li         top_pred = preds;
339*b2055c35SXin Li         preds += preds_w;
340*b2055c35SXin Li       }
341*b2055c35SXin Li     }
342*b2055c35SXin Li     PutUVMode(bw, mb->uv_mode_);
343*b2055c35SXin Li   } while (VP8IteratorNext(&it));
344*b2055c35SXin Li }
345*b2055c35SXin Li 
346*b2055c35SXin Li //------------------------------------------------------------------------------
347*b2055c35SXin Li // Paragraph 13
348*b2055c35SXin Li 
349*b2055c35SXin Li const uint8_t
350*b2055c35SXin Li     VP8CoeffsUpdateProba[NUM_TYPES][NUM_BANDS][NUM_CTX][NUM_PROBAS] = {
351*b2055c35SXin Li   { { { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
352*b2055c35SXin Li       { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
353*b2055c35SXin Li       { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 }
354*b2055c35SXin Li     },
355*b2055c35SXin Li     { { 176, 246, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
356*b2055c35SXin Li       { 223, 241, 252, 255, 255, 255, 255, 255, 255, 255, 255 },
357*b2055c35SXin Li       { 249, 253, 253, 255, 255, 255, 255, 255, 255, 255, 255 }
358*b2055c35SXin Li     },
359*b2055c35SXin Li     { { 255, 244, 252, 255, 255, 255, 255, 255, 255, 255, 255 },
360*b2055c35SXin Li       { 234, 254, 254, 255, 255, 255, 255, 255, 255, 255, 255 },
361*b2055c35SXin Li       { 253, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 }
362*b2055c35SXin Li     },
363*b2055c35SXin Li     { { 255, 246, 254, 255, 255, 255, 255, 255, 255, 255, 255 },
364*b2055c35SXin Li       { 239, 253, 254, 255, 255, 255, 255, 255, 255, 255, 255 },
365*b2055c35SXin Li       { 254, 255, 254, 255, 255, 255, 255, 255, 255, 255, 255 }
366*b2055c35SXin Li     },
367*b2055c35SXin Li     { { 255, 248, 254, 255, 255, 255, 255, 255, 255, 255, 255 },
368*b2055c35SXin Li       { 251, 255, 254, 255, 255, 255, 255, 255, 255, 255, 255 },
369*b2055c35SXin Li       { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 }
370*b2055c35SXin Li     },
371*b2055c35SXin Li     { { 255, 253, 254, 255, 255, 255, 255, 255, 255, 255, 255 },
372*b2055c35SXin Li       { 251, 254, 254, 255, 255, 255, 255, 255, 255, 255, 255 },
373*b2055c35SXin Li       { 254, 255, 254, 255, 255, 255, 255, 255, 255, 255, 255 }
374*b2055c35SXin Li     },
375*b2055c35SXin Li     { { 255, 254, 253, 255, 254, 255, 255, 255, 255, 255, 255 },
376*b2055c35SXin Li       { 250, 255, 254, 255, 254, 255, 255, 255, 255, 255, 255 },
377*b2055c35SXin Li       { 254, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 }
378*b2055c35SXin Li     },
379*b2055c35SXin Li     { { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
380*b2055c35SXin Li       { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
381*b2055c35SXin Li       { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 }
382*b2055c35SXin Li     }
383*b2055c35SXin Li   },
384*b2055c35SXin Li   { { { 217, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
385*b2055c35SXin Li       { 225, 252, 241, 253, 255, 255, 254, 255, 255, 255, 255 },
386*b2055c35SXin Li       { 234, 250, 241, 250, 253, 255, 253, 254, 255, 255, 255 }
387*b2055c35SXin Li     },
388*b2055c35SXin Li     { { 255, 254, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
389*b2055c35SXin Li       { 223, 254, 254, 255, 255, 255, 255, 255, 255, 255, 255 },
390*b2055c35SXin Li       { 238, 253, 254, 254, 255, 255, 255, 255, 255, 255, 255 }
391*b2055c35SXin Li     },
392*b2055c35SXin Li     { { 255, 248, 254, 255, 255, 255, 255, 255, 255, 255, 255 },
393*b2055c35SXin Li       { 249, 254, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
394*b2055c35SXin Li       { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 }
395*b2055c35SXin Li     },
396*b2055c35SXin Li     { { 255, 253, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
397*b2055c35SXin Li       { 247, 254, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
398*b2055c35SXin Li       { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 }
399*b2055c35SXin Li     },
400*b2055c35SXin Li     { { 255, 253, 254, 255, 255, 255, 255, 255, 255, 255, 255 },
401*b2055c35SXin Li       { 252, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
402*b2055c35SXin Li       { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 }
403*b2055c35SXin Li     },
404*b2055c35SXin Li     { { 255, 254, 254, 255, 255, 255, 255, 255, 255, 255, 255 },
405*b2055c35SXin Li       { 253, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
406*b2055c35SXin Li       { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 }
407*b2055c35SXin Li     },
408*b2055c35SXin Li     { { 255, 254, 253, 255, 255, 255, 255, 255, 255, 255, 255 },
409*b2055c35SXin Li       { 250, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
410*b2055c35SXin Li       { 254, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 }
411*b2055c35SXin Li     },
412*b2055c35SXin Li     { { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
413*b2055c35SXin Li       { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
414*b2055c35SXin Li       { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 }
415*b2055c35SXin Li     }
416*b2055c35SXin Li   },
417*b2055c35SXin Li   { { { 186, 251, 250, 255, 255, 255, 255, 255, 255, 255, 255 },
418*b2055c35SXin Li       { 234, 251, 244, 254, 255, 255, 255, 255, 255, 255, 255 },
419*b2055c35SXin Li       { 251, 251, 243, 253, 254, 255, 254, 255, 255, 255, 255 }
420*b2055c35SXin Li     },
421*b2055c35SXin Li     { { 255, 253, 254, 255, 255, 255, 255, 255, 255, 255, 255 },
422*b2055c35SXin Li       { 236, 253, 254, 255, 255, 255, 255, 255, 255, 255, 255 },
423*b2055c35SXin Li       { 251, 253, 253, 254, 254, 255, 255, 255, 255, 255, 255 }
424*b2055c35SXin Li     },
425*b2055c35SXin Li     { { 255, 254, 254, 255, 255, 255, 255, 255, 255, 255, 255 },
426*b2055c35SXin Li       { 254, 254, 254, 255, 255, 255, 255, 255, 255, 255, 255 },
427*b2055c35SXin Li       { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 }
428*b2055c35SXin Li     },
429*b2055c35SXin Li     { { 255, 254, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
430*b2055c35SXin Li       { 254, 254, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
431*b2055c35SXin Li       { 254, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 }
432*b2055c35SXin Li     },
433*b2055c35SXin Li     { { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
434*b2055c35SXin Li       { 254, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
435*b2055c35SXin Li       { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 }
436*b2055c35SXin Li     },
437*b2055c35SXin Li     { { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
438*b2055c35SXin Li       { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
439*b2055c35SXin Li       { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 }
440*b2055c35SXin Li     },
441*b2055c35SXin Li     { { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
442*b2055c35SXin Li       { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
443*b2055c35SXin Li       { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 }
444*b2055c35SXin Li     },
445*b2055c35SXin Li     { { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
446*b2055c35SXin Li       { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
447*b2055c35SXin Li       { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 }
448*b2055c35SXin Li     }
449*b2055c35SXin Li   },
450*b2055c35SXin Li   { { { 248, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
451*b2055c35SXin Li       { 250, 254, 252, 254, 255, 255, 255, 255, 255, 255, 255 },
452*b2055c35SXin Li       { 248, 254, 249, 253, 255, 255, 255, 255, 255, 255, 255 }
453*b2055c35SXin Li     },
454*b2055c35SXin Li     { { 255, 253, 253, 255, 255, 255, 255, 255, 255, 255, 255 },
455*b2055c35SXin Li       { 246, 253, 253, 255, 255, 255, 255, 255, 255, 255, 255 },
456*b2055c35SXin Li       { 252, 254, 251, 254, 254, 255, 255, 255, 255, 255, 255 }
457*b2055c35SXin Li     },
458*b2055c35SXin Li     { { 255, 254, 252, 255, 255, 255, 255, 255, 255, 255, 255 },
459*b2055c35SXin Li       { 248, 254, 253, 255, 255, 255, 255, 255, 255, 255, 255 },
460*b2055c35SXin Li       { 253, 255, 254, 254, 255, 255, 255, 255, 255, 255, 255 }
461*b2055c35SXin Li     },
462*b2055c35SXin Li     { { 255, 251, 254, 255, 255, 255, 255, 255, 255, 255, 255 },
463*b2055c35SXin Li       { 245, 251, 254, 255, 255, 255, 255, 255, 255, 255, 255 },
464*b2055c35SXin Li       { 253, 253, 254, 255, 255, 255, 255, 255, 255, 255, 255 }
465*b2055c35SXin Li     },
466*b2055c35SXin Li     { { 255, 251, 253, 255, 255, 255, 255, 255, 255, 255, 255 },
467*b2055c35SXin Li       { 252, 253, 254, 255, 255, 255, 255, 255, 255, 255, 255 },
468*b2055c35SXin Li       { 255, 254, 255, 255, 255, 255, 255, 255, 255, 255, 255 }
469*b2055c35SXin Li     },
470*b2055c35SXin Li     { { 255, 252, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
471*b2055c35SXin Li       { 249, 255, 254, 255, 255, 255, 255, 255, 255, 255, 255 },
472*b2055c35SXin Li       { 255, 255, 254, 255, 255, 255, 255, 255, 255, 255, 255 }
473*b2055c35SXin Li     },
474*b2055c35SXin Li     { { 255, 255, 253, 255, 255, 255, 255, 255, 255, 255, 255 },
475*b2055c35SXin Li       { 250, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
476*b2055c35SXin Li       { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 }
477*b2055c35SXin Li     },
478*b2055c35SXin Li     { { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
479*b2055c35SXin Li       { 254, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 },
480*b2055c35SXin Li       { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 }
481*b2055c35SXin Li     }
482*b2055c35SXin Li   }
483*b2055c35SXin Li };
484*b2055c35SXin Li 
VP8WriteProbas(VP8BitWriter * const bw,const VP8EncProba * const probas)485*b2055c35SXin Li void VP8WriteProbas(VP8BitWriter* const bw, const VP8EncProba* const probas) {
486*b2055c35SXin Li   int t, b, c, p;
487*b2055c35SXin Li   for (t = 0; t < NUM_TYPES; ++t) {
488*b2055c35SXin Li     for (b = 0; b < NUM_BANDS; ++b) {
489*b2055c35SXin Li       for (c = 0; c < NUM_CTX; ++c) {
490*b2055c35SXin Li         for (p = 0; p < NUM_PROBAS; ++p) {
491*b2055c35SXin Li           const uint8_t p0 = probas->coeffs_[t][b][c][p];
492*b2055c35SXin Li           const int update = (p0 != VP8CoeffsProba0[t][b][c][p]);
493*b2055c35SXin Li           if (VP8PutBit(bw, update, VP8CoeffsUpdateProba[t][b][c][p])) {
494*b2055c35SXin Li             VP8PutBits(bw, p0, 8);
495*b2055c35SXin Li           }
496*b2055c35SXin Li         }
497*b2055c35SXin Li       }
498*b2055c35SXin Li     }
499*b2055c35SXin Li   }
500*b2055c35SXin Li   if (VP8PutBitUniform(bw, probas->use_skip_proba_)) {
501*b2055c35SXin Li     VP8PutBits(bw, probas->skip_proba_, 8);
502*b2055c35SXin Li   }
503*b2055c35SXin Li }
504*b2055c35SXin Li 
505