/*
 * Copyright (c) Souldbminer, Lightos and Horizon OC Contributors
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms and conditions of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

#include "actmon.hpp"

namespace actmon {

    u32 DevReg(ActmonDev dev, u32 off) {
        return ActmonDevBase + (static_cast<u32>(dev) * ActmonDevSize) + off;
    }

    void EnableDev(ActmonDev dev, u32 freqKhz, u32 weight) {
        MMIO32(DevReg(dev, DevInitAvg))     = freqKhz * ActmonPeriodMs / 2;
        MMIO32(DevReg(dev, DevCountWeight)) = weight;
        MMIO32(DevReg(dev, DevCtrl))        = ActmonDevCtrlEnb | ActmonDevCtrlEnbPeriodic | ActmonDevCtrlKVal3;
    }

    void Init() {
        const u32 emcFreqKhz = static_cast<u32>(freq::MeasurePtoFreqHz(freq::ClkPtoEmc, 1) / 1000);

        if (!(MMIO32(ActmonBase + ActmonGlbStatus) & ActmonMcallMonAct)) {
            MMIO32(ActmonBase + ActmonGlbPeriodCtrl) = (ActmonPeriodMs - 1) & 0xFF;
            EnableDev(DevMcAll, emcFreqKhz, 256 * 4);
        }

        if (!(MMIO32(ActmonBase + ActmonGlbStatus) & ActmonMccpuMonAct)) {
            EnableDev(DevMcCpu, emcFreqKhz, 256 * 4);
        }
    }

    void Update(HocClkBpmpSharedInfo &info) {
        const u32 emcFreqKhz = info.freqMem;
        if (emcFreqKhz == 0) {
            return;
        }

        const u32 avgAll = MMIO32(DevReg(DevMcAll, DevAvgCount));
        const u32 avgCpu = MMIO32(DevReg(DevMcCpu, DevAvgCount));

        // Get 1000 -> 100.0.
        info.emcLoadAll = static_cast<u32>((static_cast<u64>(avgAll) * 10 * 100) / (emcFreqKhz * ActmonPeriodMs));
        info.emcLoadCpu = static_cast<u32>((static_cast<u64>(avgCpu) * 10 * 100) / (emcFreqKhz * ActmonPeriodMs));

        info.emcBwAll = static_cast<u32>((static_cast<u64>(emcFreqKhz) * 16 * info.emcLoadAll) / 1000000);
        info.emcBwCpu = static_cast<u32>((static_cast<u64>(emcFreqKhz) * 16 * info.emcLoadCpu) / 1000000);

        // Not 100% accurate but should be enough.
        info.emcBwGpu = (info.emcBwAll > info.emcBwCpu) ? (info.emcBwAll - info.emcBwCpu) : 0;
    }

} // namespace actmon