Implement bitstream debug for daala_ec
Change-Id: I809eb52e8a632189c49b8ea0a2b5de760cc2a34c
diff --git a/aom_util/debug_util.h b/aom_util/debug_util.h
index c52e385..2ed56ea 100644
--- a/aom_util/debug_util.h
+++ b/aom_util/debug_util.h
@@ -13,6 +13,9 @@
#define AOM_UTIL_DEBUG_UTIL_H_
#include "./aom_config.h"
+#if CONFIG_DAALA_EC
+#include "aom_dsp/prob.h"
+#endif
#ifdef __cplusplus
extern "C" {
@@ -30,8 +33,13 @@
int bitstream_queue_get_read(void);
void bitstream_queue_record_write(void);
void bitstream_queue_reset_write(void);
+#if CONFIG_DAALA_EC
+void bitstream_queue_pop(int *result, aom_cdf_prob *cdf, int *nsymbs);
+void bitstream_queue_push(int result, const aom_cdf_prob *cdf, int nsymbs);
+#else
void bitstream_queue_pop(int *result, int *prob);
void bitstream_queue_push(int result, int prob);
+#endif
void bitstream_queue_set_skip_write(int skip);
void bitstream_queue_set_skip_read(int skip);
void bitstream_queue_set_frame_write(int frame_idx);