| Index: webrtc/modules/desktop_capture/desktop_capturer_differ_wrapper_unittest.cc
|
| diff --git a/webrtc/modules/desktop_capture/desktop_capturer_differ_wrapper_unittest.cc b/webrtc/modules/desktop_capture/desktop_capturer_differ_wrapper_unittest.cc
|
| deleted file mode 100644
|
| index 594ad1327441486e7744dd1eb254acfa063adf5e..0000000000000000000000000000000000000000
|
| --- a/webrtc/modules/desktop_capture/desktop_capturer_differ_wrapper_unittest.cc
|
| +++ /dev/null
|
| @@ -1,291 +0,0 @@
|
| -/*
|
| - * 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/desktop_capturer_differ_wrapper.h"
|
| -
|
| -#include <initializer_list>
|
| -#include <memory>
|
| -#include <utility>
|
| -#include <vector>
|
| -
|
| -#include "webrtc/base/random.h"
|
| -#include "webrtc/base/timeutils.h"
|
| -#include "webrtc/modules/desktop_capture/desktop_geometry.h"
|
| -#include "webrtc/modules/desktop_capture/desktop_region.h"
|
| -#include "webrtc/modules/desktop_capture/differ_block.h"
|
| -#include "webrtc/modules/desktop_capture/fake_screen_capturer.h"
|
| -#include "webrtc/modules/desktop_capture/mock_desktop_capturer_callback.h"
|
| -#include "webrtc/system_wrappers/include/cpu_features_wrapper.h"
|
| -#include "webrtc/test/gtest.h"
|
| -#include "webrtc/typedefs.h"
|
| -
|
| -namespace webrtc {
|
| -
|
| -namespace {
|
| -
|
| -// Compares and asserts |frame|.updated_region() equals to |rects|. This
|
| -// function does not care about the order of the |rects| and it does not expect
|
| -// DesktopRegion to return an exact area for each rectangle in |rects|.
|
| -template <template <typename, typename...> class T = std::initializer_list,
|
| - typename... Rect>
|
| -void AssertUpdatedRegionIs(const DesktopFrame& frame,
|
| - const T<DesktopRect, Rect...>& rects) {
|
| - DesktopRegion region;
|
| - for (const auto& rect : rects) {
|
| - region.AddRect(rect);
|
| - }
|
| - ASSERT_TRUE(frame.updated_region().Equals(region));
|
| -}
|
| -
|
| -// Compares and asserts |frame|.updated_region() covers all rectangles in
|
| -// |rects|, but does not cover areas other than a kBlockSize expansion. This
|
| -// function does not care about the order of the |rects|, and it does not expect
|
| -// DesktopRegion to return an exact area of each rectangle in |rects|.
|
| -template <template <typename, typename...> class T = std::initializer_list,
|
| - typename... Rect>
|
| -void AssertUpdatedRegionCovers(const DesktopFrame& frame,
|
| - const T<DesktopRect, Rect...>& rects) {
|
| - DesktopRegion region;
|
| - for (const auto& rect : rects) {
|
| - region.AddRect(rect);
|
| - }
|
| -
|
| - // Intersect of |rects| and |frame|.updated_region() should be |rects|. i.e.
|
| - // |frame|.updated_region() should be a superset of |rects|.
|
| - DesktopRegion intersect(region);
|
| - intersect.IntersectWith(frame.updated_region());
|
| - ASSERT_TRUE(region.Equals(intersect));
|
| -
|
| - // Difference between |rects| and |frame|.updated_region() should not cover
|
| - // areas which have larger than twice of kBlockSize width and height.
|
| - //
|
| - // Explanation of the 'twice' of kBlockSize (indeed kBlockSize * 2 - 2) is
|
| - // following,
|
| - // (Each block in the following grid is a 8 x 8 pixels area. X means the real
|
| - // updated area, m means the updated area marked by
|
| - // DesktopCapturerDifferWrapper.)
|
| - // +---+---+---+---+---+---+---+---+
|
| - // | X | m | m | m | m | m | m | m |
|
| - // +---+---+---+---+---+---+---+---+
|
| - // | m | m | m | m | m | m | m | m |
|
| - // +---+---+---+---+---+---+---+---+
|
| - // | m | m | m | m | m | m | m | m |
|
| - // +---+---+---+---+---+---+---+---+
|
| - // | m | m | m | m | m | m | m | X |
|
| - // +---+---+---+---+---+---+---+---+
|
| - // The top left [0, 0] - [8, 8] and right bottom [56, 24] - [64, 32] blocks of
|
| - // this area are updated. But since DesktopCapturerDifferWrapper compares
|
| - // 32 x 32 blocks by default, this entire area is marked as updated. So the
|
| - // [8, 8] - [56, 32] is expected to be covered in the difference.
|
| - //
|
| - // But if [0, 0] - [8, 8] and [64, 24] - [72, 32] blocks are updated,
|
| - // +---+---+---+---+---+---+---+---+---+---+---+---+
|
| - // | X | m | m | m | | | | | m | m | m | m |
|
| - // +---+---+---+---+---+---+---+---+---+---+---+---+
|
| - // | m | m | m | m | | | | | m | m | m | m |
|
| - // +---+---+---+---+---+---+---+---+---+---+---+---+
|
| - // | m | m | m | m | | | | | m | m | m | m |
|
| - // +---+---+---+---+---+---+---+---+---+---+---+---+
|
| - // | m | m | m | m | | | | | X | m | m | m |
|
| - // +---+---+---+---+---+---+---+---+---+---+---+---+
|
| - // the [8, 8] - [64, 32] is not expected to be covered in the difference. As
|
| - // DesktopCapturerDifferWrapper should only mark [0, 0] - [32, 32] and
|
| - // [64, 0] - [96, 32] as updated.
|
| - DesktopRegion differ(frame.updated_region());
|
| - differ.Subtract(region);
|
| - for (DesktopRegion::Iterator it(differ); !it.IsAtEnd(); it.Advance()) {
|
| - ASSERT_TRUE(it.rect().width() <= kBlockSize * 2 - 2 ||
|
| - it.rect().height() <= kBlockSize * 2 - 2);
|
| - }
|
| -}
|
| -
|
| -// Executes a DesktopCapturerDifferWrapper::Capture() and compares its output
|
| -// DesktopFrame::updated_region() with |updated_region| if |check_result| is
|
| -// true. If |exactly_match| is true, AssertUpdatedRegionIs() will be used,
|
| -// otherwise AssertUpdatedRegionCovers() will be used.
|
| -template <template <typename, typename...> class T = std::initializer_list,
|
| - typename... Rect>
|
| -void ExecuteDifferWrapperCase(BlackWhiteDesktopFramePainter* frame_painter,
|
| - DesktopCapturerDifferWrapper* capturer,
|
| - MockDesktopCapturerCallback* callback,
|
| - const T<DesktopRect, Rect...>& updated_region,
|
| - bool check_result,
|
| - bool exactly_match) {
|
| - EXPECT_CALL(*callback,
|
| - OnCaptureResultPtr(DesktopCapturer::Result::SUCCESS, testing::_))
|
| - .Times(1)
|
| - .WillOnce(testing::Invoke([&updated_region, check_result, exactly_match](
|
| - DesktopCapturer::Result result,
|
| - std::unique_ptr<DesktopFrame>* frame) {
|
| - ASSERT_EQ(result, DesktopCapturer::Result::SUCCESS);
|
| - if (check_result) {
|
| - if (exactly_match) {
|
| - AssertUpdatedRegionIs(**frame, updated_region);
|
| - } else {
|
| - AssertUpdatedRegionCovers(**frame, updated_region);
|
| - }
|
| - }
|
| - }));
|
| - for (const auto& rect : updated_region) {
|
| - frame_painter->updated_region()->AddRect(rect);
|
| - }
|
| - capturer->CaptureFrame();
|
| -}
|
| -
|
| -// Executes a DesktopCapturerDifferWrapper::Capture(), if updated_region() is
|
| -// not set, this function will reset DesktopCapturerDifferWrapper internal
|
| -// DesktopFrame into black.
|
| -void ExecuteCapturer(DesktopCapturerDifferWrapper* capturer,
|
| - MockDesktopCapturerCallback* callback) {
|
| - EXPECT_CALL(*callback,
|
| - OnCaptureResultPtr(DesktopCapturer::Result::SUCCESS, testing::_))
|
| - .Times(1);
|
| - capturer->CaptureFrame();
|
| -}
|
| -
|
| -void ExecuteDifferWrapperTest(bool with_hints,
|
| - bool enlarge_updated_region,
|
| - bool random_updated_region,
|
| - bool check_result) {
|
| - const bool updated_region_should_exactly_match =
|
| - with_hints && !enlarge_updated_region && !random_updated_region;
|
| - BlackWhiteDesktopFramePainter frame_painter;
|
| - PainterDesktopFrameGenerator frame_generator;
|
| - frame_generator.set_desktop_frame_painter(&frame_painter);
|
| - std::unique_ptr<FakeDesktopCapturer<>> fake(new FakeDesktopCapturer<>());
|
| - fake->set_frame_generator(&frame_generator);
|
| - DesktopCapturerDifferWrapper capturer(std::move(fake));
|
| - MockDesktopCapturerCallback callback;
|
| - frame_generator.set_provide_updated_region_hints(with_hints);
|
| - frame_generator.set_enlarge_updated_region(enlarge_updated_region);
|
| - frame_generator.set_add_random_updated_region(random_updated_region);
|
| -
|
| - capturer.Start(&callback);
|
| -
|
| - EXPECT_CALL(callback,
|
| - OnCaptureResultPtr(DesktopCapturer::Result::SUCCESS, testing::_))
|
| - .Times(1)
|
| - .WillOnce(testing::Invoke([](DesktopCapturer::Result result,
|
| - std::unique_ptr<DesktopFrame>* frame) {
|
| - ASSERT_EQ(result, DesktopCapturer::Result::SUCCESS);
|
| - AssertUpdatedRegionIs(**frame,
|
| - {DesktopRect::MakeSize((*frame)->size())});
|
| - }));
|
| - capturer.CaptureFrame();
|
| -
|
| - ExecuteDifferWrapperCase(&frame_painter, &capturer, &callback,
|
| - {DesktopRect::MakeLTRB(100, 100, 200, 200),
|
| - DesktopRect::MakeLTRB(300, 300, 400, 400)},
|
| - check_result, updated_region_should_exactly_match);
|
| - ExecuteCapturer(&capturer, &callback);
|
| -
|
| - ExecuteDifferWrapperCase(
|
| - &frame_painter, &capturer, &callback,
|
| - {DesktopRect::MakeLTRB(0, 0, 40, 40),
|
| - DesktopRect::MakeLTRB(0, frame_generator.size()->height() - 40, 40,
|
| - frame_generator.size()->height()),
|
| - DesktopRect::MakeLTRB(frame_generator.size()->width() - 40, 0,
|
| - frame_generator.size()->width(), 40),
|
| - DesktopRect::MakeLTRB(frame_generator.size()->width() - 40,
|
| - frame_generator.size()->height() - 40,
|
| - frame_generator.size()->width(),
|
| - frame_generator.size()->height())},
|
| - check_result, updated_region_should_exactly_match);
|
| -
|
| - Random random(rtc::TimeMillis());
|
| - // Fuzzing tests.
|
| - for (int i = 0; i < 1000; i++) {
|
| - if (enlarge_updated_region) {
|
| - frame_generator.set_enlarge_range(random.Rand(1, 50));
|
| - }
|
| - frame_generator.size()->set(random.Rand(500, 2000), random.Rand(500, 2000));
|
| - ExecuteCapturer(&capturer, &callback);
|
| - std::vector<DesktopRect> updated_region;
|
| - for (int j = random.Rand(50); j >= 0; j--) {
|
| - // At least a 1 x 1 updated region.
|
| - const int left = random.Rand(0, frame_generator.size()->width() - 2);
|
| - const int top = random.Rand(0, frame_generator.size()->height() - 2);
|
| - const int right = random.Rand(left + 1, frame_generator.size()->width());
|
| - const int bottom = random.Rand(top + 1, frame_generator.size()->height());
|
| - updated_region.push_back(DesktopRect::MakeLTRB(left, top, right, bottom));
|
| - }
|
| - ExecuteDifferWrapperCase(&frame_painter, &capturer, &callback,
|
| - updated_region, check_result,
|
| - updated_region_should_exactly_match);
|
| - }
|
| -}
|
| -
|
| -} // namespace
|
| -
|
| -TEST(DesktopCapturerDifferWrapperTest, CaptureWithoutHints) {
|
| - ExecuteDifferWrapperTest(false, false, false, true);
|
| -}
|
| -
|
| -TEST(DesktopCapturerDifferWrapperTest, CaptureWithHints) {
|
| - ExecuteDifferWrapperTest(true, false, false, true);
|
| -}
|
| -
|
| -TEST(DesktopCapturerDifferWrapperTest, CaptureWithEnlargedHints) {
|
| - ExecuteDifferWrapperTest(true, true, false, true);
|
| -}
|
| -
|
| -TEST(DesktopCapturerDifferWrapperTest, CaptureWithRandomHints) {
|
| - ExecuteDifferWrapperTest(true, false, true, true);
|
| -}
|
| -
|
| -TEST(DesktopCapturerDifferWrapperTest, CaptureWithEnlargedAndRandomHints) {
|
| - ExecuteDifferWrapperTest(true, true, true, true);
|
| -}
|
| -
|
| -// When hints are provided, DesktopCapturerDifferWrapper has a slightly better
|
| -// performance in current configuration, but not so significant. Following is
|
| -// one run result.
|
| -// [ RUN ] DISABLED_CaptureWithoutHintsPerf
|
| -// [ OK ] DISABLED_CaptureWithoutHintsPerf (7118 ms)
|
| -// [ RUN ] DISABLED_CaptureWithHintsPerf
|
| -// [ OK ] DISABLED_CaptureWithHintsPerf (5580 ms)
|
| -// [ RUN ] DISABLED_CaptureWithEnlargedHintsPerf
|
| -// [ OK ] DISABLED_CaptureWithEnlargedHintsPerf (5974 ms)
|
| -// [ RUN ] DISABLED_CaptureWithRandomHintsPerf
|
| -// [ OK ] DISABLED_CaptureWithRandomHintsPerf (6184 ms)
|
| -// [ RUN ] DISABLED_CaptureWithEnlargedAndRandomHintsPerf
|
| -// [ OK ] DISABLED_CaptureWithEnlargedAndRandomHintsPerf (6347 ms)
|
| -TEST(DesktopCapturerDifferWrapperTest, DISABLED_CaptureWithoutHintsPerf) {
|
| - int64_t started = rtc::TimeMillis();
|
| - ExecuteDifferWrapperTest(false, false, false, false);
|
| - ASSERT_LE(rtc::TimeMillis() - started, 15000);
|
| -}
|
| -
|
| -TEST(DesktopCapturerDifferWrapperTest, DISABLED_CaptureWithHintsPerf) {
|
| - int64_t started = rtc::TimeMillis();
|
| - ExecuteDifferWrapperTest(true, false, false, false);
|
| - ASSERT_LE(rtc::TimeMillis() - started, 15000);
|
| -}
|
| -
|
| -TEST(DesktopCapturerDifferWrapperTest, DISABLED_CaptureWithEnlargedHintsPerf) {
|
| - int64_t started = rtc::TimeMillis();
|
| - ExecuteDifferWrapperTest(true, true, false, false);
|
| - ASSERT_LE(rtc::TimeMillis() - started, 15000);
|
| -}
|
| -
|
| -TEST(DesktopCapturerDifferWrapperTest, DISABLED_CaptureWithRandomHintsPerf) {
|
| - int64_t started = rtc::TimeMillis();
|
| - ExecuteDifferWrapperTest(true, false, true, false);
|
| - ASSERT_LE(rtc::TimeMillis() - started, 15000);
|
| -}
|
| -
|
| -TEST(DesktopCapturerDifferWrapperTest,
|
| - DISABLED_CaptureWithEnlargedAndRandomHintsPerf) {
|
| - int64_t started = rtc::TimeMillis();
|
| - ExecuteDifferWrapperTest(true, true, true, false);
|
| - ASSERT_LE(rtc::TimeMillis() - started, 15000);
|
| -}
|
| -
|
| -} // namespace webrtc
|
|
|