f883b42cab
Cherry-Picked the following commits: 0defd8f Changed "WebM" to "AOMedia" & "webm" to "aomedia" 54e6676 Replace "VPx" by "AVx" 5082a36 Change "Vpx" to "Avx" 7df44f1 Replace "Vp9" w/ "Av1" 967f722 Remove kVp9CodecId 828f30c Change "Vp8" to "AOM" 030b5ff AUTHORS regenerated 2524cae Add ref-mv experimental flag 016762b Change copyright notice to AOMedia form 81e5526 Replace vp9 w/ av1 9b94565 Add missing files fa8ca9f Change "vp9" to "av1" ec838b7 Convert "vp8" to "aom" 80edfa0 Change "VP9" to "AV1" d1a11fb Change "vp8" to "aom" 7b58251 Point to WebM test data dd1a5c8 Replace "VP8" with "AOM" ff00fc0 Change "VPX" to "AOM" 01dee0b Change "vp10" to "av1" in source code cebe6f0 Convert "vpx" to "aom" 17b0567 rename vp10*.mk to av1_*.mk fe5f8a8 rename files vp10_* to av1_* Change-Id: I6fc3d18eb11fc171e46140c836ad5339cf6c9419
141 lines
4.4 KiB
C
141 lines
4.4 KiB
C
/*
|
|
* 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 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_AOM_HIGHBITDEPTH
|
|
typedef int32_t EXTRABIT;
|
|
#else
|
|
typedef int16_t EXTRABIT;
|
|
#endif
|
|
|
|
typedef struct {
|
|
int16_t token;
|
|
EXTRABIT extra;
|
|
} TOKENVALUE;
|
|
|
|
typedef struct {
|
|
const aom_prob *context_tree;
|
|
#if CONFIG_ANS
|
|
const rans_dec_lut *token_cdf;
|
|
#endif // CONFIG_ANS
|
|
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_ANS
|
|
extern const struct av1_token av1_coef_encodings[];
|
|
#endif // !CONFIG_ANS
|
|
|
|
int av1_is_skippable_in_plane(MACROBLOCK *x, BLOCK_SIZE bsize, int plane);
|
|
int av1_has_high_freq_in_plane(MACROBLOCK *x, BLOCK_SIZE bsize, int plane);
|
|
|
|
struct AV1_COMP;
|
|
struct ThreadData;
|
|
|
|
#if CONFIG_VAR_TX
|
|
void av1_tokenize_sb_inter(struct AV1_COMP *cpi, struct ThreadData *td,
|
|
TOKENEXTRA **t, int dry_run, int mi_row, int mi_col,
|
|
BLOCK_SIZE bsize);
|
|
#endif
|
|
|
|
void av1_tokenize_palette_sb(struct ThreadData *const td, BLOCK_SIZE bsize,
|
|
int plane, TOKENEXTRA **t);
|
|
void av1_tokenize_sb(struct AV1_COMP *cpi, struct ThreadData *td,
|
|
TOKENEXTRA **t, int dry_run, BLOCK_SIZE bsize);
|
|
#if CONFIG_SUPERTX
|
|
void av1_tokenize_sb_supertx(struct AV1_COMP *cpi, struct ThreadData *td,
|
|
TOKENEXTRA **t, int dry_run, BLOCK_SIZE bsize);
|
|
#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];
|
|
extern const int av1_cat6_high_cost[64];
|
|
extern const int av1_cat6_high10_high_cost[256];
|
|
extern const int av1_cat6_high12_high_cost[1024];
|
|
static INLINE int av1_get_cost(int16_t token, EXTRABIT extrabits,
|
|
const int *cat6_high_table) {
|
|
if (token != CATEGORY6_TOKEN)
|
|
return av1_extra_bits[token].cost[extrabits >> 1];
|
|
return av1_cat6_low_cost[(extrabits >> 1) & 0xff] +
|
|
cat6_high_table[extrabits >> 9];
|
|
}
|
|
|
|
#if CONFIG_AOM_HIGHBITDEPTH
|
|
static INLINE const int *av1_get_high_cost_table(int bit_depth) {
|
|
return bit_depth == 8 ? av1_cat6_high_cost
|
|
: (bit_depth == 10 ? av1_cat6_high10_high_cost
|
|
: av1_cat6_high12_high_cost);
|
|
}
|
|
#else
|
|
static INLINE const int *av1_get_high_cost_table(int bit_depth) {
|
|
(void)bit_depth;
|
|
return av1_cat6_high_cost;
|
|
}
|
|
#endif // CONFIG_AOM_HIGHBITDEPTH
|
|
|
|
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,
|
|
const int *cat6_high_table) {
|
|
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] +
|
|
cat6_high_table[extrabits >> 8];
|
|
}
|
|
*token = av1_dct_cat_lt_10_value_tokens[v].token;
|
|
return av1_dct_cat_lt_10_value_cost[v];
|
|
}
|
|
|
|
#ifdef __cplusplus
|
|
} // extern "C"
|
|
#endif
|
|
|
|
#endif // AV1_ENCODER_TOKENIZE_H_
|