blob: 9161fb94418891482f659ebb7ae5ae6e064a7330 [file] [log] [blame]
// Copyright 2019 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 <limits>
#include <memory>
#include "base/files/file_path.h"
#include "base/memory/scoped_refptr.h"
#include "base/sequence_checker.h"
#include "base/synchronization/condition_variable.h"
#include "base/synchronization/lock.h"
#include "base/threading/thread.h"
#include "media/gpu/test/video_frame_helpers.h"
namespace media {
class VideoFrameMapper;
namespace test {
// The video frame file writer class implements functionality to write video
// frames to file. The supported output formats are PNG and raw I420 YUV.
class VideoFrameFileWriter : public VideoFrameProcessor {
// Supported output formats.
enum class OutputFormat {
kPNG = 0,
~VideoFrameFileWriter() override;
// Create an instance of the video frame file writer.
// |output_folder| specifies the folder video frames will be written to.
// |output_format| specifies the output file format.
// |output_limit| limits the max number of files that can be written.
static std::unique_ptr<VideoFrameFileWriter> Create(
const base::FilePath& output_folder,
OutputFormat output_format = OutputFormat::kPNG,
size_t output_limit = std::numeric_limits<size_t>::max(),
const base::FilePath::StringType& output_file_prefix =
// Interface VideoFrameProcessor
void ProcessVideoFrame(scoped_refptr<const VideoFrame> video_frame,
size_t frame_index) override;
// Wait until all currently scheduled frame write operations are done.
bool WaitUntilDone() override;
VideoFrameFileWriter(const base::FilePath& output_folder,
OutputFormat output_format,
size_t output_limit,
const base::FilePath::StringType& output_prefix);
// Initialize the video frame file writer.
bool Initialize();
// Writes the specified video frame to file on the |file_writer_thread_|.
void ProcessVideoFrameTask(scoped_refptr<const VideoFrame> video_frame,
size_t frame_index);
// Write the video frame to disk in PNG format.
void WriteVideoFramePNG(scoped_refptr<const VideoFrame> video_frame,
const base::FilePath& filename);
// Write the video frame to disk in I420 YUV format.
void WriteVideoFrameYUV(scoped_refptr<const VideoFrame> video_frame,
const base::FilePath& filename);
// Output folder the frames will be written to.
const base::FilePath output_folder_;
// Output format of the frames.
const OutputFormat output_format_;
// The maximum number of frames that can be written.
const size_t output_limit_;
// The prefix of the output file.
const base::FilePath::StringType output_file_prefix_;
// The video frame mapper used to gain access to the raw video frame memory.
std::unique_ptr<VideoFrameMapper> video_frame_mapper_;
// The number of frames currently queued for writing.
size_t num_frames_writing_ GUARDED_BY(frame_writer_lock_);
// The number of frames currently written or queued to be written.
size_t num_frames_writes_requested_ = 0u;
// Thread on which video frame writing is done.
base::Thread frame_writer_thread_;
mutable base::Lock frame_writer_lock_;
mutable base::ConditionVariable frame_writer_cv_;
} // namespace test
} // namespace media