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

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

Issue 2697453002: [DesktopCapturer] FallbackDesktopCapturerWrapper and its tests (Closed)
Patch Set: Resolve review comments 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() = default;
18 : callback_(nullptr),
19 result_(DesktopCapturer::Result::SUCCESS),
20 generator_(nullptr) {}
21
22 FakeDesktopCapturer::~FakeDesktopCapturer() = default; 18 FakeDesktopCapturer::~FakeDesktopCapturer() = default;
23 19
24 void FakeDesktopCapturer::set_result(DesktopCapturer::Result result) { 20 void FakeDesktopCapturer::set_result(DesktopCapturer::Result result) {
25 result_ = result; 21 result_ = result;
26 } 22 }
27 23
24 int FakeDesktopCapturer::num_frames_captured() const {
25 return num_frames_captured_;
26 }
27
28 int FakeDesktopCapturer::num_capture_attempts() const {
29 return num_capture_attempts_;
30 }
31
28 // Uses the |generator| provided as DesktopFrameGenerator, FakeDesktopCapturer 32 // Uses the |generator| provided as DesktopFrameGenerator, FakeDesktopCapturer
29 // does 33 // does
30 // not take the ownership of |generator|. 34 // not take the ownership of |generator|.
31 void FakeDesktopCapturer::set_frame_generator( 35 void FakeDesktopCapturer::set_frame_generator(
32 DesktopFrameGenerator* generator) { 36 DesktopFrameGenerator* generator) {
33 generator_ = generator; 37 generator_ = generator;
34 } 38 }
35 39
36 void FakeDesktopCapturer::Start(DesktopCapturer::Callback* callback) { 40 void FakeDesktopCapturer::Start(DesktopCapturer::Callback* callback) {
37 callback_ = callback; 41 callback_ = callback;
38 } 42 }
39 43
40 void FakeDesktopCapturer::CaptureFrame() { 44 void FakeDesktopCapturer::CaptureFrame() {
45 num_capture_attempts_++;
41 if (generator_) { 46 if (generator_) {
47 if (result_ != DesktopCapturer::Result::SUCCESS) {
48 callback_->OnCaptureResult(result_, nullptr);
49 return;
50 }
51
42 std::unique_ptr<DesktopFrame> frame( 52 std::unique_ptr<DesktopFrame> frame(
43 generator_->GetNextFrame(shared_memory_factory_.get())); 53 generator_->GetNextFrame(shared_memory_factory_.get()));
44 if (frame) { 54 if (frame) {
55 num_frames_captured_++;
45 callback_->OnCaptureResult(result_, std::move(frame)); 56 callback_->OnCaptureResult(result_, std::move(frame));
46 } else { 57 } else {
47 callback_->OnCaptureResult(DesktopCapturer::Result::ERROR_TEMPORARY, 58 callback_->OnCaptureResult(DesktopCapturer::Result::ERROR_TEMPORARY,
48 nullptr); 59 nullptr);
49 } 60 }
50 return; 61 return;
51 } 62 }
52 callback_->OnCaptureResult(DesktopCapturer::Result::ERROR_PERMANENT, nullptr); 63 callback_->OnCaptureResult(DesktopCapturer::Result::ERROR_PERMANENT, nullptr);
53 } 64 }
54 65
55 void FakeDesktopCapturer::SetSharedMemoryFactory( 66 void FakeDesktopCapturer::SetSharedMemoryFactory(
56 std::unique_ptr<SharedMemoryFactory> shared_memory_factory) { 67 std::unique_ptr<SharedMemoryFactory> shared_memory_factory) {
57 shared_memory_factory_ = std::move(shared_memory_factory); 68 shared_memory_factory_ = std::move(shared_memory_factory);
58 } 69 }
59 70
60 bool FakeDesktopCapturer::GetSourceList(DesktopCapturer::SourceList* sources) { 71 bool FakeDesktopCapturer::GetSourceList(DesktopCapturer::SourceList* sources) {
61 sources->push_back({kWindowId, "A-Fake-DesktopCapturer-Window"}); 72 sources->push_back({kWindowId, "A-Fake-DesktopCapturer-Window"});
62 sources->push_back({kScreenId}); 73 sources->push_back({kScreenId});
63 return true; 74 return true;
64 } 75 }
65 76
66 bool FakeDesktopCapturer::SelectSource(DesktopCapturer::SourceId id) { 77 bool FakeDesktopCapturer::SelectSource(DesktopCapturer::SourceId id) {
67 return id == kWindowId || id == kScreenId || id == kFullDesktopScreenId; 78 return id == kWindowId || id == kScreenId || id == kFullDesktopScreenId;
68 } 79 }
69 80
70 } // namespace webrtc 81 } // namespace webrtc
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698