From 3c3c9a26c65a4a399310bd563f197aaee8dc7d15 Mon Sep 17 00:00:00 2001
From: Ari Koivula <ari@koivu.la>
Date: Mon, 12 May 2014 08:45:54 +0300
Subject: [PATCH] Move scan order selection to a function.

---
 src/encoder.c | 45 ++++++++++++++++++++++-----------------------
 src/tables.h  |  4 ++--
 2 files changed, 24 insertions(+), 25 deletions(-)

diff --git a/src/encoder.c b/src/encoder.c
index c38257ed..99171250 100644
--- a/src/encoder.c
+++ b/src/encoder.c
@@ -2871,6 +2871,24 @@ static void reconstruct_chroma(const encoder_state * const encoder_state, cu_inf
   }
 }
 
+coeff_scan_order_t get_scan_order(int intra_mode, int depth)
+{
+  coeff_scan_order_t scan_idx = SCAN_DIAG;
+
+  // Scan mode is diagonal, except for 4x4+8x8 luma and 4x4 chroma, where:
+  // - angular 6-14 = vertical
+  // - angular 22-30 = horizontal
+  if (depth >= 3) {
+    if (intra_mode >= 6 && intra_mode <= 14) {
+      scan_idx = SCAN_VER;
+    } else if (intra_mode >= 22 && intra_mode <= 30) {
+      scan_idx = SCAN_HOR;
+    }
+  }
+
+  return scan_idx;
+}
+
 /**
  * This function calculates the residual coefficients for a region of the LCU
  * (defined by x, y and depth) and updates the reconstruction with the
@@ -2988,33 +3006,14 @@ void encode_transform_tree(encoder_state * const encoder_state, int32_t x, int32
 
     // Pick coeff scan mode according to intra prediction mode.
     if (cur_cu->type == CU_INTRA) {
-      int pu_index = PU_INDEX(x_pu, y_pu);
-      int luma_mode = cur_cu->intra[pu_index].mode;
       int chroma_mode = cur_cu->intra[0].mode_chroma;
       if (chroma_mode == 36) {
-        chroma_mode = luma_mode;
-      }
-      scan_idx_luma = SCAN_DIAG;
-      scan_idx_chroma = SCAN_DIAG;
-
-      // Scan mode is diagonal, except for 4x4+8x8 luma and 4x4 chroma, where:
-      // - angular 6-14 = vertical
-      // - angular 22-30 = horizontal
-      if (width <= 8) {
-        if (luma_mode >= 6 && luma_mode <= 14) {
-          scan_idx_luma = SCAN_VER;
-        } else if (luma_mode >= 22 && luma_mode <= 30) {
-          scan_idx_luma = SCAN_HOR;
-        }
-
-        if (chroma_mode >= 6 && chroma_mode <= 14) {
-          scan_idx_chroma = SCAN_VER;
-        } else if (chroma_mode >= 22 && chroma_mode <= 30) {
-          scan_idx_chroma = SCAN_HOR;
-        }
+        chroma_mode = cur_cu->intra[PU_INDEX(x_pu, y_pu)].mode;
       }
+      scan_idx_luma = get_scan_order(cur_cu->intra[PU_INDEX(x_pu, y_pu)].mode, depth);
+      scan_idx_chroma = get_scan_order(chroma_mode, depth);
     }
-
+    
     // Copy Luma and Chroma to the pred-block
     for(y = 0; y < width; y++) {
       for(x = 0; x < width; x++) {
diff --git a/src/tables.h b/src/tables.h
index 3541d172..04b5cd34 100644
--- a/src/tables.h
+++ b/src/tables.h
@@ -84,12 +84,12 @@ static const uint32_t * const g_sig_last_scan_cg[4][3] = {
 };
 
 
-enum COEFF_SCAN_TYPE
+typedef enum
 {
   SCAN_DIAG = 0, // up-right diagonal scan
   SCAN_HOR,      // horizontal first scan
   SCAN_VER       // vertical first scan
-};
+} coeff_scan_order_t;
 
 
 /**