#include <gtest/gtest.h>
#include "cpu.h"
#include "cpu_core.h"
#include "get_intra_predictor.h"
#include "typedefs.h"
#include "ls_defines.h"
#include "macros.h"

using namespace WelsDec;
#define GENERATE_4x4_UT(pred, ref, ASM, CPUFLAGS) \
  TEST(DecoderIntraPredictionTest, pred) { \
  const int32_t kiStride = 32; \
  int32_t iRunTimes = 1000; \
  ENFORCE_STACK_ALIGN_1D (uint8_t, pPredBuffer, 12 * kiStride, 4); \
  ENFORCE_STACK_ALIGN_1D (uint8_t, pRefBuffer, 12 * kiStride, 4); \
if (ASM) {\
  int32_t  iNumberofCPUCore = 1; \
  uint32_t uiCPUFlags = WelsCPUFeatureDetect( &iNumberofCPUCore); \
  if ((uiCPUFlags & CPUFLAGS) == 0) {\
    return; \
  } \
}\
  while(iRunTimes--) {\
  for (int i = 0; i < 12; i++) {\
    pRefBuffer[kiStride * 3 + i] = pPredBuffer[kiStride * 3 + i] = rand() & 255; \
    pRefBuffer[i * kiStride + 3] = pPredBuffer[i * kiStride + 3] = rand() & 255; \
  } \
  pred (&pPredBuffer[kiStride * 4 + 4], kiStride); \
  ref (&pRefBuffer[kiStride * 4 + 4], kiStride); \
  bool ok = true; \
  for (int i = 0; i < 4; i++) \
    for (int j = 0; j < 4; j++) \
      if (pPredBuffer[(i+4) * kiStride + j + 4] != pRefBuffer[(i+4) * kiStride + j + 4]) { \
        ok = false; \
        break; \
      } \
  EXPECT_EQ(ok, true);\
  } \
  }

#define PREDV(size) \
void LumaI##size##x##size##PredV(uint8_t *pPred, const int32_t kiStride) {\
  int i; \
  for (i = 0; i < size; i++) {\
    memcpy(pPred + i * kiStride, pPred - kiStride, size * sizeof(uint8_t)); \
  } \
}

#define PREDH(size) \
void LumaI##size##x##size##PredH(uint8_t *pPred, const int32_t kiStride) {\
  for (int i = 0; i < size; i++) { \
    memset(pPred + i * kiStride, pPred[i * kiStride - 1], size * sizeof(uint8_t));\
  }\
}

#define PREDDC(size, log) \
void LumaI##size##x##size##PredDC(uint8_t *pPred, const int32_t kiStride) {\
  int iSum = size; \
  for (int i = 0; i < size; i ++) \
    iSum += pPred[-1 + i * kiStride] + pPred[i - kiStride]; \
  uint8_t uiMean = iSum >>(log+1);\
  for (int i = 0; i < size; i ++) \
    memset(pPred + i * kiStride, uiMean, size * sizeof(uint8_t)); \
}

#define PREDDCLeft(size, log) \
void LumaI##size##x##size##PredDCLeft(uint8_t *pPred, const int32_t kiStride) {\
  int iSum = size/2; \
  for (int i = 0; i < size; i ++) \
    iSum += pPred[-1 + i * kiStride]; \
  uint8_t uiMean = iSum >>(log);\
  for (int i = 0; i < size; i ++) \
    memset(pPred + i * kiStride, uiMean, size * sizeof(uint8_t)); \
}

#define PREDDCTop(size, log) \
void LumaI##size##x##size##PredDCTop(uint8_t *pPred, const int32_t kiStride) {\
  int iSum = size/2; \
  for (int i = 0; i < size; i ++) \
    iSum += pPred[i - kiStride]; \
  uint8_t uiMean = iSum >>(log);\
  for (int i = 0; i < size; i ++) \
    memset(pPred + i * kiStride, uiMean, size * sizeof(uint8_t)); \
}

#define PREDDCNone(size, log) \
void LumaI##size##x##size##PredDCNone(uint8_t *pPred, const int32_t kiStride) {\
  uint8_t uiMean = 128;\
  for (int i = 0; i < size; i ++) \
    memset(pPred + i * kiStride, uiMean, size * sizeof(uint8_t)); \
}


/*down pLeft*/
void WelsI4x4LumaPredDDL_ref (uint8_t* pPred, const int32_t kiStride) {
  const int32_t kiStride2	= kiStride << 1;
  const int32_t kiStride3	= kiStride + kiStride2;
  /*get pTop*/
  uint8_t* ptop			= &pPred[-kiStride];
  const uint8_t kuiT0		= *ptop;
  const uint8_t kuiT1		= * (ptop + 1);
  const uint8_t kuiT2		= * (ptop + 2);
  const uint8_t kuiT3		= * (ptop + 3);
  const uint8_t kuiT4		= * (ptop + 4);
  const uint8_t kuiT5		= * (ptop + 5);
  const uint8_t kuiT6		= * (ptop + 6);
  const uint8_t kuiT7		= * (ptop + 7);
  const uint8_t kuiDDL0	= (2 + kuiT0 + kuiT2 + (kuiT1 << 1)) >> 2;	// kDDL0
  const uint8_t kuiDDL1	= (2 + kuiT1 + kuiT3 + (kuiT2 << 1)) >> 2;	// kDDL1
  const uint8_t kuiDDL2	= (2 + kuiT2 + kuiT4 + (kuiT3 << 1)) >> 2;	// kDDL2
  const uint8_t kuiDDL3	= (2 + kuiT3 + kuiT5 + (kuiT4 << 1)) >> 2;	// kDDL3
  const uint8_t kuiDDL4	= (2 + kuiT4 + kuiT6 + (kuiT5 << 1)) >> 2;	// kDDL4
  const uint8_t kuiDDL5	= (2 + kuiT5 + kuiT7 + (kuiT6 << 1)) >> 2;	// kDDL5
  const uint8_t kuiDDL6	= (2 + kuiT6 + kuiT7 + (kuiT7 << 1)) >> 2;	// kDDL6
  const uint8_t kuiList[8] = { kuiDDL0, kuiDDL1, kuiDDL2, kuiDDL3, kuiDDL4, kuiDDL5, kuiDDL6, 0 };

  ST32 (pPred          , LD32 (kuiList));
  ST32 (pPred + kiStride , LD32 (kuiList + 1));
  ST32 (pPred + kiStride2, LD32 (kuiList + 2));
  ST32 (pPred + kiStride3, LD32 (kuiList + 3));
}

/*down pLeft*/
void WelsI4x4LumaPredDDLTop_ref (uint8_t* pPred, const int32_t kiStride) {
  const int32_t kiStride2	= kiStride << 1;
  const int32_t kiStride3	= kiStride + kiStride2;
  /*get pTop*/
  uint8_t* ptop			= &pPred[-kiStride];
  const uint8_t kuiT0		= *ptop;
  const uint8_t kuiT1		= * (ptop + 1);
  const uint8_t kuiT2		= * (ptop + 2);
  const uint8_t kuiT3		= * (ptop + 3);
  const uint16_t kuiT01	= 1 + kuiT0 + kuiT1;
  const uint16_t kuiT12	= 1 + kuiT1 + kuiT2;
  const uint16_t kuiT23	= 1 + kuiT2 + kuiT3;
  const uint16_t kuiT33	= 1 + (kuiT3 << 1);
  const uint8_t kuiDLT0	= (kuiT01 + kuiT12) >> 2;	// kDLT0
  const uint8_t kuiDLT1	= (kuiT12 + kuiT23) >> 2;	// kDLT1
  const uint8_t kuiDLT2	= (kuiT23 + kuiT33) >> 2;	// kDLT2
  const uint8_t kuiDLT3	= kuiT33 >> 1;			// kDLT3
  const uint8_t kuiList[8] = { kuiDLT0, kuiDLT1, kuiDLT2, kuiDLT3, kuiDLT3, kuiDLT3, kuiDLT3 , kuiDLT3 };

  ST32 (pPred,           LD32 (kuiList));
  ST32 (pPred + kiStride,  LD32 (kuiList + 1));
  ST32 (pPred + kiStride2, LD32 (kuiList + 2));
  ST32 (pPred + kiStride3, LD32 (kuiList + 3));
}


/*down right*/
void WelsI4x4LumaPredDDR_ref (uint8_t* pPred, const int32_t kiStride) {
  const int32_t kiStride2	= kiStride << 1;
  const int32_t kiStride3	= kiStride + kiStride2;
  uint8_t* ptopleft		= &pPred[- (kiStride + 1)];
  uint8_t* pleft			= &pPred[-1];
  const uint8_t kuiLT		= *ptopleft;
  /*get pLeft and pTop*/
  const uint8_t kuiL0		= *pleft;
  const uint8_t kuiL1		= * (pleft + kiStride);
  const uint8_t kuiL2		= * (pleft + kiStride2);
  const uint8_t kuiL3		= * (pleft + kiStride3);
  const uint8_t kuiT0		= * (ptopleft + 1);
  const uint8_t kuiT1		= * (ptopleft + 2);
  const uint8_t kuiT2		= * (ptopleft + 3);
  const uint8_t kuiT3		= * (ptopleft + 4);
  const uint16_t kuiTL0	= 1 + kuiLT + kuiL0;
  const uint16_t kuiLT0	= 1 + kuiLT + kuiT0;
  const uint16_t kuiT01	= 1 + kuiT0 + kuiT1;
  const uint16_t kuiT12	= 1 + kuiT1 + kuiT2;
  const uint16_t kuiT23	= 1 + kuiT2 + kuiT3;
  const uint16_t kuiL01	= 1 + kuiL0 + kuiL1;
  const uint16_t kuiL12	= 1 + kuiL1 + kuiL2;
  const uint16_t kuiL23	= 1 + kuiL2 + kuiL3;
  const uint8_t kuiDDR0	= (kuiTL0 + kuiLT0) >> 2;	// kuiDDR0
  const uint8_t kuiDDR1	= (kuiLT0 + kuiT01) >> 2;	// kuiDDR1
  const uint8_t kuiDDR2	= (kuiT01 + kuiT12) >> 2;	// kuiDDR2
  const uint8_t kuiDDR3	= (kuiT12 + kuiT23) >> 2;	// kuiDDR3
  const uint8_t kuiDDR4	= (kuiTL0 + kuiL01) >> 2;	// kuiDDR4
  const uint8_t kuiDDR5	= (kuiL01 + kuiL12) >> 2;	// kuiDDR5
  const uint8_t kuiDDR6	= (kuiL12 + kuiL23) >> 2;	// kuiDDR6
  const uint8_t kuiList[8] = { kuiDDR6, kuiDDR5, kuiDDR4, kuiDDR0, kuiDDR1, kuiDDR2, kuiDDR3, 0	};

  ST32 (pPred          , LD32 (kuiList + 3));
  ST32 (pPred + kiStride , LD32 (kuiList + 2));
  ST32 (pPred + kiStride2, LD32 (kuiList + 1));
  ST32 (pPred + kiStride3, LD32 (kuiList));
}


/*vertical pLeft*/
void WelsI4x4LumaPredVL_ref (uint8_t* pPred, const int32_t kiStride) {
  const int32_t kiStride2	= kiStride << 1;
  const int32_t kiStride3	= kiStride + kiStride2;
  uint8_t* ptopleft		= &pPred[- (kiStride + 1)];
  /*get pTop*/
  const uint8_t kuiT0		    = * (ptopleft + 1);
  const uint8_t kuiT1		    = * (ptopleft + 2);
  const uint8_t kuiT2		    = * (ptopleft + 3);
  const uint8_t kuiT3		    = * (ptopleft + 4);
  const uint8_t kuiT4		    = * (ptopleft + 5);
  const uint8_t kuiT5		    = * (ptopleft + 6);
  const uint8_t kuiT6		    = * (ptopleft + 7);
  const uint16_t kuiT01		= 1 + kuiT0 + kuiT1;
  const uint16_t kuiT12		= 1 + kuiT1 + kuiT2;
  const uint16_t kuiT23		= 1 + kuiT2 + kuiT3;
  const uint16_t kuiT34		= 1 + kuiT3 + kuiT4;
  const uint16_t kuiT45		= 1 + kuiT4 + kuiT5;
  const uint16_t kuiT56		= 1 + kuiT5 + kuiT6;
  const uint8_t kuiVL0		= kuiT01 >> 1;			// kuiVL0
  const uint8_t kuiVL1		= kuiT12 >> 1;			// kuiVL1
  const uint8_t kuiVL2		= kuiT23 >> 1;			// kuiVL2
  const uint8_t kuiVL3		= kuiT34 >> 1;			// kuiVL3
  const uint8_t kuiVL4		= kuiT45 >> 1;			// kuiVL4
  const uint8_t kuiVL5		= (kuiT01 + kuiT12) >> 2;	// kuiVL5
  const uint8_t kuiVL6		= (kuiT12 + kuiT23) >> 2;	// kuiVL6
  const uint8_t kuiVL7		= (kuiT23 + kuiT34) >> 2;	// kuiVL7
  const uint8_t kuiVL8		= (kuiT34 + kuiT45) >> 2;	// kuiVL8
  const uint8_t kuiVL9		= (kuiT45 + kuiT56) >> 2;	// kuiVL9
  const uint8_t kuiList[10]	= { kuiVL0, kuiVL1, kuiVL2, kuiVL3, kuiVL4, kuiVL5, kuiVL6, kuiVL7, kuiVL8, kuiVL9 };

  ST32 (pPred,           LD32 (kuiList));
  ST32 (pPred + kiStride,  LD32 (kuiList + 5));
  ST32 (pPred + kiStride2, LD32 (kuiList + 1));
  ST32 (pPred + kiStride3, LD32 (kuiList + 6));
}

/*vertical pLeft*/
void WelsI4x4LumaPredVLTop_ref (uint8_t* pPred, const int32_t kiStride) {
  const int32_t kiStride2	    = kiStride << 1;
  const int32_t kiStride3	    = kiStride + kiStride2;
  uint8_t* ptopleft		    = &pPred[- (kiStride + 1)];
  /*get pTop*/
  const uint8_t kuiT0		    = * (ptopleft + 1);
  const uint8_t kuiT1		    = * (ptopleft + 2);
  const uint8_t kuiT2		    = * (ptopleft + 3);
  const uint8_t kuiT3		    = * (ptopleft + 4);
  const uint16_t kuiT01		= 1 + kuiT0 + kuiT1;
  const uint16_t kuiT12		= 1 + kuiT1 + kuiT2;
  const uint16_t kuiT23		= 1 + kuiT2 + kuiT3;
  const uint16_t kuiT33		= 1 + (kuiT3 << 1);
  const uint8_t kuiVL0		= kuiT01 >> 1;
  const uint8_t kuiVL1		= kuiT12 >> 1;
  const uint8_t kuiVL2		= kuiT23 >> 1;
  const uint8_t kuiVL3		= kuiT33 >> 1;
  const uint8_t kuiVL4		= (kuiT01 + kuiT12) >> 2;
  const uint8_t kuiVL5		= (kuiT12 + kuiT23) >> 2;
  const uint8_t kuiVL6		= (kuiT23 + kuiT33) >> 2;
  const uint8_t kuiVL7		= kuiVL3;
  const uint8_t kuiList[10]	= { kuiVL0, kuiVL1, kuiVL2, kuiVL3, kuiVL3, kuiVL4, kuiVL5, kuiVL6, kuiVL7, kuiVL7 };

  ST32 (pPred          , LD32 (kuiList));
  ST32 (pPred + kiStride , LD32 (kuiList + 5));
  ST32 (pPred + kiStride2, LD32 (kuiList + 1));
  ST32 (pPred + kiStride3, LD32 (kuiList + 6));
}


/*vertical right*/
void WelsI4x4LumaPredVR_ref (uint8_t* pPred, const int32_t kiStride) {
  const int32_t kiStride2	    = kiStride << 1;
  const int32_t kiStride3	    = kiStride + kiStride2;
  const uint8_t kuiLT		    = pPred[-kiStride - 1];
  /*get pLeft and pTop*/
  const uint8_t kuiL0		    = pPred[         -1];
  const uint8_t kuiL1		    = pPred[kiStride - 1];
  const uint8_t kuiL2		    = pPred[kiStride2 - 1];
  const uint8_t kuiT0		    = pPred[ -kiStride];
  const uint8_t kuiT1		    = pPred[1 - kiStride];
  const uint8_t kuiT2		    = pPred[2 - kiStride];
  const uint8_t kuiT3		    = pPred[3 - kiStride];
  const uint8_t kuiVR0		= (1 + kuiLT + kuiT0) >> 1;	// kuiVR0
  const uint8_t kuiVR1		= (1 + kuiT0 + kuiT1) >> 1;	// kuiVR1
  const uint8_t kuiVR2		= (1 + kuiT1 + kuiT2) >> 1;	// kuiVR2
  const uint8_t kuiVR3		= (1 + kuiT2 + kuiT3) >> 1;	// kuiVR3
  const uint8_t kuiVR4		= (2 + kuiL0 + (kuiLT << 1) + kuiT0) >> 2;	// kuiVR4
  const uint8_t kuiVR5		= (2 + kuiLT + (kuiT0 << 1) + kuiT1) >> 2;	// kuiVR5
  const uint8_t kuiVR6		= (2 + kuiT0 + (kuiT1 << 1) + kuiT2) >> 2;	// kuiVR6
  const uint8_t kuiVR7		= (2 + kuiT1 + (kuiT2 << 1) + kuiT3) >> 2;	// kuiVR7
  const uint8_t kuiVR8		= (2 + kuiLT + (kuiL0 << 1) + kuiL1) >> 2;	// kuiVR8
  const uint8_t kuiVR9		= (2 + kuiL0 + (kuiL1 << 1) + kuiL2) >> 2;	// kuiVR9
  const uint8_t kuiList[10]	= { kuiVR8, kuiVR0, kuiVR1, kuiVR2, kuiVR3, kuiVR9, kuiVR4, kuiVR5, kuiVR6, kuiVR7 };

  ST32 (pPred          , LD32 (kuiList + 1));
  ST32 (pPred + kiStride , LD32 (kuiList + 6));
  ST32 (pPred + kiStride2, LD32 (kuiList));
  ST32 (pPred + kiStride3, LD32 (kuiList + 5));
}

/*horizontal up*/
void WelsI4x4LumaPredHU_ref (uint8_t* pPred, const int32_t kiStride) {
  const int32_t kiStride2	    = kiStride << 1;
  const int32_t kiStride3	    = kiStride + kiStride2;
  /*get pLeft*/
  const uint8_t kuiL0		    = pPred[         -1];
  const uint8_t kuiL1		    = pPred[kiStride - 1];
  const uint8_t kuiL2		    = pPred[kiStride2 - 1];
  const uint8_t kuiL3		    = pPred[kiStride3 - 1];
  const uint16_t kuiL01		= 1 + kuiL0 + kuiL1;
  const uint16_t kuiL12		= 1 + kuiL1 + kuiL2;
  const uint16_t kuiL23		= 1 + kuiL2 + kuiL3;
  const uint8_t kuiHU0		= kuiL01 >> 1;
  const uint8_t kuiHU1		= (kuiL01 + kuiL12) >> 2;
  const uint8_t kuiHU2		= kuiL12 >> 1;
  const uint8_t kuiHU3		= (kuiL12 + kuiL23) >> 2;
  const uint8_t kuiHU4		= kuiL23 >> 1;
  const uint8_t kuiHU5		= (1 + kuiL23 + (kuiL3 << 1)) >> 2;
  const uint8_t kuiList[10]	= { kuiHU0, kuiHU1, kuiHU2, kuiHU3, kuiHU4, kuiHU5, kuiL3, kuiL3, kuiL3, kuiL3 };

  ST32 (pPred          , LD32 (kuiList));
  ST32 (pPred + kiStride , LD32 (kuiList + 2));
  ST32 (pPred + kiStride2, LD32 (kuiList + 4));
  ST32 (pPred + kiStride3, LD32 (kuiList + 6));
}

/*horizontal down*/
void WelsI4x4LumaPredHD_ref (uint8_t* pPred, const int32_t kiStride) {
  const int32_t kiStride2 	= kiStride << 1;
  const int32_t kiStride3	    = kiStride + kiStride2;
  const uint8_t kuiLT		    = pPred[- (kiStride + 1)];
  /*get pLeft and pTop*/
  const uint8_t kuiL0		    = pPred[-1          ];
  const uint8_t kuiL1		    = pPred[-1 + kiStride ];
  const uint8_t kuiL2		    = pPred[-1 + kiStride2];
  const uint8_t kuiL3		    = pPred[-1 + kiStride3];
  const uint8_t kuiT0		    = pPred[-kiStride   ];
  const uint8_t kuiT1		    = pPred[-kiStride + 1 ];
  const uint8_t kuiT2		    = pPred[-kiStride + 2 ];
  const uint16_t kuiTL0		= 1 + kuiLT + kuiL0;
  const uint16_t kuiLT0		= 1 + kuiLT + kuiT0;
  const uint16_t kuiT01		= 1 + kuiT0 + kuiT1;
  const uint16_t kuiT12		= 1 + kuiT1 + kuiT2;
  const uint16_t kuiL01		= 1 + kuiL0 + kuiL1;
  const uint16_t kuiL12		= 1 + kuiL1 + kuiL2;
  const uint16_t kuiL23		= 1 + kuiL2 + kuiL3;
  const uint8_t kuiHD0		= kuiTL0 >> 1;
  const uint8_t kuiHD1		= (kuiTL0 + kuiLT0) >> 2;
  const uint8_t kuiHD2		= (kuiLT0 + kuiT01) >> 2;
  const uint8_t kuiHD3		= (kuiT01 + kuiT12) >> 2;
  const uint8_t kuiHD4		= kuiL01 >> 1;
  const uint8_t kuiHD5		= (kuiTL0 + kuiL01) >> 2;
  const uint8_t kuiHD6		= kuiL12 >> 1;
  const uint8_t kuiHD7		= (kuiL01 + kuiL12) >> 2;
  const uint8_t kuiHD8		= kuiL23 >> 1;
  const uint8_t kuiHD9	    = (kuiL12 + kuiL23) >> 2;
  const uint8_t kuiList[10]	= { kuiHD8, kuiHD9, kuiHD6, kuiHD7, kuiHD4, kuiHD5, kuiHD0, kuiHD1, kuiHD2, kuiHD3 };

  ST32 (pPred          , LD32 (kuiList + 6));
  ST32 (pPred + kiStride , LD32 (kuiList + 4));
  ST32 (pPred + kiStride2, LD32 (kuiList + 2));
  ST32 (pPred + kiStride3, LD32 (kuiList));
}
// Unit test for Luma 4x4 cases
PREDV (4)
GENERATE_4x4_UT (WelsI4x4LumaPredV_c, LumaI4x4PredV, 0, 0)

PREDH (4)
GENERATE_4x4_UT (WelsI4x4LumaPredH_c, LumaI4x4PredH, 0, 0)

PREDDC (4, 2)
GENERATE_4x4_UT (WelsI4x4LumaPredDc_c, LumaI4x4PredDC, 0, 0)

PREDDCLeft (4, 2)
GENERATE_4x4_UT (WelsI4x4LumaPredDcLeft_c, LumaI4x4PredDCLeft, 0, 0)

PREDDCTop (4, 2)
GENERATE_4x4_UT (WelsI4x4LumaPredDcTop_c, LumaI4x4PredDCTop, 0, 0)

PREDDCNone (4, 2)
GENERATE_4x4_UT (WelsI4x4LumaPredDcNA_c, LumaI4x4PredDCNone, 0, 0)
GENERATE_4x4_UT (WelsI4x4LumaPredDDL_c, WelsI4x4LumaPredDDL_ref, 0, 0)
GENERATE_4x4_UT (WelsI4x4LumaPredDDLTop_c, WelsI4x4LumaPredDDLTop_ref, 0, 0)
GENERATE_4x4_UT (WelsI4x4LumaPredDDR_c, WelsI4x4LumaPredDDR_ref, 0, 0)
GENERATE_4x4_UT (WelsI4x4LumaPredVR_c, WelsI4x4LumaPredVR_ref, 0, 0)
GENERATE_4x4_UT (WelsI4x4LumaPredVL_c, WelsI4x4LumaPredVL_ref, 0, 0)
GENERATE_4x4_UT (WelsI4x4LumaPredVLTop_c, WelsI4x4LumaPredVLTop_ref, 0, 0)
GENERATE_4x4_UT (WelsI4x4LumaPredHU_c, WelsI4x4LumaPredHU_ref, 0, 0)
GENERATE_4x4_UT (WelsI4x4LumaPredHD_c, WelsI4x4LumaPredHD_ref, 0, 0)

#define GENERATE_8x8_UT(pred, ref, ASM, CPUFLAGS) \
TEST(DecoderIntraPredictionTest, pred) {\
const int32_t kiStride = 32; \
int iRunTimes = 1000; \
ENFORCE_STACK_ALIGN_1D (uint8_t, pRefBuffer, 18 * kiStride, 16); \
ENFORCE_STACK_ALIGN_1D (uint8_t, pPredBuffer, 18 * kiStride, 16); \
if (ASM) { \
  int32_t iTmp = 1; \
  uint32_t uiCPUFlags = WelsCPUFeatureDetect(&iTmp); \
  if ((uiCPUFlags & CPUFLAGS) == 0) {\
    return; \
  } \
} \
while(iRunTimes--) {\
for (int i = 0; i < 17; i ++) {\
  pRefBuffer[kiStride + i] = pPredBuffer[kiStride + i] = rand() & 255; \
  pRefBuffer[(i+1) * kiStride - 1] = pPredBuffer[(i+1) * kiStride - 1] = rand() & 255; \
}\
pred(&pPredBuffer[2*kiStride], kiStride); \
ref(&pRefBuffer[2*kiStride], kiStride); \
bool ok = true; \
for (int i = 0; i < 8; i ++)\
  for(int j = 0; j < 8; j ++)\
    if (pPredBuffer[(i+2) * kiStride + j] != pRefBuffer[(i+2) * kiStride + j]) {\
      ok = false; \
      break; \
    } \
    EXPECT_EQ(ok, true); \
} \
}

void WelsIChromaPredPlane_ref (uint8_t* pPred, const int32_t kiStride) {
  int32_t a = 0, b = 0, c = 0, H = 0, V = 0;
  int32_t i, j;
  uint8_t* pTop = &pPred[-kiStride];
  uint8_t* pLeft = &pPred[-1];

  for (i = 0 ; i < 4 ; i ++) {
    H += (i + 1) * (pTop[4 + i] - pTop[2 - i]);
    V += (i + 1) * (pLeft[ (4 + i) * kiStride] - pLeft[ (2 - i) * kiStride]);
  }

  a = (pLeft[7 * kiStride] + pTop[7]) << 4;
  b = (17 * H + 16) >> 5;
  c = (17 * V + 16) >> 5;

  for (i = 0 ; i < 8 ; i ++) {
    for (j = 0 ; j < 8 ; j ++) {
      int32_t iTmp = (a + b * (j - 3) + c * (i - 3) + 16) >> 5;
      pPred[j] = (iTmp < 0) ? 0 : ((iTmp > 255) ? 255 : iTmp);
    }
    pPred += kiStride;
  }
}


void WelsIChromaPredDc_ref (uint8_t* pPred, const int32_t kiStride) {
  const int32_t kiL1		= kiStride - 1;
  const int32_t kiL2		= kiL1 + kiStride;
  const int32_t kiL3		= kiL2 + kiStride;
  const int32_t kiL4		= kiL3 + kiStride;
  const int32_t kiL5		= kiL4 + kiStride;
  const int32_t kiL6		= kiL5 + kiStride;
  const int32_t kiL7		= kiL6 + kiStride;
  /*caculate the kMean value*/
  const uint8_t kuiM1		= (pPred[-kiStride] + pPred[1 - kiStride] + pPred[2 - kiStride] + pPred[3 - kiStride] +
                           pPred[-1] + pPred[kiL1] + pPred[kiL2] + pPred[kiL3] + 4) >> 3 ;
  const uint32_t kuiSum2	= pPred[4 - kiStride] + pPred[5 - kiStride] + pPred[6 - kiStride] + pPred[7 - kiStride];
  const uint32_t kuiSum3	= pPred[kiL4] + pPred[kiL5] + pPred[kiL6] + pPred[kiL7];
  const uint8_t kuiM2		= (kuiSum2 + 2) >> 2;
  const uint8_t kuiM3		= (kuiSum3 + 2) >> 2;
  const uint8_t kuiM4		= (kuiSum2 + kuiSum3 + 4) >> 3;
  const uint8_t kuiMUP[8]	= {kuiM1, kuiM1, kuiM1, kuiM1, kuiM2, kuiM2, kuiM2, kuiM2};
  const uint8_t kuiMDown[8]	= {kuiM3, kuiM3, kuiM3, kuiM3, kuiM4, kuiM4, kuiM4, kuiM4};
  const uint64_t kuiUP64		= LD64 (kuiMUP);
  const uint64_t kuiDN64		= LD64 (kuiMDown);

  ST64 (pPred       , kuiUP64);
  ST64 (pPred + kiL1 + 1, kuiUP64);
  ST64 (pPred + kiL2 + 1, kuiUP64);
  ST64 (pPred + kiL3 + 1, kuiUP64);
  ST64 (pPred + kiL4 + 1, kuiDN64);
  ST64 (pPred + kiL5 + 1, kuiDN64);
  ST64 (pPred + kiL6 + 1, kuiDN64);
  ST64 (pPred + kiL7 + 1, kuiDN64);
}

void WelsIChromaPredDcLeft_ref (uint8_t* pPred, const int32_t kiStride) {
  const int32_t kiL1	=   -1 + kiStride;
  const int32_t kiL2	= kiL1 + kiStride;
  const int32_t kiL3	= kiL2 + kiStride;
  const int32_t kiL4	= kiL3 + kiStride;
  const int32_t kiL5	= kiL4 + kiStride;
  const int32_t kiL6	= kiL5 + kiStride;
  const int32_t kiL7	= kiL6 + kiStride;
  /*caculate the kMean value*/
  const uint8_t kuiMUP   = (pPred[-1] + pPred[kiL1] + pPred[kiL2] + pPred[kiL3] + 2) >> 2 ;
  const uint8_t kuiMDown = (pPred[kiL4] + pPred[kiL5] + pPred[kiL6] + pPred[kiL7] + 2) >> 2;
  const uint64_t kuiUP64 = 0x0101010101010101ULL * kuiMUP;
  const uint64_t kuiDN64 = 0x0101010101010101ULL * kuiMDown;

  ST64 (pPred       , kuiUP64);
  ST64 (pPred + kiL1 + 1, kuiUP64);
  ST64 (pPred + kiL2 + 1, kuiUP64);
  ST64 (pPred + kiL3 + 1, kuiUP64);
  ST64 (pPred + kiL4 + 1, kuiDN64);
  ST64 (pPred + kiL5 + 1, kuiDN64);
  ST64 (pPred + kiL6 + 1, kuiDN64);
  ST64 (pPred + kiL7 + 1, kuiDN64);
}

void WelsIChromaPredDcTop_ref (uint8_t* pPred, const int32_t kiStride) {
  int32_t iTmp			= (kiStride << 3) - kiStride;
  /*caculate the kMean value*/
  const uint8_t kuiM1	    = (pPred[-kiStride] + pPred[1 - kiStride] + pPred[2 - kiStride] + pPred[3 - kiStride] + 2) >> 2;
  const uint8_t kuiM2	    = (pPred[4 - kiStride] + pPred[5 - kiStride] + pPred[6 - kiStride] + pPred[7 - kiStride] + 2) >>
                            2;
  const uint8_t kuiM[8]	= {kuiM1, kuiM1, kuiM1, kuiM1, kuiM2, kuiM2, kuiM2, kuiM2};

  uint8_t i = 7;

  do {
    ST64 (pPred + iTmp, LD64 (kuiM));

    iTmp -= kiStride;
  } while (i-- > 0);
}
PREDV (8)
PREDH (8)
PREDDCNone (8, 3)
GENERATE_8x8_UT (WelsIChromaPredDcNA_c, LumaI8x8PredDCNone, 0, 0)
GENERATE_8x8_UT (WelsIChromaPredPlane_c, WelsIChromaPredPlane_ref, 0, 0)
GENERATE_8x8_UT (WelsIChromaPredDc_c, WelsIChromaPredDc_ref, 0, 0)
GENERATE_8x8_UT (WelsIChromaPredDcTop_c, WelsIChromaPredDcTop_ref, 0, 0)
GENERATE_8x8_UT (WelsIChromaPredDcLeft_c, WelsIChromaPredDcLeft_ref, 0, 0)
GENERATE_8x8_UT (WelsIChromaPredH_c, LumaI8x8PredH, 0, 0)
GENERATE_8x8_UT (WelsIChromaPredV_c, LumaI8x8PredV, 0, 0)
#define GENERATE_16x16_UT(pred, ref, ASM, CPUFLAGS) \
TEST(DecoderIntraPredictionTest, pred) {\
const int32_t kiStride = 32; \
int32_t iRunTimes = 1000; \
ENFORCE_STACK_ALIGN_1D (uint8_t, pRefBuffer, 18 * kiStride, 16); \
ENFORCE_STACK_ALIGN_1D (uint8_t, pPredBuffer, 18 * kiStride, 16); \
if (ASM) { \
  int32_t iTmp = 1; \
  uint32_t uiCPUFlags = WelsCPUFeatureDetect( &iTmp); \
  if ((uiCPUFlags & CPUFLAGS) == 0) {\
    return ; \
  } \
}\
while(iRunTimes--) {\
for (int i = 0; i < 17; i ++) {\
  pRefBuffer[kiStride + i] = pPredBuffer[kiStride + i] = rand() & 255; \
  pRefBuffer[(i+1) * kiStride - 1] = pPredBuffer[(i+1) * kiStride - 1] = rand() & 255; \
}\
pred(&pPredBuffer[2*kiStride], kiStride); \
ref(&pRefBuffer[2*kiStride], kiStride); \
bool ok = true; \
for (int i = 0; i < 16; i ++)\
  for(int j = 0; j < 16; j ++)\
    if (pPredBuffer[(i+2) * kiStride + j] != pRefBuffer[(i+2) * kiStride + j]) {\
      ok = false; \
      break; \
    } \
    EXPECT_EQ(ok, true); \
} \
}
void WelsI16x16LumaPredPlane_ref (uint8_t* pPred, const int32_t kiStride) {
  int32_t a = 0, b = 0, c = 0, H = 0, V = 0;
  int32_t i, j;
  uint8_t* pTop = &pPred[-kiStride];
  uint8_t* pLeft = &pPred[-1];

  for (i = 0 ; i < 8 ; i ++) {
    H += (i + 1) * (pTop[8 + i] - pTop[6 - i]);
    V += (i + 1) * (pLeft[ (8 + i) * kiStride] - pLeft[ (6 - i) * kiStride]);
  }

  a = (pLeft[15 * kiStride] + pTop[15]) << 4;
  b = (5 * H + 32) >> 6;
  c = (5 * V + 32) >> 6;

  for (i = 0 ; i < 16 ; i ++) {
    for (j = 0 ; j < 16 ; j ++) {
      int32_t iTmp = (a + b * (j - 7) + c * (i - 7) + 16) >> 5;
      pPred[j] = (iTmp < 0) ? 0 : ((iTmp > 255) ? 255 : iTmp);
    }
    pPred += kiStride;
  }
}

PREDV (16)
PREDH (16)
PREDDC (16, 4)
PREDDCTop (16, 4)
PREDDCLeft (16, 4)
PREDDCNone (16, 4)

GENERATE_16x16_UT (WelsI16x16LumaPredDcNA_c, LumaI16x16PredDCNone, 0, 0)
GENERATE_16x16_UT (WelsI16x16LumaPredPlane_c, WelsI16x16LumaPredPlane_ref, 0, 0)
GENERATE_16x16_UT (WelsI16x16LumaPredDcLeft_c, LumaI16x16PredDCLeft, 0, 0)
GENERATE_16x16_UT (WelsI16x16LumaPredDcTop_c, LumaI16x16PredDCTop, 0, 0)
GENERATE_16x16_UT (WelsI16x16LumaPredDc_c, LumaI16x16PredDC, 0, 0)
GENERATE_16x16_UT (WelsI16x16LumaPredH_c, LumaI16x16PredH, 0, 0)
GENERATE_16x16_UT (WelsI16x16LumaPredV_c, LumaI16x16PredV, 0, 0)
#if defined(X86_ASM)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredH_sse2, LumaI4x4PredH, 1, WELS_CPU_SSE2)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredDDR_mmx, WelsI4x4LumaPredDDR_ref, 1, WELS_CPU_MMX)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredHD_mmx, WelsI4x4LumaPredHD_ref, 1, WELS_CPU_MMX)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredHU_mmx, WelsI4x4LumaPredHU_ref, 1, WELS_CPU_MMX)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredVR_mmx, WelsI4x4LumaPredVR_ref, 1, WELS_CPU_MMX)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredDDL_mmx, WelsI4x4LumaPredDDL_ref, 1, WELS_CPU_MMX)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredVL_mmx, WelsI4x4LumaPredVL_ref, 1, WELS_CPU_MMX)
GENERATE_8x8_UT (WelsDecoderIChromaPredDcTop_sse2, WelsIChromaPredDcTop_ref, 1, WELS_CPU_SSE2)
GENERATE_8x8_UT (WelsDecoderIChromaPredDc_sse2, WelsIChromaPredDc_ref, 1, WELS_CPU_SSE2)
GENERATE_8x8_UT (WelsDecoderIChromaPredPlane_sse2, WelsIChromaPredPlane_ref, 1, WELS_CPU_SSE2)
GENERATE_8x8_UT (WelsDecoderIChromaPredH_mmx, LumaI8x8PredH, 1, WELS_CPU_MMX)
GENERATE_8x8_UT (WelsDecoderIChromaPredV_mmx, LumaI8x8PredV, 1, WELS_CPU_MMX)
GENERATE_8x8_UT (WelsDecoderIChromaPredDcLeft_mmx, WelsIChromaPredDcLeft_ref, 1, WELS_CPU_MMX)
GENERATE_8x8_UT (WelsDecoderIChromaPredDcNA_mmx, LumaI8x8PredDCNone, 1, WELS_CPU_MMX)
GENERATE_16x16_UT (WelsDecoderI16x16LumaPredPlane_sse2, WelsI16x16LumaPredPlane_ref, 1, WELS_CPU_SSE2)
GENERATE_16x16_UT (WelsDecoderI16x16LumaPredH_sse2, LumaI16x16PredH, 1, WELS_CPU_SSE2)
GENERATE_16x16_UT (WelsDecoderI16x16LumaPredV_sse2, LumaI16x16PredV, 1, WELS_CPU_SSE2)
GENERATE_16x16_UT (WelsDecoderI16x16LumaPredDc_sse2, LumaI16x16PredDC, 1, WELS_CPU_SSE2)
GENERATE_16x16_UT (WelsDecoderI16x16LumaPredDcTop_sse2, LumaI16x16PredDCTop, 1, WELS_CPU_SSE2)
GENERATE_16x16_UT (WelsDecoderI16x16LumaPredDcNA_sse2, LumaI16x16PredDCNone, 1, WELS_CPU_SSE2)
#endif

#if defined(HAVE_NEON)
GENERATE_16x16_UT (WelsDecoderI16x16LumaPredV_neon, LumaI16x16PredV, 1, WELS_CPU_NEON)
GENERATE_16x16_UT (WelsDecoderI16x16LumaPredH_neon, LumaI16x16PredH, 1, WELS_CPU_NEON)
GENERATE_16x16_UT (WelsDecoderI16x16LumaPredDc_neon, LumaI16x16PredDC, 1, WELS_CPU_NEON)
GENERATE_16x16_UT (WelsDecoderI16x16LumaPredPlane_neon, WelsI16x16LumaPredPlane_ref, 1, WELS_CPU_NEON)

GENERATE_4x4_UT (WelsDecoderI4x4LumaPredV_neon, LumaI4x4PredV, 1, WELS_CPU_NEON)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredH_neon, LumaI4x4PredH, 1, WELS_CPU_NEON)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredDDL_neon, WelsI4x4LumaPredDDL_ref, 1, WELS_CPU_NEON)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredDDR_neon, WelsI4x4LumaPredDDR_ref, 1, WELS_CPU_NEON)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredVL_neon, WelsI4x4LumaPredVL_ref, 1, WELS_CPU_NEON)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredVR_neon, WelsI4x4LumaPredVR_ref, 1, WELS_CPU_NEON)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredHU_neon, WelsI4x4LumaPredHU_ref, 1, WELS_CPU_NEON)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredHD_neon, WelsI4x4LumaPredHD_ref, 1, WELS_CPU_NEON)

GENERATE_8x8_UT (WelsDecoderIChromaPredV_neon, LumaI8x8PredV, 1, WELS_CPU_NEON)
GENERATE_8x8_UT (WelsDecoderIChromaPredH_neon, LumaI8x8PredH, 1, WELS_CPU_NEON)
GENERATE_8x8_UT (WelsDecoderIChromaPredDc_neon, WelsIChromaPredDc_ref, 1, WELS_CPU_NEON)
GENERATE_8x8_UT (WelsDecoderIChromaPredPlane_neon, WelsIChromaPredPlane_ref, 1, WELS_CPU_NEON)
#endif

#if defined(HAVE_NEON_AARCH64)
GENERATE_16x16_UT (WelsDecoderI16x16LumaPredV_AArch64_neon, LumaI16x16PredV, 1, WELS_CPU_NEON)
GENERATE_16x16_UT (WelsDecoderI16x16LumaPredH_AArch64_neon, LumaI16x16PredH, 1, WELS_CPU_NEON)
GENERATE_16x16_UT (WelsDecoderI16x16LumaPredDc_AArch64_neon, LumaI16x16PredDC, 1, WELS_CPU_NEON)
GENERATE_16x16_UT (WelsDecoderI16x16LumaPredDcTop_AArch64_neon, LumaI16x16PredDCTop, 1, WELS_CPU_NEON)
GENERATE_16x16_UT (WelsDecoderI16x16LumaPredDcLeft_AArch64_neon, LumaI16x16PredDCLeft, 1, WELS_CPU_NEON)
GENERATE_16x16_UT (WelsDecoderI16x16LumaPredPlane_AArch64_neon, WelsI16x16LumaPredPlane_ref, 1, WELS_CPU_NEON)

GENERATE_4x4_UT (WelsDecoderI4x4LumaPredH_AArch64_neon, LumaI4x4PredH, 1, WELS_CPU_NEON)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredDDL_AArch64_neon, WelsI4x4LumaPredDDL_ref, 1, WELS_CPU_NEON)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredDDLTop_AArch64_neon, WelsI4x4LumaPredDDLTop_ref, 1, WELS_CPU_NEON)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredVL_AArch64_neon, WelsI4x4LumaPredVL_ref, 1, WELS_CPU_NEON)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredVLTop_AArch64_neon, WelsI4x4LumaPredVLTop_ref, 1, WELS_CPU_NEON)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredVR_AArch64_neon, WelsI4x4LumaPredVR_ref, 1, WELS_CPU_NEON)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredHU_AArch64_neon, WelsI4x4LumaPredHU_ref, 1, WELS_CPU_NEON)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredHD_AArch64_neon, WelsI4x4LumaPredHD_ref, 1, WELS_CPU_NEON)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredDc_AArch64_neon, LumaI4x4PredDC, 1, WELS_CPU_NEON)
GENERATE_4x4_UT (WelsDecoderI4x4LumaPredDcTop_AArch64_neon, LumaI4x4PredDCTop, 1, WELS_CPU_NEON)

GENERATE_8x8_UT (WelsDecoderIChromaPredV_AArch64_neon, LumaI8x8PredV, 1, WELS_CPU_NEON)
GENERATE_8x8_UT (WelsDecoderIChromaPredH_AArch64_neon, LumaI8x8PredH, 1, WELS_CPU_NEON)
GENERATE_8x8_UT (WelsDecoderIChromaPredDc_AArch64_neon, WelsIChromaPredDc_ref, 1, WELS_CPU_NEON)
GENERATE_8x8_UT (WelsDecoderIChromaPredPlane_AArch64_neon, WelsIChromaPredPlane_ref, 1, WELS_CPU_NEON)
GENERATE_8x8_UT (WelsDecoderIChromaPredDcTop_AArch64_neon, WelsIChromaPredDcTop_ref, 1, WELS_CPU_NEON)
#endif
