Showing the changes to test/decoder/DecUT_IntraPrediction.cpp (1 of 1 file). Show all
Unified · Split
| @@ -5,13 +5,15 @@ · 4 unmodified lines | ||
| 5 | 5 | #include "get_intra_predictor.h" |
| 6 | 6 | #include "typedefs.h" |
| 7 | 7 | #include "ls_defines.h" |
| 8 | +#include "macros.h" | |
| 9 | + | |
| 8 | 10 | using namespace WelsDec; |
| 9 | 11 | #define GENERATE_4x4_UT(pred, ref, ASM, CPUFLAGS) \ |
| 10 | 12 | TEST(DecoderIntraPredictionTest, pred) { \ |
| 11 | 13 | const int32_t kiStride = 32; \ |
| 12 | 14 | int32_t iRunTimes = 1000; \ |
| 13 | − uint8_t pPredBuffer[9 * kiStride]; \ | |
| 14 | − uint8_t pRefBuffer[9 * kiStride]; \ | |
| 15 | + ENFORCE_STACK_ALIGN_1D (uint8_t, pPredBuffer, 12 * kiStride, 4); \ | |
| 16 | + ENFORCE_STACK_ALIGN_1D (uint8_t, pRefBuffer, 12 * kiStride, 4); \ | |
| 15 | 17 | if (ASM) {\ |
| 16 | 18 | int32_t iNumberofCPUCore = 1; \ |
| 17 | 19 | uint32_t uiCPUFlags = WelsCPUFeatureDetect( &iNumberofCPUCore); \ |
| @@ -21,16 +23,16 @@ if (ASM) {\ · 3 unmodified lines | ||
| 21 | 23 | }\ |
| 22 | 24 | srand((unsigned int)time(NULL)); \ |
| 23 | 25 | while(iRunTimes--) {\ |
| 24 | − for (int i = 0; i < 9; i++) {\ | |
| 25 | − pRefBuffer[i] = pPredBuffer[i] = rand() & 255; \ | |
| 26 | − pRefBuffer[i * kiStride] = pPredBuffer[i * kiStride] = rand() & 255; \ | |
| 26 | + for (int i = 0; i < 12; i++) {\ | |
| 27 | + pRefBuffer[kiStride * 3 + i] = pPredBuffer[kiStride * 3 + i] = rand() & 255; \ | |
| 28 | + pRefBuffer[i * kiStride + 3] = pPredBuffer[i * kiStride + 3] = rand() & 255; \ | |
| 27 | 29 | } \ |
| 28 | − pred (&pPredBuffer[kiStride + 1], kiStride); \ | |
| 29 | − ref (&pRefBuffer[kiStride + 1], kiStride); \ | |
| 30 | + pred (&pPredBuffer[kiStride * 4 + 4], kiStride); \ | |
| 31 | + ref (&pRefBuffer[kiStride * 4 + 4], kiStride); \ | |
| 30 | 32 | bool ok = true; \ |
| 31 | 33 | for (int i = 0; i < 4; i++) \ |
| 32 | 34 | for (int j = 0; j < 4; j++) \ |
| 33 | − if (pPredBuffer[(i+1) * kiStride + j + 1] != pRefBuffer[(i+1) * kiStride + j + 1]) { \ | |
| 35 | + if (pPredBuffer[(i+4) * kiStride + j + 4] != pRefBuffer[(i+4) * kiStride + j + 4]) { \ | |
| 34 | 36 | ok = false; \ |
| 35 | 37 | break; \ |
| 36 | 38 | } \ |