1 /*
2  * Copyright (C) 2020 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 
17 #include <PowerStatsAidl.h>
18 #include <ZumaCommonDataProviders.h>
19 #include <AocStateResidencyDataProvider.h>
20 #include <CpupmStateResidencyDataProvider.h>
21 #include <DevfreqStateResidencyDataProvider.h>
22 #include <DisplayMrrStateResidencyDataProvider.h>
23 #include <AdaptiveDvfsStateResidencyDataProvider.h>
24 #include <TpuDvfsStateResidencyDataProvider.h>
25 #include <UfsStateResidencyDataProvider.h>
26 #include <dataproviders/GenericStateResidencyDataProvider.h>
27 #include <dataproviders/IioEnergyMeterDataProvider.h>
28 #include <dataproviders/PowerStatsEnergyConsumer.h>
29 #include <dataproviders/PowerStatsEnergyAttribution.h>
30 #include <dataproviders/PixelStateResidencyDataProvider.h>
31 
32 #include <android-base/logging.h>
33 #include <android-base/properties.h>
34 #include <android/binder_manager.h>
35 #include <android/binder_process.h>
36 #include <log/log.h>
37 #include <sys/stat.h>
38 
39 using aidl::android::hardware::power::stats::AdaptiveDvfsStateResidencyDataProvider;
40 using aidl::android::hardware::power::stats::AocStateResidencyDataProvider;
41 using aidl::android::hardware::power::stats::CpupmStateResidencyDataProvider;
42 using aidl::android::hardware::power::stats::DevfreqStateResidencyDataProvider;
43 using aidl::android::hardware::power::stats::DisplayMrrStateResidencyDataProvider;
44 using aidl::android::hardware::power::stats::DvfsStateResidencyDataProvider;
45 using aidl::android::hardware::power::stats::UfsStateResidencyDataProvider;
46 using aidl::android::hardware::power::stats::EnergyConsumerType;
47 using aidl::android::hardware::power::stats::GenericStateResidencyDataProvider;
48 using aidl::android::hardware::power::stats::IioEnergyMeterDataProvider;
49 using aidl::android::hardware::power::stats::PixelStateResidencyDataProvider;
50 using aidl::android::hardware::power::stats::PowerStatsEnergyConsumer;
51 using aidl::android::hardware::power::stats::TpuDvfsStateResidencyDataProvider;
52 
53 // TODO (b/181070764) (b/182941084):
54 // Remove this when Wifi/BT energy consumption models are available or revert before ship
55 using aidl::android::hardware::power::stats::EnergyConsumerResult;
56 using aidl::android::hardware::power::stats::Channel;
57 using aidl::android::hardware::power::stats::EnergyMeasurement;
58 class PlaceholderEnergyConsumer : public PowerStats::IEnergyConsumer {
59   public:
PlaceholderEnergyConsumer(std::shared_ptr<PowerStats> p,EnergyConsumerType type,std::string name)60     PlaceholderEnergyConsumer(std::shared_ptr<PowerStats> p, EnergyConsumerType type,
61             std::string name) : kType(type), kName(name), mPowerStats(p), mChannelId(-1) {
62         std::vector<Channel> channels;
63         mPowerStats->getEnergyMeterInfo(&channels);
64 
65         for (const auto &c : channels) {
66             if (c.name == "VSYS_PWR_WLAN_BT") {
67                 mChannelId = c.id;
68                 break;
69             }
70         }
71     }
getInfo()72     std::pair<EnergyConsumerType, std::string> getInfo() override { return {kType, kName}; }
73 
getEnergyConsumed()74     std::optional<EnergyConsumerResult> getEnergyConsumed() override {
75         int64_t totalEnergyUWs = 0;
76         int64_t timestampMs = 0;
77         if (mChannelId != -1) {
78             std::vector<EnergyMeasurement> measurements;
79             if (mPowerStats->readEnergyMeter({mChannelId}, &measurements).isOk()) {
80                 for (const auto &m : measurements) {
81                     totalEnergyUWs += m.energyUWs;
82                     timestampMs = m.timestampMs;
83                 }
84             } else {
85                 LOG(ERROR) << "Failed to read energy meter";
86                 return {};
87             }
88         }
89 
90         return EnergyConsumerResult{.timestampMs = timestampMs,
91                                 .energyUWs = totalEnergyUWs>>1};
92     }
93 
getConsumerName()94     std::string getConsumerName() override {
95         return kName;
96     };
97 
98   private:
99     const EnergyConsumerType kType;
100     const std::string kName;
101     std::shared_ptr<PowerStats> mPowerStats;
102     int32_t mChannelId;
103 };
104 
addPlaceholderEnergyConsumers(std::shared_ptr<PowerStats> p)105 void addPlaceholderEnergyConsumers(std::shared_ptr<PowerStats> p) {
106     p->addEnergyConsumer(
107             std::make_unique<PlaceholderEnergyConsumer>(p, EnergyConsumerType::WIFI, "Wifi"));
108     p->addEnergyConsumer(
109             std::make_unique<PlaceholderEnergyConsumer>(p, EnergyConsumerType::BLUETOOTH, "BT"));
110 }
111 
addAoC(std::shared_ptr<PowerStats> p)112 void addAoC(std::shared_ptr<PowerStats> p) {
113     // AoC clock is synced from "libaoc.c"
114     static const uint64_t AOC_CLOCK = 24576;
115     std::string base = "/sys/devices/platform/17000000.aoc/";
116     std::string prefix = base + "control/";
117 
118     // Add AoC cores (a32, ff1, hf0, and hf1)
119     std::vector<std::pair<std::string, std::string>> coreIds = {
120             {"AoC-A32", prefix + "a32_"},
121             {"AoC-FF1", prefix + "ff1_"},
122             {"AoC-HF1", prefix + "hf1_"},
123             {"AoC-HF0", prefix + "hf0_"},
124     };
125     std::vector<std::pair<std::string, std::string>> coreStates = {
126             {"DWN", "off"}, {"RET", "retention"}, {"WFI", "wfi"}};
127     p->addStateResidencyDataProvider(std::make_unique<AocStateResidencyDataProvider>(coreIds,
128             coreStates, AOC_CLOCK));
129 
130     // Add AoC voltage stats
131     std::vector<std::pair<std::string, std::string>> voltageIds = {
132             {"AoC-Voltage", prefix + "voltage_"},
133     };
134     std::vector<std::pair<std::string, std::string>> voltageStates = {{"NOM", "nominal"},
135                                                                       {"SUD", "super_underdrive"},
136                                                                       {"UUD", "ultra_underdrive"},
137                                                                       {"UD", "underdrive"}};
138     p->addStateResidencyDataProvider(
139             std::make_unique<AocStateResidencyDataProvider>(voltageIds, voltageStates, AOC_CLOCK));
140 
141     // Add AoC monitor mode
142     std::vector<std::pair<std::string, std::string>> monitorIds = {
143             {"AoC", prefix + "monitor_"},
144     };
145     std::vector<std::pair<std::string, std::string>> monitorStates = {
146             {"MON", "mode"},
147     };
148     p->addStateResidencyDataProvider(
149             std::make_unique<AocStateResidencyDataProvider>(monitorIds, monitorStates, AOC_CLOCK));
150 
151     // Add AoC restart count
152     const GenericStateResidencyDataProvider::StateResidencyConfig restartCountConfig = {
153             .entryCountSupported = true,
154             .entryCountPrefix = "",
155             .totalTimeSupported = false,
156             .lastEntrySupported = false,
157     };
158     const std::vector<std::pair<std::string, std::string>> restartCountHeaders = {
159             std::make_pair("RESTART", ""),
160     };
161     std::vector<GenericStateResidencyDataProvider::PowerEntityConfig> cfgs;
162     cfgs.emplace_back(
163             generateGenericStateResidencyConfigs(restartCountConfig, restartCountHeaders),
164             "AoC-Count", "");
165     p->addStateResidencyDataProvider(std::make_unique<GenericStateResidencyDataProvider>(
166             base + "restart_count", cfgs));
167 }
168 
addDvfsStats(std::shared_ptr<PowerStats> p)169 void addDvfsStats(std::shared_ptr<PowerStats> p) {
170     // A constant to represent the number of nanoseconds in one millisecond
171     const int NS_TO_MS = 1000000;
172     std::string path = "/sys/devices/platform/acpm_stats/fvp_stats";
173 
174     std::vector<std::pair<std::string, std::string>> adpCfgs = {
175         std::make_pair("CL0", "/sys/devices/system/cpu/cpufreq/policy0/stats"),
176         std::make_pair("CL1", "/sys/devices/system/cpu/cpufreq/policy4/stats"),
177         std::make_pair("CL2", "/sys/devices/system/cpu/cpufreq/policy8/stats"),
178         std::make_pair("MIF",
179                 "/sys/devices/platform/17000010.devfreq_mif/devfreq/17000010.devfreq_mif")};
180 
181     p->addStateResidencyDataProvider(std::make_unique<AdaptiveDvfsStateResidencyDataProvider>(
182             path, NS_TO_MS, adpCfgs));
183 
184     std::vector<DvfsStateResidencyDataProvider::Config> cfgs;
185     cfgs.push_back({"AUR", {
186         std::make_pair("1065MHz", "1065000"),
187         std::make_pair("861MHz", "861000"),
188         std::make_pair("713MHz", "713000"),
189         std::make_pair("525MHz", "525000"),
190         std::make_pair("355MHz", "355000"),
191         std::make_pair("256MHz", "256000"),
192         std::make_pair("178MHz", "178000"),
193     }});
194 
195     p->addStateResidencyDataProvider(std::make_unique<DvfsStateResidencyDataProvider>(
196             path, NS_TO_MS, cfgs));
197 
198     // TPU DVFS
199     const int TICK_TO_MS = 100;
200     std::vector<std::string> freqs = {
201             "1119000",
202             "1066000",
203             "845000",
204             "712000",
205             "627000",
206             "455000",
207             "226000"
208     };
209     p->addStateResidencyDataProvider(std::make_unique<TpuDvfsStateResidencyDataProvider>(
210             "/sys/devices/platform/1a000000.rio/tpu_usage", freqs, TICK_TO_MS));
211 }
212 
addSoC(std::shared_ptr<PowerStats> p)213 void addSoC(std::shared_ptr<PowerStats> p) {
214     // A constant to represent the number of nanoseconds in one millisecond.
215     const int NS_TO_MS = 1000000;
216 
217     // ACPM stats are reported in nanoseconds. The transform function
218     // converts nanoseconds to milliseconds.
219     std::function<uint64_t(uint64_t)> acpmNsToMs = [](uint64_t a) { return a / NS_TO_MS; };
220     const GenericStateResidencyDataProvider::StateResidencyConfig lpmStateConfig = {
221             .entryCountSupported = true,
222             .entryCountPrefix = "success_count:",
223             .totalTimeSupported = true,
224             .totalTimePrefix = "total_time_ns:",
225             .totalTimeTransform = acpmNsToMs,
226             .lastEntrySupported = true,
227             .lastEntryPrefix = "last_entry_time_ns:",
228             .lastEntryTransform = acpmNsToMs,
229     };
230     const GenericStateResidencyDataProvider::StateResidencyConfig downStateConfig = {
231             .entryCountSupported = true,
232             .entryCountPrefix = "down_count:",
233             .totalTimeSupported = true,
234             .totalTimePrefix = "total_down_time_ns:",
235             .totalTimeTransform = acpmNsToMs,
236             .lastEntrySupported = true,
237             .lastEntryPrefix = "last_down_time_ns:",
238             .lastEntryTransform = acpmNsToMs,
239     };
240     const GenericStateResidencyDataProvider::StateResidencyConfig reqStateConfig = {
241             .entryCountSupported = true,
242             .entryCountPrefix = "req_up_count:",
243             .totalTimeSupported = true,
244             .totalTimePrefix = "total_req_up_time_ns:",
245             .totalTimeTransform = acpmNsToMs,
246             .lastEntrySupported = true,
247             .lastEntryPrefix = "last_req_up_time_ns:",
248             .lastEntryTransform = acpmNsToMs,
249 
250     };
251     const std::vector<std::pair<std::string, std::string>> powerStateHeaders = {
252             std::make_pair("SICD", "SICD"),
253             std::make_pair("SLEEP", "SLEEP"),
254             std::make_pair("SLEEP_SLCMON", "SLEEP_SLCMON"),
255             std::make_pair("SLEEP_HSI1ON", "SLEEP_HSI1ON"),
256             std::make_pair("STOP", "STOP"),
257     };
258     const std::vector<std::pair<std::string, std::string>> mifReqStateHeaders = {
259             std::make_pair("AOC", "AOC"),
260             std::make_pair("GSA", "GSA"),
261             std::make_pair("TPU", "TPU"),
262             std::make_pair("AUR", "AUR"),
263     };
264     const std::vector<std::pair<std::string, std::string>> slcReqStateHeaders = {
265             std::make_pair("AOC", "AOC"),
266     };
267 
268     std::vector<GenericStateResidencyDataProvider::PowerEntityConfig> cfgs;
269     cfgs.emplace_back(generateGenericStateResidencyConfigs(lpmStateConfig, powerStateHeaders),
270             "LPM", "LPM:");
271     cfgs.emplace_back(generateGenericStateResidencyConfigs(downStateConfig, powerStateHeaders),
272             "MIF", "MIF:");
273     cfgs.emplace_back(generateGenericStateResidencyConfigs(reqStateConfig, mifReqStateHeaders),
274             "MIF-REQ", "MIF_REQ:");
275     cfgs.emplace_back(generateGenericStateResidencyConfigs(downStateConfig, powerStateHeaders),
276             "SLC", "SLC:");
277     cfgs.emplace_back(generateGenericStateResidencyConfigs(reqStateConfig, slcReqStateHeaders),
278             "SLC-REQ", "SLC_REQ:");
279 
280     p->addStateResidencyDataProvider(std::make_unique<GenericStateResidencyDataProvider>(
281             "/sys/devices/platform/acpm_stats/soc_stats", cfgs));
282 }
283 
setEnergyMeter(std::shared_ptr<PowerStats> p)284 void setEnergyMeter(std::shared_ptr<PowerStats> p) {
285     std::vector<const std::string> deviceNames { "s2mpg14-odpm", "s2mpg15-odpm" };
286     p->setEnergyMeterDataProvider(std::make_unique<IioEnergyMeterDataProvider>(deviceNames, true));
287 }
288 
addCPUclusters(std::shared_ptr<PowerStats> p)289 void addCPUclusters(std::shared_ptr<PowerStats> p) {
290     // A constant to represent the number of nanoseconds in one millisecond.
291     const int NS_TO_MS = 1000000;
292 
293     std::function<uint64_t(uint64_t)> acpmNsToMs = [](uint64_t a) { return a / NS_TO_MS; };
294     const GenericStateResidencyDataProvider::StateResidencyConfig cpuStateConfig = {
295             .entryCountSupported = true,
296             .entryCountPrefix = "down_count:",
297             .totalTimeSupported = true,
298             .totalTimePrefix = "total_down_time_ns:",
299             .totalTimeTransform = acpmNsToMs,
300             .lastEntrySupported = true,
301             .lastEntryPrefix = "last_down_time_ns:",
302             .lastEntryTransform = acpmNsToMs,
303     };
304 
305     const std::vector<std::pair<std::string, std::string>> cpuStateHeaders = {
306             std::make_pair("DOWN", ""),
307     };
308 
309     std::vector<GenericStateResidencyDataProvider::PowerEntityConfig> cfgs;
310     for (std::string name : {
311             "CLUSTER0",
312             "CLUSTER1",
313             "CLUSTER2"}) {
314         cfgs.emplace_back(generateGenericStateResidencyConfigs(cpuStateConfig, cpuStateHeaders),
315             name, name);
316     }
317 
318     p->addStateResidencyDataProvider(std::make_unique<GenericStateResidencyDataProvider>(
319             "/sys/devices/platform/acpm_stats/core_stats", cfgs));
320 
321     CpupmStateResidencyDataProvider::Config config = {
322         .entities = {
323             std::make_pair("CPU0", "cpu0"),
324             std::make_pair("CPU1", "cpu1"),
325             std::make_pair("CPU2", "cpu2"),
326             std::make_pair("CPU3", "cpu3"),
327             std::make_pair("CPU4", "cpu4"),
328             std::make_pair("CPU5", "cpu5"),
329             std::make_pair("CPU6", "cpu6"),
330             std::make_pair("CPU7", "cpu7"),
331             std::make_pair("CPU8", "cpu8")},
332         .states = {
333             std::make_pair("DOWN", "[state1]")}};
334 
335     CpupmStateResidencyDataProvider::SleepConfig sleepConfig = {"LPM:", "SLEEP", "total_time_ns:"};
336 
337     p->addStateResidencyDataProvider(std::make_unique<CpupmStateResidencyDataProvider>(
338             "/sys/devices/system/cpu/cpupm/cpupm/time_in_state", config,
339             "/sys/devices/platform/acpm_stats/soc_stats", sleepConfig));
340 
341     p->addEnergyConsumer(PowerStatsEnergyConsumer::createMeterConsumer(p,
342             EnergyConsumerType::CPU_CLUSTER, "CPUCL0", {"S4M_VDD_CPUCL0"}));
343     p->addEnergyConsumer(PowerStatsEnergyConsumer::createMeterConsumer(p,
344             EnergyConsumerType::CPU_CLUSTER, "CPUCL1", {"S3M_VDD_CPUCL1"}));
345     p->addEnergyConsumer(PowerStatsEnergyConsumer::createMeterConsumer(p,
346             EnergyConsumerType::CPU_CLUSTER, "CPUCL2", {"S2M_VDD_CPUCL2"}));
347 }
348 
addGPU(std::shared_ptr<PowerStats> p)349 void addGPU(std::shared_ptr<PowerStats> p) {
350     // Add gpu energy consumer
351     std::map<std::string, int32_t> stateCoeffs;
352     std::string path = "/sys/devices/platform/1f000000.mali";
353 
354     stateCoeffs = {
355         {"150000",  637},
356         {"302000", 1308},
357         {"337000", 1461},
358         {"376000", 1650},
359         {"419000", 1861},
360         {"467000", 2086},
361         {"521000", 2334},
362         {"580000", 2558},
363         {"649000", 2886},
364         {"723000", 3244},
365         {"807000", 3762},
366         {"890000", 4333}};
367 
368     p->addEnergyConsumer(PowerStatsEnergyConsumer::createMeterAndAttrConsumer(p,
369             EnergyConsumerType::OTHER, "GPU", {"S2S_VDD_G3D", "S8S_VDD_G3D_L2"},
370             {{UID_TIME_IN_STATE, path + "/uid_time_in_state"}},
371             stateCoeffs));
372 
373     p->addStateResidencyDataProvider(std::make_unique<DevfreqStateResidencyDataProvider>("GPU",
374             path));
375 }
376 
addMobileRadio(std::shared_ptr<PowerStats> p)377 void addMobileRadio(std::shared_ptr<PowerStats> p)
378 {
379     // A constant to represent the number of microseconds in one millisecond.
380     const int US_TO_MS = 1000;
381 
382     // modem power_stats are reported in microseconds. The transform function
383     // converts microseconds to milliseconds.
384     std::function<uint64_t(uint64_t)> modemUsToMs = [](uint64_t a) { return a / US_TO_MS; };
385     const GenericStateResidencyDataProvider::StateResidencyConfig powerStateConfig = {
386             .entryCountSupported = true,
387             .entryCountPrefix = "count:",
388             .totalTimeSupported = true,
389             .totalTimePrefix = "duration_usec:",
390             .totalTimeTransform = modemUsToMs,
391             .lastEntrySupported = true,
392             .lastEntryPrefix = "last_entry_timestamp_usec:",
393             .lastEntryTransform = modemUsToMs,
394     };
395     const std::vector<std::pair<std::string, std::string>> powerStateHeaders = {
396             std::make_pair("SLEEP", "SLEEP:"),
397     };
398 
399     std::vector<GenericStateResidencyDataProvider::PowerEntityConfig> cfgs;
400     cfgs.emplace_back(generateGenericStateResidencyConfigs(powerStateConfig, powerStateHeaders),
401             "MODEM", "");
402 
403     p->addStateResidencyDataProvider(std::make_unique<GenericStateResidencyDataProvider>(
404             "/sys/devices/platform/cpif/modem/power_stats", cfgs));
405 
406     p->addEnergyConsumer(PowerStatsEnergyConsumer::createMeterConsumer(p,
407             EnergyConsumerType::MOBILE_RADIO, "MODEM",
408             {"VSYS_PWR_MODEM", "VSYS_PWR_RFFE", "VSYS_PWR_MMWAVE"}));
409 }
410 
addGNSS(std::shared_ptr<PowerStats> p)411 void addGNSS(std::shared_ptr<PowerStats> p)
412 {
413     // A constant to represent the number of microseconds in one millisecond.
414     const int US_TO_MS = 1000;
415 
416     // gnss power_stats are reported in microseconds. The transform function
417     // converts microseconds to milliseconds.
418     std::function<uint64_t(uint64_t)> gnssUsToMs = [](uint64_t a) { return a / US_TO_MS; };
419 
420     const GenericStateResidencyDataProvider::StateResidencyConfig gnssStateConfig = {
421         .entryCountSupported = true,
422         .entryCountPrefix = "count:",
423         .totalTimeSupported = true,
424         .totalTimePrefix = "duration_usec:",
425         .totalTimeTransform = gnssUsToMs,
426         .lastEntrySupported = true,
427         .lastEntryPrefix = "last_entry_timestamp_usec:",
428         .lastEntryTransform = gnssUsToMs,
429     };
430 
431     const std::vector<std::pair<std::string, std::string>> gnssStateHeaders = {
432         std::make_pair("ON", "GPS_ON:"),
433         std::make_pair("OFF", "GPS_OFF:"),
434     };
435 
436     std::vector<GenericStateResidencyDataProvider::PowerEntityConfig> cfgs;
437     cfgs.emplace_back(generateGenericStateResidencyConfigs(gnssStateConfig, gnssStateHeaders),
438             "GPS", "");
439 
440     p->addStateResidencyDataProvider(std::make_unique<GenericStateResidencyDataProvider>(
441             "/dev/bbd_pwrstat", cfgs));
442 
443     p->addEnergyConsumer(PowerStatsEnergyConsumer::createMeterConsumer(p,
444             EnergyConsumerType::GNSS, "GPS", {"L9S_GNSS_CORE"}));
445 }
446 
addPCIe(std::shared_ptr<PowerStats> p)447 void addPCIe(std::shared_ptr<PowerStats> p) {
448     // Add PCIe power entities for Modem and WiFi
449     const GenericStateResidencyDataProvider::StateResidencyConfig pcieStateConfig = {
450         .entryCountSupported = true,
451         .entryCountPrefix = "Cumulative count:",
452         .totalTimeSupported = true,
453         .totalTimePrefix = "Cumulative duration msec:",
454         .lastEntrySupported = true,
455         .lastEntryPrefix = "Last entry timestamp msec:",
456     };
457     const std::vector<std::pair<std::string, std::string>> pcieStateHeaders = {
458         std::make_pair("UP", "Link up:"),
459         std::make_pair("DOWN", "Link down:"),
460     };
461 
462     // Add PCIe - Modem
463     const std::vector<GenericStateResidencyDataProvider::PowerEntityConfig> pcieModemCfgs = {
464         {generateGenericStateResidencyConfigs(pcieStateConfig, pcieStateHeaders), "PCIe-Modem",
465                 "Version: 1"}
466     };
467 
468     p->addStateResidencyDataProvider(std::make_unique<GenericStateResidencyDataProvider>(
469             "/sys/devices/platform/12100000.pcie/power_stats", pcieModemCfgs));
470 
471     // Add PCIe - WiFi
472     const std::vector<GenericStateResidencyDataProvider::PowerEntityConfig> pcieWifiCfgs = {
473         {generateGenericStateResidencyConfigs(pcieStateConfig, pcieStateHeaders),
474             "PCIe-WiFi", "Version: 1"}
475     };
476 
477     p->addStateResidencyDataProvider(std::make_unique<GenericStateResidencyDataProvider>(
478             "/sys/devices/platform/13120000.pcie/power_stats", pcieWifiCfgs));
479 }
480 
addWifi(std::shared_ptr<PowerStats> p)481 void addWifi(std::shared_ptr<PowerStats> p) {
482     // The transform function converts microseconds to milliseconds.
483     std::function<uint64_t(uint64_t)> usecToMs = [](uint64_t a) { return a / 1000; };
484     const GenericStateResidencyDataProvider::StateResidencyConfig stateConfig = {
485         .entryCountSupported = true,
486         .entryCountPrefix = "count:",
487         .totalTimeSupported = true,
488         .totalTimePrefix = "duration_usec:",
489         .totalTimeTransform = usecToMs,
490         .lastEntrySupported = true,
491         .lastEntryPrefix = "last_entry_timestamp_usec:",
492         .lastEntryTransform = usecToMs,
493     };
494     const GenericStateResidencyDataProvider::StateResidencyConfig pcieStateConfig = {
495         .entryCountSupported = true,
496         .entryCountPrefix = "count:",
497         .totalTimeSupported = true,
498         .totalTimePrefix = "duration_usec:",
499         .totalTimeTransform = usecToMs,
500         .lastEntrySupported = false,
501     };
502 
503     const std::vector<std::pair<std::string, std::string>> stateHeaders = {
504         std::make_pair("AWAKE", "AWAKE:"),
505         std::make_pair("ASLEEP", "ASLEEP:"),
506 
507     };
508     const std::vector<std::pair<std::string, std::string>> pcieStateHeaders = {
509         std::make_pair("L0", "L0:"),
510         std::make_pair("L1", "L1:"),
511         std::make_pair("L1_1", "L1_1:"),
512         std::make_pair("L1_2", "L1_2:"),
513         std::make_pair("L2", "L2:"),
514     };
515 
516     const std::vector<GenericStateResidencyDataProvider::PowerEntityConfig> cfgs = {
517         {generateGenericStateResidencyConfigs(stateConfig, stateHeaders), "WIFI", "WIFI"},
518         {generateGenericStateResidencyConfigs(pcieStateConfig, pcieStateHeaders), "WIFI-PCIE",
519                 "WIFI-PCIE"}
520     };
521 
522     p->addStateResidencyDataProvider(std::make_unique<GenericStateResidencyDataProvider>(
523             "/sys/wifi/power_stats", cfgs));
524 }
525 
addUfs(std::shared_ptr<PowerStats> p)526 void addUfs(std::shared_ptr<PowerStats> p) {
527     p->addStateResidencyDataProvider(std::make_unique<UfsStateResidencyDataProvider>(
528             "/sys/bus/platform/devices/13200000.ufs/ufs_stats/"));
529 }
530 
addPowerDomains(std::shared_ptr<PowerStats> p)531 void addPowerDomains(std::shared_ptr<PowerStats> p) {
532     // A constant to represent the number of nanoseconds in one millisecond.
533     const int NS_TO_MS = 1000000;
534 
535     std::function<uint64_t(uint64_t)> acpmNsToMs = [](uint64_t a) { return a / NS_TO_MS; };
536     const GenericStateResidencyDataProvider::StateResidencyConfig cpuStateConfig = {
537             .entryCountSupported = true,
538             .entryCountPrefix = "on_count:",
539             .totalTimeSupported = true,
540             .totalTimePrefix = "total_on_time_ns:",
541             .totalTimeTransform = acpmNsToMs,
542             .lastEntrySupported = true,
543             .lastEntryPrefix = "last_on_time_ns:",
544             .lastEntryTransform = acpmNsToMs,
545     };
546 
547     const std::vector<std::pair<std::string, std::string>> cpuStateHeaders = {
548             std::make_pair("ON", ""),
549     };
550 
551     std::vector<GenericStateResidencyDataProvider::PowerEntityConfig> cfgs;
552     for (std::string name : {
553             "pd-tpu",
554             "pd-ispfe",
555             "pd-eh",
556             "pd-bw",
557             "pd-aur",
558             "pd-yuvp",
559             "pd-tnr",
560             "pd-rgbp",
561             "pd-mfc",
562             "pd-mcsc",
563             "pd-gse",
564             "pd-gdc",
565             "pd-g2d",
566             "pd-dpuf1",
567             "pd-dpuf0",
568             "pd-dpub",
569             "pd-embedded_g3d",
570             "pd-g3d"}) {
571         cfgs.emplace_back(generateGenericStateResidencyConfigs(cpuStateConfig, cpuStateHeaders),
572             name, name + ":");
573     }
574 
575     p->addStateResidencyDataProvider(std::make_unique<GenericStateResidencyDataProvider>(
576             "/sys/devices/platform/acpm_stats/pd_stats", cfgs));
577 }
578 
addDevfreq(std::shared_ptr<PowerStats> p)579 void addDevfreq(std::shared_ptr<PowerStats> p) {
580     p->addStateResidencyDataProvider(std::make_unique<DevfreqStateResidencyDataProvider>(
581             "INT",
582             "/sys/devices/platform/17000020.devfreq_int/devfreq/17000020.devfreq_int"));
583 
584     p->addStateResidencyDataProvider(std::make_unique<DevfreqStateResidencyDataProvider>(
585             "INTCAM",
586             "/sys/devices/platform/17000030.devfreq_intcam/devfreq/17000030.devfreq_intcam"));
587 
588     p->addStateResidencyDataProvider(std::make_unique<DevfreqStateResidencyDataProvider>(
589             "DISP",
590             "/sys/devices/platform/17000040.devfreq_disp/devfreq/17000040.devfreq_disp"));
591 
592     p->addStateResidencyDataProvider(std::make_unique<DevfreqStateResidencyDataProvider>(
593             "CAM",
594             "/sys/devices/platform/17000050.devfreq_cam/devfreq/17000050.devfreq_cam"));
595 
596     p->addStateResidencyDataProvider(std::make_unique<DevfreqStateResidencyDataProvider>(
597             "TNR",
598             "/sys/devices/platform/17000060.devfreq_tnr/devfreq/17000060.devfreq_tnr"));
599 
600     p->addStateResidencyDataProvider(std::make_unique<DevfreqStateResidencyDataProvider>(
601             "MFC",
602             "/sys/devices/platform/17000070.devfreq_mfc/devfreq/17000070.devfreq_mfc"));
603 
604     p->addStateResidencyDataProvider(std::make_unique<DevfreqStateResidencyDataProvider>(
605             "BW",
606             "/sys/devices/platform/17000080.devfreq_bw/devfreq/17000080.devfreq_bw"));
607 
608     p->addStateResidencyDataProvider(std::make_unique<DevfreqStateResidencyDataProvider>(
609             "DSU",
610             "/sys/devices/platform/17000090.devfreq_dsu/devfreq/17000090.devfreq_dsu"));
611 
612     p->addStateResidencyDataProvider(std::make_unique<DevfreqStateResidencyDataProvider>(
613             "BCI",
614             "/sys/devices/platform/170000a0.devfreq_bci/devfreq/170000a0.devfreq_bci"));
615 }
616 
addTPU(std::shared_ptr<PowerStats> p)617 void addTPU(std::shared_ptr<PowerStats> p) {
618     std::map<std::string, int32_t> stateCoeffs;
619 
620     stateCoeffs = {
621         // TODO (b/197721618): Measuring the TPU power numbers
622         {"226000",  10},
623         {"455000",  20},
624         {"627000",  30},
625         {"712000",  40},
626         {"845000",  50},
627         {"967000",  60},
628         {"1119000", 70}};
629 
630     p->addEnergyConsumer(PowerStatsEnergyConsumer::createMeterAndAttrConsumer(p,
631             EnergyConsumerType::OTHER, "TPU", {"S7M_VDD_TPU"},
632             {{UID_TIME_IN_STATE, "/sys/devices/platform/1a000000.rio/tpu_usage"}},
633             stateCoeffs));
634 }
635 
636 /**
637  * Unlike other data providers, which source power entity state residency data from the kernel,
638  * this data provider acts as a general-purpose channel for state residency data providers
639  * that live in user space. Entities are defined here and user space clients of this provider's
640  * vendor service register callbacks to provide state residency data for their given pwoer entity.
641  */
addPixelStateResidencyDataProvider(std::shared_ptr<PowerStats> p)642 void addPixelStateResidencyDataProvider(std::shared_ptr<PowerStats> p) {
643 
644     auto pixelSdp = std::make_unique<PixelStateResidencyDataProvider>();
645 
646     pixelSdp->addEntity("Bluetooth", {{0, "Idle"}, {1, "Active"}, {2, "Tx"}, {3, "Rx"}});
647 
648     pixelSdp->start();
649 
650     p->addStateResidencyDataProvider(std::move(pixelSdp));
651 }
652 
addDisplayMRR(std::shared_ptr<PowerStats> p)653 void addDisplayMRR(std::shared_ptr<PowerStats> p) {
654     p->addStateResidencyDataProvider(std::make_unique<DisplayMrrStateResidencyDataProvider>(
655             "Display", "/sys/class/drm/card0/device/primary-panel/"));
656 }
657 
addZumaCommonDataProviders(std::shared_ptr<PowerStats> p)658 void addZumaCommonDataProviders(std::shared_ptr<PowerStats> p) {
659     setEnergyMeter(p);
660 
661     addAoC(p);
662     addPixelStateResidencyDataProvider(p);
663     addCPUclusters(p);
664     addSoC(p);
665     addGNSS(p);
666     addMobileRadio(p);
667     addNFC(p);
668     addPCIe(p);
669     addWifi(p);
670     addTPU(p);
671     addUfs(p);
672     addPowerDomains(p);
673     addDvfsStats(p);
674     addDevfreq(p);
675     addGPU(p);
676 }
677 
addNFC(std::shared_ptr<PowerStats> p)678 void addNFC(std::shared_ptr<PowerStats> p) {
679     const GenericStateResidencyDataProvider::StateResidencyConfig nfcStateConfig = {
680         .entryCountSupported = true,
681         .entryCountPrefix = "Cumulative count:",
682         .totalTimeSupported = true,
683         .totalTimePrefix = "Cumulative duration msec:",
684         .lastEntrySupported = true,
685         .lastEntryPrefix = "Last entry timestamp msec:",
686     };
687     const std::vector<std::pair<std::string, std::string>> nfcStateHeaders = {
688         std::make_pair("IDLE", "Idle mode:"),
689         std::make_pair("ACTIVE", "Active mode:"),
690         std::make_pair("ACTIVE-RW", "Active Reader/Writer mode:"),
691     };
692 
693     std::vector<GenericStateResidencyDataProvider::PowerEntityConfig> cfgs;
694     cfgs.emplace_back(generateGenericStateResidencyConfigs(nfcStateConfig, nfcStateHeaders),
695             "NFC", "NFC subsystem");
696 
697     std::string path;
698     struct stat buffer;
699     for (int i = 0; i < 10; i++) {
700         std::string idx = std::to_string(i);
701         path = "/sys/devices/platform/10c80000.hsi2c/i2c-" + idx + "/" + idx + "-0008/power_stats";
702         if (!stat(path.c_str(), &buffer))
703             break;
704     }
705     p->addStateResidencyDataProvider(std::make_unique<GenericStateResidencyDataProvider>(
706             path, cfgs));
707 }
708