Commit a496c673 authored by Debargha Mukherjee's avatar Debargha Mukherjee
Browse files

Update cdf for intra blocks

This was missing before. Hopefully it will improve costing.

Change-Id: Id7dbe963fe5e26798b853419804bdd128537a5d9
parent 0c9ea803
......@@ -4476,6 +4476,19 @@ static void sum_intra_stats(FRAME_COUNTS *counts, MACROBLOCKD *xd,
(void)counts;
const BLOCK_SIZE bsize = mbmi->sb_type;
// Update intra tx size cdf
if (block_signals_txsize(bsize) && !xd->lossless[mbmi->segment_id] &&
allow_update_cdf) {
const TX_SIZE tx_size = mbmi->tx_size;
const int tx_size_ctx = get_tx_size_context(xd);
const int32_t tx_size_cat = intra_tx_size_cat_lookup[bsize];
const TX_SIZE coded_tx_size = txsize_sqr_up_map[tx_size];
const int depth = tx_size_to_depth(coded_tx_size, tx_size_cat);
const int max_depths = tx_size_cat_to_max_depth(tx_size_cat);
update_cdf(fc->tx_size_cdf[tx_size_cat][tx_size_ctx], depth,
max_depths + 1);
}
if (intraonly) {
#if CONFIG_ENTROPY_STATS
const PREDICTION_MODE above = av1_above_block_mode(mi, above_mi, 0);
......
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