EC_ADAPT: refactor and fix MinArfFreq unit tests.
Ensure that cdfs are synced with pdfs after every
forward update.
Change-Id: I5677f78300156c8622f1728d7a343ff6c3a4ea64
diff --git a/av1/encoder/bitstream.c b/av1/encoder/bitstream.c
index aa55db3..818137e 100644
--- a/av1/encoder/bitstream.c
+++ b/av1/encoder/bitstream.c
@@ -565,13 +565,9 @@
prob_diff_update(
av1_switchable_interp_tree, cm->fc->switchable_interp_prob[j],
counts->switchable_interp[j], SWITCHABLE_FILTERS, probwt, w);
-#if CONFIG_DAALA_EC
- av1_tree_to_cdf(av1_switchable_interp_tree,
- cm->fc->switchable_interp_prob[j],
- cm->fc->switchable_interp_cdf[j]);
-#endif
}
}
+#endif
#if CONFIG_EXT_TX
static void update_ext_tx_probs(AV1_COMMON *cm, aom_writer *w) {
@@ -631,6 +627,7 @@
}
#else
+#if !CONFIG_EC_ADAPT || !CONFIG_DAALA_EC
static void update_ext_tx_probs(AV1_COMMON *cm, aom_writer *w) {
const int savings_thresh = av1_cost_one(GROUP_DIFF_UPDATE_PROB) -
@@ -657,13 +654,10 @@
for (j = 0; j < TX_TYPES; ++j) {
prob_diff_update(av1_ext_tx_tree, cm->fc->intra_ext_tx_prob[i][j],
cm->counts.intra_ext_tx[i][j], TX_TYPES, probwt, w);
-#if CONFIG_DAALA_EC
- av1_tree_to_cdf(av1_ext_tx_tree, cm->fc->intra_ext_tx_prob[i][j],
- cm->fc->intra_ext_tx_cdf[i][j]);
-#endif
}
}
}
+
savings = 0;
for (i = TX_4X4; i < EXT_TX_SIZES; ++i) {
savings +=
@@ -676,10 +670,6 @@
for (i = TX_4X4; i < EXT_TX_SIZES; ++i) {
prob_diff_update(av1_ext_tx_tree, cm->fc->inter_ext_tx_prob[i],
cm->counts.inter_ext_tx[i], TX_TYPES, probwt, w);
-#if CONFIG_DAALA_EC
- av1_tree_to_cdf(av1_ext_tx_tree, cm->fc->inter_ext_tx_prob[i],
- cm->fc->inter_ext_tx_cdf[i]);
-#endif
}
}
}
@@ -2345,9 +2335,6 @@
}
default: assert(0);
}
-#if CONFIG_RANS || CONFIG_DAALA_EC
- av1_coef_pareto_cdfs(cpi->common.fc);
-#endif // CONFIG_EC_MULTISYMBOL
}
#if CONFIG_ENTROPY
@@ -2922,12 +2909,8 @@
prob_diff_update(av1_segment_tree, cm->fc->seg.tree_probs,
cm->counts.seg.tree_total, MAX_SEGMENTS, probwt, w);
}
-#if CONFIG_DAALA_EC
- av1_tree_to_cdf(av1_segment_tree, cm->fc->seg.tree_probs,
- cm->fc->seg.tree_cdf);
-#endif
}
-#endif // CONFIG_EC_ADAPT,CONFIG_DAALA_EC
+#endif
static void write_txfm_mode(TX_MODE mode, struct aom_write_bit_buffer *wb) {
aom_wb_write_bit(wb, mode == TX_MODE_SELECT);
@@ -3749,10 +3732,6 @@
for (i = 0; i < INTRA_MODES; ++i) {
prob_diff_update(av1_intra_mode_tree, fc->uv_mode_prob[i],
counts->uv_mode[i], INTRA_MODES, probwt, header_bc);
-#if CONFIG_DAALA_EC
- av1_tree_to_cdf(av1_intra_mode_tree, fc->uv_mode_prob[i],
- fc->uv_mode_cdf[i]);
-#endif
}
#if CONFIG_EXT_PARTITION_TYPES
@@ -3766,12 +3745,8 @@
for (i = 0; i < PARTITION_CONTEXTS; ++i) {
prob_diff_update(av1_partition_tree, fc->partition_prob[i],
counts->partition[i], PARTITION_TYPES, probwt, header_bc);
-#if CONFIG_DAALA_EC
- av1_tree_to_cdf(av1_partition_tree, cm->fc->partition_prob[i],
- cm->fc->partition_cdf[i]);
-#endif
}
-#endif // CONFIG_EXT_PARTITION_TYPES
+#endif // CONFIG_EC_ADAPT, CONFIG_DAALA_EC
#if CONFIG_EXT_INTRA
for (i = 0; i < INTRA_FILTERS + 1; ++i)
@@ -3787,31 +3762,22 @@
#if !CONFIG_EC_ADAPT || !CONFIG_DAALA_EC
for (i = 0; i < INTRA_MODES; ++i)
- for (j = 0; j < INTRA_MODES; ++j) {
+ for (j = 0; j < INTRA_MODES; ++j)
prob_diff_update(av1_intra_mode_tree, cm->kf_y_prob[i][j],
counts->kf_y_mode[i][j], INTRA_MODES, probwt,
header_bc);
-#if CONFIG_DAALA_EC
- av1_tree_to_cdf(av1_intra_mode_tree, cm->kf_y_prob[i][j],
- cm->kf_y_cdf[i][j]);
-#endif
- }
#endif // CONFIG_EC_ADAPT, CONFIG_DAALA_EC
} else {
-#if !CONFIG_EC_ADAPT || !CONFIG_DAALA_EC
#if CONFIG_REF_MV
update_inter_mode_probs(cm, header_bc, counts);
#else
+#if !CONFIG_EC_ADAPT || !CONFIG_DAALA_EC
for (i = 0; i < INTER_MODE_CONTEXTS; ++i) {
prob_diff_update(av1_inter_mode_tree, cm->fc->inter_mode_probs[i],
counts->inter_mode[i], INTER_MODES, probwt, header_bc);
-#if CONFIG_DAALA_EC
- av1_tree_to_cdf(av1_inter_mode_tree, cm->fc->inter_mode_probs[i],
- cm->fc->inter_mode_cdf[i]);
-#endif
}
#endif
-#endif // CONFIG_EC_ADAPT, CONFIG_DAALA_EC
+#endif
#if CONFIG_EXT_INTER
update_inter_compound_mode_probs(cm, probwt, header_bc);
@@ -3895,12 +3861,8 @@
for (i = 0; i < BLOCK_SIZE_GROUPS; ++i) {
prob_diff_update(av1_intra_mode_tree, cm->fc->y_mode_prob[i],
counts->y_mode[i], INTRA_MODES, probwt, header_bc);
-#if CONFIG_DAALA_EC
- av1_tree_to_cdf(av1_intra_mode_tree, cm->fc->y_mode_prob[i],
- cm->fc->y_mode_cdf[i]);
-#endif
}
-#endif // CONFIG_EC_ADAPT, CONFIG_DAALA_EC
+#endif
av1_write_nmv_probs(cm, cm->allow_high_precision_mv, header_bc,
#if CONFIG_REF_MV
@@ -3922,6 +3884,13 @@
write_global_motion(cpi, header_bc);
#endif // CONFIG_GLOBAL_MOTION
}
+#if CONFIG_EC_MULTISYMBOL
+ av1_coef_pareto_cdfs(fc);
+ av1_set_mv_cdfs(&fc->nmvc);
+#if CONFIG_DAALA_EC
+ av1_set_mode_cdfs(cm);
+#endif
+#endif
#if CONFIG_ANS
ans_write_init(&header_ans, data);
buf_ans_flush(header_bc, &header_ans);
