15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/host/audio_scheduler.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind.h" 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/location.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/single_thread_task_runner.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/codec/audio_encoder.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/host/audio_capturer.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/proto/audio.pb.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/protocol/audio_stub.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace remoting { 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 18c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)AudioScheduler::AudioScheduler( 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<base::SingleThreadTaskRunner> audio_task_runner, 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<base::SingleThreadTaskRunner> network_task_runner, 212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<AudioCapturer> audio_capturer, 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<AudioEncoder> audio_encoder, 23c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) protocol::AudioStub* audio_stub) 24c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) : audio_task_runner_(audio_task_runner), 25c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) network_task_runner_(network_task_runner), 26c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) audio_capturer_(audio_capturer.Pass()), 27c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) audio_encoder_(audio_encoder.Pass()), 28c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) audio_stub_(audio_stub), 29c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) network_stopped_(false), 30c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) enabled_(true) { 31c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DCHECK(network_task_runner_->BelongsToCurrentThread()); 32c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DCHECK(audio_capturer_); 33c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DCHECK(audio_encoder_); 34c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DCHECK(audio_stub_); 35c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 37c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)void AudioScheduler::Start() { 38c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DCHECK(network_task_runner_->BelongsToCurrentThread()); 392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 40c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) audio_task_runner_->PostTask( 41c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) FROM_HERE, base::Bind(&AudioScheduler::StartOnAudioThread, this)); 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void AudioScheduler::Stop() { 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(network_task_runner_->BelongsToCurrentThread()); 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(audio_stub_); 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Clear |audio_stub_| to prevent audio packets being delivered to the client. 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) audio_stub_ = NULL; 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) audio_task_runner_->PostTask( 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FROM_HERE, 532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Bind(&AudioScheduler::StopOnAudioThread, this)); 542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)AudioScheduler::~AudioScheduler() { 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AudioScheduler::StartOnAudioThread() { 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(audio_task_runner_->BelongsToCurrentThread()); 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(kxing): Do something with the return value. 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) audio_capturer_->Start( 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&AudioScheduler::EncodeAudioPacket, this)); 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void AudioScheduler::StopOnAudioThread() { 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(audio_task_runner_->BelongsToCurrentThread()); 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) audio_capturer_->Stop(); 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void AudioScheduler::Pause(bool pause) { 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!audio_task_runner_->BelongsToCurrentThread()) { 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) audio_task_runner_->PostTask( 752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) FROM_HERE, base::Bind(&AudioScheduler::Pause, this, pause)); 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) enabled_ = !pause; 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AudioScheduler::EncodeAudioPacket(scoped_ptr<AudioPacket> packet) { 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(audio_task_runner_->BelongsToCurrentThread()); 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(packet.get()); 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!enabled_) 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<AudioPacket> encoded_packet = 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) audio_encoder_->Encode(packet.Pass()); 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The audio encoder returns a NULL audio packet if there's no audio to send. 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (encoded_packet.get()) { 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) network_task_runner_->PostTask( 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FROM_HERE, base::Bind(&AudioScheduler::SendAudioPacket, 962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) this, base::Passed(&encoded_packet))); 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AudioScheduler::SendAudioPacket(scoped_ptr<AudioPacket> packet) { 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(network_task_runner_->BelongsToCurrentThread()); 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(packet.get()); 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!audio_stub_) 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) audio_stub_->ProcessAudioPacket(packet.Pass(), base::Closure()); 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace remoting 111