8fefe9889d
FallbackDesktopCapturerWrapper is a DesktopCapturer implementation, which owns two DesktopCapturer implementations. If the main DesktopCapturer fails, it uses the secondary capturer. The logic is now used in ScreenCapturerWinMagnifier, and it can also be shared in ScreenCapturerWinDirectx to fallback to Gdi capturer on privilege prompt or login screen. BUG=684937 Review-Url: https://codereview.webrtc.org/2697453002 Cr-Commit-Position: refs/heads/master@{#16677}
77 lines
2.9 KiB
C++
77 lines
2.9 KiB
C++
/*
|
|
* 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.
|
|
*/
|
|
|
|
#ifndef WEBRTC_MODULES_DESKTOP_CAPTURE_FAKE_DESKTOP_CAPTURER_H_
|
|
#define WEBRTC_MODULES_DESKTOP_CAPTURE_FAKE_DESKTOP_CAPTURER_H_
|
|
|
|
#include <memory>
|
|
|
|
#include "webrtc/modules/desktop_capture/desktop_capturer.h"
|
|
#include "webrtc/modules/desktop_capture/desktop_capture_types.h"
|
|
#include "webrtc/modules/desktop_capture/desktop_frame_generator.h"
|
|
#include "webrtc/modules/desktop_capture/shared_memory.h"
|
|
|
|
namespace webrtc {
|
|
|
|
// A fake implementation of DesktopCapturer or its derived interfaces to
|
|
// generate DesktopFrame for testing purpose.
|
|
//
|
|
// Consumers can provide a FrameGenerator instance to generate instances of
|
|
// DesktopFrame to return for each Capture() function call.
|
|
// If no FrameGenerator provided, FakeDesktopCapturer will always return a
|
|
// nullptr DesktopFrame.
|
|
//
|
|
// Double buffering is guaranteed by the FrameGenerator. FrameGenerator
|
|
// implements in desktop_frame_generator.h guarantee double buffering, they
|
|
// creates a new instance of DesktopFrame each time.
|
|
class FakeDesktopCapturer : public DesktopCapturer {
|
|
public:
|
|
FakeDesktopCapturer();
|
|
~FakeDesktopCapturer() override;
|
|
|
|
// Decides the result which will be returned in next Capture() callback.
|
|
void set_result(DesktopCapturer::Result result);
|
|
|
|
// Uses the |generator| provided as DesktopFrameGenerator, FakeDesktopCapturer
|
|
// does not take the ownership of |generator|.
|
|
void set_frame_generator(DesktopFrameGenerator* generator);
|
|
|
|
// Count of DesktopFrame(s) have been returned by this instance. This field
|
|
// would never be negative.
|
|
int num_frames_captured() const;
|
|
|
|
// Count of CaptureFrame() calls have been made. This field would never be
|
|
// negative.
|
|
int num_capture_attempts() const;
|
|
|
|
// DesktopCapturer interface
|
|
void Start(DesktopCapturer::Callback* callback) override;
|
|
void CaptureFrame() override;
|
|
void SetSharedMemoryFactory(
|
|
std::unique_ptr<SharedMemoryFactory> shared_memory_factory) override;
|
|
bool GetSourceList(DesktopCapturer::SourceList* sources) override;
|
|
bool SelectSource(DesktopCapturer::SourceId id) override;
|
|
|
|
private:
|
|
static constexpr DesktopCapturer::SourceId kWindowId = 1378277495;
|
|
static constexpr DesktopCapturer::SourceId kScreenId = 1378277496;
|
|
|
|
DesktopCapturer::Callback* callback_ = nullptr;
|
|
std::unique_ptr<SharedMemoryFactory> shared_memory_factory_;
|
|
DesktopCapturer::Result result_ = Result::SUCCESS;
|
|
DesktopFrameGenerator* generator_ = nullptr;
|
|
int num_frames_captured_ = 0;
|
|
int num_capture_attempts_ = 0;
|
|
};
|
|
|
|
} // namespace webrtc
|
|
|
|
#endif // WEBRTC_MODULES_DESKTOP_CAPTURE_FAKE_DESKTOP_CAPTURER_H_
|