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 MODULES_DESKTOP_CAPTURE_FAKE_DESKTOP_CAPTURER_H_
|
|
#define MODULES_DESKTOP_CAPTURE_FAKE_DESKTOP_CAPTURER_H_
|
|
|
|
#include <memory>
|
|
|
|
#include "modules/desktop_capture/desktop_capturer.h"
|
|
#include "modules/desktop_capture/desktop_frame_generator.h"
|
|
#include "modules/desktop_capture/shared_memory.h"
|
|
#include "rtc_base/system/rtc_export.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 RTC_EXPORT 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 // MODULES_DESKTOP_CAPTURE_FAKE_DESKTOP_CAPTURER_H_
|