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 extern const int16_t *vp9_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
60 * fields are not.
61 */
62 extern const TOKENVALUE *vp9_dct_value_tokens_ptr;
63 extern const TOKENVALUE *vp9_dct_cat_lt_10_value_tokens;
64 extern const int16_t vp9_cat6_low_cost[256];
65 extern const int16_t vp9_cat6_high_cost[128];
66 extern const int16_t vp9_cat6_high10_high_cost[512];
67 extern const int16_t vp9_cat6_high12_high_cost[2048];
vp9_get_cost(int16_t token,EXTRABIT extrabits,const int16_t * cat6_high_table)68 static INLINE int16_t vp9_get_cost(int16_t token, EXTRABIT extrabits,
69 const int16_t *cat6_high_table) {
70 if (token != CATEGORY6_TOKEN)
71 return vp9_extra_bits[token].cost[extrabits];
72 return vp9_cat6_low_cost[extrabits & 0xff]
73 + cat6_high_table[extrabits >> 8];
74 }
75
76 #if CONFIG_VP9_HIGHBITDEPTH
vp9_get_high_cost_table(int bit_depth)77 static INLINE const int16_t* vp9_get_high_cost_table(int bit_depth) {
78 return bit_depth == 8 ? vp9_cat6_high_cost
79 : (bit_depth == 10 ? vp9_cat6_high10_high_cost :
80 vp9_cat6_high12_high_cost);
81 }
82 #else
vp9_get_high_cost_table(int bit_depth)83 static INLINE const int16_t* vp9_get_high_cost_table(int bit_depth) {
84 (void) bit_depth;
85 return vp9_cat6_high_cost;
86 }
87 #endif // CONFIG_VP9_HIGHBITDEPTH
88
vp9_get_token_extra(int v,int16_t * token,EXTRABIT * extra)89 static INLINE void vp9_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;
94 else
95 *extra = -2 * v - 2 * CAT6_MIN_VAL + 1;
96 return;
97 }
98 *token = vp9_dct_cat_lt_10_value_tokens[v].token;
99 *extra = vp9_dct_cat_lt_10_value_tokens[v].extra;
100 }
vp9_get_token(int v)101 static INLINE int16_t vp9_get_token(int v) {
102 if (v >= CAT6_MIN_VAL || v <= -CAT6_MIN_VAL)
103 return 10;
104 return vp9_dct_cat_lt_10_value_tokens[v].token;
105 }
106
107
108 #ifdef __cplusplus
109 } // extern "C"
110 #endif
111
112 #endif // VP9_ENCODER_VP9_TOKENIZE_H_
113