From f964702aea4cdacdb387980d8494c13387c0cb7e Mon Sep 17 00:00:00 2001
From: bouhaa <boukehaarsma23@gmail.com>
Date: Sat, 29 Jul 2023 19:48:38 +0200
Subject: [PATCH] integrate zenergy into Mangohud

Reuses most of the RAPL code paths
---
 src/cpu.cpp | 45 +++++++++++++++++++++++++++++++++++++++++++++
 src/cpu.h   | 18 ++++++++++++++++++
 2 files changed, 63 insertions(+)

diff --git a/src/cpu.cpp b/src/cpu.cpp
index 11af9a2b..ace436f9 100644
--- a/src/cpu.cpp
+++ b/src/cpu.cpp
@@ -364,6 +364,33 @@ static bool get_cpu_power_zenpower(CPUPowerData* cpuPowerData, float& power) {
     return true;
 }
 
+static bool get_cpu_power_zenergy(CPUPowerData* cpuPowerData, float& power) {
+    CPUPowerData_zenergy* powerData_zenergy = (CPUPowerData_zenergy*)cpuPowerData;
+    if (!powerData_zenergy->energyCounterFile)
+        return false;
+
+    rewind(powerData_zenergy->energyCounterFile);
+    fflush(powerData_zenergy->energyCounterFile);
+
+    uint64_t energyCounterValue = 0;
+    if (fscanf(powerData_zenergy->energyCounterFile, "%" SCNu64, &energyCounterValue) != 1)
+        return false;
+
+    Clock::time_point now = Clock::now();
+    Clock::duration timeDiff = now - powerData_zenergy->lastCounterValueTime;
+    int64_t timeDiffMicro = std::chrono::duration_cast<std::chrono::microseconds>(timeDiff).count();
+    uint64_t energyCounterDiff = energyCounterValue - powerData_zenergy->lastCounterValue;
+
+
+    if (powerData_zenergy->lastCounterValue > 0 && energyCounterValue > powerData_zenergy->lastCounterValue)
+        power = (float) energyCounterDiff / (float) timeDiffMicro;
+
+    powerData_zenergy->lastCounterValue = energyCounterValue;
+    powerData_zenergy->lastCounterValueTime = now;
+
+    return true;
+}
+
 static bool get_cpu_power_rapl(CPUPowerData* cpuPowerData, float& power) {
     CPUPowerData_rapl* powerData_rapl = (CPUPowerData_rapl*)cpuPowerData;
 
@@ -409,6 +436,9 @@ bool CPUStats::UpdateCpuPower() {
         case CPU_POWER_ZENPOWER:
             if (!get_cpu_power_zenpower(m_cpuPowerData.get(), power)) return false;
             break;
+        case CPU_POWER_ZENERGY:
+            if (!get_cpu_power_zenergy(m_cpuPowerData.get(), power)) return false;
+            break;
         case CPU_POWER_RAPL:
             if (!get_cpu_power_rapl(m_cpuPowerData.get(), power)) return false;
             break;
@@ -553,6 +583,18 @@ static CPUPowerData_zenpower* init_cpu_power_data_zenpower(const std::string pat
     return powerData.release();
 }
 
+static CPUPowerData_zenergy* init_cpu_power_data_zenergy(const std::string path) {
+    auto powerData = std::make_unique<CPUPowerData_zenergy>();
+    std::string energyCounterPath;
+
+    if(!find_input(path, "energy", energyCounterPath, "Esocket0")) return nullptr;
+
+    SPDLOG_DEBUG("hwmon: using input: {}", energyCounterPath);
+    powerData->energyCounterFile = fopen(energyCounterPath.c_str(), "r");
+
+    return powerData.release();
+}
+
 static CPUPowerData_rapl* init_cpu_power_data_rapl(const std::string path) {
     auto powerData = std::make_unique<CPUPowerData_rapl>();
 
@@ -586,6 +628,9 @@ bool CPUStats::InitCpuPowerData() {
         } else if (name == "zenpower") {
             cpuPowerData = (CPUPowerData*)init_cpu_power_data_zenpower(path);
             break;
+        } else if (name == "zenergy") {
+            cpuPowerData = (CPUPowerData*)init_cpu_power_data_zenergy(path);
+            break;
         } else if (name == "coretemp") {
             intel = true;
         }
diff --git a/src/cpu.h b/src/cpu.h
index ef31735e..ef1741cb 100644
--- a/src/cpu.h
+++ b/src/cpu.h
@@ -50,6 +50,7 @@ typedef struct CPUData_ {
 enum {
    CPU_POWER_K10TEMP,
    CPU_POWER_ZENPOWER,
+   CPU_POWER_ZENERGY,
    CPU_POWER_RAPL,
    CPU_POWER_AMDGPU
 };
@@ -102,6 +103,23 @@ struct CPUPowerData_zenpower : public CPUPowerData {
    FILE* socPowerFile {nullptr};
 };
 
+struct CPUPowerData_zenergy : public CPUPowerData {
+   CPUPowerData_zenergy() {
+      this->source = CPU_POWER_ZENERGY;
+      this->lastCounterValue = 0;
+      this->lastCounterValueTime = Clock::now();
+   };
+
+   ~CPUPowerData_zenergy() {
+      if(this->energyCounterFile)
+         fclose(this->energyCounterFile);
+   };
+
+   FILE* energyCounterFile {nullptr};
+   uint64_t lastCounterValue;
+   Clock::time_point lastCounterValueTime;
+};
+
 struct CPUPowerData_rapl : public CPUPowerData {
    CPUPowerData_rapl() {
       this->source = CPU_POWER_RAPL;