Fix waitingForDrawn callback not called when adjusting for IME am: 3188386ff8

am: b2089aba28

Change-Id: I4381e9828e37d0e939390ee24da3395059a30e9a

services/core/java/com/android/server/wm/DockedStackDividerController.java

@@ -653,6 +653,14 @@ public class DockedStackDividerController implements DimLayerUser {
                 mDelayedImeWin = imeWin;
                 imeWin.mWinAnimator.startDelayingAnimationStart();
             }
+
+            // If we are already waiting for something to be drawn, clear out the old one so it
+            // still gets executed.
+            // TODO: Have a real system where we can wait on different windows to be drawn with
+            // different callbacks.
+            if (mService.mWaitingForDrawnCallback != null) {
+                mService.mWaitingForDrawnCallback.run();
+            }
             mService.mWaitingForDrawnCallback = () -> {
                 mAnimationStartDelayed = false;
                 if (mDelayedImeWin != null) {