mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-13 05:40:42 +01:00
cec4343fb4
Bug: webrtc:12338 Change-Id: I300ba78fc4423db7030e555d7e51d2cb2246e9a4 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/227162 Reviewed-by: Harald Alvestrand <hta@webrtc.org> Commit-Queue: Artem Titov <titovartem@webrtc.org> Cr-Commit-Position: refs/heads/master@{#34678}
76 lines
2.9 KiB
C++
76 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_
|