commit a73bdb045b2580fd5d81c25d46d7dfba653a4603
author Yaowu Xu <yaowu@google.com> Thu Nov 23 15:31:38 2017 -0800
committer Yaowu Xu <yaowu@google.com> Fri Nov 24 02:51:10 2017 +0000
tree ba09e2343c504d4bf7aa7f067094b41bdb80e65b
parent 97800fd0466efd528f891b4652d10ffc00a2cba0

Revert "[CFL] Cap Averaging Area"

This reverts commit 873d8a46e8da8328709ea65c2ae7b08515a6ae7b.

BUG=aomedia:1063

Change-Id: Ief39ff278f63b9ab5b1f4692dc61ff997e0a5474

av1/common/cfl.c

diff --git a/av1/common/cfl.c b/av1/common/cfl.c
index e3b6308..d930ec2 100644
--- a/av1/common/cfl.c
+++ b/av1/common/cfl.c
@@ -70,42 +70,38 @@
 static void cfl_subtract_averages(CFL_CTX *cfl, TX_SIZE tx_size) {
   const int width = cfl->uv_width;
   const int height = cfl->uv_height;
-
-  const int avg_height_log2 =
-      AOMMIN(tx_size_high_log2[tx_size], 1 - cfl->subsampling_y + 4);
-  const int avg_width_log2 =
-      AOMMIN(tx_size_wide_log2[tx_size], 1 - cfl->subsampling_x + 4);
-  const int avg_height = 1 << avg_height_log2;
-  const int avg_width = 1 << avg_width_log2;
-  const int num_pel_log2 = avg_height_log2 + avg_width_log2;
-  const int block_row_stride = MAX_SB_SIZE << avg_height_log2;
+  const int tx_height = tx_size_high[tx_size];
+  const int tx_width = tx_size_wide[tx_size];
+  const int block_row_stride = MAX_SB_SIZE << tx_size_high_log2[tx_size];
+  const int num_pel_log2 =
+      (tx_size_high_log2[tx_size] + tx_size_wide_log2[tx_size]);
 
   int16_t *pred_buf_q3 = cfl->pred_buf_q3;
 
   cfl_pad(cfl, width, height);
 
-  for (int b_j = 0; b_j < height; b_j += avg_height) {
-    for (int b_i = 0; b_i < width; b_i += avg_width) {
+  for (int b_j = 0; b_j < height; b_j += tx_height) {
+    for (int b_i = 0; b_i < width; b_i += tx_width) {
       int sum_q3 = 0;
-      int16_t *avg_pred_buf_q3 = pred_buf_q3;
-      for (int t_j = 0; t_j < avg_height; t_j++) {
-        for (int t_i = b_i; t_i < b_i + avg_width; t_i++) {
-          sum_q3 += avg_pred_buf_q3[t_i];
+      int16_t *tx_pred_buf_q3 = pred_buf_q3;
+      for (int t_j = 0; t_j < tx_height; t_j++) {
+        for (int t_i = b_i; t_i < b_i + tx_width; t_i++) {
+          sum_q3 += tx_pred_buf_q3[t_i];
         }
-        avg_pred_buf_q3 += MAX_SB_SIZE;
+        tx_pred_buf_q3 += MAX_SB_SIZE;
       }
       int avg_q3 = (sum_q3 + (1 << (num_pel_log2 - 1))) >> num_pel_log2;
       // Loss is never more than 1/2 (in Q3)
       assert(abs((avg_q3 * (1 << num_pel_log2)) - sum_q3) <=
              1 << num_pel_log2 >> 1);
 
-      avg_pred_buf_q3 = pred_buf_q3;
-      for (int t_j = 0; t_j < avg_height; t_j++) {
-        for (int t_i = b_i; t_i < b_i + avg_width; t_i++) {
-          avg_pred_buf_q3[t_i] -= avg_q3;
+      tx_pred_buf_q3 = pred_buf_q3;
+      for (int t_j = 0; t_j < tx_height; t_j++) {
+        for (int t_i = b_i; t_i < b_i + tx_width; t_i++) {
+          tx_pred_buf_q3[t_i] -= avg_q3;
         }
 
-        avg_pred_buf_q3 += MAX_SB_SIZE;
+        tx_pred_buf_q3 += MAX_SB_SIZE;
       }
     }
     pred_buf_q3 += block_row_stride;