2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved.
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.
11 #ifndef VP10_ENCODER_TOKENIZE_H_
12 #define VP10_ENCODER_TOKENIZE_H_
14 #include "vp10/common/entropy.h"
16 #include "vp10/encoder/block.h"
17 #include "vp10/encoder/treewriter.h"
23 #define EOSB_TOKEN 127 // Not signalled, encoder only
25 #if CONFIG_VP9_HIGHBITDEPTH
26 typedef int32_t EXTRABIT;
28 typedef int16_t EXTRABIT;
38 const vpx_prob *context_tree;
41 uint8_t skip_eob_node;
44 extern const vpx_tree_index vp10_coef_tree[];
45 extern const vpx_tree_index vp10_coef_con_tree[];
46 extern const struct vp10_token vp10_coef_encodings[];
48 int vp10_is_skippable_in_plane(MACROBLOCK *x, BLOCK_SIZE bsize, int plane);
49 int vp10_has_high_freq_in_plane(MACROBLOCK *x, BLOCK_SIZE bsize, int plane);
54 void vp10_tokenize_sb(struct VP9_COMP *cpi, struct ThreadData *td,
55 TOKENEXTRA **t, int dry_run, BLOCK_SIZE bsize);
57 extern const int16_t *vp10_dct_value_cost_ptr;
58 /* TODO: The Token field should be broken out into a separate char array to
59 * improve cache locality, since it's needed for costing when the rest of the
62 extern const TOKENVALUE *vp10_dct_value_tokens_ptr;
63 extern const TOKENVALUE *vp10_dct_cat_lt_10_value_tokens;
64 extern const int16_t vp10_cat6_low_cost[256];
65 extern const int16_t vp10_cat6_high_cost[128];
66 extern const int16_t vp10_cat6_high10_high_cost[512];
67 extern const int16_t vp10_cat6_high12_high_cost[2048];
68 static INLINE int16_t vp10_get_cost(int16_t token, EXTRABIT extrabits,
69 const int16_t *cat6_high_table) {
70 if (token != CATEGORY6_TOKEN)
71 return vp10_extra_bits[token].cost[extrabits];
72 return vp10_cat6_low_cost[extrabits & 0xff]
73 + cat6_high_table[extrabits >> 8];
76 #if CONFIG_VP9_HIGHBITDEPTH
77 static INLINE const int16_t* vp10_get_high_cost_table(int bit_depth) {
78 return bit_depth == 8 ? vp10_cat6_high_cost
79 : (bit_depth == 10 ? vp10_cat6_high10_high_cost :
80 vp10_cat6_high12_high_cost);
83 static INLINE const int16_t* vp10_get_high_cost_table(int bit_depth) {
85 return vp10_cat6_high_cost;
87 #endif // CONFIG_VP9_HIGHBITDEPTH
89 static INLINE void vp10_get_token_extra(int v, int16_t *token, EXTRABIT *extra) {
90 if (v >= CAT6_MIN_VAL || v <= -CAT6_MIN_VAL) {
91 *token = CATEGORY6_TOKEN;
92 if (v >= CAT6_MIN_VAL)
93 *extra = 2 * v - 2 * CAT6_MIN_VAL;
95 *extra = -2 * v - 2 * CAT6_MIN_VAL + 1;
98 *token = vp10_dct_cat_lt_10_value_tokens[v].token;
99 *extra = vp10_dct_cat_lt_10_value_tokens[v].extra;
101 static INLINE int16_t vp10_get_token(int v) {
102 if (v >= CAT6_MIN_VAL || v <= -CAT6_MIN_VAL)
104 return vp10_dct_cat_lt_10_value_tokens[v].token;
112 #endif // VP10_ENCODER_TOKENIZE_H_