WM changes to support new screenshot functionality.

Bug: 251205791
Test: CTS test
Change-Id: Ib854829b9aca4c2f5c54afa99a49701c95170a49
5 files changed