ref: c795cfe18e91def226efca69e0af9eef82143e85
dir: /test/decoder_test.cpp/
#include <gtest/gtest.h> #include <stdint.h> #include <limits.h> #include <fstream> #include "codec_api.h" #include "utils/BufferedData.h" #include "utils/HashFunctions.h" static void UpdateHashFromPlane(SHA_CTX* ctx, const uint8_t* plane, int width, int height, int stride) { for (int i = 0; i < height; i++) { SHA1_Update(ctx, plane, width); plane += stride; } } /** * @return frame size (>= 0), or -1 for memory allocation error. */ static int ReadFrame(std::ifstream* file, BufferedData* buf) { // start code of a frame is {0, 0, 0, 1} int zeroCount = 0; char b; for (;;) { file->read(&b, 1); if (file->gcount() != 1) { break; } if (!buf->Push(b)) { return -1; } 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()) { return buf->Length() - 4; } else { // seeking fails return -1; } } else if (b == 0) { zeroCount = 3; } else { zeroCount = 0; } } } return buf->Length(); } /** * @return true if a frame is decoded successfully, otherwise false. */ static bool DecodeAndProcess(ISVCDecoder* decoder, const uint8_t* src, int sliceSize, SHA_CTX* ctx) { void* data[3]; SBufferInfo bufInfo; memset(data, 0, sizeof(data)); memset(&bufInfo, 0, sizeof(SBufferInfo)); DECODING_STATE rv = decoder->DecodeFrame2(src, sliceSize, data, &bufInfo); if (rv != dsErrorFree) { return false; } if (bufInfo.iBufferStatus == 1) { // y plane UpdateHashFromPlane(ctx, static_cast<uint8_t*>(data[0]), bufInfo.UsrData.sSystemBuffer.iWidth, bufInfo.UsrData.sSystemBuffer.iHeight, bufInfo.UsrData.sSystemBuffer.iStride[0]); // u plane UpdateHashFromPlane(ctx, static_cast<uint8_t*>(data[1]), bufInfo.UsrData.sSystemBuffer.iWidth / 2, bufInfo.UsrData.sSystemBuffer.iHeight / 2, bufInfo.UsrData.sSystemBuffer.iStride[1]); // v plane UpdateHashFromPlane(ctx, static_cast<uint8_t*>(data[2]), bufInfo.UsrData.sSystemBuffer.iWidth / 2, bufInfo.UsrData.sSystemBuffer.iHeight / 2, bufInfo.UsrData.sSystemBuffer.iStride[1]); } return true; } static void CompareFileToHash(ISVCDecoder* decoder, const char* fileName, const char* hashStr) { std::ifstream file(fileName, std::ios::in | std::ios::binary); ASSERT_TRUE(file.is_open()); unsigned char digest[SHA_DIGEST_LENGTH]; SHA_CTX ctx; SHA1_Init(&ctx); BufferedData buf; int sliceSize; while ((sliceSize = ReadFrame(&file, &buf)) > 0) { if (DecodeAndProcess(decoder, buf.data(), sliceSize, &ctx)) { buf.Clear(); } else { SHA1_Final(digest, &ctx); FAIL() << "unable to decode frame"; } } if (sliceSize < 0) { SHA1_Final(digest, &ctx); FAIL() << "unable to allocate memory"; } int32_t iEndOfStreamFlag = 1; decoder->SetOption(DECODER_OPTION_END_OF_STREAM, &iEndOfStreamFlag); // Get pending last frame if (!DecodeAndProcess(decoder, NULL, 0, &ctx)) { SHA1_Final(digest, &ctx); FAIL() << "unable to decode last frame"; } SHA1_Final(digest, &ctx); ASSERT_TRUE(CompareHash(digest, hashStr)); } class DecoderInitTest : public ::testing::Test { public: DecoderInitTest() : decoder_(NULL) {} virtual void SetUp() { long rv = CreateDecoder(&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, INIT_TYPE_PARAMETER_BASED); ASSERT_EQ(0, rv); } virtual void TearDown() { if (decoder_ != NULL) { decoder_->Uninitialize(); DestroyDecoder(decoder_); } } protected: ISVCDecoder* decoder_; }; TEST_F(DecoderInitTest, JustInit) { } struct FileParam { const char* fileName; const char* hashStr; }; class DecoderOutputTest : public DecoderInitTest, public ::testing::WithParamInterface<FileParam> { }; TEST_P(DecoderOutputTest, CompareOutput) { FileParam p = GetParam(); CompareFileToHash(decoder_, p.fileName, p.hashStr); } static const FileParam kFileParamArray[] = { {"res/test_vd_1d.264", "5827d2338b79ff82cd091c707823e466197281d3"}, {"res/test_vd_rc.264", "eea02e97bfec89d0418593a8abaaf55d02eaa1ca"}, {"res/Static.264", "91dd4a7a796805b2cd015cae8fd630d96c663f42"} }; INSTANTIATE_TEST_CASE_P(DecodeFile, DecoderOutputTest, ::testing::ValuesIn(kFileParamArray));