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 20extern "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 32typedef struct { 33 int16_t token; 34 EXTRABIT extra; 35} TOKENVALUE; 36 37typedef struct { 38 const vpx_prob *context_tree; 39 EXTRABIT extra; 40 uint8_t token; 41 uint8_t skip_eob_node; 42} TOKENEXTRA; 43 44extern const vpx_tree_index vp9_coef_tree[]; 45extern const vpx_tree_index vp9_coef_con_tree[]; 46extern const struct vp9_token vp9_coef_encodings[]; 47 48int vp9_is_skippable_in_plane(MACROBLOCK *x, BLOCK_SIZE bsize, int plane); 49int vp9_has_high_freq_in_plane(MACROBLOCK *x, BLOCK_SIZE bsize, int plane); 50 51struct VP9_COMP; 52struct ThreadData; 53 54void vp9_tokenize_sb(struct VP9_COMP *cpi, struct ThreadData *td, 55 TOKENEXTRA **t, int dry_run, BLOCK_SIZE bsize); 56 57extern 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 */ 62extern const TOKENVALUE *vp9_dct_value_tokens_ptr; 63extern const TOKENVALUE *vp9_dct_cat_lt_10_value_tokens; 64extern const int16_t vp9_cat6_low_cost[256]; 65extern const int16_t vp9_cat6_high_cost[128]; 66extern const int16_t vp9_cat6_high10_high_cost[512]; 67extern const int16_t vp9_cat6_high12_high_cost[2048]; 68static 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 77static 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 83static 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 89static 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} 101static 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