Index: remoting/host/video_scheduler.cc |
diff --git a/remoting/host/video_scheduler.cc b/remoting/host/video_scheduler.cc |
deleted file mode 100644 |
index ea57e8a3cad7e8f0956891cd01b8081a10788486..0000000000000000000000000000000000000000 |
--- a/remoting/host/video_scheduler.cc |
+++ /dev/null |
@@ -1,310 +0,0 @@ |
-// Copyright (c) 2012 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/video_scheduler.h" |
- |
-#include <algorithm> |
- |
-#include "base/bind.h" |
-#include "base/callback.h" |
-#include "base/logging.h" |
-#include "base/memory/scoped_ptr.h" |
-#include "base/message_loop/message_loop_proxy.h" |
-#include "base/task_runner_util.h" |
-#include "base/time/time.h" |
-#include "remoting/host/capture_scheduler.h" |
-#include "remoting/proto/control.pb.h" |
-#include "remoting/proto/internal.pb.h" |
-#include "remoting/proto/video.pb.h" |
-#include "remoting/protocol/cursor_shape_stub.h" |
-#include "remoting/protocol/video_stub.h" |
-#include "third_party/webrtc/modules/desktop_capture/desktop_capturer.h" |
-#include "third_party/webrtc/modules/desktop_capture/desktop_frame.h" |
-#include "third_party/webrtc/modules/desktop_capture/mouse_cursor.h" |
- |
-namespace remoting { |
- |
-namespace { |
- |
-// Helper used to encode frames on the encode thread. |
-// |
-// TODO(sergeyu): This functions doesn't do much beside calling |
-// VideoEncoder::Encode(). It's only needed to handle empty frames properly and |
-// that logic can be moved to VideoEncoder implementations. |
-scoped_ptr<VideoPacket> EncodeFrame(VideoEncoder* encoder, |
- scoped_ptr<webrtc::DesktopFrame> frame) { |
- // If there is nothing to encode then send an empty packet. |
- if (!frame || frame->updated_region().is_empty()) |
- return make_scoped_ptr(new VideoPacket()); |
- |
- return encoder->Encode(*frame); |
-} |
- |
-} // namespace |
- |
-// Interval between empty keep-alive frames. These frames are sent only when the |
-// stream is paused or inactive for some other reason (e.g. when blocked on |
-// capturer). To prevent PseudoTCP from resetting congestion window this value |
-// must be smaller than the minimum RTO used in PseudoTCP, which is 250ms. |
-static const int kKeepAlivePacketIntervalMs = 200; |
- |
-static bool g_enable_timestamps = false; |
- |
-// static |
-void VideoScheduler::EnableTimestampsForTests() { |
- g_enable_timestamps = true; |
-} |
- |
-VideoScheduler::VideoScheduler( |
- scoped_refptr<base::SingleThreadTaskRunner> capture_task_runner, |
- scoped_refptr<base::SingleThreadTaskRunner> encode_task_runner, |
- scoped_refptr<base::SingleThreadTaskRunner> network_task_runner, |
- scoped_ptr<webrtc::DesktopCapturer> capturer, |
- scoped_ptr<webrtc::MouseCursorMonitor> mouse_cursor_monitor, |
- scoped_ptr<VideoEncoder> encoder, |
- protocol::CursorShapeStub* cursor_stub, |
- protocol::VideoStub* video_stub) |
- : capture_task_runner_(capture_task_runner), |
- encode_task_runner_(encode_task_runner), |
- network_task_runner_(network_task_runner), |
- capturer_(capturer.Pass()), |
- mouse_cursor_monitor_(mouse_cursor_monitor.Pass()), |
- encoder_(encoder.Pass()), |
- cursor_stub_(cursor_stub), |
- video_stub_(video_stub), |
- latest_event_timestamp_(0) { |
- DCHECK(network_task_runner_->BelongsToCurrentThread()); |
- DCHECK(capturer_); |
- DCHECK(mouse_cursor_monitor_); |
- DCHECK(encoder_); |
- DCHECK(cursor_stub_); |
- DCHECK(video_stub_); |
-} |
- |
-// Public methods -------------------------------------------------------------- |
- |
-void VideoScheduler::Start() { |
- DCHECK(network_task_runner_->BelongsToCurrentThread()); |
- |
- keep_alive_timer_.reset(new base::DelayTimer<VideoScheduler>( |
- FROM_HERE, base::TimeDelta::FromMilliseconds(kKeepAlivePacketIntervalMs), |
- this, &VideoScheduler::SendKeepAlivePacket)); |
- |
- capture_scheduler_.reset(new CaptureScheduler( |
- base::Bind(&VideoScheduler::CaptureNextFrame, this))); |
- capture_scheduler_->Start(); |
- |
- capture_task_runner_->PostTask( |
- FROM_HERE, base::Bind(&VideoScheduler::StartOnCaptureThread, this)); |
-} |
- |
-void VideoScheduler::Stop() { |
- DCHECK(network_task_runner_->BelongsToCurrentThread()); |
- |
- // Clear stubs to prevent further updates reaching the client. |
- cursor_stub_ = nullptr; |
- video_stub_ = nullptr; |
- |
- capture_scheduler_.reset(); |
- keep_alive_timer_.reset(); |
- |
- capture_task_runner_->PostTask( |
- FROM_HERE, base::Bind(&VideoScheduler::StopOnCaptureThread, this)); |
-} |
- |
-void VideoScheduler::Pause(bool pause) { |
- DCHECK(network_task_runner_->BelongsToCurrentThread()); |
- |
- capture_scheduler_->Pause(pause); |
-} |
- |
-void VideoScheduler::SetLatestEventTimestamp(int64 latest_event_timestamp) { |
- DCHECK(network_task_runner_->BelongsToCurrentThread()); |
- |
- latest_event_timestamp_ = latest_event_timestamp; |
-} |
- |
-void VideoScheduler::SetLosslessEncode(bool want_lossless) { |
- DCHECK(network_task_runner_->BelongsToCurrentThread()); |
- |
- encode_task_runner_->PostTask( |
- FROM_HERE, base::Bind(&VideoEncoder::SetLosslessEncode, |
- base::Unretained(encoder_.get()), want_lossless)); |
-} |
- |
-void VideoScheduler::SetLosslessColor(bool want_lossless) { |
- DCHECK(network_task_runner_->BelongsToCurrentThread()); |
- |
- encode_task_runner_->PostTask( |
- FROM_HERE, base::Bind(&VideoEncoder::SetLosslessColor, |
- base::Unretained(encoder_.get()), want_lossless)); |
-} |
- |
-// Private methods ----------------------------------------------------------- |
- |
-VideoScheduler::~VideoScheduler() { |
- // Destroy the capturer and encoder on their respective threads. |
- capture_task_runner_->DeleteSoon(FROM_HERE, capturer_.release()); |
- capture_task_runner_->DeleteSoon(FROM_HERE, mouse_cursor_monitor_.release()); |
- encode_task_runner_->DeleteSoon(FROM_HERE, encoder_.release()); |
-} |
- |
-// Capturer thread ------------------------------------------------------------- |
- |
-webrtc::SharedMemory* VideoScheduler::CreateSharedMemory(size_t size) { |
- return nullptr; |
-} |
- |
-void VideoScheduler::OnCaptureCompleted(webrtc::DesktopFrame* frame) { |
- DCHECK(capture_task_runner_->BelongsToCurrentThread()); |
- |
- network_task_runner_->PostTask( |
- FROM_HERE, base::Bind(&VideoScheduler::EncodeAndSendFrame, this, |
- base::Passed(make_scoped_ptr(frame)))); |
-} |
- |
-void VideoScheduler::OnMouseCursor(webrtc::MouseCursor* cursor) { |
- DCHECK(capture_task_runner_->BelongsToCurrentThread()); |
- |
- scoped_ptr<webrtc::MouseCursor> owned_cursor(cursor); |
- |
- scoped_ptr<protocol::CursorShapeInfo> cursor_proto( |
- new protocol::CursorShapeInfo()); |
- cursor_proto->set_width(cursor->image()->size().width()); |
- cursor_proto->set_height(cursor->image()->size().height()); |
- cursor_proto->set_hotspot_x(cursor->hotspot().x()); |
- cursor_proto->set_hotspot_y(cursor->hotspot().y()); |
- |
- cursor_proto->set_data(std::string()); |
- uint8_t* current_row = cursor->image()->data(); |
- for (int y = 0; y < cursor->image()->size().height(); ++y) { |
- cursor_proto->mutable_data()->append( |
- current_row, |
- current_row + cursor->image()->size().width() * |
- webrtc::DesktopFrame::kBytesPerPixel); |
- current_row += cursor->image()->stride(); |
- } |
- |
- network_task_runner_->PostTask( |
- FROM_HERE, base::Bind(&VideoScheduler::SendCursorShape, this, |
- base::Passed(&cursor_proto))); |
-} |
- |
-void VideoScheduler::OnMouseCursorPosition( |
- webrtc::MouseCursorMonitor::CursorState state, |
- const webrtc::DesktopVector& position) { |
- // We're not subscribing to mouse position changes. |
- NOTREACHED(); |
-} |
- |
-void VideoScheduler::StartOnCaptureThread() { |
- DCHECK(capture_task_runner_->BelongsToCurrentThread()); |
- |
- mouse_cursor_monitor_->Init(this, webrtc::MouseCursorMonitor::SHAPE_ONLY); |
- capturer_->Start(this); |
-} |
- |
-void VideoScheduler::StopOnCaptureThread() { |
- DCHECK(capture_task_runner_->BelongsToCurrentThread()); |
- |
- // This doesn't deleted already captured frames, so encoder can keep using the |
- // frames that were captured previously. |
- capturer_.reset(); |
- |
- mouse_cursor_monitor_.reset(); |
-} |
- |
-void VideoScheduler::CaptureNextFrameOnCaptureThread() { |
- DCHECK(capture_task_runner_->BelongsToCurrentThread()); |
- |
- // Capture mouse shape first and then screen content. |
- mouse_cursor_monitor_->Capture(); |
- capturer_->Capture(webrtc::DesktopRegion()); |
-} |
- |
-// Network thread -------------------------------------------------------------- |
- |
-void VideoScheduler::CaptureNextFrame() { |
- DCHECK(network_task_runner_->BelongsToCurrentThread()); |
- |
- capture_task_runner_->PostTask( |
- FROM_HERE, |
- base::Bind(&VideoScheduler::CaptureNextFrameOnCaptureThread, this)); |
-} |
- |
-void VideoScheduler::EncodeAndSendFrame( |
- scoped_ptr<webrtc::DesktopFrame> frame) { |
- DCHECK(network_task_runner_->BelongsToCurrentThread()); |
- |
- if (!video_stub_) |
- return; |
- |
- capture_scheduler_->OnCaptureCompleted(); |
- |
- // Even when |frame| is nullptr we still need to post it to the encode thread |
- // to make sure frames are freed in the same order they are received and |
- // that we don't start capturing frame n+2 before frame n is freed. |
- base::PostTaskAndReplyWithResult( |
- encode_task_runner_.get(), FROM_HERE, |
- base::Bind(&EncodeFrame, encoder_.get(), base::Passed(&frame)), |
- base::Bind(&VideoScheduler::SendEncodedFrame, this, |
- latest_event_timestamp_, base::TimeTicks::Now())); |
-} |
- |
-void VideoScheduler::SendEncodedFrame(int64 latest_event_timestamp, |
- base::TimeTicks timestamp, |
- scoped_ptr<VideoPacket> packet) { |
- DCHECK(network_task_runner_->BelongsToCurrentThread()); |
- |
- if (!video_stub_) |
- return; |
- |
- if (g_enable_timestamps) |
- packet->set_timestamp(timestamp.ToInternalValue()); |
- |
- packet->set_latest_event_timestamp(latest_event_timestamp); |
- |
- capture_scheduler_->OnFrameEncoded( |
- base::TimeDelta::FromMilliseconds(packet->encode_time_ms())); |
- |
- video_stub_->ProcessVideoPacket( |
- packet.Pass(), base::Bind(&VideoScheduler::OnVideoPacketSent, this)); |
-} |
- |
-void VideoScheduler::OnVideoPacketSent() { |
- DCHECK(network_task_runner_->BelongsToCurrentThread()); |
- |
- if (!video_stub_) |
- return; |
- |
- capture_scheduler_->OnFrameSent(); |
- keep_alive_timer_->Reset(); |
-} |
- |
-void VideoScheduler::SendKeepAlivePacket() { |
- DCHECK(network_task_runner_->BelongsToCurrentThread()); |
- |
- video_stub_->ProcessVideoPacket( |
- make_scoped_ptr(new VideoPacket()), |
- base::Bind(&VideoScheduler::OnKeepAlivePacketSent, this)); |
-} |
- |
-void VideoScheduler::OnKeepAlivePacketSent() { |
- DCHECK(network_task_runner_->BelongsToCurrentThread()); |
- |
- if (keep_alive_timer_) |
- keep_alive_timer_->Reset(); |
-} |
- |
-void VideoScheduler::SendCursorShape( |
- scoped_ptr<protocol::CursorShapeInfo> cursor_shape) { |
- DCHECK(network_task_runner_->BelongsToCurrentThread()); |
- |
- if (!cursor_stub_) |
- return; |
- |
- cursor_stub_->SetCursorShape(*cursor_shape); |
-} |
- |
-} // namespace remoting |