ref: 37deb06700b5e68d8c79d74e50426d5b55fff590
dir: /test/processing/ProcessUT_ScrollDetection.cpp/
#include <gtest/gtest.h> #include <math.h> #include <string.h> #include "cpu.h" #include "cpu_core.h" #include "IWelsVP.h" #include "ScrollDetection.h" #include "ScrollDetectionFuncs.h" #include "utils/DataGenerator.h" using namespace nsWelsVP; #define ASSERT_MEMORY_FAIL2X(A, B) \ if (NULL == B) { \ delete []A;\ ASSERT_TRUE(0); \ } TEST(ScrollDetectionTest,TestScroll) { unsigned char* pSrc, *pRef; int iWidthSets[4] = {640,1024,1280,1980}; int iHeightSets[4] = {360,768,720,1080}; int iStride = 0; int iIdx = 0; for(int i=0; i<4; i++){ int iWidth = iWidthSets[i]; int iHeight = iHeightSets[i]; iStride = iWidth + 16; pSrc = new unsigned char[iHeight*iStride]; ASSERT_TRUE(NULL != pSrc); pRef = new unsigned char[iHeight*iStride]; ASSERT_MEMORY_FAIL2X(pSrc, pRef) RandomPixelDataGenerator(pRef, iWidth, iHeight, iStride, iIdx ); int iMvRange = iHeight/3; int iScrollMv = rand()%(iMvRange<<1) - iMvRange; unsigned char* pSrcTmp = pSrc; unsigned char* pRefTmp = pRef; for (int j=0;j<iHeight;j++) { if ((j+iScrollMv)>=0 && (j+iScrollMv)<iHeight) for (int i=0;i<iWidth;i++) { memcpy(pSrcTmp , &pRefTmp[(j+iScrollMv)*iStride], iWidth*sizeof(unsigned char)); } else { for (int i=0;i<iWidth;i++) pSrcTmp[i] = rand()%256; } pSrcTmp += iStride; } SPixMap sSrcMap = { { 0 } }; SPixMap sRefMap = { { 0 } }; sSrcMap.pPixel[0] = pSrc; sRefMap.pPixel[0] = pRef; sSrcMap.iStride[0] = sRefMap.iStride[0] = iStride; sSrcMap.sRect.iRectWidth = sRefMap.sRect.iRectWidth = iWidth; sSrcMap.sRect.iRectHeight = sRefMap.sRect.iRectHeight = iHeight; SScrollDetectionParam sScrollDetectionResult; WelsMemset (&sScrollDetectionResult, 0, sizeof (sScrollDetectionResult)); int iCoreNum = 1; unsigned int uiCPUFlag = WelsCPUFeatureDetect (&iCoreNum); CScrollDetection *pTest =new CScrollDetection(uiCPUFlag); int iMethodIdx = METHOD_SCROLL_DETECTION; pTest->Set(iMethodIdx, (&sScrollDetectionResult)); int ret = pTest->Process(iMethodIdx,&sSrcMap, &sRefMap); EXPECT_EQ(ret,0); pTest->Get(iMethodIdx, (&sScrollDetectionResult)); EXPECT_EQ(sScrollDetectionResult.bScrollDetectFlag,1); EXPECT_EQ(sScrollDetectionResult.iScrollMvY,iScrollMv); delete pTest; delete []pSrc; delete []pRef; } }