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 35976ee..d2fd5f2 100644 --- a/aom_dsp/bitreader.h +++ b/aom_dsp/bitreader.h
@@ -79,6 +79,18 @@ #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 58d6e61..e499200 100644 --- a/aom_dsp/buf_ans.h +++ b/aom_dsp/buf_ans.h
@@ -28,8 +28,6 @@ #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 d9ef887..8e1c782 100644 --- a/aom_dsp/daalaboolreader.c +++ b/aom_dsp/daalaboolreader.c
@@ -24,3 +24,7 @@ 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 0ed0de4..fd69a27 100644 --- a/aom_dsp/daalaboolreader.h +++ b/aom_dsp/daalaboolreader.h
@@ -29,6 +29,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); +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 8ec7ffc..c3b7782 100644 --- a/aom_dsp/dkboolreader.c +++ b/aom_dsp/dkboolreader.c
@@ -29,7 +29,7 @@ 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 fe68ecc..2fd2b37 100644 --- a/aom_dsp/dkboolreader.h +++ b/aom_dsp/dkboolreader.h
@@ -45,6 +45,7 @@ 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 @@ 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 7821b39..03fe2bf 100644 --- a/av1/common/entropymode.h +++ b/av1/common/entropymode.h
@@ -133,8 +133,8 @@ #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