Chrome OS: Ash support for >=3 displays.

ash/wm/drag_window_resizer.cc

 // Copyright (c) 2012 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 "ash/wm/drag_window_resizer.h"
 
 #include "ash/display/mouse_cursor_event_filter.h"
 #include "ash/screen_util.h"
 #include "ash/shell.h"
 #include "ash/wm/coordinate_conversion.h"
@@ skipping 12 matching lines @@
 #include "ui/wm/core/coordinate_conversion.h"
 #include "ui/wm/core/window_util.h"
 
 namespace ash {
 namespace {
 
 // The maximum opacity of the drag phantom window.
 const float kMaxOpacity = 0.8f;
 
 // Returns true if Ash has more than one root window.
-bool HasSecondaryRootWindow() {
+bool HasSecondaryRootWindows() {
   return Shell::GetAllRootWindows().size() > 1;
 }
 
-// When there are two root windows, returns one of the root windows which is not
-// |root_window|. Returns NULL if only one root window exists.
-aura::Window* GetAnotherRootWindow(aura::Window* root_window) {
+// When there are more than one root windows, returns all root windows which are
+// not |root_window|. Returns an empty vector if only one root window exists.
+std::vector<aura::Window*> GetOtherRootWindows(aura::Window* root_window) {
   aura::Window::Windows root_windows = Shell::GetAllRootWindows();
+  std::vector<aura::Window*> other_root_windows;

[oshima, 2014/12/05 22:18:15]

any reason not to use aura::Window::Windows?

[hshi1, 2014/12/05 23:25:12]

Done.

   if (root_windows.size() < 2)
-    return NULL;
-  DCHECK_EQ(2U, root_windows.size());
-  if (root_windows[0] == root_window)
-    return root_windows[1];
-  return root_windows[0];
+    return other_root_windows;
+  for (size_t i = 0; i < root_windows.size(); ++i) {
+    if (root_windows[i] != root_window)
+      other_root_windows.push_back(root_windows[i]);
+  }
+  return other_root_windows;
 }
 
 }  // namespace
 
 // static
 DragWindowResizer* DragWindowResizer::instance_ = NULL;
 
 DragWindowResizer::~DragWindowResizer() {
   if (window_state_)
     window_state_->DeleteDragDetails();
@@ skipping 14 matching lines @@
 
 void DragWindowResizer::Drag(const gfx::Point& location, int event_flags) {
   base::WeakPtr<DragWindowResizer> resizer(weak_ptr_factory_.GetWeakPtr());
   next_window_resizer_->Drag(location, event_flags);
 
   if (!resizer)
     return;
 
   last_mouse_location_ = location;
   // Show a phantom window for dragging in another root window.
-  if (HasSecondaryRootWindow()) {
+  if (HasSecondaryRootWindows()) {
     gfx::Point location_in_screen = location;
     ::wm::ConvertPointToScreen(GetTarget()->parent(), &location_in_screen);
     const bool in_original_root =
         wm::GetRootWindowAt(location_in_screen) == GetTarget()->GetRootWindow();
     UpdateDragWindow(GetTarget()->bounds(), in_original_root);
   } else {
-    drag_window_controller_.reset();
+    drag_window_controllers_.clear();
   }
 }
 
 void DragWindowResizer::CompleteDrag() {
   next_window_resizer_->CompleteDrag();
 
   GetTarget()->layer()->SetOpacity(details().initial_opacity);
-  drag_window_controller_.reset();
+  drag_window_controllers_.clear();
 
   // Check if the destination is another display.
   gfx::Point last_mouse_location_in_screen = last_mouse_location_;
   ::wm::ConvertPointToScreen(GetTarget()->parent(),
                              &last_mouse_location_in_screen);
   gfx::Screen* screen = Shell::GetScreen();
   const gfx::Display dst_display =
       screen->GetDisplayNearestPoint(last_mouse_location_in_screen);
 
   if (dst_display.id() !=
@@ skipping 24 matching lines @@
     ash::wm::AdjustBoundsToEnsureMinimumWindowVisibility(
         dst_display.bounds(), &dst_bounds);
 
     GetTarget()->SetBoundsInScreen(dst_bounds, dst_display);
   }
 }
 
 void DragWindowResizer::RevertDrag() {
   next_window_resizer_->RevertDrag();
 
-  drag_window_controller_.reset();
+  drag_window_controllers_.clear();
   GetTarget()->layer()->SetOpacity(details().initial_opacity);
 }
 
 DragWindowResizer::DragWindowResizer(WindowResizer* next_window_resizer,
                                      wm::WindowState* window_state)
     : WindowResizer(window_state),
       next_window_resizer_(next_window_resizer),
       weak_ptr_factory_(this) {
   // The pointer should be confined in one display during resizing a window
   // because the window cannot span two displays at the same time anyway. The
   // exception is window/tab dragging operation. During that operation,
   // |mouse_warp_mode_| should be set to WARP_DRAG so that the user could move a
   // window/tab to another display.
   MouseCursorEventFilter* mouse_cursor_filter =
       Shell::GetInstance()->mouse_cursor_filter();
   mouse_cursor_filter->set_mouse_warp_mode(
       ShouldAllowMouseWarp() ?
       MouseCursorEventFilter::WARP_DRAG : MouseCursorEventFilter::WARP_NONE);
   if (ShouldAllowMouseWarp())
     mouse_cursor_filter->ShowSharedEdgeIndicator(GetTarget()->GetRootWindow());
   instance_ = this;
 }
 
 void DragWindowResizer::UpdateDragWindow(const gfx::Rect& bounds,
                                          bool in_original_root) {
   if (details().window_component != HTCAPTION || !ShouldAllowMouseWarp())
     return;
 
   // It's available. Show a phantom window on the display if needed.
-  aura::Window* another_root =
-      GetAnotherRootWindow(GetTarget()->GetRootWindow());
-  const gfx::Rect root_bounds_in_screen(another_root->GetBoundsInScreen());
-  const gfx::Rect bounds_in_screen =
+  std::vector<aura::Window*> other_roots =
+      GetOtherRootWindows(GetTarget()->GetRootWindow());
+  size_t drag_window_controller_count = 0;
+  for (size_t i = 0; i < other_roots.size(); ++i) {
+    aura::Window* another_root = other_roots[i];
+    const gfx::Rect root_bounds_in_screen(another_root->GetBoundsInScreen());
+    const gfx::Rect bounds_in_screen =
       ScreenUtil::ConvertRectToScreen(GetTarget()->parent(), bounds);
-  gfx::Rect bounds_in_another_root =
-      gfx::IntersectRects(root_bounds_in_screen, bounds_in_screen);
-  const float fraction_in_another_window =
-      (bounds_in_another_root.width() * bounds_in_another_root.height()) /
-      static_cast<float>(bounds.width() * bounds.height());
+    gfx::Rect bounds_in_another_root =
+        gfx::IntersectRects(root_bounds_in_screen, bounds_in_screen);
+    const float fraction_in_another_window =
+        (bounds_in_another_root.width() * bounds_in_another_root.height()) /
+        static_cast<float>(bounds.width() * bounds.height());
 
-  if (fraction_in_another_window > 0) {
-    if (!drag_window_controller_) {
-      drag_window_controller_.reset(
-          new DragWindowController(GetTarget()));
-      // Always show the drag phantom on the |another_root| window.
-      drag_window_controller_->SetDestinationDisplay(
-          Shell::GetScreen()->GetDisplayNearestWindow(another_root));
-      drag_window_controller_->Show();
+    if (fraction_in_another_window > 0) {
+      if (drag_window_controllers_.size() < ++drag_window_controller_count)
+        drag_window_controllers_.resize(drag_window_controller_count);
+      scoped_ptr<DragWindowController>& drag_window_controller =
+        drag_window_controllers_.back();
+      if (!drag_window_controller) {
+        drag_window_controller.reset(new DragWindowController(GetTarget()));
+        // Always show the drag phantom on the |another_root| window.
+        drag_window_controller->SetDestinationDisplay(
+            Shell::GetScreen()->GetDisplayNearestWindow(another_root));
+        drag_window_controller->Show();
+      } else {
+        // No animation.
+        drag_window_controller->SetBounds(bounds_in_screen);
+      }
+      const float phantom_opacity =
+        !in_original_root ? 1 : (kMaxOpacity * fraction_in_another_window);

[oshima, 2014/12/05 22:18:15]

indent

[hshi1, 2014/12/05 23:25:12]

Done.

+      const float window_opacity =
+          in_original_root ? 1
+                           : (kMaxOpacity * (1 - fraction_in_another_window));
+      drag_window_controller->SetOpacity(phantom_opacity);
+      GetTarget()->layer()->SetOpacity(window_opacity);
     } else {
-      // No animation.
-      drag_window_controller_->SetBounds(bounds_in_screen);
+      GetTarget()->layer()->SetOpacity(1.0f);
     }
-    const float phantom_opacity =
-      !in_original_root ? 1 : (kMaxOpacity * fraction_in_another_window);
-    const float window_opacity =
-        in_original_root ? 1 : (kMaxOpacity * (1 - fraction_in_another_window));
-    drag_window_controller_->SetOpacity(phantom_opacity);
-    GetTarget()->layer()->SetOpacity(window_opacity);
-  } else {
-    drag_window_controller_.reset();
-    GetTarget()->layer()->SetOpacity(1.0f);
   }
+
+  // If we have more drag window controllers allocated than needed, release the
+  // excess controllers by shrinking the vector |drag_window_controller_|.
+  DCHECK_GE(drag_window_controllers_.size(), drag_window_controller_count);
+  if (drag_window_controllers_.size() > drag_window_controller_count)
+    drag_window_controllers_.resize(drag_window_controller_count);
 }
 
 bool DragWindowResizer::ShouldAllowMouseWarp() {
   return (details().window_component == HTCAPTION) &&
       !::wm::GetTransientParent(GetTarget()) &&
       (GetTarget()->type() == ui::wm::WINDOW_TYPE_NORMAL ||
        GetTarget()->type() == ui::wm::WINDOW_TYPE_PANEL);
 }
 
 }  // namespace ash