Delay display DOZE on LS => AOD

ScreenOffAnimationController only handles the
states from UNLOCKED => AOD, so we weren't delaying
the display state change to DOZE when we were animating
from LS => AOD.

Test: manual
Fixes: 219030172
Change-Id: Ia0aff7c634a01855db02d01a194db7850554b77b

packages/SystemUI/src/com/android/systemui/statusbar/phone/DozeParameters.java

@@ -309,7 +309,12 @@ public class DozeParameters implements
      * delayed for a few seconds. This might be useful to play animations without reducing FPS.
      */
     public boolean shouldDelayDisplayDozeTransition() {
-        return mScreenOffAnimationController.shouldDelayDisplayDozeTransition();
+        return willAnimateFromLockScreenToAod()
+                || mScreenOffAnimationController.shouldDelayDisplayDozeTransition();
+    }
+
+    private boolean willAnimateFromLockScreenToAod() {
+        return getAlwaysOn() && mKeyguardShowing;
     }
 
     /**