mouse_cursor_monitor_proxy.cc 5.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166
  1. // Copyright 2016 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. #include "remoting/host/mouse_cursor_monitor_proxy.h"
  5. #include <memory>
  6. #include "base/bind.h"
  7. #include "base/location.h"
  8. #include "base/logging.h"
  9. #include "base/task/single_thread_task_runner.h"
  10. #include "base/threading/thread_task_runner_handle.h"
  11. #include "build/chromeos_buildflags.h"
  12. #include "third_party/webrtc/modules/desktop_capture/desktop_capture_options.h"
  13. #include "third_party/webrtc/modules/desktop_capture/desktop_frame.h"
  14. #include "third_party/webrtc/modules/desktop_capture/mouse_cursor.h"
  15. #include "third_party/webrtc/modules/desktop_capture/mouse_cursor_monitor.h"
  16. #if BUILDFLAG(IS_CHROMEOS_ASH)
  17. #include "remoting/host/chromeos/mouse_cursor_monitor_aura.h"
  18. #endif
  19. namespace remoting {
  20. class MouseCursorMonitorProxy::Core
  21. : public webrtc::MouseCursorMonitor::Callback {
  22. public:
  23. explicit Core(base::WeakPtr<MouseCursorMonitorProxy> proxy);
  24. Core(const Core&) = delete;
  25. Core& operator=(const Core&) = delete;
  26. ~Core() override;
  27. void CreateMouseCursorMonitor(const webrtc::DesktopCaptureOptions& options);
  28. void Init(webrtc::MouseCursorMonitor::Mode mode);
  29. void Capture();
  30. void SetMouseCursorMonitorForTests(
  31. std::unique_ptr<webrtc::MouseCursorMonitor> mouse_cursor_monitor);
  32. private:
  33. // webrtc::MouseCursorMonitor::Callback implementation.
  34. void OnMouseCursor(webrtc::MouseCursor* mouse_cursor) override;
  35. void OnMouseCursorPosition(const webrtc::DesktopVector& position) override;
  36. base::ThreadChecker thread_checker_;
  37. base::WeakPtr<MouseCursorMonitorProxy> proxy_;
  38. scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner_;
  39. std::unique_ptr<webrtc::MouseCursorMonitor> mouse_cursor_monitor_;
  40. };
  41. MouseCursorMonitorProxy::Core::Core(
  42. base::WeakPtr<MouseCursorMonitorProxy> proxy)
  43. : proxy_(proxy), caller_task_runner_(base::ThreadTaskRunnerHandle::Get()) {
  44. thread_checker_.DetachFromThread();
  45. }
  46. MouseCursorMonitorProxy::Core::~Core() {
  47. DCHECK(thread_checker_.CalledOnValidThread());
  48. }
  49. void MouseCursorMonitorProxy::Core::CreateMouseCursorMonitor(
  50. const webrtc::DesktopCaptureOptions& options) {
  51. DCHECK(thread_checker_.CalledOnValidThread());
  52. #if BUILDFLAG(IS_CHROMEOS_ASH)
  53. mouse_cursor_monitor_ = std::make_unique<MouseCursorMonitorAura>();
  54. #else // BUILDFLAG(IS_CHROMEOS_ASH)
  55. mouse_cursor_monitor_.reset(webrtc::MouseCursorMonitor::CreateForScreen(
  56. options, webrtc::kFullDesktopScreenId));
  57. #endif // BUILDFLAG(IS_CHROMEOS_ASH)
  58. if (!mouse_cursor_monitor_)
  59. LOG(ERROR) << "Failed to initialize MouseCursorMonitor.";
  60. }
  61. void MouseCursorMonitorProxy::Core::Init(
  62. webrtc::MouseCursorMonitor::Mode mode) {
  63. DCHECK(thread_checker_.CalledOnValidThread());
  64. if (mouse_cursor_monitor_)
  65. mouse_cursor_monitor_->Init(this, mode);
  66. }
  67. void MouseCursorMonitorProxy::Core::Capture() {
  68. DCHECK(thread_checker_.CalledOnValidThread());
  69. if (mouse_cursor_monitor_)
  70. mouse_cursor_monitor_->Capture();
  71. }
  72. void MouseCursorMonitorProxy::Core::SetMouseCursorMonitorForTests(
  73. std::unique_ptr<webrtc::MouseCursorMonitor> mouse_cursor_monitor) {
  74. mouse_cursor_monitor_ = std::move(mouse_cursor_monitor);
  75. }
  76. void MouseCursorMonitorProxy::Core::OnMouseCursor(webrtc::MouseCursor* cursor) {
  77. DCHECK(thread_checker_.CalledOnValidThread());
  78. std::unique_ptr<webrtc::MouseCursor> owned_cursor(cursor);
  79. caller_task_runner_->PostTask(
  80. FROM_HERE, base::BindOnce(&MouseCursorMonitorProxy::OnMouseCursor, proxy_,
  81. std::move(owned_cursor)));
  82. }
  83. void MouseCursorMonitorProxy::Core::OnMouseCursorPosition(
  84. const webrtc::DesktopVector& position) {
  85. DCHECK(thread_checker_.CalledOnValidThread());
  86. caller_task_runner_->PostTask(
  87. FROM_HERE, base::BindOnce(&MouseCursorMonitorProxy::OnMouseCursorPosition,
  88. proxy_, position));
  89. }
  90. MouseCursorMonitorProxy::MouseCursorMonitorProxy(
  91. scoped_refptr<base::SingleThreadTaskRunner> capture_task_runner,
  92. const webrtc::DesktopCaptureOptions& options)
  93. : capture_task_runner_(capture_task_runner) {
  94. core_ = std::make_unique<Core>(weak_factory_.GetWeakPtr());
  95. capture_task_runner_->PostTask(
  96. FROM_HERE, base::BindOnce(&Core::CreateMouseCursorMonitor,
  97. base::Unretained(core_.get()), options));
  98. }
  99. MouseCursorMonitorProxy::~MouseCursorMonitorProxy() {
  100. capture_task_runner_->DeleteSoon(FROM_HERE, core_.release());
  101. }
  102. void MouseCursorMonitorProxy::Init(Callback* callback, Mode mode) {
  103. DCHECK(thread_checker_.CalledOnValidThread());
  104. callback_ = callback;
  105. capture_task_runner_->PostTask(
  106. FROM_HERE,
  107. base::BindOnce(&Core::Init, base::Unretained(core_.get()), mode));
  108. }
  109. void MouseCursorMonitorProxy::Capture() {
  110. DCHECK(thread_checker_.CalledOnValidThread());
  111. capture_task_runner_->PostTask(
  112. FROM_HERE, base::BindOnce(&Core::Capture, base::Unretained(core_.get())));
  113. }
  114. void MouseCursorMonitorProxy::SetMouseCursorMonitorForTests(
  115. std::unique_ptr<webrtc::MouseCursorMonitor> mouse_cursor_monitor) {
  116. capture_task_runner_->PostTask(
  117. FROM_HERE, base::BindOnce(&Core::SetMouseCursorMonitorForTests,
  118. base::Unretained(core_.get()),
  119. std::move(mouse_cursor_monitor)));
  120. }
  121. void MouseCursorMonitorProxy::OnMouseCursor(
  122. std::unique_ptr<webrtc::MouseCursor> cursor) {
  123. DCHECK(thread_checker_.CalledOnValidThread());
  124. callback_->OnMouseCursor(cursor.release());
  125. }
  126. void MouseCursorMonitorProxy::OnMouseCursorPosition(
  127. const webrtc::DesktopVector& position) {
  128. DCHECK(thread_checker_.CalledOnValidThread());
  129. callback_->OnMouseCursorPosition(position);
  130. }
  131. } // namespace remoting