From 49ad845c3356e6f96010f81359b9dc191cc92246 Mon Sep 17 00:00:00 2001
From: Ari Koivula <ari@koivu.la>
Date: Mon, 6 Oct 2014 19:19:51 +0300
Subject: [PATCH] Add cabac bits for part_mode.

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

diff --git a/src/search.c b/src/search.c
index b0a6ba1a..822b1563 100644
--- a/src/search.c
+++ b/src/search.c
@@ -1522,6 +1522,12 @@ static double search_cu(encoder_state * const encoder_state, int x, int y, int d
       split_cost += CTX_ENTROPY_FBITS(ctx, 1);
     }
 
+    if (cur_cu->type == CU_INTRA && depth == MAX_DEPTH) {
+      const cabac_ctx *ctx = &(encoder_state->cabac.ctx.part_size_model[0]);
+      cost += CTX_ENTROPY_FBITS(ctx, 1);  // 2Nx2N
+      split_cost += CTX_ENTROPY_FBITS(ctx, 0);  // NxN
+    }
+
     // If skip mode was selected for the block, skip further search.
     // Skip mode means there's no coefficients in the block, so splitting
     // might not give any better results but takes more time to do.