ref: e6f838e33106cc2e3f5b9739cd8fdae77da8921b
parent: 3f839e367ced86077db3992a5c69fce1d0fc4233
author: zhiliang wang <[email protected]>
date: Thu Aug 21 11:36:57 EDT 2014
Add arm UT for Sad,IntraSad,IntraPred.
--- a/codec/encoder/core/inc/sample.h
+++ b/codec/encoder/core/inc/sample.h
@@ -53,6 +53,8 @@
int32_t WelsSampleSadIntra16x16Combined3_c (uint8_t*, int32_t, uint8_t*, int32_t, int32_t*, int32_t, uint8_t*);
int32_t WelsSampleSatdIntra8x8Combined3_c (uint8_t*, int32_t, uint8_t*, int32_t, int32_t*, int32_t, uint8_t*,
uint8_t*, uint8_t*);
+int32_t WelsSampleSadIntra8x8Combined3_c (uint8_t*, int32_t, uint8_t*, int32_t, int32_t*, int32_t, uint8_t*,
+ uint8_t*, uint8_t*);
#if defined(__cplusplus)
extern "C" {--- a/test/decoder/DecUT_IntraPrediction.cpp
+++ b/test/decoder/DecUT_IntraPrediction.cpp
@@ -624,3 +624,29 @@
GENERATE_8x8_UT (WelsDecoderIChromaPredDc_neon, WelsIChromaPredDc_ref, 1, WELS_CPU_NEON)
GENERATE_8x8_UT (WelsDecoderIChromaPredPlane_neon, WelsIChromaPredPlane_ref, 1, WELS_CPU_NEON)
#endif
+
+#if defined(HAVE_NEON_AARCH64)
+GENERATE_16x16_UT (WelsDecoderI16x16LumaPredV_AArch64_neon, LumaI16x16PredV, 1, WELS_CPU_NEON)
+GENERATE_16x16_UT (WelsDecoderI16x16LumaPredH_AArch64_neon, LumaI16x16PredH, 1, WELS_CPU_NEON)
+GENERATE_16x16_UT (WelsDecoderI16x16LumaPredDc_AArch64_neon, LumaI16x16PredDC, 1, WELS_CPU_NEON)
+GENERATE_16x16_UT (WelsDecoderI16x16LumaPredDcTop_AArch64_neon, LumaI16x16PredDCTop, 1, WELS_CPU_NEON)
+GENERATE_16x16_UT (WelsDecoderI16x16LumaPredDcLeft_AArch64_neon, LumaI16x16PredDCLeft, 1, WELS_CPU_NEON)
+GENERATE_16x16_UT (WelsDecoderI16x16LumaPredPlane_AArch64_neon, WelsI16x16LumaPredPlane_ref, 1, WELS_CPU_NEON)
+
+GENERATE_4x4_UT (WelsDecoderI4x4LumaPredH_AArch64_neon, LumaI4x4PredH, 1, WELS_CPU_NEON)
+GENERATE_4x4_UT (WelsDecoderI4x4LumaPredDDL_AArch64_neon, WelsI4x4LumaPredDDL_ref, 1, WELS_CPU_NEON)
+GENERATE_4x4_UT (WelsDecoderI4x4LumaPredDDLTop_AArch64_neon, WelsI4x4LumaPredDDLTop_ref, 1, WELS_CPU_NEON)
+GENERATE_4x4_UT (WelsDecoderI4x4LumaPredVL_AArch64_neon, WelsI4x4LumaPredVL_ref, 1, WELS_CPU_NEON)
+GENERATE_4x4_UT (WelsDecoderI4x4LumaPredVLTop_AArch64_neon, WelsI4x4LumaPredVLTop_ref, 1, WELS_CPU_NEON)
+GENERATE_4x4_UT (WelsDecoderI4x4LumaPredVR_AArch64_neon, WelsI4x4LumaPredVR_ref, 1, WELS_CPU_NEON)
+GENERATE_4x4_UT (WelsDecoderI4x4LumaPredHU_AArch64_neon, WelsI4x4LumaPredHU_ref, 1, WELS_CPU_NEON)
+GENERATE_4x4_UT (WelsDecoderI4x4LumaPredHD_AArch64_neon, WelsI4x4LumaPredHD_ref, 1, WELS_CPU_NEON)
+GENERATE_4x4_UT (WelsDecoderI4x4LumaPredDc_AArch64_neon, LumaI4x4PredDC, 1, WELS_CPU_NEON)
+GENERATE_4x4_UT (WelsDecoderI4x4LumaPredDcTop_AArch64_neon, LumaI4x4PredDCTop, 1, WELS_CPU_NEON)
+
+GENERATE_8x8_UT (WelsDecoderIChromaPredV_AArch64_neon, LumaI8x8PredV, 1, WELS_CPU_NEON)
+GENERATE_8x8_UT (WelsDecoderIChromaPredH_AArch64_neon, LumaI8x8PredH, 1, WELS_CPU_NEON)
+GENERATE_8x8_UT (WelsDecoderIChromaPredDc_AArch64_neon, WelsIChromaPredDc_ref, 1, WELS_CPU_NEON)
+GENERATE_8x8_UT (WelsDecoderIChromaPredPlane_AArch64_neon, WelsIChromaPredPlane_ref, 1, WELS_CPU_NEON)
+GENERATE_8x8_UT (WelsDecoderIChromaPredDcTop_AArch64_neon, WelsIChromaPredDcTop_ref, 1, WELS_CPU_NEON)
+#endif
--- a/test/encoder/EncUT_Sample.cpp
+++ b/test/encoder/EncUT_Sample.cpp
@@ -8,136 +8,123 @@
#include "get_intra_predictor.h"
using namespace WelsEnc;
-#ifdef X86_ASM
-TEST (IntraSadSatdFuncTest, WelsIntra16x16Combined3Sad_ssse3) {- const int32_t iLineSizeDec = 32;
- const int32_t iLineSizeEnc = 32;
- int32_t tmpa, tmpb;
- int32_t iBestMode_c, iBestMode_a, iLambda = 50;
- CMemoryAlign cMemoryAlign (0);
- int32_t iCpuCores = 0;
- uint32_t m_uiCpuFeatureFlag = WelsCPUFeatureDetect (&iCpuCores);
- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSSE3))
- return;
- uint8_t* pDec = (uint8_t*)cMemoryAlign.WelsMalloc (iLineSizeDec << 5, "pDec");
- uint8_t* pEnc = (uint8_t*)cMemoryAlign.WelsMalloc (iLineSizeEnc << 5, "pEnc");
- uint8_t* pDst = (uint8_t*)cMemoryAlign.WelsMalloc (512, "pDst");
- for (int i = 0; i < (iLineSizeDec << 5); i++)
- pDec[i] = rand() % 256;
- for (int i = 0; i < (iLineSizeEnc << 5); i++)
- pEnc[i] = rand() % 256;
- for (int i = 0; i < 512; i++)
- pDst[i] = rand() % 256;
- tmpa = WelsSampleSadIntra16x16Combined3_c (pDec + 128, iLineSizeDec, pEnc, iLineSizeEnc, &iBestMode_c, iLambda, pDst);
- tmpb = WelsIntra16x16Combined3Sad_ssse3 (pDec + 128, iLineSizeDec, pEnc, iLineSizeEnc, &iBestMode_a, iLambda, pDst);
-
- ASSERT_EQ (tmpa, tmpb);
- ASSERT_EQ (iBestMode_c, iBestMode_a);
-
- cMemoryAlign.WelsFree (pDec, "pDec");
- cMemoryAlign.WelsFree (pEnc, "pEnc");
- cMemoryAlign.WelsFree (pDst, "pDst");
+#define GENERATE_Intra16x16_UT(func, ref, ASM, CPUFLAGS) \
+TEST (IntraSadSatdFuncTest, func) { \+ const int32_t iLineSizeDec = 32; \
+ const int32_t iLineSizeEnc = 32; \
+ int32_t tmpa, tmpb; \
+ int32_t iBestMode_c, iBestMode_a, iLambda = 50; \
+ if (ASM) {\+ int32_t iCpuCores = 0; \
+ uint32_t m_uiCpuFeatureFlag = WelsCPUFeatureDetect (&iCpuCores); \
+ if (0 == (m_uiCpuFeatureFlag & CPUFLAGS)) \
+ return; \
+ } \
+ ENFORCE_STACK_ALIGN_1D (uint8_t, pDec, iLineSizeDec << 5, 16); \
+ ENFORCE_STACK_ALIGN_1D (uint8_t, pEnc, iLineSizeEnc << 5, 16); \
+ ENFORCE_STACK_ALIGN_1D (uint8_t, pDst, 512, 16); \
+ for (int i = 0; i < (iLineSizeDec << 5); i++) \
+ pDec[i] = rand() % 256; \
+ for (int i = 0; i < (iLineSizeEnc << 5); i++) \
+ pEnc[i] = rand() % 256; \
+ for (int i = 0; i < 512; i++) \
+ pDst[i] = rand() % 256; \
+ tmpa = ref (pDec + 128, iLineSizeDec, pEnc, iLineSizeEnc, &iBestMode_c, iLambda, pDst); \
+ tmpb = func (pDec + 128, iLineSizeDec, pEnc, iLineSizeEnc, &iBestMode_a, iLambda, pDst); \
+ ASSERT_EQ (tmpa, tmpb); \
+ ASSERT_EQ (iBestMode_c, iBestMode_a); \
}
-TEST (IntraSadSatdFuncTest, WelsIntra16x16Combined3Satd_sse41) {- const int32_t iLineSizeDec = 32;
- const int32_t iLineSizeEnc = 32;
- int32_t tmpa, tmpb;
- int32_t iBestMode_c, iBestMode_a, iLambda = 50;
- CMemoryAlign cMemoryAlign (0);
- int32_t iCpuCores = 0;
- uint32_t m_uiCpuFeatureFlag = WelsCPUFeatureDetect (&iCpuCores);
- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE41))
- return;
- uint8_t* pDec = (uint8_t*)cMemoryAlign.WelsMalloc (iLineSizeDec << 5, "pDec");
- uint8_t* pEnc = (uint8_t*)cMemoryAlign.WelsMalloc (iLineSizeEnc << 5, "pEnc");
- uint8_t* pDst = (uint8_t*)cMemoryAlign.WelsMalloc (512, "pDst");
- for (int i = 0; i < (iLineSizeDec << 5); i++)
- pDec[i] = rand() % 256;
- for (int i = 0; i < (iLineSizeEnc << 5); i++)
- pEnc[i] = rand() % 256;
- for (int i = 0; i < 512; i++)
- pDst[i] = rand() % 256;
- tmpa = WelsSampleSatdIntra16x16Combined3_c (pDec + 128, iLineSizeDec, pEnc, iLineSizeEnc, &iBestMode_c, iLambda, pDst);
- tmpb = WelsIntra16x16Combined3Satd_sse41 (pDec + 128, iLineSizeDec, pEnc, iLineSizeEnc, &iBestMode_a, iLambda, pDst);
- ASSERT_EQ (tmpa, tmpb);
- ASSERT_EQ (iBestMode_c, iBestMode_a);
- cMemoryAlign.WelsFree (pDec, "pDec");
- cMemoryAlign.WelsFree (pEnc, "pEnc");
- cMemoryAlign.WelsFree (pDst, "pDst");
+#define GENERATE_Intra4x4_UT(func, ASM, CPUFLAGS) \
+TEST (IntraSadSatdFuncTest, func) { \+ const int32_t iLineSizeDec = 32; \
+ const int32_t iLineSizeEnc = 32; \
+ int32_t tmpa, tmpb; \
+ int32_t iBestMode_c, iBestMode_a, iLambda = 50; \
+ int32_t lambda[2] = {iLambda << 2, iLambda}; \+ int32_t iPredMode = rand() % 3; \
+ if (ASM) {\+ int32_t iCpuCores = 0; \
+ uint32_t m_uiCpuFeatureFlag = WelsCPUFeatureDetect (&iCpuCores); \
+ if (0 == (m_uiCpuFeatureFlag & CPUFLAGS)) \
+ return; \
+ } \
+ ENFORCE_STACK_ALIGN_1D (uint8_t, pDec, iLineSizeDec << 5, 16); \
+ ENFORCE_STACK_ALIGN_1D (uint8_t, pEnc, iLineSizeEnc << 5, 16); \
+ ENFORCE_STACK_ALIGN_1D (uint8_t, pDst, 512, 16); \
+ for (int i = 0; i < (iLineSizeDec << 5); i++) \
+ pDec[i] = rand() % 256; \
+ for (int i = 0; i < (iLineSizeEnc << 5); i++) \
+ pEnc[i] = rand() % 256; \
+ for (int i = 0; i < 512; i++) \
+ pDst[i] = rand() % 256; \
+ tmpa = WelsSampleSatdIntra4x4Combined3_c (pDec + 128, iLineSizeDec, pEnc, iLineSizeEnc, pDst, &iBestMode_c, \
+ lambda[iPredMode == 2], lambda[iPredMode == 1], lambda[iPredMode == 0]); \
+ tmpb = func (pDec + 128, iLineSizeDec, pEnc, iLineSizeEnc, pDst, &iBestMode_a, \
+ lambda[iPredMode == 2], lambda[iPredMode == 1], lambda[iPredMode == 0]); \
+ ASSERT_EQ (tmpa, tmpb); \
+ ASSERT_EQ (iBestMode_c, iBestMode_a); \
}
-TEST (IntraSadSatdFuncTest, WelsSampleSatdThree4x4_sse2) {- const int32_t iLineSizeDec = 32;
- const int32_t iLineSizeEnc = 32;
- int32_t tmpa, tmpb;
- int32_t iBestMode_c, iBestMode_a, iLambda = 50;
- int32_t lambda[2] = {iLambda << 2, iLambda};- int32_t iPredMode = rand() % 3;
- CMemoryAlign cMemoryAlign (0);
- int32_t iCpuCores = 0;
- uint32_t m_uiCpuFeatureFlag = WelsCPUFeatureDetect (&iCpuCores);
- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE2))
- return;
- uint8_t* pDec = (uint8_t*)cMemoryAlign.WelsMalloc (iLineSizeDec << 5, "pDec");
- uint8_t* pEnc = (uint8_t*)cMemoryAlign.WelsMalloc (iLineSizeEnc << 5, "pEnc");
- uint8_t* pDst = (uint8_t*)cMemoryAlign.WelsMalloc (512, "pDst");
- for (int i = 0; i < (iLineSizeDec << 5); i++)
- pDec[i] = rand() % 256;
- for (int i = 0; i < (iLineSizeEnc << 5); i++)
- pEnc[i] = rand() % 256;
- for (int i = 0; i < 512; i++)
- pDst[i] = rand() % 256;
- tmpa = WelsSampleSatdIntra4x4Combined3_c (pDec + 128, iLineSizeDec, pEnc, iLineSizeEnc, pDst, &iBestMode_c,
- lambda[iPredMode == 2], lambda[iPredMode == 1], lambda[iPredMode == 0]);
- tmpb = WelsSampleSatdThree4x4_sse2 (pDec + 128, iLineSizeDec, pEnc, iLineSizeEnc, pDst, &iBestMode_a,
- lambda[iPredMode == 2], lambda[iPredMode == 1], lambda[iPredMode == 0]);
- ASSERT_EQ (tmpa, tmpb);
- ASSERT_EQ (iBestMode_c, iBestMode_a);
- cMemoryAlign.WelsFree (pDec, "pDec");
- cMemoryAlign.WelsFree (pEnc, "pEnc");
- cMemoryAlign.WelsFree (pDst, "pDst");
+#define GENERATE_Intra8x8_UT(func, ref, ASM, CPUFLAGS) \
+TEST (IntraSadSatdFuncTest, func) { \+ const int32_t iLineSizeDec = 32; \
+ const int32_t iLineSizeEnc = 32; \
+ int32_t tmpa, tmpb; \
+ int32_t iBestMode_c, iBestMode_a, iLambda = 50; \
+ if (ASM) {\+ int32_t iCpuCores = 0; \
+ uint32_t m_uiCpuFeatureFlag = WelsCPUFeatureDetect (&iCpuCores); \
+ if (0 == (m_uiCpuFeatureFlag & CPUFLAGS)) \
+ return; \
+ } \
+ ENFORCE_STACK_ALIGN_1D (uint8_t, pDecCb, iLineSizeDec << 5, 16); \
+ ENFORCE_STACK_ALIGN_1D (uint8_t, pEncCb, iLineSizeEnc << 5, 16); \
+ ENFORCE_STACK_ALIGN_1D (uint8_t, pDecCr, iLineSizeDec << 5, 16); \
+ ENFORCE_STACK_ALIGN_1D (uint8_t, pEncCr, iLineSizeEnc << 5, 16); \
+ ENFORCE_STACK_ALIGN_1D (uint8_t, pDstChma, 512, 16); \
+ for (int i = 0; i < (iLineSizeDec << 5); i++) { \+ pDecCb[i] = rand() % 256; \
+ pDecCr[i] = rand() % 256; \
+ } \
+ for (int i = 0; i < (iLineSizeEnc << 5); i++) { \+ pEncCb[i] = rand() % 256; \
+ pEncCr[i] = rand() % 256; \
+ } \
+ for (int i = 0; i < 512; i++) \
+ pDstChma[i] = rand() % 256; \
+ tmpa = ref (pDecCb + 128, iLineSizeDec, pEncCb, iLineSizeEnc, &iBestMode_c, iLambda, \
+ pDstChma, pDecCr + 128, pEncCr); \
+ tmpb = func (pDecCb + 128, iLineSizeDec, pEncCb, iLineSizeEnc, &iBestMode_a, iLambda, \
+ pDstChma, pDecCr + 128, pEncCr); \
+ ASSERT_EQ (tmpa, tmpb); \
+ ASSERT_EQ (iBestMode_c, iBestMode_a); \
}
-TEST (IntraSadSatdFuncTest, WelsIntraChroma8x8Combined3Satd_sse41) {- const int32_t iLineSizeDec = 32;
- const int32_t iLineSizeEnc = 32;
- int32_t tmpa, tmpb;
- int32_t iBestMode_c, iBestMode_a, iLambda = 50;
- CMemoryAlign cMemoryAlign (0);
- int32_t iCpuCores = 0;
- uint32_t m_uiCpuFeatureFlag = WelsCPUFeatureDetect (&iCpuCores);
- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE41))
- return;
- uint8_t* pDecCb = (uint8_t*)cMemoryAlign.WelsMalloc (iLineSizeDec << 5, "pDecCb");
- uint8_t* pEncCb = (uint8_t*)cMemoryAlign.WelsMalloc (iLineSizeEnc << 5, "pEncCb");
- uint8_t* pDecCr = (uint8_t*)cMemoryAlign.WelsMalloc (iLineSizeDec << 5, "pDecCr");
- uint8_t* pEncCr = (uint8_t*)cMemoryAlign.WelsMalloc (iLineSizeEnc << 5, "pEncCr");
- uint8_t* pDstChma = (uint8_t*)cMemoryAlign.WelsMalloc (512, "pDstChma");
- for (int i = 0; i < (iLineSizeDec << 5); i++) {- pDecCb[i] = rand() % 256;
- pDecCr[i] = rand() % 256;
- }
- for (int i = 0; i < (iLineSizeEnc << 5); i++) {- pEncCb[i] = rand() % 256;
- pEncCr[i] = rand() % 256;
- }
- for (int i = 0; i < 512; i++)
- pDstChma[i] = rand() % 256;
- tmpa = WelsSampleSatdIntra8x8Combined3_c (pDecCb + 128, iLineSizeDec, pEncCb, iLineSizeEnc, &iBestMode_c, iLambda,
- pDstChma, pDecCr + 128, pEncCr);
- tmpb = WelsIntraChroma8x8Combined3Satd_sse41 (pDecCb + 128, iLineSizeDec, pEncCb, iLineSizeEnc, &iBestMode_a, iLambda,
- pDstChma, pDecCr + 128, pEncCr);
- ASSERT_EQ (tmpa, tmpb);
- ASSERT_EQ (iBestMode_c, iBestMode_a);
- cMemoryAlign.WelsFree (pDecCb, "pDecCb");
- cMemoryAlign.WelsFree (pEncCb, "pEncCb");
- cMemoryAlign.WelsFree (pDecCr, "pDecCr");
- cMemoryAlign.WelsFree (pEncCr, "pEncCr");
- cMemoryAlign.WelsFree (pDstChma, "pDstChma");
-}
+#ifdef X86_ASM
+GENERATE_Intra16x16_UT(WelsIntra16x16Combined3Sad_ssse3, WelsSampleSadIntra16x16Combined3_c, 1, WELS_CPU_SSSE3)
+GENERATE_Intra16x16_UT(WelsIntra16x16Combined3Satd_sse41, WelsSampleSatdIntra16x16Combined3_c, 1, WELS_CPU_SSE41)
+GENERATE_Intra8x8_UT(WelsIntraChroma8x8Combined3Satd_sse41, WelsSampleSatdIntra8x8Combined3_c, 1, WELS_CPU_SSE41)
+GENERATE_Intra4x4_UT(WelsSampleSatdThree4x4_sse2, 1, WELS_CPU_SSE2)
#endif
+
+#ifdef HAVE_NEON
+GENERATE_Intra16x16_UT(WelsIntra16x16Combined3Sad_neon, WelsSampleSadIntra16x16Combined3_c, 1, WELS_CPU_NEON)
+GENERATE_Intra16x16_UT(WelsIntra16x16Combined3Satd_neon, WelsSampleSatdIntra16x16Combined3_c, 1, WELS_CPU_NEON)
+GENERATE_Intra8x8_UT(WelsIntra8x8Combined3Satd_neon, WelsSampleSatdIntra8x8Combined3_c, 1, WELS_CPU_NEON)
+GENERATE_Intra8x8_UT(WelsIntra8x8Combined3Sad_neon, WelsSampleSadIntra8x8Combined3_c, 1, WELS_CPU_NEON)
+GENERATE_Intra4x4_UT(WelsIntra4x4Combined3Satd_neon, 1, WELS_CPU_NEON)
+#endif
+
+#ifdef HAVE_NEON_AARCH64
+GENERATE_Intra16x16_UT(WelsIntra16x16Combined3Sad_AArch64_neon, WelsSampleSadIntra16x16Combined3_c, 1, WELS_CPU_NEON)
+GENERATE_Intra16x16_UT(WelsIntra16x16Combined3Satd_AArch64_neon, WelsSampleSatdIntra16x16Combined3_c, 1, WELS_CPU_NEON)
+GENERATE_Intra8x8_UT(WelsIntra8x8Combined3Satd_AArch64_neon, WelsSampleSatdIntra8x8Combined3_c, 1, WELS_CPU_NEON)
+GENERATE_Intra8x8_UT(WelsIntra8x8Combined3Sad_AArch64_neon, WelsSampleSadIntra8x8Combined3_c, 1, WELS_CPU_NEON)
+GENERATE_Intra4x4_UT(WelsIntra4x4Combined3Satd_AArch64_neon, 1, WELS_CPU_NEON)
+#endif
+
#define ASSERT_MEMORY_FAIL2X(A, B) \
if (NULL == B) { \pMemAlign->WelsFree(A, "Sad_SrcA");\
@@ -457,7 +444,7 @@
EXPECT_EQ (m_pSad[0] + m_pSad[1] + m_pSad[2] + m_pSad[3], iSumSad);
}
-#ifdef X86_ASM
+
class SadSatdAssemblyFuncTest : public testing::Test {public:
virtual void SetUp() {@@ -489,327 +476,159 @@
CMemoryAlign* pMemAlign;
};
-TEST_F (SadSatdAssemblyFuncTest, WelsSampleSad4x4_mmx) {- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_MMXEXT))
- return;
-
- for (int i = 0; i < (m_iStrideA << 2); i++)
- m_pPixSrcA[i] = rand() % 256;
- for (int i = 0; i < (m_iStrideB << 2); i++)
- m_pPixSrcB[i] = rand() % 256;
-
- EXPECT_EQ (WelsSampleSad4x4_c (m_pPixSrcA, m_iStrideA, m_pPixSrcB, m_iStrideB), WelsSampleSad4x4_mmx (m_pPixSrcA,
- m_iStrideA, m_pPixSrcB, m_iStrideB));
+#define GENERATE_Sad4x4_UT(func, ref, CPUFLAGS) \
+TEST_F (SadSatdAssemblyFuncTest, func) { \+ if (0 == (m_uiCpuFeatureFlag & CPUFLAGS)) \
+ return; \
+ for (int i = 0; i < (m_iStrideA << 2); i++) \
+ m_pPixSrcA[i] = rand() % 256; \
+ for (int i = 0; i < (m_iStrideB << 2); i++) \
+ m_pPixSrcB[i] = rand() % 256; \
+ EXPECT_EQ (ref (m_pPixSrcA, m_iStrideA, m_pPixSrcB, m_iStrideB), func (m_pPixSrcA, \
+ m_iStrideA, m_pPixSrcB, m_iStrideB)); \
}
-
-TEST_F (SadSatdAssemblyFuncTest, WelsSampleSad8x8_sse21) {- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE2))
- return;
-
- for (int i = 0; i < (m_iStrideA << 3); i++)
- m_pPixSrcA[i] = rand() % 256;
- for (int i = 0; i < (m_iStrideB << 3); i++)
- m_pPixSrcB[i] = rand() % 256;
-
- EXPECT_EQ (WelsSampleSad8x8_c (m_pPixSrcA, m_iStrideA, m_pPixSrcB, m_iStrideB), WelsSampleSad8x8_sse21 (m_pPixSrcA,
- m_iStrideA, m_pPixSrcB, m_iStrideB));
+#define GENERATE_Sad8x8_UT(func, ref, CPUFLAGS) \
+TEST_F (SadSatdAssemblyFuncTest, func) { \+ if (0 == (m_uiCpuFeatureFlag & CPUFLAGS)) \
+ return; \
+ for (int i = 0; i < (m_iStrideA << 3); i++) \
+ m_pPixSrcA[i] = rand() % 256; \
+ for (int i = 0; i < (m_iStrideB << 3); i++) \
+ m_pPixSrcB[i] = rand() % 256; \
+ EXPECT_EQ (ref (m_pPixSrcA, m_iStrideA, m_pPixSrcB, m_iStrideB), func (m_pPixSrcA, \
+ m_iStrideA, m_pPixSrcB, m_iStrideB)); \
}
-TEST_F (SadSatdAssemblyFuncTest, WelsSampleSad8x16_sse2) {- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE2))
- return;
- for (int i = 0; i < (m_iStrideA << 4); i++)
- m_pPixSrcA[i] = rand() % 256;
- for (int i = 0; i < (m_iStrideB << 4); i++)
- m_pPixSrcB[i] = rand() % 256;
-
- EXPECT_EQ (WelsSampleSad8x16_c (m_pPixSrcA, m_iStrideA, m_pPixSrcB, m_iStrideB), WelsSampleSad8x16_sse2 (m_pPixSrcA,
- m_iStrideA, m_pPixSrcB, m_iStrideB));
+#define GENERATE_Sad8x16_UT(func, ref, CPUFLAGS) \
+TEST_F (SadSatdAssemblyFuncTest, func) { \+ if (0 == (m_uiCpuFeatureFlag & CPUFLAGS)) \
+ return; \
+ for (int i = 0; i < (m_iStrideA << 4); i++) \
+ m_pPixSrcA[i] = rand() % 256; \
+ for (int i = 0; i < (m_iStrideB << 4); i++) \
+ m_pPixSrcB[i] = rand() % 256; \
+ EXPECT_EQ (ref (m_pPixSrcA, m_iStrideA, m_pPixSrcB, m_iStrideB), func (m_pPixSrcA, \
+ m_iStrideA, m_pPixSrcB, m_iStrideB)); \
}
-TEST_F (SadSatdAssemblyFuncTest, WelsSampleSad16x8_sse2) {- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE2))
- return;
- for (int i = 0; i < (m_iStrideA << 3); i++)
- m_pPixSrcA[i] = rand() % 256;
- for (int i = 0; i < (m_iStrideB << 3); i++)
- m_pPixSrcB[i] = rand() % 256;
-
- EXPECT_EQ (WelsSampleSad16x8_c (m_pPixSrcA, m_iStrideA, m_pPixSrcB, m_iStrideB), WelsSampleSad16x8_sse2 (m_pPixSrcA,
- m_iStrideA, m_pPixSrcB, m_iStrideB));
+#define GENERATE_Sad16x8_UT(func, ref, CPUFLAGS) \
+TEST_F (SadSatdAssemblyFuncTest, func) { \+ if (0 == (m_uiCpuFeatureFlag & CPUFLAGS)) \
+ return; \
+ for (int i = 0; i < (m_iStrideA << 3); i++) \
+ m_pPixSrcA[i] = rand() % 256; \
+ for (int i = 0; i < (m_iStrideB << 3); i++) \
+ m_pPixSrcB[i] = rand() % 256; \
+ EXPECT_EQ (ref (m_pPixSrcA, m_iStrideA, m_pPixSrcB, m_iStrideB), func (m_pPixSrcA, \
+ m_iStrideA, m_pPixSrcB, m_iStrideB)); \
}
-TEST_F (SadSatdAssemblyFuncTest, WelsSampleSad16x16_sse2) {- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE2))
- return;
- for (int i = 0; i < (m_iStrideA << 4); i++)
- m_pPixSrcA[i] = rand() % 256;
- for (int i = 0; i < (m_iStrideB << 4); i++)
- m_pPixSrcB[i] = rand() % 256;
-
- EXPECT_EQ (WelsSampleSad16x16_c (m_pPixSrcA, m_iStrideA, m_pPixSrcB, m_iStrideB), WelsSampleSad16x16_sse2 (m_pPixSrcA,
- m_iStrideA, m_pPixSrcB, m_iStrideB));
+#define GENERATE_Sad16x16_UT(func, ref, CPUFLAGS) \
+TEST_F (SadSatdAssemblyFuncTest, func) { \+ if (0 == (m_uiCpuFeatureFlag & CPUFLAGS)) \
+ return; \
+ for (int i = 0; i < (m_iStrideA << 4); i++) \
+ m_pPixSrcA[i] = rand() % 256; \
+ for (int i = 0; i < (m_iStrideB << 4); i++) \
+ m_pPixSrcB[i] = rand() % 256; \
+ EXPECT_EQ (ref (m_pPixSrcA, m_iStrideA, m_pPixSrcB, m_iStrideB), func (m_pPixSrcA, \
+ m_iStrideA, m_pPixSrcB, m_iStrideB)); \
}
-TEST_F (SadSatdAssemblyFuncTest, WelsSampleSatd4x4_sse2) {- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE2))
- return;
- for (int i = 0; i < (m_iStrideA << 2); i++)
- m_pPixSrcA[i] = rand() % 256;
- for (int i = 0; i < (m_iStrideB << 2); i++)
- m_pPixSrcB[i] = rand() % 256;
+#ifdef X86_ASM
+GENERATE_Sad4x4_UT(WelsSampleSad4x4_mmx, WelsSampleSad4x4_c, WELS_CPU_MMXEXT)
+GENERATE_Sad8x8_UT(WelsSampleSad8x8_sse21, WelsSampleSad8x8_c, WELS_CPU_SSE2)
+GENERATE_Sad8x16_UT(WelsSampleSad8x16_sse2, WelsSampleSad8x16_c, WELS_CPU_SSE2)
+GENERATE_Sad16x8_UT(WelsSampleSad16x8_sse2, WelsSampleSad16x8_c, WELS_CPU_SSE2)
+GENERATE_Sad16x16_UT(WelsSampleSad16x16_sse2, WelsSampleSad16x16_c, WELS_CPU_SSE2)
- EXPECT_EQ (WelsSampleSatd4x4_c (m_pPixSrcA, m_iStrideA, m_pPixSrcB, m_iStrideB), WelsSampleSatd4x4_sse2 (m_pPixSrcA,
- m_iStrideA, m_pPixSrcB, m_iStrideB));
-}
+GENERATE_Sad4x4_UT(WelsSampleSatd4x4_sse2, WelsSampleSatd4x4_c, WELS_CPU_SSE2)
+GENERATE_Sad8x8_UT(WelsSampleSatd8x8_sse2, WelsSampleSatd8x8_c, WELS_CPU_SSE2)
+GENERATE_Sad8x16_UT(WelsSampleSatd8x16_sse2, WelsSampleSatd8x16_c, WELS_CPU_SSE2)
+GENERATE_Sad16x8_UT(WelsSampleSatd16x8_sse2, WelsSampleSatd16x8_c, WELS_CPU_SSE2)
+GENERATE_Sad16x16_UT(WelsSampleSatd16x16_sse2, WelsSampleSatd16x16_c, WELS_CPU_SSE2)
-TEST_F (SadSatdAssemblyFuncTest, WelsSampleSatd8x8_sse2) {- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE2))
- return;
- for (int i = 0; i < (m_iStrideA << 3); i++)
- m_pPixSrcA[i] = rand() % 256;
- for (int i = 0; i < (m_iStrideB << 3); i++)
- m_pPixSrcB[i] = rand() % 256;
+GENERATE_Sad4x4_UT(WelsSampleSatd4x4_sse41, WelsSampleSatd4x4_c, WELS_CPU_SSE41)
+GENERATE_Sad8x8_UT(WelsSampleSatd8x8_sse41, WelsSampleSatd8x8_c, WELS_CPU_SSE41)
+GENERATE_Sad8x16_UT(WelsSampleSatd8x16_sse41, WelsSampleSatd8x16_c, WELS_CPU_SSE41)
+GENERATE_Sad16x8_UT(WelsSampleSatd16x8_sse41, WelsSampleSatd16x8_c, WELS_CPU_SSE41)
+GENERATE_Sad16x16_UT(WelsSampleSatd16x16_sse41, WelsSampleSatd16x16_c, WELS_CPU_SSE41)
+#endif
- EXPECT_EQ (WelsSampleSatd8x8_c (m_pPixSrcA, m_iStrideA, m_pPixSrcB, m_iStrideB), WelsSampleSatd8x8_sse2 (m_pPixSrcA,
- m_iStrideA, m_pPixSrcB, m_iStrideB));
-}
+#ifdef HAVE_NEON
+GENERATE_Sad4x4_UT(WelsSampleSad4x4_neon, WelsSampleSad4x4_c, WELS_CPU_NEON)
+GENERATE_Sad8x8_UT(WelsSampleSad8x8_neon, WelsSampleSad8x8_c, WELS_CPU_NEON)
+GENERATE_Sad8x16_UT(WelsSampleSad8x16_neon, WelsSampleSad8x16_c, WELS_CPU_NEON)
+GENERATE_Sad16x8_UT(WelsSampleSad16x8_neon, WelsSampleSad16x8_c, WELS_CPU_NEON)
+GENERATE_Sad16x16_UT(WelsSampleSad16x16_neon, WelsSampleSad16x16_c, WELS_CPU_NEON)
-TEST_F (SadSatdAssemblyFuncTest, WelsSampleSatd8x16_sse2) {- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE2))
- return;
- for (int i = 0; i < (m_iStrideA << 4); i++)
- m_pPixSrcA[i] = rand() % 256;
- for (int i = 0; i < (m_iStrideB << 4); i++)
- m_pPixSrcB[i] = rand() % 256;
+GENERATE_Sad4x4_UT(WelsSampleSatd4x4_neon, WelsSampleSatd4x4_c, WELS_CPU_NEON)
+GENERATE_Sad8x8_UT(WelsSampleSatd8x8_neon, WelsSampleSatd8x8_c, WELS_CPU_NEON)
+GENERATE_Sad8x16_UT(WelsSampleSatd8x16_neon, WelsSampleSatd8x16_c, WELS_CPU_NEON)
+GENERATE_Sad16x8_UT(WelsSampleSatd16x8_neon, WelsSampleSatd16x8_c, WELS_CPU_NEON)
+GENERATE_Sad16x16_UT(WelsSampleSatd16x16_neon, WelsSampleSatd16x16_c, WELS_CPU_NEON)
+#endif
- EXPECT_EQ (WelsSampleSatd8x16_c (m_pPixSrcA, m_iStrideA, m_pPixSrcB, m_iStrideB), WelsSampleSatd8x16_sse2 (m_pPixSrcA,
- m_iStrideA, m_pPixSrcB, m_iStrideB));
-}
+#ifdef HAVE_NEON_AARCH64
+GENERATE_Sad4x4_UT(WelsSampleSad4x4_AArch64_neon, WelsSampleSad4x4_c, WELS_CPU_NEON)
+GENERATE_Sad8x8_UT(WelsSampleSad8x8_AArch64_neon, WelsSampleSad8x8_c, WELS_CPU_NEON)
+GENERATE_Sad8x16_UT(WelsSampleSad8x16_AArch64_neon, WelsSampleSad8x16_c, WELS_CPU_NEON)
+GENERATE_Sad16x8_UT(WelsSampleSad16x8_AArch64_neon, WelsSampleSad16x8_c, WELS_CPU_NEON)
+GENERATE_Sad16x16_UT(WelsSampleSad16x16_AArch64_neon, WelsSampleSad16x16_c, WELS_CPU_NEON)
-TEST_F (SadSatdAssemblyFuncTest, WelsSampleSatd16x8_sse2) {- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE2))
- return;
- for (int i = 0; i < (m_iStrideA << 3); i++)
- m_pPixSrcA[i] = rand() % 256;
- for (int i = 0; i < (m_iStrideB << 3); i++)
- m_pPixSrcB[i] = rand() % 256;
+GENERATE_Sad4x4_UT(WelsSampleSatd4x4_AArch64_neon, WelsSampleSatd4x4_c, WELS_CPU_NEON)
+GENERATE_Sad8x8_UT(WelsSampleSatd8x8_AArch64_neon, WelsSampleSatd8x8_c, WELS_CPU_NEON)
+GENERATE_Sad8x16_UT(WelsSampleSatd8x16_AArch64_neon, WelsSampleSatd8x16_c, WELS_CPU_NEON)
+GENERATE_Sad16x8_UT(WelsSampleSatd16x8_AArch64_neon, WelsSampleSatd16x8_c, WELS_CPU_NEON)
+GENERATE_Sad16x16_UT(WelsSampleSatd16x16_AArch64_neon, WelsSampleSatd16x16_c, WELS_CPU_NEON)
+#endif
- EXPECT_EQ (WelsSampleSatd16x8_c (m_pPixSrcA, m_iStrideA, m_pPixSrcB, m_iStrideB), WelsSampleSatd16x8_sse2 (m_pPixSrcA,
- m_iStrideA, m_pPixSrcB, m_iStrideB));
+#define GENERATE_SadFour_UT(func, CPUFLAGS, width, height) \
+TEST_F (SadSatdAssemblyFuncTest, func) { \+ if (0 == (m_uiCpuFeatureFlag & CPUFLAGS)) \
+ return; \
+ for (int i = 0; i < (m_iStrideA << 5); i++) \
+ m_pPixSrcA[i] = rand() % 256; \
+ for (int i = 0; i < (m_iStrideB << 5); i++) \
+ m_pPixSrcB[i] = rand() % 256; \
+ uint8_t* pPixA = m_pPixSrcA; \
+ uint8_t* pPixB = m_pPixSrcB + m_iStrideB; \
+ int32_t iSumSad = 0; \
+ for (int i = 0; i < height; i++) { \+ for (int j = 0; j < width; j++) { \+ iSumSad += abs (pPixA[j] - pPixB[j - 1]); \
+ iSumSad += abs (pPixA[j] - pPixB[j + 1]); \
+ iSumSad += abs (pPixA[j] - pPixB[j - m_iStrideB]); \
+ iSumSad += abs (pPixA[j] - pPixB[j + m_iStrideB]); \
+ } \
+ pPixA += m_iStrideA; \
+ pPixB += m_iStrideB; \
+ } \
+ func (m_pPixSrcA, m_iStrideA, m_pPixSrcB + m_iStrideB, m_iStrideB, m_pSad); \
+ EXPECT_EQ (m_pSad[0] + m_pSad[1] + m_pSad[2] + m_pSad[3], iSumSad); \
}
-TEST_F (SadSatdAssemblyFuncTest, WelsSampleSatd16x16_sse2) {- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE2))
- return;
- for (int i = 0; i < (m_iStrideA << 4); i++)
- m_pPixSrcA[i] = rand() % 256;
- for (int i = 0; i < (m_iStrideB << 4); i++)
- m_pPixSrcB[i] = rand() % 256;
+#ifdef X86_ASM
+GENERATE_SadFour_UT(WelsSampleSadFour4x4_sse2, WELS_CPU_SSE2, 4, 4)
+GENERATE_SadFour_UT(WelsSampleSadFour8x8_sse2, WELS_CPU_SSE2, 8, 8)
+GENERATE_SadFour_UT(WelsSampleSadFour8x16_sse2, WELS_CPU_SSE2, 8, 16)
+GENERATE_SadFour_UT(WelsSampleSadFour16x8_sse2, WELS_CPU_SSE2, 16, 8)
+GENERATE_SadFour_UT(WelsSampleSadFour16x16_sse2, WELS_CPU_SSE2, 16, 16)
+#endif
- EXPECT_EQ (WelsSampleSatd16x16_c (m_pPixSrcA, m_iStrideA, m_pPixSrcB, m_iStrideB), WelsSampleSatd16x16_sse2 (m_pPixSrcA,
- m_iStrideA, m_pPixSrcB, m_iStrideB));
-}
-
-TEST_F (SadSatdAssemblyFuncTest, WelsSampleSatd4x4_sse41) {- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE41))
- return;
- for (int i = 0; i < (m_iStrideA << 2); i++)
- m_pPixSrcA[i] = rand() % 256;
- for (int i = 0; i < (m_iStrideB << 2); i++)
- m_pPixSrcB[i] = rand() % 256;
-
- EXPECT_EQ (WelsSampleSatd4x4_c (m_pPixSrcA, m_iStrideA, m_pPixSrcB, m_iStrideB), WelsSampleSatd4x4_sse41 (m_pPixSrcA,
- m_iStrideA, m_pPixSrcB, m_iStrideB));
-}
-
-TEST_F (SadSatdAssemblyFuncTest, WelsSampleSatd8x8_sse41) {- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE41))
- return;
- for (int i = 0; i < (m_iStrideA << 3); i++)
- m_pPixSrcA[i] = rand() % 256;
- for (int i = 0; i < (m_iStrideB << 3); i++)
- m_pPixSrcB[i] = rand() % 256;
-
- EXPECT_EQ (WelsSampleSatd8x8_c (m_pPixSrcA, m_iStrideA, m_pPixSrcB, m_iStrideB), WelsSampleSatd8x8_sse41 (m_pPixSrcA,
- m_iStrideA, m_pPixSrcB, m_iStrideB));
-}
-
-TEST_F (SadSatdAssemblyFuncTest, WelsSampleSatd8x16_sse41) {- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE41))
- return;
- for (int i = 0; i < (m_iStrideA << 4); i++)
- m_pPixSrcA[i] = rand() % 256;
- for (int i = 0; i < (m_iStrideB << 4); i++)
- m_pPixSrcB[i] = rand() % 256;
-
- EXPECT_EQ (WelsSampleSatd8x16_c (m_pPixSrcA, m_iStrideA, m_pPixSrcB, m_iStrideB), WelsSampleSatd8x16_sse41 (m_pPixSrcA,
- m_iStrideA, m_pPixSrcB, m_iStrideB));
-}
-
-TEST_F (SadSatdAssemblyFuncTest, WelsSampleSatd16x8_sse41) {- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE41))
- return;
- for (int i = 0; i < (m_iStrideA << 3); i++)
- m_pPixSrcA[i] = rand() % 256;
- for (int i = 0; i < (m_iStrideB << 3); i++)
- m_pPixSrcB[i] = rand() % 256;
-
- EXPECT_EQ (WelsSampleSatd16x8_c (m_pPixSrcA, m_iStrideA, m_pPixSrcB, m_iStrideB), WelsSampleSatd16x8_sse41 (m_pPixSrcA,
- m_iStrideA, m_pPixSrcB, m_iStrideB));
-}
-
-TEST_F (SadSatdAssemblyFuncTest, WelsSampleSatd16x16_sse41) {- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE41))
- return;
- for (int i = 0; i < (m_iStrideA << 4); i++)
- m_pPixSrcA[i] = rand() % 256;
- for (int i = 0; i < (m_iStrideB << 4); i++)
- m_pPixSrcB[i] = rand() % 256;
-
- EXPECT_EQ (WelsSampleSatd16x16_c (m_pPixSrcA, m_iStrideA, m_pPixSrcB, m_iStrideB),
- WelsSampleSatd16x16_sse41 (m_pPixSrcA, m_iStrideA, m_pPixSrcB, m_iStrideB));
-}
-
-TEST_F (SadSatdAssemblyFuncTest, WelsSampleSadFour16x16_sse2) {- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE2))
- return;
-
- for (int i = 0; i < (m_iStrideA << 5); i++)
- m_pPixSrcA[i] = rand() % 256;
- for (int i = 0; i < (m_iStrideB << 5); i++)
- m_pPixSrcB[i] = rand() % 256;
- uint8_t* pPixA = m_pPixSrcA;
- uint8_t* pPixB = m_pPixSrcB + m_iStrideB;
-
- int32_t iSumSad = 0;
- for (int i = 0; i < 16; i++) {- for (int j = 0; j < 16; j++) {- iSumSad += abs (pPixA[j] - pPixB[j - 1]);
- iSumSad += abs (pPixA[j] - pPixB[j + 1]);
- iSumSad += abs (pPixA[j] - pPixB[j - m_iStrideB]);
- iSumSad += abs (pPixA[j] - pPixB[j + m_iStrideB]);
- }
-
- pPixA += m_iStrideA;
- pPixB += m_iStrideB;
- }
-
- WelsSampleSadFour16x16_sse2 (m_pPixSrcA, m_iStrideA, m_pPixSrcB + m_iStrideB, m_iStrideB, m_pSad);
- EXPECT_EQ (m_pSad[0] + m_pSad[1] + m_pSad[2] + m_pSad[3], iSumSad);
-}
-
-TEST_F (SadSatdAssemblyFuncTest, WelsSampleSadFour16x8_sse2) {- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE2))
- return;
-
- for (int i = 0; i < (m_iStrideA << 5); i++)
- m_pPixSrcA[i] = rand() % 256;
- for (int i = 0; i < (m_iStrideB << 5); i++)
- m_pPixSrcB[i] = rand() % 256;
- uint8_t* pPixA = m_pPixSrcA;
- uint8_t* pPixB = m_pPixSrcB + m_iStrideB;
-
- int32_t iSumSad = 0;
- for (int i = 0; i < 8; i++) {- for (int j = 0; j < 16; j++) {- iSumSad += abs (pPixA[j] - pPixB[j - 1]);
- iSumSad += abs (pPixA[j] - pPixB[j + 1]);
- iSumSad += abs (pPixA[j] - pPixB[j - m_iStrideB]);
- iSumSad += abs (pPixA[j] - pPixB[j + m_iStrideB]);
- }
-
- pPixA += m_iStrideA;
- pPixB += m_iStrideB;
- }
-
- WelsSampleSadFour16x8_sse2 (m_pPixSrcA, m_iStrideA, m_pPixSrcB + m_iStrideB, m_iStrideB, m_pSad);
- EXPECT_EQ (m_pSad[0] + m_pSad[1] + m_pSad[2] + m_pSad[3], iSumSad);
-}
-
-TEST_F (SadSatdAssemblyFuncTest, WelsSampleSadFour8x16_sse2) {- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE2))
- return;
-
- for (int i = 0; i < (m_iStrideA << 5); i++)
- m_pPixSrcA[i] = rand() % 256;
- for (int i = 0; i < (m_iStrideB << 5); i++)
- m_pPixSrcB[i] = rand() % 256;
- uint8_t* pPixA = m_pPixSrcA;
- uint8_t* pPixB = m_pPixSrcB + m_iStrideB;
-
- int32_t iSumSad = 0;
- for (int i = 0; i < 16; i++) {- for (int j = 0; j < 8; j++) {- iSumSad += abs (pPixA[j] - pPixB[j - 1]);
- iSumSad += abs (pPixA[j] - pPixB[j + 1]);
- iSumSad += abs (pPixA[j] - pPixB[j - m_iStrideB]);
- iSumSad += abs (pPixA[j] - pPixB[j + m_iStrideB]);
- }
-
- pPixA += m_iStrideA;
- pPixB += m_iStrideB;
- }
-
- WelsSampleSadFour8x16_sse2 (m_pPixSrcA, m_iStrideA, m_pPixSrcB + m_iStrideB, m_iStrideB, m_pSad);
- EXPECT_EQ (m_pSad[0] + m_pSad[1] + m_pSad[2] + m_pSad[3], iSumSad);
-}
-
-TEST_F (SadSatdAssemblyFuncTest, WelsSampleSadFour8x8_sse2) {- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE2))
- return;
-
- for (int i = 0; i < (m_iStrideA << 4); i++)
- m_pPixSrcA[i] = rand() % 256;
- for (int i = 0; i < (m_iStrideB << 4); i++)
- m_pPixSrcB[i] = rand() % 256;
- uint8_t* pPixA = m_pPixSrcA;
- uint8_t* pPixB = m_pPixSrcB + m_iStrideB;
-
- int32_t iSumSad = 0;
- for (int i = 0; i < 8; i++) {- for (int j = 0; j < 8; j++) {- iSumSad += abs (pPixA[j] - pPixB[j - 1]);
- iSumSad += abs (pPixA[j] - pPixB[j + 1]);
- iSumSad += abs (pPixA[j] - pPixB[j - m_iStrideB]);
- iSumSad += abs (pPixA[j] - pPixB[j + m_iStrideB]);
- }
-
- pPixA += m_iStrideA;
- pPixB += m_iStrideB;
- }
-
- WelsSampleSadFour8x8_sse2 (m_pPixSrcA, m_iStrideA, m_pPixSrcB + m_iStrideB, m_iStrideB, m_pSad);
- EXPECT_EQ (m_pSad[0] + m_pSad[1] + m_pSad[2] + m_pSad[3], iSumSad);
-}
-
-
-TEST_F (SadSatdAssemblyFuncTest, WelsSampleSadFour4x4_sse2) {- if (0 == (m_uiCpuFeatureFlag & WELS_CPU_SSE2))
- return;
-
- for (int i = 0; i < (m_iStrideA << 3); i++)
- m_pPixSrcA[i] = rand() % 256;
- for (int i = 0; i < (m_iStrideB << 3); i++)
- m_pPixSrcB[i] = rand() % 256;
- uint8_t* pPixA = m_pPixSrcA;
- uint8_t* pPixB = m_pPixSrcB + m_iStrideB;
-
- int32_t iSumSad = 0;
- for (int i = 0; i < 4; i++) {- for (int j = 0; j < 4; j++) {- iSumSad += abs (pPixA[j] - pPixB[j - 1]);
- iSumSad += abs (pPixA[j] - pPixB[j + 1]);
- iSumSad += abs (pPixA[j] - pPixB[j - m_iStrideB]);
- iSumSad += abs (pPixA[j] - pPixB[j + m_iStrideB]);
- }
-
- pPixA += m_iStrideA;
- pPixB += m_iStrideB;
- }
-
- WelsSampleSadFour4x4_sse2 (m_pPixSrcA, m_iStrideA, m_pPixSrcB + m_iStrideB, m_iStrideB, m_pSad);
- EXPECT_EQ (m_pSad[0] + m_pSad[1] + m_pSad[2] + m_pSad[3], iSumSad);
-}
+#ifdef HAVE_NEON
+GENERATE_SadFour_UT(WelsSampleSadFour4x4_neon, WELS_CPU_NEON, 4, 4)
+GENERATE_SadFour_UT(WelsSampleSadFour8x8_neon, WELS_CPU_NEON, 8, 8)
+GENERATE_SadFour_UT(WelsSampleSadFour8x16_neon, WELS_CPU_NEON, 8, 16)
+GENERATE_SadFour_UT(WelsSampleSadFour16x8_neon, WELS_CPU_NEON, 16, 8)
+GENERATE_SadFour_UT(WelsSampleSadFour16x16_neon, WELS_CPU_NEON, 16, 16)
#endif
+
+#ifdef HAVE_NEON_AARCH64
+GENERATE_SadFour_UT(WelsSampleSadFour4x4_AArch64_neon, WELS_CPU_NEON, 4, 4)
+GENERATE_SadFour_UT(WelsSampleSadFour8x8_AArch64_neon, WELS_CPU_NEON, 8, 8)
+GENERATE_SadFour_UT(WelsSampleSadFour8x16_AArch64_neon, WELS_CPU_NEON, 8, 16)
+GENERATE_SadFour_UT(WelsSampleSadFour16x8_AArch64_neon, WELS_CPU_NEON, 16, 8)
+GENERATE_SadFour_UT(WelsSampleSadFour16x16_AArch64_neon, WELS_CPU_NEON, 16, 16)
+#endif
\ No newline at end of file