#include <gtest/gtest.h> #include "codec_def.h"#include "codec_app_def.h"#include "codec_api.h"#include "wels_common_basis.h"#include "mem_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);  //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();  //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 membersvoid 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() & 3);  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;} //Mock input data for testvoid 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/Uninitializevoid 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_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_DATAFORMATvoid 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_STREAMvoid 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);  }   //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, false);   //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, true);   //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, true); //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, false); //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, true); //decoder should have EOS == true   Uninit();}  //DECODER_OPTION_VCL_NAL//Here Test illegal bitstream input//legal bitstream decoding test, please see api testvoid 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_IDvoid DecoderInterfaceTest::TestTemporalId() {  //TODO} //DECODER_OPTION_FRAME_NUMvoid DecoderInterfaceTest::TestFrameNum() {  //TODO} //DECODER_OPTION_IDR_PIC_IDvoid DecoderInterfaceTest::TestIdrPicId() {  //TODO} //DECODER_OPTION_LTR_MARKING_FLAGvoid DecoderInterfaceTest::TestLtrMarkingFlag() {  //TODO} //DECODER_OPTION_LTR_MARKED_FRAME_NUMvoid DecoderInterfaceTest::TestLtrMarkedFrameNum() {  //TODO} //DECODER_OPTION_ERROR_CON_IDCvoid DecoderInterfaceTest::TestErrorConIdc() {  //TODO} //DECODER_OPTION_TRACE_LEVELvoid DecoderInterfaceTest::TestTraceLevel() {  //TODO} //DECODER_OPTION_TRACE_CALLBACKvoid DecoderInterfaceTest::TestTraceCallback() {  //TODO} //DECODER_OPTION_TRACE_CALLBACK_CONTEXTvoid DecoderInterfaceTest::TestTraceCallbackContext() {  //TODO}  //TEST here for whole testsTEST_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();}