Add support for power and screen state for custom power components

Bug: 352833173
Bug: 352835319
Test: atest PowerStatsTests; atest PowerStatsTestRavenwood
Flag: com.android.server.power.optimization.battery_usage_stats_by_power_and_screen_state
Change-Id: I8db026a4838024574b9357fda4ddbaca0da4522c