ref: f5cc1c082fb481d491014b84194e07c5c612c1e5
dir: /test/api/BaseThreadDecoderTest.cpp/
#include <fstream> #include <gtest/gtest.h> #include "codec_def.h" #include "codec_app_def.h" #include "utils/BufferedData.h" #include "BaseThreadDecoderTest.h" static void Write2File (FILE* pFp, unsigned char* pData[3], int iStride[2], int iWidth, int iHeight) { int i; unsigned char* pPtr = NULL; pPtr = pData[0]; for (i = 0; i < iHeight; i++) { fwrite (pPtr, 1, iWidth, pFp); pPtr += iStride[0]; } iHeight = iHeight / 2; iWidth = iWidth / 2; pPtr = pData[1]; for (i = 0; i < iHeight; i++) { fwrite (pPtr, 1, iWidth, pFp); pPtr += iStride[1]; } pPtr = pData[2]; for (i = 0; i < iHeight; i++) { fwrite (pPtr, 1, iWidth, pFp); pPtr += iStride[1]; } } static void Process (SBufferInfo* pInfo, FILE* pFp) { if (pFp && pInfo->pDst[0] && pInfo->pDst[1] && pInfo->pDst[2] && pInfo) { int iStride[2]; int iWidth = pInfo->UsrData.sSystemBuffer.iWidth; int iHeight = pInfo->UsrData.sSystemBuffer.iHeight; iStride[0] = pInfo->UsrData.sSystemBuffer.iStride[0]; iStride[1] = pInfo->UsrData.sSystemBuffer.iStride[1]; Write2File (pFp, (unsigned char**)pInfo->pDst, iStride, iWidth, iHeight); } } static bool ReadFrame (std::ifstream* file, BufferedData* buf) { // start code of a frame is {0, 0, 1} or {0, 0, 0, 1} char b; buf->Clear(); int32_t sps_count = 0; int32_t non_idr_pict_count = 0; int32_t idr_pict_count = 0; int32_t zeroCount = 0; for (;;) { file->read (&b, 1); if (file->gcount() != 1) { // end of file return true; } if (!buf->PushBack (b)) { std::cout << "unable to allocate memory" << std::endl; return false; } if (buf->Length() < 5) { continue; } uint8_t nal_unit_type = 0; int32_t startcode_len_plus_one = 0; if (buf->Length() == 5) { if (buf->data()[2] == 1) { nal_unit_type = buf->data()[3] & 0x1F; } else { nal_unit_type = buf->data()[4] & 0x1F; } } else { if (zeroCount < 2) { zeroCount = b != 0 ? 0 : zeroCount + 1; } if (zeroCount == 2) { file->read (&b, 1); if (file->gcount() != 1) { // end of file return true; } if (!buf->PushBack (b)) { std::cout << "unable to allocate memory" << std::endl; return false; } if (b == 1) { //0x000001 file->read (&b, 1); if (file->gcount() != 1) { // end of file return true; } if (!buf->PushBack (b)) { std::cout << "unable to allocate memory" << std::endl; return false; } nal_unit_type = b & 0x1F; startcode_len_plus_one = 4; zeroCount = 0; } else if (b == 0) { file->read (&b, 1); if (file->gcount() != 1) { // end of file return true; } if (!buf->PushBack (b)) { std::cout << "unable to allocate memory" << std::endl; return false; } if (b == 1) { //0x00000001 file->read (&b, 1); if (file->gcount() != 1) { // end of file return true; } if (!buf->PushBack (b)) { std::cout << "unable to allocate memory" << std::endl; return false; } nal_unit_type = b & 0x1F; startcode_len_plus_one = 5; zeroCount = 0; } else { zeroCount = 0; } } else { zeroCount = 0; } } } if (nal_unit_type == 1) { if (++non_idr_pict_count == 1 && idr_pict_count == 1) { file->seekg (-startcode_len_plus_one, file->cur).good(); buf->SetLength (buf->Length() - startcode_len_plus_one); return true; } if (non_idr_pict_count == 2) { file->seekg (-startcode_len_plus_one, file->cur).good(); buf->SetLength (buf->Length() - startcode_len_plus_one); return true; } } else if (nal_unit_type == 5) { if (++idr_pict_count == 1 && non_idr_pict_count == 1) { file->seekg (-startcode_len_plus_one, file->cur).good(); buf->SetLength (buf->Length() - startcode_len_plus_one); return true; } if (idr_pict_count == 2) { file->seekg (-startcode_len_plus_one, file->cur).good(); buf->SetLength (buf->Length() - startcode_len_plus_one); return true; } } else if (nal_unit_type == 7) { if ((++sps_count == 1) && (non_idr_pict_count == 1 || idr_pict_count == 1)) { file->seekg (-startcode_len_plus_one, file->cur).good(); buf->SetLength (buf->Length() - startcode_len_plus_one); return true; } } } } BaseThreadDecoderTest::BaseThreadDecoderTest() : decoder_ (NULL), uiTimeStamp (0), pYuvFile (NULL), bEnableYuvDumpTest (false), decodeStatus_ (OpenFile) { } int32_t BaseThreadDecoderTest::SetUp() { long rv = WelsCreateDecoder (&decoder_); EXPECT_EQ (0, rv); EXPECT_TRUE (decoder_ != NULL); if (decoder_ == NULL) { return rv; } SDecodingParam decParam; memset (&decParam, 0, sizeof (SDecodingParam)); decParam.uiTargetDqLayer = UCHAR_MAX; decParam.eEcActiveIdc = ERROR_CON_SLICE_COPY; decParam.sVideoProperty.eVideoBsType = VIDEO_BITSTREAM_DEFAULT; int iThreadCount = (rand() % 2) + 2; fprintf (stderr, "iThreadCount=%d\n", iThreadCount); decoder_->SetOption (DECODER_OPTION_NUM_OF_THREADS, &iThreadCount); rv = decoder_->Initialize (&decParam); EXPECT_EQ (0, rv); return (int32_t)rv; } void BaseThreadDecoderTest::TearDown() { if (decoder_ != NULL) { decoder_->Uninitialize(); WelsDestroyDecoder (decoder_); } } void BaseThreadDecoderTest::DecodeFrame (const uint8_t* src, size_t sliceSize, Callback* cbk) { SBufferInfo bufInfo; memset (pData, 0, sizeof (pData)); memset (&bufInfo, 0, sizeof (SBufferInfo)); bufInfo.uiInBsTimeStamp = ++uiTimeStamp; DECODING_STATE rv = decoder_->DecodeFrameNoDelay (src, (int) sliceSize, pData, &bufInfo); ASSERT_TRUE (rv == dsErrorFree); sBufInfo = bufInfo; if (sBufInfo.iBufferStatus == 1 && cbk != NULL) { if (bEnableYuvDumpTest) { Process (&sBufInfo, pYuvFile); } const Frame frame = { { // y plane sBufInfo.pDst[0], bufInfo.UsrData.sSystemBuffer.iWidth, bufInfo.UsrData.sSystemBuffer.iHeight, bufInfo.UsrData.sSystemBuffer.iStride[0] }, { // u plane sBufInfo.pDst[1], sBufInfo.UsrData.sSystemBuffer.iWidth / 2, sBufInfo.UsrData.sSystemBuffer.iHeight / 2, sBufInfo.UsrData.sSystemBuffer.iStride[1] }, { // v plane sBufInfo.pDst[2], sBufInfo.UsrData.sSystemBuffer.iWidth / 2, sBufInfo.UsrData.sSystemBuffer.iHeight / 2, sBufInfo.UsrData.sSystemBuffer.iStride[1] }, }; cbk->onDecodeFrame (frame); } } void BaseThreadDecoderTest::FlushFrame (Callback* cbk) { SBufferInfo bufInfo; memset (pData, 0, sizeof (pData)); memset (&bufInfo, 0, sizeof (SBufferInfo)); DECODING_STATE rv = decoder_->FlushFrame (pData, &bufInfo); ASSERT_TRUE (rv == dsErrorFree); sBufInfo = bufInfo; if (sBufInfo.iBufferStatus == 1 && cbk != NULL) { if (bEnableYuvDumpTest) { Process (&sBufInfo, pYuvFile); } const Frame frame = { { // y plane sBufInfo.pDst[0], sBufInfo.UsrData.sSystemBuffer.iWidth, sBufInfo.UsrData.sSystemBuffer.iHeight, sBufInfo.UsrData.sSystemBuffer.iStride[0] }, { // u plane sBufInfo.pDst[1], sBufInfo.UsrData.sSystemBuffer.iWidth / 2, sBufInfo.UsrData.sSystemBuffer.iHeight / 2, sBufInfo.UsrData.sSystemBuffer.iStride[1] }, { // v plane sBufInfo.pDst[2], sBufInfo.UsrData.sSystemBuffer.iWidth / 2, sBufInfo.UsrData.sSystemBuffer.iHeight / 2, sBufInfo.UsrData.sSystemBuffer.iStride[1] }, }; cbk->onDecodeFrame (frame); } } bool BaseThreadDecoderTest::ThreadDecodeFile (const char* fileName, Callback* cbk) { std::ifstream file (fileName, std::ios::in | std::ios::binary); if (!file.is_open()) return false; std::string outFileName = std::string (fileName); size_t pos = outFileName.find_last_of ("."); if (bEnableYuvDumpTest) { outFileName = outFileName.substr (0, pos) + std::string (".yuv"); pYuvFile = fopen (outFileName.c_str(), "wb"); } int iBufIndex = 0; uiTimeStamp = 0; memset (&sBufInfo, 0, sizeof (SBufferInfo)); while (true) { if (false == ReadFrame (&file, &buf[iBufIndex])) return false; if (::testing::Test::HasFatalFailure()) { return false; } if (buf[iBufIndex].Length() == 0) { break; } DecodeFrame (buf[iBufIndex].data(), buf[iBufIndex].Length(), cbk); if (::testing::Test::HasFatalFailure()) { return false; } if (++iBufIndex >= 16) { iBufIndex = 0; } } int32_t iEndOfStreamFlag = 1; decoder_->SetOption (DECODER_OPTION_END_OF_STREAM, &iEndOfStreamFlag); // Flush out last frames in decoder buffer int32_t num_of_frames_in_buffer = 0; decoder_->GetOption (DECODER_OPTION_NUM_OF_FRAMES_REMAINING_IN_BUFFER, &num_of_frames_in_buffer); for (int32_t i = 0; i < num_of_frames_in_buffer; ++i) { FlushFrame (cbk); } if (bEnableYuvDumpTest) { fclose (pYuvFile); } return true; } bool BaseThreadDecoderTest::Open (const char* fileName) { if (decodeStatus_ == OpenFile) { file_.open (fileName, std::ios_base::out | std::ios_base::binary); if (file_.is_open()) { decodeStatus_ = Decoding; return true; } } return false; } bool BaseThreadDecoderTest::DecodeNextFrame (Callback* cbk) { switch (decodeStatus_) { case Decoding: if (false == ReadFrame (&file_, &buf_)) return false; if (::testing::Test::HasFatalFailure()) { return false; } if (buf_.Length() == 0) { decodeStatus_ = EndOfStream; return true; } DecodeFrame (buf_.data(), buf_.Length(), cbk); if (::testing::Test::HasFatalFailure()) { return false; } return true; case EndOfStream: { int32_t iEndOfStreamFlag = 1; decoder_->SetOption (DECODER_OPTION_END_OF_STREAM, &iEndOfStreamFlag); DecodeFrame (NULL, 0, cbk); decodeStatus_ = End; break; } case OpenFile: case End: break; } return false; }