ash/test/display_manager_test_api.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "ash/test/display_manager_test_api.h"
   6
   7 #include <vector>
   8
   9 #include "ash/display/display_info.h"
  10 #include "ash/display/display_manager.h"
  11 #include "ash/display/display_util.h"
  12 #include "ash/display/extended_mouse_warp_controller.h"
  13 #include "ash/display/mouse_cursor_event_filter.h"
  14 #include "ash/display/unified_mouse_warp_controller.h"
  15 #include "ash/shell.h"
  16 #include "base/strings/string_split.h"
  17 #include "ui/aura/env.h"
  18 #include "ui/aura/window_event_dispatcher.h"
  19 #include "ui/events/test/event_generator.h"
  20 #include "ui/gfx/display.h"
  21
  22 namespace ash {
  23 namespace test {
  24 typedef std::vector<gfx::Display> DisplayList;
  25 typedef DisplayInfo DisplayInfo;
  26 typedef std::vector<DisplayInfo> DisplayInfoList;
  27
  28 namespace {
  29
  30 std::vector<DisplayInfo> CreateDisplayInfoListFromString(
  31     const std::string specs,
  32     DisplayManager* display_manager) {
  33   std::vector<DisplayInfo> display_info_list;
  34   std::vector<std::string> parts;
  35   base::SplitString(specs, ',', &parts);
  36   size_t index = 0;
  37   for (std::vector<std::string>::const_iterator iter = parts.begin();
  38        iter != parts.end(); ++iter, ++index) {
  39     int64 id = index < display_manager->GetNumDisplays() ?
  40         display_manager->GetDisplayAt(index).id() :
  41         gfx::Display::kInvalidDisplayID;
  42     display_info_list.push_back(
  43         DisplayInfo::CreateFromSpecWithID(*iter, id));
  44   }
  45   return display_info_list;
  46 }
  47
  48 }  // namespace
  49
  50 // static
  51 bool DisplayManagerTestApi::TestIfMouseWarpsAt(
  52     ui::test::EventGenerator& event_generator,
  53     const gfx::Point& point_in_screen) {
  54   aura::Window* context = Shell::GetAllRootWindows()[0];
  55   DisplayManager* display_manager = Shell::GetInstance()->display_manager();
  56   if (display_manager->IsInUnifiedMode()) {
  57     static_cast<UnifiedMouseWarpController*>(
  58         Shell::GetInstance()
  59             ->mouse_cursor_filter()
  60             ->mouse_warp_controller_for_test())
  61         ->allow_non_native_event_for_test();
  62     int orig_index = FindDisplayIndexContainingPoint(
  63         display_manager->software_mirroring_display_list(), point_in_screen);
  64     if (orig_index < 0)
  65       return false;
  66     event_generator.MoveMouseTo(point_in_screen);
  67
  68     int new_index = FindDisplayIndexContainingPoint(
  69         display_manager->software_mirroring_display_list(),
  70         aura::Env::GetInstance()->last_mouse_location());
  71     if (new_index < 0)
  72       return false;
  73     return orig_index != new_index;
  74   } else {
  75     static_cast<ExtendedMouseWarpController*>(
  76         Shell::GetInstance()
  77             ->mouse_cursor_filter()
  78             ->mouse_warp_controller_for_test())
  79         ->allow_non_native_event_for_test();
  80     gfx::Screen* screen = gfx::Screen::GetScreenFor(context);
  81     gfx::Display original_display =
  82         screen->GetDisplayNearestPoint(point_in_screen);
  83     event_generator.MoveMouseTo(point_in_screen);
  84     return original_display.id() !=
  85            screen->GetDisplayNearestPoint(
  86                        aura::Env::GetInstance()->last_mouse_location()).id();
  87   }
  88 }
  89
  90 DisplayManagerTestApi::DisplayManagerTestApi(DisplayManager* display_manager)
  91     : display_manager_(display_manager) {}
  92
  93 DisplayManagerTestApi::~DisplayManagerTestApi() {}
  94
  95 void DisplayManagerTestApi::UpdateDisplay(
  96     const std::string& display_specs) {
  97   std::vector<DisplayInfo> display_info_list =
  98       CreateDisplayInfoListFromString(display_specs, display_manager_);
  99   bool is_host_origin_set = false;
 100   for (size_t i = 0; i < display_info_list.size(); ++i) {
 101     const DisplayInfo& display_info = display_info_list[i];
 102     if (display_info.bounds_in_native().origin() != gfx::Point(0, 0)) {
 103       is_host_origin_set = true;
 104       break;
 105     }
 106   }
 107
 108   // On non-testing environment, when a secondary display is connected, a new
 109   // native (i.e. X) window for the display is always created below the
 110   // previous one for GPU performance reasons. Try to emulate the behavior
 111   // unless host origins are explicitly set.
 112   if (!is_host_origin_set) {
 113     // Sart from (1,1) so that windows won't overlap with native mouse cursor.
 114     // See |AshTestBase::SetUp()|.
 115     int next_y = 1;
 116     for (std::vector<DisplayInfo>::iterator iter = display_info_list.begin();
 117          iter != display_info_list.end(); ++iter) {
 118       gfx::Rect bounds(iter->bounds_in_native().size());
 119       bounds.set_x(1);
 120       bounds.set_y(next_y);
 121       next_y += bounds.height();
 122       iter->SetBounds(bounds);
 123     }
 124   }
 125
 126   display_manager_->OnNativeDisplaysChanged(display_info_list);
 127   display_manager_->UpdateInternalDisplayModeListForTest();
 128 }
 129
 130 int64 DisplayManagerTestApi::SetFirstDisplayAsInternalDisplay() {
 131   const gfx::Display& internal = display_manager_->active_display_list_[0];
 132   SetInternalDisplayId(internal.id());
 133   return gfx::Display::InternalDisplayId();
 134 }
 135
 136 void DisplayManagerTestApi::SetInternalDisplayId(int64 id) {
 137   gfx::Display::SetInternalDisplayId(id);
 138   display_manager_->UpdateInternalDisplayModeListForTest();
 139 }
 140
 141 void DisplayManagerTestApi::DisableChangeDisplayUponHostResize() {
 142   display_manager_->set_change_display_upon_host_resize(false);
 143 }
 144
 145 void DisplayManagerTestApi::SetAvailableColorProfiles(
 146     int64 display_id,
 147     const std::vector<ui::ColorCalibrationProfile>& profiles) {
 148   display_manager_->display_info_[display_id].set_available_color_profiles(
 149       profiles);
 150 }
 151
 152 }  // namespace test
 153 }  // namespace ash