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 <memory> |
| 12 |
11 #include "webrtc/modules/desktop_capture/desktop_and_cursor_composer.h" | 13 #include "webrtc/modules/desktop_capture/desktop_and_cursor_composer.h" |
12 | 14 |
13 #include "testing/gtest/include/gtest/gtest.h" | 15 #include "testing/gtest/include/gtest/gtest.h" |
14 #include "webrtc/base/scoped_ptr.h" | |
15 #include "webrtc/modules/desktop_capture/desktop_capture_options.h" | 16 #include "webrtc/modules/desktop_capture/desktop_capture_options.h" |
16 #include "webrtc/modules/desktop_capture/desktop_frame.h" | 17 #include "webrtc/modules/desktop_capture/desktop_frame.h" |
17 #include "webrtc/modules/desktop_capture/mouse_cursor.h" | 18 #include "webrtc/modules/desktop_capture/mouse_cursor.h" |
18 #include "webrtc/modules/desktop_capture/shared_desktop_frame.h" | 19 #include "webrtc/modules/desktop_capture/shared_desktop_frame.h" |
19 #include "webrtc/modules/desktop_capture/window_capturer.h" | 20 #include "webrtc/modules/desktop_capture/window_capturer.h" |
20 #include "webrtc/system_wrappers/include/logging.h" | 21 #include "webrtc/system_wrappers/include/logging.h" |
21 | 22 |
22 namespace webrtc { | 23 namespace webrtc { |
23 | 24 |
24 namespace { | 25 namespace { |
(...skipping 55 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
80 callback_->OnCaptureCompleted(next_frame_.release()); | 81 callback_->OnCaptureCompleted(next_frame_.release()); |
81 } | 82 } |
82 | 83 |
83 void SetNextFrame(DesktopFrame* next_frame) { | 84 void SetNextFrame(DesktopFrame* next_frame) { |
84 next_frame_.reset(next_frame); | 85 next_frame_.reset(next_frame); |
85 } | 86 } |
86 | 87 |
87 private: | 88 private: |
88 Callback* callback_; | 89 Callback* callback_; |
89 | 90 |
90 rtc::scoped_ptr<DesktopFrame> next_frame_; | 91 std::unique_ptr<DesktopFrame> next_frame_; |
91 }; | 92 }; |
92 | 93 |
93 class FakeMouseMonitor : public MouseCursorMonitor { | 94 class FakeMouseMonitor : public MouseCursorMonitor { |
94 public: | 95 public: |
95 FakeMouseMonitor() : changed_(true) {} | 96 FakeMouseMonitor() : changed_(true) {} |
96 | 97 |
97 void SetState(CursorState state, const DesktopVector& pos) { | 98 void SetState(CursorState state, const DesktopVector& pos) { |
98 state_ = state; | 99 state_ = state; |
99 position_ = pos; | 100 position_ = pos; |
100 } | 101 } |
101 | 102 |
102 void SetHotspot(const DesktopVector& hotspot) { | 103 void SetHotspot(const DesktopVector& hotspot) { |
103 if (!hotspot_.equals(hotspot)) | 104 if (!hotspot_.equals(hotspot)) |
104 changed_ = true; | 105 changed_ = true; |
105 hotspot_ = hotspot; | 106 hotspot_ = hotspot; |
106 } | 107 } |
107 | 108 |
108 void Init(Callback* callback, Mode mode) override { callback_ = callback; } | 109 void Init(Callback* callback, Mode mode) override { callback_ = callback; } |
109 | 110 |
110 void Capture() override { | 111 void Capture() override { |
111 if (changed_) { | 112 if (changed_) { |
112 rtc::scoped_ptr<DesktopFrame> image( | 113 std::unique_ptr<DesktopFrame> image( |
113 new BasicDesktopFrame(DesktopSize(kCursorWidth, kCursorHeight))); | 114 new BasicDesktopFrame(DesktopSize(kCursorWidth, kCursorHeight))); |
114 uint32_t* data = reinterpret_cast<uint32_t*>(image->data()); | 115 uint32_t* data = reinterpret_cast<uint32_t*>(image->data()); |
115 memset(data, 0, image->stride() * kCursorHeight); | 116 memset(data, 0, image->stride() * kCursorHeight); |
116 | 117 |
117 // Set four pixels near the hotspot and leave all other blank. | 118 // Set four pixels near the hotspot and leave all other blank. |
118 for (int y = 0; y < kTestCursorSize; ++y) { | 119 for (int y = 0; y < kTestCursorSize; ++y) { |
119 for (int x = 0; x < kTestCursorSize; ++x) { | 120 for (int x = 0; x < kTestCursorSize; ++x) { |
120 data[(hotspot_.y() + y) * kCursorWidth + (hotspot_.x() + x)] = | 121 data[(hotspot_.y() + y) * kCursorWidth + (hotspot_.x() + x)] = |
121 kTestCursorData[y][x]; | 122 kTestCursorData[y][x]; |
122 } | 123 } |
(...skipping 46 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
169 | 170 |
170 // DesktopCapturer::Callback interface | 171 // DesktopCapturer::Callback interface |
171 void OnCaptureCompleted(DesktopFrame* frame) override { frame_.reset(frame); } | 172 void OnCaptureCompleted(DesktopFrame* frame) override { frame_.reset(frame); } |
172 | 173 |
173 protected: | 174 protected: |
174 // Owned by |blender_|. | 175 // Owned by |blender_|. |
175 FakeScreenCapturer* fake_screen_; | 176 FakeScreenCapturer* fake_screen_; |
176 FakeMouseMonitor* fake_cursor_; | 177 FakeMouseMonitor* fake_cursor_; |
177 | 178 |
178 DesktopAndCursorComposer blender_; | 179 DesktopAndCursorComposer blender_; |
179 rtc::scoped_ptr<DesktopFrame> frame_; | 180 std::unique_ptr<DesktopFrame> frame_; |
180 }; | 181 }; |
181 | 182 |
182 // Verify DesktopAndCursorComposer can handle the case when the screen capturer | 183 // Verify DesktopAndCursorComposer can handle the case when the screen capturer |
183 // fails. | 184 // fails. |
184 TEST_F(DesktopAndCursorComposerTest, Error) { | 185 TEST_F(DesktopAndCursorComposerTest, Error) { |
185 blender_.Start(this); | 186 blender_.Start(this); |
186 | 187 |
187 fake_cursor_->SetHotspot(DesktopVector()); | 188 fake_cursor_->SetHotspot(DesktopVector()); |
188 fake_cursor_->SetState(MouseCursorMonitor::INSIDE, DesktopVector()); | 189 fake_cursor_->SetState(MouseCursorMonitor::INSIDE, DesktopVector()); |
189 fake_screen_->SetNextFrame(NULL); | 190 fake_screen_->SetNextFrame(NULL); |
190 | 191 |
191 blender_.Capture(DesktopRegion()); | 192 blender_.Capture(DesktopRegion()); |
192 | 193 |
193 EXPECT_EQ(frame_, static_cast<DesktopFrame*>(NULL)); | 194 EXPECT_FALSE(frame_); |
194 } | 195 } |
195 | 196 |
196 TEST_F(DesktopAndCursorComposerTest, Blend) { | 197 TEST_F(DesktopAndCursorComposerTest, Blend) { |
197 struct { | 198 struct { |
198 int x, y; | 199 int x, y; |
199 int hotspot_x, hotspot_y; | 200 int hotspot_x, hotspot_y; |
200 bool inside; | 201 bool inside; |
201 } tests[] = { | 202 } tests[] = { |
202 {0, 0, 0, 0, true}, | 203 {0, 0, 0, 0, true}, |
203 {50, 50, 0, 0, true}, | 204 {50, 50, 0, 0, true}, |
(...skipping 17 matching lines...) Expand all Loading... |
221 | 222 |
222 DesktopVector hotspot(tests[i].hotspot_x, tests[i].hotspot_y); | 223 DesktopVector hotspot(tests[i].hotspot_x, tests[i].hotspot_y); |
223 fake_cursor_->SetHotspot(hotspot); | 224 fake_cursor_->SetHotspot(hotspot); |
224 | 225 |
225 MouseCursorMonitor::CursorState state = tests[i].inside | 226 MouseCursorMonitor::CursorState state = tests[i].inside |
226 ? MouseCursorMonitor::INSIDE | 227 ? MouseCursorMonitor::INSIDE |
227 : MouseCursorMonitor::OUTSIDE; | 228 : MouseCursorMonitor::OUTSIDE; |
228 DesktopVector pos(tests[i].x, tests[i].y); | 229 DesktopVector pos(tests[i].x, tests[i].y); |
229 fake_cursor_->SetState(state, pos); | 230 fake_cursor_->SetState(state, pos); |
230 | 231 |
231 rtc::scoped_ptr<SharedDesktopFrame> frame( | 232 std::unique_ptr<SharedDesktopFrame> frame( |
232 SharedDesktopFrame::Wrap(CreateTestFrame())); | 233 SharedDesktopFrame::Wrap(CreateTestFrame())); |
233 fake_screen_->SetNextFrame(frame->Share()); | 234 fake_screen_->SetNextFrame(frame->Share()); |
234 | 235 |
235 blender_.Capture(DesktopRegion()); | 236 blender_.Capture(DesktopRegion()); |
236 | 237 |
237 VerifyFrame(*frame_, state, pos); | 238 VerifyFrame(*frame_, state, pos); |
238 | 239 |
239 // Verify that the cursor is erased before the frame buffer is returned to | 240 // Verify that the cursor is erased before the frame buffer is returned to |
240 // the screen capturer. | 241 // the screen capturer. |
241 frame_.reset(); | 242 frame_.reset(); |
242 VerifyFrame(*frame, MouseCursorMonitor::OUTSIDE, DesktopVector()); | 243 VerifyFrame(*frame, MouseCursorMonitor::OUTSIDE, DesktopVector()); |
243 } | 244 } |
244 } | 245 } |
245 | 246 |
246 } // namespace | 247 } // namespace |
247 | 248 |
248 } // namespace webrtc | 249 } // namespace webrtc |
OLD | NEW |