move windows when disconnecting secondary monitors

ash/root_window_controller.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/root_window_controller.h"
 
 #include "ash/shell.h"
 #include "ash/shell_factory.h"
 #include "ash/shell_window_ids.h"
 #include "ash/wm/base_layout_manager.h"
 #include "ash/wm/event_client_impl.h"
 #include "ash/wm/property_util.h"
 #include "ash/wm/root_window_layout_manager.h"
 #include "ash/wm/screen_dimmer.h"
 #include "ash/wm/system_modal_container_layout_manager.h"
 #include "ash/wm/toplevel_window_event_filter.h"
 #include "ash/wm/visibility_controller.h"
 #include "ash/wm/workspace/workspace_manager.h"
 #include "ash/wm/workspace_controller.h"
+#include "ui/aura/client/activation_client.h"
+#include "ui/aura/client/aura_constants.h"
 #include "ui/aura/client/tooltip_client.h"
+#include "ui/aura/focus_manager.h"
 #include "ui/aura/root_window.h"
 
 namespace ash {
 namespace {
 
 // Creates a new window for use as a container.
 aura::Window* CreateContainer(int window_id,
                               const char* name,
                               aura::Window* parent) {
   aura::Window* container = new aura::Window(NULL);
   container->set_id(window_id);
   container->SetName(name);
   container->Init(ui::LAYER_NOT_DRAWN);
   parent->AddChild(container);
   if (window_id != internal::kShellWindowId_UnparentedControlContainer)
     container->Show();
   return container;
 }
 
+void MoveAllWindows(aura::RootWindow* src,
+                    aura::RootWindow* dst) {
+  // Windows move only from secondary displays to the primary
+  // display, so no need to move windows in the containers that are
+  // available only in the primary display (launcher, panels etc)
+  const int kContainerIdsToMove[] = {
+    internal::kShellWindowId_DefaultContainer,
+    internal::kShellWindowId_AlwaysOnTopContainer,
+    internal::kShellWindowId_SystemModalContainer,
+    internal::kShellWindowId_LockSystemModalContainer,
+  };
+
+  for (size_t i = 0; i < arraysize(kContainerIdsToMove); i++) {
+    int id = kContainerIdsToMove[i];
+    aura::Window* src_container = Shell::GetContainer(src, id);
+    aura::Window* dst_container = Shell::GetContainer(dst, id);
+    aura::Window::Windows children = src_container->children();
+    for (aura::Window::Windows::iterator iter = children.begin();
+         iter != children.end(); ++iter) {
+      aura::Window* window = *iter;
+      // Don't move modal screen.
+      if ((id == internal::kShellWindowId_SystemModalContainer ||
+           id == internal::kShellWindowId_LockSystemModalContainer) &&
+          window->GetProperty(aura::client::kModalKey) == ui::MODAL_TYPE_NONE) {
+        continue;
+      }
+      dst_container->AddChild(window);
+    }
+  }
+}
+
 // Creates each of the special window containers that holds windows of various
 // types in the shell UI.
 void CreateContainersInRootWindow(aura::RootWindow* root_window) {
   // These containers are just used by PowerButtonController to animate groups
   // of containers simultaneously without messing up the current transformations
   // on those containers.  These are direct children of the root window; all of
   // the other containers are their children.
   aura::Window* non_lock_screen_containers = CreateContainer(
       internal::kShellWindowId_NonLockScreenContainersContainer,
       "NonLockScreenContainersContainer",
@@ skipping 157 matching lines @@
   while (!root_window_->children().empty()) {
     aura::Window* child = root_window_->children()[0];
     delete child;
   }
 }
 
 bool RootWindowController::IsInMaximizedMode() const {
   return workspace_controller_->workspace_manager()->IsInMaximizedMode();
 }
 
+void RootWindowController::MoveWindowsTo(aura::RootWindow* dst) {
+  aura::Window* focused = dst->GetFocusManager()->GetFocusedWindow();
+  aura::client::ActivationClient* activation_client =
+      aura::client::GetActivationClient(dst);
+  aura::Window* active = activation_client->GetActiveWindow();
+  // Deactivate the window to close menu / bubble windows.
+  activation_client->DeactivateWindow(active);
+  // Close the background menu if any.
+  Shell::GetInstance()->HideBackgroundMenu();

[sky, 2012/06/26 04:10:44]

I don't want anything special like this. There should be something the menu code
observes to properly clean things up.

+
+  MoveAllWindows(root_window_.get(), dst);
+
+  // Restore focused / active window.
+  if (focused) {
+    DCHECK(dst->Contains(focused));
+    dst->GetFocusManager()->SetFocusedWindow(focused, NULL);
+  } else if (active) {
+    DCHECK(dst->Contains(active));
+    activation_client->ActivateWindow(active);
+  }
+}
+
 }  // namespace internal
 }  // namespace ash