Add support for AndroidX to ScreenOffAnimationGuard

Also adds a bit of leeway to not report on frames close to screen off.

Bug: 420685896
Test: ran ScreenOffAnimationGuardTest
Flag: com.android.systemui.screen_off_animation_guard_enabled

Change-Id: I3f027efcee4adc3c86727ffcfbc9c8236657c9b0