From 880d09b27ae698d2e6409c0fb85787a47a3ead5d Mon Sep 17 00:00:00 2001
From: Ari Koivula <ari@koivu.la>
Date: Tue, 15 Apr 2014 16:34:59 +0300
Subject: [PATCH] Add sao_type_idx to SAO mode cost estimation.

---
 src/sao.c | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/sao.c b/src/sao.c
index 3c9deab7..c25c746c 100644
--- a/src/sao.c
+++ b/src/sao.c
@@ -155,6 +155,9 @@ static int sao_mode_bits_edge(int edge_class, int offsets[NUM_SAO_EDGE_CATEGORIE
     }
   }
 
+  // TR coded type_idx_, edge = 2 = cMax
+  mode_bits += 1;
+
   // TR coded offsets.
   {
     sao_eo_cat edge_cat;
@@ -198,6 +201,9 @@ static int sao_mode_bits_band(int band_position, int offsets[4],
     }
   }
 
+  // TR coded sao_type_idx_, band = 1
+  mode_bits += 2;
+
   // TR coded offsets and possible FL coded offset signs.
   {
     int i;