Replace AudioScheduler with AudioPump.

remoting/host/audio_pump.cc

+// Copyright 2015 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 "remoting/host/audio_pump.h"
+
+#include "base/bind.h"
+#include "base/location.h"
+#include "base/logging.h"
+#include "base/single_thread_task_runner.h"
+#include "base/thread_task_runner_handle.h"
+#include "remoting/codec/audio_encoder.h"
+#include "remoting/host/audio_capturer.h"
+#include "remoting/proto/audio.pb.h"
+#include "remoting/protocol/audio_stub.h"
+
+namespace remoting {
+
+class AudioPump::Core {
+ public:
+  Core(base::WeakPtr<AudioPump> pump,
+       scoped_ptr<AudioCapturer> audio_capturer,
+       scoped_ptr<AudioEncoder> audio_encoder);
+  ~Core();
+
+  void Start();
+  void Pause(bool pause);
+
+ private:
+  void EncodeAudioPacket(scoped_ptr<AudioPacket> packet);
+
+  base::ThreadChecker thread_checker_;
+
+  base::WeakPtr<AudioPump> pump_;
+
+  scoped_refptr<base::SingleThreadTaskRunner> pump_task_runner_;
+
+  scoped_ptr<AudioCapturer> audio_capturer_;
+
+  scoped_ptr<AudioEncoder> audio_encoder_;
+
+  bool enabled_;
+
+  DISALLOW_COPY_AND_ASSIGN(Core);
+};
+
+AudioPump::Core::Core(
+    base::WeakPtr<AudioPump> pump,
+    scoped_ptr<AudioCapturer> audio_capturer,
+    scoped_ptr<AudioEncoder> audio_encoder)
+    : pump_(pump),
+      pump_task_runner_(base::ThreadTaskRunnerHandle::Get()),
+      audio_capturer_(audio_capturer.Pass()),
+      audio_encoder_(audio_encoder.Pass()),
+      enabled_(true) {
+  thread_checker_.DetachFromThread();
+}
+
+AudioPump::Core::~Core() {
+  DCHECK(thread_checker_.CalledOnValidThread());
+}
+
+void AudioPump::Core::Start() {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
+  audio_capturer_->Start(
+      base::Bind(&Core::EncodeAudioPacket, base::Unretained(this)));
+}
+
+void AudioPump::Core::Pause(bool pause) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
+  enabled_ = !pause;
+}
+
+void AudioPump::Core::EncodeAudioPacket(scoped_ptr<AudioPacket> packet) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(packet);
+
+  if (!enabled_)
+    return;
+
+  scoped_ptr<AudioPacket> encoded_packet =
+      audio_encoder_->Encode(packet.Pass());
+
+  // The audio encoder returns a null audio packet if there's no audio to send.
+  if (!encoded_packet)
+    return;
+
+  pump_task_runner_->PostTask(FROM_HERE,
+                              base::Bind(&AudioPump::SendAudioPacket, pump_,
+                                         base::Passed(&encoded_packet)));
+}
+
+AudioPump::AudioPump(
+    scoped_refptr<base::SingleThreadTaskRunner> audio_task_runner,
+    scoped_ptr<AudioCapturer> audio_capturer,
+    scoped_ptr<AudioEncoder> audio_encoder,
+    protocol::AudioStub* audio_stub)
+    : audio_task_runner_(audio_task_runner),
+      audio_stub_(audio_stub),
+      weak_factory_(this) {
+  DCHECK(audio_stub_);
+
+  core_.reset(new Core(weak_factory_.GetWeakPtr(), audio_capturer.Pass(),
+                       audio_encoder.Pass()));
+
+  audio_task_runner_->PostTask(
+      FROM_HERE, base::Bind(&Core::Start, base::Unretained(core_.get())));
+}
+
+AudioPump::~AudioPump() {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
+  audio_task_runner_->DeleteSoon(FROM_HERE, core_.release());
+}
+
+void AudioPump::Pause(bool pause) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
+  audio_task_runner_->PostTask(
+      FROM_HERE,
+      base::Bind(&Core::Pause, base::Unretained(core_.get()), pause));
+}
+
+void AudioPump::SendAudioPacket(scoped_ptr<AudioPacket> packet) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(packet);
+
+  audio_stub_->ProcessAudioPacket(packet.Pass(), base::Closure());
+}
+
+}  // namespace remoting