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 #ifndef VP9_ENCODER_VP9_TOKENIZE_H_
12 #define VP9_ENCODER_VP9_TOKENIZE_H_
13
14 #include "vp9/common/vp9_entropy.h"
15
16 #include "vp9/encoder/vp9_block.h"
17 #include "vp9/encoder/vp9_treewriter.h"
18
19 #ifdef __cplusplus
20 extern "C" {
21 #endif
22
23 #define EOSB_TOKEN 127 // Not signalled, encoder only
24
25 #if CONFIG_VP9_HIGHBITDEPTH
26 typedef int32_t EXTRABIT;
27 #else
28 typedef int16_t EXTRABIT;
29 #endif
30
31
32 typedef struct {
33 int16_t token;
34 EXTRABIT extra;
35 } TOKENVALUE;
36
37 typedef struct {
38 const vpx_prob *context_tree;
39 EXTRABIT extra;
40 uint8_t token;
41 uint8_t skip_eob_node;
42 } TOKENEXTRA;
43
44 extern const vpx_tree_index vp9_coef_tree[];
45 extern const vpx_tree_index vp9_coef_con_tree[];
46 extern const struct vp9_token vp9_coef_encodings[];
47
48 int vp9_is_skippable_in_plane(MACROBLOCK *x, BLOCK_SIZE bsize, int plane);
49 int vp9_has_high_freq_in_plane(MACROBLOCK *x, BLOCK_SIZE bsize, int plane);
50
51 struct VP9_COMP;
52 struct ThreadData;
53
54 void vp9_tokenize_sb(struct VP9_COMP *cpi, struct ThreadData *td,
55 TOKENEXTRA **t, int dry_run, BLOCK_SIZE bsize);
56
57 typedef struct {
58 const vpx_prob *prob;
59 int len;
60 int base_val;
61 const int16_t *cost;
62 } vp9_extra_bit;
63
64 // indexed by token value
65 extern const vp9_extra_bit vp9_extra_bits[ENTROPY_TOKENS];
66 #if CONFIG_VP9_HIGHBITDEPTH
67 extern const vp9_extra_bit vp9_extra_bits_high10[ENTROPY_TOKENS];
68 extern const vp9_extra_bit vp9_extra_bits_high12[ENTROPY_TOKENS];
69 #endif // CONFIG_VP9_HIGHBITDEPTH
70
71 extern const int16_t *vp9_dct_value_cost_ptr;
72 /* TODO: The Token field should be broken out into a separate char array to
73 * improve cache locality, since it's needed for costing when the rest of the
74 * fields are not.
75 */
76 extern const TOKENVALUE *vp9_dct_value_tokens_ptr;
77 extern const TOKENVALUE *vp9_dct_cat_lt_10_value_tokens;
78 extern const int16_t vp9_cat6_low_cost[256];
79 extern const int16_t vp9_cat6_high_cost[128];
80 extern const int16_t vp9_cat6_high10_high_cost[512];
81 extern const int16_t vp9_cat6_high12_high_cost[2048];
vp9_get_cost(int16_t token,EXTRABIT extrabits,const int16_t * cat6_high_table)82 static INLINE int16_t vp9_get_cost(int16_t token, EXTRABIT extrabits,
83 const int16_t *cat6_high_table) {
84 if (token != CATEGORY6_TOKEN)
85 return vp9_extra_bits[token].cost[extrabits];
86 return vp9_cat6_low_cost[extrabits & 0xff]
87 + cat6_high_table[extrabits >> 8];
88 }
89
90 #if CONFIG_VP9_HIGHBITDEPTH
vp9_get_high_cost_table(int bit_depth)91 static INLINE const int16_t* vp9_get_high_cost_table(int bit_depth) {
92 return bit_depth == 8 ? vp9_cat6_high_cost
93 : (bit_depth == 10 ? vp9_cat6_high10_high_cost :
94 vp9_cat6_high12_high_cost);
95 }
96 #else
vp9_get_high_cost_table(int bit_depth)97 static INLINE const int16_t* vp9_get_high_cost_table(int bit_depth) {
98 (void) bit_depth;
99 return vp9_cat6_high_cost;
100 }
101 #endif // CONFIG_VP9_HIGHBITDEPTH
102
vp9_get_token_extra(int v,int16_t * token,EXTRABIT * extra)103 static INLINE void vp9_get_token_extra(int v, int16_t *token, EXTRABIT *extra) {
104 if (v >= CAT6_MIN_VAL || v <= -CAT6_MIN_VAL) {
105 *token = CATEGORY6_TOKEN;
106 if (v >= CAT6_MIN_VAL)
107 *extra = 2 * v - 2 * CAT6_MIN_VAL;
108 else
109 *extra = -2 * v - 2 * CAT6_MIN_VAL + 1;
110 return;
111 }
112 *token = vp9_dct_cat_lt_10_value_tokens[v].token;
113 *extra = vp9_dct_cat_lt_10_value_tokens[v].extra;
114 }
vp9_get_token(int v)115 static INLINE int16_t vp9_get_token(int v) {
116 if (v >= CAT6_MIN_VAL || v <= -CAT6_MIN_VAL)
117 return 10;
118 return vp9_dct_cat_lt_10_value_tokens[v].token;
119 }
120
121
122 #ifdef __cplusplus
123 } // extern "C"
124 #endif
125
126 #endif // VP9_ENCODER_VP9_TOKENIZE_H_
127