media/mojo/services/demuxer_stream_provider_shim.cc

   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
   3 // found in the LICENSE file.
   4
   5 #include "media/mojo/services/demuxer_stream_provider_shim.h"
   6
   7 #include "base/bind.h"
   8 #include "base/bind_helpers.h"
   9 #include "base/callback_helpers.h"
  10
  11 namespace media {
  12
  13 DemuxerStreamProviderShim::DemuxerStreamProviderShim(
  14     mojo::DemuxerStreamPtr audio,
  15     mojo::DemuxerStreamPtr video,
  16     const base::Closure& demuxer_ready_cb)
  17     : demuxer_ready_cb_(demuxer_ready_cb),
  18       streams_ready_(0),
  19       weak_factory_(this) {
  20   DCHECK(audio || video);
  21   DCHECK(!demuxer_ready_cb_.is_null());
  22
  23   if (audio) {
  24     streams_.push_back(new MojoDemuxerStreamAdapter(
  25         audio.Pass(),
  26         base::Bind(&DemuxerStreamProviderShim::OnStreamReady,
  27                    weak_factory_.GetWeakPtr())));
  28   }
  29
  30   if (video) {
  31     streams_.push_back(new MojoDemuxerStreamAdapter(
  32         video.Pass(),
  33         base::Bind(&DemuxerStreamProviderShim::OnStreamReady,
  34                    weak_factory_.GetWeakPtr())));
  35   }
  36 }
  37
  38 DemuxerStreamProviderShim::~DemuxerStreamProviderShim() {
  39 }
  40
  41 DemuxerStream* DemuxerStreamProviderShim::GetStream(DemuxerStream::Type type) {
  42   DCHECK(demuxer_ready_cb_.is_null());
  43   for (auto* stream : streams_) {
  44     if (stream->type() == type)
  45       return stream;
  46   }
  47
  48   return nullptr;
  49 }
  50
  51 void DemuxerStreamProviderShim::OnStreamReady() {
  52   if (++streams_ready_ == streams_.size())
  53     base::ResetAndReturn(&demuxer_ready_cb_).Run();
  54 }
  55
  56 }  // namespace media