| 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 |