ref: 821483a4e9173c442c368f1ebef8eef6c376faf6
dir: /test/encoder/EncUT_EncoderExt.cpp/
#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(); }