ref: dcd388006ed5e1aef891c71098d804891790302e
parent: 9993198a3c6ffac6aef85b1ae2946f12ff45d541
parent: 801b664201c573b5fa8c7e52a1386cd8cfe72dfd
author: Licai Guo <[email protected]>
date: Fri Apr 4 09:27:33 EDT 2014
Merge pull request #629 from JuannyWang/mbAux [Encoder UT] add encoder mbAux unittest
--- /dev/null
+++ b/test/encoder/EncUT_EncoderMbAux.cpp
@@ -1,0 +1,494 @@
+#include<gtest/gtest.h>
+#include<stdlib.h>
+#include<time.h>
+#include "ls_defines.h"
+#include "encode_mb_aux.h"
+#include "wels_common_basis.h"
+
+using namespace WelsSVCEnc;
+
+#define ALLOC_MEMORY(type, name, num) type* name = (type*)cMemoryAlign.WelsMalloc(num*sizeof(type), #name);
+#define FREE_MEMORY(name) cMemoryAlign.WelsFree(name, #name);
+TEST(EncodeMbAuxTest, TestScan_4x4_ac_c) {+ CMemoryAlign cMemoryAlign(0);
+ ALLOC_MEMORY(int16_t, iLevel, 16);
+ ALLOC_MEMORY(int16_t, iDctA, 16);
+ ALLOC_MEMORY(int16_t, iDctB, 16);
+ srand((unsigned int)time(NULL));
+ for(int i=0;i<16;i++) {+ iDctA[i]=rand()%256+1;
+ iDctB[i]=iDctA[i];
+ }
+ WelsScan4x4Ac_c( iLevel, iDctA );
+ EXPECT_EQ(iLevel[0],iDctB[1]);
+ EXPECT_EQ(iLevel[1],iDctB[4]);
+ EXPECT_EQ(iLevel[2],iDctB[8]);
+ EXPECT_EQ(iLevel[3],iDctB[5]);
+ EXPECT_EQ(iLevel[4],iDctB[2]);
+ EXPECT_EQ(iLevel[5],iDctB[3]);
+ EXPECT_EQ(iLevel[6],iDctB[6]);
+ EXPECT_EQ(iLevel[7],iDctB[9]);
+ EXPECT_EQ(iLevel[8],iDctB[12]);
+ EXPECT_EQ(iLevel[9],iDctB[13]);
+ EXPECT_EQ(iLevel[10],iDctB[10]);
+ EXPECT_EQ(iLevel[11],iDctB[7]);
+ EXPECT_EQ(iLevel[12],iDctB[11]);
+ EXPECT_EQ(iLevel[13],iDctB[14]);
+ EXPECT_EQ(iLevel[14],iDctB[15]);
+ EXPECT_EQ(iLevel[15],0);
+ FREE_MEMORY(iLevel);
+ FREE_MEMORY(iDctA);
+ FREE_MEMORY(iDctB);
+}
+
+#ifdef X86_ASM
+TEST(EncodeMbAuxTest, TestScan_4x4_ac_sse2) {+ CMemoryAlign cMemoryAlign(0);
+ ALLOC_MEMORY(int16_t, iLevelA, 16);
+ ALLOC_MEMORY(int16_t, iLevelB, 16);
+ ALLOC_MEMORY(int16_t, iDct, 16);
+ srand((unsigned int)time(NULL));
+ for(int i=0;i<16;i++) {+ iDct[i]=rand()%256+1;
+ }
+ WelsScan4x4Ac_c( iLevelA, iDct );
+ WelsScan4x4Ac_sse2( iLevelB, iDct );
+ for(int j=0; j<16;j++)
+ EXPECT_EQ(iLevelA[j], iLevelB[j]);
+ FREE_MEMORY(iLevelA);
+ FREE_MEMORY(iLevelB);
+ FREE_MEMORY(iDct);
+}
+TEST(EncodeMbAuxTest, WelsScan4x4DcAc_sse2) {+ CMemoryAlign cMemoryAlign(0);
+ ALLOC_MEMORY(int16_t, iLevelA, 32);
+ ALLOC_MEMORY(int16_t, iLevelB, 32);
+ ALLOC_MEMORY(int16_t, iDct, 32);
+ srand((unsigned int)time(NULL));
+ for(int i = 0; i < 32; i++)
+ iDct[i] = (rand() & 32767) - 16384;
+ WelsScan4x4DcAc_sse2(iLevelA, iDct);
+ WelsScan4x4DcAc_c(iLevelB, iDct);
+ for(int i = 0; i < 16; i++)
+ EXPECT_EQ(iLevelA[i], iLevelB[i]);
+ FREE_MEMORY(iLevelA);
+ FREE_MEMORY(iLevelB);
+ FREE_MEMORY(iDct);
+}
+#endif
+TEST(EncodeMbAuxTest, TestScan_4x4_dcc) {+ CMemoryAlign cMemoryAlign(0);
+ ALLOC_MEMORY(int16_t, iLevel, 16);
+ ALLOC_MEMORY(int16_t, iDctA, 16);
+ ALLOC_MEMORY(int16_t, iDctB, 16);
+ srand((unsigned int)time(NULL));
+ for(int i=0;i<16;i++)
+ iDctA[i] = iDctB[i] = rand()%256+1;
+ WelsScan4x4Dc( iLevel, iDctA );
+ EXPECT_EQ(iLevel[0],iDctB[0]);
+ EXPECT_EQ(iLevel[1],iDctB[1]);
+ EXPECT_EQ(iLevel[2],iDctB[4]);
+ EXPECT_EQ(iLevel[3],iDctB[8]);
+ EXPECT_EQ(iLevel[4],iDctB[5]);
+ EXPECT_EQ(iLevel[5],iDctB[2]);
+ EXPECT_EQ(iLevel[6],iDctB[3]);
+ EXPECT_EQ(iLevel[7],iDctB[6]);
+ EXPECT_EQ(iLevel[8],iDctB[9]);
+ EXPECT_EQ(iLevel[9],iDctB[12]);
+ EXPECT_EQ(iLevel[10],iDctB[13]);
+ EXPECT_EQ(iLevel[11],iDctB[10]);
+ EXPECT_EQ(iLevel[12],iDctB[7]);
+ EXPECT_EQ(iLevel[13],iDctB[11]);
+ EXPECT_EQ(iLevel[14],iDctB[14]);
+ EXPECT_EQ(iLevel[15],iDctB[15]);
+ FREE_MEMORY(iLevel);
+ FREE_MEMORY(iDctA);
+ FREE_MEMORY(iDctB);
+}
+static inline void PixelSubWH( int16_t *iDiff, int iSize, uint8_t *pPix1, int iStride1, uint8_t *pPix2, int iStride2 ) {+ int y, x;
+ for( y = 0; y < iSize; y++ ) {+ for( x = 0; x < iSize; x++ )
+ iDiff[x + y*iSize] = pPix1[x] - pPix2[x];
+ pPix1 += iStride1;
+ pPix2 += iStride2;
+ }
+ }
+
+#define FENC_STRIDE 16
+#define FDEC_STRIDE 32
+static void Sub4x4DctAnchor( int16_t iDct[4][4], uint8_t *pPix1, uint8_t *pPix2 ) {+ int16_t iDiff[4][4];
+ int16_t tmp[4][4];
+ int i;
+ PixelSubWH( (int16_t*)iDiff, 4, pPix1, FENC_STRIDE, pPix2, FDEC_STRIDE );
+ for( i = 0; i < 4; i++ ) {+ const int a03 = iDiff[i][0] + iDiff[i][3];
+ const int a12 = iDiff[i][1] + iDiff[i][2];
+ const int s03 = iDiff[i][0] - iDiff[i][3];
+ const int s12 = iDiff[i][1] - iDiff[i][2];
+ tmp[0][i] = a03 + a12;
+ tmp[1][i] = 2*s03 + s12;
+ tmp[2][i] = a03 - a12;
+ tmp[3][i] = s03 - 2*s12;
+ }
+ for( i = 0; i < 4; i++ ) {+ const int a03 = tmp[i][0] + tmp[i][3];
+ const int a12 = tmp[i][1] + tmp[i][2];
+ const int s03 = tmp[i][0] - tmp[i][3];
+ const int s12 = tmp[i][1] - tmp[i][2];
+ iDct[i][0] = a03 + a12;
+ iDct[i][1] = 2*s03 + s12;
+ iDct[i][2] = a03 - a12;
+ iDct[i][3] = s03 - 2*s12;
+ }
+}
+
+static void Sub8x8DctAnchor( int16_t iDct[4][4][4], uint8_t *pPix1, uint8_t *pPix2 ) {+ Sub4x4DctAnchor( iDct[0], &pPix1[0], &pPix2[0] );
+ Sub4x4DctAnchor( iDct[1], &pPix1[4], &pPix2[4] );
+ Sub4x4DctAnchor( iDct[2], &pPix1[4*FENC_STRIDE+0], &pPix2[4*FDEC_STRIDE+0] );
+ Sub4x4DctAnchor( iDct[3], &pPix1[4*FENC_STRIDE+4], &pPix2[4*FDEC_STRIDE+4] );
+}
+TEST(EncodeMbAuxTest, WelsDctT4_c) {+ int16_t iDctRef[4][4];
+ uint8_t uiPix1[16*FENC_STRIDE], uiPix2[16*FDEC_STRIDE];
+ int16_t iDct[16];
+ srand((unsigned int)time(NULL));
+ for(int i = 0; i < 4; i++)
+ for(int j = 0; j < 4; j++)
+ uiPix1[i*FENC_STRIDE+j] = uiPix2[i*FDEC_STRIDE+j] = rand() & 255;
+ Sub4x4DctAnchor(iDctRef, uiPix1, uiPix2);
+ WelsDctT4_c(iDct, uiPix1, FENC_STRIDE, uiPix2, FDEC_STRIDE);
+ for(int i = 0; i < 4; i++)
+ for(int j = 0; j < 4; j++)
+ EXPECT_EQ(iDctRef[j][i], iDct[i*4+j]);
+}
+TEST(EncodeMbAuxTest, WelsDctFourT4_c) {+ int16_t iDctRef[4][4][4]; uint8_t uiPix1[16*FENC_STRIDE], uiPix2[16*FDEC_STRIDE];
+ int16_t iDct[16*4];
+ srand((unsigned int)time(NULL));
+ for(int i = 0; i < 8; i++)
+ for(int j = 0; j < 8; j++)
+ uiPix1[i*FENC_STRIDE+j] = uiPix2[i*FDEC_STRIDE+j] = rand() & 255;
+ Sub8x8DctAnchor(iDctRef, uiPix1, uiPix2);
+ WelsDctFourT4_c(iDct, uiPix1, FENC_STRIDE, uiPix2, FDEC_STRIDE);
+ for(int k = 0; k < 4; k++)
+ for(int i = 0; i < 4; i++)
+ for(int j = 0; j < 4; j++)
+ EXPECT_EQ(iDctRef[k][j][i], iDct[k*16+i*4+j]);
+}
+
+#ifdef X86_ASM
+TEST(EncodeMbAuxTest, WelsDctT4_mmx) {+ int16_t iDctC[16], iDctM[16];
+ uint8_t uiPix1[16*FENC_STRIDE], uiPix2[16*FDEC_STRIDE];
+ srand((unsigned int)time(NULL));
+ for(int i = 0; i < 4; i++)
+ for(int j = 0; j < 4; j++)
+ uiPix1[i*FENC_STRIDE+j] = uiPix2[i*FDEC_STRIDE+j] = rand() & 255;
+ WelsDctT4_c(iDctC, uiPix1, FENC_STRIDE, uiPix2, FDEC_STRIDE);
+ WelsDctT4_mmx(iDctM, uiPix1, FENC_STRIDE, uiPix2, FDEC_STRIDE);
+ for(int i = 0; i < 16; i++)
+ EXPECT_EQ(iDctC[i], iDctM[i]);
+}
+
+TEST(EncodeMbAuxTest, WelsDctFourT4_sse2) {+ CMemoryAlign cMemoryAlign(0);
+ ALLOC_MEMORY(uint8_t, uiPix1, 16*FENC_STRIDE);
+ ALLOC_MEMORY(uint8_t, uiPix2, 16*FDEC_STRIDE);
+ ALLOC_MEMORY(int16_t, iDctC, 16*4);
+ ALLOC_MEMORY(int16_t, iDctS, 16*4);
+ srand((unsigned int)time(NULL));
+ for(int i = 0; i < 8; i++)
+ for(int j = 0; j < 8; j++)
+ uiPix1[i*FENC_STRIDE+j] = uiPix2[i*FDEC_STRIDE+j] = rand() & 255;
+ WelsDctFourT4_c(iDctC, uiPix1, FENC_STRIDE, uiPix2, FDEC_STRIDE);
+ WelsDctFourT4_sse2(iDctS, uiPix1, FENC_STRIDE, uiPix2, FDEC_STRIDE);
+ for(int i = 0; i < 64; i++)
+ EXPECT_EQ(iDctC[i], iDctS[i]);
+ FREE_MEMORY(uiPix1);
+ FREE_MEMORY(uiPix2);
+ FREE_MEMORY(iDctC);
+ FREE_MEMORY(iDctS);
+}
+
+TEST(EncodeMbAuxTest, WelsCalculateSingleCtr4x4_sse2) {+ CMemoryAlign cMemoryAlign(0);
+ ALLOC_MEMORY(int16_t, iDctC, 16);
+ ALLOC_MEMORY(int16_t, iDctS, 16);
+ srand((unsigned int)time(NULL));
+ for(int i = 0; i < 16; i++)
+ iDctC[i] = iDctS[i] = (rand() & 65535 ) - 32768;
+ WelsCalculateSingleCtr4x4_c(iDctC);
+ WelsCalculateSingleCtr4x4_sse2(iDctS);
+ for(int i = 0; i < 16; i++)
+ EXPECT_EQ(iDctC[i], iDctS[i]);
+ FREE_MEMORY(iDctC);
+ FREE_MEMORY(iDctS);
+}
+#endif
+
+void copy(uint8_t *pDst, int32_t iDStride, uint8_t *pSrc, int32_t iSStride, int32_t iWidth, int32_t iHeight) {+ for(int i = 0; i < iHeight; i++)
+ memcpy(pDst+i*iDStride, pSrc+i*iSStride, iWidth);
+}
+
+#define GENERATE_UT_FOR_COPY(width, height, function) \
+TEST(EncodeMbAuxTest, function) { \+ const int iSStride = 64; \
+ const int iDStride = 64; \
+ uint8_t _ref_src[64*64], _ref_dst[64*64], _dst[64*64]; \
+ uint8_t *ref_src = (uint8_t*)((((unsigned long)(_ref_src + 15)) >> 4) << 4); \
+ uint8_t *ref_dst = (uint8_t*)((((unsigned long)(_ref_dst + 15)) >> 4) << 4); \
+ uint8_t *dst = (uint8_t*)((((unsigned long)(_dst + 15)) >> 4) << 4); \
+ srand((unsigned int)time(NULL)); \
+ for(int i = 0; i < height; i++) \
+ for(int j = 0; j < width; j++) \
+ ref_src[i*iSStride+j] = rand() & 255; \
+ function(dst, iDStride, ref_src, iSStride); \
+ copy(ref_dst, iDStride, ref_src, iSStride, width, height); \
+ for(int i = 0; i < height; i++) \
+ for(int j = 0; j < width; j++) \
+ EXPECT_EQ(ref_dst[i*iDStride+j], dst[i*iDStride+j]); \
+}
+
+GENERATE_UT_FOR_COPY(4,4, WelsCopy4x4);
+GENERATE_UT_FOR_COPY(8,8, WelsCopy8x8_c);
+GENERATE_UT_FOR_COPY(8, 16, WelsCopy8x16_c);
+GENERATE_UT_FOR_COPY(16, 8, WelsCopy16x8_c);
+GENERATE_UT_FOR_COPY(16, 16, WelsCopy16x16_c);
+#ifdef X86_ASM
+GENERATE_UT_FOR_COPY(16, 8, WelsCopy16x8NotAligned_sse2);
+GENERATE_UT_FOR_COPY(16, 16, WelsCopy16x16NotAligned_sse2);
+GENERATE_UT_FOR_COPY(16, 16, WelsCopy16x16_sse2);
+#endif
+TEST(EncodeMbAuxTest, WelsGetNoneZeroCount_c) {+ int16_t _iLevel[32];
+ int16_t *pLevel = (int16_t*) (((((unsigned long)_iLevel) + 15) >> 4) << 4);
+ srand((unsigned int)time(NULL));
+ int32_t result = 0;
+ for(int i = 0; i < 16; i++) {+ pLevel[i] = (rand() & 0x07) - 4;
+ if(pLevel[i]) result ++;
+ }
+ int32_t nnz = WelsGetNoneZeroCount_c(pLevel);
+ EXPECT_EQ(nnz, result);
+}
+#ifdef X86_ASM
+TEST(EncodeMbAuxTest, WelsGetNoneZeroCount_sse2) {+ int16_t _iLevel[32];
+ int16_t *pLevel = (int16_t*) (((((unsigned long)_iLevel) + 15) >> 4) << 4);
+ srand((unsigned int)time(NULL));
+ int32_t result = 0;
+ for(int i = 0; i < 16; i++) {+ pLevel[i] = (rand() & 0x07) - 4;
+ if(pLevel[i]) result ++;
+ }
+ int32_t nnz = WelsGetNoneZeroCount_sse2(pLevel);
+ EXPECT_EQ(nnz, result);
+}
+#endif
+#define WELS_ABS_LC(a) ((sign ^ (int32_t)(a)) - sign)
+#define NEW_QUANT(pDct, ff, mf) (((ff)+ WELS_ABS_LC(pDct))*(mf)) >>16
+#define WELS_NEW_QUANT(pDct,ff,mf) WELS_ABS_LC(NEW_QUANT(pDct, ff, mf))
+void WelsQuantFour4x4MaxAnchor(int16_t *pDct, int16_t* ff, int16_t *mf, int16_t *max) {+ int32_t i, j, k, sign;
+ int16_t max_abs;
+ for( k = 0; k < 4; k++) {+ max_abs = 0;
+ for( i = 0; i < 16; i++ ) {+ j = i & 0x07;
+ sign = WELS_SIGN(pDct[i]);
+ pDct[i] = NEW_QUANT(pDct[i], ff[j], mf[j]);
+ if( max_abs < pDct[i]) max_abs = pDct[i];
+ pDct[i] = WELS_ABS_LC(pDct[i]);
+ }
+ pDct += 16;
+ max[k] = max_abs;
+ }
+}
+TEST(EncodeMbAuxTest, WelsQuantFour4x4Max_c) {+ int16_t ff[8], mf[8];
+ int16_t iDctA[64], iMaxA[16];
+ int16_t iDctC[64], iMaxC[16];
+ srand((unsigned int)time(NULL));
+ for(int i = 0; i < 8; i++) {+ ff[i] = rand() & 32767;
+ mf[i] = rand() & 32767;
+ }
+ for(int i = 0; i < 64; i++)
+ iDctA[i] = iDctC[i] = (rand() & 65535) - 32767;
+ WelsQuantFour4x4MaxAnchor(iDctA, ff, mf, iMaxA);
+ WelsQuantFour4x4Max_c(iDctC, ff, mf, iMaxC);
+ for(int i = 0; i < 64; i++)
+ EXPECT_EQ(iDctA[i],iDctC[i]);
+ for(int i = 0; i < 4; i++)
+ EXPECT_EQ(iMaxA[i], iMaxC[i]);
+}
+#ifdef X86_ASM
+TEST(EncodeMbAuxTest, WelsQuantFour4x4Max_sse2) {+ CMemoryAlign cMemoryAlign(0);
+ ALLOC_MEMORY(int16_t, ff, 8);
+ ALLOC_MEMORY(int16_t, mf, 8);
+ ALLOC_MEMORY(int16_t, iDctC, 64);
+ ALLOC_MEMORY(int16_t, iDctS, 64);
+ ALLOC_MEMORY(int16_t, iMaxC, 16);
+ ALLOC_MEMORY(int16_t, iMaxS, 16);
+ srand((unsigned int)time(NULL));
+ for(int i = 0; i < 8; i++) {+ ff[i] = rand() & 32767;
+ mf[i] = rand() & 32767;
+ }
+ for(int i = 0; i < 64; i++)
+ iDctC[i] = iDctS[i] = (rand() & 65535) - 32767;
+ WelsQuantFour4x4Max_c(iDctC, ff, mf, iMaxC);
+ WelsQuantFour4x4Max_sse2(iDctS, ff, mf, iMaxS);
+ for(int i = 0; i < 64; i++)
+ EXPECT_EQ(iDctC[i], iDctS[i]);
+ for(int i = 0; i < 4; i++)
+ EXPECT_EQ(iMaxC[i], iMaxS[i]);
+ FREE_MEMORY(ff);
+ FREE_MEMORY(mf);
+ FREE_MEMORY(iDctC);
+ FREE_MEMORY(iDctS);
+ FREE_MEMORY(iMaxC);
+ FREE_MEMORY(iMaxS);
+}
+#endif
+int32_t WelsHadamardQuant2x2SkipAnchor(int16_t *rs, int16_t ff, int16_t mf) {+ int16_t pDct[4], s[4];
+ int16_t threshold = ((1<<16)-1)/mf - ff;
+ s[0] = rs[0] + rs[32];
+ s[1] = rs[0] - rs[32];
+ s[2] = rs[16] + rs[48];
+ s[3] = rs[16] - rs[48];
+ pDct[0] = s[0] + s[2];
+ pDct[1] = s[0] - s[2];
+ pDct[2] = s[1] + s[3];
+ pDct[3] = s[1] - s[3];
+ return ((WELS_ABS(pDct[0]) > threshold) || (WELS_ABS(pDct[1]) > threshold) || (WELS_ABS(pDct[2]) > threshold) || (WELS_ABS(pDct[3]) > threshold));
+}
+
+TEST(EncodeMbAuxTest, WelsHadamardQuant2x2Skip_c) {+ int16_t iRS[64];
+ int16_t ff, mf;
+ srand((unsigned int)time(NULL));
+ for(int i = 0; i < 64; i++)
+ iRS[i] = (rand() & 32767) - 16384;
+ ff = rand() & 32767;
+ mf = rand() & 32767;
+ EXPECT_EQ(WelsHadamardQuant2x2Skip_c(iRS, ff, mf), WelsHadamardQuant2x2SkipAnchor(iRS, ff, mf));
+}
+
+int32_t WelsHadamardQuant2x2Anchor(int16_t *rs, const int16_t ff, int16_t mf, int16_t * pDct, int16_t * block) {+ int16_t s[4];
+ int32_t sign, i, dc_nzc = 0;
+
+ s[0] = rs[0] + rs[32];
+ s[1] = rs[0] - rs[32];
+ s[2] = rs[16] + rs[48];
+ s[3] = rs[16] - rs[48];
+
+ rs[0] = 0;
+ rs[16] = 0;
+ rs[32] = 0;
+ rs[48] = 0;
+
+ pDct[0] = s[0] + s[2];
+ pDct[1] = s[0] - s[2];
+ pDct[2] = s[1] + s[3];
+ pDct[3] = s[1] - s[3];
+
+ sign = WELS_SIGN(pDct[0]);
+ pDct[0] = WELS_NEW_QUANT(pDct[0], ff, mf);
+ sign = WELS_SIGN(pDct[1]);
+ pDct[1] = WELS_NEW_QUANT(pDct[1], ff, mf);
+ sign = WELS_SIGN(pDct[2]);
+ pDct[2] = WELS_NEW_QUANT(pDct[2], ff, mf);
+ sign = WELS_SIGN(pDct[3]);
+ pDct[3] = WELS_NEW_QUANT(pDct[3], ff, mf);
+ ST64( block, LD64(pDct) );
+ for(i=0; i<4; i++)
+ dc_nzc += (block[i] != 0);
+ return dc_nzc;
+}
+
+TEST(EncodeMbAuxTest, WelsHadamardQuant2x2_c) {+ int16_t iRsC[64], iRsA[64];
+ int16_t ff, mf;
+ int16_t iBlockA[16], iBlockC[16], iDctA[4], iDctC[4];
+ srand((unsigned int)time(NULL));
+ for(int i = 0; i < 64; i++)
+ iRsA[i] = iRsC[i] = (rand() & 32767) - 16384;
+ for(int i = 0; i < 4; i++)
+ iDctA[i] = iDctC[i] = (rand() & 32767) - 16384;
+ ff = rand() & 32767;
+ mf = rand() & 32767;
+
+ int32_t iRetA = WelsHadamardQuant2x2Anchor(iRsA, ff, mf, iDctA, iBlockA);
+ int32_t iRetC = WelsHadamardQuant2x2_c(iRsC, ff, mf, iDctC, iBlockC);
+ EXPECT_EQ(iRetA, iRetC);
+ for(int i = 0; i < 4; i++)
+ EXPECT_EQ(iDctA[i], iDctC[i]);
+}
+
+void WelsHadamardT4DcAnchor( int16_t *pLumaDc, int16_t *pDct) {+ int32_t p[16], s[4];
+ int32_t i, iIdx;
+ for(i = 0 ; i < 16 ; i +=4) {+ iIdx = ((i&0x08) << 4) +((i&0x04) << 3);
+ s[0] = pDct[iIdx ] + pDct[iIdx+80];
+ s[3] = pDct[iIdx ] - pDct[iIdx+80];
+ s[1] = pDct[iIdx+16] + pDct[iIdx+64];
+ s[2] = pDct[iIdx+16] - pDct[iIdx+64];
+ p[i ] = s[0] + s[1];
+ p[i+2] = s[0] - s[1];
+ p[i+1] = s[3] + s[2];
+ p[i+3] = s[3] - s[2];
+ }
+ for(i = 0 ; i < 4 ; i ++) {+ s[0] = p[i ] + p[i+12];
+ s[3] = p[i ] - p[i+12];
+ s[1] = p[i+4] + p[i+ 8];
+ s[2] = p[i+4] - p[i+ 8];
+ pLumaDc[i ] = WELS_CLIP3((s[0] + s[1] + 1) >> 1, -32768, 32767);
+ pLumaDc[i+8 ] = WELS_CLIP3((s[0] - s[1] + 1) >> 1, -32768, 32767);
+ pLumaDc[i+4 ] = WELS_CLIP3((s[3] + s[2] + 1) >> 1, -32768, 32767);
+ pLumaDc[i+12] = WELS_CLIP3((s[3] - s[2] + 1) >> 1, -32768, 32767);
+ }
+}
+TEST(EncodeMbAuxTest, WelsHadamardT4Dc_c) {+ CMemoryAlign cMemoryAlign(0);
+ ALLOC_MEMORY(int16_t, iDct, 128*16);
+ ALLOC_MEMORY(int16_t, iLumaDcR, 16);
+ ALLOC_MEMORY(int16_t, iLumaDcC, 16);
+ srand((unsigned int)time(NULL));
+ for(int i = 0; i < 128*16; i++)
+ iDct[i] = (rand() & 32767) - 16384;
+ WelsHadamardT4DcAnchor(iLumaDcR, iDct);
+ WelsHadamardT4Dc_c(iLumaDcC, iDct);
+ for(int i = 0;i < 16; i++)
+ EXPECT_EQ(iLumaDcR[i], iLumaDcC[i]);
+ FREE_MEMORY(iDct);
+ FREE_MEMORY(iLumaDcR);
+ FREE_MEMORY(iLumaDcC);
+}
+#ifdef X86_ASM
+TEST(EncodeMbAuxTest, WelsHadamardT4Dc_sse2) {+ CMemoryAlign cMemoryAlign(0);
+ ALLOC_MEMORY(int16_t, iDct, 128*16);
+ ALLOC_MEMORY(int16_t, iLumaDcC, 16);
+ ALLOC_MEMORY(int16_t, iLumaDcS, 16);
+ srand((unsigned int)time(NULL));
+ for(int i = 0; i < 128*16; i++)
+ iDct[i] = (rand() & 32767) - 16384;
+ WelsHadamardT4Dc_c(iLumaDcC, iDct);
+ WelsHadamardT4Dc_sse2(iLumaDcS, iDct);
+ for(int i = 0;i < 16; i++)
+ EXPECT_EQ(iLumaDcC[i], iLumaDcS[i]);
+ FREE_MEMORY(iDct);
+ FREE_MEMORY(iLumaDcC);
+ FREE_MEMORY(iLumaDcS);
+}
+#endif
--- a/test/encoder/targets.mk
+++ b/test/encoder/targets.mk
@@ -1,6 +1,7 @@
ENCODER_UNITTEST_SRCDIR=test/encoder
ENCODER_UNITTEST_CPP_SRCS=\
$(ENCODER_UNITTEST_SRCDIR)/EncUT_EncoderMb.cpp\
+ $(ENCODER_UNITTEST_SRCDIR)/EncUT_EncoderMbAux.cpp\
$(ENCODER_UNITTEST_SRCDIR)/EncUT_ExpandPic.cpp\
$(ENCODER_UNITTEST_SRCDIR)/EncUT_ExpGolomb.cpp\
$(ENCODER_UNITTEST_SRCDIR)/EncUT_GetIntraPredictor.cpp\