commit 800df032b178e670641a5be2ca9868ee816e11ae
author Angie Chiang <angiebird@google.com> Wed Mar 22 11:14:12 2017 -0700
committer Angie Chiang <angiebird@google.com> Wed Mar 22 20:22:06 2017 +0000
tree 4bfcc9b5de810318ef5240b0dd65bd46d9bde593
parent 7d7ead9c22eb3647bf1bd73c636fb2206c4f6115

Add forward probability update for lv_map

The feature is implemented in the following two functions.
av1_write_txb_probs
av1_read_txb_probs

Change-Id: I0b646e17ec54d7a10a77a6853439217091455af1

av1/encoder/bitstream.c

diff --git a/av1/encoder/bitstream.c b/av1/encoder/bitstream.c
index 8adcb15..3bbce0f 100644
--- a/av1/encoder/bitstream.c
+++ b/av1/encoder/bitstream.c
@@ -2849,6 +2849,7 @@
 #endif
 }
 
+#if !CONFIG_LV_MAP
 #if !CONFIG_PVQ && !(CONFIG_EC_ADAPT && CONFIG_NEW_TOKENSET)
 static void build_tree_distribution(AV1_COMP *cpi, TX_SIZE tx_size,
                                     av1_coeff_stats *coef_branch_ct,
@@ -3325,8 +3326,9 @@
   }
 #endif  // CONFIG_SUBFRAME_PROB_UPDATE
 }
-#endif
+#endif  // !(CONFIG_EC_ADAPT && CONFIG_NEW_TOKENSET)
 #endif  // !CONFIG_EC_ADAPT
+#endif  // !CONFIG_LV_MAP
 
 #if CONFIG_LOOP_RESTORATION
 static void encode_restoration_mode(AV1_COMMON *cm,
@@ -4601,11 +4603,16 @@
 #if !CONFIG_EC_ADAPT
   update_txfm_probs(cm, header_bc, counts);
 #endif
+#if CONFIG_LV_MAP
+  av1_write_txb_probs(cpi, header_bc);
+#else
 #if !CONFIG_PVQ
 #if !(CONFIG_EC_ADAPT && CONFIG_NEW_TOKENSET)
   update_coef_probs(cpi, header_bc);
 #endif  // !(CONFIG_EC_ADAPT && CONFIG_NEW_TOKENSET)
 #endif  // CONFIG_PVQ
+#endif  // CONFIG_LV_MAP
+
 #if CONFIG_VAR_TX
   update_txfm_partition_probs(cm, header_bc, counts, probwt);
 #endif