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