Tablet-like power button behavior on Convertible/Tablet ChromeOS devices

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_
 
 #include "ash/ash_export.h"
 #include "base/macros.h"
 #include "base/time/time.h"
+#include "base/timer/timer.h"
 #include "ui/events/event_handler.h"
 
 #if defined(OS_CHROMEOS)
 #include "chromeos/dbus/power_manager_client.h"
 #include "ui/display/chromeos/display_configurator.h"
 #endif
 
 namespace gfx {
 class Rect;
 class Size;
 }
 
 namespace ui {
 class Layer;
 }
 
 namespace ash {
 
-namespace test {
-class PowerButtonControllerTest;
-}
-
 class LockStateController;
 
 // Handles power & lock button events which may result in the locking or
 // shutting down of the system as well as taking screen shots while in maximize
 // mode.
 class ASH_EXPORT PowerButtonController
     : public ui::EventHandler
 // TODO(derat): Remove these ifdefs after DisplayConfigurator becomes
 // cross-platform.
 #if defined(OS_CHROMEOS)
       ,
       public ui::DisplayConfigurator::Observer,
       public chromeos::PowerManagerClient::Observer
 #endif
 {
  public:
+  // Helper class used by tablet power button tests to access internal state.
+  class ASH_EXPORT TestApi {
+   public:
+    explicit TestApi(PowerButtonController* controller);
+    virtual ~TestApi();
+
+    bool TabletPowerButtonTimerIsRunning() const;

[Daniel Erat, 2016/11/04 20:18:34]

nit: add comments to both of these documenting what they do

[Qiang(Joe) Xu, 2016/11/04 23:18:21]

Done.

+
+    void TriggerTabletPowerButtonTimeout();
+
+   private:
+    PowerButtonController* controller_;  // Not owned.
+
+    DISALLOW_COPY_AND_ASSIGN(TestApi);
+  };
+
   explicit PowerButtonController(LockStateController* controller);
   ~PowerButtonController() override;
 
   void set_has_legacy_power_button_for_test(bool legacy) {
     has_legacy_power_button_ = legacy;
   }
 
   void set_enable_quick_lock_for_test(bool enable_quick_lock) {
     enable_quick_lock_ = enable_quick_lock;
   }
 
   // Called when the current screen brightness changes.
   void OnScreenBrightnessChanged(double percent);
 
   // 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);
 
   // ui::EventHandler:
   void OnKeyEvent(ui::KeyEvent* event) override;
+  void OnMouseEvent(ui::MouseEvent* event) override;
 
 #if defined(OS_CHROMEOS)
   // Overriden from ui::DisplayConfigurator::Observer:
   void OnDisplayModeChanged(
       const ui::DisplayConfigurator::DisplayStateList& outputs) override;
 
   // Overridden from chromeos::PowerManagerClient::Observer:
   void PowerButtonEventReceived(bool down,
                                 const base::TimeTicks& timestamp) override;
 #endif
 
  private:
-  friend class test::PowerButtonControllerTest;
+  // Called by OnPowerButtonEvent on convertible/tablet devices.
+  void OnTabletPowerButtonEvent(bool down, const base::TimeTicks& timestamp);
+
+  // Set backlights to |forced_off| if needed.
+  void SetBacklightsForcedOff(bool forced_off);
+
+  // Sends a request to powerd to get the backlights forced off state so that
+  // |backlights_forced_off_| can be initialized.
+  void GetInitialBacklightsForcedOff();
+
+  // Initializes |backlights_forced_off_|.
+  void OnGotInitialBacklightsForcedOff(bool is_forced_off);
+
+  // Starts |tablet_power_button_timer_| when the power button is pressed while
+  // in tablet mode.
+  void StartTabletPowerButtonTimer();
+
+  // Called by |tablet_power_button_timer_| to start the pre-shutdown animation.
+  void OnTabletPowerButtonTimeout();
+
+  // Locks the screen if the "require password to wake from sleep" pref is set
+  // and locking is possible.
+  void LockScreenIfRequired();
+
+  // Returns true if device is a convertible/tablet device or has
+  // kAshEnableTouchViewTesting in test, otherwise false.
+  bool IsTabletModeSupported() const;
+
+  // Returns true if device is currently in tablet/maximize mode, otherwise
+  // false.
+  bool IsTabletModeActive() const;
 
   // Are the power or lock buttons currently held?
   bool power_button_down_;
   bool lock_button_down_;
 
   // True when the volume down button is being held down.
   bool volume_down_pressed_;
 
 #if defined(OS_CHROMEOS)
   // Volume to be restored after a screenshot is taken by pressing the power
@@ skipping 11 matching lines @@
 
   // Was a command-line switch set telling us that we're running on hardware
   // that misreports power button releases?
   bool has_legacy_power_button_;
 
   // Enables quick, non-cancellable locking of the screen when in maximize mode.
   bool enable_quick_lock_;
 
   LockStateController* controller_;  // Not owned.
 
+  // True if the screen was off when the power button was pressed.
+  bool power_button_down_while_screen_off_;
+
+  // Current forced-off state of backlights.
+  bool backlights_forced_off_;
+
+  // Started when the tablet power button is pressed and stopped when it's
+  // released. Runs OnTabletPowerButtonTimeout() to start shutdown.
+  base::OneShotTimer tablet_power_button_timer_;
+
   DISALLOW_COPY_AND_ASSIGN(PowerButtonController);
 };
 
 }  // namespace ash
 
 #endif  // ASH_WM_POWER_BUTTON_CONTROLLER_H_