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 #include "vp9/common/vp9_entropy.h" 12 #include "vp9/common/vp9_blockd.h" 13 #include "vp9/common/vp9_onyxc_int.h" 14 #include "vp9/common/vp9_entropymode.h" 15 #include "vpx_mem/vpx_mem.h" 16 #include "vpx/vpx_integer.h" 17 18 #define MODEL_NODES (ENTROPY_NODES - UNCONSTRAINED_NODES) 19 20 DECLARE_ALIGNED(16, const uint8_t, vp9_norm[256]) = { 21 0, 7, 6, 6, 5, 5, 5, 5, 4, 4, 4, 4, 4, 4, 4, 4, 22 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 23 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 24 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 25 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 26 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 27 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 28 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 29 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 30 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 31 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 32 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 33 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 34 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 35 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 36 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 37 }; 38 39 DECLARE_ALIGNED(16, const uint8_t, 40 vp9_coefband_trans_8x8plus[1024]) = { 41 0, 1, 1, 2, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4, 4, 4, 42 4, 4, 4, 4, 4, 5, 43 // beyond MAXBAND_INDEX+1 all values are filled as 5 44 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 45 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 46 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 47 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 48 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 49 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 50 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 51 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 52 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 53 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 54 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 55 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 56 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 57 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 58 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 59 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 60 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 61 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 62 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 63 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 64 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 65 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 66 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 67 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 68 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 69 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 70 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 71 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 72 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 73 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 74 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 75 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 76 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 77 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 78 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 79 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 80 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 81 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 82 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 83 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 84 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, 86 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 87 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 88 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 89 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 90 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 91 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 92 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 93 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 94 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 95 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 96 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 97 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 98 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 99 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 100 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 101 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 102 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 103 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 104 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 105 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 106 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 107 }; 108 109 DECLARE_ALIGNED(16, const uint8_t, 110 vp9_coefband_trans_4x4[16]) = { 111 0, 1, 1, 2, 2, 2, 3, 3, 3, 3, 4, 4, 4, 5, 5, 5, 112 }; 113 114 DECLARE_ALIGNED(16, const uint8_t, vp9_pt_energy_class[MAX_ENTROPY_TOKENS]) = { 115 0, 1, 2, 3, 3, 4, 4, 5, 5, 5, 5, 5 116 }; 117 118 119 120 /* Array indices are identical to previously-existing CONTEXT_NODE indices */ 121 122 const vp9_tree_index vp9_coef_tree[TREE_SIZE(MAX_ENTROPY_TOKENS)] = { 123 -DCT_EOB_TOKEN, 2, /* 0 = EOB */ 124 -ZERO_TOKEN, 4, /* 1 = ZERO */ 125 -ONE_TOKEN, 6, /* 2 = ONE */ 126 8, 12, /* 3 = LOW_VAL */ 127 -TWO_TOKEN, 10, /* 4 = TWO */ 128 -THREE_TOKEN, -FOUR_TOKEN, /* 5 = THREE */ 129 14, 16, /* 6 = HIGH_LOW */ 130 -DCT_VAL_CATEGORY1, -DCT_VAL_CATEGORY2, /* 7 = CAT_ONE */ 131 18, 20, /* 8 = CAT_THREEFOUR */ 132 -DCT_VAL_CATEGORY3, -DCT_VAL_CATEGORY4, /* 9 = CAT_THREE */ 133 -DCT_VAL_CATEGORY5, -DCT_VAL_CATEGORY6 /* 10 = CAT_FIVE */ 134 }; 135 136 struct vp9_token vp9_coef_encodings[MAX_ENTROPY_TOKENS]; 137 138 /* Trees for extra bits. Probabilities are constant and 139 do not depend on previously encoded bits */ 140 141 static const vp9_prob Pcat1[] = { 159}; 142 static const vp9_prob Pcat2[] = { 165, 145}; 143 static const vp9_prob Pcat3[] = { 173, 148, 140}; 144 static const vp9_prob Pcat4[] = { 176, 155, 140, 135}; 145 static const vp9_prob Pcat5[] = { 180, 157, 141, 134, 130}; 146 static const vp9_prob Pcat6[] = { 147 254, 254, 254, 252, 249, 243, 230, 196, 177, 153, 140, 133, 130, 129 148 }; 149 150 const vp9_tree_index vp9_coefmodel_tree[6] = { 151 -DCT_EOB_MODEL_TOKEN, 2, /* 0 = EOB */ 152 -ZERO_TOKEN, 4, /* 1 = ZERO */ 153 -ONE_TOKEN, -TWO_TOKEN, 154 }; 155 156 // Model obtained from a 2-sided zero-centerd distribuition derived 157 // from a Pareto distribution. The cdf of the distribution is: 158 // cdf(x) = 0.5 + 0.5 * sgn(x) * [1 - {alpha/(alpha + |x|)} ^ beta] 159 // 160 // For a given beta and a given probablity of the 1-node, the alpha 161 // is first solved, and then the {alpha, beta} pair is used to generate 162 // the probabilities for the rest of the nodes. 163 164 // beta = 8 165 static const vp9_prob modelcoefprobs_pareto8[COEFPROB_MODELS][MODEL_NODES] = { 166 { 3, 86, 128, 6, 86, 23, 88, 29}, 167 { 9, 86, 129, 17, 88, 61, 94, 76}, 168 { 15, 87, 129, 28, 89, 93, 100, 110}, 169 { 20, 88, 130, 38, 91, 118, 106, 136}, 170 { 26, 89, 131, 48, 92, 139, 111, 156}, 171 { 31, 90, 131, 58, 94, 156, 117, 171}, 172 { 37, 90, 132, 66, 95, 171, 122, 184}, 173 { 42, 91, 132, 75, 97, 183, 127, 194}, 174 { 47, 92, 133, 83, 98, 193, 132, 202}, 175 { 52, 93, 133, 90, 100, 201, 137, 208}, 176 { 57, 94, 134, 98, 101, 208, 142, 214}, 177 { 62, 94, 135, 105, 103, 214, 146, 218}, 178 { 66, 95, 135, 111, 104, 219, 151, 222}, 179 { 71, 96, 136, 117, 106, 224, 155, 225}, 180 { 76, 97, 136, 123, 107, 227, 159, 228}, 181 { 80, 98, 137, 129, 109, 231, 162, 231}, 182 { 84, 98, 138, 134, 110, 234, 166, 233}, 183 { 89, 99, 138, 140, 112, 236, 170, 235}, 184 { 93, 100, 139, 145, 113, 238, 173, 236}, 185 { 97, 101, 140, 149, 115, 240, 176, 238}, 186 {101, 102, 140, 154, 116, 242, 179, 239}, 187 {105, 103, 141, 158, 118, 243, 182, 240}, 188 {109, 104, 141, 162, 119, 244, 185, 241}, 189 {113, 104, 142, 166, 120, 245, 187, 242}, 190 {116, 105, 143, 170, 122, 246, 190, 243}, 191 {120, 106, 143, 173, 123, 247, 192, 244}, 192 {123, 107, 144, 177, 125, 248, 195, 244}, 193 {127, 108, 145, 180, 126, 249, 197, 245}, 194 {130, 109, 145, 183, 128, 249, 199, 245}, 195 {134, 110, 146, 186, 129, 250, 201, 246}, 196 {137, 111, 147, 189, 131, 251, 203, 246}, 197 {140, 112, 147, 192, 132, 251, 205, 247}, 198 {143, 113, 148, 194, 133, 251, 207, 247}, 199 {146, 114, 149, 197, 135, 252, 208, 248}, 200 {149, 115, 149, 199, 136, 252, 210, 248}, 201 {152, 115, 150, 201, 138, 252, 211, 248}, 202 {155, 116, 151, 204, 139, 253, 213, 249}, 203 {158, 117, 151, 206, 140, 253, 214, 249}, 204 {161, 118, 152, 208, 142, 253, 216, 249}, 205 {163, 119, 153, 210, 143, 253, 217, 249}, 206 {166, 120, 153, 212, 144, 254, 218, 250}, 207 {168, 121, 154, 213, 146, 254, 220, 250}, 208 {171, 122, 155, 215, 147, 254, 221, 250}, 209 {173, 123, 155, 217, 148, 254, 222, 250}, 210 {176, 124, 156, 218, 150, 254, 223, 250}, 211 {178, 125, 157, 220, 151, 254, 224, 251}, 212 {180, 126, 157, 221, 152, 254, 225, 251}, 213 {183, 127, 158, 222, 153, 254, 226, 251}, 214 {185, 128, 159, 224, 155, 255, 227, 251}, 215 {187, 129, 160, 225, 156, 255, 228, 251}, 216 {189, 131, 160, 226, 157, 255, 228, 251}, 217 {191, 132, 161, 227, 159, 255, 229, 251}, 218 {193, 133, 162, 228, 160, 255, 230, 252}, 219 {195, 134, 163, 230, 161, 255, 231, 252}, 220 {197, 135, 163, 231, 162, 255, 231, 252}, 221 {199, 136, 164, 232, 163, 255, 232, 252}, 222 {201, 137, 165, 233, 165, 255, 233, 252}, 223 {202, 138, 166, 233, 166, 255, 233, 252}, 224 {204, 139, 166, 234, 167, 255, 234, 252}, 225 {206, 140, 167, 235, 168, 255, 235, 252}, 226 {207, 141, 168, 236, 169, 255, 235, 252}, 227 {209, 142, 169, 237, 171, 255, 236, 252}, 228 {210, 144, 169, 237, 172, 255, 236, 252}, 229 {212, 145, 170, 238, 173, 255, 237, 252}, 230 {214, 146, 171, 239, 174, 255, 237, 253}, 231 {215, 147, 172, 240, 175, 255, 238, 253}, 232 {216, 148, 173, 240, 176, 255, 238, 253}, 233 {218, 149, 173, 241, 177, 255, 239, 253}, 234 {219, 150, 174, 241, 179, 255, 239, 253}, 235 {220, 152, 175, 242, 180, 255, 240, 253}, 236 {222, 153, 176, 242, 181, 255, 240, 253}, 237 {223, 154, 177, 243, 182, 255, 240, 253}, 238 {224, 155, 178, 244, 183, 255, 241, 253}, 239 {225, 156, 178, 244, 184, 255, 241, 253}, 240 {226, 158, 179, 244, 185, 255, 242, 253}, 241 {228, 159, 180, 245, 186, 255, 242, 253}, 242 {229, 160, 181, 245, 187, 255, 242, 253}, 243 {230, 161, 182, 246, 188, 255, 243, 253}, 244 {231, 163, 183, 246, 189, 255, 243, 253}, 245 {232, 164, 184, 247, 190, 255, 243, 253}, 246 {233, 165, 185, 247, 191, 255, 244, 253}, 247 {234, 166, 185, 247, 192, 255, 244, 253}, 248 {235, 168, 186, 248, 193, 255, 244, 253}, 249 {236, 169, 187, 248, 194, 255, 244, 253}, 250 {236, 170, 188, 248, 195, 255, 245, 253}, 251 {237, 171, 189, 249, 196, 255, 245, 254}, 252 {238, 173, 190, 249, 197, 255, 245, 254}, 253 {239, 174, 191, 249, 198, 255, 245, 254}, 254 {240, 175, 192, 249, 199, 255, 246, 254}, 255 {240, 177, 193, 250, 200, 255, 246, 254}, 256 {241, 178, 194, 250, 201, 255, 246, 254}, 257 {242, 179, 195, 250, 202, 255, 246, 254}, 258 {242, 181, 196, 250, 203, 255, 247, 254}, 259 {243, 182, 197, 251, 204, 255, 247, 254}, 260 {244, 184, 198, 251, 205, 255, 247, 254}, 261 {244, 185, 199, 251, 206, 255, 247, 254}, 262 {245, 186, 200, 251, 207, 255, 247, 254}, 263 {246, 188, 201, 252, 207, 255, 248, 254}, 264 {246, 189, 202, 252, 208, 255, 248, 254}, 265 {247, 191, 203, 252, 209, 255, 248, 254}, 266 {247, 192, 204, 252, 210, 255, 248, 254}, 267 {248, 194, 205, 252, 211, 255, 248, 254}, 268 {248, 195, 206, 252, 212, 255, 249, 254}, 269 {249, 197, 207, 253, 213, 255, 249, 254}, 270 {249, 198, 208, 253, 214, 255, 249, 254}, 271 {250, 200, 210, 253, 215, 255, 249, 254}, 272 {250, 201, 211, 253, 215, 255, 249, 254}, 273 {250, 203, 212, 253, 216, 255, 249, 254}, 274 {251, 204, 213, 253, 217, 255, 250, 254}, 275 {251, 206, 214, 254, 218, 255, 250, 254}, 276 {252, 207, 216, 254, 219, 255, 250, 254}, 277 {252, 209, 217, 254, 220, 255, 250, 254}, 278 {252, 211, 218, 254, 221, 255, 250, 254}, 279 {253, 213, 219, 254, 222, 255, 250, 254}, 280 {253, 214, 221, 254, 223, 255, 250, 254}, 281 {253, 216, 222, 254, 224, 255, 251, 254}, 282 {253, 218, 224, 254, 225, 255, 251, 254}, 283 {254, 220, 225, 254, 225, 255, 251, 254}, 284 {254, 222, 227, 255, 226, 255, 251, 254}, 285 {254, 224, 228, 255, 227, 255, 251, 254}, 286 {254, 226, 230, 255, 228, 255, 251, 254}, 287 {255, 228, 231, 255, 230, 255, 251, 254}, 288 {255, 230, 233, 255, 231, 255, 252, 254}, 289 {255, 232, 235, 255, 232, 255, 252, 254}, 290 {255, 235, 237, 255, 233, 255, 252, 254}, 291 {255, 238, 240, 255, 235, 255, 252, 255}, 292 {255, 241, 243, 255, 236, 255, 252, 254}, 293 {255, 246, 247, 255, 239, 255, 253, 255} 294 }; 295 296 static void extend_model_to_full_distribution(vp9_prob p, 297 vp9_prob *tree_probs) { 298 const int l = (p - 1) / 2; 299 const vp9_prob (*model)[MODEL_NODES] = modelcoefprobs_pareto8; 300 if (p & 1) { 301 vpx_memcpy(tree_probs + UNCONSTRAINED_NODES, 302 model[l], MODEL_NODES * sizeof(vp9_prob)); 303 } else { 304 // interpolate 305 int i; 306 for (i = UNCONSTRAINED_NODES; i < ENTROPY_NODES; ++i) 307 tree_probs[i] = (model[l][i - UNCONSTRAINED_NODES] + 308 model[l + 1][i - UNCONSTRAINED_NODES]) >> 1; 309 } 310 } 311 312 void vp9_model_to_full_probs(const vp9_prob *model, vp9_prob *full) { 313 if (full != model) 314 vpx_memcpy(full, model, sizeof(vp9_prob) * UNCONSTRAINED_NODES); 315 extend_model_to_full_distribution(model[PIVOT_NODE], full); 316 } 317 318 static vp9_tree_index cat1[2], cat2[4], cat3[6], cat4[8], cat5[10], cat6[28]; 319 320 static void init_bit_tree(vp9_tree_index *p, int n) { 321 int i = 0; 322 323 while (++i < n) { 324 p[0] = p[1] = i << 1; 325 p += 2; 326 } 327 328 p[0] = p[1] = 0; 329 } 330 331 static void init_bit_trees() { 332 init_bit_tree(cat1, 1); 333 init_bit_tree(cat2, 2); 334 init_bit_tree(cat3, 3); 335 init_bit_tree(cat4, 4); 336 init_bit_tree(cat5, 5); 337 init_bit_tree(cat6, 14); 338 } 339 340 const vp9_extra_bit vp9_extra_bits[MAX_ENTROPY_TOKENS] = { 341 { 0, 0, 0, 0}, 342 { 0, 0, 0, 1}, 343 { 0, 0, 0, 2}, 344 { 0, 0, 0, 3}, 345 { 0, 0, 0, 4}, 346 { cat1, Pcat1, 1, 5}, 347 { cat2, Pcat2, 2, 7}, 348 { cat3, Pcat3, 3, 11}, 349 { cat4, Pcat4, 4, 19}, 350 { cat5, Pcat5, 5, 35}, 351 { cat6, Pcat6, 14, 67}, 352 { 0, 0, 0, 0} 353 }; 354 355 #include "vp9/common/vp9_default_coef_probs.h" 356 357 void vp9_default_coef_probs(VP9_COMMON *cm) { 358 vp9_copy(cm->fc.coef_probs[TX_4X4], default_coef_probs_4x4); 359 vp9_copy(cm->fc.coef_probs[TX_8X8], default_coef_probs_8x8); 360 vp9_copy(cm->fc.coef_probs[TX_16X16], default_coef_probs_16x16); 361 vp9_copy(cm->fc.coef_probs[TX_32X32], default_coef_probs_32x32); 362 } 363 364 void vp9_coef_tree_initialize() { 365 init_bit_trees(); 366 vp9_tokens_from_tree(vp9_coef_encodings, vp9_coef_tree); 367 } 368 369 // #define COEF_COUNT_TESTING 370 371 #define COEF_COUNT_SAT 24 372 #define COEF_MAX_UPDATE_FACTOR 112 373 #define COEF_COUNT_SAT_KEY 24 374 #define COEF_MAX_UPDATE_FACTOR_KEY 112 375 #define COEF_COUNT_SAT_AFTER_KEY 24 376 #define COEF_MAX_UPDATE_FACTOR_AFTER_KEY 128 377 378 static void adapt_coef_probs(VP9_COMMON *cm, TX_SIZE tx_size, 379 unsigned int count_sat, 380 unsigned int update_factor) { 381 const FRAME_CONTEXT *pre_fc = &cm->frame_contexts[cm->frame_context_idx]; 382 383 vp9_coeff_probs_model *dst_coef_probs = cm->fc.coef_probs[tx_size]; 384 const vp9_coeff_probs_model *pre_coef_probs = pre_fc->coef_probs[tx_size]; 385 vp9_coeff_count_model *coef_counts = cm->counts.coef[tx_size]; 386 unsigned int (*eob_branch_count)[REF_TYPES][COEF_BANDS][PREV_COEF_CONTEXTS] = 387 cm->counts.eob_branch[tx_size]; 388 int i, j, k, l, m; 389 unsigned int branch_ct[UNCONSTRAINED_NODES][2]; 390 391 for (i = 0; i < BLOCK_TYPES; ++i) 392 for (j = 0; j < REF_TYPES; ++j) 393 for (k = 0; k < COEF_BANDS; ++k) 394 for (l = 0; l < PREV_COEF_CONTEXTS; ++l) { 395 if (l >= 3 && k == 0) 396 continue; 397 vp9_tree_probs_from_distribution(vp9_coefmodel_tree, branch_ct, 398 coef_counts[i][j][k][l]); 399 branch_ct[0][1] = eob_branch_count[i][j][k][l] - branch_ct[0][0]; 400 for (m = 0; m < UNCONSTRAINED_NODES; ++m) 401 dst_coef_probs[i][j][k][l][m] = merge_probs( 402 pre_coef_probs[i][j][k][l][m], 403 branch_ct[m], 404 count_sat, update_factor); 405 } 406 } 407 408 void vp9_adapt_coef_probs(VP9_COMMON *cm) { 409 TX_SIZE t; 410 unsigned int count_sat, update_factor; 411 412 if (frame_is_intra_only(cm)) { 413 update_factor = COEF_MAX_UPDATE_FACTOR_KEY; 414 count_sat = COEF_COUNT_SAT_KEY; 415 } else if (cm->last_frame_type == KEY_FRAME) { 416 update_factor = COEF_MAX_UPDATE_FACTOR_AFTER_KEY; /* adapt quickly */ 417 count_sat = COEF_COUNT_SAT_AFTER_KEY; 418 } else { 419 update_factor = COEF_MAX_UPDATE_FACTOR; 420 count_sat = COEF_COUNT_SAT; 421 } 422 for (t = TX_4X4; t <= TX_32X32; t++) 423 adapt_coef_probs(cm, t, count_sat, update_factor); 424 } 425
