libyuv: Adding psnr/ssim to libyuv and updating unit tests according to latest conventions.
Review URL: http://webrtc-codereview.appspot.com/331007 git-svn-id: http://webrtc.googlecode.com/svn/trunk@1253 4adac7df-926f-26a2-2b94-8c16560cd09d
This commit is contained in:
mikhal@webrtc.org
246
src/common_video/libyuv/libyuv_unittest.cc
Normal file
246
src/common_video/libyuv/libyuv_unittest.cc
Normal file
@@ -0,0 +1,246 @@
|
||||
/*
|
||||
* 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 <math.h>
|
||||
#include <string.h>
|
||||
|
||||
#include "common_video/libyuv/include/libyuv.h"
|
||||
#include "gtest/gtest.h"
|
||||
#include "system_wrappers/interface/tick_util.h"
|
||||
#include "testsupport/fileutils.h"
|
||||
|
||||
namespace webrtc {
|
||||
|
||||
int PrintFrame(const uint8_t* frame, int width, int height) {
|
||||
if (frame == NULL)
|
||||
return -1;
|
||||
int k = 0;
|
||||
for (int i = 0; i < height; i++) {
|
||||
for (int j = 0; j < width; j++) {
|
||||
printf("%d ", frame[k++]);
|
||||
}
|
||||
printf(" \n");
|
||||
}
|
||||
printf(" \n");
|
||||
return 0;
|
||||
}
|
||||
|
||||
int PrintFrame(const uint8_t* frame, int width,
|
||||
int height, const char* str) {
|
||||
if (frame == NULL)
|
||||
return -1;
|
||||
printf("%s %dx%d \n", str, width, height);
|
||||
|
||||
const uint8_t* frame_y = frame;
|
||||
const uint8_t* frame_u = frame_y + width * height;
|
||||
const uint8_t* frame_v = frame_u + width * height / 4;
|
||||
|
||||
int ret = 0;
|
||||
ret += PrintFrame(frame_y, width, height);
|
||||
ret += PrintFrame(frame_u, width / 2, height / 2);
|
||||
ret += PrintFrame(frame_v, width / 2, height / 2);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
void CreateImage(int width, int height,
|
||||
uint8_t* frame, int offset,
|
||||
int height_factor, int width_factor) {
|
||||
if (frame == NULL)
|
||||
return;
|
||||
for (int i = 0; i < height; i++) {
|
||||
for (int j = 0; j < width; j++) {
|
||||
*frame = static_cast<uint8_t>((i + offset) * height_factor
|
||||
+ j * width_factor);
|
||||
frame++;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
class TestLibYuv : public ::testing::Test {
|
||||
protected:
|
||||
TestLibYuv();
|
||||
virtual void SetUp();
|
||||
virtual void TearDown();
|
||||
|
||||
FILE* source_file_;
|
||||
const int width_;
|
||||
const int height_;
|
||||
const int frame_length_;
|
||||
};
|
||||
|
||||
// TODO (mikhal): Use scoped_ptr when handling buffers.
|
||||
TestLibYuv::TestLibYuv()
|
||||
: source_file_(NULL),
|
||||
width_(352),
|
||||
height_(288),
|
||||
frame_length_(CalcBufferSize(kI420, 352, 288)) {
|
||||
}
|
||||
|
||||
void TestLibYuv::SetUp() {
|
||||
const std::string input_file_name = webrtc::test::ProjectRootPath() +
|
||||
"resources/foreman_cif.yuv";
|
||||
source_file_ = fopen(input_file_name.c_str(), "rb");
|
||||
ASSERT_TRUE(source_file_ != NULL) << "Cannot read file: "<<
|
||||
input_file_name << "\n";
|
||||
}
|
||||
|
||||
void TestLibYuv::TearDown() {
|
||||
if (source_file_ != NULL) {
|
||||
ASSERT_EQ(0, fclose(source_file_));
|
||||
}
|
||||
source_file_ = NULL;
|
||||
}
|
||||
|
||||
TEST_F(TestLibYuv, ConvertSanityTest) {
|
||||
// TODO(mikhal)
|
||||
}
|
||||
|
||||
TEST_F(TestLibYuv, ConvertTest) {
|
||||
// Reading YUV frame - testing on the first frame of the foreman sequence
|
||||
int j = 0;
|
||||
std::string output_file_name = webrtc::test::OutputPath() +
|
||||
"LibYuvTest_conversion.yuv";
|
||||
FILE* output_file = fopen(output_file_name.c_str(), "wb");
|
||||
ASSERT_TRUE(output_file != NULL);
|
||||
|
||||
double psnr = 0;
|
||||
|
||||
uint8_t* orig_buffer = new uint8_t[frame_length_];
|
||||
EXPECT_GT(fread(orig_buffer, 1, frame_length_, source_file_), 0U);
|
||||
|
||||
// printf("\nConvert #%d I420 <-> RGB24\n", j);
|
||||
uint8_t* res_rgb_buffer2 = new uint8_t[width_ * height_ * 3];
|
||||
uint8_t* res_i420_buffer = new uint8_t[frame_length_];
|
||||
EXPECT_EQ(0, ConvertFromI420(kRGB24, orig_buffer, width_, height_,
|
||||
res_rgb_buffer2, false, kRotateNone));
|
||||
EXPECT_EQ(0, ConvertToI420(kRGB24, res_rgb_buffer2, width_, height_,
|
||||
res_i420_buffer, false, kRotateNone));
|
||||
|
||||
fwrite(res_i420_buffer, frame_length_, 1, output_file);
|
||||
//ImagePSNRfromBuffer(orig_buffer, res_i420_buffer, width_, height_, &psnr);
|
||||
psnr = I420PSNR(orig_buffer, res_i420_buffer, width_, height_);
|
||||
// Optimization Speed- quality trade-off => 45 dB only.
|
||||
EXPECT_GT(ceil(psnr), 45);
|
||||
j++;
|
||||
delete [] res_rgb_buffer2;
|
||||
|
||||
// printf("\nConvert #%d I420 <-> UYVY\n", j);
|
||||
uint8_t* out_uyvy_buffer = new uint8_t[width_ * height_ * 2];
|
||||
EXPECT_EQ(0, ConvertFromI420(kUYVY, orig_buffer, width_,
|
||||
height_, out_uyvy_buffer, false, kRotateNone));
|
||||
|
||||
EXPECT_EQ(0, ConvertToI420(kUYVY, out_uyvy_buffer, width_, height_,
|
||||
res_i420_buffer, false, kRotateNone));
|
||||
//ImagePSNRfromBuffer(orig_buffer, res_i420_buffer, width_, height_, &psnr);
|
||||
psnr = I420PSNR(orig_buffer, res_i420_buffer, width_, height_);
|
||||
EXPECT_EQ(48.0, psnr);
|
||||
fwrite(res_i420_buffer, frame_length_, 1, output_file);
|
||||
|
||||
j++;
|
||||
delete [] out_uyvy_buffer;
|
||||
|
||||
// printf("\nConvert #%d I420 <-> I420 \n", j);
|
||||
uint8_t* out_i420_buffer = new uint8_t[width_ * height_ * 3 / 2 ];
|
||||
EXPECT_EQ(0, ConvertToI420(kI420, orig_buffer, width_, height_,
|
||||
out_i420_buffer, false, kRotateNone));
|
||||
EXPECT_EQ(0, ConvertFromI420(kI420 , out_i420_buffer, width_, height_,
|
||||
res_i420_buffer, false, kRotateNone));
|
||||
fwrite(res_i420_buffer, frame_length_, 1, output_file);
|
||||
//ImagePSNRfromBuffer(orig_buffer, res_i420_buffer, width_, height_, &psnr);
|
||||
psnr = I420PSNR(orig_buffer, res_i420_buffer, width_, height_);
|
||||
EXPECT_EQ(48.0, psnr);
|
||||
j++;
|
||||
delete [] out_i420_buffer;
|
||||
|
||||
// printf("\nConvert #%d I420 <-> YV12\n", j);
|
||||
uint8_t* outYV120Buffer = new uint8_t[frame_length_];
|
||||
|
||||
EXPECT_EQ(0, ConvertFromI420(kYV12, orig_buffer, width_, height_,
|
||||
outYV120Buffer, false, kRotateNone));
|
||||
EXPECT_EQ(0, ConvertYV12ToI420(outYV120Buffer, width_, height_,
|
||||
res_i420_buffer));
|
||||
fwrite(res_i420_buffer, frame_length_, 1, output_file);
|
||||
|
||||
//ImagePSNRfromBuffer(orig_buffer, res_i420_buffer, width_, height_, &psnr);
|
||||
psnr = I420PSNR(orig_buffer, res_i420_buffer, width_, height_);
|
||||
EXPECT_EQ(48.0, psnr);
|
||||
j++;
|
||||
delete [] outYV120Buffer;
|
||||
|
||||
// printf("\nConvert #%d I420 <-> YUY2\n", j);
|
||||
uint8_t* out_yuy2_buffer = new uint8_t[width_ * height_ * 2];
|
||||
|
||||
EXPECT_EQ(0, ConvertFromI420(kYUY2, orig_buffer, width_, height_,
|
||||
out_yuy2_buffer, false, kRotateNone));
|
||||
EXPECT_EQ(0, ConvertToI420(kYUY2, out_yuy2_buffer, width_, height_,
|
||||
res_i420_buffer, false, kRotateNone));
|
||||
|
||||
fwrite(res_i420_buffer, frame_length_, 1, output_file);
|
||||
//ImagePSNRfromBuffer(orig_buffer, res_i420_buffer, width_, height_, &psnr);
|
||||
psnr = I420PSNR(orig_buffer, res_i420_buffer, width_, height_);
|
||||
EXPECT_EQ(48.0, psnr);
|
||||
delete [] out_yuy2_buffer;
|
||||
|
||||
delete [] res_i420_buffer;
|
||||
delete [] orig_buffer;
|
||||
}
|
||||
|
||||
TEST_F(TestLibYuv, MirrorTest) {
|
||||
// TODO (mikhal): Add an automated test to confirm output.
|
||||
std::string str;
|
||||
int width = 16;
|
||||
int height = 8;
|
||||
int factor_y = 1;
|
||||
int factor_u = 1;
|
||||
int factor_v = 1;
|
||||
int start_buffer_offset = 10;
|
||||
int length = webrtc::CalcBufferSize(kI420, width, height);
|
||||
|
||||
uint8_t* test_frame = new uint8_t[length];
|
||||
memset(test_frame, 255, length);
|
||||
|
||||
// Create input frame
|
||||
uint8_t* in_frame = test_frame;
|
||||
uint8_t* in_frame_cb = in_frame + width * height;
|
||||
uint8_t* in_frame_cr = in_frame_cb + (width * height) / 4;
|
||||
CreateImage(width, height, in_frame, 10, factor_y, 1); // Y
|
||||
CreateImage(width / 2, height / 2, in_frame_cb, 100, factor_u, 1); // Cb
|
||||
CreateImage(width / 2, height / 2, in_frame_cr, 200, factor_v, 1); // Cr
|
||||
EXPECT_EQ(0, PrintFrame(test_frame, width, height, "InputFrame"));
|
||||
|
||||
uint8_t* test_frame2 = new uint8_t[length + start_buffer_offset * 2];
|
||||
memset(test_frame2, 255, length + start_buffer_offset * 2);
|
||||
uint8_t* out_frame = test_frame2;
|
||||
|
||||
// LeftRight
|
||||
std::cout << "Test Mirror function: LeftRight" << std::endl;
|
||||
EXPECT_EQ(0, MirrorI420LeftRight(in_frame, out_frame, width, height));
|
||||
EXPECT_EQ(0, PrintFrame(test_frame2, width, height, "OutputFrame"));
|
||||
EXPECT_EQ(0, MirrorI420LeftRight(out_frame, test_frame, width, height));
|
||||
|
||||
EXPECT_EQ(0, memcmp(in_frame, test_frame, length));
|
||||
|
||||
// UpDown
|
||||
std::cout << "Test Mirror function: UpDown" << std::endl;
|
||||
EXPECT_EQ(0, MirrorI420UpDown(in_frame, out_frame, width, height));
|
||||
EXPECT_EQ(0, PrintFrame(test_frame2, width, height, "OutputFrame"));
|
||||
EXPECT_EQ(0, MirrorI420UpDown(out_frame, test_frame, width, height));
|
||||
|
||||
EXPECT_EQ(0, memcmp(in_frame, test_frame, length));
|
||||
|
||||
// TODO(mikhal): Write to a file, and ask to look at the file.
|
||||
|
||||
std::cout << "Do the mirrored frames look correct?" << std::endl;
|
||||
delete [] test_frame;
|
||||
delete [] test_frame2;
|
||||
}
|
||||
|
||||
} // namespace
|
||||
Reference in New Issue
Block a user