From bb8dfbdee2a6a26484af19661ee395552a28cbba Mon Sep 17 00:00:00 2001 From: "mikhal@webrtc.org" Date: Tue, 1 Nov 2011 22:07:16 +0000 Subject: [PATCH] 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 --- .../main/test/unit_test/unit_test.cc | 497 +++++++++--------- 1 file changed, 257 insertions(+), 240 deletions(-) diff --git a/src/modules/video_processing/main/test/unit_test/unit_test.cc b/src/modules/video_processing/main/test/unit_test/unit_test.cc index 107dacc75..1ce06541c 100644 --- a/src/modules/video_processing/main/test/unit_test/unit_test.cc +++ b/src/modules/video_processing/main/test/unit_test/unit_test.cc @@ -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(stats.mean), - static_cast(stats.numPixels), - static_cast(stats.subSamplHeight), - static_cast(stats.subSamplWidth), - static_cast(stats.sum)); + printf("\nFrameStats\n"); + printf("mean: %u\nnumPixels: %u\nsubSamplWidth: " + "%u\nsumSamplHeight: %u\nsum: %u\n\n", + static_cast(stats.mean), + static_cast(stats.numPixels), + static_cast(stats.subSamplHeight), + static_cast(stats.subSamplWidth), + static_cast(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(avgRuntime / frameNum / NumRuns)); - static_cast(avgRuntime)); - printf("Min run time = %d us / frame\n\n", - //static_cast(minRuntime / frameNum)); - static_cast(minRuntime)); + sourceFrame.Free(); + printf("\nAverage run time = %d us / frame\n", + //static_cast(avgRuntime / frameNum / NumRuns)); + static_cast(avgRuntime)); + printf("Min run time = %d us / frame\n\n", + //static_cast(minRuntime / frameNum)); + static_cast(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(); }