NResourceMonitor.h

#ifndef Ndmspc_NResourceMonitor_H
#define Ndmspc_NResourceMonitor_H
#include <sys/resource.h>
#include <chrono>
#include <TObject.h>
#include <THnSparse.h>
 
namespace Ndmspc {

class NResourceMonitor : public TObject {
  public:
  NResourceMonitor();

  virtual ~NResourceMonitor();

  virtual void Print(Option_t * option = "") const;

  THnSparse * Initialize(THnSparse * hns);

  void Fill(Int_t * coords, int threadId);

  THnSparse * GetHnSparse() const { return fHnSparse; }

  rusage & GetUsageStart() { return fUsageStart; }

  rusage & GetUsageEnd() { return fUsageEnd; }

  double GetTimeDiffInSeconds() const;

  double GetCpuUsage() const;

  long GetMemoryUsageDiff() const { return fUsageEnd.ru_maxrss - fUsageStart.ru_maxrss; }

  void Start();

  void End();
 
  private:
  THnSparse *                                    fHnSparse{nullptr}; 
  rusage                                         fUsageStart;        
  rusage                                         fUsageEnd;          
  std::chrono::high_resolution_clock::time_point fWallStart;         
  std::chrono::high_resolution_clock::time_point fWallEnd;           
  std::vector<std::string>                       fNames = {"time", "cpu", "mem"}; 

  double timevalToDouble(const timeval & tv) const
  {
    return static_cast<double>(tv.tv_sec) + static_cast<double>(tv.tv_usec) / 1000000.0;
  }

  ClassDef(NResourceMonitor, 1);
};
} // namespace Ndmspc
#endif