toolbox/resourceMonitor.hh

Go to the documentation of this file.

/* resource monitor
 */

#ifndef resourceMonitor_hh
#define resourceMonitor_hh

#include <iostream>

#include <sys/time.h>
#include <sys/resource.h>
#include <unistd.h>
#include <string.h>
#include <time.h>

extern "C" { 
#ifdef __sun__
  // Solaris 2.5 does not have getrusage in <sys/resource.h>
  int getrusage(int who, struct rusage* ru);
#endif
}

namespace concepts {

  // ******************************************************* ResourceMonitor **

  class ResourceMonitor {
    friend std::ostream& operator <<(std::ostream& os, ResourceMonitor& rm) { 
      return rm.sample(os); }

    rusage ru_;
  public:
    ResourceMonitor() { getrusage(RUSAGE_SELF, &ru_); }

    std::ostream& sample(std::ostream& os);

    float utime();
  };
  
}

class ResourceTimer {
public:

  ResourceTimer() {
    memset(&t_old, 0, sizeof(t_old));
    memset(&t_cur, 0, sizeof(t_cur));
    start();
  }

  ~ResourceTimer() { }

  void start();
  double stop(int nrun = 1);

  double reset(int nrun = 1) {
    double t = stop(nrun);
    //start();
    t_old = t_cur;
    return t;
  }


  timespec t_old;
  timespec t_cur;
};

#endif // resourceMonitor_hh