| Index: content/common/gpu/media/vaapi_h264_decoder_unittest.cc
|
| diff --git a/content/common/gpu/media/vaapi_h264_decoder_unittest.cc b/content/common/gpu/media/vaapi_h264_decoder_unittest.cc
|
| deleted file mode 100644
|
| index cc8fc188d99eab524369e9a73114d0ed0491c6c8..0000000000000000000000000000000000000000
|
| --- a/content/common/gpu/media/vaapi_h264_decoder_unittest.cc
|
| +++ /dev/null
|
| @@ -1,385 +0,0 @@
|
| -// Copyright 2014 The Chromium Authors. All rights reserved.
|
| -// Use of this source code is governed by a BSD-style license that can be
|
| -// found in the LICENSE file.
|
| -
|
| -#include <string>
|
| -
|
| -// This has to be included first.
|
| -// See http://code.google.com/p/googletest/issues/detail?id=371
|
| -#include "testing/gtest/include/gtest/gtest.h"
|
| -
|
| -#include "base/at_exit.h"
|
| -#include "base/bind.h"
|
| -#include "base/command_line.h"
|
| -#include "base/files/file_util.h"
|
| -#include "base/logging.h"
|
| -#include "base/message_loop/message_loop.h"
|
| -#include "content/common/gpu/media/h264_decoder.h"
|
| -#include "media/base/video_decoder_config.h"
|
| -#include "third_party/libyuv/include/libyuv.h"
|
| -
|
| -#if defined(USE_OZONE)
|
| -#include "ui/ozone/public/ozone_platform.h"
|
| -#endif
|
| -
|
| -// This program is run like this:
|
| -// DISPLAY=:0 ./vaapi_h264_decoder_unittest --input_file input.h264
|
| -// [--output_file output.i420] [--md5sum expected_md5_hex]
|
| -//
|
| -// The input is read from input.h264. The output is written to output.i420 if it
|
| -// is given. It also verifies the MD5 sum of the decoded I420 data if the
|
| -// expected MD5 sum is given.
|
| -
|
| -namespace content {
|
| -namespace {
|
| -
|
| -// These are the command line parameters
|
| -base::FilePath g_input_file;
|
| -base::FilePath g_output_file;
|
| -std::string g_md5sum;
|
| -
|
| -// These default values are used if nothing is specified in the command line.
|
| -const base::FilePath::CharType* kDefaultInputFile =
|
| - FILE_PATH_LITERAL("test-25fps.h264");
|
| -const char* kDefaultMD5Sum = "3af866863225b956001252ebeccdb71d";
|
| -
|
| -// This class encapsulates the use of VaapiH264Decoder to a simpler interface.
|
| -// It reads an H.264 Annex B bytestream from a file and outputs the decoded
|
| -// frames (in I420 format) to another file. The output file can be played by
|
| -// $ mplayer test_dec.yuv -demuxer rawvideo -rawvideo w=1920:h=1080
|
| -//
|
| -// To use the class, construct an instance, call Initialize() to specify the
|
| -// input and output file paths, then call Run() to decode the whole stream and
|
| -// output the frames.
|
| -//
|
| -// This class must be created, called and destroyed on a single thread, and
|
| -// does nothing internally on any other thread.
|
| -class VaapiH264DecoderLoop {
|
| - public:
|
| - VaapiH264DecoderLoop();
|
| - ~VaapiH264DecoderLoop();
|
| -
|
| - // Initialize the decoder. Return true if successful.
|
| - bool Initialize(base::FilePath input_file, base::FilePath output_file);
|
| -
|
| - // Run the decode loop. The decoded data is written to the file specified by
|
| - // output_file in Initialize(). Return true if all decoding is successful.
|
| - bool Run();
|
| -
|
| - // Get the MD5 sum of the decoded data.
|
| - std::string GetMD5Sum();
|
| -
|
| - private:
|
| - // Callback from the decoder when a picture is decoded.
|
| - void OutputPicture(int32 input_id,
|
| - const scoped_refptr<VASurface>& va_surface);
|
| -
|
| - // Recycle one surface and put it on available_surfaces_ list.
|
| - void RecycleSurface(VASurfaceID va_surface_id);
|
| -
|
| - // Give all surfaces in available_surfaces_ to the decoder.
|
| - void RefillSurfaces();
|
| -
|
| - // Free the current set of surfaces and allocate a new set of
|
| - // surfaces. Returns true when successful.
|
| - bool AllocateNewSurfaces();
|
| -
|
| - // Use the data in the frame: write to file and update MD5 sum.
|
| - bool ProcessVideoFrame(const scoped_refptr<media::VideoFrame>& frame);
|
| -
|
| - scoped_ptr<VaapiWrapper> wrapper_;
|
| - scoped_ptr<VaapiH264Decoder> decoder_;
|
| - std::string data_; // data read from input_file
|
| - base::FilePath output_file_; // output data is written to this file
|
| - std::vector<VASurfaceID> available_surfaces_;
|
| -
|
| - // These members (num_outputted_pictures_, num_surfaces_)
|
| - // need to be initialized and possibly freed manually.
|
| - int num_outputted_pictures_; // number of pictures already outputted
|
| - size_t num_surfaces_; // number of surfaces in the current set of surfaces
|
| - base::MD5Context md5_context_;
|
| -};
|
| -
|
| -VaapiH264DecoderLoop::VaapiH264DecoderLoop()
|
| - : num_outputted_pictures_(0), num_surfaces_(0) {
|
| - base::MD5Init(&md5_context_);
|
| -}
|
| -
|
| -VaapiH264DecoderLoop::~VaapiH264DecoderLoop() {
|
| -}
|
| -
|
| -void LogOnError(VaapiH264Decoder::VAVDAH264DecoderFailure error) {
|
| - LOG(FATAL) << "Oh noes! Decoder failed: " << error;
|
| -}
|
| -
|
| -bool VaapiH264DecoderLoop::Initialize(base::FilePath input_file,
|
| - base::FilePath output_file) {
|
| - media::VideoCodecProfile profile = media::H264PROFILE_BASELINE;
|
| - base::Closure report_error_cb =
|
| - base::Bind(&LogOnError, VaapiH264Decoder::VAAPI_ERROR);
|
| - wrapper_ = VaapiWrapper::CreateForVideoCodec(VaapiWrapper::kDecode, profile,
|
| - report_error_cb);
|
| - if (!wrapper_.get()) {
|
| - LOG(ERROR) << "Can't create vaapi wrapper";
|
| - return false;
|
| - }
|
| -
|
| - decoder_.reset(new VaapiH264Decoder(
|
| - wrapper_.get(),
|
| - base::Bind(&VaapiH264DecoderLoop::OutputPicture, base::Unretained(this)),
|
| - base::Bind(&LogOnError)));
|
| -
|
| - if (!base::ReadFileToString(input_file, &data_)) {
|
| - LOG(ERROR) << "failed to read input data from " << input_file.value();
|
| - return false;
|
| - }
|
| -
|
| - const int input_id = 0; // We don't use input_id in this class.
|
| - decoder_->SetStream(
|
| - reinterpret_cast<const uint8*>(data_.c_str()), data_.size(), input_id);
|
| -
|
| - // This creates or truncates output_file.
|
| - // Without it, AppendToFile() will not work.
|
| - if (!output_file.empty()) {
|
| - if (base::WriteFile(output_file, NULL, 0) != 0) {
|
| - return false;
|
| - }
|
| - output_file_ = output_file;
|
| - }
|
| -
|
| - return true;
|
| -}
|
| -
|
| -bool VaapiH264DecoderLoop::Run() {
|
| - while (1) {
|
| - switch (decoder_->Decode()) {
|
| - case VaapiH264Decoder::kDecodeError:
|
| - LOG(ERROR) << "Decode Error";
|
| - return false;
|
| - case VaapiH264Decoder::kAllocateNewSurfaces:
|
| - DVLOG(1) << "Allocate new surfaces";
|
| - if (!AllocateNewSurfaces()) {
|
| - LOG(ERROR) << "Failed to allocate new surfaces";
|
| - return false;
|
| - }
|
| - break;
|
| - case VaapiH264Decoder::kRanOutOfStreamData: {
|
| - bool rc = decoder_->Flush();
|
| - DVLOG(1) << "Flush returns " << rc;
|
| - return rc;
|
| - }
|
| - case VaapiH264Decoder::kRanOutOfSurfaces:
|
| - DVLOG(1) << "Ran out of surfaces";
|
| - RefillSurfaces();
|
| - break;
|
| - }
|
| - }
|
| -}
|
| -
|
| -std::string VaapiH264DecoderLoop::GetMD5Sum() {
|
| - base::MD5Digest digest;
|
| - base::MD5Final(&digest, &md5_context_);
|
| - return MD5DigestToBase16(digest);
|
| -}
|
| -
|
| -scoped_refptr<media::VideoFrame> CopyNV12ToI420(VAImage* image, void* mem) {
|
| - int width = image->width;
|
| - int height = image->height;
|
| -
|
| - DVLOG(1) << "CopyNV12ToI420 width=" << width << ", height=" << height;
|
| -
|
| - const gfx::Size coded_size(width, height);
|
| - const gfx::Rect visible_rect(width, height);
|
| - const gfx::Size natural_size(width, height);
|
| -
|
| - scoped_refptr<media::VideoFrame> frame =
|
| - media::VideoFrame::CreateFrame(media::VideoFrame::I420,
|
| - coded_size,
|
| - visible_rect,
|
| - natural_size,
|
| - base::TimeDelta());
|
| -
|
| - uint8_t* mem_byte_ptr = static_cast<uint8_t*>(mem);
|
| - uint8_t* src_y = mem_byte_ptr + image->offsets[0];
|
| - uint8_t* src_uv = mem_byte_ptr + image->offsets[1];
|
| - int src_stride_y = image->pitches[0];
|
| - int src_stride_uv = image->pitches[1];
|
| -
|
| - uint8_t* dst_y = frame->data(media::VideoFrame::kYPlane);
|
| - uint8_t* dst_u = frame->data(media::VideoFrame::kUPlane);
|
| - uint8_t* dst_v = frame->data(media::VideoFrame::kVPlane);
|
| - int dst_stride_y = frame->stride(media::VideoFrame::kYPlane);
|
| - int dst_stride_u = frame->stride(media::VideoFrame::kUPlane);
|
| - int dst_stride_v = frame->stride(media::VideoFrame::kVPlane);
|
| -
|
| - int rc = libyuv::NV12ToI420(src_y,
|
| - src_stride_y,
|
| - src_uv,
|
| - src_stride_uv,
|
| - dst_y,
|
| - dst_stride_y,
|
| - dst_u,
|
| - dst_stride_u,
|
| - dst_v,
|
| - dst_stride_v,
|
| - width,
|
| - height);
|
| - CHECK_EQ(0, rc);
|
| - return frame;
|
| -}
|
| -
|
| -bool VaapiH264DecoderLoop::ProcessVideoFrame(
|
| - const scoped_refptr<media::VideoFrame>& frame) {
|
| - frame->HashFrameForTesting(&md5_context_);
|
| -
|
| - if (output_file_.empty())
|
| - return true;
|
| -
|
| - for (size_t i = 0; i < media::VideoFrame::NumPlanes(frame->format()); i++) {
|
| - int to_write = media::VideoFrame::PlaneAllocationSize(
|
| - frame->format(), i, frame->coded_size());
|
| - const char* buf = reinterpret_cast<const char*>(frame->data(i));
|
| - if (!base::AppendToFile(output_file_, buf, to_write))
|
| - return false;
|
| - }
|
| - return true;
|
| -}
|
| -
|
| -void VaapiH264DecoderLoop::OutputPicture(
|
| - int32 input_id,
|
| - const scoped_refptr<VASurface>& va_surface) {
|
| - DVLOG(1) << "OutputPicture: picture " << num_outputted_pictures_++;
|
| -
|
| - VAImage image;
|
| - void* mem;
|
| -
|
| - if (!wrapper_->GetDerivedVaImage(va_surface->id(), &image, &mem)) {
|
| - LOG(ERROR) << "Cannot get VAImage.";
|
| - return;
|
| - }
|
| -
|
| - if (image.format.fourcc != VA_FOURCC_NV12) {
|
| - LOG(ERROR) << "Unexpected image format: " << image.format.fourcc;
|
| - wrapper_->ReturnVaImage(&image);
|
| - return;
|
| - }
|
| -
|
| - // Convert NV12 to I420 format.
|
| - scoped_refptr<media::VideoFrame> frame = CopyNV12ToI420(&image, mem);
|
| -
|
| - if (frame.get()) {
|
| - if (!ProcessVideoFrame(frame)) {
|
| - LOG(ERROR) << "Write to file failed";
|
| - }
|
| - } else {
|
| - LOG(ERROR) << "Cannot convert image to I420.";
|
| - }
|
| -
|
| - wrapper_->ReturnVaImage(&image);
|
| -}
|
| -
|
| -void VaapiH264DecoderLoop::RecycleSurface(VASurfaceID va_surface_id) {
|
| - available_surfaces_.push_back(va_surface_id);
|
| -}
|
| -
|
| -void VaapiH264DecoderLoop::RefillSurfaces() {
|
| - for (size_t i = 0; i < available_surfaces_.size(); i++) {
|
| - VASurface::ReleaseCB release_cb = base::Bind(
|
| - &VaapiH264DecoderLoop::RecycleSurface, base::Unretained(this));
|
| - scoped_refptr<VASurface> surface(new VASurface(
|
| - available_surfaces_[i], decoder_->GetPicSize(), release_cb));
|
| - decoder_->ReuseSurface(surface);
|
| - }
|
| - available_surfaces_.clear();
|
| -}
|
| -
|
| -bool VaapiH264DecoderLoop::AllocateNewSurfaces() {
|
| - CHECK_EQ(num_surfaces_, available_surfaces_.size())
|
| - << "not all surfaces are returned";
|
| -
|
| - available_surfaces_.clear();
|
| - wrapper_->DestroySurfaces();
|
| -
|
| - gfx::Size size = decoder_->GetPicSize();
|
| - num_surfaces_ = decoder_->GetRequiredNumOfPictures();
|
| - return wrapper_->CreateSurfaces(size, num_surfaces_, &available_surfaces_);
|
| -}
|
| -
|
| -TEST(VaapiH264DecoderTest, TestDecode) {
|
| - base::FilePath input_file = g_input_file;
|
| - base::FilePath output_file = g_output_file;
|
| - std::string md5sum = g_md5sum;
|
| -
|
| - // If nothing specified, use the default file in the source tree.
|
| - if (input_file.empty() && output_file.empty() && md5sum.empty()) {
|
| - input_file = base::FilePath(kDefaultInputFile);
|
| - md5sum = kDefaultMD5Sum;
|
| - } else {
|
| - ASSERT_FALSE(input_file.empty()) << "Need to specify --input_file";
|
| - }
|
| -
|
| - DVLOG(1) << "Input File: " << input_file.value();
|
| - DVLOG(1) << "Output File: " << output_file.value();
|
| - DVLOG(1) << "Expected MD5 sum: " << md5sum;
|
| -
|
| - content::VaapiH264DecoderLoop loop;
|
| - ASSERT_TRUE(loop.Initialize(input_file, output_file))
|
| - << "initialize decoder loop failed";
|
| - ASSERT_TRUE(loop.Run()) << "run decoder loop failed";
|
| -
|
| - if (!md5sum.empty()) {
|
| - std::string actual = loop.GetMD5Sum();
|
| - DVLOG(1) << "Actual MD5 sum: " << actual;
|
| - EXPECT_EQ(md5sum, actual);
|
| - }
|
| -}
|
| -
|
| -} // namespace
|
| -} // namespace content
|
| -
|
| -int main(int argc, char** argv) {
|
| - testing::InitGoogleTest(&argc, argv); // Removes gtest-specific args.
|
| - base::CommandLine::Init(argc, argv);
|
| -
|
| - base::ShadowingAtExitManager at_exit_manager;
|
| -
|
| - // Needed to enable DVLOG through --vmodule.
|
| - logging::LoggingSettings settings;
|
| - settings.logging_dest = logging::LOG_TO_SYSTEM_DEBUG_LOG;
|
| - CHECK(logging::InitLogging(settings));
|
| -
|
| - // Process command line.
|
| - base::CommandLine* cmd_line = base::CommandLine::ForCurrentProcess();
|
| - CHECK(cmd_line);
|
| -
|
| - base::CommandLine::SwitchMap switches = cmd_line->GetSwitches();
|
| - for (base::CommandLine::SwitchMap::const_iterator it = switches.begin();
|
| - it != switches.end();
|
| - ++it) {
|
| - if (it->first == "input_file") {
|
| - content::g_input_file = base::FilePath(it->second);
|
| - continue;
|
| - }
|
| - if (it->first == "output_file") {
|
| - content::g_output_file = base::FilePath(it->second);
|
| - continue;
|
| - }
|
| - if (it->first == "md5sum") {
|
| - content::g_md5sum = it->second;
|
| - continue;
|
| - }
|
| - if (it->first == "v" || it->first == "vmodule")
|
| - continue;
|
| - if (it->first == "ozone-platform")
|
| - continue;
|
| - LOG(FATAL) << "Unexpected switch: " << it->first << ":" << it->second;
|
| - }
|
| -
|
| -#if defined(USE_OZONE)
|
| - base::MessageLoopForUI main_loop;
|
| - ui::OzonePlatform::InitializeForUI();
|
| - ui::OzonePlatform::InitializeForGPU();
|
| -#endif
|
| -
|
| - return RUN_ALL_TESTS();
|
| -}
|
|
|