1 // Copyright 2014 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
5 #include "media/cast/test/fake_video_encode_accelerator.h"
8 #include "base/location.h"
9 #include "base/logging.h"
10 #include "base/single_thread_task_runner.h"
16 static const unsigned int kMinimumInputCount = 1;
17 static const size_t kMinimumOutputBufferSize = 123456;
19 FakeVideoEncodeAccelerator::FakeVideoEncodeAccelerator(
20 const scoped_refptr<base::SingleThreadTaskRunner>& task_runner,
21 std::vector<uint32>* stored_bitrates)
22 : task_runner_(task_runner),
23 stored_bitrates_(stored_bitrates),
26 will_initialization_succeed_(true),
27 weak_this_factory_(this) {
28 DCHECK(stored_bitrates_);
31 FakeVideoEncodeAccelerator::~FakeVideoEncodeAccelerator() {
32 weak_this_factory_.InvalidateWeakPtrs();
35 std::vector<VideoEncodeAccelerator::SupportedProfile>
36 FakeVideoEncodeAccelerator::GetSupportedProfiles() {
37 return std::vector<VideoEncodeAccelerator::SupportedProfile>();
40 bool FakeVideoEncodeAccelerator::Initialize(
41 media::VideoFrame::Format input_format,
42 const gfx::Size& input_visible_size,
43 VideoCodecProfile output_profile,
44 uint32 initial_bitrate,
46 if (!will_initialization_succeed_)
49 if (output_profile != media::VP8PROFILE_ANY &&
50 output_profile != media::H264PROFILE_MAIN) {
53 task_runner_->PostTask(
55 base::Bind(&FakeVideoEncodeAccelerator::DoRequireBitstreamBuffers,
56 weak_this_factory_.GetWeakPtr(),
59 kMinimumOutputBufferSize));
63 void FakeVideoEncodeAccelerator::Encode(const scoped_refptr<VideoFrame>& frame,
64 bool force_keyframe) {
66 DCHECK(!available_buffer_ids_.empty());
68 // Fake that we have encoded the frame; resulting in using the full output
70 int32 id = available_buffer_ids_.front();
71 available_buffer_ids_.pop_front();
73 bool is_key_fame = force_keyframe;
78 task_runner_->PostTask(
80 base::Bind(&FakeVideoEncodeAccelerator::DoBitstreamBufferReady,
81 weak_this_factory_.GetWeakPtr(),
83 kMinimumOutputBufferSize,
87 void FakeVideoEncodeAccelerator::UseOutputBitstreamBuffer(
88 const BitstreamBuffer& buffer) {
89 available_buffer_ids_.push_back(buffer.id());
92 void FakeVideoEncodeAccelerator::RequestEncodingParametersChange(
95 stored_bitrates_->push_back(bitrate);
98 void FakeVideoEncodeAccelerator::Destroy() { delete this; }
100 void FakeVideoEncodeAccelerator::SendDummyFrameForTesting(bool key_frame) {
101 DoBitstreamBufferReady(0, 23, key_frame);
104 void FakeVideoEncodeAccelerator::DoRequireBitstreamBuffers(
105 unsigned int input_count,
106 const gfx::Size& input_coded_size,
107 size_t output_buffer_size) const {
108 client_->RequireBitstreamBuffers(
109 input_count, input_coded_size, output_buffer_size);
112 void FakeVideoEncodeAccelerator::DoBitstreamBufferReady(
113 int32 bitstream_buffer_id,
115 bool key_frame) const {
116 client_->BitstreamBufferReady(bitstream_buffer_id, payload_size, key_frame);