MediaResource refactoring to support multiple streams

media/mojo/services/media_resource_shim.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
-#include "media/mojo/services/demuxer_stream_provider_shim.h"
+#include "media/mojo/services/media_resource_shim.h"
 
 #include <utility>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/callback_helpers.h"
 
 namespace media {
 
-DemuxerStreamProviderShim::DemuxerStreamProviderShim(
+MediaResourceShim::MediaResourceShim(
     std::vector<mojom::DemuxerStreamPtr> streams,
     const base::Closure& demuxer_ready_cb)
     : demuxer_ready_cb_(demuxer_ready_cb),
       streams_ready_(0),
       weak_factory_(this) {
   DCHECK(!streams.empty());
   DCHECK(!demuxer_ready_cb_.is_null());
 
   for (auto& s : streams) {
     streams_.emplace_back(new MojoDemuxerStreamAdapter(
-        std::move(s), base::Bind(&DemuxerStreamProviderShim::OnStreamReady,
+        std::move(s), base::Bind(&MediaResourceShim::OnStreamReady,
                                  weak_factory_.GetWeakPtr())));
   }
 }
 
-DemuxerStreamProviderShim::~DemuxerStreamProviderShim() {
+MediaResourceShim::~MediaResourceShim() {}
+
+std::vector<DemuxerStream*> MediaResourceShim::GetStreams() {
+  DCHECK(demuxer_ready_cb_.is_null());
+  std::vector<DemuxerStream*> result;
+  for (auto& stream : streams_)
+    result.push_back(stream.get());

[tguilbert, 2017/01/24 23:24:41]

Is there a difference between this code using the vector copy constructor?

[servolk, 2017/01/25 00:57:32]

We can't use a copy constructor, since |streams_| is a vector of
std::unique_ptrs which can't be implicitly converted to a vector of
DemuxerStream*, so we have to do this explicitly.

[tguilbert, 2017/01/25 02:36:46]

Oops, you are right, I missed |streams_|'s type.

+  return result;
 }
 
-DemuxerStream* DemuxerStreamProviderShim::GetStream(DemuxerStream::Type type) {
-  DCHECK(demuxer_ready_cb_.is_null());
-  for (auto& stream : streams_) {
-    if (stream->type() == type)
-      return stream.get();
-  }
-
-  return nullptr;
-}
-
-void DemuxerStreamProviderShim::OnStreamReady() {
+void MediaResourceShim::OnStreamReady() {
   if (++streams_ready_ == streams_.size())
     base::ResetAndReturn(&demuxer_ready_cb_).Run();
 }
 
 }  // namespace media