From 09ebbfb39f1f3edf7e40b5af2bbf7636af4f574f Mon Sep 17 00:00:00 2001
From: Thomas Davies <thdavies@cisco.com>
Date: Thu, 20 Oct 2016 18:28:47 +0100
Subject: [PATCH] EC_ADAPT: send updates for the correct nodes.

EOB and ZERO token are not currently adapted.

Change-Id: Ie7d657b71fcb157b09e40874fb06a8b7cd95cc70
---
 av1/decoder/decodeframe.c | 2 +-
 av1/encoder/bitstream.c   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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