bool YUVPixelDataGenerator (uint8_t* pPointer, int32_t iWidth, int32_t iHeight, int32_t iStride) {
#define SRC_FRAME_WIDTH (160)
#define SRC_FRAME_HEIGHT (96)
if (SRC_FRAME_WIDTH - iWidth <= 0 || SRC_FRAME_HEIGHT - iHeight <= 0) {
return false;
}
const int32_t kiFrameSize = SRC_FRAME_WIDTH * SRC_FRAME_HEIGHT;
BufferedData sBuf;
sBuf.SetLength (kiFrameSize);
if (sBuf.Length() != (size_t)kiFrameSize) {
return false;
}
FileInputStream fileStream;
if (!fileStream.Open ("res/CiscoVT2people_160x96_6fps.yuv")) {
return false;
}
if (fileStream.read (sBuf.data(), kiFrameSize) == kiFrameSize) {
int32_t iStartPosX = rand() % (SRC_FRAME_WIDTH - iWidth);
int32_t iStartPosY = rand() % (SRC_FRAME_HEIGHT - iHeight);
uint8_t* pSrcPointer = sBuf.data() + iStartPosX + iStartPosY * SRC_FRAME_WIDTH;
uint8_t* pLocalPointer = pPointer;
for (int j = 0; j < iHeight; j++) {
memcpy (pLocalPointer, pSrcPointer, iWidth * sizeof (uint8_t));
pLocalPointer += iStride;
pSrcPointer += SRC_FRAME_WIDTH;
}
return true;
}
return false;
}
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));
}
