Fix issue where screen off is disabled after expanding then collapsing the shade.

The root cause is that we only get a callback to update whether we want to control screen off when the shade expansion changes, not when the isExpanding value changes. Sometimes, this value can still be true briefly when the callback is triggered.

We can safely fix this by checking isFullyCollapsed rather than isExpanded || isExpanding. isFullyCollapsed is updated at the same time as isExpanded, so it will be accurate when the callback is triggered.

Fixes: 194212105
Test: expand the shade, collapse it, press power button, see animation
Test: expand shade, press power, don't see animation
Test: press power while in the middle of expanding or collapsing the shade, see no animation
Change-Id: I0396aa4ee6408820d700ac23bea04897f37c4443

packages/SystemUI/src/com/android/systemui/statusbar/phone/UnlockedScreenOffAnimationController.kt

@@ -199,10 +199,9 @@ class UnlockedScreenOffAnimationController @Inject constructor(
 
         // We currently draw both the light reveal scrim, and the AOD UI, in the shade. If it's
         // already expanded and showing notifications/QS, the animation looks really messy. For now,
-        // disable it if the notification panel is expanded.
+        // disable it if the notification panel is not fully collapsed.
         if (!this::statusBar.isInitialized ||
-                statusBar.notificationPanelViewController.isFullyExpanded ||
-                statusBar.notificationPanelViewController.isExpanding) {
+                !statusBar.notificationPanelViewController.isFullyCollapsed) {
             return false
         }