diff options
12 files changed, 401 insertions, 10 deletions
diff --git a/services/core/java/com/android/server/am/BatteryStatsService.java b/services/core/java/com/android/server/am/BatteryStatsService.java index aee02997793b..ab6eaa8a8051 100644 --- a/services/core/java/com/android/server/am/BatteryStatsService.java +++ b/services/core/java/com/android/server/am/BatteryStatsService.java @@ -126,6 +126,7 @@ import com.android.server.power.stats.BatteryStatsImpl; import com.android.server.power.stats.BatteryUsageStatsProvider; import com.android.server.power.stats.CpuAggregatedPowerStatsProcessor; import com.android.server.power.stats.MobileRadioAggregatedPowerStatsProcessor; +import com.android.server.power.stats.PhoneCallAggregatedPowerStatsProcessor; import com.android.server.power.stats.PowerStatsAggregator; import com.android.server.power.stats.PowerStatsExporter; import com.android.server.power.stats.PowerStatsScheduler; @@ -489,6 +490,9 @@ public final class BatteryStatsService extends IBatteryStats.Stub AggregatedPowerStatsConfig.STATE_PROCESS_STATE) .setProcessor( new MobileRadioAggregatedPowerStatsProcessor(mPowerProfile)); + config.trackPowerComponent(BatteryConsumer.POWER_COMPONENT_PHONE, + BatteryConsumer.POWER_COMPONENT_MOBILE_RADIO) + .setProcessor(new PhoneCallAggregatedPowerStatsProcessor()); return config; } diff --git a/services/core/java/com/android/server/power/stats/AggregatedPowerStats.java b/services/core/java/com/android/server/power/stats/AggregatedPowerStats.java index dae83320482c..e1b4b88ed1df 100644 --- a/services/core/java/com/android/server/power/stats/AggregatedPowerStats.java +++ b/services/core/java/com/android/server/power/stats/AggregatedPowerStats.java @@ -69,7 +69,7 @@ class AggregatedPowerStats { aggregatedPowerStatsConfig.getPowerComponentsAggregatedStatsConfigs(); mPowerComponentStats = new PowerComponentAggregatedPowerStats[configs.size()]; for (int i = 0; i < configs.size(); i++) { - mPowerComponentStats[i] = new PowerComponentAggregatedPowerStats(configs.get(i)); + mPowerComponentStats[i] = new PowerComponentAggregatedPowerStats(this, configs.get(i)); } } diff --git a/services/core/java/com/android/server/power/stats/AggregatedPowerStatsConfig.java b/services/core/java/com/android/server/power/stats/AggregatedPowerStatsConfig.java index b625a5d5a59e..945b866a0d4a 100644 --- a/services/core/java/com/android/server/power/stats/AggregatedPowerStatsConfig.java +++ b/services/core/java/com/android/server/power/stats/AggregatedPowerStatsConfig.java @@ -85,6 +85,9 @@ public class AggregatedPowerStatsConfig { * Configures which states should be tracked as separate dimensions for the entire device. */ public PowerComponent trackDeviceStates(@TrackedState int... states) { + if (mTrackedDeviceStates != null) { + throw new IllegalStateException("Component is already configured"); + } mTrackedDeviceStates = states; return this; } @@ -93,6 +96,9 @@ public class AggregatedPowerStatsConfig { * Configures which states should be tracked as separate dimensions on a per-UID basis. */ public PowerComponent trackUidStates(@TrackedState int... states) { + if (mTrackedUidStates != null) { + throw new IllegalStateException("Component is already configured"); + } mTrackedUidStates = states; return this; } @@ -153,6 +159,7 @@ public class AggregatedPowerStatsConfig { } return false; } + } private final List<PowerComponent> mPowerComponents = new ArrayList<>(); @@ -168,6 +175,33 @@ public class AggregatedPowerStatsConfig { return builder; } + /** + * Creates a configuration for the specified power component, which is a subcomponent + * of a different power component. The tracked states will be the same as the parent + * component's. + */ + public PowerComponent trackPowerComponent(int powerComponentId, + int parentPowerComponentId) { + PowerComponent parent = null; + for (int i = 0; i < mPowerComponents.size(); i++) { + PowerComponent powerComponent = mPowerComponents.get(i); + if (powerComponent.getPowerComponentId() == parentPowerComponentId) { + parent = powerComponent; + break; + } + } + + if (parent == null) { + throw new IllegalArgumentException( + "Parent component " + parentPowerComponentId + " is not configured"); + } + + PowerComponent powerComponent = trackPowerComponent(powerComponentId); + powerComponent.mTrackedDeviceStates = parent.mTrackedDeviceStates; + powerComponent.mTrackedUidStates = parent.mTrackedUidStates; + return powerComponent; + } + public List<PowerComponent> getPowerComponentsAggregatedStatsConfigs() { return mPowerComponents; } diff --git a/services/core/java/com/android/server/power/stats/MobileRadioAggregatedPowerStatsProcessor.java b/services/core/java/com/android/server/power/stats/MobileRadioAggregatedPowerStatsProcessor.java index bc6fbcc98744..5eab7c591dcb 100644 --- a/services/core/java/com/android/server/power/stats/MobileRadioAggregatedPowerStatsProcessor.java +++ b/services/core/java/com/android/server/power/stats/MobileRadioAggregatedPowerStatsProcessor.java @@ -405,8 +405,7 @@ public class MobileRadioAggregatedPowerStatsProcessor extends AggregatedPowerSta return "idle: " + mStatsLayout.getDeviceIdleTime(stats) + " sleep: " + mStatsLayout.getDeviceSleepTime(stats) + " scan: " + mStatsLayout.getDeviceScanTime(stats) - + " power-data: " + mStatsLayout.getDevicePowerEstimate(stats) - + " power-call: " + mStatsLayout.getDeviceCallPowerEstimate(stats); + + " power: " + mStatsLayout.getDevicePowerEstimate(stats); } @Override diff --git a/services/core/java/com/android/server/power/stats/MobileRadioPowerStatsCollector.java b/services/core/java/com/android/server/power/stats/MobileRadioPowerStatsCollector.java index 098c7a7f7a60..e28f9a9d1c25 100644 --- a/services/core/java/com/android/server/power/stats/MobileRadioPowerStatsCollector.java +++ b/services/core/java/com/android/server/power/stats/MobileRadioPowerStatsCollector.java @@ -140,7 +140,7 @@ public class MobileRadioPowerStatsCollector extends PowerStatsCollector { MobileRadioStatsArrayLayout() {} MobileRadioStatsArrayLayout(@NonNull PowerStats.Descriptor descriptor) { - fromExtras(descriptor.extras); + super(descriptor); } void addDeviceMobileActivity() { diff --git a/services/core/java/com/android/server/power/stats/PhoneCallAggregatedPowerStatsProcessor.java b/services/core/java/com/android/server/power/stats/PhoneCallAggregatedPowerStatsProcessor.java new file mode 100644 index 000000000000..0222859bc802 --- /dev/null +++ b/services/core/java/com/android/server/power/stats/PhoneCallAggregatedPowerStatsProcessor.java @@ -0,0 +1,96 @@ +/* + * Copyright (C) 2024 The Android Open Source Project + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package com.android.server.power.stats; + +import android.os.BatteryConsumer; +import android.os.PersistableBundle; + +import com.android.internal.os.PowerStats; + +public class PhoneCallAggregatedPowerStatsProcessor extends AggregatedPowerStatsProcessor { + private final PowerStatsCollector.StatsArrayLayout mStatsArrayLayout; + private final PowerStats.Descriptor mDescriptor; + private final long[] mTmpDeviceStats; + private PowerStats.Descriptor mMobileRadioStatsDescriptor; + private MobileRadioPowerStatsCollector.MobileRadioStatsArrayLayout mMobileRadioStatsArrayLayout; + private long[] mTmpMobileRadioDeviceStats; + + public PhoneCallAggregatedPowerStatsProcessor() { + mStatsArrayLayout = new PowerStatsCollector.StatsArrayLayout(); + mStatsArrayLayout.addDeviceSectionPowerEstimate(); + PersistableBundle extras = new PersistableBundle(); + mStatsArrayLayout.toExtras(extras); + mDescriptor = new PowerStats.Descriptor(BatteryConsumer.POWER_COMPONENT_PHONE, + mStatsArrayLayout.getDeviceStatsArrayLength(), null, 0, 0, extras); + mTmpDeviceStats = new long[mDescriptor.statsArrayLength]; + } + + @Override + void finish(PowerComponentAggregatedPowerStats stats) { + stats.setPowerStatsDescriptor(mDescriptor); + + PowerComponentAggregatedPowerStats mobileRadioStats = + stats.getAggregatedPowerStats().getPowerComponentStats( + BatteryConsumer.POWER_COMPONENT_MOBILE_RADIO); + if (mobileRadioStats == null) { + return; + } + + if (mMobileRadioStatsDescriptor == null) { + mMobileRadioStatsDescriptor = mobileRadioStats.getPowerStatsDescriptor(); + if (mMobileRadioStatsDescriptor == null) { + return; + } + + mMobileRadioStatsArrayLayout = + new MobileRadioPowerStatsCollector.MobileRadioStatsArrayLayout( + mMobileRadioStatsDescriptor); + mTmpMobileRadioDeviceStats = new long[mMobileRadioStatsDescriptor.statsArrayLength]; + } + + MultiStateStats.States[] deviceStateConfig = + mobileRadioStats.getConfig().getDeviceStateConfig(); + + // Phone call power estimates have already been calculated by the mobile radio stats + // processor. All that remains to be done is copy the estimates over. + MultiStateStats.States.forEachTrackedStateCombination(deviceStateConfig, + states -> { + mobileRadioStats.getDeviceStats(mTmpMobileRadioDeviceStats, states); + double callPowerEstimate = + mMobileRadioStatsArrayLayout.getDeviceCallPowerEstimate( + mTmpMobileRadioDeviceStats); + mStatsArrayLayout.setDevicePowerEstimate(mTmpDeviceStats, callPowerEstimate); + stats.setDeviceStats(states, mTmpDeviceStats); + }); + } + + @Override + String deviceStatsToString(PowerStats.Descriptor descriptor, long[] stats) { + return "power: " + mStatsArrayLayout.getDevicePowerEstimate(stats); + } + + @Override + String stateStatsToString(PowerStats.Descriptor descriptor, int key, long[] stats) { + // Unsupported for this power component + return null; + } + + @Override + String uidStatsToString(PowerStats.Descriptor descriptor, long[] stats) { + // Unsupported for this power component + return null; + } +} diff --git a/services/core/java/com/android/server/power/stats/PowerComponentAggregatedPowerStats.java b/services/core/java/com/android/server/power/stats/PowerComponentAggregatedPowerStats.java index a438145a1587..6d58307dbefa 100644 --- a/services/core/java/com/android/server/power/stats/PowerComponentAggregatedPowerStats.java +++ b/services/core/java/com/android/server/power/stats/PowerComponentAggregatedPowerStats.java @@ -52,10 +52,12 @@ class PowerComponentAggregatedPowerStats { private static final long UNKNOWN = -1; public final int powerComponentId; - private final MultiStateStats.States[] mDeviceStateConfig; - private final MultiStateStats.States[] mUidStateConfig; + @NonNull + private final AggregatedPowerStats mAggregatedPowerStats; @NonNull private final AggregatedPowerStatsConfig.PowerComponent mConfig; + private final MultiStateStats.States[] mDeviceStateConfig; + private final MultiStateStats.States[] mUidStateConfig; private final int[] mDeviceStates; private MultiStateStats.Factory mStatsFactory; @@ -72,7 +74,9 @@ class PowerComponentAggregatedPowerStats { public MultiStateStats stats; } - PowerComponentAggregatedPowerStats(AggregatedPowerStatsConfig.PowerComponent config) { + PowerComponentAggregatedPowerStats(@NonNull AggregatedPowerStats aggregatedPowerStats, + @NonNull AggregatedPowerStatsConfig.PowerComponent config) { + mAggregatedPowerStats = aggregatedPowerStats; mConfig = config; powerComponentId = config.getPowerComponentId(); mDeviceStateConfig = config.getDeviceStateConfig(); @@ -82,6 +86,11 @@ class PowerComponentAggregatedPowerStats { } @NonNull + AggregatedPowerStats getAggregatedPowerStats() { + return mAggregatedPowerStats; + } + + @NonNull public AggregatedPowerStatsConfig.PowerComponent getConfig() { return mConfig; } @@ -91,6 +100,10 @@ class PowerComponentAggregatedPowerStats { return mPowerStatsDescriptor; } + public void setPowerStatsDescriptor(PowerStats.Descriptor powerStatsDescriptor) { + mPowerStatsDescriptor = powerStatsDescriptor; + } + void setState(@AggregatedPowerStatsConfig.TrackedState int stateId, int state, long timestampMs) { if (mDeviceStats == null) { @@ -143,6 +156,9 @@ class PowerComponentAggregatedPowerStats { } void setDeviceStats(@AggregatedPowerStatsConfig.TrackedState int[] states, long[] values) { + if (mDeviceStats == null) { + createDeviceStats(0); + } mDeviceStats.setStats(states, values); } diff --git a/services/core/java/com/android/server/power/stats/PowerStatsCollector.java b/services/core/java/com/android/server/power/stats/PowerStatsCollector.java index e969f4f99417..c8095f3a17db 100644 --- a/services/core/java/com/android/server/power/stats/PowerStatsCollector.java +++ b/services/core/java/com/android/server/power/stats/PowerStatsCollector.java @@ -83,6 +83,13 @@ public abstract class PowerStatsCollector { private int mDevicePowerEstimatePosition; private int mUidPowerEstimatePosition; + public StatsArrayLayout() { + } + + public StatsArrayLayout(PowerStats.Descriptor descriptor) { + fromExtras(descriptor.extras); + } + public int getDeviceStatsArrayLength() { return mDeviceStatsArrayLength; } diff --git a/services/tests/powerstatstests/Android.bp b/services/tests/powerstatstests/Android.bp index f3c56e213254..bce6a1296ee9 100644 --- a/services/tests/powerstatstests/Android.bp +++ b/services/tests/powerstatstests/Android.bp @@ -45,6 +45,7 @@ filegroup { "src/com/android/server/power/stats/MobileRadioPowerCalculatorTest.java", "src/com/android/server/power/stats/MobileRadioPowerStatsCollectorTest.java", "src/com/android/server/power/stats/MultiStateStatsTest.java", + "src/com/android/server/power/stats/PhoneCallAggregatedPowerStatsProcessorTest.java", "src/com/android/server/power/stats/PowerStatsAggregatorTest.java", "src/com/android/server/power/stats/PowerStatsCollectorTest.java", "src/com/android/server/power/stats/PowerStatsExporterTest.java", diff --git a/services/tests/powerstatstests/src/com/android/server/power/stats/CpuAggregatedPowerStatsProcessorTest.java b/services/tests/powerstatstests/src/com/android/server/power/stats/CpuAggregatedPowerStatsProcessorTest.java index 2f13eec5cea5..7ed4a82ef5c5 100644 --- a/services/tests/powerstatstests/src/com/android/server/power/stats/CpuAggregatedPowerStatsProcessorTest.java +++ b/services/tests/powerstatstests/src/com/android/server/power/stats/CpuAggregatedPowerStatsProcessorTest.java @@ -30,6 +30,7 @@ import static com.android.server.power.stats.AggregatedPowerStatsConfig.STATE_SC import static com.google.common.truth.Truth.assertThat; import static org.junit.Assert.fail; +import static org.mockito.Mockito.mock; import android.os.BatteryConsumer; import android.os.PersistableBundle; @@ -207,7 +208,7 @@ public class CpuAggregatedPowerStatsProcessorTest { MockPowerComponentAggregatedPowerStats(AggregatedPowerStatsConfig.PowerComponent config, boolean useEnergyConsumers) { - super(config); + super(new AggregatedPowerStats(mock(AggregatedPowerStatsConfig.class)), config); mStatsLayout = new CpuPowerStatsCollector.CpuStatsArrayLayout(); mStatsLayout.addDeviceSectionCpuTimeByScalingStep(3); mStatsLayout.addDeviceSectionCpuTimeByCluster(2); diff --git a/services/tests/powerstatstests/src/com/android/server/power/stats/MobileRadioAggregatedPowerStatsProcessorTest.java b/services/tests/powerstatstests/src/com/android/server/power/stats/MobileRadioAggregatedPowerStatsProcessorTest.java index afb48270491e..089981e11b9a 100644 --- a/services/tests/powerstatstests/src/com/android/server/power/stats/MobileRadioAggregatedPowerStatsProcessorTest.java +++ b/services/tests/powerstatstests/src/com/android/server/power/stats/MobileRadioAggregatedPowerStatsProcessorTest.java @@ -178,7 +178,8 @@ public class MobileRadioAggregatedPowerStatsProcessorTest { .setProcessor(processor); PowerComponentAggregatedPowerStats aggregatedStats = - new PowerComponentAggregatedPowerStats(config); + new PowerComponentAggregatedPowerStats( + new AggregatedPowerStats(mock(AggregatedPowerStatsConfig.class)), config); aggregatedStats.setState(STATE_POWER, POWER_STATE_OTHER, 0); aggregatedStats.setState(STATE_SCREEN, SCREEN_STATE_ON, 0); @@ -323,7 +324,8 @@ public class MobileRadioAggregatedPowerStatsProcessorTest { .setProcessor(processor); PowerComponentAggregatedPowerStats aggregatedStats = - new PowerComponentAggregatedPowerStats(config); + new PowerComponentAggregatedPowerStats( + new AggregatedPowerStats(mock(AggregatedPowerStatsConfig.class)), config); aggregatedStats.setState(STATE_POWER, POWER_STATE_OTHER, 0); aggregatedStats.setState(STATE_SCREEN, SCREEN_STATE_ON, 0); diff --git a/services/tests/powerstatstests/src/com/android/server/power/stats/PhoneCallAggregatedPowerStatsProcessorTest.java b/services/tests/powerstatstests/src/com/android/server/power/stats/PhoneCallAggregatedPowerStatsProcessorTest.java new file mode 100644 index 000000000000..58924d99c480 --- /dev/null +++ b/services/tests/powerstatstests/src/com/android/server/power/stats/PhoneCallAggregatedPowerStatsProcessorTest.java @@ -0,0 +1,231 @@ +/* + * Copyright (C) 2024 The Android Open Source Project + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package com.android.server.power.stats; + +import static com.android.server.power.stats.AggregatedPowerStatsConfig.POWER_STATE_OTHER; +import static com.android.server.power.stats.AggregatedPowerStatsConfig.SCREEN_STATE_ON; +import static com.android.server.power.stats.AggregatedPowerStatsConfig.SCREEN_STATE_OTHER; +import static com.android.server.power.stats.AggregatedPowerStatsConfig.STATE_POWER; +import static com.android.server.power.stats.AggregatedPowerStatsConfig.STATE_PROCESS_STATE; +import static com.android.server.power.stats.AggregatedPowerStatsConfig.STATE_SCREEN; + +import static com.google.common.truth.Truth.assertThat; + +import static org.mockito.ArgumentMatchers.any; +import static org.mockito.ArgumentMatchers.anyInt; +import static org.mockito.Mockito.doAnswer; +import static org.mockito.Mockito.when; + +import android.content.Context; +import android.content.pm.PackageManager; +import android.hardware.power.stats.EnergyConsumerType; +import android.net.NetworkStats; +import android.os.BatteryConsumer; +import android.os.Handler; +import android.os.OutcomeReceiver; +import android.platform.test.ravenwood.RavenwoodRule; +import android.telephony.ModemActivityInfo; +import android.telephony.TelephonyManager; + +import com.android.internal.os.Clock; + +import org.junit.Before; +import org.junit.Rule; +import org.junit.Test; +import org.mockito.Mock; +import org.mockito.MockitoAnnotations; + +import java.util.function.IntSupplier; +import java.util.function.LongSupplier; +import java.util.function.Supplier; + +public class PhoneCallAggregatedPowerStatsProcessorTest { + @Rule(order = 0) + public final RavenwoodRule mRavenwood = new RavenwoodRule.Builder() + .setProvideMainThread(true) + .build(); + + private static final double PRECISION = 0.00001; + private static final int VOLTAGE_MV = 3500; + + @Rule(order = 1) + public final BatteryUsageStatsRule mStatsRule = new BatteryUsageStatsRule(); + @Mock + private Context mContext; + @Mock + private PowerStatsUidResolver mPowerStatsUidResolver; + @Mock + private PackageManager mPackageManager; + @Mock + private PowerStatsCollector.ConsumedEnergyRetriever mConsumedEnergyRetriever; + @Mock + private Supplier<NetworkStats> mNetworkStatsSupplier; + @Mock + private TelephonyManager mTelephonyManager; + @Mock + private LongSupplier mCallDurationSupplier; + @Mock + private LongSupplier mScanDurationSupplier; + + private final MobileRadioPowerStatsCollector.Injector mInjector = + new MobileRadioPowerStatsCollector.Injector() { + @Override + public Handler getHandler() { + return mStatsRule.getHandler(); + } + + @Override + public Clock getClock() { + return mStatsRule.getMockClock(); + } + + @Override + public PowerStatsUidResolver getUidResolver() { + return mPowerStatsUidResolver; + } + + @Override + public PackageManager getPackageManager() { + return mPackageManager; + } + + @Override + public PowerStatsCollector.ConsumedEnergyRetriever getConsumedEnergyRetriever() { + return mConsumedEnergyRetriever; + } + + @Override + public IntSupplier getVoltageSupplier() { + return () -> VOLTAGE_MV; + } + + @Override + public Supplier<NetworkStats> getMobileNetworkStatsSupplier() { + return mNetworkStatsSupplier; + } + + @Override + public TelephonyManager getTelephonyManager() { + return mTelephonyManager; + } + + @Override + public LongSupplier getCallDurationSupplier() { + return mCallDurationSupplier; + } + + @Override + public LongSupplier getPhoneSignalScanDurationSupplier() { + return mScanDurationSupplier; + } + }; + + @Before + public void setup() { + MockitoAnnotations.initMocks(this); + + when(mContext.getPackageManager()).thenReturn(mPackageManager); + when(mPackageManager.hasSystemFeature(PackageManager.FEATURE_TELEPHONY)).thenReturn(true); + when(mPowerStatsUidResolver.mapUid(anyInt())) + .thenAnswer(invocation -> invocation.getArgument(0)); + + // No power monitoring hardware + when(mConsumedEnergyRetriever.getEnergyConsumerIds(EnergyConsumerType.MOBILE_RADIO)) + .thenReturn(new int[0]); + + mStatsRule.setTestPowerProfile("power_profile_test_legacy_modem"); + } + + @Test + public void copyEstimatesFromMobileRadioPowerStats() { + MobileRadioAggregatedPowerStatsProcessor mobileStatsProcessor = + new MobileRadioAggregatedPowerStatsProcessor(mStatsRule.getPowerProfile()); + + PhoneCallAggregatedPowerStatsProcessor phoneStatsProcessor = + new PhoneCallAggregatedPowerStatsProcessor(); + + AggregatedPowerStatsConfig aggregatedPowerStatsConfig = new AggregatedPowerStatsConfig(); + aggregatedPowerStatsConfig.trackPowerComponent(BatteryConsumer.POWER_COMPONENT_MOBILE_RADIO) + .trackDeviceStates(STATE_POWER, STATE_SCREEN) + .trackUidStates(STATE_POWER, STATE_SCREEN, STATE_PROCESS_STATE) + .setProcessor(mobileStatsProcessor); + aggregatedPowerStatsConfig.trackPowerComponent(BatteryConsumer.POWER_COMPONENT_PHONE, + BatteryConsumer.POWER_COMPONENT_MOBILE_RADIO) + .setProcessor(phoneStatsProcessor); + + AggregatedPowerStats aggregatedPowerStats = + new AggregatedPowerStats(aggregatedPowerStatsConfig); + PowerComponentAggregatedPowerStats mobileRadioStats = + aggregatedPowerStats.getPowerComponentStats( + BatteryConsumer.POWER_COMPONENT_MOBILE_RADIO); + + aggregatedPowerStats.setDeviceState(STATE_POWER, POWER_STATE_OTHER, 0); + aggregatedPowerStats.setDeviceState(STATE_SCREEN, SCREEN_STATE_ON, 0); + + MobileRadioPowerStatsCollector collector = new MobileRadioPowerStatsCollector(mInjector, 0); + collector.setEnabled(true); + + // Initial empty ModemActivityInfo. + mockModemActivityInfo(new ModemActivityInfo(0L, 0L, 0L, new int[5], 0L)); + + // Establish a baseline + aggregatedPowerStats.addPowerStats(collector.collectStats(), 0); + + // Turn the screen off after 2.5 seconds + aggregatedPowerStats.setDeviceState(STATE_SCREEN, SCREEN_STATE_OTHER, 2500); + + ModemActivityInfo mai = new ModemActivityInfo(10000, 2000, 3000, + new int[]{100, 200, 300, 400, 500}, 600); + mockModemActivityInfo(mai); + + // A phone call was made + when(mCallDurationSupplier.getAsLong()).thenReturn(7000L); + + mStatsRule.setTime(10_000, 10_000); + + aggregatedPowerStats.addPowerStats(collector.collectStats(), 10_000); + + mobileStatsProcessor.finish(mobileRadioStats); + + PowerComponentAggregatedPowerStats stats = + aggregatedPowerStats.getPowerComponentStats(BatteryConsumer.POWER_COMPONENT_PHONE); + phoneStatsProcessor.finish(stats); + + PowerStatsCollector.StatsArrayLayout statsLayout = + new PowerStatsCollector.StatsArrayLayout(stats.getPowerStatsDescriptor()); + + long[] deviceStats = new long[stats.getPowerStatsDescriptor().statsArrayLength]; + stats.getDeviceStats(deviceStats, states(POWER_STATE_OTHER, SCREEN_STATE_ON)); + assertThat(statsLayout.getDevicePowerEstimate(deviceStats)) + .isWithin(PRECISION).of(0.7); + stats.getDeviceStats(deviceStats, states(POWER_STATE_OTHER, SCREEN_STATE_OTHER)); + assertThat(statsLayout.getDevicePowerEstimate(deviceStats)) + .isWithin(PRECISION).of(2.1); + } + + private void mockModemActivityInfo(ModemActivityInfo emptyMai) { + doAnswer(invocation -> { + OutcomeReceiver<ModemActivityInfo, TelephonyManager.ModemActivityInfoException> + receiver = invocation.getArgument(1); + receiver.onResult(emptyMai); + return null; + }).when(mTelephonyManager).requestModemActivityInfo(any(), any()); + } + + private int[] states(int... states) { + return states; + } +} |