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 |