ref: 84ff16c0156e71de20af1cf149f26e3433213174
dir: /test/api/BaseDecoderTest.cpp/
#include <fstream> #include <gtest/gtest.h> #include "codec_def.h" #include "codec_app_def.h" #include "utils/BufferedData.h" #include "BaseDecoderTest.h" static void ReadFrame(std::ifstream* file, BufferedData* buf) { // start code of a frame is {0, 0, 0, 1} int zeroCount = 0; char b; buf->Clear(); for (;;) { file->read(&b, 1); if (file->gcount() != 1) { // end of file return; } if (!buf->PushBack(b)) { FAIL() << "unable to allocate memory"; } if (buf->Length() <= 4) { continue; } if (zeroCount < 3) { zeroCount = b != 0 ? 0 : zeroCount + 1; } else { if (b == 1) { if (file->seekg(-4, file->cur).good()) { buf->SetLength(buf->Length() - 4); return; } else { FAIL() << "unable to seek file"; } } else if (b == 0) { zeroCount = 3; } else { zeroCount = 0; } } } } BaseDecoderTest::BaseDecoderTest() : decoder_(NULL), decodeStatus_(OpenFile) {} void BaseDecoderTest::SetUp() { long rv = WelsCreateDecoder(&decoder_); ASSERT_EQ(0, rv); ASSERT_TRUE(decoder_ != NULL); SDecodingParam decParam; memset(&decParam, 0, sizeof(SDecodingParam)); decParam.iOutputColorFormat = videoFormatI420; decParam.uiTargetDqLayer = UCHAR_MAX; decParam.uiEcActiveFlag = 1; decParam.sVideoProperty.eVideoBsType = VIDEO_BITSTREAM_DEFAULT; rv = decoder_->Initialize(&decParam); ASSERT_EQ(0, rv); } void BaseDecoderTest::TearDown() { if (decoder_ != NULL) { decoder_->Uninitialize(); WelsDestroyDecoder(decoder_); } } void BaseDecoderTest::DecodeFrame(const uint8_t* src, int sliceSize, Callback* cbk) { void* data[3]; SBufferInfo bufInfo; memset(data, 0, sizeof(data)); memset(&bufInfo, 0, sizeof(SBufferInfo)); DECODING_STATE rv = decoder_->DecodeFrame2(src, sliceSize, data, &bufInfo); ASSERT_TRUE(rv == dsErrorFree); if (bufInfo.iBufferStatus == 1 && cbk != NULL) { const Frame frame = { { // y plane static_cast<uint8_t*>(data[0]), bufInfo.UsrData.sSystemBuffer.iWidth, bufInfo.UsrData.sSystemBuffer.iHeight, bufInfo.UsrData.sSystemBuffer.iStride[0] }, { // u plane static_cast<uint8_t*>(data[1]), bufInfo.UsrData.sSystemBuffer.iWidth / 2, bufInfo.UsrData.sSystemBuffer.iHeight / 2, bufInfo.UsrData.sSystemBuffer.iStride[1] }, { // v plane static_cast<uint8_t*>(data[2]), bufInfo.UsrData.sSystemBuffer.iWidth / 2, bufInfo.UsrData.sSystemBuffer.iHeight / 2, bufInfo.UsrData.sSystemBuffer.iStride[1] }, }; cbk->onDecodeFrame(frame); } } void BaseDecoderTest::DecodeFile(const char* fileName, Callback* cbk) { std::ifstream file(fileName, std::ios::in | std::ios::binary); ASSERT_TRUE(file.is_open()); BufferedData buf; while (true) { ReadFrame(&file, &buf); if (::testing::Test::HasFatalFailure()) { return; } if (buf.Length() == 0) { break; } DecodeFrame(buf.data(), buf.Length(), cbk); if (::testing::Test::HasFatalFailure()) { return; } } int32_t iEndOfStreamFlag = 1; decoder_->SetOption(DECODER_OPTION_END_OF_STREAM, &iEndOfStreamFlag); // Get pending last frame DecodeFrame(NULL, 0, cbk); } bool BaseDecoderTest::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 BaseDecoderTest::DecodeNextFrame(Callback* cbk) { switch (decodeStatus_) { case Decoding: ReadFrame(&file_, &buf_); 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; }