Cancel partial screenshot UI when lock happens.

ash/shell.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/shell.h"
 
 #include <algorithm>
 
 #include "ash/app_list/app_list.h"
 #include "ash/ash_switches.h"
@@ skipping 637 matching lines @@
   // KeyRewriterEventFilter must be the first one.
   DCHECK(!GetRootWindowEventFilterCount());
   key_rewriter_filter_.reset(new internal::KeyRewriterEventFilter);
   AddRootWindowEventFilter(key_rewriter_filter_.get());
 
   // PartialScreenshotEventFilter must be the second one to capture key
   // events when the taking partial screenshot UI is there.
   DCHECK_EQ(1U, GetRootWindowEventFilterCount());
   partial_screenshot_filter_.reset(new internal::PartialScreenshotEventFilter);
   AddRootWindowEventFilter(partial_screenshot_filter_.get());
+  AddShellObserver(partial_screenshot_filter_.get());
 
   // Then AcceleratorFilter and InputMethodEventFilter must be added (in this
   // order) since they have the second highest priority.
   DCHECK_EQ(2U, GetRootWindowEventFilterCount());
 #if !defined(OS_MACOSX)
   accelerator_filter_.reset(new internal::AcceleratorFilter);
   AddRootWindowEventFilter(accelerator_filter_.get());
   DCHECK_EQ(3U, GetRootWindowEventFilterCount());
 #endif
   input_method_filter_.reset(new internal::InputMethodEventFilter);
@@ skipping 95 matching lines @@
   window_modality_controller_.reset(new internal::WindowModalityController);
   AddRootWindowEventFilter(window_modality_controller_.get());
 
   visibility_controller_.reset(new internal::VisibilityController);
 
   tooltip_controller_.reset(new internal::TooltipController);
   AddRootWindowEventFilter(tooltip_controller_.get());
 
   drag_drop_controller_.reset(new internal::DragDropController);
   power_button_controller_.reset(new PowerButtonController);
+  AddShellObserver(power_button_controller_.get());
   video_detector_.reset(new VideoDetector);
   window_cycle_controller_.reset(new WindowCycleController);
   monitor_controller_.reset(new internal::MonitorController);
 }
 
 aura::Window* Shell::GetContainer(int container_id) {
   return const_cast<aura::Window*>(
       const_cast<const Shell*>(this)->GetContainer(container_id));
 }
 
@@ skipping 68 matching lines @@
                                      const gfx::Insets& insets) {
   aura::Monitor* monitor = aura::Env::GetInstance()->monitor_manager()->
       GetMonitorNearestWindow(contains);
   if (monitor->work_area_insets() == insets)
     return;
   monitor->set_work_area_insets(insets);
   FOR_EACH_OBSERVER(ShellObserver, observers_,
                     OnMonitorWorkAreaInsetsChanged());
 }
 
+void Shell::OnLoginStateChange(bool logged_in, bool is_guest) {
+  FOR_EACH_OBSERVER(ShellObserver, observers_,
+                    OnLoginStateChange(logged_in, is_guest));
+}
+
+void Shell::OnExit() {
+  FOR_EACH_OBSERVER(ShellObserver, observers_, OnExit());
+}
+
+void Shell::OnLockStateChange(bool locked) {
+  FOR_EACH_OBSERVER(ShellObserver, observers_, OnLockStateChange(locked));
+}
+
 void Shell::CreateLauncher() {
   if (launcher_.get())
     return;
 
   aura::Window* default_container =
       GetContainer(internal::kShellWindowId_DefaultContainer);
   launcher_.reset(new Launcher(default_container));
 
   launcher_->SetFocusCycler(focus_cycler_.get());
   shelf_->SetLauncher(launcher_.get());
@@ skipping 65 matching lines @@
   // Create desktop background widget.
   desktop_background_controller_->SetDesktopBackgroundImageMode();
 }
 
 void Shell::DisableWorkspaceGridLayout() {
   if (workspace_controller_.get())
     workspace_controller_->workspace_manager()->set_grid_size(0);
 }
 
 }  // namespace ash