Allow retrieving a window from the current state during transition

This is necessary to determine, for example, the location of the PIP window for interaction

Bug: 171049720
Test: atest FlickerLibTest
Change-Id: Ic5f48456a67745b339b619616333abc700f3e963