OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2013 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2013 The WebRTC project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #include "webrtc/modules/desktop_capture/window_capturer.h" | 11 #include "webrtc/modules/desktop_capture/window_capturer.h" |
12 | 12 |
13 #include <assert.h> | 13 #include <assert.h> |
14 | 14 |
15 #include <memory> | 15 #include <memory> |
16 | 16 |
17 #include "webrtc/base/checks.h" | 17 #include "webrtc/base/checks.h" |
18 #include "webrtc/base/constructormagic.h" | 18 #include "webrtc/base/constructormagic.h" |
19 #include "webrtc/base/win32.h" | 19 #include "webrtc/base/win32.h" |
20 #include "webrtc/modules/desktop_capture/desktop_frame_win.h" | 20 #include "webrtc/modules/desktop_capture/desktop_frame_win.h" |
21 #include "webrtc/modules/desktop_capture/win/window_capture_utils.h" | 21 #include "webrtc/modules/desktop_capture/win/window_capture_utils.h" |
22 #include "webrtc/system_wrappers/include/logging.h" | 22 #include "webrtc/system_wrappers/include/logging.h" |
23 | 23 |
24 namespace webrtc { | 24 namespace webrtc { |
25 | 25 |
26 namespace { | 26 namespace { |
27 | 27 |
28 BOOL CALLBACK WindowsEnumerationHandler(HWND hwnd, LPARAM param) { | 28 BOOL CALLBACK WindowsEnumerationHandler(HWND hwnd, LPARAM param) { |
29 WindowCapturer::WindowList* list = | 29 DesktopCapturer::SourceList* list = |
30 reinterpret_cast<WindowCapturer::WindowList*>(param); | 30 reinterpret_cast<DesktopCapturer::SourceList*>(param); |
31 | 31 |
32 // Skip windows that are invisible, minimized, have no title, or are owned, | 32 // Skip windows that are invisible, minimized, have no title, or are owned, |
33 // unless they have the app window style set. | 33 // unless they have the app window style set. |
34 int len = GetWindowTextLength(hwnd); | 34 int len = GetWindowTextLength(hwnd); |
35 HWND owner = GetWindow(hwnd, GW_OWNER); | 35 HWND owner = GetWindow(hwnd, GW_OWNER); |
36 LONG exstyle = GetWindowLong(hwnd, GWL_EXSTYLE); | 36 LONG exstyle = GetWindowLong(hwnd, GWL_EXSTYLE); |
37 if (len == 0 || IsIconic(hwnd) || !IsWindowVisible(hwnd) || | 37 if (len == 0 || IsIconic(hwnd) || !IsWindowVisible(hwnd) || |
38 (owner && !(exstyle & WS_EX_APPWINDOW))) { | 38 (owner && !(exstyle & WS_EX_APPWINDOW))) { |
39 return TRUE; | 39 return TRUE; |
40 } | 40 } |
(...skipping 14 matching lines...) Expand all Loading... |
55 // Windows 8 introduced a "Modern App" identified by their class name being | 55 // Windows 8 introduced a "Modern App" identified by their class name being |
56 // either ApplicationFrameWindow or windows.UI.Core.coreWindow. The | 56 // either ApplicationFrameWindow or windows.UI.Core.coreWindow. The |
57 // associated windows cannot be captured, so we skip them. | 57 // associated windows cannot be captured, so we skip them. |
58 // http://crbug.com/526883. | 58 // http://crbug.com/526883. |
59 if (rtc::IsWindows8OrLater() && | 59 if (rtc::IsWindows8OrLater() && |
60 (wcscmp(class_name, L"ApplicationFrameWindow") == 0 || | 60 (wcscmp(class_name, L"ApplicationFrameWindow") == 0 || |
61 wcscmp(class_name, L"Windows.UI.Core.CoreWindow") == 0)) { | 61 wcscmp(class_name, L"Windows.UI.Core.CoreWindow") == 0)) { |
62 return TRUE; | 62 return TRUE; |
63 } | 63 } |
64 | 64 |
65 WindowCapturer::Window window; | 65 DesktopCapturer::Source window; |
66 window.id = reinterpret_cast<WindowId>(hwnd); | 66 window.id = reinterpret_cast<WindowId>(hwnd); |
67 | 67 |
68 const size_t kTitleLength = 500; | 68 const size_t kTitleLength = 500; |
69 WCHAR window_title[kTitleLength]; | 69 WCHAR window_title[kTitleLength]; |
70 // Truncate the title if it's longer than kTitleLength. | 70 // Truncate the title if it's longer than kTitleLength. |
71 GetWindowText(hwnd, window_title, kTitleLength); | 71 GetWindowText(hwnd, window_title, kTitleLength); |
72 window.title = rtc::ToUtf8(window_title); | 72 window.title = rtc::ToUtf8(window_title); |
73 | 73 |
74 // Skip windows when we failed to convert the title or it is empty. | 74 // Skip windows when we failed to convert the title or it is empty. |
75 if (window.title.empty()) | 75 if (window.title.empty()) |
76 return TRUE; | 76 return TRUE; |
77 | 77 |
78 list->push_back(window); | 78 list->push_back(window); |
79 | 79 |
80 return TRUE; | 80 return TRUE; |
81 } | 81 } |
82 | 82 |
83 class WindowCapturerWin : public WindowCapturer { | 83 class WindowCapturerWin : public WindowCapturer { |
84 public: | 84 public: |
85 WindowCapturerWin(); | 85 WindowCapturerWin(); |
86 ~WindowCapturerWin() override; | 86 ~WindowCapturerWin() override; |
87 | 87 |
88 // WindowCapturer interface. | |
89 bool GetWindowList(WindowList* windows) override; | |
90 bool SelectWindow(WindowId id) override; | |
91 bool BringSelectedWindowToFront() override; | |
92 | |
93 // DesktopCapturer interface. | 88 // DesktopCapturer interface. |
94 void Start(Callback* callback) override; | 89 void Start(Callback* callback) override; |
95 void CaptureFrame() override; | 90 void CaptureFrame() override; |
| 91 bool GetSourceList(SourceList* sources) override; |
| 92 bool SelectSource(SourceId id) override; |
| 93 bool FocusOnSelectedSource() override; |
96 | 94 |
97 private: | 95 private: |
98 Callback* callback_ = nullptr; | 96 Callback* callback_ = nullptr; |
99 | 97 |
100 // HWND and HDC for the currently selected window or nullptr if window is not | 98 // HWND and HDC for the currently selected window or nullptr if window is not |
101 // selected. | 99 // selected. |
102 HWND window_ = nullptr; | 100 HWND window_ = nullptr; |
103 | 101 |
104 DesktopSize previous_size_; | 102 DesktopSize previous_size_; |
105 | 103 |
106 AeroChecker aero_checker_; | 104 AeroChecker aero_checker_; |
107 | 105 |
108 // This map is used to avoid flickering for the case when SelectWindow() calls | 106 // This map is used to avoid flickering for the case when SelectWindow() calls |
109 // are interleaved with Capture() calls. | 107 // are interleaved with Capture() calls. |
110 std::map<HWND, DesktopSize> window_size_map_; | 108 std::map<HWND, DesktopSize> window_size_map_; |
111 | 109 |
112 RTC_DISALLOW_COPY_AND_ASSIGN(WindowCapturerWin); | 110 RTC_DISALLOW_COPY_AND_ASSIGN(WindowCapturerWin); |
113 }; | 111 }; |
114 | 112 |
115 WindowCapturerWin::WindowCapturerWin() {} | 113 WindowCapturerWin::WindowCapturerWin() {} |
116 WindowCapturerWin::~WindowCapturerWin() {} | 114 WindowCapturerWin::~WindowCapturerWin() {} |
117 | 115 |
118 bool WindowCapturerWin::GetWindowList(WindowList* windows) { | 116 bool WindowCapturerWin::GetSourceList(SourceList* sources) { |
119 WindowList result; | 117 SourceList result; |
120 LPARAM param = reinterpret_cast<LPARAM>(&result); | 118 LPARAM param = reinterpret_cast<LPARAM>(&result); |
121 if (!EnumWindows(&WindowsEnumerationHandler, param)) | 119 if (!EnumWindows(&WindowsEnumerationHandler, param)) |
122 return false; | 120 return false; |
123 windows->swap(result); | 121 sources->swap(result); |
124 | 122 |
125 std::map<HWND, DesktopSize> new_map; | 123 std::map<HWND, DesktopSize> new_map; |
126 for (const auto& item : *windows) { | 124 for (const auto& item : *sources) { |
127 HWND hwnd = reinterpret_cast<HWND>(item.id); | 125 HWND hwnd = reinterpret_cast<HWND>(item.id); |
128 new_map[hwnd] = window_size_map_[hwnd]; | 126 new_map[hwnd] = window_size_map_[hwnd]; |
129 } | 127 } |
130 window_size_map_.swap(new_map); | 128 window_size_map_.swap(new_map); |
131 | 129 |
132 return true; | 130 return true; |
133 } | 131 } |
134 | 132 |
135 bool WindowCapturerWin::SelectWindow(WindowId id) { | 133 bool WindowCapturerWin::SelectSource(SourceId id) { |
136 HWND window = reinterpret_cast<HWND>(id); | 134 HWND window = reinterpret_cast<HWND>(id); |
137 if (!IsWindow(window) || !IsWindowVisible(window) || IsIconic(window)) | 135 if (!IsWindow(window) || !IsWindowVisible(window) || IsIconic(window)) |
138 return false; | 136 return false; |
139 window_ = window; | 137 window_ = window; |
140 // When a window is not in the map, window_size_map_[window] will create an | 138 // When a window is not in the map, window_size_map_[window] will create an |
141 // item with DesktopSize (0, 0). | 139 // item with DesktopSize (0, 0). |
142 previous_size_ = window_size_map_[window]; | 140 previous_size_ = window_size_map_[window]; |
143 return true; | 141 return true; |
144 } | 142 } |
145 | 143 |
146 bool WindowCapturerWin::BringSelectedWindowToFront() { | 144 bool WindowCapturerWin::FocusOnSelectedSource() { |
147 if (!window_) | 145 if (!window_) |
148 return false; | 146 return false; |
149 | 147 |
150 if (!IsWindow(window_) || !IsWindowVisible(window_) || IsIconic(window_)) | 148 if (!IsWindow(window_) || !IsWindowVisible(window_) || IsIconic(window_)) |
151 return false; | 149 return false; |
152 | 150 |
153 return BringWindowToTop(window_) != FALSE && | 151 return BringWindowToTop(window_) != FALSE && |
154 SetForegroundWindow(window_) != FALSE; | 152 SetForegroundWindow(window_) != FALSE; |
155 } | 153 } |
156 | 154 |
(...skipping 113 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
270 return new WindowCapturerWin(); | 268 return new WindowCapturerWin(); |
271 } | 269 } |
272 | 270 |
273 // static | 271 // static |
274 std::unique_ptr<DesktopCapturer> DesktopCapturer::CreateRawWindowCapturer( | 272 std::unique_ptr<DesktopCapturer> DesktopCapturer::CreateRawWindowCapturer( |
275 const DesktopCaptureOptions& options) { | 273 const DesktopCaptureOptions& options) { |
276 return std::unique_ptr<DesktopCapturer>(new WindowCapturerWin()); | 274 return std::unique_ptr<DesktopCapturer>(new WindowCapturerWin()); |
277 } | 275 } |
278 | 276 |
279 } // namespace webrtc | 277 } // namespace webrtc |
OLD | NEW |