updating vpm unit_test following r858
Review URL: http://webrtc-codereview.appspot.com/255005 git-svn-id: http://webrtc.googlecode.com/svn/trunk@865 4adac7df-926f-26a2-2b94-8c16560cd09d
This commit is contained in:
mikhal@webrtc.org
parent
7395d3d8e9
commit
bb8dfbdee2
@ -14,352 +14,369 @@
|
||||
|
||||
using webrtc::Trace;
|
||||
|
||||
void TestSize(VideoFrame& sourceFrame, WebRtc_UWord32 targetWidth, WebRtc_UWord32 targetHeight,
|
||||
void TestSize(VideoFrame& sourceFrame,
|
||||
WebRtc_UWord32 targetWidth, WebRtc_UWord32 targetHeight,
|
||||
WebRtc_UWord32 mode, VideoProcessingModule *vpm);
|
||||
|
||||
class VPMEnvironment : public ::testing::Environment
|
||||
{
|
||||
public:
|
||||
virtual void SetUp()
|
||||
{
|
||||
Trace::CreateTrace();
|
||||
ASSERT_EQ(0, Trace::SetTraceFile("VPMTrace.txt"));
|
||||
}
|
||||
virtual void SetUp()
|
||||
{
|
||||
Trace::CreateTrace();
|
||||
ASSERT_EQ(0, Trace::SetTraceFile("VPMTrace.txt"));
|
||||
}
|
||||
|
||||
virtual void TearDown()
|
||||
{
|
||||
Trace::ReturnTrace();
|
||||
}
|
||||
virtual void TearDown()
|
||||
{
|
||||
Trace::ReturnTrace();
|
||||
}
|
||||
};
|
||||
VideoProcessingModuleTest::VideoProcessingModuleTest() :
|
||||
_vpm(NULL),
|
||||
_sourceFile(NULL),
|
||||
_width(352),
|
||||
_height(288),
|
||||
_frameLength(CalcBufferSize(kI420, 352, 288))
|
||||
_vpm(NULL),
|
||||
_sourceFile(NULL),
|
||||
_width(352),
|
||||
_height(288),
|
||||
_frameLength(CalcBufferSize(kI420, 352, 288))
|
||||
{
|
||||
}
|
||||
|
||||
void VideoProcessingModuleTest::SetUp()
|
||||
{
|
||||
_vpm = VideoProcessingModule::Create(0);
|
||||
ASSERT_TRUE(_vpm != NULL);
|
||||
_vpm = VideoProcessingModule::Create(0);
|
||||
ASSERT_TRUE(_vpm != NULL);
|
||||
|
||||
ASSERT_EQ(0, _videoFrame.VerifyAndAllocate(_frameLength));
|
||||
_videoFrame.SetWidth(_width);
|
||||
_videoFrame.SetHeight(_height);
|
||||
ASSERT_EQ(0, _videoFrame.VerifyAndAllocate(_frameLength));
|
||||
_videoFrame.SetWidth(_width);
|
||||
_videoFrame.SetHeight(_height);
|
||||
|
||||
_sourceFile = fopen("testFiles/foreman_cif.yuv","rb");
|
||||
ASSERT_TRUE(_sourceFile != NULL) <<
|
||||
"Cannot read source file: testFiles/foreman_cif.yuv\n";
|
||||
_sourceFile = fopen("testFiles/foreman_cif.yuv","rb");
|
||||
ASSERT_TRUE(_sourceFile != NULL) <<
|
||||
"Cannot read source file: testFiles/foreman_cif.yuv\n";
|
||||
}
|
||||
|
||||
void VideoProcessingModuleTest::TearDown()
|
||||
{
|
||||
if (_sourceFile != NULL)
|
||||
{
|
||||
ASSERT_EQ(0, fclose(_sourceFile));
|
||||
}
|
||||
_sourceFile = NULL;
|
||||
if (_sourceFile != NULL) {
|
||||
ASSERT_EQ(0, fclose(_sourceFile));
|
||||
}
|
||||
_sourceFile = NULL;
|
||||
|
||||
if (_vpm != NULL)
|
||||
{
|
||||
VideoProcessingModule::Destroy(_vpm);
|
||||
}
|
||||
_vpm = NULL;
|
||||
if (_vpm != NULL) {
|
||||
VideoProcessingModule::Destroy(_vpm);
|
||||
}
|
||||
_vpm = NULL;
|
||||
}
|
||||
|
||||
TEST_F(VideoProcessingModuleTest, GetVersionTest)
|
||||
{
|
||||
WebRtc_Word8 version[255];
|
||||
WebRtc_UWord32 remainingBuffer = sizeof(version);
|
||||
WebRtc_UWord32 position = 0;
|
||||
WebRtc_Word8 version[255];
|
||||
WebRtc_UWord32 remainingBuffer = sizeof(version);
|
||||
WebRtc_UWord32 position = 0;
|
||||
|
||||
EXPECT_EQ(-1, VideoProcessingModule::GetVersion(NULL, remainingBuffer, position));
|
||||
EXPECT_EQ(-1, _vpm->GetVersion(NULL, remainingBuffer, position));
|
||||
EXPECT_EQ(-1, VideoProcessingModule::GetVersion(NULL, remainingBuffer,
|
||||
position));
|
||||
EXPECT_EQ(-1, _vpm->GetVersion(NULL, remainingBuffer, position));
|
||||
|
||||
WebRtc_UWord32 badRemainingBuffer = 5;
|
||||
EXPECT_EQ(-3, VideoProcessingModule::GetVersion(version, badRemainingBuffer, position));
|
||||
EXPECT_EQ(-3, _vpm->GetVersion(version, badRemainingBuffer, position));
|
||||
WebRtc_UWord32 badRemainingBuffer = 5;
|
||||
EXPECT_EQ(-3, VideoProcessingModule::GetVersion(version, badRemainingBuffer,
|
||||
position));
|
||||
EXPECT_EQ(-3, _vpm->GetVersion(version, badRemainingBuffer, position));
|
||||
|
||||
EXPECT_EQ(0, VideoProcessingModule::GetVersion(version, remainingBuffer, position));
|
||||
EXPECT_EQ(remainingBuffer, sizeof(version) - position);
|
||||
printf("\n%s\n\n", version);
|
||||
EXPECT_EQ(0, VideoProcessingModule::GetVersion(version, remainingBuffer,
|
||||
position));
|
||||
EXPECT_EQ(remainingBuffer, sizeof(version) - position);
|
||||
printf("\n%s\n\n", version);
|
||||
|
||||
EXPECT_EQ(0, _vpm->GetVersion(&version[position], remainingBuffer, position));
|
||||
EXPECT_EQ(remainingBuffer, sizeof(version) - position);
|
||||
EXPECT_EQ(0, _vpm->GetVersion(&version[position], remainingBuffer,
|
||||
position));
|
||||
EXPECT_EQ(remainingBuffer, sizeof(version) - position);
|
||||
}
|
||||
|
||||
TEST_F(VideoProcessingModuleTest, HandleNullBuffer)
|
||||
{
|
||||
VideoProcessingModule::FrameStats stats;
|
||||
ASSERT_EQ(0, _vpm->GetFrameStats(stats, _videoFrame));
|
||||
// Video frame with unallocated buffer.
|
||||
VideoFrame videoFrame;
|
||||
videoFrame.SetWidth(_width);
|
||||
videoFrame.SetHeight(_height);
|
||||
VideoProcessingModule::FrameStats stats;
|
||||
ASSERT_EQ(0, _vpm->GetFrameStats(stats, _videoFrame));
|
||||
// Video frame with unallocated buffer.
|
||||
VideoFrame videoFrame;
|
||||
videoFrame.SetWidth(_width);
|
||||
videoFrame.SetHeight(_height);
|
||||
|
||||
EXPECT_EQ(-3, _vpm->GetFrameStats(stats, NULL, _width, _height));
|
||||
EXPECT_EQ(-3, _vpm->GetFrameStats(stats, videoFrame));
|
||||
EXPECT_EQ(-3, _vpm->GetFrameStats(stats, NULL, _width, _height));
|
||||
EXPECT_EQ(-3, _vpm->GetFrameStats(stats, videoFrame));
|
||||
|
||||
EXPECT_EQ(-1, _vpm->ColorEnhancement(NULL, _width, _height));
|
||||
EXPECT_EQ(-1, _vpm->ColorEnhancement(videoFrame));
|
||||
EXPECT_EQ(-1, _vpm->ColorEnhancement(NULL, _width, _height));
|
||||
EXPECT_EQ(-1, _vpm->ColorEnhancement(videoFrame));
|
||||
|
||||
EXPECT_EQ(-1, _vpm->Deflickering(NULL, _width, _height, 0, stats));
|
||||
EXPECT_EQ(-1, _vpm->Deflickering(videoFrame, stats));
|
||||
EXPECT_EQ(-1, _vpm->Deflickering(NULL, _width, _height, 0, stats));
|
||||
EXPECT_EQ(-1, _vpm->Deflickering(videoFrame, stats));
|
||||
|
||||
EXPECT_EQ(-1, _vpm->Denoising(NULL, _width, _height));
|
||||
EXPECT_EQ(-1, _vpm->Denoising(videoFrame));
|
||||
EXPECT_EQ(-1, _vpm->Denoising(NULL, _width, _height));
|
||||
EXPECT_EQ(-1, _vpm->Denoising(videoFrame));
|
||||
|
||||
EXPECT_EQ(-3, _vpm->BrightnessDetection(NULL, _width, _height, stats));
|
||||
EXPECT_EQ(-3, _vpm->BrightnessDetection(videoFrame, stats));
|
||||
|
||||
EXPECT_EQ(VPM_PARAMETER_ERROR, _vpm->PreprocessFrame(NULL, NULL));
|
||||
EXPECT_EQ(-3, _vpm->BrightnessDetection(NULL, _width, _height, stats));
|
||||
EXPECT_EQ(-3, _vpm->BrightnessDetection(videoFrame, stats));
|
||||
|
||||
EXPECT_EQ(VPM_PARAMETER_ERROR, _vpm->PreprocessFrame(NULL, NULL));
|
||||
}
|
||||
|
||||
TEST_F(VideoProcessingModuleTest, HandleBadStats)
|
||||
{
|
||||
VideoProcessingModule::FrameStats stats;
|
||||
|
||||
ASSERT_EQ(_frameLength, fread(_videoFrame.Buffer(), 1, _frameLength, _sourceFile));
|
||||
|
||||
EXPECT_EQ(-1, _vpm->Deflickering(_videoFrame.Buffer(), _width, _height, 0, stats));
|
||||
EXPECT_EQ(-1, _vpm->Deflickering(_videoFrame, stats));
|
||||
VideoProcessingModule::FrameStats stats;
|
||||
|
||||
EXPECT_EQ(-3, _vpm->BrightnessDetection(_videoFrame.Buffer(), _width, _height, stats));
|
||||
EXPECT_EQ(-3, _vpm->BrightnessDetection(_videoFrame, stats));
|
||||
ASSERT_EQ(_frameLength, fread(_videoFrame.Buffer(), 1, _frameLength,
|
||||
_sourceFile));
|
||||
|
||||
EXPECT_EQ(-1, _vpm->Deflickering(_videoFrame.Buffer(), _width, _height, 0,
|
||||
stats));
|
||||
EXPECT_EQ(-1, _vpm->Deflickering(_videoFrame, stats));
|
||||
|
||||
EXPECT_EQ(-3, _vpm->BrightnessDetection(_videoFrame.Buffer(), _width,
|
||||
_height, stats));
|
||||
EXPECT_EQ(-3, _vpm->BrightnessDetection(_videoFrame, stats));
|
||||
}
|
||||
|
||||
TEST_F(VideoProcessingModuleTest, HandleBadSize)
|
||||
{
|
||||
VideoProcessingModule::FrameStats stats;
|
||||
ASSERT_EQ(0, _vpm->GetFrameStats(stats, _videoFrame));
|
||||
VideoProcessingModule::FrameStats stats;
|
||||
ASSERT_EQ(0, _vpm->GetFrameStats(stats, _videoFrame));
|
||||
|
||||
// Bad width
|
||||
_videoFrame.SetWidth(0);
|
||||
EXPECT_EQ(-3, _vpm->GetFrameStats(stats, _videoFrame.Buffer(), 0, _height));
|
||||
EXPECT_EQ(-3, _vpm->GetFrameStats(stats, _videoFrame));
|
||||
|
||||
EXPECT_EQ(-1, _vpm->ColorEnhancement(_videoFrame.Buffer(), 0, _height));
|
||||
EXPECT_EQ(-1, _vpm->ColorEnhancement(_videoFrame));
|
||||
// Bad width
|
||||
_videoFrame.SetWidth(0);
|
||||
EXPECT_EQ(-3, _vpm->GetFrameStats(stats, _videoFrame.Buffer(), 0, _height));
|
||||
EXPECT_EQ(-3, _vpm->GetFrameStats(stats, _videoFrame));
|
||||
|
||||
EXPECT_EQ(-1, _vpm->Deflickering(_videoFrame.Buffer(), 0, _height, 0, stats));
|
||||
EXPECT_EQ(-1, _vpm->Deflickering(_videoFrame, stats));
|
||||
EXPECT_EQ(-1, _vpm->ColorEnhancement(_videoFrame.Buffer(), 0, _height));
|
||||
EXPECT_EQ(-1, _vpm->ColorEnhancement(_videoFrame));
|
||||
|
||||
EXPECT_EQ(-1, _vpm->Denoising(_videoFrame.Buffer(), 0, _height));
|
||||
EXPECT_EQ(-1, _vpm->Denoising(_videoFrame));
|
||||
EXPECT_EQ(-1, _vpm->Deflickering(_videoFrame.Buffer(), 0, _height, 0,
|
||||
stats));
|
||||
EXPECT_EQ(-1, _vpm->Deflickering(_videoFrame, stats));
|
||||
|
||||
EXPECT_EQ(-3, _vpm->BrightnessDetection(_videoFrame.Buffer(), 0, _height, stats));
|
||||
EXPECT_EQ(-3, _vpm->BrightnessDetection(_videoFrame, stats));
|
||||
EXPECT_EQ(-1, _vpm->Denoising(_videoFrame.Buffer(), 0, _height));
|
||||
EXPECT_EQ(-1, _vpm->Denoising(_videoFrame));
|
||||
|
||||
EXPECT_EQ(-3, _vpm->BrightnessDetection(_videoFrame.Buffer(), 0, _height,
|
||||
stats));
|
||||
EXPECT_EQ(-3, _vpm->BrightnessDetection(_videoFrame, stats));
|
||||
|
||||
|
||||
// Bad height
|
||||
_videoFrame.SetWidth(_width);
|
||||
_videoFrame.SetHeight(0);
|
||||
EXPECT_EQ(-3, _vpm->GetFrameStats(stats, _videoFrame.Buffer(), _width, 0));
|
||||
EXPECT_EQ(-3, _vpm->GetFrameStats(stats, _videoFrame));
|
||||
|
||||
EXPECT_EQ(-1, _vpm->ColorEnhancement(_videoFrame.Buffer(), _width, 0));
|
||||
EXPECT_EQ(-1, _vpm->ColorEnhancement(_videoFrame));
|
||||
// Bad height
|
||||
_videoFrame.SetWidth(_width);
|
||||
_videoFrame.SetHeight(0);
|
||||
EXPECT_EQ(-3, _vpm->GetFrameStats(stats, _videoFrame.Buffer(), _width, 0));
|
||||
EXPECT_EQ(-3, _vpm->GetFrameStats(stats, _videoFrame));
|
||||
|
||||
EXPECT_EQ(-1, _vpm->Deflickering(_videoFrame.Buffer(), _width, 0, 0, stats));
|
||||
EXPECT_EQ(-1, _vpm->Deflickering(_videoFrame, stats));
|
||||
EXPECT_EQ(-1, _vpm->ColorEnhancement(_videoFrame.Buffer(), _width, 0));
|
||||
EXPECT_EQ(-1, _vpm->ColorEnhancement(_videoFrame));
|
||||
|
||||
EXPECT_EQ(-1, _vpm->Denoising(_videoFrame.Buffer(), _width, 0));
|
||||
EXPECT_EQ(-1, _vpm->Denoising(_videoFrame));
|
||||
EXPECT_EQ(-1, _vpm->Deflickering(_videoFrame.Buffer(), _width, 0, 0,
|
||||
stats));
|
||||
EXPECT_EQ(-1, _vpm->Deflickering(_videoFrame, stats));
|
||||
|
||||
EXPECT_EQ(-3, _vpm->BrightnessDetection(_videoFrame.Buffer(), _width, 0, stats));
|
||||
EXPECT_EQ(-3, _vpm->BrightnessDetection(_videoFrame, stats));
|
||||
EXPECT_EQ(-1, _vpm->Denoising(_videoFrame.Buffer(), _width, 0));
|
||||
EXPECT_EQ(-1, _vpm->Denoising(_videoFrame));
|
||||
|
||||
EXPECT_EQ(VPM_PARAMETER_ERROR, _vpm->SetTargetResolution(0,0,0));
|
||||
EXPECT_EQ(VPM_PARAMETER_ERROR, _vpm->SetMaxFrameRate(0));
|
||||
EXPECT_EQ(-3, _vpm->BrightnessDetection(_videoFrame.Buffer(), _width, 0,
|
||||
stats));
|
||||
EXPECT_EQ(-3, _vpm->BrightnessDetection(_videoFrame, stats));
|
||||
|
||||
VideoFrame *outFrame = NULL;
|
||||
EXPECT_EQ(VPM_PARAMETER_ERROR, _vpm->PreprocessFrame(&_videoFrame, &outFrame));
|
||||
EXPECT_EQ(VPM_PARAMETER_ERROR, _vpm->SetTargetResolution(0,0,0));
|
||||
EXPECT_EQ(VPM_PARAMETER_ERROR, _vpm->SetMaxFrameRate(0));
|
||||
|
||||
VideoFrame *outFrame = NULL;
|
||||
EXPECT_EQ(VPM_PARAMETER_ERROR, _vpm->PreprocessFrame(&_videoFrame,
|
||||
&outFrame));
|
||||
}
|
||||
|
||||
TEST_F(VideoProcessingModuleTest, IdenticalResultsAfterReset)
|
||||
{
|
||||
VideoFrame videoFrame2;
|
||||
VideoProcessingModule::FrameStats stats;
|
||||
VideoFrame videoFrame2;
|
||||
VideoProcessingModule::FrameStats stats;
|
||||
|
||||
ASSERT_EQ(0, videoFrame2.VerifyAndAllocate(_frameLength));
|
||||
videoFrame2.SetWidth(_width);
|
||||
videoFrame2.SetHeight(_height);
|
||||
ASSERT_EQ(0, videoFrame2.VerifyAndAllocate(_frameLength));
|
||||
videoFrame2.SetWidth(_width);
|
||||
videoFrame2.SetHeight(_height);
|
||||
|
||||
// Only testing non-static functions here.
|
||||
ASSERT_EQ(_frameLength, fread(_videoFrame.Buffer(), 1, _frameLength, _sourceFile));
|
||||
ASSERT_EQ(0, _vpm->GetFrameStats(stats, _videoFrame));
|
||||
memcpy(videoFrame2.Buffer(), _videoFrame.Buffer(), _frameLength);
|
||||
ASSERT_EQ(0, _vpm->Deflickering(_videoFrame, stats));
|
||||
_vpm->Reset();
|
||||
// Retrieve frame stats again in case Deflickering() has zeroed them.
|
||||
ASSERT_EQ(0, _vpm->GetFrameStats(stats, videoFrame2));
|
||||
ASSERT_EQ(0, _vpm->Deflickering(videoFrame2, stats));
|
||||
EXPECT_EQ(0, memcmp(_videoFrame.Buffer(), videoFrame2.Buffer(), _frameLength));
|
||||
// Only testing non-static functions here.
|
||||
ASSERT_EQ(_frameLength, fread(_videoFrame.Buffer(), 1, _frameLength,
|
||||
_sourceFile));
|
||||
ASSERT_EQ(0, _vpm->GetFrameStats(stats, _videoFrame));
|
||||
memcpy(videoFrame2.Buffer(), _videoFrame.Buffer(), _frameLength);
|
||||
ASSERT_EQ(0, _vpm->Deflickering(_videoFrame, stats));
|
||||
_vpm->Reset();
|
||||
// Retrieve frame stats again in case Deflickering() has zeroed them.
|
||||
ASSERT_EQ(0, _vpm->GetFrameStats(stats, videoFrame2));
|
||||
ASSERT_EQ(0, _vpm->Deflickering(videoFrame2, stats));
|
||||
EXPECT_EQ(0, memcmp(_videoFrame.Buffer(), videoFrame2.Buffer(),
|
||||
_frameLength));
|
||||
|
||||
ASSERT_EQ(_frameLength, fread(_videoFrame.Buffer(), 1, _frameLength, _sourceFile));
|
||||
memcpy(videoFrame2.Buffer(), _videoFrame.Buffer(), _frameLength);
|
||||
ASSERT_GE(_vpm->Denoising(_videoFrame), 0);
|
||||
_vpm->Reset();
|
||||
ASSERT_GE(_vpm->Denoising(videoFrame2), 0);
|
||||
EXPECT_EQ(0, memcmp(_videoFrame.Buffer(), videoFrame2.Buffer(), _frameLength));
|
||||
|
||||
ASSERT_EQ(_frameLength, fread(_videoFrame.Buffer(), 1, _frameLength, _sourceFile));
|
||||
ASSERT_EQ(0, _vpm->GetFrameStats(stats, _videoFrame));
|
||||
memcpy(videoFrame2.Buffer(), _videoFrame.Buffer(), _frameLength);
|
||||
ASSERT_EQ(0, _vpm->BrightnessDetection(_videoFrame, stats));
|
||||
_vpm->Reset();
|
||||
ASSERT_EQ(0, _vpm->BrightnessDetection(videoFrame2, stats));
|
||||
EXPECT_EQ(0, memcmp(_videoFrame.Buffer(), videoFrame2.Buffer(), _frameLength));
|
||||
ASSERT_EQ(_frameLength, fread(_videoFrame.Buffer(), 1, _frameLength,
|
||||
_sourceFile));
|
||||
memcpy(videoFrame2.Buffer(), _videoFrame.Buffer(), _frameLength);
|
||||
ASSERT_GE(_vpm->Denoising(_videoFrame), 0);
|
||||
_vpm->Reset();
|
||||
ASSERT_GE(_vpm->Denoising(videoFrame2), 0);
|
||||
EXPECT_EQ(0, memcmp(_videoFrame.Buffer(), videoFrame2.Buffer(),
|
||||
_frameLength));
|
||||
|
||||
ASSERT_EQ(_frameLength, fread(_videoFrame.Buffer(), 1, _frameLength,
|
||||
_sourceFile));
|
||||
ASSERT_EQ(0, _vpm->GetFrameStats(stats, _videoFrame));
|
||||
memcpy(videoFrame2.Buffer(), _videoFrame.Buffer(), _frameLength);
|
||||
ASSERT_EQ(0, _vpm->BrightnessDetection(_videoFrame, stats));
|
||||
_vpm->Reset();
|
||||
ASSERT_EQ(0, _vpm->BrightnessDetection(videoFrame2, stats));
|
||||
EXPECT_EQ(0, memcmp(_videoFrame.Buffer(), videoFrame2.Buffer(),
|
||||
_frameLength));
|
||||
}
|
||||
|
||||
TEST_F(VideoProcessingModuleTest, FrameStats)
|
||||
{
|
||||
VideoProcessingModule::FrameStats stats;
|
||||
ASSERT_EQ(_frameLength, fread(_videoFrame.Buffer(), 1, _frameLength, _sourceFile));
|
||||
VideoProcessingModule::FrameStats stats;
|
||||
ASSERT_EQ(_frameLength, fread(_videoFrame.Buffer(), 1, _frameLength,
|
||||
_sourceFile));
|
||||
|
||||
EXPECT_EQ(false, _vpm->ValidFrameStats(stats));
|
||||
EXPECT_EQ(0, _vpm->GetFrameStats(stats, _videoFrame));
|
||||
EXPECT_EQ(true, _vpm->ValidFrameStats(stats));
|
||||
EXPECT_EQ(false, _vpm->ValidFrameStats(stats));
|
||||
EXPECT_EQ(0, _vpm->GetFrameStats(stats, _videoFrame));
|
||||
EXPECT_EQ(true, _vpm->ValidFrameStats(stats));
|
||||
|
||||
printf("\nFrameStats\n");
|
||||
printf("mean: %u\nnumPixels: %u\nsubSamplWidth: %u\nsumSamplHeight: %u\nsum: %u\n\n",
|
||||
static_cast<unsigned int>(stats.mean),
|
||||
static_cast<unsigned int>(stats.numPixels),
|
||||
static_cast<unsigned int>(stats.subSamplHeight),
|
||||
static_cast<unsigned int>(stats.subSamplWidth),
|
||||
static_cast<unsigned int>(stats.sum));
|
||||
printf("\nFrameStats\n");
|
||||
printf("mean: %u\nnumPixels: %u\nsubSamplWidth: "
|
||||
"%u\nsumSamplHeight: %u\nsum: %u\n\n",
|
||||
static_cast<unsigned int>(stats.mean),
|
||||
static_cast<unsigned int>(stats.numPixels),
|
||||
static_cast<unsigned int>(stats.subSamplHeight),
|
||||
static_cast<unsigned int>(stats.subSamplWidth),
|
||||
static_cast<unsigned int>(stats.sum));
|
||||
|
||||
_vpm->ClearFrameStats(stats);
|
||||
EXPECT_EQ(false, _vpm->ValidFrameStats(stats));
|
||||
_vpm->ClearFrameStats(stats);
|
||||
EXPECT_EQ(false, _vpm->ValidFrameStats(stats));
|
||||
}
|
||||
|
||||
TEST_F(VideoProcessingModuleTest, PreprocessorLogic)
|
||||
{
|
||||
// disable temporal sampling
|
||||
_vpm->EnableTemporalDecimation(false);
|
||||
ASSERT_EQ(VPM_OK, _vpm->SetMaxFrameRate(30));
|
||||
ASSERT_EQ(VPM_OK, _vpm->SetTargetResolution(100, 100, 15));
|
||||
//revert
|
||||
_vpm->EnableTemporalDecimation(true);
|
||||
ASSERT_EQ(VPM_OK, _vpm->SetTargetResolution(100, 100, 30));
|
||||
// disable spatial sampling
|
||||
_vpm->SetInputFrameResampleMode(kNoRescaling);
|
||||
ASSERT_EQ(VPM_OK, _vpm->SetTargetResolution(100, 100, 30));
|
||||
VideoFrame *outFrame = NULL;
|
||||
ASSERT_EQ(VPM_OK, _vpm->PreprocessFrame(&_videoFrame, &outFrame));
|
||||
ASSERT_EQ(_videoFrame.Height() , outFrame->Height());
|
||||
ASSERT_EQ(_videoFrame.Width() , outFrame->Width());
|
||||
|
||||
// Disable temporal sampling
|
||||
_vpm->EnableTemporalDecimation(false);
|
||||
ASSERT_EQ(VPM_OK, _vpm->SetMaxFrameRate(30));
|
||||
ASSERT_EQ(VPM_OK, _vpm->SetTargetResolution(100, 100, 15));
|
||||
// Revert
|
||||
_vpm->EnableTemporalDecimation(true);
|
||||
ASSERT_EQ(VPM_OK, _vpm->SetTargetResolution(100, 100, 30));
|
||||
// Disable spatial sampling
|
||||
_vpm->SetInputFrameResampleMode(kNoRescaling);
|
||||
ASSERT_EQ(VPM_OK, _vpm->SetTargetResolution(100, 100, 30));
|
||||
VideoFrame *outFrame = NULL;
|
||||
ASSERT_EQ(VPM_OK, _vpm->PreprocessFrame(&_videoFrame, &outFrame));
|
||||
// No rescaling=> output frame = NULL
|
||||
ASSERT_TRUE(outFrame == NULL);
|
||||
}
|
||||
|
||||
TEST_F(VideoProcessingModuleTest, Resampler)
|
||||
{
|
||||
enum { NumRuns = 1 };
|
||||
enum { NumRuns = 1 };
|
||||
|
||||
WebRtc_Word64 minRuntime = 0;
|
||||
WebRtc_Word64 avgRuntime = 0;
|
||||
WebRtc_Word64 minRuntime = 0;
|
||||
WebRtc_Word64 avgRuntime = 0;
|
||||
|
||||
TickTime t0;
|
||||
TickTime t1;
|
||||
TickInterval accTicks;
|
||||
WebRtc_Word32 height = 288;
|
||||
WebRtc_Word32 width = 352;
|
||||
WebRtc_Word32 lengthSourceFrame = width*height*3/2;
|
||||
TickTime t0;
|
||||
TickTime t1;
|
||||
TickInterval accTicks;
|
||||
WebRtc_Word32 height = 288;
|
||||
WebRtc_Word32 width = 352;
|
||||
WebRtc_Word32 lengthSourceFrame = width*height*3/2;
|
||||
|
||||
rewind(_sourceFile);
|
||||
ASSERT_TRUE(_sourceFile != NULL) <<
|
||||
"Cannot read input file \n";
|
||||
rewind(_sourceFile);
|
||||
ASSERT_TRUE(_sourceFile != NULL) <<
|
||||
"Cannot read input file \n";
|
||||
|
||||
// CA not needed here
|
||||
_vpm->EnableContentAnalysis(false);
|
||||
// no temporal decimation
|
||||
_vpm->EnableTemporalDecimation(false);
|
||||
// CA not needed here
|
||||
_vpm->EnableContentAnalysis(false);
|
||||
// no temporal decimation
|
||||
_vpm->EnableTemporalDecimation(false);
|
||||
|
||||
// reading test frame
|
||||
VideoFrame sourceFrame;
|
||||
ASSERT_EQ(0, sourceFrame.VerifyAndAllocate(lengthSourceFrame));
|
||||
fread(sourceFrame.Buffer(), 1, lengthSourceFrame, _sourceFile);
|
||||
ASSERT_EQ(0, sourceFrame.SetLength(lengthSourceFrame));
|
||||
sourceFrame.SetHeight(height);
|
||||
sourceFrame.SetWidth(width);
|
||||
// Reading test frame
|
||||
VideoFrame sourceFrame;
|
||||
ASSERT_EQ(0, sourceFrame.VerifyAndAllocate(lengthSourceFrame));
|
||||
fread(sourceFrame.Buffer(), 1, lengthSourceFrame, _sourceFile);
|
||||
ASSERT_EQ(0, sourceFrame.SetLength(lengthSourceFrame));
|
||||
sourceFrame.SetHeight(height);
|
||||
sourceFrame.SetWidth(width);
|
||||
|
||||
for (WebRtc_UWord32 runIdx = 0; runIdx < NumRuns; runIdx++)
|
||||
{
|
||||
// initiate test timer
|
||||
t0 = TickTime::Now();
|
||||
for (WebRtc_UWord32 runIdx = 0; runIdx < NumRuns; runIdx++)
|
||||
{
|
||||
// initiate test timer
|
||||
t0 = TickTime::Now();
|
||||
|
||||
// kFastRescaling
|
||||
_vpm->SetInputFrameResampleMode(kFastRescaling);
|
||||
//TESTING DIFFERENT SIZES
|
||||
TestSize(sourceFrame, 100, 50, 1, _vpm); // Cut, decimation 1x, interpolate
|
||||
TestSize(sourceFrame, 352/2, 288/2, 1, _vpm); // Even decimation
|
||||
TestSize(sourceFrame, 352, 288, 1, _vpm); // No resampling
|
||||
TestSize(sourceFrame, 2*352, 2*288,1, _vpm); // Even upsampling
|
||||
TestSize(sourceFrame, 400, 256, 1, _vpm); // Upsampling 1.5x and cut
|
||||
TestSize(sourceFrame, 960, 720, 1, _vpm); // Upsampling 3.5x and cut
|
||||
TestSize(sourceFrame, 1280, 720, 1, _vpm); // Upsampling 4x and cut
|
||||
// kFastRescaling
|
||||
_vpm->SetInputFrameResampleMode(kFastRescaling);
|
||||
// TESTING DIFFERENT SIZES
|
||||
TestSize(sourceFrame, 100, 50, 1, _vpm); // Cut, decimation 1x, interpolate
|
||||
TestSize(sourceFrame, 352/2, 288/2, 1, _vpm); // Even decimation
|
||||
TestSize(sourceFrame, 352, 288, 1, _vpm); // No resampling
|
||||
TestSize(sourceFrame, 2*352, 2*288,1, _vpm); // Even upsampling
|
||||
TestSize(sourceFrame, 400, 256, 1, _vpm); // Upsampling 1.5x and cut
|
||||
TestSize(sourceFrame, 960, 720, 1, _vpm); // Upsampling 3.5x and cut
|
||||
TestSize(sourceFrame, 1280, 720, 1, _vpm); // Upsampling 4x and cut
|
||||
|
||||
//kBiLinear
|
||||
_vpm->SetInputFrameResampleMode(kBiLinear);
|
||||
//TESTING DIFFERENT SIZES
|
||||
TestSize(sourceFrame, 352/4, 288/4, 2, _vpm);
|
||||
TestSize(sourceFrame, 352/2, 288/2, 2, _vpm);
|
||||
TestSize(sourceFrame, 2*352, 2*288,2, _vpm);
|
||||
TestSize(sourceFrame, 480, 640, 2, _vpm);
|
||||
TestSize(sourceFrame, 960, 720, 2, _vpm);
|
||||
TestSize(sourceFrame, 1280, 720, 2, _vpm);
|
||||
// stop timer
|
||||
t1 = TickTime::Now();
|
||||
accTicks += t1 - t0;
|
||||
// kBiLinear
|
||||
_vpm->SetInputFrameResampleMode(kBiLinear);
|
||||
// TESTING DIFFERENT SIZES
|
||||
TestSize(sourceFrame, 352/4, 288/4, 2, _vpm);
|
||||
TestSize(sourceFrame, 352/2, 288/2, 2, _vpm);
|
||||
TestSize(sourceFrame, 2*352, 2*288,2, _vpm);
|
||||
TestSize(sourceFrame, 480, 640, 2, _vpm);
|
||||
TestSize(sourceFrame, 960, 720, 2, _vpm);
|
||||
TestSize(sourceFrame, 1280, 720, 2, _vpm);
|
||||
// stop timer
|
||||
t1 = TickTime::Now();
|
||||
accTicks += t1 - t0;
|
||||
|
||||
if (accTicks.Microseconds() < minRuntime || runIdx == 0)
|
||||
{
|
||||
minRuntime = accTicks.Microseconds();
|
||||
}
|
||||
avgRuntime += accTicks.Microseconds();
|
||||
if (accTicks.Microseconds() < minRuntime || runIdx == 0) {
|
||||
minRuntime = accTicks.Microseconds();
|
||||
}
|
||||
avgRuntime += accTicks.Microseconds();
|
||||
}
|
||||
|
||||
sourceFrame.Free();
|
||||
|
||||
printf("\nAverage run time = %d us / frame\n",
|
||||
//static_cast<int>(avgRuntime / frameNum / NumRuns));
|
||||
static_cast<int>(avgRuntime));
|
||||
printf("Min run time = %d us / frame\n\n",
|
||||
//static_cast<int>(minRuntime / frameNum));
|
||||
static_cast<int>(minRuntime));
|
||||
sourceFrame.Free();
|
||||
|
||||
printf("\nAverage run time = %d us / frame\n",
|
||||
//static_cast<int>(avgRuntime / frameNum / NumRuns));
|
||||
static_cast<int>(avgRuntime));
|
||||
printf("Min run time = %d us / frame\n\n",
|
||||
//static_cast<int>(minRuntime / frameNum));
|
||||
static_cast<int>(minRuntime));
|
||||
}
|
||||
|
||||
void TestSize(VideoFrame& sourceFrame, WebRtc_UWord32 targetWidth, WebRtc_UWord32 targetHeight,
|
||||
void TestSize(VideoFrame& sourceFrame, WebRtc_UWord32 targetWidth,
|
||||
WebRtc_UWord32 targetHeight,
|
||||
WebRtc_UWord32 mode, VideoProcessingModule *vpm)
|
||||
{
|
||||
VideoFrame *outFrame = NULL;
|
||||
std::ostringstream filename;
|
||||
filename << "Resampler_"<< mode <<"_" << targetWidth << "x" << targetHeight << "_30Hz_P420.yuv";
|
||||
std::cout << "Watch " << filename.str() << " and verify that it is okay." << std::endl;
|
||||
FILE* standAloneFile = fopen(filename.str().c_str(), "wb");
|
||||
ASSERT_EQ(VPM_OK, vpm->SetTargetResolution(targetWidth, targetHeight, 30));
|
||||
ASSERT_EQ(VPM_OK, vpm->PreprocessFrame(&sourceFrame, &outFrame));
|
||||
std::ostringstream filename;
|
||||
filename << "Resampler_"<< mode <<"_" << targetWidth << "x" <<
|
||||
targetHeight << "_30Hz_P420.yuv";
|
||||
std::cout << "Watch " << filename.str() << " and verify that it is okay."
|
||||
<< std::endl;
|
||||
FILE* standAloneFile = fopen(filename.str().c_str(), "wb");
|
||||
ASSERT_EQ(VPM_OK, vpm->SetTargetResolution(targetWidth, targetHeight, 30));
|
||||
ASSERT_EQ(VPM_OK, vpm->PreprocessFrame(&sourceFrame, &outFrame));
|
||||
// Length should be updated only if frame was resampled
|
||||
if (targetWidth != sourceFrame.Width() ||
|
||||
targetHeight != sourceFrame.Height()) {
|
||||
ASSERT_EQ((targetWidth * targetHeight * 3 / 2), outFrame->Length());
|
||||
|
||||
// Write to file for visual inspection
|
||||
fwrite(outFrame->Buffer(), 1, outFrame->Length(), standAloneFile);
|
||||
|
||||
outFrame->Free();
|
||||
fclose(standAloneFile);
|
||||
}
|
||||
fclose(standAloneFile);
|
||||
}
|
||||
|
||||
int main(int argc, char** argv)
|
||||
{
|
||||
::testing::InitGoogleTest(&argc, argv);
|
||||
VPMEnvironment* env = new VPMEnvironment;
|
||||
::testing::AddGlobalTestEnvironment(env);
|
||||
|
||||
::testing::InitGoogleTest(&argc, argv);
|
||||
VPMEnvironment* env = new VPMEnvironment;
|
||||
::testing::AddGlobalTestEnvironment(env);
|
||||
|
||||
return RUN_ALL_TESTS();
|
||||
return RUN_ALL_TESTS();
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user