SystemUI: finish circle battery animation after unplugging

Change-Id: I4a06323429625af7eaba334a3e799190f94a0555
Signed-off-by: Roman Birg <roman@cyngn.com>

1 files changed, 11 insertions, 9 deletions

diff --git a/packages/SystemUI/src/com/android/systemui/BatteryMeterView.java b/packages/SystemUI/src/com/android/systemui/BatteryMeterView.java
index 41d4aae..244b7f7 100755
--- a/packages/SystemUI/src/com/android/systemui/BatteryMeterView.java
+++ b/packages/SystemUI/src/com/android/systemui/BatteryMeterView.java
@@ -928,22 +928,24 @@ public class BatteryMeterView extends View implements DemoMode,
             if (!tracker.shouldIndicateCharging()
                     || tracker.status == BatteryManager.BATTERY_STATUS_FULL
                     || tracker.level == 0) {
-                if (mIsAnimating) {
-                    mIsAnimating = false;
-                    mAnimOffset = 0;
-                }
-                return;
+                mIsAnimating = false;
+            } else {
+                mIsAnimating = true;
             }
 
-            mIsAnimating = true;
-
             if (mAnimOffset > 360) {
                 mAnimOffset = 0;
-            } else {
+            }
+
+            boolean continueAnimation = mIsAnimating || mAnimOffset != 0;
+
+            if (continueAnimation) {
                 mAnimOffset += 3;
             }
 
-            postInvalidateDelayed(50);
+            if (continueAnimation) {
+                postInvalidateDelayed(50);
+            }
         }
 
         /**