Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(471)

Side by Side Diff: webrtc/modules/desktop_capture/fake_desktop_capturer.cc

Issue 2697453002: [DesktopCapturer] FallbackDesktopCapturerWrapper and its tests (Closed)
Patch Set: Created 3 years, 10 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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 "webrtc/modules/desktop_capture/fake_desktop_capturer.h" 11 #include "webrtc/modules/desktop_capture/fake_desktop_capturer.h"
12 12
13 #include <utility> 13 #include <utility>
14 14
15 namespace webrtc { 15 namespace webrtc {
16 16
17 FakeDesktopCapturer::FakeDesktopCapturer() 17 FakeDesktopCapturer::FakeDesktopCapturer()
18 : callback_(nullptr), 18 : callback_(nullptr),
19 result_(DesktopCapturer::Result::SUCCESS), 19 result_(DesktopCapturer::Result::SUCCESS),
20 generator_(nullptr) {} 20 generator_(nullptr) {}
Jamie 2017/02/15 18:43:48 You can remove these initializers since you're now
Hzj_jie 2017/02/15 21:20:36 I definitely forgot to remove these lines after th
21 21
22 FakeDesktopCapturer::~FakeDesktopCapturer() = default; 22 FakeDesktopCapturer::~FakeDesktopCapturer() = default;
23 23
24 void FakeDesktopCapturer::set_result(DesktopCapturer::Result result) { 24 void FakeDesktopCapturer::set_result(DesktopCapturer::Result result) {
25 result_ = result; 25 result_ = result;
26 } 26 }
27 27
28 int FakeDesktopCapturer::frame_captured() const {
29 return frame_captured_;
30 }
31
32 int FakeDesktopCapturer::capture_attempts() const {
33 return capture_attempts_;
34 }
35
28 // Uses the |generator| provided as DesktopFrameGenerator, FakeDesktopCapturer 36 // Uses the |generator| provided as DesktopFrameGenerator, FakeDesktopCapturer
29 // does 37 // does
30 // not take the ownership of |generator|. 38 // not take the ownership of |generator|.
31 void FakeDesktopCapturer::set_frame_generator( 39 void FakeDesktopCapturer::set_frame_generator(
32 DesktopFrameGenerator* generator) { 40 DesktopFrameGenerator* generator) {
33 generator_ = generator; 41 generator_ = generator;
34 } 42 }
35 43
36 void FakeDesktopCapturer::Start(DesktopCapturer::Callback* callback) { 44 void FakeDesktopCapturer::Start(DesktopCapturer::Callback* callback) {
37 callback_ = callback; 45 callback_ = callback;
38 } 46 }
39 47
40 void FakeDesktopCapturer::CaptureFrame() { 48 void FakeDesktopCapturer::CaptureFrame() {
49 capture_attempts_++;
41 if (generator_) { 50 if (generator_) {
51 if (result_ != DesktopCapturer::Result::SUCCESS) {
52 callback_->OnCaptureResult(result_, nullptr);
53 return;
54 }
55
42 std::unique_ptr<DesktopFrame> frame( 56 std::unique_ptr<DesktopFrame> frame(
43 generator_->GetNextFrame(shared_memory_factory_.get())); 57 generator_->GetNextFrame(shared_memory_factory_.get()));
44 if (frame) { 58 if (frame) {
59 frame_captured_++;
45 callback_->OnCaptureResult(result_, std::move(frame)); 60 callback_->OnCaptureResult(result_, std::move(frame));
46 } else { 61 } else {
47 callback_->OnCaptureResult(DesktopCapturer::Result::ERROR_TEMPORARY, 62 callback_->OnCaptureResult(DesktopCapturer::Result::ERROR_TEMPORARY,
48 nullptr); 63 nullptr);
49 } 64 }
50 return; 65 return;
51 } 66 }
52 callback_->OnCaptureResult(DesktopCapturer::Result::ERROR_PERMANENT, nullptr); 67 callback_->OnCaptureResult(DesktopCapturer::Result::ERROR_PERMANENT, nullptr);
53 } 68 }
54 69
55 void FakeDesktopCapturer::SetSharedMemoryFactory( 70 void FakeDesktopCapturer::SetSharedMemoryFactory(
56 std::unique_ptr<SharedMemoryFactory> shared_memory_factory) { 71 std::unique_ptr<SharedMemoryFactory> shared_memory_factory) {
57 shared_memory_factory_ = std::move(shared_memory_factory); 72 shared_memory_factory_ = std::move(shared_memory_factory);
58 } 73 }
59 74
60 bool FakeDesktopCapturer::GetSourceList(DesktopCapturer::SourceList* sources) { 75 bool FakeDesktopCapturer::GetSourceList(DesktopCapturer::SourceList* sources) {
61 sources->push_back({kWindowId, "A-Fake-DesktopCapturer-Window"}); 76 sources->push_back({kWindowId, "A-Fake-DesktopCapturer-Window"});
62 sources->push_back({kScreenId}); 77 sources->push_back({kScreenId});
63 return true; 78 return true;
64 } 79 }
65 80
66 bool FakeDesktopCapturer::SelectSource(DesktopCapturer::SourceId id) { 81 bool FakeDesktopCapturer::SelectSource(DesktopCapturer::SourceId id) {
67 return id == kWindowId || id == kScreenId || id == kFullDesktopScreenId; 82 return id == kWindowId || id == kScreenId || id == kFullDesktopScreenId;
68 } 83 }
69 84
70 } // namespace webrtc 85 } // namespace webrtc
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698