Make screenshots use the focused display for some screenshot types

Modify FocusedDisplayRepository to make testing easier.

Bug: 362720389
Test: atest TakeScreenshotExecutorTest
Flag: com.android.systemui.screenshot_multidisplay_focus_change
Change-Id: Ie8326399b23c5bc7aa668981a4bfe38f4c21a926