118 lines
3.7 KiB
C++
118 lines
3.7 KiB
C++
#include<gtest/gtest.h>
|
|
#include<stdlib.h>
|
|
#include "codec_api.h"
|
|
#include "welsEncoderExt.h"
|
|
|
|
using namespace WelsSVCEnc;
|
|
|
|
class EncoderInterfaceTest : public ::testing::Test {
|
|
public:
|
|
virtual void SetUp() {
|
|
pPtrEnc = new CWelsH264SVCEncoder();
|
|
pParamExt = new SEncParamExt();
|
|
pSrcPic = new SSourcePicture;
|
|
pOption = new SEncParamExt();
|
|
pYUV = NULL;
|
|
|
|
m_iWidth = 1280;
|
|
m_iHeight = 720;
|
|
m_iPicResSize = m_iWidth * m_iHeight * 3 >> 1;
|
|
pYUV = new uint8_t [m_iPicResSize];
|
|
}
|
|
void TemporalLayerSettingTest();
|
|
virtual void TearDown() {
|
|
delete pPtrEnc;
|
|
delete pParamExt;
|
|
delete pOption;
|
|
delete pSrcPic;
|
|
delete []pYUV;
|
|
}
|
|
public:
|
|
CWelsH264SVCEncoder* pPtrEnc;
|
|
SEncParamExt* pParamExt;
|
|
SSourcePicture* pSrcPic;
|
|
SEncParamExt* pOption;
|
|
uint8_t* pYUV;
|
|
|
|
int32_t m_iWidth;
|
|
int32_t m_iHeight;
|
|
int32_t m_iPicResSize;
|
|
};
|
|
|
|
void EncoderInterfaceTest::TemporalLayerSettingTest() {
|
|
|
|
pParamExt->iInputCsp = 23;
|
|
pParamExt->iPicWidth = m_iWidth;
|
|
pParamExt->iPicHeight = m_iHeight;
|
|
pParamExt->iTargetBitrate = 60000;
|
|
pParamExt->sSpatialLayers[0].iVideoHeight = pParamExt->iPicHeight;
|
|
pParamExt->sSpatialLayers[0].iVideoWidth = pParamExt->iPicWidth;
|
|
pParamExt->sSpatialLayers[0].iSpatialBitrate = 50000;
|
|
pParamExt->iTemporalLayerNum = 1;
|
|
pParamExt->iSpatialLayerNum = 1;
|
|
|
|
int32_t iResult = pPtrEnc->InitializeExt (pParamExt);
|
|
EXPECT_EQ (iResult, static_cast<int32_t> (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));
|
|
|
|
iResult = pPtrEnc->EncodeFrame (pSrcPic, &sFbi);
|
|
EXPECT_EQ (iResult, static_cast<int32_t> (cmResultSuccess));
|
|
EXPECT_EQ (sFbi.eFrameType, static_cast<int32_t> (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));
|
|
|
|
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));
|
|
|
|
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));
|
|
|
|
pOption ->iTemporalLayerNum = 2;
|
|
iResult = pPtrEnc->SetOption (eOptionId, pOption);
|
|
EXPECT_EQ (iResult, static_cast<int32_t> (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));
|
|
|
|
pOption ->iTemporalLayerNum = 4;
|
|
iResult = pPtrEnc->SetOption (eOptionId, pOption);
|
|
EXPECT_EQ (iResult, static_cast<int32_t> (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));
|
|
|
|
pPtrEnc->Uninitialize();
|
|
|
|
}
|
|
|
|
TEST_F (EncoderInterfaceTest, TestTemporalLayerSetting) {
|
|
TemporalLayerSettingTest();
|
|
} |