Commit 85e3b960 authored by Angie Chiang's avatar Angie Chiang

Add symbol rate accounting code for lv_map

Change-Id: I03adc36f5e5f839d23c4f50a864d317d008e30a6
parent dff57e0a
...@@ -89,7 +89,8 @@ set(AOM_AV1_DECODER_SOURCES ...@@ -89,7 +89,8 @@ set(AOM_AV1_DECODER_SOURCES
"${AOM_ROOT}/av1/decoder/dsubexp.c" "${AOM_ROOT}/av1/decoder/dsubexp.c"
"${AOM_ROOT}/av1/decoder/dsubexp.h" "${AOM_ROOT}/av1/decoder/dsubexp.h"
"${AOM_ROOT}/av1/decoder/dthread.c" "${AOM_ROOT}/av1/decoder/dthread.c"
"${AOM_ROOT}/av1/decoder/dthread.h") "${AOM_ROOT}/av1/decoder/dthread.h"
"${AOM_ROOT}/av1/decoder/symbolrate.h")
set(AOM_AV1_ENCODER_SOURCES set(AOM_AV1_ENCODER_SOURCES
"${AOM_ROOT}/av1/av1_cx_iface.c" "${AOM_ROOT}/av1/av1_cx_iface.c"
......
...@@ -32,6 +32,7 @@ AV1_DX_SRCS-yes += decoder/decoder.c ...@@ -32,6 +32,7 @@ AV1_DX_SRCS-yes += decoder/decoder.c
AV1_DX_SRCS-yes += decoder/decoder.h AV1_DX_SRCS-yes += decoder/decoder.h
AV1_DX_SRCS-yes += decoder/dsubexp.c AV1_DX_SRCS-yes += decoder/dsubexp.c
AV1_DX_SRCS-yes += decoder/dsubexp.h AV1_DX_SRCS-yes += decoder/dsubexp.h
AV1_DX_SRCS-yes += decoder/symbolrate.h
ifeq ($(CONFIG_ACCOUNTING),yes) ifeq ($(CONFIG_ACCOUNTING),yes)
AV1_DX_SRCS-yes += decoder/accounting.h AV1_DX_SRCS-yes += decoder/accounting.h
......
...@@ -441,6 +441,10 @@ typedef struct FRAME_COUNTS { ...@@ -441,6 +441,10 @@ typedef struct FRAME_COUNTS {
#endif // CONFIG_CTX1D #endif // CONFIG_CTX1D
#endif // CONFIG_LV_MAP #endif // CONFIG_LV_MAP
#if CONFIG_SYMBOLRATE
unsigned int symbol_num[2]; // 0: entropy symbol 1: non-entropy symbol
#endif
unsigned int newmv_mode[NEWMV_MODE_CONTEXTS][2]; unsigned int newmv_mode[NEWMV_MODE_CONTEXTS][2];
unsigned int zeromv_mode[ZEROMV_MODE_CONTEXTS][2]; unsigned int zeromv_mode[ZEROMV_MODE_CONTEXTS][2];
unsigned int refmv_mode[REFMV_MODE_CONTEXTS][2]; unsigned int refmv_mode[REFMV_MODE_CONTEXTS][2];
......
...@@ -64,6 +64,7 @@ ...@@ -64,6 +64,7 @@
#endif #endif
#include "av1/decoder/detokenize.h" #include "av1/decoder/detokenize.h"
#include "av1/decoder/dsubexp.h" #include "av1/decoder/dsubexp.h"
#include "av1/decoder/symbolrate.h"
#if CONFIG_WARPED_MOTION || CONFIG_GLOBAL_MOTION #if CONFIG_WARPED_MOTION || CONFIG_GLOBAL_MOTION
#include "av1/common/warped_motion.h" #include "av1/common/warped_motion.h"
...@@ -5199,7 +5200,7 @@ static int read_compressed_header(AV1Decoder *pbi, const uint8_t *data, ...@@ -5199,7 +5200,7 @@ static int read_compressed_header(AV1Decoder *pbi, const uint8_t *data,
#endif #endif
#if CONFIG_LV_MAP #if CONFIG_LV_MAP
av1_read_txb_probs(fc, cm->tx_mode, &r); av1_read_txb_probs(fc, cm->tx_mode, &r, &cm->counts);
#endif // CONFIG_LV_MAP #endif // CONFIG_LV_MAP
#if !CONFIG_NEW_MULTISYMBOL #if !CONFIG_NEW_MULTISYMBOL
...@@ -5704,6 +5705,9 @@ void av1_decode_tg_tiles_and_wrapup(AV1Decoder *pbi, const uint8_t *data, ...@@ -5704,6 +5705,9 @@ void av1_decode_tg_tiles_and_wrapup(AV1Decoder *pbi, const uint8_t *data,
#if CONFIG_LV_MAP #if CONFIG_LV_MAP
av1_adapt_coef_probs(cm); av1_adapt_coef_probs(cm);
#endif // CONFIG_LV_MAP #endif // CONFIG_LV_MAP
#if CONFIG_SYMBOLRATE
av1_dump_symbol_rate(cm);
#endif
av1_adapt_intra_frame_probs(cm); av1_adapt_intra_frame_probs(cm);
av1_average_tile_coef_cdfs(pbi->common.fc, tile_ctxs, cdf_ptrs, av1_average_tile_coef_cdfs(pbi->common.fc, tile_ctxs, cdf_ptrs,
cm->tile_rows * cm->tile_cols); cm->tile_rows * cm->tile_cols);
......
...@@ -15,16 +15,20 @@ ...@@ -15,16 +15,20 @@
#include "av1/decoder/decodemv.h" #include "av1/decoder/decodemv.h"
#include "av1/decoder/decodetxb.h" #include "av1/decoder/decodetxb.h"
#include "av1/decoder/dsubexp.h" #include "av1/decoder/dsubexp.h"
#include "av1/decoder/symbolrate.h"
#define ACCT_STR __func__ #define ACCT_STR __func__
static int read_golomb(MACROBLOCKD *xd, aom_reader *r) { static int read_golomb(MACROBLOCKD *xd, aom_reader *r, FRAME_COUNTS *counts) {
#if !CONFIG_SYMBOLRATE
(void)counts;
#endif
int x = 1; int x = 1;
int length = 0; int length = 0;
int i = 0; int i = 0;
while (!i) { while (!i) {
i = aom_read_bit(r, ACCT_STR); i = av1_read_record_bit(counts, r, ACCT_STR);
++length; ++length;
if (length >= 32) { if (length >= 32) {
aom_internal_error(xd->error_info, AOM_CODEC_CORRUPT_FRAME, aom_internal_error(xd->error_info, AOM_CODEC_CORRUPT_FRAME,
...@@ -35,7 +39,7 @@ static int read_golomb(MACROBLOCKD *xd, aom_reader *r) { ...@@ -35,7 +39,7 @@ static int read_golomb(MACROBLOCKD *xd, aom_reader *r) {
for (i = 0; i < length - 1; ++i) { for (i = 0; i < length - 1; ++i) {
x <<= 1; x <<= 1;
x += aom_read_bit(r, ACCT_STR); x += av1_read_record_bit(counts, r, ACCT_STR);
} }
return x - 1; return x - 1;
...@@ -72,8 +76,9 @@ static INLINE int read_nz_map(aom_reader *r, tran_low_t *tcoeffs, int plane, ...@@ -72,8 +76,9 @@ static INLINE int read_nz_map(aom_reader *r, tran_low_t *tcoeffs, int plane,
if (c < seg_eob - 1) { if (c < seg_eob - 1) {
#if LV_MAP_PROB #if LV_MAP_PROB
is_nz = aom_read_symbol(r, fc->nz_map_cdf[txs_ctx][plane_type][coeff_ctx], is_nz = av1_read_record_symbol(
2, ACCT_STR); counts, r, fc->nz_map_cdf[txs_ctx][plane_type][coeff_ctx], 2,
ACCT_STR);
#else #else
is_nz = aom_read(r, nz_map[coeff_ctx], ACCT_STR); is_nz = aom_read(r, nz_map[coeff_ctx], ACCT_STR);
#endif #endif
...@@ -93,8 +98,9 @@ static INLINE int read_nz_map(aom_reader *r, tran_low_t *tcoeffs, int plane, ...@@ -93,8 +98,9 @@ static INLINE int read_nz_map(aom_reader *r, tran_low_t *tcoeffs, int plane,
if (is_nz) { if (is_nz) {
#if LV_MAP_PROB #if LV_MAP_PROB
int is_eob = aom_read_symbol( int is_eob = av1_read_record_symbol(
r, fc->eob_flag_cdf[txs_ctx][plane_type][eob_ctx], 2, ACCT_STR); counts, r, fc->eob_flag_cdf[txs_ctx][plane_type][eob_ctx], 2,
ACCT_STR);
#else #else
int is_eob = aom_read(r, eob_flag[eob_ctx], ACCT_STR); int is_eob = aom_read(r, eob_flag[eob_ctx], ACCT_STR);
#endif #endif
...@@ -124,8 +130,8 @@ static INLINE int read_nz_map_vert(aom_reader *r, tran_low_t *tcoeffs, ...@@ -124,8 +130,8 @@ static INLINE int read_nz_map_vert(aom_reader *r, tran_low_t *tcoeffs,
for (int col = 0; col < width; ++col) { for (int col = 0; col < width; ++col) {
int el_ctx = get_empty_line_ctx(col, eob_ls); int el_ctx = get_empty_line_ctx(col, eob_ls);
#if LV_MAP_PROB #if LV_MAP_PROB
int empty_line = aom_read_symbol( int empty_line = av1_read_record_symbol(
r, fc->empty_line_cdf[txs_ctx][plane_type][tx_class][el_ctx], 2, counts, r, fc->empty_line_cdf[txs_ctx][plane_type][tx_class][el_ctx], 2,
ACCT_STR); ACCT_STR);
#else #else
int empty_line = aom_read( int empty_line = aom_read(
...@@ -141,8 +147,9 @@ static INLINE int read_nz_map_vert(aom_reader *r, tran_low_t *tcoeffs, ...@@ -141,8 +147,9 @@ static INLINE int read_nz_map_vert(aom_reader *r, tran_low_t *tcoeffs,
get_nz_map_ctx(tcoeffs, coeff_idx, bwl, height, tx_type); get_nz_map_ctx(tcoeffs, coeff_idx, bwl, height, tx_type);
if (row + 1 != height) { if (row + 1 != height) {
#if LV_MAP_PROB #if LV_MAP_PROB
int is_nz = aom_read_symbol( int is_nz = av1_read_record_symbol(
r, fc->nz_map_cdf[txs_ctx][plane_type][coeff_ctx], 2, ACCT_STR); counts, r, fc->nz_map_cdf[txs_ctx][plane_type][coeff_ctx], 2,
ACCT_STR);
#else #else
int is_nz = aom_read(r, nz_map[coeff_ctx], ACCT_STR); int is_nz = aom_read(r, nz_map[coeff_ctx], ACCT_STR);
#endif #endif
...@@ -153,8 +160,9 @@ static INLINE int read_nz_map_vert(aom_reader *r, tran_low_t *tcoeffs, ...@@ -153,8 +160,9 @@ static INLINE int read_nz_map_vert(aom_reader *r, tran_low_t *tcoeffs,
if (row + 1 != height) { if (row + 1 != height) {
int eob_ctx = get_hv_eob_ctx(col, row, eob_ls); int eob_ctx = get_hv_eob_ctx(col, row, eob_ls);
#if LV_MAP_PROB #if LV_MAP_PROB
int is_eob = aom_read_symbol( int is_eob = av1_read_record_symbol(
r, fc->hv_eob_cdf[txs_ctx][plane_type][tx_class][eob_ctx], 2, counts, r,
fc->hv_eob_cdf[txs_ctx][plane_type][tx_class][eob_ctx], 2,
ACCT_STR); ACCT_STR);
#else #else
int is_eob = aom_read( int is_eob = aom_read(
...@@ -198,8 +206,8 @@ static INLINE int read_nz_map_horiz(aom_reader *r, tran_low_t *tcoeffs, ...@@ -198,8 +206,8 @@ static INLINE int read_nz_map_horiz(aom_reader *r, tran_low_t *tcoeffs,
for (int row = 0; row < height; ++row) { for (int row = 0; row < height; ++row) {
int el_ctx = get_empty_line_ctx(row, eob_ls); int el_ctx = get_empty_line_ctx(row, eob_ls);
#if LV_MAP_PROB #if LV_MAP_PROB
int empty_line = aom_read_symbol( int empty_line = av1_read_record_symbol(
r, fc->empty_line_cdf[txs_ctx][plane_type][tx_class][el_ctx], 2, counts, r, fc->empty_line_cdf[txs_ctx][plane_type][tx_class][el_ctx], 2,
ACCT_STR); ACCT_STR);
#else #else
int empty_line = aom_read( int empty_line = aom_read(
...@@ -215,8 +223,9 @@ static INLINE int read_nz_map_horiz(aom_reader *r, tran_low_t *tcoeffs, ...@@ -215,8 +223,9 @@ static INLINE int read_nz_map_horiz(aom_reader *r, tran_low_t *tcoeffs,
get_nz_map_ctx(tcoeffs, coeff_idx, bwl, height, tx_type); get_nz_map_ctx(tcoeffs, coeff_idx, bwl, height, tx_type);
if (col + 1 != width) { if (col + 1 != width) {
#if LV_MAP_PROB #if LV_MAP_PROB
int is_nz = aom_read_symbol( int is_nz = av1_read_record_symbol(
r, fc->nz_map_cdf[txs_ctx][plane_type][coeff_ctx], 2, ACCT_STR); counts, r, fc->nz_map_cdf[txs_ctx][plane_type][coeff_ctx], 2,
ACCT_STR);
#else #else
int is_nz = aom_read(r, nz_map[coeff_ctx], ACCT_STR); int is_nz = aom_read(r, nz_map[coeff_ctx], ACCT_STR);
#endif #endif
...@@ -226,8 +235,9 @@ static INLINE int read_nz_map_horiz(aom_reader *r, tran_low_t *tcoeffs, ...@@ -226,8 +235,9 @@ static INLINE int read_nz_map_horiz(aom_reader *r, tran_low_t *tcoeffs,
eob = AOMMAX(eob, iscan[coeff_idx] + 1); eob = AOMMAX(eob, iscan[coeff_idx] + 1);
int eob_ctx = get_hv_eob_ctx(row, col, eob_ls); int eob_ctx = get_hv_eob_ctx(row, col, eob_ls);
#if LV_MAP_PROB #if LV_MAP_PROB
int is_eob = aom_read_symbol( int is_eob = av1_read_record_symbol(
r, fc->hv_eob_cdf[txs_ctx][plane_type][tx_class][eob_ctx], 2, counts, r,
fc->hv_eob_cdf[txs_ctx][plane_type][tx_class][eob_ctx], 2,
ACCT_STR); ACCT_STR);
#else #else
int is_eob = int is_eob =
...@@ -276,8 +286,9 @@ uint8_t av1_read_coeffs_txb(const AV1_COMMON *const cm, MACROBLOCKD *xd, ...@@ -276,8 +286,9 @@ uint8_t av1_read_coeffs_txb(const AV1_COMMON *const cm, MACROBLOCKD *xd,
memset(tcoeffs, 0, sizeof(*tcoeffs) * seg_eob); memset(tcoeffs, 0, sizeof(*tcoeffs) * seg_eob);
#if LV_MAP_PROB #if LV_MAP_PROB
int all_zero = aom_read_symbol( int all_zero = av1_read_record_symbol(
r, ec_ctx->txb_skip_cdf[txs_ctx][txb_ctx->txb_skip_ctx], 2, ACCT_STR); counts, r, ec_ctx->txb_skip_cdf[txs_ctx][txb_ctx->txb_skip_ctx], 2,
ACCT_STR);
#else #else
int all_zero = int all_zero =
aom_read(r, ec_ctx->txb_skip[txs_ctx][txb_ctx->txb_skip_ctx], ACCT_STR); aom_read(r, ec_ctx->txb_skip[txs_ctx][txb_ctx->txb_skip_ctx], ACCT_STR);
...@@ -313,8 +324,9 @@ uint8_t av1_read_coeffs_txb(const AV1_COMMON *const cm, MACROBLOCKD *xd, ...@@ -313,8 +324,9 @@ uint8_t av1_read_coeffs_txb(const AV1_COMMON *const cm, MACROBLOCKD *xd,
read_nz_map(r, tcoeffs, plane, scan, tx_size, tx_type, ec_ctx, counts); read_nz_map(r, tcoeffs, plane, scan, tx_size, tx_type, ec_ctx, counts);
} else { } else {
#if LV_MAP_PROB #if LV_MAP_PROB
const int eob_mode = aom_read_symbol( const int eob_mode = av1_read_record_symbol(
r, ec_ctx->eob_mode_cdf[txs_ctx][plane_type][tx_class], 2, ACCT_STR); counts, r, ec_ctx->eob_mode_cdf[txs_ctx][plane_type][tx_class], 2,
ACCT_STR);
#else #else
const int eob_mode = const int eob_mode =
aom_read(r, ec_ctx->eob_mode[txs_ctx][plane_type][tx_class], ACCT_STR); aom_read(r, ec_ctx->eob_mode[txs_ctx][plane_type][tx_class], ACCT_STR);
...@@ -354,9 +366,9 @@ uint8_t av1_read_coeffs_txb(const AV1_COMMON *const cm, MACROBLOCKD *xd, ...@@ -354,9 +366,9 @@ uint8_t av1_read_coeffs_txb(const AV1_COMMON *const cm, MACROBLOCKD *xd,
ctx = get_base_ctx(tcoeffs, scan[c], bwl, height, i + 1); ctx = get_base_ctx(tcoeffs, scan[c], bwl, height, i + 1);
#if LV_MAP_PROB #if LV_MAP_PROB
if (aom_read_symbol(r, if (av1_read_record_symbol(
ec_ctx->coeff_base_cdf[txs_ctx][plane_type][i][ctx], counts, r, ec_ctx->coeff_base_cdf[txs_ctx][plane_type][i][ctx], 2,
2, ACCT_STR)) ACCT_STR))
#else #else
if (aom_read(r, coeff_base[ctx], ACCT_STR)) if (aom_read(r, coeff_base[ctx], ACCT_STR))
#endif #endif
...@@ -369,15 +381,16 @@ uint8_t av1_read_coeffs_txb(const AV1_COMMON *const cm, MACROBLOCKD *xd, ...@@ -369,15 +381,16 @@ uint8_t av1_read_coeffs_txb(const AV1_COMMON *const cm, MACROBLOCKD *xd,
if (c == 0) { if (c == 0) {
int dc_sign_ctx = txb_ctx->dc_sign_ctx; int dc_sign_ctx = txb_ctx->dc_sign_ctx;
#if LV_MAP_PROB #if LV_MAP_PROB
sign = aom_read_symbol( sign = av1_read_record_symbol(
r, ec_ctx->dc_sign_cdf[plane_type][dc_sign_ctx], 2, ACCT_STR); counts, r, ec_ctx->dc_sign_cdf[plane_type][dc_sign_ctx], 2,
ACCT_STR);
#else #else
sign = sign =
aom_read(r, ec_ctx->dc_sign[plane_type][dc_sign_ctx], ACCT_STR); aom_read(r, ec_ctx->dc_sign[plane_type][dc_sign_ctx], ACCT_STR);
#endif #endif
if (counts) ++counts->dc_sign[plane_type][dc_sign_ctx][sign]; if (counts) ++counts->dc_sign[plane_type][dc_sign_ctx][sign];
} else { } else {
sign = aom_read_bit(r, ACCT_STR); sign = av1_read_record_bit(counts, r, ACCT_STR);
} }
if (sign) *v = -(*v); if (sign) *v = -(*v);
continue; continue;
...@@ -401,14 +414,14 @@ uint8_t av1_read_coeffs_txb(const AV1_COMMON *const cm, MACROBLOCKD *xd, ...@@ -401,14 +414,14 @@ uint8_t av1_read_coeffs_txb(const AV1_COMMON *const cm, MACROBLOCKD *xd,
if (c == 0) { if (c == 0) {
int dc_sign_ctx = txb_ctx->dc_sign_ctx; int dc_sign_ctx = txb_ctx->dc_sign_ctx;
#if LV_MAP_PROB #if LV_MAP_PROB
sign = aom_read_symbol(r, ec_ctx->dc_sign_cdf[plane_type][dc_sign_ctx], 2, sign = av1_read_record_symbol(
ACCT_STR); counts, r, ec_ctx->dc_sign_cdf[plane_type][dc_sign_ctx], 2, ACCT_STR);
#else #else
sign = aom_read(r, ec_ctx->dc_sign[plane_type][dc_sign_ctx], ACCT_STR); sign = aom_read(r, ec_ctx->dc_sign[plane_type][dc_sign_ctx], ACCT_STR);
#endif #endif
if (counts) ++counts->dc_sign[plane_type][dc_sign_ctx][sign]; if (counts) ++counts->dc_sign[plane_type][dc_sign_ctx][sign];
} else { } else {
sign = aom_read_bit(r, ACCT_STR); sign = av1_read_record_bit(counts, r, ACCT_STR);
} }
ctx = get_br_ctx(tcoeffs, scan[c], bwl, height); ctx = get_br_ctx(tcoeffs, scan[c], bwl, height);
...@@ -418,9 +431,9 @@ uint8_t av1_read_coeffs_txb(const AV1_COMMON *const cm, MACROBLOCKD *xd, ...@@ -418,9 +431,9 @@ uint8_t av1_read_coeffs_txb(const AV1_COMMON *const cm, MACROBLOCKD *xd,
#if BR_NODE #if BR_NODE
for (idx = 0; idx < BASE_RANGE_SETS; ++idx) { for (idx = 0; idx < BASE_RANGE_SETS; ++idx) {
#if LV_MAP_PROB #if LV_MAP_PROB
if (aom_read_symbol(r, if (av1_read_record_symbol(
ec_ctx->coeff_br_cdf[txs_ctx][plane_type][idx][ctx], counts, r, ec_ctx->coeff_br_cdf[txs_ctx][plane_type][idx][ctx], 2,
2, ACCT_STR)) ACCT_STR))
#else // LV_MAP_PROB #else // LV_MAP_PROB
if (aom_read(r, ec_ctx->coeff_br[txs_ctx][plane_type][idx][ctx], if (aom_read(r, ec_ctx->coeff_br[txs_ctx][plane_type][idx][ctx],
ACCT_STR)) ACCT_STR))
...@@ -433,9 +446,9 @@ uint8_t av1_read_coeffs_txb(const AV1_COMMON *const cm, MACROBLOCKD *xd, ...@@ -433,9 +446,9 @@ uint8_t av1_read_coeffs_txb(const AV1_COMMON *const cm, MACROBLOCKD *xd,
if (counts) ++counts->coeff_br[txs_ctx][plane_type][idx][ctx][1]; if (counts) ++counts->coeff_br[txs_ctx][plane_type][idx][ctx][1];
for (tok = 0; tok < extra_bits; ++tok) { for (tok = 0; tok < extra_bits; ++tok) {
#if LV_MAP_PROB #if LV_MAP_PROB
if (aom_read_symbol(r, if (av1_read_record_symbol(
ec_ctx->coeff_lps_cdf[txs_ctx][plane_type][ctx], counts, r, ec_ctx->coeff_lps_cdf[txs_ctx][plane_type][ctx], 2,
2, ACCT_STR)) ACCT_STR))
#else #else
if (aom_read(r, ec_ctx->coeff_lps[txs_ctx][plane_type][ctx], if (aom_read(r, ec_ctx->coeff_lps[txs_ctx][plane_type][ctx],
ACCT_STR)) ACCT_STR))
...@@ -463,7 +476,8 @@ uint8_t av1_read_coeffs_txb(const AV1_COMMON *const cm, MACROBLOCKD *xd, ...@@ -463,7 +476,8 @@ uint8_t av1_read_coeffs_txb(const AV1_COMMON *const cm, MACROBLOCKD *xd,
#else #else
for (idx = 0; idx < COEFF_BASE_RANGE; ++idx) { for (idx = 0; idx < COEFF_BASE_RANGE; ++idx) {
#if LV_MAP_PROB #if LV_MAP_PROB
if (aom_read_symbol(r, ec_ctx->coeff_lps_cdf[txs_ctx][plane_type][ctx], 2, if (av1_read_record_symbol(
counts, r, ec_ctx->coeff_lps_cdf[txs_ctx][plane_type][ctx], 2,
ACCT_STR)) ACCT_STR))
#else #else
if (aom_read(r, ec_ctx->coeff_lps[txs_ctx][plane_type][ctx], ACCT_STR)) if (aom_read(r, ec_ctx->coeff_lps[txs_ctx][plane_type][ctx], ACCT_STR))
...@@ -482,7 +496,7 @@ uint8_t av1_read_coeffs_txb(const AV1_COMMON *const cm, MACROBLOCKD *xd, ...@@ -482,7 +496,7 @@ uint8_t av1_read_coeffs_txb(const AV1_COMMON *const cm, MACROBLOCKD *xd,
#endif #endif
// decode 0-th order Golomb code // decode 0-th order Golomb code
*v = read_golomb(xd, r) + COEFF_BASE_RANGE + 1 + NUM_BASE_LEVELS; *v = read_golomb(xd, r, counts) + COEFF_BASE_RANGE + 1 + NUM_BASE_LEVELS;
if (sign) *v = -(*v); if (sign) *v = -(*v);
cul_level += abs(*v); cul_level += abs(*v);
} }
...@@ -540,10 +554,13 @@ uint8_t av1_read_coeffs_txb_facade(AV1_COMMON *cm, MACROBLOCKD *xd, ...@@ -540,10 +554,13 @@ uint8_t av1_read_coeffs_txb_facade(AV1_COMMON *cm, MACROBLOCKD *xd,
} }
static void read_txb_probs(FRAME_CONTEXT *fc, const TX_SIZE tx_size, static void read_txb_probs(FRAME_CONTEXT *fc, const TX_SIZE tx_size,
aom_reader *r) { aom_reader *r, FRAME_COUNTS *counts) {
#if !CONFIG_SYMBOLRATE
(void)counts;
#endif
int plane, ctx, level; int plane, ctx, level;
if (aom_read_bit(r, ACCT_STR) == 0) return; if (av1_read_record_bit(counts, r, ACCT_STR) == 0) return;
for (ctx = 0; ctx < TXB_SKIP_CONTEXTS; ++ctx) for (ctx = 0; ctx < TXB_SKIP_CONTEXTS; ++ctx)
av1_diff_update_prob(r, &fc->txb_skip[tx_size][ctx], ACCT_STR); av1_diff_update_prob(r, &fc->txb_skip[tx_size][ctx], ACCT_STR);
...@@ -567,7 +584,8 @@ static void read_txb_probs(FRAME_CONTEXT *fc, const TX_SIZE tx_size, ...@@ -567,7 +584,8 @@ static void read_txb_probs(FRAME_CONTEXT *fc, const TX_SIZE tx_size,
av1_diff_update_prob(r, &fc->coeff_lps[tx_size][plane][ctx], ACCT_STR); av1_diff_update_prob(r, &fc->coeff_lps[tx_size][plane][ctx], ACCT_STR);
} }
void av1_read_txb_probs(FRAME_CONTEXT *fc, TX_MODE tx_mode, aom_reader *r) { void av1_read_txb_probs(FRAME_CONTEXT *fc, TX_MODE tx_mode, aom_reader *r,
FRAME_COUNTS *counts) {
const TX_SIZE max_tx_size = tx_mode_to_biggest_tx_size[tx_mode]; const TX_SIZE max_tx_size = tx_mode_to_biggest_tx_size[tx_mode];
TX_SIZE tx_size; TX_SIZE tx_size;
int ctx, plane; int ctx, plane;
...@@ -581,5 +599,5 @@ void av1_read_txb_probs(FRAME_CONTEXT *fc, TX_MODE tx_mode, aom_reader *r) { ...@@ -581,5 +599,5 @@ void av1_read_txb_probs(FRAME_CONTEXT *fc, TX_MODE tx_mode, aom_reader *r) {
av1_diff_update_prob(r, &fc->dc_sign[plane][ctx], ACCT_STR); av1_diff_update_prob(r, &fc->dc_sign[plane][ctx], ACCT_STR);
for (tx_size = TX_4X4; tx_size <= max_tx_size; ++tx_size) for (tx_size = TX_4X4; tx_size <= max_tx_size; ++tx_size)
read_txb_probs(fc, tx_size, r); read_txb_probs(fc, tx_size, r, counts);
} }
...@@ -28,5 +28,6 @@ uint8_t av1_read_coeffs_txb_facade(AV1_COMMON *cm, MACROBLOCKD *xd, ...@@ -28,5 +28,6 @@ uint8_t av1_read_coeffs_txb_facade(AV1_COMMON *cm, MACROBLOCKD *xd,
int plane, tran_low_t *tcoeffs, int plane, tran_low_t *tcoeffs,
TX_SIZE tx_size, int16_t *max_scan_line, TX_SIZE tx_size, int16_t *max_scan_line,
int *eob); int *eob);
void av1_read_txb_probs(FRAME_CONTEXT *fc, TX_MODE tx_mode, aom_reader *r); void av1_read_txb_probs(FRAME_CONTEXT *fc, TX_MODE tx_mode, aom_reader *r,
FRAME_COUNTS *counts);
#endif // DECODETXB_H_ #endif // DECODETXB_H_
/*
* Copyright (c) 2017, 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.
*/
#include "aom_dsp/bitreader.h"
#ifndef AV1_DECODER_SYMBOLRATE_H_
#define AV1_DECODER_SYMBOLRATE_H_
#if CONFIG_SYMBOLRATE
static INLINE void av1_dump_symbol_rate(struct AV1Common *cm) {
const FRAME_COUNTS *counts = &cm->counts;
printf("%d %d\n", counts->symbol_num[0], counts->symbol_num[1]);
}
static INLINE int av1_read_record_symbol(FRAME_COUNTS *counts, aom_reader *r,
aom_cdf_prob *cdf, int nsymbs,
const char *str) {
(void)str;
if (counts) ++counts->symbol_num[0];
return aom_read_symbol(r, cdf, nsymbs, str);
}
static INLINE int av1_read_record_bit(FRAME_COUNTS *counts, aom_reader *r,
const char *str) {
(void)str;
if (counts) ++counts->symbol_num[1];
return aom_read_bit(r, str);
}
#else // CONFIG_SYMBOLRATE
#define av1_read_record_symbol(counts, r, cdf, nsymbs, ACCT_STR_NAME) \
aom_read_symbol(r, cdf, nsymbs, ACCT_STR_NAME)
#define av1_read_record_bit(counts, r, ACCT_STR_NAME) \
aom_read_bit(r, ACCT_STR_NAME)
#endif // CONFIG_SYMBOLRATE
#endif // AV1_DECODER_SYMBOLRATE_H_
...@@ -102,6 +102,7 @@ set(CONFIG_LOWBITDEPTH 1 CACHE NUMBER "Enables low bit depth support.") ...@@ -102,6 +102,7 @@ set(CONFIG_LOWBITDEPTH 1 CACHE NUMBER "Enables low bit depth support.")
set(CONFIG_REALTIME_ONLY 0 CACHE NUMBER "Support only realtime encodes.") set(CONFIG_REALTIME_ONLY 0 CACHE NUMBER "Support only realtime encodes.")
set(CONFIG_SIZE_LIMIT 0 CACHE NUMBER "Limit max decode width/height.") set(CONFIG_SIZE_LIMIT 0 CACHE NUMBER "Limit max decode width/height.")
set(CONFIG_SPATIAL_RESAMPLING 1 CACHE NUMBER "Spatial resampling.") set(CONFIG_SPATIAL_RESAMPLING 1 CACHE NUMBER "Spatial resampling.")
set(CONFIG_SYMBOLRATE 0 CACHE NUMBER "Enables symbol rate accounting.")
# AV1 experiment flags. # AV1 experiment flags.
set(CONFIG_ADAPT_SCAN 0 CACHE NUMBER "AV1 experiment flag.") set(CONFIG_ADAPT_SCAN 0 CACHE NUMBER "AV1 experiment flag.")
......
...@@ -394,6 +394,7 @@ CONFIG_LIST=" ...@@ -394,6 +394,7 @@ CONFIG_LIST="
decode_perf_tests decode_perf_tests
encode_perf_tests encode_perf_tests
bitstream_debug bitstream_debug
symbolrate
coefficient_range_checking coefficient_range_checking
lowbitdepth lowbitdepth
highbitdepth highbitdepth
...@@ -452,6 +453,7 @@ CMDLINE_SELECT=" ...@@ -452,6 +453,7 @@ CMDLINE_SELECT="
encode_perf_tests encode_perf_tests
coefficient_range_checking coefficient_range_checking
bitstream_debug bitstream_debug
symbolrate
lowbitdepth lowbitdepth
highbitdepth highbitdepth
experimental experimental
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment