| /* |
| * Copyright (c) 2010 The WebM project authors. All Rights Reserved. |
| * |
| * Use of this source code is governed by a BSD-style license |
| * that can be found in the LICENSE file in the root of the source |
| * tree. An additional intellectual property rights grant can be found |
| * in the file PATENTS. All contributing project authors may |
| * be found in the AUTHORS file in the root of the source tree. |
| */ |
| |
| |
| #include <stdio.h> |
| |
| #include "entropy.h" |
| #include "string.h" |
| #include "blockd.h" |
| #include "onyxc_int.h" |
| #include "entropymode.h" |
| #include "vpx_mem/vpx_mem.h" |
| |
| #define uchar unsigned char /* typedefs can clash */ |
| #define uint unsigned int |
| |
| typedef const uchar cuchar; |
| typedef const uint cuint; |
| |
| typedef vp8_prob Prob; |
| |
| #include "coefupdateprobs.h" |
| |
| DECLARE_ALIGNED(16, const unsigned char, vp8_norm[256]) = |
| { |
| 0, 7, 6, 6, 5, 5, 5, 5, 4, 4, 4, 4, 4, 4, 4, 4, |
| 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, |
| 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, |
| 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, |
| 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, |
| 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, |
| 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, |
| 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, |
| 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, |
| 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, |
| 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, |
| 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, |
| 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, |
| 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, |
| 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, |
| 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 |
| }; |
| |
| DECLARE_ALIGNED(16, cuchar, vp8_coef_bands[16]) = |
| { 0, 1, 2, 3, 6, 4, 5, 6, 6, 6, 6, 6, 6, 7, 7, 7}; |
| |
| DECLARE_ALIGNED(16, cuchar, vp8_prev_token_class[MAX_ENTROPY_TOKENS]) = |
| #if CONFIG_EXPANDED_COEF_CONTEXT |
| { 0, 1, 2, 2, 3, 3, 3, 3, 3, 3, 3, 0}; |
| #else |
| { 0, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0}; |
| #endif |
| |
| DECLARE_ALIGNED(16, const int, vp8_default_zig_zag1d[16]) = |
| { |
| 0, 1, 4, 8, |
| 5, 2, 3, 6, |
| 9, 12, 13, 10, |
| 7, 11, 14, 15, |
| }; |
| DECLARE_ALIGNED(64, cuchar, vp8_coef_bands_8x8[64]) = { 0, 1, 2, 3, 5, 4, 4, 5, |
| 5, 3, 6, 3, 5, 4, 6, 6, |
| 6, 5, 5, 6, 6, 6, 6, 6, |
| 6, 6, 6, 6, 6, 6, 6, 6, |
| 6, 6, 6, 6, 7, 7, 7, 7, |
| 7, 7, 7, 7, 7, 7, 7, 7, |
| 7, 7, 7, 7, 7, 7, 7, 7, |
| 7, 7, 7, 7, 7, 7, 7, 7 |
| }; |
| DECLARE_ALIGNED(64, const int, vp8_default_zig_zag1d_8x8[64]) = |
| { |
| 0, 1, 8, 16, 9, 2, 3, 10, 17, 24, 32, 25, 18, 11, 4, 5, |
| 12, 19, 26, 33, 40, 48, 41, 34, 27, 20, 13, 6, 7, 14, 21, 28, |
| 35, 42, 49, 56, 57, 50, 43, 36, 29, 22, 15, 23, 30, 37, 44, 51, |
| 58, 59, 52, 45, 38, 31, 39, 46, 53, 60, 61, 54, 47, 55, 62, 63, |
| }; |
| |
| |
| DECLARE_ALIGNED(16, short, vp8_default_zig_zag_mask[16]); |
| DECLARE_ALIGNED(64, short, vp8_default_zig_zag_mask_8x8[64]);//int64_t |
| |
| /* Array indices are identical to previously-existing CONTEXT_NODE indices */ |
| |
| const vp8_tree_index vp8_coef_tree[ 22] = /* corresponding _CONTEXT_NODEs */ |
| { |
| -DCT_EOB_TOKEN, 2, /* 0 = EOB */ |
| -ZERO_TOKEN, 4, /* 1 = ZERO */ |
| -ONE_TOKEN, 6, /* 2 = ONE */ |
| 8, 12, /* 3 = LOW_VAL */ |
| -TWO_TOKEN, 10, /* 4 = TWO */ |
| -THREE_TOKEN, -FOUR_TOKEN, /* 5 = THREE */ |
| 14, 16, /* 6 = HIGH_LOW */ |
| -DCT_VAL_CATEGORY1, -DCT_VAL_CATEGORY2, /* 7 = CAT_ONE */ |
| 18, 20, /* 8 = CAT_THREEFOUR */ |
| -DCT_VAL_CATEGORY3, -DCT_VAL_CATEGORY4, /* 9 = CAT_THREE */ |
| -DCT_VAL_CATEGORY5, -DCT_VAL_CATEGORY6 /* 10 = CAT_FIVE */ |
| }; |
| |
| struct vp8_token_struct vp8_coef_encodings[MAX_ENTROPY_TOKENS]; |
| |
| /* Trees for extra bits. Probabilities are constant and |
| do not depend on previously encoded bits */ |
| |
| static const Prob Pcat1[] = { 159}; |
| static const Prob Pcat2[] = { 165, 145}; |
| static const Prob Pcat3[] = { 173, 148, 140}; |
| static const Prob Pcat4[] = { 176, 155, 140, 135}; |
| static const Prob Pcat5[] = { 180, 157, 141, 134, 130}; |
| static const Prob Pcat6[] = |
| { 254, 254, 252, 249, 243, 230, 196, 177, 153, 140, 133, 130, 129}; |
| |
| static vp8_tree_index cat1[2], cat2[4], cat3[6], cat4[8], cat5[10], cat6[26]; |
| |
| void vp8_init_scan_order_mask() |
| { |
| int i; |
| |
| for (i = 0; i < 16; i++) |
| { |
| vp8_default_zig_zag_mask[vp8_default_zig_zag1d[i]] = 1 << i; |
| } |
| for (i = 0; i < 64; i++) |
| { |
| vp8_default_zig_zag_mask_8x8[vp8_default_zig_zag1d_8x8[i]] = 1 << i; |
| } |
| } |
| |
| static void init_bit_tree(vp8_tree_index *p, int n) |
| { |
| int i = 0; |
| |
| while (++i < n) |
| { |
| p[0] = p[1] = i << 1; |
| p += 2; |
| } |
| |
| p[0] = p[1] = 0; |
| } |
| |
| static void init_bit_trees() |
| { |
| init_bit_tree(cat1, 1); |
| init_bit_tree(cat2, 2); |
| init_bit_tree(cat3, 3); |
| init_bit_tree(cat4, 4); |
| init_bit_tree(cat5, 5); |
| init_bit_tree(cat6, 13); |
| } |
| |
| vp8_extra_bit_struct vp8_extra_bits[12] = |
| { |
| { 0, 0, 0, 0}, |
| { 0, 0, 0, 1}, |
| { 0, 0, 0, 2}, |
| { 0, 0, 0, 3}, |
| { 0, 0, 0, 4}, |
| { cat1, Pcat1, 1, 5}, |
| { cat2, Pcat2, 2, 7}, |
| { cat3, Pcat3, 3, 11}, |
| { cat4, Pcat4, 4, 19}, |
| { cat5, Pcat5, 5, 35}, |
| { cat6, Pcat6, 13, 67}, |
| { 0, 0, 0, 0} |
| }; |
| |
| #if CONFIG_NEWUPDATE |
| const vp8_prob updprobs[4] = {128, 136, 120, 112}; |
| #endif |
| |
| #include "default_coef_probs.h" |
| |
| void vp8_default_coef_probs(VP8_COMMON *pc) |
| { |
| vpx_memcpy(pc->fc.coef_probs, default_coef_probs, |
| sizeof(default_coef_probs)); |
| |
| vpx_memcpy(pc->fc.coef_probs_8x8, vp8_default_coef_probs_8x8, |
| sizeof(vp8_default_coef_probs_8x8)); |
| |
| } |
| |
| void vp8_coef_tree_initialize() |
| { |
| init_bit_trees(); |
| vp8_tokens_from_tree(vp8_coef_encodings, vp8_coef_tree); |
| } |
| |
| #if CONFIG_ADAPTIVE_ENTROPY |
| |
| //#define COEF_COUNT_TESTING |
| |
| #define COEF_COUNT_SAT 24 |
| #define COEF_MAX_UPDATE_FACTOR 112 |
| #define COEF_COUNT_SAT_KEY 24 |
| #define COEF_MAX_UPDATE_FACTOR_KEY 112 |
| #define COEF_COUNT_SAT_AFTER_KEY 24 |
| #define COEF_MAX_UPDATE_FACTOR_AFTER_KEY 128 |
| |
| void vp8_adapt_coef_probs(VP8_COMMON *cm) |
| { |
| int t, i, j, k, count; |
| unsigned int branch_ct[ENTROPY_NODES][2]; |
| vp8_prob coef_probs[ENTROPY_NODES]; |
| int update_factor; /* denominator 256 */ |
| int factor; |
| int count_sat; |
| |
| //printf("Frame type: %d\n", cm->frame_type); |
| if (cm->frame_type == KEY_FRAME) |
| { |
| update_factor = COEF_MAX_UPDATE_FACTOR_KEY; |
| count_sat = COEF_COUNT_SAT_KEY; |
| } |
| else if (cm->last_frame_type == KEY_FRAME) |
| { |
| update_factor = COEF_MAX_UPDATE_FACTOR_AFTER_KEY; /* adapt quickly */ |
| count_sat = COEF_COUNT_SAT_AFTER_KEY; |
| } |
| else |
| { |
| update_factor = COEF_MAX_UPDATE_FACTOR; |
| count_sat = COEF_COUNT_SAT; |
| } |
| |
| #ifdef COEF_COUNT_TESTING |
| { |
| printf("static const unsigned int\ncoef_counts" |
| "[BLOCK_TYPES] [COEF_BANDS]" |
| "[PREV_COEF_CONTEXTS] [MAX_ENTROPY_TOKENS] = {\n"); |
| for (i = 0; i<BLOCK_TYPES; ++i) |
| { |
| printf(" {\n"); |
| for (j = 0; j<COEF_BANDS; ++j) |
| { |
| printf(" {\n"); |
| for (k = 0; k<PREV_COEF_CONTEXTS; ++k) |
| { |
| printf(" {"); |
| for (t = 0; t<MAX_ENTROPY_TOKENS; ++t) printf("%d, ", cm->fc.coef_counts[i][j][k][t]); |
| printf("},\n"); |
| } |
| printf(" },\n"); |
| } |
| printf(" },\n"); |
| } |
| printf("};\n"); |
| printf("static const unsigned int\ncoef_counts_8x8" |
| "[BLOCK_TYPES_8X8] [COEF_BANDS]" |
| "[PREV_COEF_CONTEXTS] [MAX_ENTROPY_TOKENS] = {\n"); |
| for (i = 0; i<BLOCK_TYPES_8X8; ++i) |
| { |
| printf(" {\n"); |
| for (j = 0; j<COEF_BANDS; ++j) |
| { |
| printf(" {\n"); |
| for (k = 0; k<PREV_COEF_CONTEXTS; ++k) |
| { |
| printf(" {"); |
| for (t = 0; t<MAX_ENTROPY_TOKENS; ++t) printf("%d, ", cm->fc.coef_counts_8x8[i][j][k][t]); |
| printf("},\n"); |
| } |
| printf(" },\n"); |
| } |
| printf(" },\n"); |
| } |
| printf("};\n"); |
| } |
| #endif |
| |
| for (i = 0; i<BLOCK_TYPES; ++i) |
| for (j = 0; j<COEF_BANDS; ++j) |
| for (k = 0; k<PREV_COEF_CONTEXTS; ++k) |
| { |
| #if CONFIG_EXPANDED_COEF_CONTEXT |
| if (k >=3 && ((i == 0 && j == 1) || (i > 0 && j == 0))) |
| continue; |
| #endif |
| vp8_tree_probs_from_distribution( |
| MAX_ENTROPY_TOKENS, vp8_coef_encodings, vp8_coef_tree, |
| coef_probs, branch_ct, cm->fc.coef_counts [i][j][k], |
| 256, 1); |
| for (t=0; t<ENTROPY_NODES; ++t) |
| { |
| int prob; |
| count = branch_ct[t][0] + branch_ct[t][1]; |
| count = count > count_sat ? count_sat : count; |
| factor = (update_factor * count / count_sat); |
| prob = ((int)cm->fc.pre_coef_probs[i][j][k][t] * (256-factor) + |
| (int)coef_probs[t] * factor + 128) >> 8; |
| if (prob <= 0) cm->fc.coef_probs[i][j][k][t] = 1; |
| else if (prob > 255) cm->fc.coef_probs[i][j][k][t] = 255; |
| else cm->fc.coef_probs[i][j][k][t] = prob; |
| } |
| } |
| |
| for (i = 0; i<BLOCK_TYPES_8X8; ++i) |
| for (j = 0; j<COEF_BANDS; ++j) |
| for (k = 0; k<PREV_COEF_CONTEXTS; ++k) |
| { |
| #if CONFIG_EXPANDED_COEF_CONTEXT |
| if (k >=3 && ((i == 0 && j == 1) || (i > 0 && j == 0))) |
| continue; |
| #endif |
| vp8_tree_probs_from_distribution( |
| MAX_ENTROPY_TOKENS, vp8_coef_encodings, vp8_coef_tree, |
| coef_probs, branch_ct, cm->fc.coef_counts_8x8 [i][j][k], |
| 256, 1); |
| for (t=0; t<ENTROPY_NODES; ++t) |
| { |
| int prob; |
| count = branch_ct[t][0] + branch_ct[t][1]; |
| count = count > count_sat ? count_sat : count; |
| factor = (update_factor * count / count_sat); |
| prob = ((int)cm->fc.pre_coef_probs_8x8[i][j][k][t] * (256-factor) + |
| (int)coef_probs[t] * factor + 128) >> 8; |
| if (prob <= 0) cm->fc.coef_probs_8x8[i][j][k][t] = 1; |
| else if (prob > 255) cm->fc.coef_probs_8x8[i][j][k][t] = 255; |
| else cm->fc.coef_probs_8x8[i][j][k][t] = prob; |
| } |
| } |
| } |
| #endif |
