Add context to gfx::Screen calls in support of simultaneous desktop+ash

ash/shell.cc

Index: ash/shell.cc
diff --git a/ash/shell.cc b/ash/shell.cc
index 9de3e0918b66331d66c866ee55f098d6613fafdd..48c13e81cb1324321d535f27852c9c23dc210d41 100644
--- a/ash/shell.cc
+++ b/ash/shell.cc
@@ -194,7 +194,8 @@ Shell::Shell(ShellDelegate* delegate)
       panel_layout_manager_(NULL),
       status_area_widget_(NULL),
       browser_context_(NULL) {
-  gfx::Screen::SetInstance(screen_);
+  gfx::Screen::SetScreenInstance(gfx::SCREEN_TYPE_NATIVE, screen_);
+  gfx::Screen::SetScreenInstance(gfx::SCREEN_TYPE_ALTERNATE, screen_);
   ui_controls::InstallUIControlsAura(internal::CreateUIControls());
 #if defined(OS_CHROMEOS)
   output_configurator_->AddObserver(output_configurator_animation_.get());
@@ -337,6 +338,11 @@ aura::RootWindow* Shell::GetActiveRootWindow() {
 }
 
 // static
+gfx::Screen* Shell::GetAshScreen() {
+  return gfx::Screen::GetScreenFor(GetPrimaryRootWindow());
+}
+
+// static
 Shell::RootWindowList Shell::GetAllRootWindows() {
   return Shell::GetInstance()->display_controller()->
       GetAllRootWindows();