From: Nathan E. Egge <negge@mozilla.com>
Date: Mon, 20 Jun 2016 17:44:22 +0000 (-0400)
Subject: Add API for writing trees using a CDF.
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=44460148b21073a7a4f1d349099c51f2ed7f3ded;p=libvpx

Add API for writing trees using a CDF.

Added aom_write_tree_cdf() and aom_read_tree_cdf() function calls to
 bitwriter.h and bitreader.h respectively.
These calls take a multisymbol CDF and an index and directly encode the
 symbol using the enabled entropy coder.
Currently only the daala entropy encoder supports this (enabled with
 --enable-daala_ec) and a compile error is thrown otherwise.

Change-Id: I2fa1e87af4352c94384e0cfdbfd170ac99cf3705
---

diff --git a/aom_dsp/bitreader.h b/aom_dsp/bitreader.h
index 57626e7a5..1a0cc45ab 100644
--- a/aom_dsp/bitreader.h
+++ b/aom_dsp/bitreader.h
@@ -26,6 +26,7 @@
 #include "aom_dsp/dkboolreader.h"
 #endif
 #include "aom_dsp/prob.h"
+#include "av1/common/odintrin.h"
 
 #ifdef __cplusplus
 extern "C" {
@@ -136,6 +137,19 @@ static INLINE int aom_read_symbol(aom_reader *r, const aom_cdf_prob *cdf,
 #endif
 }
 
+static INLINE int aom_read_tree_cdf(aom_reader *r, const uint16_t *cdf,
+                                    int nsymbs) {
+#if CONFIG_DAALA_EC
+  return daala_read_tree_cdf(r, cdf, nsymbs);
+#else
+  (void)r;
+  (void)cdf;
+  (void)nsymbs;
+  assert(0 && "Unsupported bitreader operation");
+  return -1;
+#endif
+}
+
 #ifdef __cplusplus
 }  // extern "C"
 #endif
diff --git a/aom_dsp/bitwriter.h b/aom_dsp/bitwriter.h
index 86f97a353..d2569329a 100644
--- a/aom_dsp/bitwriter.h
+++ b/aom_dsp/bitwriter.h
@@ -116,6 +116,19 @@ static INLINE void aom_write_symbol(aom_writer *w, int symb,
 #endif
 }
 
+static INLINE void aom_write_tree_cdf(aom_writer *w, int symb,
+                                      const uint16_t *cdf, int nsymbs) {
+#if CONFIG_DAALA_EC
+  daala_write_tree_cdf(w, symb, cdf, nsymbs);
+#else
+  (void)w;
+  (void)symb;
+  (void)cdf;
+  (void)nsymbs;
+  assert(0 && "Unsupported bitwriter operation");
+#endif
+}
+
 #ifdef __cplusplus
 }  // extern "C"
 #endif
diff --git a/aom_dsp/daalaboolreader.h b/aom_dsp/daalaboolreader.h
index 3ccea1d00..f0a7e2542 100644
--- a/aom_dsp/daalaboolreader.h
+++ b/aom_dsp/daalaboolreader.h
@@ -66,6 +66,11 @@ static INLINE int daala_read_tree_bits(daala_reader *r,
   return -i;
 }
 
+static INLINE int daala_read_tree_cdf(daala_reader *r, const uint16_t *cdf,
+                                      int nsymbs) {
+  return od_ec_decode_cdf_q15(&r->ec, cdf, nsymbs, "aom");
+}
+
 #ifdef __cplusplus
 }  // extern "C"
 #endif
diff --git a/aom_dsp/daalaboolwriter.h b/aom_dsp/daalaboolwriter.h
index f9102f000..c71b12e74 100644
--- a/aom_dsp/daalaboolwriter.h
+++ b/aom_dsp/daalaboolwriter.h
@@ -78,6 +78,11 @@ static INLINE void daala_write_tree_bits(daala_writer *w,
   } while (len);
 }
 
+static INLINE void daala_write_tree_cdf(daala_writer *w, int symb,
+                                        const uint16_t *cdf, int nsymbs) {
+  od_ec_encode_cdf_q15(&w->ec, symb, cdf, nsymbs);
+}
+
 #ifdef __cplusplus
 }  // extern "C"
 #endif