OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2016 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 <string.h> | 11 #include <string.h> |
12 | 12 |
13 #include <algorithm> | 13 #include <algorithm> |
14 #include <initializer_list> | 14 #include <initializer_list> |
| 15 #include <iostream> // TODO(zijiehe): Remove once flaky has been resolved. |
15 #include <memory> | 16 #include <memory> |
16 #include <utility> | 17 #include <utility> |
17 | 18 |
| 19 // TODO(zijiehe): Remove once flaky has been resolved. |
| 20 #include "webrtc/base/base64.h" |
18 #include "webrtc/base/checks.h" | 21 #include "webrtc/base/checks.h" |
19 #include "webrtc/base/constructormagic.h" | 22 #include "webrtc/base/constructormagic.h" |
20 #include "webrtc/base/logging.h" | 23 #include "webrtc/base/logging.h" |
21 #include "webrtc/modules/desktop_capture/desktop_capturer.h" | 24 #include "webrtc/modules/desktop_capture/desktop_capturer.h" |
22 #include "webrtc/modules/desktop_capture/desktop_capture_options.h" | 25 #include "webrtc/modules/desktop_capture/desktop_capture_options.h" |
23 #include "webrtc/modules/desktop_capture/desktop_frame.h" | 26 #include "webrtc/modules/desktop_capture/desktop_frame.h" |
24 #include "webrtc/modules/desktop_capture/desktop_region.h" | 27 #include "webrtc/modules/desktop_capture/desktop_region.h" |
25 #include "webrtc/modules/desktop_capture/mock_desktop_capturer_callback.h" | 28 #include "webrtc/modules/desktop_capture/mock_desktop_capturer_callback.h" |
26 #include "webrtc/modules/desktop_capture/rgba_color.h" | 29 #include "webrtc/modules/desktop_capture/rgba_color.h" |
27 #include "webrtc/modules/desktop_capture/screen_drawer.h" | 30 #include "webrtc/modules/desktop_capture/screen_drawer.h" |
(...skipping 81 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
109 // by |c|) its primary colors is |i|, and the other two are 0x7f. So we | 112 // by |c|) its primary colors is |i|, and the other two are 0x7f. So we |
110 // won't draw a black or white rectangle. | 113 // won't draw a black or white rectangle. |
111 for (int c = 0; c < 3; c++) { | 114 for (int c = 0; c < 3; c++) { |
112 // A fixed size rectangle. | 115 // A fixed size rectangle. |
113 for (int i = 0; i < kTestArea - kRectSize; i += 16) { | 116 for (int i = 0; i < kTestArea - kRectSize; i += 16) { |
114 DesktopRect rect = DesktopRect::MakeXYWH(i, i, kRectSize, kRectSize); | 117 DesktopRect rect = DesktopRect::MakeXYWH(i, i, kRectSize, kRectSize); |
115 rect.Translate(drawer->DrawableRegion().top_left()); | 118 rect.Translate(drawer->DrawableRegion().top_left()); |
116 RgbaColor color((c == 0 ? (i & 0xff) : 0x7f), | 119 RgbaColor color((c == 0 ? (i & 0xff) : 0x7f), |
117 (c == 1 ? (i & 0xff) : 0x7f), | 120 (c == 1 ? (i & 0xff) : 0x7f), |
118 (c == 2 ? (i & 0xff) : 0x7f)); | 121 (c == 2 ? (i & 0xff) : 0x7f)); |
119 TestCaptureOneFrame(capturers, drawer.get(), rect, color); | 122 // Fail fast. |
| 123 ASSERT_NO_FATAL_FAILURE( |
| 124 TestCaptureOneFrame(capturers, drawer.get(), rect, color)); |
120 } | 125 } |
121 | 126 |
122 // A variable-size rectangle. | 127 // A variable-size rectangle. |
123 for (int i = 0; i < kTestArea - kRectSize; i += 16) { | 128 for (int i = 0; i < kTestArea - kRectSize; i += 16) { |
124 DesktopRect rect = DesktopRect::MakeXYWH(kRectSize, kRectSize, i, i); | 129 DesktopRect rect = DesktopRect::MakeXYWH(kRectSize, kRectSize, i, i); |
125 rect.Translate(drawer->DrawableRegion().top_left()); | 130 rect.Translate(drawer->DrawableRegion().top_left()); |
126 RgbaColor color((c == 0 ? (i & 0xff) : 0x7f), | 131 RgbaColor color((c == 0 ? (i & 0xff) : 0x7f), |
127 (c == 1 ? (i & 0xff) : 0x7f), | 132 (c == 1 ? (i & 0xff) : 0x7f), |
128 (c == 2 ? (i & 0xff) : 0x7f)); | 133 (c == 2 ? (i & 0xff) : 0x7f)); |
129 TestCaptureOneFrame(capturers, drawer.get(), rect, color); | 134 // Fail fast. |
| 135 ASSERT_NO_FATAL_FAILURE( |
| 136 TestCaptureOneFrame(capturers, drawer.get(), rect, color)); |
130 } | 137 } |
131 } | 138 } |
132 } | 139 } |
133 | 140 |
134 void TestCaptureUpdatedRegion() { | 141 void TestCaptureUpdatedRegion() { |
135 TestCaptureUpdatedRegion({capturer_.get()}); | 142 TestCaptureUpdatedRegion({capturer_.get()}); |
136 } | 143 } |
137 | 144 |
138 #if defined(WEBRTC_WIN) | 145 #if defined(WEBRTC_WIN) |
139 // Enable allow_directx_capturer in DesktopCaptureOptions, but let | 146 // Enable allow_directx_capturer in DesktopCaptureOptions, but let |
(...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
176 RgbaColor color) { | 183 RgbaColor color) { |
177 const int wait_capture_round = 600; | 184 const int wait_capture_round = 600; |
178 drawer->Clear(); | 185 drawer->Clear(); |
179 size_t succeeded_capturers = 0; | 186 size_t succeeded_capturers = 0; |
180 for (int i = 0; i < wait_capture_round; i++) { | 187 for (int i = 0; i < wait_capture_round; i++) { |
181 drawer->DrawRectangle(rect, color); | 188 drawer->DrawRectangle(rect, color); |
182 drawer->WaitForPendingDraws(); | 189 drawer->WaitForPendingDraws(); |
183 for (size_t j = 0; j < capturers.size(); j++) { | 190 for (size_t j = 0; j < capturers.size(); j++) { |
184 if (capturers[j] == nullptr) { | 191 if (capturers[j] == nullptr) { |
185 // DesktopCapturer should return an empty updated_region() if no | 192 // DesktopCapturer should return an empty updated_region() if no |
186 // update detected. So we won't test it again if it has captured | 193 // update detected. So we won't test it again if it has captured the |
187 // the rectangle we drew. | 194 // rectangle we drew. |
188 continue; | 195 continue; |
189 } | 196 } |
190 std::unique_ptr<DesktopFrame> frame = CaptureFrame(capturers[j]); | 197 std::unique_ptr<DesktopFrame> frame = CaptureFrame(capturers[j]); |
191 if (!frame) { | 198 if (!frame) { |
192 // CaptureFrame() has triggered an assertion failure already, we | 199 // CaptureFrame() has triggered an assertion failure already, we only |
193 // only need to return here. | 200 // need to return here. |
194 return; | 201 return; |
195 } | 202 } |
196 | 203 |
197 if (ArePixelsColoredBy( | 204 if (ArePixelsColoredBy( |
198 *frame, rect, color, drawer->MayDrawIncompleteShapes())) { | 205 *frame, rect, color, drawer->MayDrawIncompleteShapes())) { |
199 capturers[j] = nullptr; | 206 capturers[j] = nullptr; |
200 succeeded_capturers++; | 207 succeeded_capturers++; |
201 } | 208 } |
| 209 // The following else if statement is for debugging purpose only, which |
| 210 // should be removed after flaky of ScreenCapturerIntegrationTest has |
| 211 // been resolved. |
| 212 else if (i == wait_capture_round - 1) { |
| 213 std::string result; |
| 214 rtc::Base64::EncodeFromArray(frame->data(), |
| 215 frame->size().height() * frame->stride(), |
| 216 &result); |
| 217 std::cout << frame->size().width() << " x " << frame->size().height() |
| 218 << std::endl; |
| 219 // Split the entire string (can be over 4M) into several lines to |
| 220 // avoid browser from stucking. |
| 221 static const size_t kLineLength = 32768; |
| 222 const char* result_end = result.c_str() + result.length(); |
| 223 for (const char* it = result.c_str(); |
| 224 it < result_end; |
| 225 it += kLineLength) { |
| 226 const size_t max_length = result_end - it; |
| 227 std::cout << std::string(it, std::min(kLineLength, max_length)) |
| 228 << std::endl; |
| 229 } |
| 230 ASSERT_TRUE(false) << "ScreenCapturerIntegrationTest may be flaky. " |
| 231 "Please kindly FYI the broken link to " |
| 232 "zijiehe@chromium.org for investigation. If I " |
| 233 "have not responded as quick as expected, " |
| 234 "disable *all* tests in " |
| 235 "screen_capturer_integration_test.cc to " |
| 236 "unblock other developers."; |
| 237 } |
202 } | 238 } |
203 | 239 |
204 if (succeeded_capturers == capturers.size()) { | 240 if (succeeded_capturers == capturers.size()) { |
205 break; | 241 break; |
206 } | 242 } |
207 } | 243 } |
208 | 244 |
209 ASSERT_EQ(succeeded_capturers, capturers.size()); | 245 ASSERT_EQ(succeeded_capturers, capturers.size()); |
210 } | 246 } |
211 | 247 |
(...skipping 71 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
283 DISABLED_MaybeCaptureUpdatedRegionWithDirectxCapturer) { | 319 DISABLED_MaybeCaptureUpdatedRegionWithDirectxCapturer) { |
284 // Even DirectX capturer is not supported in current system, we should be able | 320 // Even DirectX capturer is not supported in current system, we should be able |
285 // to select a usable capturer. | 321 // to select a usable capturer. |
286 MaybeCreateDirectxCapturer(); | 322 MaybeCreateDirectxCapturer(); |
287 TestCaptureUpdatedRegion(); | 323 TestCaptureUpdatedRegion(); |
288 } | 324 } |
289 | 325 |
290 #endif // defined(WEBRTC_WIN) | 326 #endif // defined(WEBRTC_WIN) |
291 | 327 |
292 } // namespace webrtc | 328 } // namespace webrtc |
OLD | NEW |