ref: e3c306608cf21fe1ca61cd3228a50fc71434cac9
dir: /test/encoder/EncUT_Cavlc.cpp/
#include "cpu.h" #include "macros.h" #include "set_mb_syn_cavlc.h" #include <gtest/gtest.h> #include <cmath> #include <cstddef> using namespace WelsEnc; namespace { int32_t CavlcParamCal_ref (int16_t* pCoffLevel, uint8_t* pRun, int16_t* pLevel, int32_t* pTotalCoeff, int32_t iLastIndex) { int32_t iTotalZeros = 0; int32_t iTotalCoeffs = 0; while (iLastIndex >= 0 && pCoffLevel[iLastIndex] == 0) { -- iLastIndex; } while (iLastIndex >= 0) { int32_t iCountZero = 0; pLevel[iTotalCoeffs] = pCoffLevel[iLastIndex--]; while (iLastIndex >= 0 && pCoffLevel[iLastIndex] == 0) { ++ iCountZero; -- iLastIndex; } iTotalZeros += iCountZero; pRun[iTotalCoeffs++] = iCountZero; } *pTotalCoeff = iTotalCoeffs; return iTotalZeros; } void TestCavlcParamCalWithEndIdx (PCavlcParamCalFunc func, int endIdx, bool allZero, bool allNonZero) { ENFORCE_STACK_ALIGN_1D(int16_t, coeffLevel, 16, 16); ENFORCE_STACK_ALIGN_1D(int16_t, level, 16, 16); ENFORCE_STACK_ALIGN_1D(uint8_t, run, 16, 16); uint8_t run_ref[16]; int16_t level_ref[16]; int32_t totalCoeffs = 0; int32_t totalCoeffs_ref = 0; for (int i = 0; i < 16; i++) { const int r = std::rand(); if (allZero || (i > endIdx && endIdx > 7)) coeffLevel[i] = 0; else if (allNonZero) coeffLevel[i] = r % 0xFFFF - 0x8000 ? r % 0xFFFF - 0x8000 : 0x7FFF; else coeffLevel[i] = (r >> 16 & 1) * ((r & 0xFFFF) - 0x8000); } const int32_t totalZeros_ref = CavlcParamCal_ref (coeffLevel, run_ref, level_ref, &totalCoeffs_ref, endIdx); const int32_t totalZeros = func (coeffLevel, run, level, &totalCoeffs, endIdx); ASSERT_EQ (totalCoeffs, totalCoeffs_ref); if (totalCoeffs > 0) ASSERT_EQ (totalZeros, totalZeros_ref); for (int i = 0; i < totalCoeffs_ref; i++) ASSERT_EQ (level[i], level_ref[i]); for (int i = 0; i < totalCoeffs_ref - 1; i++) ASSERT_EQ (run[i], run_ref[i]); } void TestCavlcParamCal (PCavlcParamCalFunc func) { const int endIdxes[] = { 3, 14, 15 }; const int num_test_repetitions = 10000; for (std::size_t i = 0; i < sizeof endIdxes / sizeof *endIdxes; i++) { for (int count = 0; count < num_test_repetitions; count++) TestCavlcParamCalWithEndIdx (func, endIdxes[i], count == 0, count == 1); } } } // anon ns. TEST (CavlcTest, CavlcParamCal_c) { TestCavlcParamCal (CavlcParamCal_c); } #ifdef X86_32_ASM TEST (CavlcTest, CavlcParamCal_sse2) { TestCavlcParamCal (CavlcParamCal_sse2); } #endif #ifdef X86_ASM TEST (CavlcTest, CavlcParamCal_sse42) { if (WelsCPUFeatureDetect (0) & WELS_CPU_SSE42) TestCavlcParamCal (CavlcParamCal_sse42); } #endif