1 // Copyright (c) 2012 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 "remoting/host/audio_scheduler.h"
6
7 #include "base/bind.h"
8 #include "base/location.h"
9 #include "base/logging.h"
10 #include "base/single_thread_task_runner.h"
11 #include "remoting/codec/audio_encoder.h"
12 #include "remoting/host/audio_capturer.h"
13 #include "remoting/proto/audio.pb.h"
14 #include "remoting/protocol/audio_stub.h"
15
16 namespace remoting {
17
AudioScheduler(scoped_refptr<base::SingleThreadTaskRunner> audio_task_runner,scoped_refptr<base::SingleThreadTaskRunner> network_task_runner,scoped_ptr<AudioCapturer> audio_capturer,scoped_ptr<AudioEncoder> audio_encoder,protocol::AudioStub * audio_stub)18 AudioScheduler::AudioScheduler(
19 scoped_refptr<base::SingleThreadTaskRunner> audio_task_runner,
20 scoped_refptr<base::SingleThreadTaskRunner> network_task_runner,
21 scoped_ptr<AudioCapturer> audio_capturer,
22 scoped_ptr<AudioEncoder> audio_encoder,
23 protocol::AudioStub* audio_stub)
24 : audio_task_runner_(audio_task_runner),
25 network_task_runner_(network_task_runner),
26 audio_capturer_(audio_capturer.Pass()),
27 audio_encoder_(audio_encoder.Pass()),
28 audio_stub_(audio_stub),
29 network_stopped_(false),
30 enabled_(true) {
31 DCHECK(network_task_runner_->BelongsToCurrentThread());
32 DCHECK(audio_capturer_);
33 DCHECK(audio_encoder_);
34 DCHECK(audio_stub_);
35 }
36
Start()37 void AudioScheduler::Start() {
38 DCHECK(network_task_runner_->BelongsToCurrentThread());
39
40 audio_task_runner_->PostTask(
41 FROM_HERE, base::Bind(&AudioScheduler::StartOnAudioThread, this));
42 }
43
Stop()44 void AudioScheduler::Stop() {
45 DCHECK(network_task_runner_->BelongsToCurrentThread());
46 DCHECK(audio_stub_);
47
48 // Clear |audio_stub_| to prevent audio packets being delivered to the client.
49 audio_stub_ = NULL;
50
51 audio_task_runner_->PostTask(
52 FROM_HERE,
53 base::Bind(&AudioScheduler::StopOnAudioThread, this));
54 }
55
~AudioScheduler()56 AudioScheduler::~AudioScheduler() {
57 }
58
StartOnAudioThread()59 void AudioScheduler::StartOnAudioThread() {
60 DCHECK(audio_task_runner_->BelongsToCurrentThread());
61
62 // TODO(kxing): Do something with the return value.
63 audio_capturer_->Start(
64 base::Bind(&AudioScheduler::EncodeAudioPacket, this));
65 }
66
StopOnAudioThread()67 void AudioScheduler::StopOnAudioThread() {
68 DCHECK(audio_task_runner_->BelongsToCurrentThread());
69 audio_capturer_->Stop();
70 }
71
Pause(bool pause)72 void AudioScheduler::Pause(bool pause) {
73 if (!audio_task_runner_->BelongsToCurrentThread()) {
74 audio_task_runner_->PostTask(
75 FROM_HERE, base::Bind(&AudioScheduler::Pause, this, pause));
76 return;
77 }
78
79 enabled_ = !pause;
80 }
81
EncodeAudioPacket(scoped_ptr<AudioPacket> packet)82 void AudioScheduler::EncodeAudioPacket(scoped_ptr<AudioPacket> packet) {
83 DCHECK(audio_task_runner_->BelongsToCurrentThread());
84 DCHECK(packet.get());
85
86 if (!enabled_)
87 return;
88
89 scoped_ptr<AudioPacket> encoded_packet =
90 audio_encoder_->Encode(packet.Pass());
91
92 // The audio encoder returns a NULL audio packet if there's no audio to send.
93 if (encoded_packet.get()) {
94 network_task_runner_->PostTask(
95 FROM_HERE, base::Bind(&AudioScheduler::SendAudioPacket,
96 this, base::Passed(&encoded_packet)));
97 }
98 }
99
SendAudioPacket(scoped_ptr<AudioPacket> packet)100 void AudioScheduler::SendAudioPacket(scoped_ptr<AudioPacket> packet) {
101 DCHECK(network_task_runner_->BelongsToCurrentThread());
102 DCHECK(packet.get());
103
104 if (!audio_stub_)
105 return;
106
107 audio_stub_->ProcessAudioPacket(packet.Pass(), base::Closure());
108 }
109
110 } // namespace remoting
111