Index: webrtc/modules/desktop_capture/fake_desktop_capturer.cc |
diff --git a/webrtc/modules/desktop_capture/fake_desktop_capturer.cc b/webrtc/modules/desktop_capture/fake_desktop_capturer.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..bd4c37e8b5a86eaf6a0eefd253f6afc8eb1bc19f |
--- /dev/null |
+++ b/webrtc/modules/desktop_capture/fake_desktop_capturer.cc |
@@ -0,0 +1,70 @@ |
+/* |
+ * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. |
+ * |
+ * Use of this source code is governed by a BSD-style license |
+ * that can be found in the LICENSE file in the root of the source |
+ * tree. An additional intellectual property rights grant can be found |
+ * in the file PATENTS. All contributing project authors may |
+ * be found in the AUTHORS file in the root of the source tree. |
+ */ |
+ |
+#include "webrtc/modules/desktop_capture/fake_desktop_capturer.h" |
+ |
+#include <utility> |
+ |
+namespace webrtc { |
+ |
+FakeDesktopCapturer::FakeDesktopCapturer() |
+ : callback_(nullptr), |
+ result_(DesktopCapturer::Result::SUCCESS), |
+ generator_(nullptr) {} |
+ |
+FakeDesktopCapturer::~FakeDesktopCapturer() = default; |
+ |
+void FakeDesktopCapturer::set_result(DesktopCapturer::Result result) { |
+ result_ = result; |
+} |
+ |
+// Uses the |generator| provided as DesktopFrameGenerator, FakeDesktopCapturer |
+// does |
+// not take the ownership of |generator|. |
+void FakeDesktopCapturer::set_frame_generator( |
+ DesktopFrameGenerator* generator) { |
+ generator_ = generator; |
+} |
+ |
+void FakeDesktopCapturer::Start(DesktopCapturer::Callback* callback) { |
+ callback_ = callback; |
+} |
+ |
+void FakeDesktopCapturer::CaptureFrame() { |
+ if (generator_) { |
+ std::unique_ptr<DesktopFrame> frame( |
+ generator_->GetNextFrame(shared_memory_factory_.get())); |
+ if (frame) { |
+ callback_->OnCaptureResult(result_, std::move(frame)); |
+ } else { |
+ callback_->OnCaptureResult(DesktopCapturer::Result::ERROR_TEMPORARY, |
+ nullptr); |
+ } |
+ return; |
+ } |
+ callback_->OnCaptureResult(DesktopCapturer::Result::ERROR_PERMANENT, nullptr); |
+} |
+ |
+void FakeDesktopCapturer::SetSharedMemoryFactory( |
+ std::unique_ptr<SharedMemoryFactory> shared_memory_factory) { |
+ shared_memory_factory_ = std::move(shared_memory_factory); |
+} |
+ |
+bool FakeDesktopCapturer::GetSourceList(DesktopCapturer::SourceList* sources) { |
+ sources->push_back({kWindowId, "A-Fake-DesktopCapturer-Window"}); |
+ sources->push_back({kScreenId}); |
+ return true; |
+} |
+ |
+bool FakeDesktopCapturer::SelectSource(DesktopCapturer::SourceId id) { |
+ return id == kWindowId || id == kScreenId || id == kFullDesktopScreenId; |
+} |
+ |
+} // namespace webrtc |