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

add BasicInitialize UT and some fix in create/destroy encoder in the existing interface test

Browse Source
This commit is contained in:
Sijia Chen 2014-07-11 13:50:58 +08:00
parent da0f65ea0a
commit 99dc18c0ba
2 changed files with 223 additions and 51 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
codec/encoder/core/src/encoder_ext.cpp
View File

@ -145,7 +145,14 @@ int32_t ParamValidation (SLogContext* pLogCtx, SWelsSvcCodingParam* pCfg) {
|| fMaxFrameRate - pCfg->fMaxFrameRate < -fEpsn)) {
pCfg->fMaxFrameRate = fMaxFrameRate;
}
//bitrate setting validation
if ((pCfg->iRCMode != RC_OFF_MODE) && (pCfg->iRCMode != RC_QUALITY_MODE) && (pCfg->iRCMode != RC_BITRATE_MODE)
&& (pCfg->iRCMode != RC_LOW_BW_MODE)) {
WelsLog (pLogCtx, WELS_LOG_ERROR, "ParamValidation(),Invalid iRCMode = %d\n", pCfg->iRCMode);
return ENC_RETURN_UNSUPPORTED_PARA;
}
//bitrate setting validation
if (pCfg->iRCMode != RC_OFF_MODE) {
int32_t iTotalBitrate = 0;
if (pCfg->iTargetBitrate <= 0) {

265
test/encoder/EncUT_EncoderExt.cpp
View File

@ -1,44 +1,105 @@
#include <gtest/gtest.h>
#include <stdlib.h>
#include "codec_api.h"
#include "welsEncoderExt.h"
using namespace WelsSVCEnc;
#include "codec_app_def.h"
//TODO: consider using BaseEncoderTest class from #include "../BaseEncoderTest.h"
class EncoderInterfaceTest : public ::testing::Test {
#define MB_SIZE (16)
#define MAX_WIDTH (3840)
#define MAX_HEIGHT (2160)
public:
virtual void SetUp() {
pPtrEnc = new CWelsH264SVCEncoder();
pParamExt = new SEncParamExt();
pSrcPic = new SSourcePicture;
pOption = new SEncParamExt();
pYUV = NULL;
int rv = WelsCreateSVCEncoder (&pPtrEnc);
ASSERT_EQ (0, rv);
ASSERT_TRUE (pPtrEnc != NULL);
m_iWidth = 1280;
m_iHeight = 720;
pParamExt = new SEncParamExt();
ASSERT_TRUE (pParamExt != NULL);
pSrcPic = new SSourcePicture;
ASSERT_TRUE (pSrcPic != NULL);
pOption = new SEncParamExt();
ASSERT_TRUE (pOption != NULL);
pYUV = NULL;
m_iWidth = MAX_WIDTH;
m_iHeight = MAX_HEIGHT;
m_iPicResSize = m_iWidth * m_iHeight * 3 >> 1;
pYUV = new uint8_t [m_iPicResSize];
pYUV = new unsigned char [m_iPicResSize];
ASSERT_TRUE (pYUV != NULL);
}
void TemporalLayerSettingTest();
virtual void TearDown() {
delete pPtrEnc;
delete pParamExt;
delete pOption;
delete pSrcPic;
delete []pYUV;
if (pPtrEnc) {
WelsDestroySVCEncoder (pPtrEnc);
pPtrEnc = NULL;
}
}
void TemporalLayerSettingTest();
void EncodeOneFrame (SEncParamBase* pEncParamBase);
void PrepareOneSrcFrame();
public:
CWelsH264SVCEncoder* pPtrEnc;
ISVCEncoder* pPtrEnc;
SEncParamExt* pParamExt;
SSourcePicture* pSrcPic;
SEncParamExt* pOption;
uint8_t* pYUV;
unsigned char* pYUV;
int32_t m_iWidth;
int32_t m_iHeight;
int32_t m_iPicResSize;
SFrameBSInfo sFbi;
int m_iWidth;
int m_iHeight;
int m_iPicResSize;
};
void EncoderInterfaceTest::PrepareOneSrcFrame() {
pSrcPic->iColorFormat = videoFormatI420;
pSrcPic->uiTimeStamp = 0;
pSrcPic->iPicWidth = pParamExt->iPicWidth;
pSrcPic->iPicHeight = pParamExt->iPicHeight;
for (int i = 0; i < m_iPicResSize; i++)
pYUV[i] = rand() % 256;
pSrcPic->iStride[0] = m_iWidth;
pSrcPic->iStride[1] = pSrcPic->iStride[2] = pSrcPic->iStride[0] >> 1;
pSrcPic->pData[0] = pYUV;
pSrcPic->pData[1] = pSrcPic->pData[0] + (m_iWidth * m_iHeight);
pSrcPic->pData[2] = pSrcPic->pData[1] + (m_iWidth * m_iHeight >> 2);
//clean the output
memset (&sFbi, 0, sizeof (SFrameBSInfo));
}
void EncoderInterfaceTest::EncodeOneFrame (SEncParamBase* pEncParamBase) {
int iResult;
iResult = pPtrEnc->Initialize (pEncParamBase);
EXPECT_EQ (iResult, static_cast<int> (cmResultSuccess));
PrepareOneSrcFrame();
iResult = pPtrEnc->EncodeFrame (pSrcPic, &sFbi);
EXPECT_EQ (iResult, static_cast<int> (cmResultSuccess));
EXPECT_EQ (sFbi.eFrameType, static_cast<int> (videoFrameTypeIDR));
iResult = pPtrEnc->GetOption (ENCODER_OPTION_SVC_ENCODE_PARAM_BASE, pEncParamBase);
EXPECT_EQ (iResult, static_cast<int> (cmResultSuccess));
pPtrEnc->Uninitialize();
EXPECT_EQ (iResult, static_cast<int> (cmResultSuccess));
}
void EncoderInterfaceTest::TemporalLayerSettingTest() {
pParamExt->iPicWidth = m_iWidth;
@ -50,63 +111,47 @@ void EncoderInterfaceTest::TemporalLayerSettingTest() {
pParamExt->iTemporalLayerNum = 1;
pParamExt->iSpatialLayerNum = 1;
int32_t iResult = pPtrEnc->InitializeExt (pParamExt);
EXPECT_EQ (iResult, static_cast<int32_t> (cmResultSuccess));
int iResult = pPtrEnc->InitializeExt (pParamExt);
EXPECT_EQ (iResult, static_cast<int> (cmResultSuccess));
pSrcPic->iColorFormat = videoFormatI420;
pSrcPic->uiTimeStamp = 0;
pSrcPic->iPicWidth = pParamExt->iPicWidth;
pSrcPic->iPicHeight = pParamExt->iPicHeight;
for (int32_t i = 0; i < m_iPicResSize; i++)
pYUV[i] = rand() % 256;
pSrcPic->iStride[0] = m_iWidth;
pSrcPic->iStride[1] = pSrcPic->iStride[2] = pSrcPic->iStride[0] >> 1;
pSrcPic->pData[0] = pYUV;
pSrcPic->pData[1] = pSrcPic->pData[0] + (m_iWidth * m_iHeight);
pSrcPic->pData[2] = pSrcPic->pData[1] + (m_iWidth * m_iHeight >> 2);
SFrameBSInfo sFbi;
memset (&sFbi, 0, sizeof (SFrameBSInfo));
PrepareOneSrcFrame();
iResult = pPtrEnc->EncodeFrame (pSrcPic, &sFbi);
EXPECT_EQ (iResult, static_cast<int32_t> (cmResultSuccess));
EXPECT_EQ (sFbi.eFrameType, static_cast<int32_t> (videoFrameTypeIDR));
EXPECT_EQ (iResult, static_cast<int> (cmResultSuccess));
EXPECT_EQ (sFbi.eFrameType, static_cast<int> (videoFrameTypeIDR));
pSrcPic->uiTimeStamp = 30;
iResult = pPtrEnc->EncodeFrame (pSrcPic, &sFbi);
EXPECT_EQ (iResult, static_cast<int32_t> (cmResultSuccess));
EXPECT_EQ (sFbi.eFrameType, static_cast<int32_t> (videoFrameTypeP));
EXPECT_EQ (iResult, static_cast<int> (cmResultSuccess));
EXPECT_EQ (sFbi.eFrameType, static_cast<int> (videoFrameTypeP));
memcpy (pOption, pParamExt, sizeof (SEncParamExt));
pOption ->iTemporalLayerNum = 4;
ENCODER_OPTION eOptionId = ENCODER_OPTION_SVC_ENCODE_PARAM_EXT;
iResult = pPtrEnc->SetOption (eOptionId, pOption);
EXPECT_EQ (iResult, static_cast<int32_t> (cmResultSuccess));
EXPECT_EQ (iResult, static_cast<int> (cmResultSuccess));
pSrcPic->uiTimeStamp = 60;
iResult = pPtrEnc->EncodeFrame (pSrcPic, &sFbi);
EXPECT_EQ (iResult, static_cast<int32_t> (cmResultSuccess));
EXPECT_EQ (sFbi.eFrameType, static_cast<int32_t> (videoFrameTypeIDR));
EXPECT_EQ (iResult, static_cast<int> (cmResultSuccess));
EXPECT_EQ (sFbi.eFrameType, static_cast<int> (videoFrameTypeIDR));
pOption ->iTemporalLayerNum = 2;
iResult = pPtrEnc->SetOption (eOptionId, pOption);
EXPECT_EQ (iResult, static_cast<int32_t> (cmResultSuccess));
EXPECT_EQ (iResult, static_cast<int> (cmResultSuccess));
pSrcPic->uiTimeStamp = 90;
iResult = pPtrEnc->EncodeFrame (pSrcPic, &sFbi);
EXPECT_EQ (iResult, static_cast<int32_t> (cmResultSuccess));
EXPECT_EQ (sFbi.eFrameType, static_cast<int32_t> (videoFrameTypeP));
EXPECT_EQ (iResult, static_cast<int> (cmResultSuccess));
EXPECT_EQ (sFbi.eFrameType, static_cast<int> (videoFrameTypeP));
pOption ->iTemporalLayerNum = 4;
iResult = pPtrEnc->SetOption (eOptionId, pOption);
EXPECT_EQ (iResult, static_cast<int32_t> (cmResultSuccess));
EXPECT_EQ (iResult, static_cast<int> (cmResultSuccess));
pSrcPic->uiTimeStamp = 120;
iResult = pPtrEnc->EncodeFrame (pSrcPic, &sFbi);
EXPECT_EQ (iResult, static_cast<int32_t> (cmResultSuccess));
EXPECT_EQ (sFbi.eFrameType, static_cast<int32_t> (videoFrameTypeP));
EXPECT_EQ (iResult, static_cast<int> (cmResultSuccess));
EXPECT_EQ (sFbi.eFrameType, static_cast<int> (videoFrameTypeP));
pPtrEnc->Uninitialize();
@ -115,3 +160,123 @@ void EncoderInterfaceTest::TemporalLayerSettingTest() {
TEST_F (EncoderInterfaceTest, TestTemporalLayerSetting) {
TemporalLayerSettingTest();
}
void GetValidEncParamBase (SEncParamBase* pEncParamBase) {
pEncParamBase->iUsageType = CAMERA_VIDEO_REAL_TIME;
pEncParamBase->iPicWidth = abs ((rand() * 2) + MB_SIZE) % (MAX_WIDTH+1);
pEncParamBase->iPicHeight = abs ((rand() * 2) + MB_SIZE) % (MAX_HEIGHT+1);
pEncParamBase->iTargetBitrate = rand() + 1; //!=0
pEncParamBase->iRCMode = RC_BITRATE_MODE; //-1, 0, 1, 2
pEncParamBase->fMaxFrameRate = rand() + 0.5f; //!=0
}
TEST_F (EncoderInterfaceTest, BasicInitializeTest) {
SEncParamBase sEncParamBase;
GetValidEncParamBase (&sEncParamBase);
int iResult = pPtrEnc->Initialize (&sEncParamBase);
EXPECT_EQ (iResult, static_cast<int> (cmResultSuccess));
if (iResult != cmResultSuccess) {
fprintf (stderr, "Unexpected ParamBase? \
iUsageType=%d, Pic=%dx%d, TargetBitrate=%d, iRCMode=%d, fMaxFrameRate=%.1f\n",
sEncParamBase.iUsageType, sEncParamBase.iPicWidth, sEncParamBase.iPicHeight,
sEncParamBase.iTargetBitrate, sEncParamBase.iRCMode, sEncParamBase.fMaxFrameRate);
}
PrepareOneSrcFrame();
iResult = pPtrEnc->EncodeFrame (pSrcPic, &sFbi);
EXPECT_EQ (iResult, static_cast<int> (cmResultSuccess));
EXPECT_EQ (sFbi.eFrameType, static_cast<int> (videoFrameTypeIDR));
pPtrEnc->Uninitialize();
EXPECT_EQ (iResult, static_cast<int> (cmResultSuccess));
}
TEST_F (EncoderInterfaceTest, BasicInitializeTestFalse) {
int iResult;
SEncParamBase sEncParamBase;
//iUsageType
GetValidEncParamBase (&sEncParamBase);
sEncParamBase.iUsageType = static_cast<EUsageType> (2);
iResult = pPtrEnc->Initialize (&sEncParamBase);
EXPECT_EQ (iResult, static_cast<int> (cmInitParaError));
//iPicWidth
GetValidEncParamBase (&sEncParamBase);
sEncParamBase.iPicWidth = 0;
iResult = pPtrEnc->Initialize (&sEncParamBase);
EXPECT_EQ (iResult, static_cast<int> (cmInitParaError));
GetValidEncParamBase (&sEncParamBase);
sEncParamBase.iPicWidth = -1;
iResult = pPtrEnc->Initialize (&sEncParamBase);
EXPECT_EQ (iResult, static_cast<int> (cmInitParaError));
//TODO: add checking max in interface and then enable this checking
//GetValidEncParamBase(&sEncParamBase);
//sEncParamBase.iPicWidth = rand()+(MAX_WIDTH+1);
//iResult = pPtrEnc->Initialize (&sEncParamBase);
//EXPECT_EQ (iResult, static_cast<int> (cmInitParaError));
//iPicHeight
GetValidEncParamBase (&sEncParamBase);
sEncParamBase.iPicHeight = 0;
iResult = pPtrEnc->Initialize (&sEncParamBase);
EXPECT_EQ (iResult, static_cast<int> (cmInitParaError));
GetValidEncParamBase (&sEncParamBase);
sEncParamBase.iPicHeight = -1;
iResult = pPtrEnc->Initialize (&sEncParamBase);
EXPECT_EQ (iResult, static_cast<int> (cmInitParaError));
//TODO: add checking max in interface and then enable this checking
//GetValidEncParamBase(&sEncParamBase);
//sEncParamBase.iPicWidth = rand()+(MAX_HEIGHT+1);
//iResult = pPtrEnc->Initialize (&sEncParamBase);
//EXPECT_EQ (iResult, static_cast<int> (cmInitParaError));
//iTargetBitrate
GetValidEncParamBase (&sEncParamBase);
sEncParamBase.iTargetBitrate = 0;
iResult = pPtrEnc->Initialize (&sEncParamBase);
EXPECT_EQ (iResult, static_cast<int> (cmInitParaError));
GetValidEncParamBase (&sEncParamBase);
sEncParamBase.iTargetBitrate = -1;
iResult = pPtrEnc->Initialize (&sEncParamBase);
EXPECT_EQ (iResult, static_cast<int> (cmInitParaError));
//iUsageType
GetValidEncParamBase (&sEncParamBase);
sEncParamBase.iRCMode = static_cast<RC_MODES> (3);
iResult = pPtrEnc->Initialize (&sEncParamBase);
EXPECT_EQ (iResult, static_cast<int> (cmInitParaError));
}
TEST_F (EncoderInterfaceTest, BasicInitializeTestAutoAdjustment) {
SEncParamBase sEncParamBase;
// large fMaxFrameRate
GetValidEncParamBase (&sEncParamBase);
sEncParamBase.fMaxFrameRate = 50000;
EncodeOneFrame (&sEncParamBase);
EXPECT_LE (sEncParamBase.fMaxFrameRate, 30.0);
EXPECT_GE (sEncParamBase.fMaxFrameRate, 1.0);
// fMaxFrameRate = 0
GetValidEncParamBase (&sEncParamBase);
sEncParamBase.fMaxFrameRate = 0;
EncodeOneFrame (&sEncParamBase);
EXPECT_LE (sEncParamBase.fMaxFrameRate, 30.0);
EXPECT_GE (sEncParamBase.fMaxFrameRate, 1.0);
// fMaxFrameRate = -1
GetValidEncParamBase (&sEncParamBase);
sEncParamBase.fMaxFrameRate = -1;
EncodeOneFrame (&sEncParamBase);
EXPECT_LE (sEncParamBase.fMaxFrameRate, 30.0);
EXPECT_GE (sEncParamBase.fMaxFrameRate, 1.0);
}