refine the UT case for checking IDR request

This commit is contained in:
lyao2
2014-12-11 14:32:26 +08:00
parent 5ff55a6ec2
commit 2e85e3c7d9

View File

@@ -122,7 +122,7 @@ void EncoderInterfaceTest::EncodeOneIDRandP (ISVCEncoder* pPtrEnc) {
int iResult;
iResult = pPtrEnc->EncodeFrame (pSrcPic, &sFbi);
EXPECT_EQ (iResult, static_cast<int> (cmResultSuccess));
//EXPECT_EQ (sFbi.eFrameType, static_cast<int> (videoFrameTypeIDR));
EXPECT_EQ (sFbi.eFrameType, static_cast<int> (videoFrameTypeIDR));
pSrcPic->uiTimeStamp += 30;
iResult = pPtrEnc->EncodeFrame (pSrcPic, &sFbi);
@@ -619,7 +619,13 @@ TEST_F (EncoderInterfaceTest, ForceIntraFrame) {
//call next frame to be IDR
pPtrEnc->ForceIntraFrame (bIDR);
EncodeOneIDRandP (pPtrEnc);
int iCount = 0;
do {
iResult = pPtrEnc->EncodeFrame (pSrcPic, &sFbi);
EXPECT_EQ (iResult, static_cast<int> (cmResultSuccess));
pSrcPic->uiTimeStamp += 30;
} while ((sFbi.eFrameType == static_cast<int> (videoFrameTypeSkip)) && (iCount ++ < 100));
EXPECT_EQ (sFbi.eFrameType, static_cast<int> (videoFrameTypeIDR));
pPtrEnc->Uninitialize();
EXPECT_EQ (iResult, static_cast<int> (cmResultSuccess));
@@ -659,7 +665,13 @@ TEST_F (EncoderInterfaceTest, ForceIntraFrameWithTemporal) {
//call next frame to be IDR
pPtrEnc->ForceIntraFrame (bIDR);
EncodeOneIDRandP (pPtrEnc);
int iCount = 0;
do {
iResult = pPtrEnc->EncodeFrame (pSrcPic, &sFbi);
EXPECT_EQ (iResult, static_cast<int> (cmResultSuccess));
pSrcPic->uiTimeStamp += 30;
} while ((sFbi.eFrameType == static_cast<int> (videoFrameTypeSkip)) && (iCount ++ < 100));
EXPECT_EQ (sFbi.eFrameType, static_cast<int> (videoFrameTypeIDR));
pPtrEnc->Uninitialize();
EXPECT_EQ (iResult, static_cast<int> (cmResultSuccess));