Remove root window from screen_ash

ash/screen_ash.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/screen_ash.h"
 
 #include "ash/shell.h"
 #include "ash/wm/shelf_layout_manager.h"
 #include "base/logging.h"
 #include "ui/aura/env.h"
 #include "ui/aura/monitor_manager.h"
 #include "ui/aura/root_window.h"
 #include "ui/gfx/monitor.h"
 #include "ui/gfx/screen.h"
 
 namespace ash {
 
 namespace {
 aura::MonitorManager* GetMonitorManager() {
   return aura::Env::GetInstance()->monitor_manager();
 }
 }  // namespace
 
-ScreenAsh::ScreenAsh(aura::RootWindow* root_window)
-    : root_window_(root_window) {
+ScreenAsh::ScreenAsh() {
 }
 
 ScreenAsh::~ScreenAsh() {
 }
 
 // static
 gfx::Rect ScreenAsh::GetMaximizedWindowBounds(aura::Window* window) {
-  return Shell::GetInstance()->shelf()->GetMaximizedWindowBounds(window);
+  if (window->GetRootWindow() == Shell::GetPrimaryRootWindow())
+    return Shell::GetInstance()->shelf()->GetMaximizedWindowBounds(window);
+  else
+    return gfx::Screen::GetMonitorNearestWindow(window).bounds();
 }
 
 // static
 gfx::Rect ScreenAsh::GetUnmaximizedWorkAreaBounds(aura::Window* window) {
-  return Shell::GetInstance()->shelf()->GetUnmaximizedWorkAreaBounds(window);
+  if (window->GetRootWindow() == Shell::GetPrimaryRootWindow())
+    return Shell::GetInstance()->shelf()->GetUnmaximizedWorkAreaBounds(window);
+  else
+    return gfx::Screen::GetMonitorNearestWindow(window).work_area();
 }
 
 gfx::Point ScreenAsh::GetCursorScreenPoint() {
-  return root_window_->last_mouse_location();
+  // TODO(oshima): Support multiple root window.
+  return Shell::GetPrimaryRootWindow()->last_mouse_location();
 }
 
 gfx::NativeWindow ScreenAsh::GetWindowAtCursorScreenPoint() {
   const gfx::Point point = gfx::Screen::GetCursorScreenPoint();
-  return root_window_->GetTopWindowContainingPoint(point);
+  // TODO(oshima): convert point to relateive to the root window.
+  return Shell::GetRootWindowAt(point)->GetTopWindowContainingPoint(point);
 }
 
 int ScreenAsh::GetNumMonitors() {
   return GetMonitorManager()->GetNumMonitors();
 }
 
 gfx::Monitor ScreenAsh::GetMonitorNearestWindow(gfx::NativeView window) const {
   return GetMonitorManager()->GetMonitorNearestWindow(window);
 }
 
 gfx::Monitor ScreenAsh::GetMonitorNearestPoint(const gfx::Point& point) const {
   return GetMonitorManager()->GetMonitorNearestPoint(point);
 }
 
 gfx::Monitor ScreenAsh::GetPrimaryMonitor() const {
   return GetMonitorManager()->GetMonitorAt(0);
 }
 
 }  // namespace ash