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 |