Use segment tree_cdf with CONFIG_EC_MULTISYMBOL. Change-Id: I0005c896a243275c052a0163a5da0f9230071743
diff --git a/av1/common/entropymode.c b/av1/common/entropymode.c index 28f67a4..5f72aa7 100644 --- a/av1/common/entropymode.c +++ b/av1/common/entropymode.c
@@ -1767,6 +1767,8 @@ #endif av1_tree_to_cdf_2D(av1_intra_mode_tree, av1_kf_y_mode_prob, av1_kf_y_mode_cdf, INTRA_MODES, INTRA_MODES); +#endif +#if CONFIG_EC_MULTISYMBOL av1_tree_to_cdf(av1_segment_tree, fc->seg.tree_probs, fc->seg.tree_cdf); #endif #if CONFIG_DELTA_Q @@ -1777,7 +1779,9 @@ #if CONFIG_DAALA_EC int av1_switchable_interp_ind[SWITCHABLE_FILTERS]; int av1_switchable_interp_inv[SWITCHABLE_FILTERS]; +#endif +#if CONFIG_EC_MULTISYMBOL void av1_set_mode_cdfs(struct AV1Common *cm) { FRAME_CONTEXT *fc = cm->fc; int i, j; @@ -1785,7 +1789,7 @@ av1_tree_to_cdf(av1_segment_tree, cm->fc->seg.tree_probs, cm->fc->seg.tree_cdf); } - +#if CONFIG_DAALA_EC for (i = 0; i < INTRA_MODES; ++i) av1_tree_to_cdf(av1_intra_mode_tree, fc->uv_mode_prob[i], fc->uv_mode_cdf[i]); @@ -1820,6 +1824,7 @@ av1_tree_to_cdf(av1_ext_tx_tree, fc->inter_ext_tx_prob[i], fc->inter_ext_tx_cdf[i]); #endif +#endif } #endif
diff --git a/av1/common/entropymode.h b/av1/common/entropymode.h index 63211dc..bf1867f 100644 --- a/av1/common/entropymode.h +++ b/av1/common/entropymode.h
@@ -359,7 +359,8 @@ #if CONFIG_DAALA_EC extern int av1_switchable_interp_ind[SWITCHABLE_FILTERS]; extern int av1_switchable_interp_inv[SWITCHABLE_FILTERS]; - +#endif +#if CONFIG_EC_MULTISYMBOL void av1_set_mode_cdfs(struct AV1Common *cm); #endif
diff --git a/av1/common/seg_common.h b/av1/common/seg_common.h index d833a86..8e69ffd 100644 --- a/av1/common/seg_common.h +++ b/av1/common/seg_common.h
@@ -48,7 +48,7 @@ struct segmentation_probs { aom_prob tree_probs[SEG_TREE_PROBS]; -#if CONFIG_DAALA_EC +#if CONFIG_EC_MULTISYMBOL aom_cdf_prob tree_cdf[MAX_SEGMENTS]; #endif aom_prob pred_probs[PREDICTION_PROBS];
diff --git a/av1/decoder/decodeframe.c b/av1/decoder/decodeframe.c index d83b0ad..0479687 100644 --- a/av1/decoder/decodeframe.c +++ b/av1/decoder/decodeframe.c
@@ -4303,7 +4303,7 @@ #else av1_set_mv_cdfs(&fc->nmvc); #endif -#if CONFIG_DAALA_EC +#if CONFIG_EC_MULTISYMBOL av1_set_mode_cdfs(cm); #endif #endif
diff --git a/av1/decoder/decodemv.c b/av1/decoder/decodemv.c index 8f8dc2c..78ace95 100644 --- a/av1/decoder/decodemv.c +++ b/av1/decoder/decodemv.c
@@ -297,7 +297,7 @@ #endif // CONFIG_EXT_INTER static int read_segment_id(aom_reader *r, struct segmentation_probs *segp) { -#if CONFIG_DAALA_EC +#if CONFIG_EC_MULTISYMBOL return aom_read_symbol(r, segp->tree_cdf, MAX_SEGMENTS, ACCT_STR); #else return aom_read_tree(r, av1_segment_tree, segp->tree_probs, ACCT_STR);
diff --git a/av1/encoder/bitstream.c b/av1/encoder/bitstream.c index 9a30e11..546bbf8 100644 --- a/av1/encoder/bitstream.c +++ b/av1/encoder/bitstream.c
@@ -913,7 +913,7 @@ static void write_segment_id(aom_writer *w, const struct segmentation *seg, struct segmentation_probs *segp, int segment_id) { if (seg->enabled && seg->update_map) { -#if CONFIG_DAALA_EC +#if CONFIG_EC_MULTISYMBOL aom_write_symbol(w, segment_id, segp->tree_cdf, MAX_SEGMENTS); #else aom_write_tree(w, av1_segment_tree, segp->tree_probs, segment_id, 3, 0); @@ -4351,7 +4351,7 @@ #else av1_set_mv_cdfs(&fc->nmvc); #endif -#if CONFIG_DAALA_EC +#if CONFIG_EC_MULTISYMBOL av1_set_mode_cdfs(cm); #endif #endif