Cancel partial screenshot UI when lock happens.

ash/wm/power_button_controller.h

 // 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.
 
 #ifndef ASH_WM_POWER_BUTTON_CONTROLLER_H_
 #define ASH_WM_POWER_BUTTON_CONTROLLER_H_
 #pragma once
 
 #include "ash/ash_export.h"
+#include "ash/shell.h"
+#include "ash/shell_observer.h"
 #include "base/basictypes.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/time.h"
 #include "base/timer.h"
 #include "ui/aura/root_window_observer.h"
 
 namespace gfx {
 class Size;
 }
 
@@ skipping 11 matching lines @@
 
   virtual void RequestLockScreen() = 0;
   virtual void RequestShutdown() = 0;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(PowerButtonControllerDelegate);
 };
 
 // Displays onscreen animations and locks or suspends the system in response to
 // the power button being pressed or released.
-class ASH_EXPORT PowerButtonController : public aura::RootWindowObserver {
+class ASH_EXPORT PowerButtonController : public aura::RootWindowObserver,
+                                         public ShellObserver {
  public:
   // Animations that can be applied to groups of containers.
   // Exposed here for TestApi::ContainerGroupIsAnimated().
   enum AnimationType {
     ANIMATION_SLOW_CLOSE = 0,
     ANIMATION_UNDO_SLOW_CLOSE,
     ANIMATION_FAST_CLOSE,
     ANIMATION_FADE_IN,
     ANIMATION_HIDE,
     ANIMATION_RESTORE,
@@ skipping 76 matching lines @@
   virtual ~PowerButtonController();
 
   void set_delegate(PowerButtonControllerDelegate* delegate) {
     delegate_.reset(delegate);
   }
 
   void set_has_legacy_power_button_for_test(bool legacy) {
     has_legacy_power_button_ = legacy;
   }
 
-  // Called when the user logs in.
-  void OnLoginStateChange(bool logged_in, bool is_guest);
-
-  // Called when the application is exiting.
-  void OnExit();
-
-  // Called when the screen is locked (after the lock window is visible) or
-  // unlocked.
-  void OnLockStateChange(bool locked);
-
   // Called when Chrome gets a request to display the lock screen.
   void OnStartingLock();
 
   // Called when the power or lock buttons are pressed or released.
   void OnPowerButtonEvent(bool down, const base::TimeTicks& timestamp);
   void OnLockButtonEvent(bool down, const base::TimeTicks& timestamp);
 
   // Displays the shutdown animation and requests shutdown when it's done.
   void RequestShutdown();
 
   // aura::RootWindowObserver overrides:
   virtual void OnRootWindowResized(const aura::RootWindow* root,
                                    const gfx::Size& old_size) OVERRIDE;
 
+  // ShellObserver overrides:
+  virtual void OnLoginStateChanged(Shell::LoginStatus status) OVERRIDE;
+  virtual void OnAppTerminating() OVERRIDE;
+  virtual void OnLockStateChanged(bool locked) OVERRIDE;
+
  private:
   bool logged_in_as_non_guest() const { return logged_in_ && !is_guest_; }
 
   // Requests that the screen be locked and starts |lock_fail_timer_|.
   void OnLockTimeout();
 
   // Aborts the pre-lock animation.
   void OnLockFailTimeout();
 
   // Displays the pre-shutdown animation and starts |shutdown_timer_|.
@@ skipping 78 matching lines @@
   // |background_layer_|, as the desktop background is now covering the whole
   // screen.
   base::OneShotTimer<PowerButtonController> hide_background_layer_timer_;
 
   DISALLOW_COPY_AND_ASSIGN(PowerButtonController);
 };
 
 }  // namespace ash
 
 #endif  // ASH_WM_POWER_BUTTON_CONTROLLER_H_