OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved. | 2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #ifndef VP9_ENCODER_VP9_TOKENIZE_H_ | 11 #ifndef VP9_ENCODER_VP9_TOKENIZE_H_ |
12 #define VP9_ENCODER_VP9_TOKENIZE_H_ | 12 #define VP9_ENCODER_VP9_TOKENIZE_H_ |
13 | 13 |
14 #include "vp9/common/vp9_entropy.h" | 14 #include "vp9/common/vp9_entropy.h" |
15 | 15 |
16 #include "vp9/encoder/vp9_block.h" | 16 #include "vp9/encoder/vp9_block.h" |
17 #include "vp9/encoder/vp9_treewriter.h" | 17 #include "vp9/encoder/vp9_treewriter.h" |
18 | 18 |
19 #ifdef __cplusplus | 19 #ifdef __cplusplus |
20 extern "C" { | 20 extern "C" { |
21 #endif | 21 #endif |
22 | 22 |
23 void vp9_tokenize_initialize(); | 23 #define EOSB_TOKEN 127 // Not signalled, encoder only |
24 | 24 |
25 #define EOSB_TOKEN 127 // Not signalled, encoder only | 25 #if CONFIG_VP9_HIGHBITDEPTH |
| 26 typedef int32_t EXTRABIT; |
| 27 #else |
| 28 typedef int16_t EXTRABIT; |
| 29 #endif |
| 30 |
26 | 31 |
27 typedef struct { | 32 typedef struct { |
28 int16_t token; | 33 int16_t token; |
29 #if CONFIG_VP9_HIGHBITDEPTH | 34 EXTRABIT extra; |
30 int32_t extra; | |
31 #else | |
32 int16_t extra; | |
33 #endif | |
34 } TOKENVALUE; | 35 } TOKENVALUE; |
35 | 36 |
36 typedef struct { | 37 typedef struct { |
37 const vp9_prob *context_tree; | 38 const vp9_prob *context_tree; |
38 #if CONFIG_VP9_HIGHBITDEPTH | 39 EXTRABIT extra; |
39 int32_t extra; | 40 uint8_t token; |
40 #else | 41 uint8_t skip_eob_node; |
41 int16_t extra; | |
42 #endif | |
43 uint8_t token; | |
44 uint8_t skip_eob_node; | |
45 } TOKENEXTRA; | 42 } TOKENEXTRA; |
46 | 43 |
47 extern const vp9_tree_index vp9_coef_tree[]; | 44 extern const vp9_tree_index vp9_coef_tree[]; |
48 extern const vp9_tree_index vp9_coef_con_tree[]; | 45 extern const vp9_tree_index vp9_coef_con_tree[]; |
49 extern struct vp9_token vp9_coef_encodings[]; | 46 extern const struct vp9_token vp9_coef_encodings[]; |
50 | 47 |
51 int vp9_is_skippable_in_plane(MACROBLOCK *x, BLOCK_SIZE bsize, int plane); | 48 int vp9_is_skippable_in_plane(MACROBLOCK *x, BLOCK_SIZE bsize, int plane); |
52 int vp9_has_high_freq_in_plane(MACROBLOCK *x, BLOCK_SIZE bsize, int plane); | 49 int vp9_has_high_freq_in_plane(MACROBLOCK *x, BLOCK_SIZE bsize, int plane); |
53 | 50 |
54 struct VP9_COMP; | 51 struct VP9_COMP; |
55 struct ThreadData; | 52 struct ThreadData; |
56 | 53 |
57 void vp9_tokenize_sb(struct VP9_COMP *cpi, struct ThreadData *td, | 54 void vp9_tokenize_sb(struct VP9_COMP *cpi, struct ThreadData *td, |
58 TOKENEXTRA **t, int dry_run, BLOCK_SIZE bsize); | 55 TOKENEXTRA **t, int dry_run, BLOCK_SIZE bsize); |
59 | 56 |
60 extern const int16_t *vp9_dct_value_cost_ptr; | 57 extern const int16_t *vp9_dct_value_cost_ptr; |
61 /* TODO: The Token field should be broken out into a separate char array to | 58 /* TODO: The Token field should be broken out into a separate char array to |
62 * improve cache locality, since it's needed for costing when the rest of the | 59 * improve cache locality, since it's needed for costing when the rest of the |
63 * fields are not. | 60 * fields are not. |
64 */ | 61 */ |
65 extern const TOKENVALUE *vp9_dct_value_tokens_ptr; | 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]; |
| 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 |
66 #if CONFIG_VP9_HIGHBITDEPTH | 76 #if CONFIG_VP9_HIGHBITDEPTH |
67 extern const int16_t *vp9_dct_value_cost_high10_ptr; | 77 static INLINE const int16_t* vp9_get_high_cost_table(int bit_depth) { |
68 extern const TOKENVALUE *vp9_dct_value_tokens_high10_ptr; | 78 return bit_depth == 8 ? vp9_cat6_high_cost |
69 extern const int16_t *vp9_dct_value_cost_high12_ptr; | 79 : (bit_depth == 10 ? vp9_cat6_high10_high_cost : |
70 extern const TOKENVALUE *vp9_dct_value_tokens_high12_ptr; | 80 vp9_cat6_high12_high_cost); |
| 81 } |
| 82 #else |
| 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 } |
71 #endif // CONFIG_VP9_HIGHBITDEPTH | 87 #endif // CONFIG_VP9_HIGHBITDEPTH |
72 | 88 |
| 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 } |
| 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 |
73 #ifdef __cplusplus | 108 #ifdef __cplusplus |
74 } // extern "C" | 109 } // extern "C" |
75 #endif | 110 #endif |
76 | 111 |
77 #endif // VP9_ENCODER_VP9_TOKENIZE_H_ | 112 #endif // VP9_ENCODER_VP9_TOKENIZE_H_ |
OLD | NEW |