common_video/libyuv/libyuv_unittest.cc - src.git - Git at Google

 /*
  *  Copyright (c) 2012 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 "third_party/libyuv/include/libyuv.h"

 #include <math.h>
 #include <string.h>

 #include <memory>

 #include "api/video/i420_buffer.h"
 #include "api/video/video_frame.h"
 #include "common_video/libyuv/include/webrtc_libyuv.h"
 #include "test/frame_utils.h"
 #include "test/gmock.h"
 #include "test/gtest.h"
 #include "test/testsupport/file_utils.h"

 namespace webrtc {

 namespace {
 void Calc16ByteAlignedStride(int width, int* stride_y, int* stride_uv) {
   *stride_y = 16 * ((width + 15) / 16);
   *stride_uv = 16 * ((width + 31) / 32);
 }

 int PrintPlane(const uint8_t* buf,
                int width,
                int height,
                int stride,
                FILE* file) {
   for (int i = 0; i < height; i++, buf += stride) {
     if (fwrite(buf, 1, width, file) != static_cast<unsigned int>(width))
       return -1;
   }
   return 0;
 }

 int PrintVideoFrame(const I420BufferInterface& frame, FILE* file) {
   int width = frame.width();
   int height = frame.height();
   int chroma_width = frame.ChromaWidth();
   int chroma_height = frame.ChromaHeight();

   if (PrintPlane(frame.DataY(), width, height, frame.StrideY(), file) < 0) {
     return -1;
   }
   if (PrintPlane(frame.DataU(), chroma_width, chroma_height, frame.StrideU(),
                  file) < 0) {
     return -1;
   }
   if (PrintPlane(frame.DataV(), chroma_width, chroma_height, frame.StrideV(),
                  file) < 0) {
     return -1;
   }
   return 0;
 }

 }  // Anonymous namespace

 class TestLibYuv : public ::testing::Test {
  protected:
   TestLibYuv();
   void SetUp() override;
   void TearDown() override;

   FILE* source_file_;
   std::unique_ptr<VideoFrame> orig_frame_;
   const int width_;
   const int height_;
   const int size_y_;
   const int size_uv_;
   const size_t frame_length_;
 };

 TestLibYuv::TestLibYuv()
     : source_file_(NULL),
       orig_frame_(),
       width_(352),
       height_(288),
       size_y_(width_ * height_),
       size_uv_(((width_ + 1) / 2) * ((height_ + 1) / 2)),
       frame_length_(CalcBufferSize(VideoType::kI420, 352, 288)) {}

 void TestLibYuv::SetUp() {
   const std::string input_file_name =
       webrtc::test::ResourcePath("foreman_cif", "yuv");
   source_file_ = fopen(input_file_name.c_str(), "rb");
   ASSERT_TRUE(source_file_ != NULL)
       << "Cannot read file: " << input_file_name << "\n";

   rtc::scoped_refptr<I420BufferInterface> buffer(
       test::ReadI420Buffer(width_, height_, source_file_));

   orig_frame_ =
       std::make_unique<VideoFrame>(VideoFrame::Builder()
                                        .set_video_frame_buffer(buffer)
                                        .set_rotation(webrtc::kVideoRotation_0)
                                        .set_timestamp_us(0)
                                        .build());
 }

 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.0;

   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_]);
   EXPECT_EQ(0, ConvertFromI420(*orig_frame_, VideoType::kI420, 0,
                                out_i420_buffer.get()));
   int y_size = width_ * height_;
   int u_size = res_i420_buffer->ChromaWidth() * res_i420_buffer->ChromaHeight();
   int ret = libyuv::I420Copy(
       out_i420_buffer.get(), width_, out_i420_buffer.get() + y_size,
       width_ >> 1, out_i420_buffer.get() + y_size + u_size, width_ >> 1,
       res_i420_buffer.get()->MutableDataY(), res_i420_buffer.get()->StrideY(),
       res_i420_buffer.get()->MutableDataU(), res_i420_buffer.get()->StrideU(),
       res_i420_buffer.get()->MutableDataV(), res_i420_buffer.get()->StrideV(),
       width_, height_);
   EXPECT_EQ(0, ret);

   if (PrintVideoFrame(*res_i420_buffer, output_file) < 0) {
     return;
   }
   psnr =
       I420PSNR(*orig_frame_->video_frame_buffer()->GetI420(), *res_i420_buffer);
   EXPECT_EQ(48.0, psnr);
   j++;

   printf("\nConvert #%d I420 <-> RGB24\n", j);
   std::unique_ptr<uint8_t[]> res_rgb_buffer2(new uint8_t[width_ * height_ * 3]);
   // Align the stride values for the output frame.
   int stride_y = 0;
   int stride_uv = 0;
   Calc16ByteAlignedStride(width_, &stride_y, &stride_uv);
   res_i420_buffer =
       I420Buffer::Create(width_, height_, stride_y, stride_uv, stride_uv);
   EXPECT_EQ(0, ConvertFromI420(*orig_frame_, VideoType::kRGB24, 0,
                                res_rgb_buffer2.get()));

   ret = libyuv::ConvertToI420(
       res_rgb_buffer2.get(), 0, res_i420_buffer.get()->MutableDataY(),
       res_i420_buffer.get()->StrideY(), res_i420_buffer.get()->MutableDataU(),
       res_i420_buffer.get()->StrideU(), res_i420_buffer.get()->MutableDataV(),
       res_i420_buffer.get()->StrideV(), 0, 0, width_, height_,
       res_i420_buffer->width(), res_i420_buffer->height(), libyuv::kRotate0,
       ConvertVideoType(VideoType::kRGB24));

   EXPECT_EQ(0, ret);
   if (PrintVideoFrame(*res_i420_buffer, output_file) < 0) {
     return;
   }
   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);
   j++;

   printf("\nConvert #%d I420 <-> UYVY\n", j);
   std::unique_ptr<uint8_t[]> out_uyvy_buffer(new uint8_t[width_ * height_ * 2]);
   EXPECT_EQ(0, ConvertFromI420(*orig_frame_, VideoType::kUYVY, 0,
                                out_uyvy_buffer.get()));

   ret = libyuv::ConvertToI420(
       out_uyvy_buffer.get(), 0, res_i420_buffer.get()->MutableDataY(),
       res_i420_buffer.get()->StrideY(), res_i420_buffer.get()->MutableDataU(),
       res_i420_buffer.get()->StrideU(), res_i420_buffer.get()->MutableDataV(),
       res_i420_buffer.get()->StrideV(), 0, 0, width_, height_,
       res_i420_buffer->width(), res_i420_buffer->height(), libyuv::kRotate0,
       ConvertVideoType(VideoType::kUYVY));

   EXPECT_EQ(0, ret);
   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;
   }
   j++;

   printf("\nConvert #%d I420 <-> YUY2\n", j);
   std::unique_ptr<uint8_t[]> out_yuy2_buffer(new uint8_t[width_ * height_ * 2]);
   EXPECT_EQ(0, ConvertFromI420(*orig_frame_, VideoType::kYUY2, 0,
                                out_yuy2_buffer.get()));

   ret = libyuv::ConvertToI420(
       out_yuy2_buffer.get(), 0, res_i420_buffer.get()->MutableDataY(),
       res_i420_buffer.get()->StrideY(), res_i420_buffer.get()->MutableDataU(),
       res_i420_buffer.get()->StrideU(), res_i420_buffer.get()->MutableDataV(),
       res_i420_buffer.get()->StrideV(), 0, 0, width_, height_,
       res_i420_buffer->width(), res_i420_buffer->height(), libyuv::kRotate0,
       ConvertVideoType(VideoType::kYUY2));

   EXPECT_EQ(0, ret);

   if (PrintVideoFrame(*res_i420_buffer, output_file) < 0) {
     return;
   }

   psnr =
       I420PSNR(*orig_frame_->video_frame_buffer()->GetI420(), *res_i420_buffer);
   EXPECT_EQ(48.0, psnr);

   printf("\nConvert #%d I420 <-> RGB565\n", j);
   std::unique_ptr<uint8_t[]> out_rgb565_buffer(
       new uint8_t[width_ * height_ * 2]);
   EXPECT_EQ(0, ConvertFromI420(*orig_frame_, VideoType::kRGB565, 0,
                                out_rgb565_buffer.get()));

   ret = libyuv::ConvertToI420(
       out_rgb565_buffer.get(), 0, res_i420_buffer.get()->MutableDataY(),
       res_i420_buffer.get()->StrideY(), res_i420_buffer.get()->MutableDataU(),
       res_i420_buffer.get()->StrideU(), res_i420_buffer.get()->MutableDataV(),
       res_i420_buffer.get()->StrideV(), 0, 0, width_, height_,
       res_i420_buffer->width(), res_i420_buffer->height(), libyuv::kRotate0,
       ConvertVideoType(VideoType::kRGB565));

   EXPECT_EQ(0, ret);
   if (PrintVideoFrame(*res_i420_buffer, output_file) < 0) {
     return;
   }
   j++;

   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.
   EXPECT_GT(ceil(psnr), 40);

   printf("\nConvert #%d I420 <-> ARGB8888\n", j);
   std::unique_ptr<uint8_t[]> out_argb8888_buffer(
       new uint8_t[width_ * height_ * 4]);
   EXPECT_EQ(0, ConvertFromI420(*orig_frame_, VideoType::kARGB, 0,
                                out_argb8888_buffer.get()));

   ret = libyuv::ConvertToI420(
       out_argb8888_buffer.get(), 0, res_i420_buffer.get()->MutableDataY(),
       res_i420_buffer.get()->StrideY(), res_i420_buffer.get()->MutableDataU(),
       res_i420_buffer.get()->StrideU(), res_i420_buffer.get()->MutableDataV(),
       res_i420_buffer.get()->StrideV(), 0, 0, width_, height_,
       res_i420_buffer->width(), res_i420_buffer->height(), libyuv::kRotate0,
       ConvertVideoType(VideoType::kARGB));

   EXPECT_EQ(0, ret);

   if (PrintVideoFrame(*res_i420_buffer, output_file) < 0) {
     return;
   }

   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);

   ASSERT_EQ(0, fclose(output_file));
 }

 TEST_F(TestLibYuv, ConvertAlignedFrame) {
   // Reading YUV frame - testing on the first frame of the foreman sequence
   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.0;

   int stride_y = 0;
   int stride_uv = 0;
   Calc16ByteAlignedStride(width_, &stride_y, &stride_uv);

   rtc::scoped_refptr<I420Buffer> res_i420_buffer =
       I420Buffer::Create(width_, height_, stride_y, stride_uv, stride_uv);
   std::unique_ptr<uint8_t[]> out_i420_buffer(new uint8_t[frame_length_]);
   EXPECT_EQ(0, ConvertFromI420(*orig_frame_, VideoType::kI420, 0,
                                out_i420_buffer.get()));
   int y_size = width_ * height_;
   int u_size = res_i420_buffer->ChromaWidth() * res_i420_buffer->ChromaHeight();
   int ret = libyuv::I420Copy(
       out_i420_buffer.get(), width_, out_i420_buffer.get() + y_size,
       width_ >> 1, out_i420_buffer.get() + y_size + u_size, width_ >> 1,
       res_i420_buffer.get()->MutableDataY(), res_i420_buffer.get()->StrideY(),
       res_i420_buffer.get()->MutableDataU(), res_i420_buffer.get()->StrideU(),
       res_i420_buffer.get()->MutableDataV(), res_i420_buffer.get()->StrideV(),
       width_, height_);

   EXPECT_EQ(0, ret);

   if (PrintVideoFrame(*res_i420_buffer, output_file) < 0) {
     return;
   }
   psnr =
       I420PSNR(*orig_frame_->video_frame_buffer()->GetI420(), *res_i420_buffer);
   EXPECT_EQ(48.0, psnr);
 }

 static uint8_t Average(int a, int b, int c, int d) {
   return (a + b + c + d + 2) / 4;
 }

 TEST_F(TestLibYuv, NV12Scale2x2to2x2) {
   const std::vector<uint8_t> src_y = {0, 1, 2, 3};
   const std::vector<uint8_t> src_uv = {0, 1};
   std::vector<uint8_t> dst_y(4);
   std::vector<uint8_t> dst_uv(2);

   uint8_t* tmp_buffer = nullptr;

   NV12Scale(tmp_buffer, src_y.data(), 2, src_uv.data(), 2, 2, 2, dst_y.data(),
             2, dst_uv.data(), 2, 2, 2);

   EXPECT_THAT(dst_y, ::testing::ContainerEq(src_y));
   EXPECT_THAT(dst_uv, ::testing::ContainerEq(src_uv));
 }

 TEST_F(TestLibYuv, NV12Scale4x4to2x2) {
   const uint8_t src_y[] = {0, 1, 2,  3,  4,  5,  6,  7,
                            8, 9, 10, 11, 12, 13, 14, 15};
   const uint8_t src_uv[] = {0, 1, 2, 3, 4, 5, 6, 7};
   std::vector<uint8_t> dst_y(4);
   std::vector<uint8_t> dst_uv(2);

   std::vector<uint8_t> tmp_buffer;
   const int src_chroma_width = (4 + 1) / 2;
   const int src_chroma_height = (4 + 1) / 2;
   const int dst_chroma_width = (2 + 1) / 2;
   const int dst_chroma_height = (2 + 1) / 2;
   tmp_buffer.resize(src_chroma_width * src_chroma_height * 2 +
                     dst_chroma_width * dst_chroma_height * 2);
   tmp_buffer.shrink_to_fit();

   NV12Scale(tmp_buffer.data(), src_y, 4, src_uv, 4, 4, 4, dst_y.data(), 2,
             dst_uv.data(), 2, 2, 2);

   EXPECT_THAT(dst_y, ::testing::ElementsAre(
                          Average(0, 1, 4, 5), Average(2, 3, 6, 7),
                          Average(8, 9, 12, 13), Average(10, 11, 14, 15)));
   EXPECT_THAT(dst_uv,
               ::testing::ElementsAre(Average(0, 2, 4, 6), Average(1, 3, 5, 7)));
 }

 }  // namespace webrtc
	/*
	* Copyright (c) 2012 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 "third_party/libyuv/include/libyuv.h"

	#include <math.h>
	#include <string.h>

	#include <memory>

	#include "api/video/i420_buffer.h"
	#include "api/video/video_frame.h"
	#include "common_video/libyuv/include/webrtc_libyuv.h"
	#include "test/frame_utils.h"
	#include "test/gmock.h"
	#include "test/gtest.h"
	#include "test/testsupport/file_utils.h"

	namespace webrtc {

	namespace {
	void Calc16ByteAlignedStride(int width, int* stride_y, int* stride_uv) {
	stride_y = 16 ((width + 15) / 16);
	stride_uv = 16 ((width + 31) / 32);
	}

	int PrintPlane(const uint8_t* buf,
	int width,
	int height,
	int stride,
	FILE* file) {
	for (int i = 0; i < height; i++, buf += stride) {
	if (fwrite(buf, 1, width, file) != static_cast<unsigned int>(width))
	return -1;
	}
	return 0;
	}

	int PrintVideoFrame(const I420BufferInterface& frame, FILE* file) {
	int width = frame.width();
	int height = frame.height();
	int chroma_width = frame.ChromaWidth();
	int chroma_height = frame.ChromaHeight();

	if (PrintPlane(frame.DataY(), width, height, frame.StrideY(), file) < 0) {
	return -1;
	}
	if (PrintPlane(frame.DataU(), chroma_width, chroma_height, frame.StrideU(),
	file) < 0) {
	return -1;
	}
	if (PrintPlane(frame.DataV(), chroma_width, chroma_height, frame.StrideV(),
	file) < 0) {
	return -1;
	}
	return 0;
	}

	} // Anonymous namespace

	class TestLibYuv : public ::testing::Test {
	protected:
	TestLibYuv();
	void SetUp() override;
	void TearDown() override;

	FILE* source_file_;
	std::unique_ptr<VideoFrame> orig_frame_;
	const int width_;
	const int height_;
	const int size_y_;
	const int size_uv_;
	const size_t frame_length_;
	};

	TestLibYuv::TestLibYuv()
	: source_file_(NULL),
	orig_frame_(),
	width_(352),
	height_(288),
	size_y_(width_ * height_),
	size_uv_(((width_ + 1) / 2) * ((height_ + 1) / 2)),
	frame_length_(CalcBufferSize(VideoType::kI420, 352, 288)) {}

	void TestLibYuv::SetUp() {
	const std::string input_file_name =
	webrtc::test::ResourcePath("foreman_cif", "yuv");
	source_file_ = fopen(input_file_name.c_str(), "rb");
	ASSERT_TRUE(source_file_ != NULL)
	<< "Cannot read file: " << input_file_name << "\n";

	rtc::scoped_refptr<I420BufferInterface> buffer(
	test::ReadI420Buffer(width_, height_, source_file_));

	orig_frame_ =
	std::make_unique<VideoFrame>(VideoFrame::Builder()
	.set_video_frame_buffer(buffer)
	.set_rotation(webrtc::kVideoRotation_0)
	.set_timestamp_us(0)
	.build());
	}

	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.0;

	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_]);
	EXPECT_EQ(0, ConvertFromI420(*orig_frame_, VideoType::kI420, 0,
	out_i420_buffer.get()));
	int y_size = width_ * height_;
	int u_size = res_i420_buffer->ChromaWidth() * res_i420_buffer->ChromaHeight();
	int ret = libyuv::I420Copy(
	out_i420_buffer.get(), width_, out_i420_buffer.get() + y_size,
	width_ >> 1, out_i420_buffer.get() + y_size + u_size, width_ >> 1,
	res_i420_buffer.get()->MutableDataY(), res_i420_buffer.get()->StrideY(),
	res_i420_buffer.get()->MutableDataU(), res_i420_buffer.get()->StrideU(),
	res_i420_buffer.get()->MutableDataV(), res_i420_buffer.get()->StrideV(),
	width_, height_);
	EXPECT_EQ(0, ret);

	if (PrintVideoFrame(*res_i420_buffer, output_file) < 0) {
	return;
	}
	psnr =
	I420PSNR(orig_frame_->video_frame_buffer()->GetI420(), res_i420_buffer);
	EXPECT_EQ(48.0, psnr);
	j++;

	printf("\nConvert #%d I420 <-> RGB24\n", j);
	std::unique_ptr<uint8_t[]> res_rgb_buffer2(new uint8_t[width_ * height_ * 3]);
	// Align the stride values for the output frame.
	int stride_y = 0;
	int stride_uv = 0;
	Calc16ByteAlignedStride(width_, &stride_y, &stride_uv);
	res_i420_buffer =
	I420Buffer::Create(width_, height_, stride_y, stride_uv, stride_uv);
	EXPECT_EQ(0, ConvertFromI420(*orig_frame_, VideoType::kRGB24, 0,
	res_rgb_buffer2.get()));

	ret = libyuv::ConvertToI420(
	res_rgb_buffer2.get(), 0, res_i420_buffer.get()->MutableDataY(),
	res_i420_buffer.get()->StrideY(), res_i420_buffer.get()->MutableDataU(),
	res_i420_buffer.get()->StrideU(), res_i420_buffer.get()->MutableDataV(),
	res_i420_buffer.get()->StrideV(), 0, 0, width_, height_,
	res_i420_buffer->width(), res_i420_buffer->height(), libyuv::kRotate0,
	ConvertVideoType(VideoType::kRGB24));

	EXPECT_EQ(0, ret);
	if (PrintVideoFrame(*res_i420_buffer, output_file) < 0) {
	return;
	}
	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);
	j++;

	printf("\nConvert #%d I420 <-> UYVY\n", j);
	std::unique_ptr<uint8_t[]> out_uyvy_buffer(new uint8_t[width_ * height_ * 2]);
	EXPECT_EQ(0, ConvertFromI420(*orig_frame_, VideoType::kUYVY, 0,
	out_uyvy_buffer.get()));

	ret = libyuv::ConvertToI420(
	out_uyvy_buffer.get(), 0, res_i420_buffer.get()->MutableDataY(),
	res_i420_buffer.get()->StrideY(), res_i420_buffer.get()->MutableDataU(),
	res_i420_buffer.get()->StrideU(), res_i420_buffer.get()->MutableDataV(),
	res_i420_buffer.get()->StrideV(), 0, 0, width_, height_,
	res_i420_buffer->width(), res_i420_buffer->height(), libyuv::kRotate0,
	ConvertVideoType(VideoType::kUYVY));

	EXPECT_EQ(0, ret);
	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;
	}
	j++;

	printf("\nConvert #%d I420 <-> YUY2\n", j);
	std::unique_ptr<uint8_t[]> out_yuy2_buffer(new uint8_t[width_ * height_ * 2]);
	EXPECT_EQ(0, ConvertFromI420(*orig_frame_, VideoType::kYUY2, 0,
	out_yuy2_buffer.get()));

	ret = libyuv::ConvertToI420(
	out_yuy2_buffer.get(), 0, res_i420_buffer.get()->MutableDataY(),
	res_i420_buffer.get()->StrideY(), res_i420_buffer.get()->MutableDataU(),
	res_i420_buffer.get()->StrideU(), res_i420_buffer.get()->MutableDataV(),
	res_i420_buffer.get()->StrideV(), 0, 0, width_, height_,
	res_i420_buffer->width(), res_i420_buffer->height(), libyuv::kRotate0,
	ConvertVideoType(VideoType::kYUY2));

	EXPECT_EQ(0, ret);

	if (PrintVideoFrame(*res_i420_buffer, output_file) < 0) {
	return;
	}

	psnr =
	I420PSNR(orig_frame_->video_frame_buffer()->GetI420(), res_i420_buffer);
	EXPECT_EQ(48.0, psnr);

	printf("\nConvert #%d I420 <-> RGB565\n", j);
	std::unique_ptr<uint8_t[]> out_rgb565_buffer(
	new uint8_t[width_ * height_ * 2]);
	EXPECT_EQ(0, ConvertFromI420(*orig_frame_, VideoType::kRGB565, 0,
	out_rgb565_buffer.get()));

	ret = libyuv::ConvertToI420(
	out_rgb565_buffer.get(), 0, res_i420_buffer.get()->MutableDataY(),
	res_i420_buffer.get()->StrideY(), res_i420_buffer.get()->MutableDataU(),
	res_i420_buffer.get()->StrideU(), res_i420_buffer.get()->MutableDataV(),
	res_i420_buffer.get()->StrideV(), 0, 0, width_, height_,
	res_i420_buffer->width(), res_i420_buffer->height(), libyuv::kRotate0,
	ConvertVideoType(VideoType::kRGB565));

	EXPECT_EQ(0, ret);
	if (PrintVideoFrame(*res_i420_buffer, output_file) < 0) {
	return;
	}
	j++;

	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.
	EXPECT_GT(ceil(psnr), 40);

	printf("\nConvert #%d I420 <-> ARGB8888\n", j);
	std::unique_ptr<uint8_t[]> out_argb8888_buffer(
	new uint8_t[width_ * height_ * 4]);
	EXPECT_EQ(0, ConvertFromI420(*orig_frame_, VideoType::kARGB, 0,
	out_argb8888_buffer.get()));

	ret = libyuv::ConvertToI420(
	out_argb8888_buffer.get(), 0, res_i420_buffer.get()->MutableDataY(),
	res_i420_buffer.get()->StrideY(), res_i420_buffer.get()->MutableDataU(),
	res_i420_buffer.get()->StrideU(), res_i420_buffer.get()->MutableDataV(),
	res_i420_buffer.get()->StrideV(), 0, 0, width_, height_,
	res_i420_buffer->width(), res_i420_buffer->height(), libyuv::kRotate0,
	ConvertVideoType(VideoType::kARGB));

	EXPECT_EQ(0, ret);

	if (PrintVideoFrame(*res_i420_buffer, output_file) < 0) {
	return;
	}

	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);

	ASSERT_EQ(0, fclose(output_file));
	}

	TEST_F(TestLibYuv, ConvertAlignedFrame) {
	// Reading YUV frame - testing on the first frame of the foreman sequence
	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.0;

	int stride_y = 0;
	int stride_uv = 0;
	Calc16ByteAlignedStride(width_, &stride_y, &stride_uv);

	rtc::scoped_refptr<I420Buffer> res_i420_buffer =
	I420Buffer::Create(width_, height_, stride_y, stride_uv, stride_uv);
	std::unique_ptr<uint8_t[]> out_i420_buffer(new uint8_t[frame_length_]);
	EXPECT_EQ(0, ConvertFromI420(*orig_frame_, VideoType::kI420, 0,
	out_i420_buffer.get()));
	int y_size = width_ * height_;
	int u_size = res_i420_buffer->ChromaWidth() * res_i420_buffer->ChromaHeight();
	int ret = libyuv::I420Copy(
	out_i420_buffer.get(), width_, out_i420_buffer.get() + y_size,
	width_ >> 1, out_i420_buffer.get() + y_size + u_size, width_ >> 1,
	res_i420_buffer.get()->MutableDataY(), res_i420_buffer.get()->StrideY(),
	res_i420_buffer.get()->MutableDataU(), res_i420_buffer.get()->StrideU(),
	res_i420_buffer.get()->MutableDataV(), res_i420_buffer.get()->StrideV(),
	width_, height_);

	EXPECT_EQ(0, ret);

	if (PrintVideoFrame(*res_i420_buffer, output_file) < 0) {
	return;
	}
	psnr =
	I420PSNR(orig_frame_->video_frame_buffer()->GetI420(), res_i420_buffer);
	EXPECT_EQ(48.0, psnr);
	}

	static uint8_t Average(int a, int b, int c, int d) {
	return (a + b + c + d + 2) / 4;
	}

	TEST_F(TestLibYuv, NV12Scale2x2to2x2) {
	const std::vector<uint8_t> src_y = {0, 1, 2, 3};
	const std::vector<uint8_t> src_uv = {0, 1};
	std::vector<uint8_t> dst_y(4);
	std::vector<uint8_t> dst_uv(2);

	uint8_t* tmp_buffer = nullptr;

	NV12Scale(tmp_buffer, src_y.data(), 2, src_uv.data(), 2, 2, 2, dst_y.data(),
	2, dst_uv.data(), 2, 2, 2);

	EXPECT_THAT(dst_y, ::testing::ContainerEq(src_y));
	EXPECT_THAT(dst_uv, ::testing::ContainerEq(src_uv));
	}

	TEST_F(TestLibYuv, NV12Scale4x4to2x2) {
	const uint8_t src_y[] = {0, 1, 2, 3, 4, 5, 6, 7,
	8, 9, 10, 11, 12, 13, 14, 15};
	const uint8_t src_uv[] = {0, 1, 2, 3, 4, 5, 6, 7};
	std::vector<uint8_t> dst_y(4);
	std::vector<uint8_t> dst_uv(2);

	std::vector<uint8_t> tmp_buffer;
	const int src_chroma_width = (4 + 1) / 2;
	const int src_chroma_height = (4 + 1) / 2;
	const int dst_chroma_width = (2 + 1) / 2;
	const int dst_chroma_height = (2 + 1) / 2;
	tmp_buffer.resize(src_chroma_width * src_chroma_height * 2 +
	dst_chroma_width * dst_chroma_height * 2);
	tmp_buffer.shrink_to_fit();

	NV12Scale(tmp_buffer.data(), src_y, 4, src_uv, 4, 4, 4, dst_y.data(), 2,
	dst_uv.data(), 2, 2, 2);

	EXPECT_THAT(dst_y, ::testing::ElementsAre(
	Average(0, 1, 4, 5), Average(2, 3, 6, 7),
	Average(8, 9, 12, 13), Average(10, 11, 14, 15)));
	EXPECT_THAT(dst_uv,
	::testing::ElementsAre(Average(0, 2, 4, 6), Average(1, 3, 5, 7)));
	}

	} // namespace webrtc