src/third_party/webrtc/modules/desktop_capture/screen_capturer_helper_unittest.cc

   1 /*
   2  *  Copyright (c) 2013 The WebRTC project authors. All Rights Reserved.
   3  *
   4  *  Use of this source code is governed by a BSD-style license
   5  *  that can be found in the LICENSE file in the root of the source
   6  *  tree. An additional intellectual property rights grant can be found
   7  *  in the file PATENTS.  All contributing project authors may
   8  *  be found in the AUTHORS file in the root of the source tree.
   9  */
  10
  11 #include "webrtc/modules/desktop_capture/screen_capturer_helper.h"
  12
  13 #include "testing/gtest/include/gtest/gtest.h"
  14 #include "webrtc/system_wrappers/interface/scoped_ptr.h"
  15
  16 namespace webrtc {
  17
  18 class ScreenCapturerHelperTest : public testing::Test {
  19  protected:
  20   ScreenCapturerHelper capturer_helper_;
  21 };
  22
  23 TEST_F(ScreenCapturerHelperTest, ClearInvalidRegion) {
  24   DesktopRegion region(DesktopRect::MakeXYWH(1, 2, 3, 4));
  25   capturer_helper_.InvalidateRegion(region);
  26   capturer_helper_.ClearInvalidRegion();
  27   capturer_helper_.TakeInvalidRegion(&region);
  28   EXPECT_TRUE(region.is_empty());
  29 }
  30
  31 TEST_F(ScreenCapturerHelperTest, InvalidateRegion) {
  32   DesktopRegion region;
  33   capturer_helper_.TakeInvalidRegion(&region);
  34   EXPECT_TRUE(region.is_empty());
  35
  36   region.SetRect(DesktopRect::MakeXYWH(1, 2, 3, 4));
  37   capturer_helper_.InvalidateRegion(region);
  38   capturer_helper_.TakeInvalidRegion(&region);
  39   EXPECT_TRUE(DesktopRegion(DesktopRect::MakeXYWH(1, 2, 3, 4)).Equals(region));
  40
  41   capturer_helper_.InvalidateRegion(
  42       DesktopRegion(DesktopRect::MakeXYWH(1, 2, 3, 4)));
  43   capturer_helper_.InvalidateRegion(
  44       DesktopRegion(DesktopRect::MakeXYWH(4, 2, 3, 4)));
  45   capturer_helper_.TakeInvalidRegion(&region);
  46   EXPECT_TRUE(DesktopRegion(DesktopRect::MakeXYWH(1, 2, 6, 4)).Equals(region));
  47 }
  48
  49 TEST_F(ScreenCapturerHelperTest, InvalidateScreen) {
  50   DesktopRegion region;
  51   capturer_helper_.InvalidateScreen(DesktopSize(12, 34));
  52   capturer_helper_.TakeInvalidRegion(&region);
  53   EXPECT_TRUE(DesktopRegion(DesktopRect::MakeWH(12, 34)).Equals(region));
  54 }
  55
  56 TEST_F(ScreenCapturerHelperTest, SizeMostRecent) {
  57   EXPECT_TRUE(capturer_helper_.size_most_recent().is_empty());
  58   capturer_helper_.set_size_most_recent(DesktopSize(12, 34));
  59   EXPECT_TRUE(
  60       DesktopSize(12, 34).equals(capturer_helper_.size_most_recent()));
  61 }
  62
  63 TEST_F(ScreenCapturerHelperTest, SetLogGridSize) {
  64   capturer_helper_.set_size_most_recent(DesktopSize(10, 10));
  65
  66   DesktopRegion region;
  67   capturer_helper_.TakeInvalidRegion(&region);
  68   EXPECT_TRUE(DesktopRegion().Equals(region));
  69
  70   capturer_helper_.InvalidateRegion(
  71       DesktopRegion(DesktopRect::MakeXYWH(7, 7, 1, 1)));
  72   capturer_helper_.TakeInvalidRegion(&region);
  73   EXPECT_TRUE(DesktopRegion(DesktopRect::MakeXYWH(7, 7, 1, 1)).Equals(region));
  74
  75   capturer_helper_.SetLogGridSize(-1);
  76   capturer_helper_.InvalidateRegion(
  77       DesktopRegion(DesktopRect::MakeXYWH(7, 7, 1, 1)));
  78   capturer_helper_.TakeInvalidRegion(&region);
  79   EXPECT_TRUE(DesktopRegion(DesktopRect::MakeXYWH(7, 7, 1, 1)).Equals(region));
  80
  81   capturer_helper_.SetLogGridSize(0);
  82   capturer_helper_.InvalidateRegion(
  83       DesktopRegion(DesktopRect::MakeXYWH(7, 7, 1, 1)));
  84   capturer_helper_.TakeInvalidRegion(&region);
  85   EXPECT_TRUE(DesktopRegion(DesktopRect::MakeXYWH(7, 7, 1, 1)).Equals(region));
  86
  87   capturer_helper_.SetLogGridSize(1);
  88   capturer_helper_.InvalidateRegion(
  89       DesktopRegion(DesktopRect::MakeXYWH(7, 7, 1, 1)));
  90   capturer_helper_.TakeInvalidRegion(&region);
  91
  92   EXPECT_TRUE(DesktopRegion(DesktopRect::MakeXYWH(6, 6, 2, 2)).Equals(region));
  93
  94   capturer_helper_.SetLogGridSize(2);
  95   capturer_helper_.InvalidateRegion(
  96       DesktopRegion(DesktopRect::MakeXYWH(7, 7, 1, 1)));
  97   capturer_helper_.TakeInvalidRegion(&region);
  98   EXPECT_TRUE(DesktopRegion(DesktopRect::MakeXYWH(4, 4, 4, 4)).Equals(region));
  99
 100   capturer_helper_.SetLogGridSize(0);
 101   capturer_helper_.InvalidateRegion(
 102       DesktopRegion(DesktopRect::MakeXYWH(7, 7, 1, 1)));
 103   capturer_helper_.TakeInvalidRegion(&region);
 104   EXPECT_TRUE(DesktopRegion(DesktopRect::MakeXYWH(7, 7, 1, 1)).Equals(region));
 105 }
 106
 107 void TestExpandRegionToGrid(const DesktopRegion& region, int log_grid_size,
 108                             const DesktopRegion& expanded_region_expected) {
 109   DesktopRegion expanded_region1;
 110   ScreenCapturerHelper::ExpandToGrid(region, log_grid_size, &expanded_region1);
 111   EXPECT_TRUE(expanded_region_expected.Equals(expanded_region1));
 112
 113   DesktopRegion expanded_region2;
 114   ScreenCapturerHelper::ExpandToGrid(expanded_region1, log_grid_size,
 115                                      &expanded_region2);
 116   EXPECT_TRUE(expanded_region1.Equals(expanded_region2));
 117 }
 118
 119 void TestExpandRectToGrid(int l, int t, int r, int b, int log_grid_size,
 120                           int lExpanded, int tExpanded,
 121                           int rExpanded, int bExpanded) {
 122   TestExpandRegionToGrid(DesktopRegion(DesktopRect::MakeLTRB(l, t, r, b)),
 123                          log_grid_size,
 124                          DesktopRegion(DesktopRect::MakeLTRB(
 125                              lExpanded, tExpanded, rExpanded, bExpanded)));
 126 }
 127
 128 TEST_F(ScreenCapturerHelperTest, ExpandToGrid) {
 129   const int kLogGridSize = 4;
 130   const int kGridSize = 1 << kLogGridSize;
 131   for (int i = -2; i <= 2; i++) {
 132     int x = i * kGridSize;
 133     for (int j = -2; j <= 2; j++) {
 134       int y = j * kGridSize;
 135       TestExpandRectToGrid(x + 0, y + 0, x + 1, y + 1, kLogGridSize,
 136                            x + 0, y + 0, x + kGridSize, y + kGridSize);
 137       TestExpandRectToGrid(x + 0, y + kGridSize - 1, x + 1, y + kGridSize,
 138                            kLogGridSize,
 139                            x + 0, y + 0, x + kGridSize, y + kGridSize);
 140       TestExpandRectToGrid(x + kGridSize - 1, y + kGridSize - 1,
 141                            x + kGridSize, y + kGridSize, kLogGridSize,
 142                            x + 0, y + 0, x + kGridSize, y + kGridSize);
 143       TestExpandRectToGrid(x + kGridSize - 1, y + 0,
 144                            x + kGridSize, y + 1, kLogGridSize,
 145                            x + 0, y + 0, x + kGridSize, y + kGridSize);
 146       TestExpandRectToGrid(x - 1, y + 0, x + 1, y + 1, kLogGridSize,
 147                            x - kGridSize, y + 0, x + kGridSize, y + kGridSize);
 148       TestExpandRectToGrid(x - 1, y - 1, x + 1, y + 0, kLogGridSize,
 149                            x - kGridSize, y - kGridSize, x + kGridSize, y);
 150       TestExpandRectToGrid(x + 0, y - 1, x + 1, y + 1, kLogGridSize,
 151                            x, y - kGridSize, x + kGridSize, y + kGridSize);
 152       TestExpandRectToGrid(x - 1, y - 1, x + 0, y + 1, kLogGridSize,
 153                            x - kGridSize, y - kGridSize, x, y + kGridSize);
 154
 155       // Construct a region consisting of 3 pixels and verify that it's expanded
 156       // properly to 3 squares that are kGridSize by kGridSize.
 157       for (int q = 0; q < 4; ++q) {
 158         DesktopRegion region;
 159         DesktopRegion expanded_region_expected;
 160
 161         if (q != 0) {
 162           region.AddRect(DesktopRect::MakeXYWH(x - 1, y - 1, 1, 1));
 163           expanded_region_expected.AddRect(DesktopRect::MakeXYWH(
 164               x - kGridSize, y - kGridSize, kGridSize, kGridSize));
 165         }
 166         if (q != 1) {
 167           region.AddRect(DesktopRect::MakeXYWH(x, y - 1, 1, 1));
 168           expanded_region_expected.AddRect(DesktopRect::MakeXYWH(
 169               x, y - kGridSize, kGridSize, kGridSize));
 170         }
 171         if (q != 2) {
 172           region.AddRect(DesktopRect::MakeXYWH(x - 1, y, 1, 1));
 173           expanded_region_expected.AddRect(DesktopRect::MakeXYWH(
 174               x - kGridSize, y, kGridSize, kGridSize));
 175         }
 176         if (q != 3) {
 177           region.AddRect(DesktopRect::MakeXYWH(x, y, 1, 1));
 178           expanded_region_expected.AddRect(DesktopRect::MakeXYWH(
 179               x, y, kGridSize, kGridSize));
 180         }
 181
 182         TestExpandRegionToGrid(region, kLogGridSize, expanded_region_expected);
 183       }
 184     }
 185   }
 186 }
 187
 188 }  // namespace webrtc