Index: remoting/host/mouse_shape_pump.cc |
diff --git a/remoting/host/mouse_shape_pump.cc b/remoting/host/mouse_shape_pump.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..133771ea2935ef528397d553aa358db8e7ccd8a6 |
--- /dev/null |
+++ b/remoting/host/mouse_shape_pump.cc |
@@ -0,0 +1,140 @@ |
+// 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/mouse_shape_pump.h" |
+ |
+#include "base/bind.h" |
+#include "base/location.h" |
+#include "base/single_thread_task_runner.h" |
+#include "base/thread_task_runner_handle.h" |
+#include "base/timer/timer.h" |
+#include "remoting/proto/control.pb.h" |
+#include "remoting/protocol/cursor_shape_stub.h" |
+#include "third_party/webrtc/modules/desktop_capture/desktop_frame.h" |
+#include "third_party/webrtc/modules/desktop_capture/mouse_cursor.h" |
+#include "third_party/webrtc/modules/desktop_capture/mouse_cursor_monitor.h" |
+ |
+namespace remoting { |
+ |
+// Poll mouse shape 10 times a second. |
+static const int kCursorCaptureIntervalMs = 100; |
+ |
+class MouseShapePump::Core : public webrtc::MouseCursorMonitor::Callback { |
+ public: |
+ Core(base::WeakPtr<MouseShapePump> proxy, |
+ scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner, |
+ scoped_ptr<webrtc::MouseCursorMonitor> mouse_cursor_monitor); |
+ ~Core() override; |
+ |
+ void Start(); |
+ void Capture(); |
+ |
+ private: |
+ // webrtc::MouseCursorMonitor::Callback implementation. |
+ void OnMouseCursor(webrtc::MouseCursor* mouse_cursor) override; |
+ void OnMouseCursorPosition(webrtc::MouseCursorMonitor::CursorState state, |
+ const webrtc::DesktopVector& position) override; |
+ |
+ base::ThreadChecker thread_checker_; |
+ |
+ base::WeakPtr<MouseShapePump> proxy_; |
+ scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner_; |
+ scoped_ptr<webrtc::MouseCursorMonitor> mouse_cursor_monitor_; |
+ |
+ base::Timer capture_timer_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(Core); |
+}; |
+ |
+MouseShapePump::Core::Core( |
+ base::WeakPtr<MouseShapePump> proxy, |
+ scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner, |
+ scoped_ptr<webrtc::MouseCursorMonitor> mouse_cursor_monitor) |
+ : proxy_(proxy), |
+ caller_task_runner_(caller_task_runner), |
+ mouse_cursor_monitor_(mouse_cursor_monitor.Pass()), |
+ capture_timer_(true, true) { |
+ thread_checker_.DetachFromThread(); |
+} |
+ |
+MouseShapePump::Core::~Core() { |
+ DCHECK(thread_checker_.CalledOnValidThread()); |
+} |
+ |
+void MouseShapePump::Core::Start() { |
+ DCHECK(thread_checker_.CalledOnValidThread()); |
+ |
+ mouse_cursor_monitor_->Init(this, webrtc::MouseCursorMonitor::SHAPE_ONLY); |
+ |
+ capture_timer_.Start( |
+ FROM_HERE, base::TimeDelta::FromMilliseconds(kCursorCaptureIntervalMs), |
+ base::Bind(&MouseShapePump::Core::Capture, base::Unretained(this))); |
+} |
+ |
+void MouseShapePump::Core::Capture() { |
+ DCHECK(thread_checker_.CalledOnValidThread()); |
+ |
+ mouse_cursor_monitor_->Capture(); |
+} |
+ |
+void MouseShapePump::Core::OnMouseCursor(webrtc::MouseCursor* cursor) { |
+ DCHECK(thread_checker_.CalledOnValidThread()); |
+ |
+ 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(); |
+ } |
+ |
+ caller_task_runner_->PostTask( |
+ FROM_HERE, base::Bind(&MouseShapePump::OnCursorShape, proxy_, |
+ base::Passed(&cursor_proto))); |
+} |
+ |
+void MouseShapePump::Core::OnMouseCursorPosition( |
+ webrtc::MouseCursorMonitor::CursorState state, |
+ const webrtc::DesktopVector& position) { |
+ // We're not subscribing to mouse position changes. |
+ NOTREACHED(); |
+} |
+ |
+MouseShapePump::MouseShapePump( |
+ scoped_refptr<base::SingleThreadTaskRunner> capture_task_runner, |
+ scoped_ptr<webrtc::MouseCursorMonitor> mouse_cursor_monitor, |
+ protocol::CursorShapeStub* cursor_shape_stub) |
+ : capture_task_runner_(capture_task_runner), |
+ cursor_shape_stub_(cursor_shape_stub), |
+ weak_factory_(this) { |
+ core_.reset(new Core(weak_factory_.GetWeakPtr(), |
+ base::ThreadTaskRunnerHandle::Get(), |
+ mouse_cursor_monitor.Pass())); |
+ capture_task_runner_->PostTask( |
+ FROM_HERE, base::Bind(&Core::Start, base::Unretained(core_.get()))); |
+} |
+ |
+MouseShapePump::~MouseShapePump() { |
+ capture_task_runner_->DeleteSoon(FROM_HERE, core_.release()); |
+} |
+ |
+void MouseShapePump::OnCursorShape( |
+ scoped_ptr<protocol::CursorShapeInfo> cursor) { |
+ DCHECK(thread_checker_.CalledOnValidThread()); |
+ |
+ cursor_shape_stub_->SetCursorShape(*cursor); |
+} |
+ |
+} // namespace remoting |