shithub: openh264

Download patch

ref: ca97f78ae828617e877951b9f1341ac67974ce52
parent: 8084a2c797d71dfd918a6bef35bbcb973901169b
author: Martin Storsjö <[email protected]>
date: Mon Jan 26 09:58:19 EST 2015 

Add unit tests for special cased MC functions in the encoder


--- a/test/encoder/EncUT_MotionCompensation.cpp
+++ b/test/encoder/EncUT_MotionCompensation.cpp
@@ -36,3 +36,84 @@
 DEF_CHROMA_MCTEST_SUBCASE(pfx,a,b,8,8)
 
 #include "mc_test_common.h"
+
+TEST (EncMcAvg, PixelAvg) {
+  SMcFunc sMcFunc;
+  for (int32_t k = 0; k < 2; k++) {
+    for (int32_t w = 0; w < 2; w++) {
+      int32_t width = 8 << w;
+      int32_t height = 16;
+      uint32_t uiCpuFlag = k == 0 ? 0 : WelsCPUFeatureDetect (NULL);
+      WelsInitMcFuncs (&sMcFunc, uiCpuFlag);
+      uint8_t uSrc1[MC_BUFF_HEIGHT][MC_BUFF_SRC_STRIDE];
+      uint8_t uSrc2[MC_BUFF_HEIGHT][MC_BUFF_SRC_STRIDE];
+      ENFORCE_STACK_ALIGN_2D (uint8_t, uDstAnchor, MC_BUFF_HEIGHT, MC_BUFF_DST_STRIDE, 16);
+      ENFORCE_STACK_ALIGN_2D (uint8_t, uDstTest, MC_BUFF_HEIGHT, MC_BUFF_DST_STRIDE, 16);
+      for (int32_t j = 0; j < MC_BUFF_HEIGHT; j++) {
+        for (int32_t i = 0; i < MC_BUFF_SRC_STRIDE; i++) {
+          uSrc1[j][i] = rand() % 256;
+          uSrc2[j][i] = rand() % 256;
+        }
+      }
+      PixelAvgAnchor (uDstAnchor[0], MC_BUFF_DST_STRIDE, uSrc1[0], MC_BUFF_SRC_STRIDE, uSrc2[0], MC_BUFF_SRC_STRIDE, width,
+                      height);
+      sMcFunc.pfSampleAveraging[w] (uDstTest[0], MC_BUFF_DST_STRIDE, uSrc1[0], MC_BUFF_SRC_STRIDE, uSrc2[0],
+                                    MC_BUFF_SRC_STRIDE, height);
+      for (int32_t j = 0; j < height; j++) {
+        for (int32_t i = 0; i < width; i++) {
+          ASSERT_EQ (uDstAnchor[j][i], uDstTest[j][i]);
+        }
+      }
+    }
+  }
+}
+
+TEST (EncMcHalfpel, LumaHalfpel) {
+  SMcFunc sMcFunc;
+  for (int32_t k = 0; k < 2; k++) {
+    for (int32_t w = 0; w < 2; w++) {
+      int32_t width = 8 << w;
+      int32_t height = 16;
+      uint8_t uAnchor[4][MC_BUFF_HEIGHT][MC_BUFF_SRC_STRIDE];
+      uint8_t uSrcTest[MC_BUFF_HEIGHT][MC_BUFF_SRC_STRIDE];
+      ENFORCE_STACK_ALIGN_2D (uint8_t, uDstTest, MC_BUFF_HEIGHT, MC_BUFF_DST_STRIDE, 16);
+      uint8_t* uAnchors[4];
+      int16_t pBuf[MC_BUFF_DST_STRIDE];
+      uAnchors[0] = &uAnchor[0][4][4];
+      uAnchors[1] = &uAnchor[1][4][4];
+      uAnchors[2] = &uAnchor[2][4][4];
+      uAnchors[3] = &uAnchor[3][4][4];
+
+      memset (uAnchor, 0, 4 * sizeof (uint8_t)*MC_BUFF_HEIGHT * MC_BUFF_DST_STRIDE);
+      memset (uDstTest, 0, sizeof (uint8_t)*MC_BUFF_HEIGHT * MC_BUFF_DST_STRIDE);
+      for (int32_t j = 0; j < MC_BUFF_HEIGHT; j++) {
+        for (int32_t i = 0; i < MC_BUFF_SRC_STRIDE; i++) {
+          uAnchor[0][j][i] = uSrcTest[j][i] = rand() % 256;
+        }
+      }
+
+      uint32_t uiCpuFlag = k == 0 ? 0 : WelsCPUFeatureDetect (NULL);
+      WelsInitMcFuncs (&sMcFunc, uiCpuFlag);
+
+      MCHalfPelFilterAnchor (uAnchors[1], uAnchors[2], uAnchors[3], uAnchors[0], MC_BUFF_SRC_STRIDE, width, height, pBuf + 4);
+      sMcFunc.pfLumaHalfpelHor (&uSrcTest[4][4], MC_BUFF_SRC_STRIDE, uDstTest[0], MC_BUFF_DST_STRIDE, width + 1, height);
+      for (int32_t j = 0; j < height; j++) {
+        for (int32_t i = 0; i < width; i++) {
+          ASSERT_EQ (uAnchor[1][4 + j][4 + i], uDstTest[j][i]);
+        }
+      }
+      sMcFunc.pfLumaHalfpelVer (&uSrcTest[4][4], MC_BUFF_SRC_STRIDE, uDstTest[0], MC_BUFF_DST_STRIDE, width, height + 1);
+      for (int32_t j = 0; j < height; j++) {
+        for (int32_t i = 0; i < width; i++) {
+          ASSERT_EQ (uAnchor[2][4 + j][4 + i], uDstTest[j][i]);
+        }
+      }
+      sMcFunc.pfLumaHalfpelCen (&uSrcTest[4][4], MC_BUFF_SRC_STRIDE, uDstTest[0], MC_BUFF_DST_STRIDE, width + 1, height + 1);
+      for (int32_t j = 0; j < height; j++) {
+        for (int32_t i = 0; i < width; i++) {
+          ASSERT_EQ (uAnchor[3][4 + j][4 + i], uDstTest[j][i]);
+        }
+      }
+    }
+  }
+}