commit d7ec47f0c2b6fc80ebb659732d7e6ab8ba453304
author Jingning Han <jingning@google.com> Tue Nov 29 16:34:40 2016 -0800
committer Jingning Han <jingning@google.com> Wed Nov 30 03:52:43 2016 +0000
tree 810f23ed7ff40d5e02a5e754fe2ae2333f31df50
parent 124022279913fb47c4740e0dbe99ddede87f4d39

Add inverse 2x2 transform function

This function will be used in the 4x4 coding block unit.

Change-Id: I981c2abce4ae1e1571adb0d87c76e2307e94300b

av1/common/idct.c

diff --git a/av1/common/idct.c b/av1/common/idct.c
index 156fc96..9b5536f 100644
--- a/av1/common/idct.c
+++ b/av1/common/idct.c
@@ -1069,6 +1069,35 @@
 }
 #endif  // CONFIG_TX64X64
 
+#if CONFIG_CB4X4
+void av1_inv_txfm_add_2x2(const tran_low_t *input, uint8_t *dest, int stride,
+                          int eob, TX_TYPE tx_type, int lossless) {
+  tran_high_t a1 = input[0] >> UNIT_QUANT_SHIFT;
+  tran_high_t b1 = input[1] >> UNIT_QUANT_SHIFT;
+  tran_high_t c1 = input[2] >> UNIT_QUANT_SHIFT;
+  tran_high_t d1 = input[3] >> UNIT_QUANT_SHIFT;
+
+  tran_high_t a2 = a1 + c1;
+  tran_high_t b2 = b1 + d1;
+  tran_high_t c2 = a1 - c1;
+  tran_high_t d2 = b1 - d1;
+
+  (void)tx_type;
+  (void)lossless;
+  (void)eob;
+
+  a1 = (a2 + b2) >> 1;
+  b1 = (a2 - b2) >> 1;
+  c1 = (c2 + d2) >> 1;
+  d1 = (c2 - d2) >> 1;
+
+  dest[0] = clip_pixel_add(dest[0], WRAPLOW(a1));
+  dest[1] = clip_pixel_add(dest[1], WRAPLOW(b1));
+  dest[stride] = clip_pixel_add(dest[stride], WRAPLOW(c1));
+  dest[stride + 1] = clip_pixel_add(dest[stride + 1], WRAPLOW(d1));
+}
+#endif
+
 void av1_inv_txfm_add_4x4(const tran_low_t *input, uint8_t *dest, int stride,
                           int eob, TX_TYPE tx_type, int lossless) {
   if (lossless) {
@@ -2279,6 +2308,11 @@
       // case.
       av1_inv_txfm_add_4x4(input, dest, stride, eob, tx_type, lossless);
       break;
+#if CONFIG_CB4X4
+    case TX_2X2:
+      av1_inv_txfm_add_2x2(input, dest, stride, eob, tx_type, lossless);
+      break;
+#endif
     default: assert(0 && "Invalid transform size"); break;
   }
 }