source: lemon-0.x/src/work/marci/time_measure.h @ 206:47f62d789fe7

// -*- c++ -*-
#ifndef TIME_MEASURE_H
#define TIME_MEASURE_H

#include <sys/time.h>
#include <sys/times.h>
#include <fstream>
#include <iostream>
#include <unistd.h>

double currTime() {
  timeval tv;
  //timezone tz;
  gettimeofday(&tv, 0);
  return double(tv.tv_sec)+double(tv.tv_usec)/1000000.0;
}

class TimeStamp
{
  tms ts;
  double real_time;
  
public:

  tms &getTms() {return ts;}
  const tms &getTms() const {return ts;}
  double getRealTime() const {return real_time;}
  void stamp()
  {
    timeval tv;
    times(&ts);
    gettimeofday(&tv, 0);real_time=tv.tv_sec+double(tv.tv_usec)/1e6;
  }
  
  TimeStamp()
  { ts.tms_utime=ts.tms_stime=ts.tms_cutime=ts.tms_cstime=0; real_time=0;}
  
  TimeStamp(void *) { stamp();}
  
  TimeStamp &operator+=(const TimeStamp &b)
  {
    ts.tms_utime+=b.ts.tms_utime;
    ts.tms_stime+=b.ts.tms_stime;
    ts.tms_cutime+=b.ts.tms_cutime;
    ts.tms_cstime+=b.ts.tms_cstime;
    real_time+=b.real_time;
    return *this;
  }
  TimeStamp operator+(const TimeStamp &b) const
  {
    TimeStamp t(*this);
    return t+=b;
  }
  TimeStamp &operator-=(const TimeStamp &b)
  {
    ts.tms_utime-=b.ts.tms_utime;
    ts.tms_stime-=b.ts.tms_stime;
    ts.tms_cutime-=b.ts.tms_cutime;
    ts.tms_cstime-=b.ts.tms_cstime;
    real_time-=b.real_time;
    return *this;
  }
  TimeStamp operator-(const TimeStamp &b) const
  {
    TimeStamp t(*this);
    return t-=b;
  }

  TimeStamp Ellapsed() const
  {
    TimeStamp t(NULL);
    return t-*this;
  }
  
  friend std::ostream& operator<<(std::ostream& os,const TimeStamp &t);
  
  double getUserTime() const
  {
    return double(ts.tms_utime)/sysconf(_SC_CLK_TCK);
  }
  double getSystemTime() const
  {
    return double(ts.tms_stime)/sysconf(_SC_CLK_TCK);
  }
  double getCUserTime() const
  {
    return double(ts.tms_cutime)/sysconf(_SC_CLK_TCK);
  }
  double getCSystemTime() const
  {
    return double(ts.tms_cstime)/sysconf(_SC_CLK_TCK);
  }
};

class Timer
{
  TimeStamp start_time;

  void _reset() {start_time.stamp();}
  
public: 
  Timer() {_reset();}

  operator TimeStamp ()
  {
    TimeStamp t;
    t.stamp();
    return t-start_time;
  }

  TimeStamp reset()
  {
    TimeStamp t(start_time);
    _reset();
    return start_time-t;
  }
};

inline std::ostream& operator<<(std::ostream& os,const TimeStamp &t)
{
  long cls = sysconf(_SC_CLK_TCK);
  os << "u: " << double(t.getTms().tms_utime)/cls <<
    "s, s: " << double(t.getTms().tms_stime)/cls <<
    "s, cu: " << double(t.getTms().tms_cutime)/cls <<
    "s, cs: " << double(t.getTms().tms_cstime)/cls <<
    "s, real: " << t.getRealTime() << "s";
  return os;
}

#endif //TIME_MEASURE_H