Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef MEDIA_CAST_TEST_FAKE_MOCK_VIDEO_ENCODE_ACCELERATOR_H_ | 5 #ifndef MEDIA_VIDEO_FAKE_VIDEO_ENCODE_ACCELERATOR_H_ |
| 6 #define MEDIA_CAST_TEST_FAKE_MOCK_VIDEO_ENCODE_ACCELERATOR_H_ | 6 #define MEDIA_VIDEO_FAKE_VIDEO_ENCODE_ACCELERATOR_H_ |
| 7 | |
| 8 #include "media/video/video_encode_accelerator.h" | |
| 9 | 7 |
| 10 #include <list> | 8 #include <list> |
| 9 #include <queue> | |
| 11 #include <vector> | 10 #include <vector> |
| 12 | 11 |
| 13 #include "base/memory/weak_ptr.h" | 12 #include "base/memory/weak_ptr.h" |
| 14 #include "media/base/bitstream_buffer.h" | 13 #include "media/base/bitstream_buffer.h" |
| 14 #include "media/video/video_encode_accelerator.h" | |
| 15 | 15 |
| 16 namespace base { | 16 namespace base { |
| 17 | |
| 17 class SingleThreadTaskRunner; | 18 class SingleThreadTaskRunner; |
| 19 | |
| 18 } // namespace base | 20 } // namespace base |
| 19 | 21 |
| 20 namespace media { | 22 namespace media { |
| 21 namespace cast { | |
| 22 namespace test { | |
| 23 | 23 |
| 24 class FakeVideoEncodeAccelerator : public VideoEncodeAccelerator { | 24 class FakeVideoEncodeAccelerator : public VideoEncodeAccelerator { |
| 25 public: | 25 public: |
| 26 explicit FakeVideoEncodeAccelerator( | 26 FakeVideoEncodeAccelerator( |
| 27 const scoped_refptr<base::SingleThreadTaskRunner>& task_runner, | 27 const scoped_refptr<base::SingleThreadTaskRunner>& task_runner); |
| 28 std::vector<uint32>* stored_bitrates); | |
| 29 ~FakeVideoEncodeAccelerator() override; | 28 ~FakeVideoEncodeAccelerator() override; |
| 30 | 29 |
| 31 std::vector<VideoEncodeAccelerator::SupportedProfile> GetSupportedProfiles() | 30 std::vector<VideoEncodeAccelerator::SupportedProfile> GetSupportedProfiles() |
| 32 override; | 31 override; |
| 33 bool Initialize(media::VideoFrame::Format input_format, | 32 bool Initialize(VideoFrame::Format input_format, |
| 34 const gfx::Size& input_visible_size, | 33 const gfx::Size& input_visible_size, |
| 35 VideoCodecProfile output_profile, | 34 VideoCodecProfile output_profile, |
| 36 uint32 initial_bitrate, | 35 uint32 initial_bitrate, |
| 37 Client* client) override; | 36 Client* client) override; |
| 38 | |
| 39 void Encode(const scoped_refptr<VideoFrame>& frame, | 37 void Encode(const scoped_refptr<VideoFrame>& frame, |
| 40 bool force_keyframe) override; | 38 bool force_keyframe) override; |
| 41 | |
| 42 void UseOutputBitstreamBuffer(const BitstreamBuffer& buffer) override; | 39 void UseOutputBitstreamBuffer(const BitstreamBuffer& buffer) override; |
| 43 | |
| 44 void RequestEncodingParametersChange(uint32 bitrate, | 40 void RequestEncodingParametersChange(uint32 bitrate, |
| 45 uint32 framerate) override; | 41 uint32 framerate) override; |
| 46 | |
| 47 void Destroy() override; | 42 void Destroy() override; |
| 48 | 43 |
| 44 const std::vector<uint32>& stored_bitrates() { | |
| 45 return stored_bitrates_; | |
| 46 } | |
| 49 void SendDummyFrameForTesting(bool key_frame); | 47 void SendDummyFrameForTesting(bool key_frame); |
| 50 void SetWillInitializationSucceed(bool will_initialization_succeed) { | 48 void SetWillInitializationSucceed(bool will_initialization_succeed); |
| 51 will_initialization_succeed_ = will_initialization_succeed; | |
| 52 } | |
| 53 | 49 |
| 54 private: | 50 private: |
| 55 void DoRequireBitstreamBuffers(unsigned int input_count, | 51 void DoRequireBitstreamBuffers(unsigned int input_count, |
| 56 const gfx::Size& input_coded_size, | 52 const gfx::Size& input_coded_size, |
| 57 size_t output_buffer_size) const; | 53 size_t output_buffer_size) const; |
| 54 void EncodeTask(); | |
| 58 void DoBitstreamBufferReady(int32 bitstream_buffer_id, | 55 void DoBitstreamBufferReady(int32 bitstream_buffer_id, |
| 59 size_t payload_size, | 56 size_t kMinimumOutputBufferSize, |
|
wuchengli
2014/12/16 02:21:48
s/kMinimumOutputBufferSize/payload_size/
hellner1
2014/12/16 20:14:08
Done.
| |
| 60 bool key_frame) const; | 57 bool key_frame); |
| 61 | 58 |
| 59 // Our original (constructor) calling message loop used for all tasks. | |
| 62 const scoped_refptr<base::SingleThreadTaskRunner> task_runner_; | 60 const scoped_refptr<base::SingleThreadTaskRunner> task_runner_; |
| 63 std::vector<uint32>* const stored_bitrates_; | 61 std::vector<uint32> stored_bitrates_; |
| 64 VideoEncodeAccelerator::Client* client_; | |
| 65 bool first_; | |
| 66 bool will_initialization_succeed_; | 62 bool will_initialization_succeed_; |
| 67 | 63 |
| 68 std::list<int32> available_buffer_ids_; | 64 VideoEncodeAccelerator::Client* client_; |
| 65 | |
| 66 // Keeps track of if the current frame is the first encoded frame. This | |
| 67 // is used to force a fake key frame for the first encoded frame. | |
| 68 bool first_; | |
| 69 | |
| 70 // A queue containing the necessary data for incoming frames. The boolean | |
| 71 // represent whether the queued frame should force a key frame. | |
| 72 std::queue<bool> queued_frames_; | |
|
wuchengli
2014/12/16 02:21:48
nit: add a blank line
hellner1
2014/12/16 20:14:08
Done.
| |
| 73 // A list of buffers available for putting fake encoded frames in. | |
| 74 std::list<BitstreamBuffer> available_buffers_; | |
| 69 | 75 |
| 70 base::WeakPtrFactory<FakeVideoEncodeAccelerator> weak_this_factory_; | 76 base::WeakPtrFactory<FakeVideoEncodeAccelerator> weak_this_factory_; |
| 71 | 77 |
| 72 DISALLOW_COPY_AND_ASSIGN(FakeVideoEncodeAccelerator); | 78 DISALLOW_COPY_AND_ASSIGN(FakeVideoEncodeAccelerator); |
| 73 }; | 79 }; |
| 74 | 80 |
| 75 } // namespace test | |
| 76 } // namespace cast | |
| 77 } // namespace media | 81 } // namespace media |
| 78 | 82 |
| 79 #endif // MEDIA_CAST_TEST_FAKE_MOCK_VIDEO_ENCODE_ACCELERATOR_H_ | 83 #endif // MEDIA_VIDEO_FAKE_VIDEO_ENCODE_ACCELERATOR_H_ |
| OLD | NEW |