ref: b304687197cf59ed24ee6fc4756f7f91b72f613f
dir: /test/decoder/DecUT_DecExt.cpp/
#include <gtest/gtest.h> #include "codec_def.h" #include "codec_app_def.h" #include "codec_api.h" #include "wels_common_basis.h" #include "memory_align.h" #include "ls_defines.h" using namespace WelsDec; #define BUF_SIZE 100 //payload size exclude 6 bytes: 0001, nal type and final '\0' #define PAYLOAD_SIZE (BUF_SIZE - 6) class DecoderInterfaceTest : public ::testing::Test { public: virtual void SetUp() { int rv = WelsCreateDecoder (&m_pDec); ASSERT_EQ (0, rv); ASSERT_TRUE (m_pDec != NULL); } virtual void TearDown() { if (m_pDec) { WelsDestroyDecoder (m_pDec); } } //Init members void Init(); //Uninit members void Uninit(); //Mock input data for test void MockPacketType (const EWelsNalUnitType eNalUnitType, const int iPacketLength); //Decoder real bitstream void DecoderBs (const char* sFileName); //Test Initialize/Uninitialize void TestInitUninit(); //DECODER_OPTION_DATAFORMAT void TestDataFormat(); //DECODER_OPTION_END_OF_STREAM void TestEndOfStream(); //DECODER_OPTION_VCL_NAL void TestVclNal(); //DECODER_OPTION_TEMPORAL_ID void TestTemporalId(); //DECODER_OPTION_FRAME_NUM void TestFrameNum(); //DECODER_OPTION_IDR_PIC_ID void TestIdrPicId(); //DECODER_OPTION_LTR_MARKING_FLAG void TestLtrMarkingFlag(); //DECODER_OPTION_LTR_MARKED_FRAME_NUM void TestLtrMarkedFrameNum(); //DECODER_OPTION_ERROR_CON_IDC void TestErrorConIdc(); //DECODER_OPTION_TRACE_LEVEL void TestTraceLevel(); //DECODER_OPTION_TRACE_CALLBACK void TestTraceCallback(); //DECODER_OPTION_TRACE_CALLBACK_CONTEXT void TestTraceCallbackContext(); //DECODER_OPTION_GET_DECODER_STATICTIS void TestGetDecStatistics(); //Do whole tests here void DecoderInterfaceAll(); public: ISVCDecoder* m_pDec; SDecodingParam m_sDecParam; SBufferInfo m_sBufferInfo; uint8_t* m_pData[3]; unsigned char m_szBuffer[BUF_SIZE]; //for mocking packet int m_iBufLength; //record the valid data in m_szBuffer }; //Init members void DecoderInterfaceTest::Init() { memset (&m_sBufferInfo, 0, sizeof (SBufferInfo)); memset (&m_sDecParam, 0, sizeof (SDecodingParam)); m_sDecParam.pFileNameRestructed = NULL; m_sDecParam.eOutputColorFormat = (EVideoFormatType) (rand() % 100); m_sDecParam.uiCpuLoad = rand() % 100; m_sDecParam.uiTargetDqLayer = rand() % 100; m_sDecParam.eEcActiveIdc = (ERROR_CON_IDC) (rand() & 7); m_sDecParam.sVideoProperty.size = sizeof (SVideoProperty); m_sDecParam.sVideoProperty.eVideoBsType = (VIDEO_BITSTREAM_TYPE) (rand() % 3); m_pData[0] = m_pData[1] = m_pData[2] = NULL; m_szBuffer[0] = m_szBuffer[1] = m_szBuffer[2] = 0; m_szBuffer[3] = 1; m_iBufLength = 4; CM_RETURN eRet = (CM_RETURN) m_pDec->Initialize (&m_sDecParam); if ((m_sDecParam.eOutputColorFormat != videoFormatI420) && (m_sDecParam.eOutputColorFormat != videoFormatInternal)) ASSERT_EQ (eRet, cmUnsupportedData); else ASSERT_EQ (eRet, cmResultSuccess); } void DecoderInterfaceTest::Uninit() { if (m_pDec) { CM_RETURN eRet = (CM_RETURN) m_pDec->Uninitialize(); ASSERT_EQ (eRet, cmResultSuccess); } memset (&m_sDecParam, 0, sizeof (SDecodingParam)); memset (&m_sBufferInfo, 0, sizeof (SBufferInfo)); m_pData[0] = m_pData[1] = m_pData[2] = NULL; m_iBufLength = 0; } void DecoderInterfaceTest::DecoderBs (const char* sFileName) { uint8_t* pBuf = NULL; int32_t iBufPos = 0; int32_t iFileSize; int32_t i = 0; int32_t iSliceSize; int32_t iSliceIndex = 0; int32_t iEndOfStreamFlag = 0; FILE* pH264File; uint8_t uiStartCode[4] = {0, 0, 0, 1}; #if defined(ANDROID_NDK) std::string filename = std::string ("/sdcard/") + sFileName; ASSERT_TRUE ((pH264File = fopen (filename.c_str(), "rb")) != NULL); #else ASSERT_TRUE ((pH264File = fopen (sFileName, "rb")) != NULL); #endif fseek (pH264File, 0L, SEEK_END); iFileSize = (int32_t) ftell (pH264File); fseek (pH264File, 0L, SEEK_SET); pBuf = new uint8_t[iFileSize + 4]; ASSERT_EQ (fread (pBuf, 1, iFileSize, pH264File), (unsigned int) iFileSize); memcpy (pBuf + iFileSize, &uiStartCode[0], 4); //confirmed_safe_unsafe_usage while (true) { if (iBufPos >= iFileSize) { iEndOfStreamFlag = true; if (iEndOfStreamFlag) m_pDec->SetOption (DECODER_OPTION_END_OF_STREAM, (void*)&iEndOfStreamFlag); break; } for (i = 0; i < iFileSize; i++) { if ((pBuf[iBufPos + i] == 0 && pBuf[iBufPos + i + 1] == 0 && pBuf[iBufPos + i + 2] == 0 && pBuf[iBufPos + i + 3] == 1 && i > 0)) { break; } } iSliceSize = i; m_pDec->DecodeFrame2 (pBuf + iBufPos, iSliceSize, m_pData, &m_sBufferInfo); m_pDec->DecodeFrame2 (NULL, 0, m_pData, &m_sBufferInfo); iBufPos += iSliceSize; ++ iSliceIndex; } fclose (pH264File); if (pBuf) { delete[] pBuf; pBuf = NULL; } } //Mock input data for test void DecoderInterfaceTest::MockPacketType (const EWelsNalUnitType eNalUnitType, const int iPacketLength) { switch (eNalUnitType) { case NAL_UNIT_SEI: m_szBuffer[m_iBufLength++] = 6; break; case NAL_UNIT_SPS: m_szBuffer[m_iBufLength++] = 67; break; case NAL_UNIT_PPS: m_szBuffer[m_iBufLength++] = 68; break; case NAL_UNIT_SUBSET_SPS: m_szBuffer[m_iBufLength++] = 15; break; case NAL_UNIT_PREFIX: m_szBuffer[m_iBufLength++] = 14; break; case NAL_UNIT_CODED_SLICE: m_szBuffer[m_iBufLength++] = 61; break; case NAL_UNIT_CODED_SLICE_IDR: m_szBuffer[m_iBufLength++] = 65; break; default: m_szBuffer[m_iBufLength++] = 0; //NAL_UNIT_UNSPEC_0 break; int iAddLength = iPacketLength - 5; //excluding 0001 and type if (iAddLength > PAYLOAD_SIZE) iAddLength = PAYLOAD_SIZE; for (int i = 0; i < iAddLength; ++i) { m_szBuffer[m_iBufLength++] = rand() % 256; } m_szBuffer[m_iBufLength++] = '\0'; } } //Test Initialize/Uninitialize void DecoderInterfaceTest::TestInitUninit() { int iOutput; CM_RETURN eRet; //No initialize, no GetOption can be done m_pDec->Uninitialize(); for (int i = 0; i <= (int) DECODER_OPTION_TRACE_CALLBACK_CONTEXT; ++i) { eRet = (CM_RETURN) m_pDec->GetOption ((DECODER_OPTION) i, &iOutput); EXPECT_EQ (eRet, cmInitExpected); } //Initialize first, can get input color format m_sDecParam.eOutputColorFormat = (EVideoFormatType) 20; //just for test m_sDecParam.bParseOnly = false; m_pDec->Initialize (&m_sDecParam); eRet = (CM_RETURN) m_pDec->GetOption (DECODER_OPTION_DATAFORMAT, &iOutput); EXPECT_EQ (eRet, cmResultSuccess); EXPECT_EQ ((int32_t) videoFormatI420, iOutput); //Uninitialize, no GetOption can be done m_pDec->Uninitialize(); iOutput = 21; for (int i = 0; i <= (int) DECODER_OPTION_TRACE_CALLBACK_CONTEXT; ++i) { eRet = (CM_RETURN) m_pDec->GetOption ((DECODER_OPTION) i, &iOutput); EXPECT_EQ (iOutput, 21); EXPECT_EQ (eRet, cmInitExpected); } } //DECODER_OPTION_DATAFORMAT void DecoderInterfaceTest::TestDataFormat() { int iTmp = rand(); int iOut; CM_RETURN eRet; Init(); //invalid input eRet = (CM_RETURN) m_pDec->SetOption (DECODER_OPTION_DATAFORMAT, NULL); EXPECT_EQ (eRet, cmInitParaError); //valid input eRet = (CM_RETURN) m_pDec->SetOption (DECODER_OPTION_DATAFORMAT, &iTmp); if ((iTmp != (int32_t) videoFormatI420) && (iTmp != (int32_t) videoFormatInternal)) EXPECT_EQ (eRet, cmUnsupportedData); else EXPECT_EQ (eRet, cmResultSuccess); eRet = (CM_RETURN) m_pDec->GetOption (DECODER_OPTION_DATAFORMAT, &iOut); EXPECT_EQ (eRet, cmResultSuccess); EXPECT_EQ (iOut, (int32_t) videoFormatI420); Uninit(); } //DECODER_OPTION_END_OF_STREAM void DecoderInterfaceTest::TestEndOfStream() { int iTmp, iOut; CM_RETURN eRet; Init(); //invalid input eRet = (CM_RETURN) m_pDec->SetOption (DECODER_OPTION_END_OF_STREAM, NULL); EXPECT_EQ (eRet, cmInitParaError); //valid random input for (int i = 0; i < 10; ++i) { iTmp = rand(); eRet = (CM_RETURN) m_pDec->SetOption (DECODER_OPTION_END_OF_STREAM, &iTmp); EXPECT_EQ (eRet, cmResultSuccess); eRet = (CM_RETURN) m_pDec->GetOption (DECODER_OPTION_END_OF_STREAM, &iOut); EXPECT_EQ (eRet, cmResultSuccess); EXPECT_EQ (iOut, iTmp != 0 ? 1 : 0); } //set false as input iTmp = false; eRet = (CM_RETURN) m_pDec->SetOption (DECODER_OPTION_END_OF_STREAM, &iTmp); EXPECT_EQ (eRet, cmResultSuccess); eRet = (CM_RETURN) m_pDec->GetOption (DECODER_OPTION_END_OF_STREAM, &iOut); EXPECT_EQ (eRet, cmResultSuccess); EXPECT_EQ (iOut, 0); //set true as input iTmp = true; eRet = (CM_RETURN) m_pDec->SetOption (DECODER_OPTION_END_OF_STREAM, &iTmp); EXPECT_EQ (eRet, cmResultSuccess); eRet = (CM_RETURN) m_pDec->GetOption (DECODER_OPTION_END_OF_STREAM, &iOut); EXPECT_EQ (eRet, cmResultSuccess); EXPECT_EQ (iOut, 1); //Mock data packet in //Test NULL data input for decoder, should be true for EOS eRet = (CM_RETURN) m_pDec->DecodeFrame2 (NULL, 0, m_pData, &m_sBufferInfo); EXPECT_EQ (eRet, 0); //decode should return OK eRet = (CM_RETURN) m_pDec->GetOption (DECODER_OPTION_END_OF_STREAM, &iOut); EXPECT_EQ (iOut, 1); //decoder should have EOS == true //Test valid data input for decoder, should be false for EOS MockPacketType (NAL_UNIT_UNSPEC_0, 50); eRet = (CM_RETURN) m_pDec->DecodeFrame2 (m_szBuffer, m_iBufLength, m_pData, &m_sBufferInfo); eRet = (CM_RETURN) m_pDec->GetOption (DECODER_OPTION_END_OF_STREAM, &iOut); EXPECT_EQ (iOut, 0); //decoder should have EOS == false //Test NULL data input for decoder, should be true for EOS eRet = (CM_RETURN) m_pDec->DecodeFrame2 (NULL, 0, m_pData, &m_sBufferInfo); eRet = (CM_RETURN) m_pDec->GetOption (DECODER_OPTION_END_OF_STREAM, &iOut); EXPECT_EQ (iOut, 1); //decoder should have EOS == true Uninit(); } //DECODER_OPTION_VCL_NAL //Here Test illegal bitstream input //legal bitstream decoding test, please see api test void DecoderInterfaceTest::TestVclNal() { int iTmp, iOut; CM_RETURN eRet; Init(); //Test SetOption //VclNal never supports SetOption iTmp = rand(); eRet = (CM_RETURN) m_pDec->SetOption (DECODER_OPTION_VCL_NAL, &iTmp); EXPECT_EQ (eRet, cmInitParaError); //Test GetOption //invalid input eRet = (CM_RETURN) m_pDec->GetOption (DECODER_OPTION_VCL_NAL, NULL); EXPECT_EQ (eRet, cmInitParaError); //valid input without actual decoding eRet = (CM_RETURN) m_pDec->GetOption (DECODER_OPTION_VCL_NAL, &iOut); EXPECT_EQ (eRet, cmResultSuccess); EXPECT_EQ (iOut, FEEDBACK_NON_VCL_NAL); //valid input with decoding error MockPacketType (NAL_UNIT_CODED_SLICE_IDR, 50); m_pDec->DecodeFrame2 (m_szBuffer, m_iBufLength, m_pData, &m_sBufferInfo); eRet = (CM_RETURN) m_pDec->GetOption (DECODER_OPTION_VCL_NAL, &iOut); EXPECT_EQ (eRet, cmResultSuccess); EXPECT_EQ (iOut, FEEDBACK_UNKNOWN_NAL); m_pDec->DecodeFrame2 (NULL, 0, m_pData, &m_sBufferInfo); eRet = (CM_RETURN) m_pDec->GetOption (DECODER_OPTION_VCL_NAL, &iOut); EXPECT_EQ (eRet, cmResultSuccess); EXPECT_EQ (iOut, FEEDBACK_UNKNOWN_NAL); Uninit(); } //DECODER_OPTION_TEMPORAL_ID void DecoderInterfaceTest::TestTemporalId() { //TODO } //DECODER_OPTION_FRAME_NUM void DecoderInterfaceTest::TestFrameNum() { //TODO } //DECODER_OPTION_IDR_PIC_ID void DecoderInterfaceTest::TestIdrPicId() { //TODO } //DECODER_OPTION_LTR_MARKING_FLAG void DecoderInterfaceTest::TestLtrMarkingFlag() { //TODO } //DECODER_OPTION_LTR_MARKED_FRAME_NUM void DecoderInterfaceTest::TestLtrMarkedFrameNum() { //TODO } //DECODER_OPTION_ERROR_CON_IDC void DecoderInterfaceTest::TestErrorConIdc() { int iTmp, iOut; CM_RETURN eRet; Init(); //Test GetOption //invalid input eRet = (CM_RETURN) m_pDec->GetOption (DECODER_OPTION_ERROR_CON_IDC, NULL); EXPECT_EQ (eRet, cmInitParaError); //Test GetOption //valid input eRet = (CM_RETURN) m_pDec->GetOption (DECODER_OPTION_ERROR_CON_IDC, &iOut); EXPECT_EQ (eRet, cmResultSuccess); //Test SetOption iTmp = rand() & 7; eRet = (CM_RETURN) m_pDec->SetOption (DECODER_OPTION_ERROR_CON_IDC, &iTmp); EXPECT_EQ (eRet, cmResultSuccess); //Test GetOption eRet = (CM_RETURN) m_pDec->GetOption (DECODER_OPTION_ERROR_CON_IDC, &iOut); EXPECT_EQ (eRet, cmResultSuccess); EXPECT_EQ (iTmp, iOut); Uninit(); } //DECODER_OPTION_TRACE_LEVEL void DecoderInterfaceTest::TestTraceLevel() { //TODO } //DECODER_OPTION_TRACE_CALLBACK void DecoderInterfaceTest::TestTraceCallback() { //TODO } //DECODER_OPTION_TRACE_CALLBACK_CONTEXT void DecoderInterfaceTest::TestTraceCallbackContext() { //TODO } //DECODER_OPTION_GET_STATISTICS void DecoderInterfaceTest::TestGetDecStatistics() { CM_RETURN eRet; SDecoderStatistics sDecStatic; int32_t iError = 0; Init(); // setoption not support, eRet = (CM_RETURN)m_pDec->SetOption (DECODER_OPTION_GET_STATISTICS, NULL); EXPECT_EQ (eRet, cmInitParaError); //EC on UT iError = 2; m_pDec->SetOption (DECODER_OPTION_ERROR_CON_IDC, &iError); //Decoder error bs DecoderBs ("res/Error_I_P.264"); m_pDec->GetOption (DECODER_OPTION_GET_STATISTICS, &sDecStatic); EXPECT_EQ (65u, sDecStatic.uiAvgEcRatio); EXPECT_EQ (7u, sDecStatic.uiAvgEcPropRatio); EXPECT_EQ (5u, sDecStatic.uiDecodedFrameCount); EXPECT_EQ (288u, sDecStatic.uiHeight); EXPECT_EQ (1u, sDecStatic.uiIDRCorrectNum); EXPECT_EQ (3u, sDecStatic.uiResolutionChangeTimes); EXPECT_EQ (352u, sDecStatic.uiWidth); EXPECT_EQ (4u, sDecStatic.uiEcFrameNum); EXPECT_EQ (2u, sDecStatic.uiEcIDRNum); EXPECT_EQ (0u, sDecStatic.uiIDRLostNum); Uninit(); //Decoder error bs when the first IDR lost Init(); iError = 2; m_pDec->SetOption (DECODER_OPTION_ERROR_CON_IDC, &iError); DecoderBs ("res/BA_MW_D_IDR_LOST.264"); m_pDec->GetOption (DECODER_OPTION_GET_STATISTICS, &sDecStatic); EXPECT_EQ (88u, sDecStatic.uiAvgEcRatio); EXPECT_EQ (88u, sDecStatic.uiAvgEcPropRatio); EXPECT_EQ (97u, sDecStatic.uiDecodedFrameCount); EXPECT_EQ (144u, sDecStatic.uiHeight); EXPECT_EQ (3u, sDecStatic.uiIDRCorrectNum); EXPECT_EQ (0u, sDecStatic.uiEcIDRNum); EXPECT_EQ (1u, sDecStatic.uiResolutionChangeTimes); EXPECT_EQ (176u, sDecStatic.uiWidth); EXPECT_EQ (27u, sDecStatic.uiEcFrameNum); EXPECT_EQ (1u, sDecStatic.uiIDRLostNum); Uninit(); //ecoder error bs when the first P lost Init(); iError = 2; m_pDec->SetOption (DECODER_OPTION_ERROR_CON_IDC, &iError); DecoderBs ("res/BA_MW_D_P_LOST.264"); m_pDec->GetOption (DECODER_OPTION_GET_STATISTICS, &sDecStatic); EXPECT_EQ (85u, sDecStatic.uiAvgEcRatio); EXPECT_EQ (85u, sDecStatic.uiAvgEcPropRatio); EXPECT_EQ (99u, sDecStatic.uiDecodedFrameCount); EXPECT_EQ (144u, sDecStatic.uiHeight); EXPECT_EQ (4u, sDecStatic.uiIDRCorrectNum); EXPECT_EQ (0u, sDecStatic.uiEcIDRNum); EXPECT_EQ (1u, sDecStatic.uiResolutionChangeTimes); EXPECT_EQ (176u, sDecStatic.uiWidth); EXPECT_EQ (28u, sDecStatic.uiEcFrameNum); EXPECT_EQ (0u, sDecStatic.uiIDRLostNum); Uninit(); //EC enable //EC Off UT just correc bitstream Init(); iError = 0; m_pDec->SetOption (DECODER_OPTION_ERROR_CON_IDC, &iError); DecoderBs ("res/test_vd_1d.264"); m_pDec->GetOption (DECODER_OPTION_GET_STATISTICS, &sDecStatic); EXPECT_EQ (0u, sDecStatic.uiAvgEcRatio); EXPECT_EQ (0u, sDecStatic.uiAvgEcPropRatio); EXPECT_EQ (9u, sDecStatic.uiDecodedFrameCount); EXPECT_EQ (192u, sDecStatic.uiHeight); EXPECT_EQ (1u, sDecStatic.uiIDRCorrectNum); EXPECT_EQ (1u, sDecStatic.uiResolutionChangeTimes); EXPECT_EQ (320u, sDecStatic.uiWidth); EXPECT_EQ (0u, sDecStatic.uiEcFrameNum); EXPECT_EQ (0u, sDecStatic.uiIDRLostNum); Uninit(); } //TEST here for whole tests TEST_F (DecoderInterfaceTest, DecoderInterfaceAll) { //Initialize Uninitialize TestInitUninit(); //DECODER_OPTION_DATAFORMAT TestDataFormat(); //DECODER_OPTION_END_OF_STREAM TestEndOfStream(); //DECODER_OPTION_VCL_NAL TestVclNal(); //DECODER_OPTION_TEMPORAL_ID TestTemporalId(); //DECODER_OPTION_FRAME_NUM TestFrameNum(); //DECODER_OPTION_IDR_PIC_ID TestIdrPicId(); //DECODER_OPTION_LTR_MARKING_FLAG TestLtrMarkingFlag(); //DECODER_OPTION_LTR_MARKED_FRAME_NUM TestLtrMarkedFrameNum(); //DECODER_OPTION_ERROR_CON_IDC TestErrorConIdc(); //DECODER_OPTION_TRACE_LEVEL TestTraceLevel(); //DECODER_OPTION_TRACE_CALLBACK TestTraceCallback(); //DECODER_OPTION_TRACE_CALLBACK_CONTEXT TestTraceCallbackContext(); //DECODER_OPTION_GET_STATISTICS TestGetDecStatistics(); }