OLD | NEW |
| (Empty) |
1 // Copyright 2016 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. | |
4 | |
5 #include "media/remoting/fake_remoting_demuxer_stream_provider.h" | |
6 | |
7 #include <vector> | |
8 | |
9 #include "base/callback_helpers.h" | |
10 #include "media/base/decoder_buffer.h" | |
11 #include "media/base/media_util.h" | |
12 #include "testing/gtest/include/gtest/gtest.h" | |
13 | |
14 using testing::_; | |
15 using testing::Invoke; | |
16 using testing::Return; | |
17 | |
18 namespace media { | |
19 | |
20 DummyDemuxerStream::DummyDemuxerStream(bool is_audio) { | |
21 type_ = is_audio ? DemuxerStream::AUDIO : DemuxerStream::VIDEO; | |
22 if (is_audio) { | |
23 audio_config_.Initialize(kCodecAAC, kSampleFormatS16, CHANNEL_LAYOUT_STEREO, | |
24 38400, std::vector<uint8_t>(), Unencrypted(), | |
25 base::TimeDelta(), 0); | |
26 } else { | |
27 gfx::Size size(640, 480); | |
28 gfx::Rect rect(0, 0, 640, 480); | |
29 video_config_.Initialize(kCodecH264, H264PROFILE_BASELINE, | |
30 PIXEL_FORMAT_I420, COLOR_SPACE_SD_REC601, size, | |
31 rect, size, std::vector<uint8_t>(), Unencrypted()); | |
32 } | |
33 ON_CALL(*this, Read(_)) | |
34 .WillByDefault(Invoke(this, &DummyDemuxerStream::FakeRead)); | |
35 } | |
36 | |
37 DummyDemuxerStream::~DummyDemuxerStream() = default; | |
38 | |
39 void DummyDemuxerStream::FakeRead(const ReadCB& read_cb) { | |
40 if (buffer_queue_.empty()) { | |
41 // Silent return to simulate waiting for buffer available. | |
42 pending_read_cb_ = read_cb; | |
43 return; | |
44 } | |
45 scoped_refptr<::media::DecoderBuffer> buffer = buffer_queue_.front(); | |
46 buffer_queue_.pop_front(); | |
47 read_cb.Run(kOk, buffer); | |
48 } | |
49 | |
50 AudioDecoderConfig DummyDemuxerStream::audio_decoder_config() { | |
51 return audio_config_; | |
52 } | |
53 | |
54 VideoDecoderConfig DummyDemuxerStream::video_decoder_config() { | |
55 return video_config_; | |
56 } | |
57 | |
58 DemuxerStream::Type DummyDemuxerStream::type() const { | |
59 return type_; | |
60 } | |
61 | |
62 DemuxerStream::Liveness DummyDemuxerStream::liveness() const { | |
63 return LIVENESS_UNKNOWN; | |
64 } | |
65 | |
66 bool DummyDemuxerStream::SupportsConfigChanges() { | |
67 return false; | |
68 } | |
69 | |
70 VideoRotation DummyDemuxerStream::video_rotation() { | |
71 return VIDEO_ROTATION_0; | |
72 } | |
73 | |
74 bool DummyDemuxerStream::enabled() const { | |
75 return false; | |
76 } | |
77 | |
78 void DummyDemuxerStream::CreateFakeFrame(size_t size, | |
79 bool key_frame, | |
80 int pts_ms) { | |
81 std::vector<uint8_t> buffer(size); | |
82 // Assign each byte in the buffer its index mod 256. | |
83 for (size_t i = 0; i < size; ++i) { | |
84 buffer[i] = static_cast<uint8_t>(i & 0xFF); | |
85 } | |
86 base::TimeDelta pts = base::TimeDelta::FromMilliseconds(pts_ms); | |
87 | |
88 // To DecoderBuffer | |
89 scoped_refptr<::media::DecoderBuffer> input_buffer = | |
90 ::media::DecoderBuffer::CopyFrom(buffer.data(), size); | |
91 input_buffer->set_timestamp(pts); | |
92 input_buffer->set_is_key_frame(key_frame); | |
93 | |
94 // Sends frame out if there is pending read callback. Otherwise, stores it | |
95 // in the buffer queue. | |
96 if (pending_read_cb_.is_null()) { | |
97 buffer_queue_.push_back(input_buffer); | |
98 } else { | |
99 base::ResetAndReturn(&pending_read_cb_).Run(kOk, input_buffer); | |
100 } | |
101 } | |
102 | |
103 FakeRemotingDemuxerStreamProvider::FakeRemotingDemuxerStreamProvider() | |
104 : demuxer_stream_(new DummyDemuxerStream(true)) {} | |
105 | |
106 FakeRemotingDemuxerStreamProvider::~FakeRemotingDemuxerStreamProvider() {} | |
107 | |
108 DemuxerStream* FakeRemotingDemuxerStreamProvider::GetStream( | |
109 DemuxerStream::Type type) { | |
110 if (type == DemuxerStream::AUDIO) | |
111 return reinterpret_cast<DemuxerStream*>(demuxer_stream_.get()); | |
112 return nullptr; | |
113 } | |
114 | |
115 } // namespace media | |
OLD | NEW |