1 /*
2  * Copyright (C) 2015 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package com.android.internal.os;
17 
18 import android.hardware.Sensor;
19 import android.hardware.SensorManager;
20 import android.os.BatteryConsumer;
21 import android.os.BatteryStats;
22 import android.os.BatteryUsageStats;
23 import android.os.BatteryUsageStatsQuery;
24 import android.os.UidBatteryConsumer;
25 import android.util.SparseArray;
26 
27 import java.util.List;
28 
29 public class SensorPowerCalculator extends PowerCalculator {
30     private final SparseArray<Sensor> mSensors;
31 
SensorPowerCalculator(SensorManager sensorManager)32     public SensorPowerCalculator(SensorManager sensorManager) {
33         List<Sensor> sensors = sensorManager.getSensorList(Sensor.TYPE_ALL);
34         mSensors = new SparseArray<>(sensors.size());
35         for (int i = 0; i < sensors.size(); i++) {
36             Sensor sensor = sensors.get(i);
37             mSensors.put(sensor.getHandle(), sensor);
38         }
39     }
40 
41     @Override
calculate(BatteryUsageStats.Builder builder, BatteryStats batteryStats, long rawRealtimeUs, long rawUptimeUs, BatteryUsageStatsQuery query)42     public void calculate(BatteryUsageStats.Builder builder, BatteryStats batteryStats,
43             long rawRealtimeUs, long rawUptimeUs, BatteryUsageStatsQuery query) {
44         double appsPowerMah = 0;
45         final SparseArray<UidBatteryConsumer.Builder> uidBatteryConsumerBuilders =
46                 builder.getUidBatteryConsumerBuilders();
47         for (int i = uidBatteryConsumerBuilders.size() - 1; i >= 0; i--) {
48             final UidBatteryConsumer.Builder app = uidBatteryConsumerBuilders.valueAt(i);
49             appsPowerMah += calculateApp(app, app.getBatteryStatsUid(), rawRealtimeUs);
50         }
51 
52         builder.getAggregateBatteryConsumerBuilder(
53                 BatteryUsageStats.AGGREGATE_BATTERY_CONSUMER_SCOPE_DEVICE)
54                 .setConsumedPower(BatteryConsumer.POWER_COMPONENT_SENSORS, appsPowerMah);
55         builder.getAggregateBatteryConsumerBuilder(
56                 BatteryUsageStats.AGGREGATE_BATTERY_CONSUMER_SCOPE_ALL_APPS)
57                 .setConsumedPower(BatteryConsumer.POWER_COMPONENT_SENSORS, appsPowerMah);
58     }
59 
calculateApp(UidBatteryConsumer.Builder app, BatteryStats.Uid u, long rawRealtimeUs)60     private double calculateApp(UidBatteryConsumer.Builder app, BatteryStats.Uid u,
61             long rawRealtimeUs) {
62         final double powerMah = calculatePowerMah(u, rawRealtimeUs,
63                 BatteryStats.STATS_SINCE_CHARGED);
64         app.setUsageDurationMillis(BatteryConsumer.POWER_COMPONENT_SENSORS,
65                         calculateDuration(u, rawRealtimeUs, BatteryStats.STATS_SINCE_CHARGED))
66                 .setConsumedPower(BatteryConsumer.POWER_COMPONENT_SENSORS, powerMah);
67         return powerMah;
68     }
69 
70     @Override
calculateApp(BatterySipper app, BatteryStats.Uid u, long rawRealtimeUs, long rawUptimeUs, int statsType)71     protected void calculateApp(BatterySipper app, BatteryStats.Uid u, long rawRealtimeUs,
72             long rawUptimeUs, int statsType) {
73         app.sensorPowerMah = calculatePowerMah(u, rawRealtimeUs, statsType);
74     }
75 
calculateDuration(BatteryStats.Uid u, long rawRealtimeUs, int statsType)76     private long calculateDuration(BatteryStats.Uid u, long rawRealtimeUs, int statsType) {
77         long durationMs = 0;
78         final SparseArray<? extends BatteryStats.Uid.Sensor> sensorStats = u.getSensorStats();
79         final int NSE = sensorStats.size();
80         for (int ise = 0; ise < NSE; ise++) {
81             final int sensorHandle = sensorStats.keyAt(ise);
82             if (sensorHandle == BatteryStats.Uid.Sensor.GPS) {
83                 continue;
84             }
85 
86             final BatteryStats.Uid.Sensor sensor = sensorStats.valueAt(ise);
87             final BatteryStats.Timer timer = sensor.getSensorTime();
88             durationMs += timer.getTotalTimeLocked(rawRealtimeUs, statsType) / 1000;
89         }
90         return durationMs;
91     }
92 
calculatePowerMah(BatteryStats.Uid u, long rawRealtimeUs, int statsType)93     private double calculatePowerMah(BatteryStats.Uid u, long rawRealtimeUs, int statsType) {
94         double powerMah = 0;
95         final SparseArray<? extends BatteryStats.Uid.Sensor> sensorStats = u.getSensorStats();
96         final int count = sensorStats.size();
97         for (int ise = 0; ise < count; ise++) {
98             final int sensorHandle = sensorStats.keyAt(ise);
99             // TODO(b/178127364): remove BatteryStats.Uid.Sensor.GPS and references to it.
100             if (sensorHandle == BatteryStats.Uid.Sensor.GPS) {
101                 continue;
102             }
103 
104             final BatteryStats.Uid.Sensor sensor = sensorStats.valueAt(ise);
105             final BatteryStats.Timer timer = sensor.getSensorTime();
106             final long sensorTime = timer.getTotalTimeLocked(rawRealtimeUs, statsType) / 1000;
107             if (sensorTime != 0) {
108                 Sensor s = mSensors.get(sensorHandle);
109                 if (s != null) {
110                     powerMah += (sensorTime * s.getPower()) / (1000 * 60 * 60);
111                 }
112             }
113         }
114         return powerMah;
115     }
116 }
117