cc33737a80
Removed old PSNR/SSIM implementations in: * test/testsupport/metrics/video_metrics.cc * src/modules/video_coding/codecs/test_framework/test.cc The functions in video_metrics.cc is now using code in libyuv instead. Old code in test.cc is using the same functions. The code for video_metrics.h had to be moved into a separate GYP file to avoid circular dependency error on Mac (see issue 160 for more details). The reason for this is that libyuv's unittest target depends on test_support_main. BUG= TEST=metrics_unittests in Debug+Release on Linux, Mac and Windows. Review URL: http://webrtc-codereview.appspot.com/333025 git-svn-id: http://webrtc.googlecode.com/svn/trunk@1325 4adac7df-926f-26a2-2b94-8c16560cd09d
140 lines
5.4 KiB
C++
140 lines
5.4 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 "testsupport/metrics/video_metrics.h"
|
|
|
|
#include "gtest/gtest.h"
|
|
#include "testsupport/fileutils.h"
|
|
|
|
namespace webrtc {
|
|
|
|
static const char* kEmptyFileName = "video_metrics_unittest_empty_file.tmp";
|
|
static const char* kNonExistingFileName = "video_metrics_unittest_non_existing";
|
|
static const int kWidth = 352;
|
|
static const int kHeight = 288;
|
|
|
|
static const int kMissingReferenceFileReturnCode = -1;
|
|
static const int kMissingTestFileReturnCode = -2;
|
|
static const int kEmptyFileReturnCode = -3;
|
|
static const double kPsnrPerfectResult = 48.0;
|
|
static const double kSsimPerfectResult = 1.0;
|
|
|
|
class VideoMetricsTest: public testing::Test {
|
|
protected:
|
|
VideoMetricsTest() {
|
|
video_file_ = webrtc::test::ResourcePath("foreman_cif_short", "yuv");
|
|
}
|
|
virtual ~VideoMetricsTest() {}
|
|
void SetUp() {
|
|
// Create an empty file:
|
|
FILE* dummy = fopen(kEmptyFileName, "wb");
|
|
fclose(dummy);
|
|
}
|
|
void TearDown() {
|
|
std::remove(kEmptyFileName);
|
|
}
|
|
webrtc::test::QualityMetricsResult psnr_result_;
|
|
webrtc::test::QualityMetricsResult ssim_result_;
|
|
std::string video_file_;
|
|
};
|
|
|
|
// Tests that it is possible to run with the same reference as test file
|
|
TEST_F(VideoMetricsTest, ReturnsPerfectResultForIdenticalFilesPSNR) {
|
|
EXPECT_EQ(0, I420PSNRFromFiles(video_file_.c_str(), video_file_.c_str(),
|
|
kWidth, kHeight, &psnr_result_));
|
|
EXPECT_EQ(kPsnrPerfectResult, psnr_result_.average);
|
|
}
|
|
|
|
TEST_F(VideoMetricsTest, ReturnsPerfectResultForIdenticalFilesSSIM) {
|
|
EXPECT_EQ(0, I420SSIMFromFiles(video_file_.c_str(), video_file_.c_str(),
|
|
kWidth, kHeight, &ssim_result_));
|
|
EXPECT_EQ(kSsimPerfectResult, ssim_result_.average);
|
|
}
|
|
|
|
TEST_F(VideoMetricsTest, ReturnsPerfectResultForIdenticalFilesBothMetrics) {
|
|
EXPECT_EQ(0, I420MetricsFromFiles(video_file_.c_str(), video_file_.c_str(),
|
|
kWidth, kHeight, &psnr_result_,
|
|
&ssim_result_));
|
|
EXPECT_EQ(kPsnrPerfectResult, psnr_result_.average);
|
|
EXPECT_EQ(kSsimPerfectResult, ssim_result_.average);
|
|
}
|
|
|
|
// Tests that the right return code is given when the reference file is missing.
|
|
TEST_F(VideoMetricsTest, MissingReferenceFilePSNR) {
|
|
EXPECT_EQ(kMissingReferenceFileReturnCode,
|
|
I420PSNRFromFiles(kNonExistingFileName, video_file_.c_str(),
|
|
kWidth, kHeight, &ssim_result_));
|
|
}
|
|
|
|
TEST_F(VideoMetricsTest, MissingReferenceFileSSIM) {
|
|
EXPECT_EQ(kMissingReferenceFileReturnCode,
|
|
I420SSIMFromFiles(kNonExistingFileName, video_file_.c_str(),
|
|
kWidth, kHeight, &ssim_result_));
|
|
}
|
|
|
|
TEST_F(VideoMetricsTest, MissingReferenceFileBothMetrics) {
|
|
EXPECT_EQ(kMissingReferenceFileReturnCode,
|
|
I420MetricsFromFiles(kNonExistingFileName, video_file_.c_str(),
|
|
kWidth, kHeight,
|
|
&psnr_result_, &ssim_result_));
|
|
}
|
|
|
|
// Tests that the right return code is given when the test file is missing.
|
|
TEST_F(VideoMetricsTest, MissingTestFilePSNR) {
|
|
EXPECT_EQ(kMissingTestFileReturnCode,
|
|
I420PSNRFromFiles(video_file_.c_str(), kNonExistingFileName,
|
|
kWidth, kHeight, &ssim_result_));
|
|
}
|
|
|
|
TEST_F(VideoMetricsTest, MissingTestFileSSIM) {
|
|
EXPECT_EQ(kMissingTestFileReturnCode,
|
|
I420SSIMFromFiles(video_file_.c_str(), kNonExistingFileName,
|
|
kWidth, kHeight, &ssim_result_));
|
|
}
|
|
|
|
TEST_F(VideoMetricsTest, MissingTestFileBothMetrics) {
|
|
EXPECT_EQ(kMissingTestFileReturnCode,
|
|
I420MetricsFromFiles(video_file_.c_str(), kNonExistingFileName,
|
|
kWidth, kHeight,
|
|
&psnr_result_, &ssim_result_));
|
|
}
|
|
|
|
// Tests that the method can be executed with empty files.
|
|
TEST_F(VideoMetricsTest, EmptyFilesPSNR) {
|
|
EXPECT_EQ(kEmptyFileReturnCode,
|
|
I420PSNRFromFiles(kEmptyFileName, video_file_.c_str(),
|
|
kWidth, kHeight, &ssim_result_));
|
|
EXPECT_EQ(kEmptyFileReturnCode,
|
|
I420PSNRFromFiles(video_file_.c_str(), kEmptyFileName,
|
|
kWidth, kHeight, &ssim_result_));
|
|
}
|
|
|
|
TEST_F(VideoMetricsTest, EmptyFilesSSIM) {
|
|
EXPECT_EQ(kEmptyFileReturnCode,
|
|
I420SSIMFromFiles(kEmptyFileName, video_file_.c_str(),
|
|
kWidth, kHeight, &ssim_result_));
|
|
EXPECT_EQ(kEmptyFileReturnCode,
|
|
I420SSIMFromFiles(video_file_.c_str(), kEmptyFileName,
|
|
kWidth, kHeight, &ssim_result_));
|
|
}
|
|
|
|
TEST_F(VideoMetricsTest, EmptyFilesBothMetrics) {
|
|
EXPECT_EQ(kEmptyFileReturnCode,
|
|
I420MetricsFromFiles(kEmptyFileName, video_file_.c_str(),
|
|
kWidth, kHeight,
|
|
&psnr_result_, &ssim_result_));
|
|
EXPECT_EQ(kEmptyFileReturnCode,
|
|
I420MetricsFromFiles(video_file_.c_str(), kEmptyFileName,
|
|
kWidth, kHeight,
|
|
&psnr_result_, &ssim_result_));
|
|
}
|
|
|
|
} // namespace webrtc
|