From 7fe38c4864537c15f38f4acd3c29b3eed4c404b0 Mon Sep 17 00:00:00 2001
From: Jing Ji <jji@google.com>
Date: Wed, 19 Jan 2022 01:42:20 -0800
Subject: [PATCH] Include process state data in battery usage stats snapshots.

As the background current monitoring would need to access that.

Bug: 203105544
Test: FrameworksCoreTests:com.android.internal.os
Change-Id: Id5c04c4910054675d7f6e989f8eabf4321466e39
---
 core/java/com/android/internal/os/BatteryUsageStatsStore.java | 1 +
 1 file changed, 1 insertion(+)

diff --git a/core/java/com/android/internal/os/BatteryUsageStatsStore.java b/core/java/com/android/internal/os/BatteryUsageStatsStore.java
index fd54b32bd12ff..af82f40013d89 100644
--- a/core/java/com/android/internal/os/BatteryUsageStatsStore.java
+++ b/core/java/com/android/internal/os/BatteryUsageStatsStore.java
@@ -58,6 +58,7 @@ public class BatteryUsageStatsStore {
             new BatteryUsageStatsQuery.Builder()
                     .setMaxStatsAgeMs(0)
                     .includePowerModels()
+                    .includeProcessStateData()
                     .build());
     private static final String BATTERY_USAGE_STATS_DIR = "battery-usage-stats";
     private static final String SNAPSHOT_FILE_EXTENSION = ".bus";