Motown: Framework parts for mojo transport (producer/consumer/mediapipe) and control (audiotrack).

services/media/framework_mojo/push_producer_base.cc

+// Copyright 2016 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 "base/bind.h"
+#include "base/bind_helpers.h"
+#include "base/logging.h"
+#include "base/message_loop/message_loop.h"
+#include "services/media/framework_mojo/push_producer_base.h"
+
+namespace mojo {
+namespace media {
+
+PushProducerBase::PushProducerBase() :
+    state_(MediaState::UNPREPARED),
+    end_of_stream_(false),
+    max_pushes_outstanding_(0),
+    current_pushes_outstanding_(0) {
+  task_runner_ = base::MessageLoop::current()->task_runner();
+  DCHECK(task_runner_);
+}
+
+PushProducerBase::~PushProducerBase() {
+  base::AutoLock lock(lock_);
+}
+
+void PushProducerBase::SetStatusCallback(
+    const StatusCallback& callback) {
+  status_callback_ = callback;
+}
+
+PayloadAllocator* PushProducerBase::allocator() {
+  return &mojo_allocator_;
+}
+
+void PushProducerBase::SetDemandCallback(
+    const DemandCallback& demand_callback) {
+  demand_callback_ = demand_callback;
+}
+
+void PushProducerBase::Prime() {
+  {
+    base::AutoLock lock(lock_);
+    max_pushes_outstanding_ = 10; // TODO(dalesat): Made up!
+  }
+
+  DCHECK(demand_callback_);
+  demand_callback_(Demand::kPositive);
+  SetState(MediaState::PAUSED);
+}
+
+Demand PushProducerBase::SupplyPacket(PacketPtr packet) {
+  DCHECK(packet);
+
+  // If we're no longer connected, throw the packet away.
+  if (!IsConnected()) {
+    SetState(MediaState::UNPREPARED);
+    // TODO(dalesat): More shutdown?
+    return Demand::kNegative;
+  }
+
+  Demand demand;
+
+  {
+    base::AutoLock lock(lock_);
+    DCHECK(current_pushes_outstanding_ < max_pushes_outstanding_);
+    DCHECK(!end_of_stream_) << "packet pushed after end-of-stream";
+
+    ++current_pushes_outstanding_;
+
+    if (packet->end_of_stream()) {
+      end_of_stream_ = true;
+      demand = Demand::kNegative;
+      max_pushes_outstanding_ = 0;
+    } else {
+      demand = current_pushes_outstanding_ < max_pushes_outstanding_ ?
+          Demand::kPositive :
+          Demand::kNegative;
+    }
+  }
+
+  MediaPacketPtr media_packet = CreateMediaPacket(packet);
+  task_runner_->PostTask(FROM_HERE, base::Bind(
+      &PushProducerBase::PushPacket,
+      base::Unretained(this),
+      packet.release(),
+      base::Passed(media_packet.Pass())));
+
+  return demand;
+}
+
+void PushProducerBase::PushPacket(
+    Packet* packet_raw_ptr,
+    MediaPacketPtr media_packet) {
+  PushPacketInternal(packet_raw_ptr, media_packet.Pass());
+}
+
+void PushProducerBase::PushCompleted(const PacketPtr& packet) {
+  DCHECK(packet);
+
+  Demand demand;
+
+  {
+    base::AutoLock lock(lock_);
+    DCHECK(current_pushes_outstanding_);
+    demand = --current_pushes_outstanding_ < max_pushes_outstanding_ ?
+        Demand::kPositive :
+        Demand::kNegative;
+  }
+
+  DCHECK(demand_callback_);
+  demand_callback_(demand);
+
+  if (end_of_stream_ && packet->end_of_stream()) {
+    SetState(MediaState::ENDED);
+  }
+}
+
+void PushProducerBase::OnConnectionLost() {
+  DCHECK(demand_callback_);
+  demand_callback_(Demand::kNegative);
+  SetState(MediaState::UNPREPARED);
+}
+
+void PushProducerBase::SetState(MediaState state) {
+  if (state_ != state) {
+    state_ = state;
+    if (status_callback_) {
+      status_callback_(state_);
+    }
+  }
+}
+
+MediaPacketPtr PushProducerBase::CreateMediaPacket(
+    const PacketPtr& packet) {
+  DCHECK(packet);
+
+  MediaPacketRegionPtr region = MediaPacketRegion::New();
+  region->offset = mojo_allocator_.OffsetFromPtr(packet->payload());
+  region->length = packet->size();
+
+  MediaPacketPtr media_packet = MediaPacket::New();
+  media_packet->pts = packet->presentation_time();
+  media_packet->duration = packet->duration();
+  media_packet->end_of_stream = packet->end_of_stream();
+  media_packet->payload = region.Pass();
+
+  return media_packet.Pass();
+}
+
+} // namespace media
+} // namespace mojo