Recover AOD state when leaving battery saver

We need to anable AOD again when the device is placed on a charger and
AOD was disabled by battery saver.

Test: manual
Fixes: 146879388
Change-Id: Iea0a48d8eb51239b751941bf76eb2630a86ceb11

packages/SystemUI/src/com/android/systemui/doze/DozeTriggers.java

@@ -490,6 +490,9 @@ public class DozeTriggers implements DozeMachine.Part {
         public void onPowerSaveChanged(boolean active) {
             if (mDozeHost.isPowerSaveActive()) {
                 mMachine.requestState(DozeMachine.State.DOZE);
+            } else if (mMachine.getState() == DozeMachine.State.DOZE
+                    && mConfig.alwaysOnEnabled(UserHandle.USER_CURRENT)) {
+                mMachine.requestState(DozeMachine.State.DOZE_AOD);
             }
         }