Showing the changes to test/decoder/DecUT_IdctResAddPred.cpp (1 of 23 files). Show all
Unified · Split
| @@ -0,0 +1,83 @@ | ||
| 1 | +#include <gtest/gtest.h> | |
| 2 | +#include <time.h> | |
| 3 | +#include "macros.h" | |
| 4 | +#include "../codec/decoder/core/inc/decode_mb_aux.h" | |
| 5 | +using namespace WelsDec; | |
| 6 | +void IdctResAddPred_ref (uint8_t* pPred, const int32_t kiStride, int16_t* pRs) { | |
| 7 | + int16_t iSrc[16]; | |
| 8 | + | |
| 9 | + uint8_t* pDst = pPred; | |
| 10 | + const int32_t kiStride2 = kiStride << 1; | |
| 11 | + const int32_t kiStride3 = kiStride + kiStride2; | |
| 12 | + int32_t i; | |
| 13 | + | |
| 14 | + for (i = 0; i < 4; i++) { | |
| 15 | + const int32_t kiY = i << 2; | |
| 16 | + const int32_t kiT0 = pRs[kiY] + pRs[kiY + 2]; | |
| 17 | + const int32_t kiT1 = pRs[kiY] - pRs[kiY + 2]; | |
| 18 | + const int32_t kiT2 = (pRs[kiY + 1] >> 1) - pRs[kiY + 3]; | |
| 19 | + const int32_t kiT3 = pRs[kiY + 1] + (pRs[kiY + 3] >> 1); | |
| 20 | + | |
| 21 | + iSrc[kiY] = kiT0 + kiT3; | |
| 22 | + iSrc[kiY + 1] = kiT1 + kiT2; | |
| 23 | + iSrc[kiY + 2] = kiT1 - kiT2; | |
| 24 | + iSrc[kiY + 3] = kiT0 - kiT3; | |
| 25 | + } | |
| 26 | + | |
| 27 | + for (i = 0; i < 4; i++) { | |
| 28 | + int32_t kT1 = iSrc[i] + iSrc[i + 8]; | |
| 29 | + int32_t kT2 = iSrc[i + 4] + (iSrc[i + 12] >> 1); | |
| 30 | + int32_t kT3 = (32 + kT1 + kT2) >> 6; | |
| 31 | + int32_t kT4 = (32 + kT1 - kT2) >> 6; | |
| 32 | + | |
| 33 | + pDst[i] = WelsClip1 (kT3 + pPred[i]); | |
| 34 | + pDst[i + kiStride3] = WelsClip1 (kT4 + pPred[i + kiStride3]); | |
| 35 | + | |
| 36 | + kT1 = iSrc[i] - iSrc[i + 8]; | |
| 37 | + kT2 = (iSrc[i + 4] >> 1) - iSrc[i + 12]; | |
| 38 | + pDst[i + kiStride] = WelsClip1 (((32 + kT1 + kT2) >> 6) + pDst[i + kiStride]); | |
| 39 | + pDst[i + kiStride2] = WelsClip1 (((32 + kT1 - kT2) >> 6) + pDst[i + kiStride2]); | |
| 40 | + } | |
| 41 | +} | |
| 42 | + | |
| 43 | +#define GENERATE_IDCTRESADDPRED(pred) \ | |
| 44 | +TEST(DecoderDecodeMbAux, pred) {\ | |
| 45 | + const int32_t kiStride = 32;\ | |
| 46 | + const int iBits = 12;\ | |
| 47 | + const int iMask = (1 << iBits) - 1;\ | |
| 48 | + const int iOffset = 1 << (iBits - 1);\ | |
| 49 | + int16_t iRS[16];\ | |
| 50 | + uint8_t uiPred[16*kiStride];\ | |
| 51 | + int16_t iRefRS[16];\ | |
| 52 | + uint8_t uiRefPred[16*kiStride];\ | |
| 53 | + int32_t iRunTimes = 1000;\ | |
| 54 | + srand((unsigned int)time(NULL));\ | |
| 55 | + while(iRunTimes--) {\ | |
| 56 | + for(int i = 0; i < 4; i++)\ | |
| 57 | + for(int j = 0; j < 4; j++)\ | |
| 58 | + iRefRS[i*4+j] = iRS[i*4+j] = (rand() & iMask) - iOffset;\ | |
| 59 | + for(int i = 0; i < 4; i++)\ | |
| 60 | + for(int j = 0; j < 4; j++)\ | |
| 61 | + uiRefPred[i * kiStride + j] = uiPred[i * kiStride + j] = rand() & 255;\ | |
| 62 | + pred(uiPred, kiStride, iRS);\ | |
| 63 | + IdctResAddPred_ref(uiRefPred, kiStride, iRefRS);\ | |
| 64 | + bool ok = true;\ | |
| 65 | + for(int i = 0; i < 4; i++)\ | |
| 66 | + for(int j = 0; j < 4; j++)\ | |
| 67 | + if (uiRefPred[i * kiStride + j] != uiPred[i * kiStride + j]) {\ | |
| 68 | + ok = false;\ | |
| 69 | + goto next;\ | |
| 70 | + }\ | |
| 71 | + next:\ | |
| 72 | + EXPECT_EQ(ok, true);\ | |
| 73 | + }\ | |
| 74 | +} | |
| 75 | + | |
| 76 | +GENERATE_IDCTRESADDPRED (IdctResAddPred_c) | |
| 77 | +#if defined(X86_ASM) | |
| 78 | +GENERATE_IDCTRESADDPRED (IdctResAddPred_mmx) | |
| 79 | +#endif | |
| 80 | + | |
| 81 | +#if defined(HAVE_NEON) | |
| 82 | +GENERATE_IDCTRESADDPRED (IdctResAddPred_neon) | |
| 83 | +#endif | |