119 lines
3.8 KiB
C++
119 lines
3.8 KiB
C++
|
/*
|
||
|
|
* Copyright (c) 2011 The WebRTC project authors. All Rights Reserved.
|
||
|
|
*
|
||
|
|
* Use of this source code is governed by a BSD-style license
|
||
|
|
* that can be found in the LICENSE file in the root of the source
|
||
|
|
* tree. An additional intellectual property rights grant can be found
|
||
|
|
* in the file PATENTS. All contributing project authors may
|
||
|
|
* be found in the AUTHORS file in the root of the source tree.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "unit_test.h"
|
||
|
|
#include "video_processing.h"
|
||
|
|
|
||
|
|
#include "tick_util.h"
|
||
|
|
|
||
|
|
#include <cstdio>
|
||
|
|
#include <cstdlib>
|
||
|
|
|
||
|
|
using namespace webrtc;
|
||
|
|
|
||
|
|
TEST_F(VideoProcessingModuleTest, Denoising)
|
||
|
|
{
|
||
|
|
enum { NumRuns = 10 };
|
||
|
|
WebRtc_UWord32 frameNum = 0;
|
||
|
|
|
||
|
|
WebRtc_Word64 minRuntime = 0;
|
||
|
|
WebRtc_Word64 avgRuntime = 0;
|
||
|
|
|
||
|
|
FILE* denoiseFile = fopen("denoise_testfile.yuv", "wb");
|
||
|
|
ASSERT_TRUE(denoiseFile != NULL) << "Could not open output file.\n";
|
||
|
|
|
||
|
|
FILE* noiseFile = fopen("noise_testfile.yuv", "wb");
|
||
|
|
ASSERT_TRUE(noiseFile != NULL) << "Could not open noisy file.\n";
|
||
|
|
|
||
|
|
printf("\nRun time [us / frame]:\n");
|
||
|
|
for (WebRtc_UWord32 runIdx = 0; runIdx < NumRuns; runIdx++)
|
||
|
|
{
|
||
|
|
TickTime t0;
|
||
|
|
TickTime t1;
|
||
|
|
TickInterval accTicks;
|
||
|
|
WebRtc_Word32 modifiedPixels = 0;
|
||
|
|
|
||
|
|
frameNum = 0;
|
||
|
|
while (fread(_videoFrame.Buffer(), 1, _frameLength, _sourceFile) == _frameLength)
|
||
|
|
{
|
||
|
|
frameNum++;
|
||
|
|
WebRtc_UWord8* sourceBuffer = _videoFrame.Buffer();
|
||
|
|
|
||
|
|
// Add noise to a part in video stream
|
||
|
|
// Random noise
|
||
|
|
// TODO: investigate the effectiveness of this test.
|
||
|
|
|
||
|
|
//for(WebRtc_UWord32 ir = 0; ir < _frameLength; ir++)
|
||
|
|
// sourceBuffer[ir] = 128
|
||
|
|
for (WebRtc_UWord32 ir = 0; ir < _height; ir++)
|
||
|
|
{
|
||
|
|
WebRtc_UWord32 ik = ir * _width;
|
||
|
|
for (WebRtc_UWord32 ic = 0; ic < _width; ic++)
|
||
|
|
{
|
||
|
|
WebRtc_UWord8 r = rand() % 16;
|
||
|
|
r -= 8;
|
||
|
|
if (ir < _height / 4)
|
||
|
|
r = 0;
|
||
|
|
if (ir >= 3 * _height / 4)
|
||
|
|
r = 0;
|
||
|
|
if (ic < _width / 4)
|
||
|
|
r = 0;
|
||
|
|
if (ic >= 3 * _width / 4)
|
||
|
|
r = 0;
|
||
|
|
|
||
|
|
/*WebRtc_UWord8 pixelValue = 0;
|
||
|
|
if (ir >= _height / 2)
|
||
|
|
{ // Region 3 or 4
|
||
|
|
pixelValue = 170;
|
||
|
|
}
|
||
|
|
if (ic >= _width / 2)
|
||
|
|
{ // Region 2 or 4
|
||
|
|
pixelValue += 85;
|
||
|
|
}
|
||
|
|
pixelValue += r;
|
||
|
|
sourceBuffer[ik + ic] = pixelValue;
|
||
|
|
*/
|
||
|
|
sourceBuffer[ik + ic] += r;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (runIdx == 0)
|
||
|
|
{
|
||
|
|
fwrite(_videoFrame.Buffer(), 1, _frameLength, noiseFile);
|
||
|
|
}
|
||
|
|
|
||
|
|
t0 = TickTime::Now();
|
||
|
|
ASSERT_GE(modifiedPixels = _vpm->Denoising(_videoFrame), 0);
|
||
|
|
t1 = TickTime::Now();
|
||
|
|
accTicks += t1 - t0;
|
||
|
|
|
||
|
|
if (runIdx == 0)
|
||
|
|
{
|
||
|
|
fwrite(_videoFrame.Buffer(), 1, _frameLength, denoiseFile);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
ASSERT_NE(0, feof(_sourceFile)) << "Error reading source file";
|
||
|
|
|
||
|
|
printf("%u\n", static_cast<int>(accTicks.Microseconds() / frameNum));
|
||
|
|
if (accTicks.Microseconds() < minRuntime || runIdx == 0)
|
||
|
|
{
|
||
|
|
minRuntime = accTicks.Microseconds();
|
||
|
|
}
|
||
|
|
avgRuntime += accTicks.Microseconds();
|
||
|
|
|
||
|
|
rewind(_sourceFile);
|
||
|
|
}
|
||
|
|
|
||
|
|
printf("\nAverage run time = %d us / frame\n",
|
||
|
|
static_cast<int>(avgRuntime / frameNum / NumRuns));
|
||
|
|
printf("Min run time = %d us / frame\n\n",
|
||
|
|
static_cast<int>(minRuntime / frameNum));
|
||
|
|
}
|