1 files changed, 172 insertions, 0 deletions
diff --git a/battery_mitigation/include/battery_mitigation/BatteryMitigationService.h b/battery_mitigation/include/battery_mitigation/BatteryMitigationService.h
new file mode 100644
index 00000000..9c697140
--- /dev/null
+++ b/battery_mitigation/include/battery_mitigation/BatteryMitigationService.h
@@ -0,0 +1,172 @@
+/*
+ * Copyright (C) 2023 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.
+ */
+
+#pragma once
+
+#include <android-base/chrono_utils.h>
+#include <android-base/file.h>
+#include <android-base/logging.h>
+#include <android-base/parseint.h>
+#include <android-base/properties.h>
+#include <android-base/strings.h>
+#include <errno.h>
+#include <sys/epoll.h>
+#include <sys/eventfd.h>
+#include <sys/mman.h>
+#include <sys/stat.h>
+#include <sys/system_properties.h>
+#include <unistd.h>
+#include <utils/RefBase.h>
+
+#include <algorithm>
+#include <cmath>
+#include <map>
+#include <regex>
+#include <thread>
+
+#include "MitigationConfig.h"
+
+#define MIN_SUPPORTED_PLATFORM   2 /* CDT */
+#define MAX_SUPPORTED_PLATFORM   5
+#define NSEC_PER_SEC             1000000000L
+#define BROWNOUT_EVENT_BUF_SIZE  10
+#define DUMP_TIMES               12
+#define EPOLL_MAXEVENTS 12
+#define BUF_SIZE                 128
+#define FVP_STATS_SIZE 4096
+#define STAT_NAME_SIZE           48
+#define STATS_MAX_SIZE           64
+#define PMIC_NUM 2
+#define LOOP_TRIG_STATS(idx) for (int idx = 0; idx < MAX_EVENT; idx++)
+
+namespace android {
+namespace hardware {
+namespace google {
+namespace pixel {
+
+using ::android::sp;
+using android::hardware::google::pixel::MitigationConfig;
+
+struct numericStat {
+    char name[STAT_NAME_SIZE];
+    int value;
+};
+
+struct OdpmInstantPower {
+    struct timespec time;
+    double value;
+};
+
+struct BrownoutStatsCSVFields {
+    const char *const triggered_time;
+    const char *const triggered_idx;
+    const char *const battery_soc;
+    const char *const battery_temp;
+    const char *const battery_cycle;
+    const char *const voltage_now;
+    const char *const current_now;
+    const char *const cpu0_freq;
+    const char *const cpu1_freq;
+    const char *const cpu2_freq;
+    const char *const gpu_freq;
+    const char *const tpu_freq;
+    const char *const aur_freq;
+    const char *const odpm_prefix;
+};
+
+struct BrownoutStatsCSVRow {
+    struct timespec triggered_time;
+    int triggered_idx;
+    int min_battery_soc;
+    int max_battery_temp;
+    int min_battery_cycle;
+    int min_voltage_now;
+    int max_current_now;
+    int min_cpu0_freq;
+    int min_cpu1_freq;
+    int min_cpu2_freq;
+    int min_gpu_freq;
+    int min_tpu_freq;
+    int min_aur_freq;
+
+    double max_main_odpm_instant_power[METER_CHANNEL_MAX];
+    double max_sub_odpm_instant_power[METER_CHANNEL_MAX];
+};
+
+struct BrownoutStatsExtend {
+    struct brownout_stats brownoutStats;
+    char fvpStats[FVP_STATS_SIZE];
+    struct numericStat numericStats[STATS_MAX_SIZE];
+    timeval eventReceivedTime;
+    timeval dumpTime;
+    unsigned int eventIdx;
+};
+
+class BatteryMitigationService : public RefBase {
+  public:
+    BatteryMitigationService(const struct MitigationConfig::EventThreadConfig &eventThreadCfg);
+    ~BatteryMitigationService();
+
+    void startBrownoutEventThread();
+    void stopEventThread(std::atomic_bool &thread_stop, int wakeup_event_fd,
+                         std::thread &event_thread);
+    bool isBrownoutStatsBinarySupported();
+    bool isPlatformSupported();
+    bool isTimeValid(const char*, std::chrono::system_clock::time_point);
+    bool genParsedMeal(const char*);
+    bool genLastmealCSV(const char*);
+  private:
+    struct MitigationConfig::EventThreadConfig cfg;
+
+    int storingFd;
+    int triggeredStateFd[MAX_EVENT];
+    int triggeredStateEpollFd;
+    int triggeredStateWakeupEventFd;
+    std::thread eventThread;
+    std::atomic_bool triggerThreadStop{false};
+    int brownoutStatsFd;
+    int triggeredIdxFd;
+    int triggeredIdxEpollFd;
+    int wakeupEventFd;
+    std::thread brownoutEventThread;
+    std::atomic_bool threadStop{false};
+
+    int mainPmicID;
+    int subPmicID;
+    double mainLpfBitResolutions[METER_CHANNEL_MAX];
+    double subLpfBitResolutions[METER_CHANNEL_MAX];
+    char *mainLpfChannelNames[METER_CHANNEL_MAX];
+    char *subLpfChannelNames[METER_CHANNEL_MAX];
+    std::vector<MitigationConfig::numericSysfs> totalNumericSysfsStatPaths;
+
+    void BrownoutEventThread();
+    void TriggerEventThread();
+    void initTotalNumericSysfsPaths();
+    void initPmicRelated();
+    int initFd();
+    int initTrigFd();
+    void tearDownBrownoutEventThread();
+    void tearDownTriggerEventThread();
+    int readNumericStats(struct BrownoutStatsExtend*);
+    bool parseBrownoutStatsExtend(FILE *);
+    void printBrownoutStatsExtendSummary(FILE *, struct BrownoutStatsExtend *);
+    void getBrownoutStatsCSVRow(struct BrownoutStatsExtend *, struct BrownoutStatsCSVRow *);
+};
+
+}  // namespace pixel
+}  // namespace google
+}  // namespace hardware
+}  // namespace android