commit 50f1911192048d0fce41516a9b440d7cb231a733
author Nathan E. Egge <negge@mozilla.com> Tue Jun 27 21:07:57 2017 -0400
committer Nathan Egge <negge@mozilla.com> Wed Jun 28 15:10:17 2017 +0000
tree 6ec773a744a90dcbe59224c989cbc5fcb918e2fb
parent 05b45e6b790326a533f018d900a6eb8089cfe914

Add the av1_cost_tokens_from_cdf() function.

Change-Id: I148f8c7045d179c0a1ba7f1fe33b859f66bfc7f3

av1/encoder/cost.c

diff --git a/av1/encoder/cost.c b/av1/encoder/cost.c
index e3151a5..e33df53 100644
--- a/av1/encoder/cost.c
+++ b/av1/encoder/cost.c
@@ -65,3 +65,21 @@
   costs[-tree[0]] = av1_cost_bit(probs[0], 0);
   cost(costs, tree, probs, 2, 0);
 }
+
+void av1_cost_tokens_from_cdf(int *costs, const aom_cdf_prob *cdf,
+                              const int *inv_map) {
+  int i;
+  aom_cdf_prob prev_cdf = 0;
+  for (i = 0;; ++i) {
+    const aom_cdf_prob p15 = AOM_ICDF(cdf[i]) - prev_cdf;
+    prev_cdf = AOM_ICDF(cdf[i]);
+
+    if (inv_map)
+      costs[inv_map[i]] = av1_cost_symbol(p15);
+    else
+      costs[i] = av1_cost_symbol(p15);
+
+    // Stop once we reach the end of the CDF
+    if (cdf[i] == AOM_ICDF(CDF_PROB_TOP)) break;
+  }
+}