diff options
Diffstat (limited to 'third_party/aom/av1/encoder/tokenize.h')
| -rw-r--r-- | third_party/aom/av1/encoder/tokenize.h | 151 |
1 files changed, 151 insertions, 0 deletions
diff --git a/third_party/aom/av1/encoder/tokenize.h b/third_party/aom/av1/encoder/tokenize.h new file mode 100644 index 000000000..3928111d6 --- /dev/null +++ b/third_party/aom/av1/encoder/tokenize.h @@ -0,0 +1,151 @@ +/* + * Copyright (c) 2016, Alliance for Open Media. All rights reserved + * + * This source code is subject to the terms of the BSD 2 Clause License and + * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License + * was not distributed with this source code in the LICENSE file, you can + * obtain it at www.aomedia.org/license/software. If the Alliance for Open + * Media Patent License 1.0 was not distributed with this source code in the + * PATENTS file, you can obtain it at www.aomedia.org/license/patent. + */ + +#ifndef AV1_ENCODER_TOKENIZE_H_ +#define AV1_ENCODER_TOKENIZE_H_ + +#include "av1/common/entropy.h" + +#include "av1/encoder/block.h" +#include "av1/encoder/treewriter.h" + +#ifdef __cplusplus +extern "C" { +#endif + +#define EOSB_TOKEN 127 // Not signalled, encoder only + +#if CONFIG_HIGHBITDEPTH +typedef int32_t EXTRABIT; +#else +typedef int16_t EXTRABIT; +#endif + +typedef struct { + int16_t token; + EXTRABIT extra; +} TOKENVALUE; + +typedef struct { +#if CONFIG_NEW_TOKENSET + aom_cdf_prob (*tail_cdf)[CDF_SIZE(ENTROPY_TOKENS)]; + aom_cdf_prob (*head_cdf)[CDF_SIZE(ENTROPY_TOKENS)]; + int eob_val; + int first_val; +#elif CONFIG_EC_MULTISYMBOL + aom_cdf_prob (*token_cdf)[CDF_SIZE(ENTROPY_TOKENS)]; +#endif + const aom_prob *context_tree; + EXTRABIT extra; + uint8_t token; + uint8_t skip_eob_node; +} TOKENEXTRA; + +extern const aom_tree_index av1_coef_tree[]; +extern const aom_tree_index av1_coef_con_tree[]; +#if !CONFIG_EC_MULTISYMBOL +extern const struct av1_token av1_coef_encodings[]; +#endif // !CONFIG_EC_MULTISYMBOL + +int av1_is_skippable_in_plane(MACROBLOCK *x, BLOCK_SIZE bsize, int plane); + +struct AV1_COMP; +struct ThreadData; + +typedef enum { + OUTPUT_ENABLED = 0, + DRY_RUN_NORMAL, + DRY_RUN_COSTCOEFFS, +} RUN_TYPE; + +// Note in all the tokenize functions rate if non NULL is incremented +// with the coefficient token cost only if dry_run = DRY_RUN_COSTCOEFS, +// otherwise rate is not incremented. +#if CONFIG_VAR_TX +void av1_tokenize_sb_vartx(const struct AV1_COMP *cpi, struct ThreadData *td, + TOKENEXTRA **t, RUN_TYPE dry_run, int mi_row, + int mi_col, BLOCK_SIZE bsize, int *rate); +#endif +#if CONFIG_PALETTE +void av1_tokenize_palette_sb(const struct AV1_COMP *cpi, + const struct ThreadData *const td, int plane, + TOKENEXTRA **t, RUN_TYPE dry_run, BLOCK_SIZE bsize, + int *rate); +#endif // CONFIG_PALETTE +void av1_tokenize_sb(const struct AV1_COMP *cpi, struct ThreadData *td, + TOKENEXTRA **t, RUN_TYPE dry_run, BLOCK_SIZE bsize, + int *rate, const int mi_row, const int mi_col); +#if CONFIG_SUPERTX +void av1_tokenize_sb_supertx(const struct AV1_COMP *cpi, struct ThreadData *td, + TOKENEXTRA **t, RUN_TYPE dry_run, BLOCK_SIZE bsize, + int *rate); +#endif + +extern const int16_t *av1_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 *av1_dct_value_tokens_ptr; +extern const TOKENVALUE *av1_dct_cat_lt_10_value_tokens; +extern const int *av1_dct_cat_lt_10_value_cost; +extern const int16_t av1_cat6_low_cost[256]; +#if CONFIG_HIGHBITDEPTH +#define CAT6_HIGH_COST_ENTRIES 1024 +#else +#define CAT6_HIGH_COST_ENTRIES 64 +#endif +extern const int av1_cat6_high_cost[CAT6_HIGH_COST_ENTRIES]; +extern const uint8_t av1_cat6_skipped_bits_discount[8]; + +static INLINE void av1_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 = av1_dct_cat_lt_10_value_tokens[v].token; + *extra = av1_dct_cat_lt_10_value_tokens[v].extra; +} +static INLINE int16_t av1_get_token(int v) { + if (v >= CAT6_MIN_VAL || v <= -CAT6_MIN_VAL) return 10; + return av1_dct_cat_lt_10_value_tokens[v].token; +} + +static INLINE int av1_get_token_cost(int v, int16_t *token, int cat6_bits) { + if (v >= CAT6_MIN_VAL || v <= -CAT6_MIN_VAL) { + EXTRABIT extrabits; + *token = CATEGORY6_TOKEN; + extrabits = abs(v) - CAT6_MIN_VAL; + return av1_cat6_low_cost[extrabits & 0xff] + + av1_cat6_high_cost[extrabits >> 8] - + av1_cat6_skipped_bits_discount[18 - cat6_bits]; + } + *token = av1_dct_cat_lt_10_value_tokens[v].token; + return av1_dct_cat_lt_10_value_cost[v]; +} + +#if !CONFIG_PVQ || CONFIG_VAR_TX +static INLINE int get_tx_eob(const struct segmentation *seg, int segment_id, + TX_SIZE tx_size) { + const int eob_max = tx_size_2d[tx_size]; + return segfeature_active(seg, segment_id, SEG_LVL_SKIP) ? 0 : eob_max; +} +#endif + +#ifdef __cplusplus +} // extern "C" +#endif + +#endif // AV1_ENCODER_TOKENIZE_H_ |