blob: bce82ddc25a7b1dca82b3e78c68e95af97f2be86 [file] [log] [blame]
/*
* Copyright (c) 2017 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/blank_detector_desktop_capturer_wrapper.h"
#include <memory>
#include <utility>
#include "webrtc/modules/desktop_capture/desktop_capturer.h"
#include "webrtc/modules/desktop_capture/desktop_frame.h"
#include "webrtc/modules/desktop_capture/desktop_frame_generator.h"
#include "webrtc/modules/desktop_capture/fake_desktop_capturer.h"
#include "webrtc/test/gtest.h"
namespace webrtc {
class BlankDetectorDesktopCapturerWrapperTest
: public testing::Test,
public DesktopCapturer::Callback {
public:
BlankDetectorDesktopCapturerWrapperTest();
~BlankDetectorDesktopCapturerWrapperTest() override;
protected:
void PerfTest(DesktopCapturer* capturer);
const int frame_width_ = 1024;
const int frame_height_ = 768;
std::unique_ptr<BlankDetectorDesktopCapturerWrapper> wrapper_;
DesktopCapturer* capturer_ = nullptr;
BlackWhiteDesktopFramePainter painter_;
int num_frames_captured_ = 0;
DesktopCapturer::Result last_result_ = DesktopCapturer::Result::SUCCESS;
std::unique_ptr<DesktopFrame> last_frame_;
private:
// DesktopCapturer::Callback interface.
void OnCaptureResult(DesktopCapturer::Result result,
std::unique_ptr<DesktopFrame> frame) override;
PainterDesktopFrameGenerator frame_generator_;
};
BlankDetectorDesktopCapturerWrapperTest::
BlankDetectorDesktopCapturerWrapperTest() {
frame_generator_.size()->set(frame_width_, frame_height_);
frame_generator_.set_desktop_frame_painter(&painter_);
std::unique_ptr<DesktopCapturer> capturer(new FakeDesktopCapturer());
FakeDesktopCapturer* fake_capturer =
static_cast<FakeDesktopCapturer*>(capturer.get());
fake_capturer->set_frame_generator(&frame_generator_);
capturer_ = fake_capturer;
wrapper_.reset(new BlankDetectorDesktopCapturerWrapper(
std::move(capturer), RgbaColor(0, 0, 0, 0)));
wrapper_->Start(this);
}
BlankDetectorDesktopCapturerWrapperTest::
~BlankDetectorDesktopCapturerWrapperTest() = default;
void BlankDetectorDesktopCapturerWrapperTest::OnCaptureResult(
DesktopCapturer::Result result,
std::unique_ptr<DesktopFrame> frame) {
last_result_ = result;
last_frame_ = std::move(frame);
num_frames_captured_++;
}
void BlankDetectorDesktopCapturerWrapperTest::PerfTest(
DesktopCapturer* capturer) {
for (int i = 0; i < 10000; i++) {
capturer->CaptureFrame();
ASSERT_EQ(num_frames_captured_, i + 1);
}
}
TEST_F(BlankDetectorDesktopCapturerWrapperTest, ShouldDetectBlankFrame) {
wrapper_->CaptureFrame();
ASSERT_EQ(num_frames_captured_, 1);
ASSERT_EQ(last_result_, DesktopCapturer::Result::ERROR_TEMPORARY);
ASSERT_FALSE(last_frame_);
}
TEST_F(BlankDetectorDesktopCapturerWrapperTest, ShouldPassBlankDetection) {
painter_.updated_region()->AddRect(DesktopRect::MakeXYWH(0, 0, 100, 100));
wrapper_->CaptureFrame();
ASSERT_EQ(num_frames_captured_, 1);
ASSERT_EQ(last_result_, DesktopCapturer::Result::SUCCESS);
ASSERT_TRUE(last_frame_);
painter_.updated_region()->AddRect(
DesktopRect::MakeXYWH(frame_width_ - 100, frame_height_ - 100, 100, 100));
wrapper_->CaptureFrame();
ASSERT_EQ(num_frames_captured_, 2);
ASSERT_EQ(last_result_, DesktopCapturer::Result::SUCCESS);
ASSERT_TRUE(last_frame_);
painter_.updated_region()->AddRect(
DesktopRect::MakeXYWH(0, frame_height_ - 100, 100, 100));
wrapper_->CaptureFrame();
ASSERT_EQ(num_frames_captured_, 3);
ASSERT_EQ(last_result_, DesktopCapturer::Result::SUCCESS);
ASSERT_TRUE(last_frame_);
painter_.updated_region()->AddRect(
DesktopRect::MakeXYWH(frame_width_ - 100, 0, 100, 100));
wrapper_->CaptureFrame();
ASSERT_EQ(num_frames_captured_, 4);
ASSERT_EQ(last_result_, DesktopCapturer::Result::SUCCESS);
ASSERT_TRUE(last_frame_);
painter_.updated_region()->AddRect(DesktopRect::MakeXYWH(
(frame_width_ >> 1) - 50, (frame_height_ >> 1) - 50, 100, 100));
wrapper_->CaptureFrame();
ASSERT_EQ(num_frames_captured_, 5);
ASSERT_EQ(last_result_, DesktopCapturer::Result::SUCCESS);
ASSERT_TRUE(last_frame_);
}
TEST_F(BlankDetectorDesktopCapturerWrapperTest,
ShouldNotCheckAfterANonBlankFrameReceived) {
wrapper_->CaptureFrame();
ASSERT_EQ(num_frames_captured_, 1);
ASSERT_EQ(last_result_, DesktopCapturer::Result::ERROR_TEMPORARY);
ASSERT_FALSE(last_frame_);
painter_.updated_region()->AddRect(
DesktopRect::MakeXYWH(frame_width_ - 100, 0, 100, 100));
wrapper_->CaptureFrame();
ASSERT_EQ(num_frames_captured_, 2);
ASSERT_EQ(last_result_, DesktopCapturer::Result::SUCCESS);
ASSERT_TRUE(last_frame_);
for (int i = 0; i < 100; i++) {
wrapper_->CaptureFrame();
ASSERT_EQ(num_frames_captured_, i + 3);
ASSERT_EQ(last_result_, DesktopCapturer::Result::SUCCESS);
ASSERT_TRUE(last_frame_);
}
}
// There is no perceptible impact by using BlankDetectorDesktopCapturerWrapper.
// i.e. less than 0.2ms per frame.
// [ OK ] DISABLED_Performance (10210 ms)
// [ OK ] DISABLED_PerformanceComparison (8791 ms)
TEST_F(BlankDetectorDesktopCapturerWrapperTest, DISABLED_Performance) {
PerfTest(wrapper_.get());
}
TEST_F(BlankDetectorDesktopCapturerWrapperTest,
DISABLED_PerformanceComparison) {
capturer_->Start(this);
PerfTest(capturer_);
}
} // namespace webrtc