ref: 7777e7a8d5d4993825f441abd1dd9eb35680bf60
dir: /vp9/encoder/vp9_tokenize.h/
/* * Copyright (c) 2010 The WebM project authors. All Rights Reserved. * * Use of this source code is governed by a BSD-style license * that can be found in the LICENSE file in the root of the source * tree. An additional intellectual property rights grant can be found * in the file PATENTS. All contributing project authors may * be found in the AUTHORS file in the root of the source tree. */ #ifndef VP9_ENCODER_VP9_TOKENIZE_H_ #define VP9_ENCODER_VP9_TOKENIZE_H_ #include "vp9/common/vp9_entropy.h" #include "vp9/encoder/vp9_block.h" #include "vp9/encoder/vp9_treewriter.h" #ifdef __cplusplus extern "C" { #endif #define EOSB_TOKEN 127 // Not signalled, encoder only #if CONFIG_VP9_HIGHBITDEPTH typedef int32_t EXTRABIT; #else typedef int16_t EXTRABIT; #endif typedef struct { int16_t token; EXTRABIT extra; } TOKENVALUE; typedef struct { const vpx_prob *context_tree; EXTRABIT extra; uint8_t token; uint8_t skip_eob_node; } TOKENEXTRA; extern const vpx_tree_index vp9_coef_tree[]; extern const vpx_tree_index vp9_coef_con_tree[]; extern const struct vp9_token vp9_coef_encodings[]; int vp9_is_skippable_in_plane(MACROBLOCK *x, BLOCK_SIZE bsize, int plane); int vp9_has_high_freq_in_plane(MACROBLOCK *x, BLOCK_SIZE bsize, int plane); struct VP9_COMP; struct ThreadData; void vp9_tokenize_sb(struct VP9_COMP *cpi, struct ThreadData *td, TOKENEXTRA **t, int dry_run, BLOCK_SIZE bsize); extern const int16_t *vp9_dct_value_cost_ptr; /* TODO: The Token field should be broken out into a separate char array to * improve cache locality, since it's needed for costing when the rest of the * fields are not. */ extern const TOKENVALUE *vp9_dct_value_tokens_ptr; extern const TOKENVALUE *vp9_dct_cat_lt_10_value_tokens; extern const int16_t vp9_cat6_low_cost[256]; extern const int16_t vp9_cat6_high_cost[128]; extern const int16_t vp9_cat6_high10_high_cost[512]; extern const int16_t vp9_cat6_high12_high_cost[2048]; static INLINE int16_t vp9_get_cost(int16_t token, EXTRABIT extrabits, const int16_t *cat6_high_table) { if (token != CATEGORY6_TOKEN) return vp9_extra_bits[token].cost[extrabits]; return vp9_cat6_low_cost[extrabits & 0xff] + cat6_high_table[extrabits >> 8]; } #if CONFIG_VP9_HIGHBITDEPTH static INLINE const int16_t* vp9_get_high_cost_table(int bit_depth) { return bit_depth == 8 ? vp9_cat6_high_cost : (bit_depth == 10 ? vp9_cat6_high10_high_cost : vp9_cat6_high12_high_cost); } #else static INLINE const int16_t* vp9_get_high_cost_table(int bit_depth) { (void) bit_depth; return vp9_cat6_high_cost; } #endif // CONFIG_VP9_HIGHBITDEPTH static INLINE void vp9_get_token_extra(int v, int16_t *token, EXTRABIT *extra) { if (v >= CAT6_MIN_VAL || v <= -CAT6_MIN_VAL) { *token = CATEGORY6_TOKEN; if (v >= CAT6_MIN_VAL) *extra = 2 * v - 2 * CAT6_MIN_VAL; else *extra = -2 * v - 2 * CAT6_MIN_VAL + 1; return; } *token = vp9_dct_cat_lt_10_value_tokens[v].token; *extra = vp9_dct_cat_lt_10_value_tokens[v].extra; } static INLINE int16_t vp9_get_token(int v) { if (v >= CAT6_MIN_VAL || v <= -CAT6_MIN_VAL) return 10; return vp9_dct_cat_lt_10_value_tokens[v].token; } #ifdef __cplusplus } // extern "C" #endif #endif // VP9_ENCODER_VP9_TOKENIZE_H_