ref: 945fc0e03ee855bd60e6b5628879b2ad48bb8162
dir: /test/decoder/DecUT_IdctResAddPred.cpp/
#include <gtest/gtest.h> #include "macros.h" #include "decode_mb_aux.h" #include "deblocking.h" #include "cpu.h" using namespace WelsDec; namespace { void IdctResAddPred_ref (uint8_t* pPred, const int32_t kiStride, int16_t* pRs) { int16_t iSrc[16]; uint8_t* pDst = pPred; const int32_t kiStride2 = kiStride << 1; const int32_t kiStride3 = kiStride + kiStride2; int32_t i; for (i = 0; i < 4; i++) { const int32_t kiY = i << 2; const int32_t kiT0 = pRs[kiY] + pRs[kiY + 2]; const int32_t kiT1 = pRs[kiY] - pRs[kiY + 2]; const int32_t kiT2 = (pRs[kiY + 1] >> 1) - pRs[kiY + 3]; const int32_t kiT3 = pRs[kiY + 1] + (pRs[kiY + 3] >> 1); iSrc[kiY] = kiT0 + kiT3; iSrc[kiY + 1] = kiT1 + kiT2; iSrc[kiY + 2] = kiT1 - kiT2; iSrc[kiY + 3] = kiT0 - kiT3; } for (i = 0; i < 4; i++) { int32_t kT1 = iSrc[i] + iSrc[i + 8]; int32_t kT2 = iSrc[i + 4] + (iSrc[i + 12] >> 1); int32_t kT3 = (32 + kT1 + kT2) >> 6; int32_t kT4 = (32 + kT1 - kT2) >> 6; pDst[i] = WelsClip1 (kT3 + pPred[i]); pDst[i + kiStride3] = WelsClip1 (kT4 + pPred[i + kiStride3]); kT1 = iSrc[i] - iSrc[i + 8]; kT2 = (iSrc[i + 4] >> 1) - iSrc[i + 12]; pDst[i + kiStride] = WelsClip1 (((32 + kT1 + kT2) >> 6) + pDst[i + kiStride]); pDst[i + kiStride2] = WelsClip1 (((32 + kT1 - kT2) >> 6) + pDst[i + kiStride2]); } } void SetNonZeroCount_ref (int8_t* pNonZeroCount) { int32_t i; for (i = 0; i < 24; i++) { pNonZeroCount[i] = !!pNonZeroCount[i]; } } #if defined(X86_ASM) #if defined(HAVE_AVX2) void IdctFourResAddPred_ref (uint8_t* pPred, int32_t iStride, int16_t* pRs) { IdctResAddPred_ref (pPred + 0 * iStride + 0, iStride, pRs + 0 * 16); IdctResAddPred_ref (pPred + 0 * iStride + 4, iStride, pRs + 1 * 16); IdctResAddPred_ref (pPred + 4 * iStride + 0, iStride, pRs + 2 * 16); IdctResAddPred_ref (pPred + 4 * iStride + 4, iStride, pRs + 3 * 16); } #endif #endif } // anon ns #define GENERATE_IDCTRESADDPRED(pred, flag) \ TEST(DecoderDecodeMbAux, pred) {\ const int32_t kiStride = 32;\ const int iBits = 12;\ const int iMask = (1 << iBits) - 1;\ const int iOffset = 1 << (iBits - 1);\ ENFORCE_STACK_ALIGN_1D (int16_t, iRS, 16, 16);\ ENFORCE_STACK_ALIGN_1D (uint8_t, uiPred, 16 * kiStride, 16);\ int16_t iRefRS[16];\ uint8_t uiRefPred[16*kiStride];\ int32_t iRunTimes = 1000;\ uint32_t uiCPUFlags = WelsCPUFeatureDetect(NULL); \ if ((uiCPUFlags & flag) == 0 && flag != 0) \ return; \ while(iRunTimes--) {\ for(int i = 0; i < 4; i++)\ for(int j = 0; j < 4; j++)\ iRefRS[i*4+j] = iRS[i*4+j] = (rand() & iMask) - iOffset;\ for(int i = 0; i < 4; i++)\ for(int j = 0; j < 4; j++)\ uiRefPred[i * kiStride + j] = uiPred[i * kiStride + j] = rand() & 255;\ pred(uiPred, kiStride, iRS);\ IdctResAddPred_ref(uiRefPred, kiStride, iRefRS);\ bool ok = true;\ for(int i = 0; i < 4; i++)\ for(int j = 0; j < 4; j++)\ if (uiRefPred[i * kiStride + j] != uiPred[i * kiStride + j]) {\ ok = false;\ goto next;\ }\ next:\ EXPECT_EQ(ok, true);\ }\ } #define GENERATE_IDCTFOURRESADDPRED(pred, flag) \ TEST(DecoderDecodeMbAux, pred) {\ const int32_t kiStride = 32;\ const int iBits = 12;\ const int iMask = (1 << iBits) - 1;\ const int iOffset = 1 << (iBits - 1);\ ENFORCE_STACK_ALIGN_1D (int16_t, iRS, 4 * 16, 16);\ ENFORCE_STACK_ALIGN_1D (uint8_t, uiPred, 4 * 16 * kiStride, 16);\ int16_t iRefRS[4 * 16];\ uint8_t uiRefPred[4 * 16 * kiStride];\ int8_t iNzc[6] = { 0 };\ int32_t iRunTimes = 1000;\ uint32_t uiCPUFlags = WelsCPUFeatureDetect(0); \ if ((uiCPUFlags & flag) == 0 && flag != 0) \ return; \ while (iRunTimes--) {\ for (int i = 0; i < 4; i++)\ for (int j = 0; j < 16; j++)\ iNzc[i / 2 * 4 + i % 2] += !!(iRefRS[16 * i + j] = iRS[16 * i + j] = (rand() & iMask) - iOffset);\ for (int i = 0; i < 8; i++)\ for (int j = 0; j < 8; j++)\ uiRefPred[i * kiStride + j] = uiPred[i * kiStride + j] = rand() & 255;\ pred (uiPred, kiStride, iRS, iNzc);\ IdctFourResAddPred_ref (uiRefPred, kiStride, iRefRS);\ bool ok = true;\ for (int i = 0; i < 8; i++)\ for (int j = 0; j < 8; j++)\ if (uiRefPred[i * kiStride + j] != uiPred[i * kiStride + j]) {\ ok = false;\ goto next;\ }\ next:\ EXPECT_EQ(ok, true);\ }\ } GENERATE_IDCTRESADDPRED (IdctResAddPred_c, 0) #if defined(X86_ASM) GENERATE_IDCTRESADDPRED (IdctResAddPred_mmx, WELS_CPU_MMXEXT) GENERATE_IDCTRESADDPRED (IdctResAddPred_sse2, WELS_CPU_SSE2) #if defined(HAVE_AVX2) GENERATE_IDCTRESADDPRED (IdctResAddPred_avx2, WELS_CPU_AVX2) GENERATE_IDCTFOURRESADDPRED (IdctFourResAddPred_avx2, WELS_CPU_AVX2) #endif #endif #if defined(HAVE_NEON) GENERATE_IDCTRESADDPRED (IdctResAddPred_neon, WELS_CPU_NEON) #endif #if defined(HAVE_NEON_AARCH64) GENERATE_IDCTRESADDPRED (IdctResAddPred_AArch64_neon, WELS_CPU_NEON) #endif #define GENERATE_SETNONZEROCOUNT(method, flag) \ TEST(DecoderDecodeMbAux, method) \ {\ uint32_t uiCPUFlags = WelsCPUFeatureDetect(NULL); \ if ((uiCPUFlags & flag) == 0 && flag != 0) \ return; \ int8_t iNonZeroCount[2][24];\ for(int32_t i = 0; i < 24; i++) {\ iNonZeroCount[0][i] = iNonZeroCount[1][i] = (rand() % 25);\ }\ method(iNonZeroCount[0]);\ SetNonZeroCount_ref(iNonZeroCount[1]);\ for(int32_t i =0; i<24; i++) {\ ASSERT_EQ (iNonZeroCount[0][i], iNonZeroCount[1][i]);\ }\ for(int32_t i =0; i<24; i++) {\ iNonZeroCount[0][i] = iNonZeroCount[1][i] = 0;\ }\ method(iNonZeroCount[0]);\ SetNonZeroCount_ref(iNonZeroCount[1]);\ for(int32_t i =0; i<24; i++) {\ ASSERT_EQ (iNonZeroCount[0][i], iNonZeroCount[1][i]);\ }\ for(int32_t i =0; i<24; i++) {\ iNonZeroCount[0][i] = iNonZeroCount[1][i] = 16;\ }\ method(iNonZeroCount[0]);\ SetNonZeroCount_ref(iNonZeroCount[1]);\ for(int32_t i =0; i<24; i++) {\ ASSERT_EQ (iNonZeroCount[0][i], iNonZeroCount[1][i]);\ }\ } GENERATE_SETNONZEROCOUNT (WelsNonZeroCount_c, 0) #if defined(X86_ASM) GENERATE_SETNONZEROCOUNT (WelsNonZeroCount_sse2, WELS_CPU_SSE2) #endif #if defined(HAVE_NEON) GENERATE_SETNONZEROCOUNT (WelsNonZeroCount_neon, WELS_CPU_NEON) #endif #if defined(HAVE_NEON_AARCH64) GENERATE_SETNONZEROCOUNT (WelsNonZeroCount_AArch64_neon, WELS_CPU_NEON) #endif