Index: webrtc/common_video/libyuv/libyuv_unittest.cc |
diff --git a/webrtc/common_video/libyuv/libyuv_unittest.cc b/webrtc/common_video/libyuv/libyuv_unittest.cc |
index c60085c360483b3ff44fd5bd37d7b52d905898fa..ea90ec4e90c3cef7152f794b42fe50b02191aa93 100644 |
--- a/webrtc/common_video/libyuv/libyuv_unittest.cc |
+++ b/webrtc/common_video/libyuv/libyuv_unittest.cc |
@@ -62,7 +62,7 @@ void TestLibYuv::SetUp() { |
ASSERT_TRUE(source_file_ != NULL) << "Cannot read file: "<< |
input_file_name << "\n"; |
- rtc::scoped_refptr<VideoFrameBuffer> buffer( |
+ rtc::scoped_refptr<I420BufferInterface> buffer( |
test::ReadI420Buffer(width_, height_, source_file_)); |
orig_frame_.reset(new VideoFrame(buffer, kVideoRotation_0, 0)); |
@@ -89,8 +89,8 @@ TEST_F(TestLibYuv, ConvertTest) { |
double psnr = 0.0; |
- rtc::scoped_refptr<I420Buffer> res_i420_buffer = I420Buffer::Create( |
- width_, height_, width_, (width_ + 1) / 2, (width_ + 1) / 2); |
+ rtc::scoped_refptr<I420Buffer> res_i420_buffer = |
+ I420Buffer::Create(width_, height_); |
printf("\nConvert #%d I420 <-> I420 \n", j); |
std::unique_ptr<uint8_t[]> out_i420_buffer(new uint8_t[frame_length_]); |
@@ -103,7 +103,8 @@ TEST_F(TestLibYuv, ConvertTest) { |
if (PrintVideoFrame(*res_i420_buffer, output_file) < 0) { |
return; |
} |
- psnr = I420PSNR(*orig_frame_->video_frame_buffer(), *res_i420_buffer); |
+ psnr = |
+ I420PSNR(*orig_frame_->video_frame_buffer()->GetI420(), *res_i420_buffer); |
EXPECT_EQ(48.0, psnr); |
j++; |
@@ -125,7 +126,8 @@ TEST_F(TestLibYuv, ConvertTest) { |
if (PrintVideoFrame(*res_i420_buffer, output_file) < 0) { |
return; |
} |
- psnr = I420PSNR(*orig_frame_->video_frame_buffer(), *res_i420_buffer); |
+ psnr = |
+ I420PSNR(*orig_frame_->video_frame_buffer()->GetI420(), *res_i420_buffer); |
// Optimization Speed- quality trade-off => 45 dB only (platform dependant). |
EXPECT_GT(ceil(psnr), 44); |
@@ -138,7 +140,8 @@ TEST_F(TestLibYuv, ConvertTest) { |
EXPECT_EQ(0, |
ConvertToI420(VideoType::kUYVY, out_uyvy_buffer.get(), 0, 0, width_, |
height_, 0, kVideoRotation_0, res_i420_buffer.get())); |
- psnr = I420PSNR(*orig_frame_->video_frame_buffer(), *res_i420_buffer); |
+ psnr = |
+ I420PSNR(*orig_frame_->video_frame_buffer()->GetI420(), *res_i420_buffer); |
EXPECT_EQ(48.0, psnr); |
if (PrintVideoFrame(*res_i420_buffer, output_file) < 0) { |
return; |
@@ -158,7 +161,8 @@ TEST_F(TestLibYuv, ConvertTest) { |
return; |
} |
- psnr = I420PSNR(*orig_frame_->video_frame_buffer(), *res_i420_buffer); |
+ psnr = |
+ I420PSNR(*orig_frame_->video_frame_buffer()->GetI420(), *res_i420_buffer); |
EXPECT_EQ(48.0, psnr); |
printf("\nConvert #%d I420 <-> RGB565\n", j); |
@@ -175,7 +179,8 @@ TEST_F(TestLibYuv, ConvertTest) { |
} |
j++; |
- psnr = I420PSNR(*orig_frame_->video_frame_buffer(), *res_i420_buffer); |
+ psnr = |
+ I420PSNR(*orig_frame_->video_frame_buffer()->GetI420(), *res_i420_buffer); |
// TODO(leozwang) Investigate the right psnr should be set for I420ToRGB565, |
// Another example is I420ToRGB24, the psnr is 44 |
// TODO(mikhal): Add psnr for RGB565, 1555, 4444, convert to ARGB. |
@@ -195,7 +200,8 @@ TEST_F(TestLibYuv, ConvertTest) { |
return; |
} |
- psnr = I420PSNR(*orig_frame_->video_frame_buffer(), *res_i420_buffer); |
+ psnr = |
+ I420PSNR(*orig_frame_->video_frame_buffer()->GetI420(), *res_i420_buffer); |
// TODO(leozwang) Investigate the right psnr should be set for |
// I420ToARGB8888, |
EXPECT_GT(ceil(psnr), 42); |
@@ -228,7 +234,8 @@ TEST_F(TestLibYuv, ConvertAlignedFrame) { |
if (PrintVideoFrame(*res_i420_buffer, output_file) < 0) { |
return; |
} |
- psnr = I420PSNR(*orig_frame_->video_frame_buffer(), *res_i420_buffer); |
+ psnr = |
+ I420PSNR(*orig_frame_->video_frame_buffer()->GetI420(), *res_i420_buffer); |
EXPECT_EQ(48.0, psnr); |
} |
@@ -241,7 +248,8 @@ TEST_F(TestLibYuv, RotateTest) { |
int stride_uv; |
// Assume compact layout, no padding. |
- const uint8_t *orig_buffer = orig_frame_->video_frame_buffer()->DataY(); |
+ const uint8_t* orig_buffer = |
+ orig_frame_->video_frame_buffer()->GetI420()->DataY(); |
Calc16ByteAlignedStride(rotated_width, &stride_y, &stride_uv); |
rtc::scoped_refptr<I420Buffer> rotated_res_i420_buffer = I420Buffer::Create( |
@@ -252,8 +260,7 @@ TEST_F(TestLibYuv, RotateTest) { |
EXPECT_EQ( |
0, ConvertToI420(VideoType::kI420, orig_buffer, 0, 0, width_, height_, 0, |
kVideoRotation_270, rotated_res_i420_buffer.get())); |
- rotated_res_i420_buffer = I420Buffer::Create( |
- width_, height_, width_, (width_ + 1) / 2, (width_ + 1) / 2); |
+ rotated_res_i420_buffer = I420Buffer::Create(width_, height_); |
EXPECT_EQ( |
0, ConvertToI420(VideoType::kI420, orig_buffer, 0, 0, width_, height_, 0, |
kVideoRotation_180, rotated_res_i420_buffer.get())); |