150 lines
4.0 KiB
C++
150 lines
4.0 KiB
C++
#include <gtest/gtest.h>
|
|
#include <stdint.h>
|
|
#include <fstream>
|
|
|
|
#include "codec_api.h"
|
|
|
|
#include "utils/BufferedData.h"
|
|
#include "utils/HashFunctions.h"
|
|
|
|
static int InitWithParam(ISVCEncoder* encoder, int width,
|
|
int height, float frameRate) {
|
|
SVCEncodingParam param;
|
|
memset (¶m, 0, sizeof(SVCEncodingParam));
|
|
|
|
param.sSpatialLayers[0].iVideoWidth = width;
|
|
param.sSpatialLayers[0].iVideoHeight = height;
|
|
param.sSpatialLayers[0].fFrameRate = frameRate;
|
|
param.sSpatialLayers[0].iQualityLayerNum = 1;
|
|
param.sSpatialLayers[0].iSpatialBitrate = 600000;
|
|
|
|
SSliceConfig* sliceCfg = ¶m.sSpatialLayers[0].sSliceCfg;
|
|
sliceCfg->sSliceArgument.uiSliceNum = 1;
|
|
sliceCfg->sSliceArgument.uiSliceSizeConstraint = 1500;
|
|
sliceCfg->sSliceArgument.uiSliceMbNum[0] = 960;
|
|
|
|
param.fFrameRate = param.sSpatialLayers[0].fFrameRate;
|
|
param.iPicWidth = param.sSpatialLayers[0].iVideoWidth;
|
|
param.iPicHeight = param.sSpatialLayers[0].iVideoHeight;
|
|
param.iTargetBitrate = 5000000;
|
|
param.iTemporalLayerNum = 3;
|
|
param.iSpatialLayerNum = 1;
|
|
param.bEnableBackgroundDetection = true;
|
|
param.bEnableLongTermReference = true;
|
|
param.iLtrMarkPeriod = 30;
|
|
param.iInputCsp = videoFormatI420;
|
|
param.bEnableSpsPpsIdAddition = true;
|
|
|
|
return encoder->Initialize(¶m, INIT_TYPE_PARAMETER_BASED);
|
|
}
|
|
|
|
static void UpdateHashFromFrame(const SFrameBSInfo& info, SHA_CTX* ctx) {
|
|
for (int i = 0; i < info.iLayerNum; ++i) {
|
|
const SLayerBSInfo& layerInfo = info.sLayerInfo[i];
|
|
int layerSize = 0;
|
|
for (int j = 0; j < layerInfo.iNalCount; ++j) {
|
|
layerSize += layerInfo.iNalLengthInByte[j];
|
|
}
|
|
SHA1_Update(ctx, layerInfo.pBsBuf, layerSize);
|
|
}
|
|
}
|
|
|
|
static void CompareFileToHash(ISVCEncoder* encoder,
|
|
const char* fileName, const char* hashStr,
|
|
int width, int height, float frameRate) {
|
|
std::ifstream file(fileName, std::ios::in | std::ios::binary);
|
|
ASSERT_TRUE(file.is_open());
|
|
|
|
int rv = InitWithParam(encoder, width, height, frameRate);
|
|
ASSERT_TRUE(rv == cmResultSuccess);
|
|
|
|
// I420: 1(Y) + 1/4(U) + 1/4(V)
|
|
int frameSize = width * height * 3 / 2;
|
|
|
|
BufferedData buf;
|
|
buf.SetLength(frameSize);
|
|
ASSERT_TRUE(buf.Length() == frameSize);
|
|
char* data = reinterpret_cast<char*>(buf.data());
|
|
|
|
SFrameBSInfo info;
|
|
memset(&info, 0, sizeof(SFrameBSInfo));
|
|
|
|
unsigned char digest[SHA_DIGEST_LENGTH];
|
|
SHA_CTX ctx;
|
|
SHA1_Init(&ctx);
|
|
|
|
while (file.read(data, frameSize), file.gcount() == frameSize) {
|
|
rv = encoder->EncodeFrame(buf.data(), &info);
|
|
if (rv == videoFrameTypeInvalid) {
|
|
SHA1_Final(digest, &ctx);
|
|
FAIL() << "unable to encode frame";
|
|
}
|
|
if (rv != videoFrameTypeSkip) {
|
|
UpdateHashFromFrame(info, &ctx);
|
|
}
|
|
}
|
|
|
|
SHA1_Final(digest, &ctx);
|
|
ASSERT_TRUE(CompareHash(digest, hashStr));
|
|
}
|
|
|
|
class EncoderBaseTest : public ::testing::Test {
|
|
public:
|
|
EncoderBaseTest() : encoder_(NULL) {}
|
|
|
|
virtual void SetUp() {
|
|
int rv = CreateSVCEncoder(&encoder_);
|
|
ASSERT_EQ(0, rv);
|
|
ASSERT_TRUE(encoder_ != NULL);
|
|
}
|
|
|
|
virtual void TearDown() {
|
|
if (encoder_) {
|
|
encoder_->Uninitialize();
|
|
DestroySVCEncoder(encoder_);
|
|
}
|
|
}
|
|
|
|
protected:
|
|
ISVCEncoder* encoder_;
|
|
};
|
|
|
|
TEST_F(EncoderBaseTest, JustInit) {
|
|
}
|
|
|
|
struct EncodeFileParam {
|
|
const char* fileName;
|
|
const char* hashStr;
|
|
int width;
|
|
int height;
|
|
float frameRate;
|
|
};
|
|
|
|
class EncoderOutputTest : public EncoderBaseTest ,
|
|
public ::testing::WithParamInterface<EncodeFileParam> {
|
|
};
|
|
|
|
|
|
TEST_P(EncoderOutputTest, CompareOutput) {
|
|
EncodeFileParam p = GetParam();
|
|
CompareFileToHash(encoder_, p.fileName, p.hashStr, p.width, p.height, p.frameRate);
|
|
}
|
|
|
|
static const EncodeFileParam kFileParamArray[] = {
|
|
{
|
|
"res/CiscoVT2people_320x192_12fps.yuv",
|
|
"4df5751a59eb02153e086ade9b3ecfcb8845c30b", 320, 192, 12.0f
|
|
},
|
|
{
|
|
"res/CiscoVT2people_160x96_6fps.yuv",
|
|
"6eb53b6bfdb95dfca0575bd3efe81aa58163951c", 160, 96, 6.0f
|
|
},
|
|
{
|
|
"res/Static_152_100.yuv",
|
|
"c5af55647a5ead570bd5d96651e05ea699762b8e", 152, 100, 6.0f
|
|
},
|
|
};
|
|
|
|
INSTANTIATE_TEST_CASE_P(EncodeFile, EncoderOutputTest,
|
|
::testing::ValuesIn(kFileParamArray));
|