1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "content/browser/media/capture/desktop_capturer_ash.h"
#include "content/browser/media/capture/desktop_frame_skia.h"
#include "third_party/skia/include/core/SkBitmap.h"
#include "third_party/webrtc/modules/desktop_capture/desktop_frame.h"
#include "ui/aura/env.h"
#include "ui/aura/window.h"
#include "ui/aura/window_tree_host.h"
#include "ui/display/display.h"
#include "ui/display/screen.h"
#include "ui/gfx/geometry/rect.h"
#include "ui/gfx/image/image.h"
#include "ui/snapshot/snapshot_aura.h"
namespace content {
namespace {
aura::Window* FindRootWindowForDisplayId(int64_t display_id) {
for (auto& window_tree_host : aura::Env::GetInstance()->window_tree_hosts()) {
auto* root_window = window_tree_host->window();
auto display =
display::Screen::GetScreen()->GetDisplayNearestWindow(root_window);
if (display.id() == display_id) {
return root_window;
}
}
return nullptr;
}
} // namespace
DesktopCapturerAsh::DesktopCapturerAsh() = default;
DesktopCapturerAsh::~DesktopCapturerAsh() = default;
bool DesktopCapturerAsh::GetSourceList(SourceList* result) {
for (const auto& display : display::Screen::GetScreen()->GetAllDisplays()) {
result->emplace_back(display.id(), std::string(), display.id());
}
return true;
}
bool DesktopCapturerAsh::SelectSource(SourceId id) {
display_id_ = id;
return true;
}
bool DesktopCapturerAsh::FocusOnSelectedSource() {
return true;
}
void DesktopCapturerAsh::Start(Callback* callback) {
callback_ = callback;
}
void DesktopCapturerAsh::CaptureFrame() {
aura::Window* root_window = nullptr;
if (display_id_) {
root_window = FindRootWindowForDisplayId(*display_id_);
}
if (!root_window) {
root_window = FindRootWindowForDisplayId(
display::Screen::GetScreen()->GetDisplayForNewWindows().id());
}
if (!root_window) {
// No root window to capture was found.
callback_->OnCaptureResult(Result::ERROR_PERMANENT, nullptr);
return;
}
const gfx::Rect bounds(root_window->bounds().size());
ui::GrabWindowSnapshot(
root_window, bounds,
base::BindOnce(&DesktopCapturerAsh::OnGrabWindowSnapsot,
weak_ptr_factory_.GetWeakPtr()));
}
bool DesktopCapturerAsh::IsOccluded(const webrtc::DesktopVector& pos) {
return false;
}
void DesktopCapturerAsh::SetSharedMemoryFactory(
std::unique_ptr<webrtc::SharedMemoryFactory> shared_memory_factory) {}
void DesktopCapturerAsh::SetExcludedWindow(webrtc::WindowId window) {}
void DesktopCapturerAsh::OnGrabWindowSnapsot(gfx::Image snapshot) {
callback_->OnCaptureResult(
Result::SUCCESS,
std::make_unique<content::DesktopFrameSkia>(snapshot.AsBitmap()));
}
} // namespace content
|
