Index: media/mojo/services/demuxer_stream_provider_shim.cc |
diff --git a/media/mojo/services/demuxer_stream_provider_shim.cc b/media/mojo/services/demuxer_stream_provider_shim.cc |
index 0c9d8fea64f70708b9d9e88ded2be08ea32527d4..bf29ce1aefd6de5610b0abf83eb971a1115af553 100644 |
--- a/media/mojo/services/demuxer_stream_provider_shim.cc |
+++ b/media/mojo/services/demuxer_stream_provider_shim.cc |
@@ -13,25 +13,18 @@ |
namespace media { |
DemuxerStreamProviderShim::DemuxerStreamProviderShim( |
- mojom::DemuxerStreamPtr audio, |
- mojom::DemuxerStreamPtr video, |
+ std::vector<mojom::DemuxerStreamPtr> streams, |
const base::Closure& demuxer_ready_cb) |
: demuxer_ready_cb_(demuxer_ready_cb), |
streams_ready_(0), |
weak_factory_(this) { |
- DCHECK(audio || video); |
+ DCHECK(!streams.empty()); |
DCHECK(!demuxer_ready_cb_.is_null()); |
- if (audio) { |
- streams_.push_back(new MojoDemuxerStreamAdapter( |
- std::move(audio), base::Bind(&DemuxerStreamProviderShim::OnStreamReady, |
- weak_factory_.GetWeakPtr()))); |
- } |
- |
- if (video) { |
- streams_.push_back(new MojoDemuxerStreamAdapter( |
- std::move(video), base::Bind(&DemuxerStreamProviderShim::OnStreamReady, |
- weak_factory_.GetWeakPtr()))); |
+ for (auto& s : streams) { |
+ streams_.emplace_back(new MojoDemuxerStreamAdapter( |
+ std::move(s), base::Bind(&DemuxerStreamProviderShim::OnStreamReady, |
+ weak_factory_.GetWeakPtr()))); |
} |
} |
@@ -40,9 +33,9 @@ DemuxerStreamProviderShim::~DemuxerStreamProviderShim() { |
DemuxerStream* DemuxerStreamProviderShim::GetStream(DemuxerStream::Type type) { |
DCHECK(demuxer_ready_cb_.is_null()); |
- for (auto* stream : streams_) { |
+ for (auto& stream : streams_) { |
if (stream->type() == type) |
- return stream; |
+ return stream.get(); |
xhwang
2017/01/31 00:15:21
Add a comment that we only return the first stream
servolk
2017/01/31 01:10:40
Done. Although https://codereview.chromium.org/249
|
} |
return nullptr; |