#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_ASMTEST (CavlcTest, CavlcParamCal_sse2) {  TestCavlcParamCal (CavlcParamCal_sse2);}#endif #ifdef X86_ASMTEST (CavlcTest, CavlcParamCal_sse42) {  if (WelsCPUFeatureDetect (0) & WELS_CPU_SSE42)    TestCavlcParamCal (CavlcParamCal_sse42);}#endif