Commit 0d04f573 authored by Angie Chiang's avatar Angie Chiang

Update eob_extra counts

Change-Id: Ib0e0b5f7085d3c910dcd15a3628a001af864c8fc
parent fcdda9dc
......@@ -374,6 +374,7 @@ uint8_t av1_read_coeffs_txb(const AV1_COMMON *const cm, MACROBLOCKD *xd,
int bit = av1_read_record_bin(
counts, r, ec_ctx->eob_extra_cdf[txs_ctx][plane_type][eob_pt], 2,
ACCT_STR);
if (counts) ++counts->eob_extra[txs_ctx][plane_type][eob_pt][bit];
if (bit) {
eob_extra += (1 << eob_shift);
}
......
......@@ -155,7 +155,7 @@ void av1_update_eob_context(int eob, int seg_eob, TX_SIZE txsize,
if (k_eob_offset_bits[eob_pt] > 0) {
int eob_shift = k_eob_offset_bits[eob_pt] - 1;
int bit = (eob_extra & (1 << eob_shift)) ? 1 : 0;
// TODO(angiebird): update count as well
counts->eob_extra[txsize][plane][eob_pt][bit]++;
#if LV_MAP_PROB
update_cdf(ec_ctx->eob_extra_cdf[txsize][plane][eob_pt], bit, 2);
#endif
......
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