OLD | NEW |
| (Empty) |
1 // Copyright (c) 2013 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/filters/fake_demuxer_stream.h" | |
6 | |
7 #include "base/bind.h" | |
8 #include "base/callback_helpers.h" | |
9 #include "base/location.h" | |
10 #include "base/logging.h" | |
11 #include "base/single_thread_task_runner.h" | |
12 #include "base/thread_task_runner_handle.h" | |
13 #include "media/base/bind_to_current_loop.h" | |
14 #include "media/base/decoder_buffer.h" | |
15 #include "media/base/test_helpers.h" | |
16 #include "media/base/video_frame.h" | |
17 #include "ui/gfx/geometry/rect.h" | |
18 #include "ui/gfx/geometry/size.h" | |
19 | |
20 namespace media { | |
21 | |
22 const int kStartTimestampMs = 0; | |
23 const int kDurationMs = 30; | |
24 const int kStartWidth = 320; | |
25 const int kStartHeight = 240; | |
26 const int kWidthDelta = 4; | |
27 const int kHeightDelta = 3; | |
28 const uint8 kKeyId[] = { 0x00, 0x01, 0x02, 0x03 }; | |
29 const uint8 kIv[] = { | |
30 0x20, 0x21, 0x22, 0x23, 0x24, 0x25, 0x26, 0x27, | |
31 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 | |
32 }; | |
33 | |
34 FakeDemuxerStream::FakeDemuxerStream(int num_configs, | |
35 int num_buffers_in_one_config, | |
36 bool is_encrypted) | |
37 : task_runner_(base::ThreadTaskRunnerHandle::Get()), | |
38 num_configs_(num_configs), | |
39 num_buffers_in_one_config_(num_buffers_in_one_config), | |
40 config_changes_(num_configs > 1), | |
41 is_encrypted_(is_encrypted), | |
42 read_to_hold_(-1) { | |
43 DCHECK_GT(num_configs, 0); | |
44 DCHECK_GT(num_buffers_in_one_config, 0); | |
45 Initialize(); | |
46 UpdateVideoDecoderConfig(); | |
47 } | |
48 | |
49 FakeDemuxerStream::~FakeDemuxerStream() {} | |
50 | |
51 void FakeDemuxerStream::Initialize() { | |
52 DCHECK_EQ(-1, read_to_hold_); | |
53 num_configs_left_ = num_configs_; | |
54 num_buffers_left_in_current_config_ = num_buffers_in_one_config_; | |
55 num_buffers_returned_ = 0; | |
56 current_timestamp_ = base::TimeDelta::FromMilliseconds(kStartTimestampMs); | |
57 duration_ = base::TimeDelta::FromMilliseconds(kDurationMs); | |
58 splice_timestamp_ = kNoTimestamp(); | |
59 next_coded_size_ = gfx::Size(kStartWidth, kStartHeight); | |
60 next_read_num_ = 0; | |
61 } | |
62 | |
63 void FakeDemuxerStream::Read(const ReadCB& read_cb) { | |
64 DCHECK(task_runner_->BelongsToCurrentThread()); | |
65 DCHECK(read_cb_.is_null()); | |
66 | |
67 read_cb_ = BindToCurrentLoop(read_cb); | |
68 | |
69 if (read_to_hold_ == next_read_num_) | |
70 return; | |
71 | |
72 DCHECK(read_to_hold_ == -1 || read_to_hold_ > next_read_num_); | |
73 DoRead(); | |
74 } | |
75 | |
76 AudioDecoderConfig FakeDemuxerStream::audio_decoder_config() { | |
77 DCHECK(task_runner_->BelongsToCurrentThread()); | |
78 NOTREACHED(); | |
79 return AudioDecoderConfig(); | |
80 } | |
81 | |
82 VideoDecoderConfig FakeDemuxerStream::video_decoder_config() { | |
83 DCHECK(task_runner_->BelongsToCurrentThread()); | |
84 return video_decoder_config_; | |
85 } | |
86 | |
87 // TODO(xhwang): Support audio if needed. | |
88 DemuxerStream::Type FakeDemuxerStream::type() const { | |
89 DCHECK(task_runner_->BelongsToCurrentThread()); | |
90 return VIDEO; | |
91 } | |
92 | |
93 bool FakeDemuxerStream::SupportsConfigChanges() { | |
94 return config_changes_; | |
95 } | |
96 | |
97 VideoRotation FakeDemuxerStream::video_rotation() { | |
98 return VIDEO_ROTATION_0; | |
99 } | |
100 | |
101 void FakeDemuxerStream::HoldNextRead() { | |
102 DCHECK(task_runner_->BelongsToCurrentThread()); | |
103 read_to_hold_ = next_read_num_; | |
104 } | |
105 | |
106 void FakeDemuxerStream::HoldNextConfigChangeRead() { | |
107 DCHECK(task_runner_->BelongsToCurrentThread()); | |
108 // Set |read_to_hold_| to be the next config change read. | |
109 read_to_hold_ = next_read_num_ + num_buffers_in_one_config_ - | |
110 next_read_num_ % (num_buffers_in_one_config_ + 1); | |
111 } | |
112 | |
113 void FakeDemuxerStream::SatisfyRead() { | |
114 DCHECK(task_runner_->BelongsToCurrentThread()); | |
115 DCHECK_EQ(read_to_hold_, next_read_num_); | |
116 DCHECK(!read_cb_.is_null()); | |
117 | |
118 read_to_hold_ = -1; | |
119 DoRead(); | |
120 } | |
121 | |
122 void FakeDemuxerStream::SatisfyReadAndHoldNext() { | |
123 DCHECK(task_runner_->BelongsToCurrentThread()); | |
124 DCHECK_EQ(read_to_hold_, next_read_num_); | |
125 DCHECK(!read_cb_.is_null()); | |
126 | |
127 ++read_to_hold_; | |
128 DoRead(); | |
129 } | |
130 | |
131 void FakeDemuxerStream::Reset() { | |
132 read_to_hold_ = -1; | |
133 | |
134 if (!read_cb_.is_null()) | |
135 base::ResetAndReturn(&read_cb_).Run(kAborted, NULL); | |
136 } | |
137 | |
138 void FakeDemuxerStream::SeekToStart() { | |
139 Reset(); | |
140 Initialize(); | |
141 } | |
142 | |
143 void FakeDemuxerStream::UpdateVideoDecoderConfig() { | |
144 const gfx::Rect kVisibleRect(kStartWidth, kStartHeight); | |
145 video_decoder_config_.Initialize( | |
146 kCodecVP8, VIDEO_CODEC_PROFILE_UNKNOWN, VideoFrame::YV12, | |
147 next_coded_size_, kVisibleRect, next_coded_size_, | |
148 NULL, 0, is_encrypted_, false); | |
149 next_coded_size_.Enlarge(kWidthDelta, kHeightDelta); | |
150 } | |
151 | |
152 void FakeDemuxerStream::DoRead() { | |
153 DCHECK(task_runner_->BelongsToCurrentThread()); | |
154 DCHECK(!read_cb_.is_null()); | |
155 | |
156 next_read_num_++; | |
157 | |
158 if (num_buffers_left_in_current_config_ == 0) { | |
159 // End of stream. | |
160 if (num_configs_left_ == 0) { | |
161 base::ResetAndReturn(&read_cb_).Run(kOk, | |
162 DecoderBuffer::CreateEOSBuffer()); | |
163 return; | |
164 } | |
165 | |
166 // Config change. | |
167 num_buffers_left_in_current_config_ = num_buffers_in_one_config_; | |
168 UpdateVideoDecoderConfig(); | |
169 base::ResetAndReturn(&read_cb_).Run(kConfigChanged, NULL); | |
170 return; | |
171 } | |
172 | |
173 scoped_refptr<DecoderBuffer> buffer = CreateFakeVideoBufferForTest( | |
174 video_decoder_config_, current_timestamp_, duration_); | |
175 | |
176 // TODO(xhwang): Output out-of-order buffers if needed. | |
177 if (is_encrypted_) { | |
178 buffer->set_decrypt_config(scoped_ptr<DecryptConfig>( | |
179 new DecryptConfig(std::string(kKeyId, kKeyId + arraysize(kKeyId)), | |
180 std::string(kIv, kIv + arraysize(kIv)), | |
181 std::vector<SubsampleEntry>()))); | |
182 } | |
183 buffer->set_timestamp(current_timestamp_); | |
184 buffer->set_duration(duration_); | |
185 buffer->set_splice_timestamp(splice_timestamp_); | |
186 current_timestamp_ += duration_; | |
187 | |
188 num_buffers_left_in_current_config_--; | |
189 if (num_buffers_left_in_current_config_ == 0) | |
190 num_configs_left_--; | |
191 | |
192 num_buffers_returned_++; | |
193 base::ResetAndReturn(&read_cb_).Run(kOk, buffer); | |
194 } | |
195 | |
196 } // namespace media | |
OLD | NEW |