vp10: allow forward updates for uv_mode probabilities.

See issue 1040 point 4.

Change-Id: I79e06bd71a27f45770c760c47dc71bc3767a77a0

diff --git a/vp10/decoder/decodeframe.c b/vp10/decoder/decodeframe.c
index 248912ded91b1253036f51331d64c3a52d98188e..b4c75ea7d0d3be9e9ce0ef66837543a40f7c337a 100644 (file)
--- a/vp10/decoder/decodeframe.c
+++ b/vp10/decoder/decodeframe.c
@@ -2145,6 +2145,12 @@ static int read_compressed_header(VP10Decoder *pbi, const uint8_t *data,
       for (i = 0; i < INTRA_MODES - 1; ++i)
         vp10_diff_update_prob(&r, &fc->y_mode_prob[j][i]);
 
+#if CONFIG_MISC_FIXES
+    for (j = 0; j < INTRA_MODES; j++)
+      for (i = 0; i < INTRA_MODES - 1; ++i)
+        vp10_diff_update_prob(&r, &fc->uv_mode_prob[j][i]);
+#endif
+
     for (j = 0; j < PARTITION_CONTEXTS; ++j)
       for (i = 0; i < PARTITION_TYPES - 1; ++i)
         vp10_diff_update_prob(&r, &fc->partition_prob[j][i]);

diff --git a/vp10/encoder/bitstream.c b/vp10/encoder/bitstream.c
index 3269fc7c52196495b0d10aca2d6f5d7dbdc0a215..c32ad68207f69f39813bb0b14fc9c8f54214d3d7 100644 (file)
--- a/vp10/encoder/bitstream.c
+++ b/vp10/encoder/bitstream.c
@@ -1291,6 +1291,12 @@ static size_t write_compressed_header(VP10_COMP *cpi, uint8_t *data) {
       prob_diff_update(vp10_intra_mode_tree, cm->fc->y_mode_prob[i],
                        counts->y_mode[i], INTRA_MODES, &header_bc);
 
+#if CONFIG_MISC_FIXES
+    for (i = 0; i < INTRA_MODES; ++i)
+      prob_diff_update(vp10_intra_mode_tree, cm->fc->uv_mode_prob[i],
+                       counts->uv_mode[i], INTRA_MODES, &header_bc);
+#endif
+
     for (i = 0; i < PARTITION_CONTEXTS; ++i)
       prob_diff_update(vp10_partition_tree, fc->partition_prob[i],
                        counts->partition[i], PARTITION_TYPES, &header_bc);