// Software Name : cppuprofile
// SPDX-FileCopyrightText: Copyright (c) 2022 Orange
// SPDX-License-Identifier: BSD-3-Clause
//
// This software is distributed under the BSD License;
// see the LICENSE file for more details.
//
// Author: Cédric CHEDALEUX <cedric.chedaleux@orange.com> et al.

#ifndef UPROFILEIMPL_H_
#define UPROFILEIMPL_H_

#include "igpumonitor.h"
#include "timestampunit.h"
#include "util/cpumonitor.h"
#include "util/timer.h"
#include <fstream>
#include <list>
#include <map>
#include <mutex>
#include <string>

namespace uprofile
{
class UProfileImpl
{
public:
    enum class ProfilingType {
        TIME_EXEC,
        TIME_EVENT,
        PROCESS_MEMORY,
        SYSTEM_MEMORY,
        CPU,
        GPU_USAGE,
        GPU_MEMORY
    };

    static UProfileImpl* getInstance();
    static void destroyInstance();
    virtual ~UProfileImpl();

    // Implementation
    void start(const char* file);
    void stop();
    void addGPUMonitor(IGPUMonitor* monitor);
    void removeGPUMonitor();
    void setTimestampUnit(TimestampUnit tsUnit);
    void timeBegin(const std::string& title);
    void timeEnd(const std::string& title);
    void startProcessMemoryMonitoring(int period);
    void startSystemMemoryMonitoring(int period);
    void startCPUUsageMonitoring(int period);
    void startGPUUsageMonitoring(int period);
    void startGPUMemoryMonitoring(int period);
    void getProcessMemory(int& rss, int& shared);
    void getSystemMemory(int& totalMem, int& availableMem, int& freeMem);
    vector<float> getInstantCpuUsage();

private:
    static UProfileImpl* m_uprofiler;
    UProfileImpl();

    void write(ProfilingType type, const std::list<std::string>& data);
    unsigned long long getTimestamp() const;
    static unsigned long long getEpochTime();
    static unsigned long long getTimeSinceBoot();

    void dumpCpuUsage();
    void dumpProcessMemory();
    void dumpSystemMemory();
    void dumpGpuUsage();
    void dumpGpuMemory();

    TimestampUnit m_tsUnit;
    std::map<std::string, unsigned long long> m_steps; // Store steps (title, start time)
    std::ofstream m_file;
    Timer m_processMemoryMonitorTimer;
    Timer m_systemMemoryMonitorTimer;
    Timer m_cpuMonitorTimer;
    Timer m_gpuUsageMonitorTimer;
    Timer m_gpuMemoryMonitorTimer;
    CpuMonitor m_cpuMonitor;
    IGPUMonitor* m_gpuMonitor;

    std::mutex m_fileMutex;
    std::mutex m_stepsMutex;
};

}

#endif /* UPROFILEIMPL_H_ */