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