1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
#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
|
