From: Michael Bebenita <mbebenita@mozilla.com>
Date: Wed, 3 Aug 2016 23:13:04 +0000 (-0700)
Subject: Port aom_reader_tell() support
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=868fc0b04aef06f8a3779ba92fc2cdfbbcb6c59e;p=libvpx

Port aom_reader_tell() support

This commit ports the following from aom/master:
4c46278 Add aom_reader_tell() support.
b9c9935 Remove an erroneous declaration.
56c9c3b Fix ANS build.

Change-Id: I59bd910f58c218c649a1de2a7b5fae0397e13cb1
---

diff --git a/aom_dsp/bitreader.h b/aom_dsp/bitreader.h
index 35976eeb1..d2fd5f2fd 100644
--- a/aom_dsp/bitreader.h
+++ b/aom_dsp/bitreader.h
@@ -79,6 +79,18 @@ static INLINE int aom_reader_has_error(aom_reader *r) {
 #endif
 }
 
+static INLINE ptrdiff_t aom_reader_tell(const aom_reader *r) {
+#if CONFIG_ANS
+  (void)r;
+  assert(0 && "aom_reader_tell() is unimplemented for ANS");
+  return 0;
+#elif CONFIG_DAALA_EC
+  return aom_daala_reader_tell(r);
+#else
+  return aom_dk_reader_tell(r);
+#endif
+}
+
 static INLINE int aom_read(aom_reader *r, int prob) {
 #if CONFIG_ANS
   return uabs_read(r, prob);
diff --git a/aom_dsp/buf_ans.h b/aom_dsp/buf_ans.h
index 58d6e61fc..e499200bb 100644
--- a/aom_dsp/buf_ans.h
+++ b/aom_dsp/buf_ans.h
@@ -28,8 +28,6 @@ extern "C" {
 #define ANS_METHOD_UABS 0
 #define ANS_METHOD_RANS 1
 
-struct aom_internal_error_info *error;
-
 struct buffered_ans_symbol {
   unsigned int method : 1;  // one of ANS_METHOD_UABS or ANS_METHOD_RANS
   // TODO(aconverse): Should be possible to write this in terms of start for ABS
diff --git a/aom_dsp/daalaboolreader.c b/aom_dsp/daalaboolreader.c
index d9ef88700..8e1c7821c 100644
--- a/aom_dsp/daalaboolreader.c
+++ b/aom_dsp/daalaboolreader.c
@@ -24,3 +24,7 @@ int aom_daala_reader_init(daala_reader *r, const uint8_t *buffer, int size) {
 const uint8_t *aom_daala_reader_find_end(daala_reader *r) {
   return r->buffer_end;
 }
+
+ptrdiff_t aom_daala_reader_tell(const daala_reader *r) {
+  return od_ec_dec_tell(&r->ec);
+}
diff --git a/aom_dsp/daalaboolreader.h b/aom_dsp/daalaboolreader.h
index 0ed0de4ac..fd69a274c 100644
--- a/aom_dsp/daalaboolreader.h
+++ b/aom_dsp/daalaboolreader.h
@@ -29,6 +29,7 @@ typedef struct daala_reader daala_reader;
 
 int aom_daala_reader_init(daala_reader *r, const uint8_t *buffer, int size);
 const uint8_t *aom_daala_reader_find_end(daala_reader *r);
+ptrdiff_t aom_daala_reader_tell(const daala_reader *r);
 
 static INLINE int aom_daala_read(daala_reader *r, int prob) {
   if (prob == 128) {
diff --git a/aom_dsp/dkboolreader.c b/aom_dsp/dkboolreader.c
index 8ec7ffc40..c3b7782d6 100644
--- a/aom_dsp/dkboolreader.c
+++ b/aom_dsp/dkboolreader.c
@@ -29,7 +29,7 @@ int aom_dk_reader_init(struct aom_dk_reader *r, const uint8_t *buffer,
     return 1;
   } else {
     r->buffer_end = buffer + size;
-    r->buffer = buffer;
+    r->buffer_start = r->buffer = buffer;
     r->value = 0;
     r->count = -8;
     r->range = 255;
diff --git a/aom_dsp/dkboolreader.h b/aom_dsp/dkboolreader.h
index fe68ecc0f..2fd2b378e 100644
--- a/aom_dsp/dkboolreader.h
+++ b/aom_dsp/dkboolreader.h
@@ -45,6 +45,7 @@ struct aom_dk_reader {
   BD_VALUE value;
   unsigned int range;
   int count;
+  const uint8_t *buffer_start;
   const uint8_t *buffer_end;
   const uint8_t *buffer;
   aom_decrypt_cb decrypt_cb;
@@ -60,6 +61,13 @@ void aom_dk_reader_fill(struct aom_dk_reader *r);
 
 const uint8_t *aom_dk_reader_find_end(struct aom_dk_reader *r);
 
+static INLINE ptrdiff_t aom_dk_reader_tell(const struct aom_dk_reader *r) {
+  const size_t bits_read = (r->buffer - r->buffer_start) * CHAR_BIT;
+  const int count =
+      (r->count < LOTS_OF_BITS) ? r->count : r->count - LOTS_OF_BITS;
+  return bits_read + BD_VALUE_SIZE - (count + CHAR_BIT);
+}
+
 static INLINE int aom_dk_reader_has_error(struct aom_dk_reader *r) {
   // Check if we have reached the end of the buffer.
   //
diff --git a/av1/common/entropymode.h b/av1/common/entropymode.h
index 7821b3929..03fe2bf5f 100644
--- a/av1/common/entropymode.h
+++ b/av1/common/entropymode.h
@@ -133,8 +133,8 @@ typedef struct frame_contexts {
 #endif  // CONFIG_LOOP_RESTORATION
 #if CONFIG_DAALA_EC
   aom_cdf_prob partition_cdf[PARTITION_CONTEXTS][PARTITION_TYPES];
-  aom_cdf_prob
-      switchable_interp_cdf[SWITCHABLE_FILTER_CONTEXTS][SWITCHABLE_FILTERS];
+  aom_cdf_prob switchable_interp_cdf[SWITCHABLE_FILTER_CONTEXTS]
+                                    [SWITCHABLE_FILTERS];
   aom_cdf_prob intra_ext_tx_cdf[EXT_TX_SIZES][TX_TYPES][TX_TYPES];
   aom_cdf_prob inter_ext_tx_cdf[EXT_TX_SIZES][TX_TYPES];
 #endif