| OLD | NEW |
| 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "content/browser/media/capture/desktop_capture_device.h" | 5 #include "content/browser/media/capture/desktop_capture_device.h" |
| 6 | 6 |
| 7 #include <string> | 7 #include <string> |
| 8 | 8 |
| 9 #include "base/basictypes.h" | 9 #include "base/basictypes.h" |
| 10 #include "base/synchronization/waitable_event.h" | 10 #include "base/synchronization/waitable_event.h" |
| (...skipping 56 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 67 : webrtc::DesktopFrame( | 67 : webrtc::DesktopFrame( |
| 68 frame->size(), | 68 frame->size(), |
| 69 -frame->stride(), | 69 -frame->stride(), |
| 70 frame->data() + (frame->size().height() - 1) * frame->stride(), | 70 frame->data() + (frame->size().height() - 1) * frame->stride(), |
| 71 frame->shared_memory()), | 71 frame->shared_memory()), |
| 72 original_frame_(frame) { | 72 original_frame_(frame) { |
| 73 set_dpi(frame->dpi()); | 73 set_dpi(frame->dpi()); |
| 74 set_capture_time_ms(frame->capture_time_ms()); | 74 set_capture_time_ms(frame->capture_time_ms()); |
| 75 mutable_updated_region()->Swap(frame->mutable_updated_region()); | 75 mutable_updated_region()->Swap(frame->mutable_updated_region()); |
| 76 } | 76 } |
| 77 virtual ~InvertedDesktopFrame() {} | 77 ~InvertedDesktopFrame() override {} |
| 78 | 78 |
| 79 private: | 79 private: |
| 80 scoped_ptr<webrtc::DesktopFrame> original_frame_; | 80 scoped_ptr<webrtc::DesktopFrame> original_frame_; |
| 81 | 81 |
| 82 DISALLOW_COPY_AND_ASSIGN(InvertedDesktopFrame); | 82 DISALLOW_COPY_AND_ASSIGN(InvertedDesktopFrame); |
| 83 }; | 83 }; |
| 84 | 84 |
| 85 // TODO(sergeyu): Move this to a separate file where it can be reused. | 85 // TODO(sergeyu): Move this to a separate file where it can be reused. |
| 86 class FakeScreenCapturer : public webrtc::ScreenCapturer { | 86 class FakeScreenCapturer : public webrtc::ScreenCapturer { |
| 87 public: | 87 public: |
| 88 FakeScreenCapturer() | 88 FakeScreenCapturer() |
| 89 : callback_(NULL), | 89 : callback_(NULL), |
| 90 frame_index_(0), | 90 frame_index_(0), |
| 91 generate_inverted_frames_(false) { | 91 generate_inverted_frames_(false) { |
| 92 } | 92 } |
| 93 virtual ~FakeScreenCapturer() {} | 93 ~FakeScreenCapturer() override {} |
| 94 | 94 |
| 95 void set_generate_inverted_frames(bool generate_inverted_frames) { | 95 void set_generate_inverted_frames(bool generate_inverted_frames) { |
| 96 generate_inverted_frames_ = generate_inverted_frames; | 96 generate_inverted_frames_ = generate_inverted_frames; |
| 97 } | 97 } |
| 98 | 98 |
| 99 // VideoFrameCapturer interface. | 99 // VideoFrameCapturer interface. |
| 100 virtual void Start(Callback* callback) override { | 100 void Start(Callback* callback) override { callback_ = callback; } |
| 101 callback_ = callback; | |
| 102 } | |
| 103 | 101 |
| 104 virtual void Capture(const webrtc::DesktopRegion& region) override { | 102 void Capture(const webrtc::DesktopRegion& region) override { |
| 105 webrtc::DesktopSize size; | 103 webrtc::DesktopSize size; |
| 106 if (frame_index_ % 2 == 0) { | 104 if (frame_index_ % 2 == 0) { |
| 107 size = webrtc::DesktopSize(kTestFrameWidth1, kTestFrameHeight1); | 105 size = webrtc::DesktopSize(kTestFrameWidth1, kTestFrameHeight1); |
| 108 } else { | 106 } else { |
| 109 size = webrtc::DesktopSize(kTestFrameWidth2, kTestFrameHeight2); | 107 size = webrtc::DesktopSize(kTestFrameWidth2, kTestFrameHeight2); |
| 110 } | 108 } |
| 111 frame_index_++; | 109 frame_index_++; |
| 112 | 110 |
| 113 webrtc::DesktopFrame* frame = new webrtc::BasicDesktopFrame(size); | 111 webrtc::DesktopFrame* frame = new webrtc::BasicDesktopFrame(size); |
| 114 if (generate_inverted_frames_) | 112 if (generate_inverted_frames_) |
| 115 frame = new InvertedDesktopFrame(frame); | 113 frame = new InvertedDesktopFrame(frame); |
| 116 callback_->OnCaptureCompleted(frame); | 114 callback_->OnCaptureCompleted(frame); |
| 117 } | 115 } |
| 118 | 116 |
| 119 virtual void SetMouseShapeObserver( | 117 void SetMouseShapeObserver( |
| 120 MouseShapeObserver* mouse_shape_observer) override { | 118 MouseShapeObserver* mouse_shape_observer) override {} |
| 121 } | |
| 122 | 119 |
| 123 virtual bool GetScreenList(ScreenList* screens) override { | 120 bool GetScreenList(ScreenList* screens) override { return false; } |
| 124 return false; | |
| 125 } | |
| 126 | 121 |
| 127 virtual bool SelectScreen(webrtc::ScreenId id) override { | 122 bool SelectScreen(webrtc::ScreenId id) override { return false; } |
| 128 return false; | |
| 129 } | |
| 130 | 123 |
| 131 private: | 124 private: |
| 132 Callback* callback_; | 125 Callback* callback_; |
| 133 int frame_index_; | 126 int frame_index_; |
| 134 bool generate_inverted_frames_; | 127 bool generate_inverted_frames_; |
| 135 }; | 128 }; |
| 136 | 129 |
| 137 } // namespace | 130 } // namespace |
| 138 | 131 |
| 139 class DesktopCaptureDeviceTest : public testing::Test { | 132 class DesktopCaptureDeviceTest : public testing::Test { |
| (...skipping 124 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 264 | 257 |
| 265 capture_device_->StopAndDeAllocate(); | 258 capture_device_->StopAndDeAllocate(); |
| 266 | 259 |
| 267 EXPECT_EQ(kTestFrameWidth1, format.frame_size.width()); | 260 EXPECT_EQ(kTestFrameWidth1, format.frame_size.width()); |
| 268 EXPECT_EQ(kTestFrameHeight1, format.frame_size.height()); | 261 EXPECT_EQ(kTestFrameHeight1, format.frame_size.height()); |
| 269 EXPECT_EQ(kFrameRate, format.frame_rate); | 262 EXPECT_EQ(kFrameRate, format.frame_rate); |
| 270 EXPECT_EQ(media::PIXEL_FORMAT_ARGB, format.pixel_format); | 263 EXPECT_EQ(media::PIXEL_FORMAT_ARGB, format.pixel_format); |
| 271 } | 264 } |
| 272 | 265 |
| 273 } // namespace content | 266 } // namespace content |
| OLD | NEW |