// Copyright 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "ui/display/manager/display_change_observer.h"

#include <string>

#include "base/test/scoped_feature_list.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/display/display_switches.h"
#include "ui/display/manager/display_configurator.h"
#include "ui/display/manager/fake_display_snapshot.h"
#include "ui/display/manager/managed_display_info.h"
#include "ui/display/types/display_mode.h"
#include "ui/gfx/geometry/rect.h"
#include "ui/gfx/geometry/size.h"

namespace display {

namespace {

float ComputeDeviceScaleFactor(float diagonal_inch,
                               const gfx::Rect& resolution) {
  // We assume that displays have square pixel.
  float diagonal_pixel = std::sqrt(std::pow(resolution.width(), 2) +
                                   std::pow(resolution.height(), 2));
  float dpi = diagonal_pixel / diagonal_inch;
  return DisplayChangeObserver::FindDeviceScaleFactor(dpi);
}

std::unique_ptr<DisplayMode> MakeDisplayMode(int width,
                                             int height,
                                             bool is_interlaced,
                                             float refresh_rate) {
  return std::make_unique<DisplayMode>(gfx::Size(width, height), is_interlaced,
                                       refresh_rate);
}

}  // namespace

class DisplayChangeObserverTest : public testing::Test {
 public:
  DisplayChangeObserverTest() = default;

  void SetUp() override {
    scoped_feature_list_.InitAndDisableFeature(
        features::kEnableDisplayZoomSetting);
    testing::Test::SetUp();
  }

 private:
  base::test::ScopedFeatureList scoped_feature_list_;

  DISALLOW_COPY_AND_ASSIGN(DisplayChangeObserverTest);
};

TEST_F(DisplayChangeObserverTest, GetExternalManagedDisplayModeList) {
  std::unique_ptr<DisplaySnapshot> display_snapshot =
      FakeDisplaySnapshot::Builder()
          .SetId(123)
          .SetNativeMode(MakeDisplayMode(1920, 1200, false, 60))
          // All non-interlaced (as would be seen with different refresh rates).
          .AddMode(MakeDisplayMode(1920, 1080, false, 80))
          .AddMode(MakeDisplayMode(1920, 1080, false, 70))
          .AddMode(MakeDisplayMode(1920, 1080, false, 60))
          // Interlaced vs non-interlaced.
          .AddMode(MakeDisplayMode(1280, 720, true, 60))
          .AddMode(MakeDisplayMode(1280, 720, false, 60))
          // Interlaced only.
          .AddMode(MakeDisplayMode(1024, 768, true, 70))
          .AddMode(MakeDisplayMode(1024, 768, true, 60))
          // Mixed.
          .AddMode(MakeDisplayMode(1024, 600, true, 60))
          .AddMode(MakeDisplayMode(1024, 600, false, 70))
          .AddMode(MakeDisplayMode(1024, 600, false, 60))
          // Just one interlaced mode.
          .AddMode(MakeDisplayMode(640, 480, true, 60))
          .Build();

  ManagedDisplayInfo::ManagedDisplayModeList display_modes =
      DisplayChangeObserver::GetExternalManagedDisplayModeList(
          *display_snapshot);
  ASSERT_EQ(6u, display_modes.size());
  EXPECT_EQ("640x480", display_modes[0].size().ToString());
  EXPECT_TRUE(display_modes[0].is_interlaced());
  EXPECT_EQ(display_modes[0].refresh_rate(), 60);

  EXPECT_EQ("1024x600", display_modes[1].size().ToString());
  EXPECT_FALSE(display_modes[1].is_interlaced());
  EXPECT_EQ(display_modes[1].refresh_rate(), 70);

  EXPECT_EQ("1024x768", display_modes[2].size().ToString());
  EXPECT_TRUE(display_modes[2].is_interlaced());
  EXPECT_EQ(display_modes[2].refresh_rate(), 70);

  EXPECT_EQ("1280x720", display_modes[3].size().ToString());
  EXPECT_FALSE(display_modes[3].is_interlaced());
  EXPECT_EQ(display_modes[3].refresh_rate(), 60);

  EXPECT_EQ("1920x1080", display_modes[4].size().ToString());
  EXPECT_FALSE(display_modes[4].is_interlaced());
  EXPECT_EQ(display_modes[4].refresh_rate(), 80);

  EXPECT_EQ("1920x1200", display_modes[5].size().ToString());
  EXPECT_FALSE(display_modes[5].is_interlaced());
  EXPECT_EQ(display_modes[5].refresh_rate(), 60);
}

TEST_F(DisplayChangeObserverTest, GetEmptyExternalManagedDisplayModeList) {
  FakeDisplaySnapshot display_snapshot(
      123, gfx::Point(), gfx::Size(), DISPLAY_CONNECTION_TYPE_UNKNOWN, false,
      false, false, std::string(), {}, nullptr, nullptr, 0, gfx::Size());

  ManagedDisplayInfo::ManagedDisplayModeList display_modes =
      DisplayChangeObserver::GetExternalManagedDisplayModeList(
          display_snapshot);
  EXPECT_EQ(0u, display_modes.size());
}

TEST_F(DisplayChangeObserverTest, GetInternalManagedDisplayModeList) {
  std::unique_ptr<DisplaySnapshot> display_snapshot =
      FakeDisplaySnapshot::Builder()
          .SetId(123)
          .SetNativeMode(MakeDisplayMode(1366, 768, false, 60))
          .AddMode(MakeDisplayMode(1024, 768, false, 60))
          .AddMode(MakeDisplayMode(800, 600, false, 60))
          .AddMode(MakeDisplayMode(600, 600, false, 56.2))
          .AddMode(MakeDisplayMode(640, 480, false, 59.9))
          .Build();

  ManagedDisplayInfo info(1, "", false);
  info.SetBounds(gfx::Rect(0, 0, 1366, 768));

  ManagedDisplayInfo::ManagedDisplayModeList display_modes =
      DisplayChangeObserver::GetInternalManagedDisplayModeList(
          info, *display_snapshot);
  ASSERT_EQ(5u, display_modes.size());
  EXPECT_EQ("1366x768", display_modes[0].size().ToString());
  EXPECT_FALSE(display_modes[0].native());
  EXPECT_NEAR(display_modes[0].ui_scale(), 0.5, 0.01);
  EXPECT_EQ(display_modes[0].refresh_rate(), 60);

  EXPECT_EQ("1366x768", display_modes[1].size().ToString());
  EXPECT_FALSE(display_modes[1].native());
  EXPECT_NEAR(display_modes[1].ui_scale(), 0.6, 0.01);
  EXPECT_EQ(display_modes[1].refresh_rate(), 60);

  EXPECT_EQ("1366x768", display_modes[2].size().ToString());
  EXPECT_FALSE(display_modes[2].native());
  EXPECT_NEAR(display_modes[2].ui_scale(), 0.75, 0.01);
  EXPECT_EQ(display_modes[2].refresh_rate(), 60);

  EXPECT_EQ("1366x768", display_modes[3].size().ToString());
  EXPECT_TRUE(display_modes[3].native());
  EXPECT_NEAR(display_modes[3].ui_scale(), 1.0, 0.01);
  EXPECT_EQ(display_modes[3].refresh_rate(), 60);

  EXPECT_EQ("1366x768", display_modes[4].size().ToString());
  EXPECT_FALSE(display_modes[4].native());
  EXPECT_NEAR(display_modes[4].ui_scale(), 1.125, 0.01);
  EXPECT_EQ(display_modes[4].refresh_rate(), 60);
}

TEST_F(DisplayChangeObserverTest, GetInternalHiDPIManagedDisplayModeList) {
  // Data picked from peppy.
  std::unique_ptr<DisplaySnapshot> display_snapshot =
      FakeDisplaySnapshot::Builder()
          .SetId(123)
          .SetNativeMode(MakeDisplayMode(2560, 1700, false, 60))
          .AddMode(MakeDisplayMode(2048, 1536, false, 60))
          .AddMode(MakeDisplayMode(1920, 1440, false, 60))
          .Build();

  ManagedDisplayInfo info(1, "", false);
  info.SetBounds(gfx::Rect(0, 0, 2560, 1700));
  info.set_device_scale_factor(2.0f);

  ManagedDisplayInfo::ManagedDisplayModeList display_modes =
      DisplayChangeObserver::GetInternalManagedDisplayModeList(
          info, *display_snapshot);
  ASSERT_EQ(8u, display_modes.size());
  EXPECT_EQ("2560x1700", display_modes[0].size().ToString());
  EXPECT_FALSE(display_modes[0].native());
  EXPECT_NEAR(display_modes[0].ui_scale(), 0.5, 0.01);
  EXPECT_EQ(display_modes[0].refresh_rate(), 60);

  EXPECT_EQ("2560x1700", display_modes[1].size().ToString());
  EXPECT_FALSE(display_modes[1].native());
  EXPECT_NEAR(display_modes[1].ui_scale(), 0.625, 0.01);
  EXPECT_EQ(display_modes[1].refresh_rate(), 60);

  EXPECT_EQ("2560x1700", display_modes[2].size().ToString());
  EXPECT_FALSE(display_modes[2].native());
  EXPECT_NEAR(display_modes[2].ui_scale(), 0.8, 0.01);
  EXPECT_EQ(display_modes[2].refresh_rate(), 60);

  EXPECT_EQ("2560x1700", display_modes[3].size().ToString());
  EXPECT_FALSE(display_modes[3].native());
  EXPECT_NEAR(display_modes[3].ui_scale(), 1.0, 0.01);
  EXPECT_EQ(display_modes[3].refresh_rate(), 60);

  EXPECT_EQ("2560x1700", display_modes[4].size().ToString());
  EXPECT_FALSE(display_modes[4].native());
  EXPECT_NEAR(display_modes[4].ui_scale(), 1.125, 0.01);
  EXPECT_EQ(display_modes[4].refresh_rate(), 60);

  EXPECT_EQ("2560x1700", display_modes[5].size().ToString());
  EXPECT_FALSE(display_modes[5].native());
  EXPECT_NEAR(display_modes[5].ui_scale(), 1.25, 0.01);
  EXPECT_EQ(display_modes[5].refresh_rate(), 60);

  EXPECT_EQ("2560x1700", display_modes[6].size().ToString());
  EXPECT_FALSE(display_modes[6].native());
  EXPECT_NEAR(display_modes[6].ui_scale(), 1.5, 0.01);
  EXPECT_EQ(display_modes[6].refresh_rate(), 60);

  EXPECT_EQ("2560x1700", display_modes[7].size().ToString());
  EXPECT_TRUE(display_modes[7].native());
  EXPECT_NEAR(display_modes[7].ui_scale(), 2.0, 0.01);
  EXPECT_EQ(display_modes[7].refresh_rate(), 60);
}

TEST_F(DisplayChangeObserverTest, GetInternalManagedDisplayModeList1_25) {
  // Data picked from peppy.
  std::unique_ptr<DisplaySnapshot> display_snapshot =
      FakeDisplaySnapshot::Builder()
          .SetId(123)
          .SetNativeMode(MakeDisplayMode(1920, 1080, false, 60))
          .Build();

  ManagedDisplayInfo info(1, "", false);
  info.SetBounds(gfx::Rect(0, 0, 1920, 1080));
  info.set_device_scale_factor(1.25);

  ManagedDisplayInfo::ManagedDisplayModeList display_modes =
      DisplayChangeObserver::GetInternalManagedDisplayModeList(
          info, *display_snapshot);
  ASSERT_EQ(5u, display_modes.size());
  EXPECT_EQ("1920x1080", display_modes[0].size().ToString());
  EXPECT_FALSE(display_modes[0].native());
  EXPECT_NEAR(display_modes[0].ui_scale(), 0.5, 0.01);
  EXPECT_EQ(display_modes[0].refresh_rate(), 60);

  EXPECT_EQ("1920x1080", display_modes[1].size().ToString());
  EXPECT_FALSE(display_modes[1].native());
  EXPECT_NEAR(display_modes[1].ui_scale(), 0.625, 0.01);
  EXPECT_EQ(display_modes[1].refresh_rate(), 60);

  EXPECT_EQ("1920x1080", display_modes[2].size().ToString());
  EXPECT_FALSE(display_modes[2].native());
  EXPECT_NEAR(display_modes[2].ui_scale(), 0.8, 0.01);
  EXPECT_EQ(display_modes[2].refresh_rate(), 60);

  EXPECT_EQ("1920x1080", display_modes[3].size().ToString());
  EXPECT_TRUE(display_modes[3].native());
  EXPECT_NEAR(display_modes[3].ui_scale(), 1.0, 0.01);
  EXPECT_EQ(display_modes[3].refresh_rate(), 60);

  EXPECT_EQ("1920x1080", display_modes[4].size().ToString());
  EXPECT_FALSE(display_modes[4].native());
  EXPECT_NEAR(display_modes[4].ui_scale(), 1.25, 0.01);
  EXPECT_EQ(display_modes[4].refresh_rate(), 60);
}

TEST_F(DisplayChangeObserverTest, GetExternalManagedDisplayModeList4K) {
  std::unique_ptr<DisplaySnapshot> display_snapshot =
      FakeDisplaySnapshot::Builder()
          .SetId(123)
          .SetNativeMode(MakeDisplayMode(3840, 2160, false, 30))
          .AddMode(MakeDisplayMode(1920, 1200, false, 60))
          // All non-interlaced (as would be seen with different refresh rates).
          .AddMode(MakeDisplayMode(1920, 1080, false, 80))
          .AddMode(MakeDisplayMode(1920, 1080, false, 70))
          .AddMode(MakeDisplayMode(1920, 1080, false, 60))
          // Interlaced vs non-interlaced.
          .AddMode(MakeDisplayMode(1280, 720, true, 60))
          .AddMode(MakeDisplayMode(1280, 720, false, 60))
          // Interlaced only.
          .AddMode(MakeDisplayMode(1024, 768, true, 70))
          .AddMode(MakeDisplayMode(1024, 768, true, 60))
          // Mixed.
          .AddMode(MakeDisplayMode(1024, 600, true, 60))
          .AddMode(MakeDisplayMode(1024, 600, false, 70))
          .AddMode(MakeDisplayMode(1024, 600, false, 60))
          // Just one interlaced mode.
          .AddMode(MakeDisplayMode(640, 480, true, 60))
          .Build();

  ManagedDisplayInfo::ManagedDisplayModeList display_modes =
      DisplayChangeObserver::GetExternalManagedDisplayModeList(
          *display_snapshot);
  ManagedDisplayInfo info(1, "", false);
  info.SetManagedDisplayModes(display_modes);  // Sort as external display.
  display_modes = info.display_modes();

  ASSERT_EQ(9u, display_modes.size());
  EXPECT_EQ("640x480", display_modes[0].size().ToString());
  EXPECT_TRUE(display_modes[0].is_interlaced());
  EXPECT_EQ(display_modes[0].refresh_rate(), 60);

  EXPECT_EQ("1024x600", display_modes[1].size().ToString());
  EXPECT_FALSE(display_modes[1].is_interlaced());
  EXPECT_EQ(display_modes[1].refresh_rate(), 70);

  EXPECT_EQ("1024x768", display_modes[2].size().ToString());
  EXPECT_TRUE(display_modes[2].is_interlaced());
  EXPECT_EQ(display_modes[2].refresh_rate(), 70);

  EXPECT_EQ("1280x720", display_modes[3].size().ToString());
  EXPECT_FALSE(display_modes[3].is_interlaced());
  EXPECT_EQ(display_modes[3].refresh_rate(), 60);

  EXPECT_EQ("1920x1080", display_modes[4].size().ToString());
  EXPECT_FALSE(display_modes[4].is_interlaced());
  EXPECT_EQ(display_modes[4].refresh_rate(), 80);

  EXPECT_EQ("3840x2160", display_modes[5].size().ToString());
  EXPECT_FALSE(display_modes[5].is_interlaced());
  EXPECT_FALSE(display_modes[5].native());
  EXPECT_EQ(display_modes[5].refresh_rate(), 30);
  EXPECT_EQ(display_modes[5].device_scale_factor(), 2.0);

  EXPECT_EQ("1920x1200", display_modes[6].size().ToString());
  EXPECT_FALSE(display_modes[6].is_interlaced());
  EXPECT_EQ(display_modes[6].refresh_rate(), 60);

  EXPECT_EQ("3840x2160", display_modes[7].size().ToString());
  EXPECT_FALSE(display_modes[7].is_interlaced());
  EXPECT_FALSE(display_modes[7].native());
  EXPECT_EQ(display_modes[7].refresh_rate(), 30);
  EXPECT_EQ(display_modes[7].device_scale_factor(), 1.25);

  EXPECT_EQ("3840x2160", display_modes[8].size().ToString());
  EXPECT_FALSE(display_modes[8].is_interlaced());
  EXPECT_TRUE(display_modes[8].native());
  EXPECT_EQ(display_modes[8].refresh_rate(), 30);
}

TEST_F(DisplayChangeObserverTest, FindDeviceScaleFactor) {
  EXPECT_EQ(1.0f, ComputeDeviceScaleFactor(19.5f, gfx::Rect(1600, 900)));

  // 21.5" 1920x1080
  EXPECT_EQ(1.0f, ComputeDeviceScaleFactor(21.5f, gfx::Rect(1920, 1080)));

  // 12.1" 1280x800
  EXPECT_EQ(1.0f, ComputeDeviceScaleFactor(12.1f, gfx::Rect(1280, 800)));

  // 13.3" 1920x1080
  EXPECT_EQ(1.25f, ComputeDeviceScaleFactor(13.3f, gfx::Rect(1920, 1080)));

  // 14" 1920x1080
  EXPECT_EQ(1.25f, ComputeDeviceScaleFactor(14.0f, gfx::Rect(1920, 1080)));

  // 11.6" 1920x1080
  EXPECT_EQ(1.6f, ComputeDeviceScaleFactor(11.6f, gfx::Rect(1920, 1080)));

  // 12.02" 2160x1440
  EXPECT_EQ(1.6f, ComputeDeviceScaleFactor(12.02f, gfx::Rect(2160, 1440)));

  // 12.85" 2560x1700
  EXPECT_EQ(2.0f, ComputeDeviceScaleFactor(12.85f, gfx::Rect(2560, 1700)));

  // 12.3" 2400x1600
  EXPECT_EQ(2.0f, ComputeDeviceScaleFactor(12.3f, gfx::Rect(2400, 1600)));

  // Erroneous values should still work.
  EXPECT_EQ(1.0f, DisplayChangeObserver::FindDeviceScaleFactor(-100.0f));
  EXPECT_EQ(1.0f, DisplayChangeObserver::FindDeviceScaleFactor(0.0f));
  EXPECT_EQ(2.0f, DisplayChangeObserver::FindDeviceScaleFactor(10000.0f));
}

TEST_F(DisplayChangeObserverTest,
       FindExternalDisplayNativeModeWhenOverwritten) {
  std::unique_ptr<DisplaySnapshot> display_snapshot =
      FakeDisplaySnapshot::Builder()
          .SetId(123)
          .SetNativeMode(MakeDisplayMode(1920, 1080, true, 60))
          .AddMode(MakeDisplayMode(1920, 1080, false, 60))
          .Build();

  ManagedDisplayInfo::ManagedDisplayModeList display_modes =
      DisplayChangeObserver::GetExternalManagedDisplayModeList(
          *display_snapshot);
  ASSERT_EQ(2u, display_modes.size());
  EXPECT_EQ("1920x1080", display_modes[0].size().ToString());
  EXPECT_FALSE(display_modes[0].is_interlaced());
  EXPECT_FALSE(display_modes[0].native());
  EXPECT_EQ(display_modes[0].refresh_rate(), 60);

  EXPECT_EQ("1920x1080", display_modes[1].size().ToString());
  EXPECT_TRUE(display_modes[1].is_interlaced());
  EXPECT_TRUE(display_modes[1].native());
  EXPECT_EQ(display_modes[1].refresh_rate(), 60);
}

}  // namespace display