Handles users rotating screen too early

ash/display/display_configuration_controller.h

 // Copyright 2016 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_DISPLAY_DISPLAY_CONFIGURATION_CONTROLLER_H_
 #define ASH_DISPLAY_DISPLAY_CONFIGURATION_CONTROLLER_H_
 
 #include <memory>
 
 #include "ash/ash_export.h"
 #include "ash/display/window_tree_host_manager.h"
+#include "ash/rotator/screen_rotation_animator.h"

[bruthig, 2017/03/10 22:34:07]

super-nit: If ScreenRotationAnimatorObserver was in it's own .h/.cc then I think
you'd be able to forward declare the ScreenRotationAnimator, which would
potentially speed up build times.

[wutao, 2017/03/14 16:46:37]

Moved ScreenRotationAnimatorObserver to it's own .h.

 #include "base/macros.h"
 #include "base/memory/weak_ptr.h"
 #include "ui/display/display.h"
 
 namespace display {
 class DisplayLayout;
 class DisplayManager;
 }
 
 namespace ash {
 
 namespace test {
-class ShellTestApi;
+class DisplayConfigurationControllerTestApi;
 }  // namespace test
 
 class DisplayAnimator;
 
 // This class controls Display related configuration. Specifically it:
 // * Handles animated transitions where appropriate.
 // * Limits the frequency of certain operations.
 // * Provides a single interface for UI and API classes.
 // * TODO: Forwards display configuration changed events to UI and API classes.
 class ASH_EXPORT DisplayConfigurationController
-    : public WindowTreeHostManager::Observer {
+    : public WindowTreeHostManager::Observer,
+      public ScreenRotationAnimator::ScreenRotationAnimatorObserver {
  public:
   DisplayConfigurationController(
       display::DisplayManager* display_manager,
       WindowTreeHostManager* window_tree_host_manager);
   ~DisplayConfigurationController() override;
 
   // Sets the layout for the current displays with a fade in/out
   // animation. Currently |display_id| is assumed to be the secondary
   // display.  TODO(oshima/stevenjb): Support 3+ displays.
   void SetDisplayLayout(std::unique_ptr<display::DisplayLayout> layout,
                         bool user_action);
 
   // Sets the mirror mode with a fade-in/fade-out animation. Affects all
   // displays.
   void SetMirrorMode(bool mirror, bool user_action);
 
   // Sets the display's rotation with animation if available.
   void SetDisplayRotation(int64_t display_id,
                           display::Display::Rotation rotation,
                           display::Display::RotationSource source);
 
   // Sets the primary display id.
   void SetPrimaryDisplayId(int64_t display_id, bool user_action);
 
   // WindowTreeHostManager::Observer
   void OnDisplayConfigurationChanged() override;
 
+  // ScreenRotationAnimator::ScreenRotationAnimatorObserver
+  void OnEndedOrAbortedScreenRotationAnimation(
+      ScreenRotationAnimator* animator) override;
+
  protected:
-  friend class ash::test::ShellTestApi;
+  friend class ash::test::DisplayConfigurationControllerTestApi;
 
   // Allow tests to skip animations.
   void ResetAnimatorForTest();
 
  private:
   class DisplayChangeLimiter;
 
   // Sets the timeout for the DisplayChangeLimiter if it exists. Call this
   // *before* starting any animations.
   void SetThrottleTimeout(int64_t throttle_ms);
   bool IsLimited();
   void SetDisplayLayoutImpl(std::unique_ptr<display::DisplayLayout> layout);
   void SetMirrorModeImpl(bool mirror);
   void SetPrimaryDisplayIdImpl(int64_t display_id);
 
+  // Returns true if the screen rotation animation can be completed
+  // successfully. For example an animation is not possible if |display_id_|
+  // specifies a display::Display that is not currently active. See
+  // www.crbug.com/479503.
+  bool CanAnimateScreenRotation(int64_t display_id) const;
+
+  // Returns the currently active rotation for the display with |display_id|. If
+  // an animation is in progress, the target rotation will be returned.
+  display::Display::Rotation GetCurrentScreenRotation(int64_t display_id) const;
+
+  ScreenRotationAnimator* GetScreenRotationAnimatorForDisplay(

[bruthig, 2017/03/10 22:34:07]

nit: Can you doc this method that it creates one if it doesn't exist.

[wutao, 2017/03/14 16:46:37]

Done.

+      int64_t display_id);
+
   display::DisplayManager* display_manager_;         // weak ptr
   WindowTreeHostManager* window_tree_host_manager_;  // weak ptr
   std::unique_ptr<DisplayAnimator> display_animator_;
   std::unique_ptr<DisplayChangeLimiter> limiter_;
 
+  // Tracks |display_id| to ScreenRotationAnimator mappings. The
+  // |rotation_animator_map_| is populated on demand the first time a
+  // ScreenRotationAnimator is needed for a given |display_id|.
+  // On animation ended or aborted, the animator may be deleted if there is no
+  // more pending rotation request.
+  std::unordered_map<int64_t, std::unique_ptr<ScreenRotationAnimator>>
+      rotation_animator_map_;
+
   base::WeakPtrFactory<DisplayConfigurationController> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(DisplayConfigurationController);
 };
 
 }  // namespace ash
 
 #endif  // ASH_DISPLAY_DISPLAY_CONFIGURATION_CONTROLLER_H_