Make UMO follow squishing QQS

Before this change UMO used an estimation of when the squishness looks
like, which sometimes was a bit off. Now UMO follows the actual height
of the QQS during the transition.

Flag: com.android.systemui.scene_container
Bug: 345467289
Bug: 353519527
Test: manual on the foldable, tablet and phone
Change-Id: I6b0c9188123292870d8e07945b9676bd3c065318