1 files changed, 156 insertions, 0 deletions
diff --git a/chromium/ash/shell/window_watcher.cc b/chromium/ash/shell/window_watcher.cc
new file mode 100644
index 00000000000..0963eb7cc96
--- /dev/null
+++ b/chromium/ash/shell/window_watcher.cc
@@ -0,0 +1,156 @@
+// 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/shell/window_watcher.h"
+
+#include "ash/display/display_controller.h"
+#include "ash/launcher/launcher.h"
+#include "ash/launcher/launcher_model.h"
+#include "ash/shelf/shelf_widget.h"
+#include "ash/shell.h"
+#include "ash/shell_window_ids.h"
+#include "ui/aura/root_window.h"
+#include "ui/aura/window.h"
+#include "ui/gfx/display.h"
+
+namespace ash {
+namespace shell {
+
+class WindowWatcher::WorkspaceWindowWatcher : public aura::WindowObserver {
+ public:
+  explicit WorkspaceWindowWatcher(WindowWatcher* watcher) : watcher_(watcher) {
+  }
+
+  virtual ~WorkspaceWindowWatcher() {
+  }
+
+  virtual void OnWindowAdded(aura::Window* new_window) OVERRIDE {
+    new_window->AddObserver(watcher_);
+  }
+
+  virtual void OnWillRemoveWindow(aura::Window* window) OVERRIDE {
+    DCHECK(window->children().empty());
+    window->RemoveObserver(watcher_);
+  }
+
+  void RootWindowAdded(aura::RootWindow* root) {
+    aura::Window* panel_container = ash::Shell::GetContainer(
+        root,
+        internal::kShellWindowId_PanelContainer);
+    panel_container->AddObserver(watcher_);
+
+    aura::Window* container =
+        Launcher::ForWindow(root)->shelf_widget()->window_container();
+    container->AddObserver(this);
+    for (size_t i = 0; i < container->children().size(); ++i)
+      container->children()[i]->AddObserver(watcher_);
+  }
+
+  void RootWindowRemoved(aura::RootWindow* root) {
+    aura::Window* panel_container = ash::Shell::GetContainer(
+        root,
+        internal::kShellWindowId_PanelContainer);
+    panel_container->RemoveObserver(watcher_);
+
+    aura::Window* container =
+        Launcher::ForWindow(root)->shelf_widget()->window_container();
+    container->RemoveObserver(this);
+    for (size_t i = 0; i < container->children().size(); ++i)
+      container->children()[i]->RemoveObserver(watcher_);
+  }
+
+ private:
+  WindowWatcher* watcher_;
+
+  DISALLOW_COPY_AND_ASSIGN(WorkspaceWindowWatcher);
+};
+
+WindowWatcher::WindowWatcher() {
+  workspace_window_watcher_.reset(new WorkspaceWindowWatcher(this));
+  Shell::RootWindowList root_windows = Shell::GetAllRootWindows();
+  for (Shell::RootWindowList::iterator iter = root_windows.begin();
+       iter != root_windows.end(); ++ iter) {
+    workspace_window_watcher_->RootWindowAdded(*iter);
+  }
+}
+
+WindowWatcher::~WindowWatcher() {
+  Shell::RootWindowList root_windows = Shell::GetAllRootWindows();
+  for (Shell::RootWindowList::iterator iter = root_windows.begin();
+       iter != root_windows.end(); ++ iter) {
+    workspace_window_watcher_->RootWindowRemoved(*iter);
+  }
+}
+
+aura::Window* WindowWatcher::GetWindowByID(ash::LauncherID id) {
+  IDToWindow::const_iterator i = id_to_window_.find(id);
+  return i != id_to_window_.end() ? i->second : NULL;
+}
+
+ash::LauncherID WindowWatcher::GetIDByWindow(aura::Window* window) const {
+  for (IDToWindow::const_iterator i = id_to_window_.begin();
+       i != id_to_window_.end(); ++i) {
+    if (i->second == window)
+      return i->first;
+  }
+  return 0;  // TODO: add a constant for this.
+}
+
+// aura::WindowObserver overrides:
+void WindowWatcher::OnWindowAdded(aura::Window* new_window) {
+  if (new_window->type() != aura::client::WINDOW_TYPE_NORMAL &&
+      new_window->type() != aura::client::WINDOW_TYPE_PANEL)
+    return;
+
+  static int image_count = 0;
+  ash::LauncherModel* model = Shell::GetInstance()->launcher_model();
+  ash::LauncherItem item;
+  item.type = new_window->type() == aura::client::WINDOW_TYPE_PANEL ?
+                                    ash::TYPE_APP_PANEL : ash::TYPE_TABBED;
+  id_to_window_[model->next_id()] = new_window;
+
+  SkBitmap icon_bitmap;
+  icon_bitmap.setConfig(SkBitmap::kARGB_8888_Config, 16, 16);
+  icon_bitmap.allocPixels();
+  icon_bitmap.eraseARGB(255,
+                        image_count == 0 ? 255 : 0,
+                        image_count == 1 ? 255 : 0,
+                        image_count == 2 ? 255 : 0);
+  image_count = (image_count + 1) % 3;
+  item.image = gfx::ImageSkia(gfx::ImageSkiaRep(icon_bitmap,
+                                                ui::SCALE_FACTOR_100P));
+
+  model->Add(item);
+}
+
+void WindowWatcher::OnWillRemoveWindow(aura::Window* window) {
+  for (IDToWindow::iterator i = id_to_window_.begin();
+       i != id_to_window_.end(); ++i) {
+    if (i->second == window) {
+      ash::LauncherModel* model = Shell::GetInstance()->launcher_model();
+      int index = model->ItemIndexByID(i->first);
+      DCHECK_NE(-1, index);
+      model->RemoveItemAt(index);
+      id_to_window_.erase(i);
+      break;
+    }
+  }
+}
+
+void WindowWatcher::OnDisplayBoundsChanged(const gfx::Display& display) {
+}
+
+void WindowWatcher::OnDisplayAdded(const gfx::Display& new_display) {
+  aura::RootWindow* root = Shell::GetInstance()->display_controller()->
+      GetRootWindowForDisplayId(new_display.id());
+  workspace_window_watcher_->RootWindowAdded(root);
+}
+
+void WindowWatcher::OnDisplayRemoved(const gfx::Display& old_display) {
+  // All windows in the display has already been removed, so no need to
+  // remove observers.
+}
+
+}  // namespace shell
+}  // namespace ash