generic-library/openh264
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #745 from lyao2/scrollingUT

Browse Source 
add scroll detection UT
This commit is contained in:
huili2 2014-04-25 15:07:15 +08:00
commit c0d21a23f3
5 changed files with 102 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
Makefile
View File

@ -70,7 +70,8 @@ ENCODER_INCLUDES = \
PROCESSING_INCLUDES = \
-Icodec/processing/interface \
-Icodec/processing/src/common
-Icodec/processing/src/common \
-Icodec/processing/src/scrolldetection
GTEST_INCLUDES += \
-Igtest \
@ -92,6 +93,7 @@ CODEC_UNITTEST_LDFLAGS = -L. $(call LINK_LIB,gtest) $(call LINK_LIB,decoder) $(c
CODEC_UNITTEST_DEPS = $(LIBPREFIX)gtest.$(LIBSUFFIX) $(LIBPREFIX)decoder.$(LIBSUFFIX) $(LIBPREFIX)encoder.$(LIBSUFFIX) $(LIBPREFIX)processing.$(LIBSUFFIX) $(LIBPREFIX)common.$(LIBSUFFIX)
DECODER_UNITTEST_INCLUDES = $(CODEC_UNITTEST_INCLUDES) $(DECODER_INCLUDES) -Itest -Itest/decoder
ENCODER_UNITTEST_INCLUDES = $(CODEC_UNITTEST_INCLUDES) $(ENCODER_INCLUDES) -Itest -Itest/encoder
PROCESSING_UNITTEST_INCLUDES = $(CODEC_UNITTEST_INCLUDES) $(PROCESSING_INCLUDES) -Itest -Itest/processing
API_TEST_INCLUDES = $(CODEC_UNITTEST_INCLUDES) -Itest -Itest/api
.PHONY: test gtest-bootstrap clean
@ -158,9 +160,10 @@ include build/gtest-targets.mk
include test/api/targets.mk
include test/decoder/targets.mk
include test/encoder/targets.mk
include test/processing/targets.mk
binaries: codec_unittest$(EXEEXT)
BINARIES += codec_unittest$(EXEEXT)
codec_unittest$(EXEEXT): $(DECODER_UNITTEST_OBJS) $(ENCODER_UNITTEST_OBJS) $(API_TEST_OBJS) $(CODEC_UNITTEST_DEPS)
codec_unittest$(EXEEXT): $(DECODER_UNITTEST_OBJS) $(ENCODER_UNITTEST_OBJS) $(PROCESSING_UNITTEST_OBJS) $(API_TEST_OBJS) $(CODEC_UNITTEST_DEPS)
$(QUIET)rm -f $@
$(QUIET_CXX)$(CXX) $(CXX_LINK_O) $+ $(CODEC_UNITTEST_LDFLAGS) $(LDFLAGS)
endif

1
build/mktargets.sh
View File

@ -9,5 +9,6 @@ python build/mktargets.py --directory codec/console/dec --binary h264dec
python build/mktargets.py --directory codec/console/enc --binary h264enc
python build/mktargets.py --directory test/encoder --prefix encoder_unittest
python build/mktargets.py --directory test/decoder --prefix decoder_unittest
python build/mktargets.py --directory test/processing --prefix processing_unittest
python build/mktargets.py --directory test/api --prefix api_test
python build/mktargets.py --directory gtest --library gtest --out build/gtest-targets.mk --cpp-suffix .cc --include gtest-all.cc

5
codec/processing/src/scrolldetection/ScrollDetection.cpp
View File

@ -38,9 +38,8 @@
WELSVP_NAMESPACE_BEGIN
EResult CScrollDetection::Process(int32_t iType, SPixMap* pSrcPixMap, SPixMap* pRefPixMap){
if (pRefPixMap->pPixel[0] == NULL || pRefPixMap->pPixel[1] == NULL || pRefPixMap->pPixel[2] == NULL
|| pSrcPixMap->pPixel[0] == NULL || pSrcPixMap->pPixel[1] == NULL || pSrcPixMap->pPixel[2] == NULL
|| pRefPixMap->sRect.iRectWidth != pSrcPixMap->sRect.iRectWidth || pRefPixMap->sRect.iRectHeight != pSrcPixMap->sRect.iRectHeight){
if (pRefPixMap->pPixel[0] == NULL || pSrcPixMap->pPixel[0] == NULL ||
pRefPixMap->sRect.iRectWidth != pSrcPixMap->sRect.iRectWidth || pRefPixMap->sRect.iRectHeight != pSrcPixMap->sRect.iRectHeight){
return RET_INVALIDPARAM;
}

84
test/processing/ProcessUT_ScrollDetection.cpp Normal file
View File

@ -0,0 +1,84 @@
#include <gtest/gtest.h>
#include <math.h>
#include <string.h>
#include "cpu.h"
#include "cpu_core.h"
#include "typedef.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)
{
uint8_t* pSrc, *pRef;
int32_t iWidthSets[4] = {640,1024,1280,1980};
int32_t iHeightSets[4] = {360,768,720,1080};
int32_t iStride = 0;
int32_t iIdx = 0;
for(int32_t i=0; i<4; i++){
int32_t iWidth = iWidthSets[i];
int32_t iHeight = iHeightSets[i];
iStride = iWidth + 16;
pSrc = new uint8_t[iHeight*iStride];
ASSERT_TRUE(NULL != pSrc);
pRef = new uint8_t[iHeight*iStride];
ASSERT_MEMORY_FAIL2X(pSrc, pRef)
RandomPixelDataGenerator(pRef, iWidth, iHeight, iStride, iIdx );
int32_t iMvRange = iHeight/3;
int32_t iScrollMv = rand()%(iMvRange<<1) - iMvRange;
uint8_t* pSrcTmp = pSrc;
uint8_t* pRefTmp = pRef;
for (int32_t j=0;j<iHeight;j++) {
if ((j+iScrollMv)>=0 && (j+iScrollMv)<iHeight)
for (int32_t i=0;i<iWidth;i++) {
memcpy(pSrcTmp , &pRefTmp[(j+iScrollMv)*iStride], iWidth*sizeof(uint8_t));
} else {
for (int32_t 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));
int32_t iCoreNum = 1;
uint32_t uiCPUFlag = WelsCPUFeatureDetect (&iCoreNum);
CScrollDetection *pTest =new CScrollDetection(uiCPUFlag);
int32_t iMethodIdx = METHOD_SCROLL_DETECTION;
pTest->Set(iMethodIdx, (&sScrollDetectionResult));
int32_t 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;
}
}

10
test/processing/targets.mk Normal file
View File

@ -0,0 +1,10 @@
PROCESSING_UNITTEST_SRCDIR=test/processing
PROCESSING_UNITTEST_CPP_SRCS=\
$(PROCESSING_UNITTEST_SRCDIR)/ProcessUT_ScrollDetection.cpp\
PROCESSING_UNITTEST_OBJS += $(PROCESSING_UNITTEST_CPP_SRCS:.cpp=.$(OBJ))
OBJS += $(PROCESSING_UNITTEST_OBJS)
$(PROCESSING_UNITTEST_SRCDIR)/%.$(OBJ): $(PROCESSING_UNITTEST_SRCDIR)/%.cpp
$(QUIET_CXX)$(CXX) $(CFLAGS) $(CXXFLAGS) $(INCLUDES) $(PROCESSING_UNITTEST_CFLAGS) $(PROCESSING_UNITTEST_INCLUDES) -c $(CXX_O) $<