| 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 |