// Software Name : cppuprofile
// SPDX-FileCopyrightText: Copyright (c) 2023 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 NVIDIAMONITOR_H_
#define NVIDIAMONITOR_H_

#include "api.h"
#include "igpumonitor.h"
#include <mutex>
#include <string>
#include <thread>
#include <vector>

using namespace std;

namespace uprofile
{
class NvidiaMonitor : public IGPUMonitor
{
public:
    UPROFAPI explicit NvidiaMonitor();
    UPROFAPI virtual ~NvidiaMonitor();

    UPROFAPI void start(int period) override;
    UPROFAPI void stop() override;
    UPROFAPI bool watching() const override;
    UPROFAPI float getUsage() const override;
    UPROFAPI void getMemory(int& usedMem, int& totalMem) const override;

private:
    void watchGPU(int period);
    void abortWatchGPU();

    mutable std::mutex m_mutex;
    std::unique_ptr<std::thread> m_watcherThread;
    bool m_watching = false;
    int m_totalMem = 0;
    int m_usedMem = 0;
    float m_gpuUsage = 0.f;
};

}
#endif /* NVIDIAMONITOR_H_ */