From: Thomas Davies <thdavies@cisco.com>
Date: Thu, 20 Oct 2016 17:28:47 +0000 (+0100)
Subject: EC_ADAPT: send updates for the correct nodes.
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=09ebbfb39f1f3edf7e40b5af2bbf7636af4f574f;p=libvpx

EC_ADAPT: send updates for the correct nodes.

EOB and ZERO token are not currently adapted.

Change-Id: Ie7d657b71fcb157b09e40874fb06a8b7cd95cc70
---

diff --git a/av1/decoder/decodeframe.c b/av1/decoder/decodeframe.c
index 282997170..2e69a14f3 100644
--- a/av1/decoder/decodeframe.c
+++ b/av1/decoder/decodeframe.c
@@ -1916,7 +1916,7 @@ static void read_coef_probs_common(av1_coeff_probs_model *coef_probs,
                                    aom_reader *r) {
   int i, j, k, l, m;
 #if CONFIG_EC_ADAPT
-  const int node_limit = ONE_TOKEN;
+  const int node_limit = UNCONSTRAINED_NODES - 1;
 #else
   const int node_limit = UNCONSTRAINED_NODES;
 #endif
diff --git a/av1/encoder/bitstream.c b/av1/encoder/bitstream.c
index 0fdf17e32..aa55db347 100644
--- a/av1/encoder/bitstream.c
+++ b/av1/encoder/bitstream.c
@@ -2206,7 +2206,7 @@ static void update_coef_probs_common(aom_writer *const bc, AV1_COMP *cpi,
   av1_coeff_probs_model *old_coef_probs = cpi->common.fc->coef_probs[tx_size];
   const aom_prob upd = DIFF_UPDATE_PROB;
 #if CONFIG_EC_ADAPT
-  const int entropy_nodes_update = ONE_TOKEN;
+  const int entropy_nodes_update = UNCONSTRAINED_NODES - 1;
 #else
   const int entropy_nodes_update = UNCONSTRAINED_NODES;
 #endif