timer.h

/*                                                                           
Developed by Sandeep Sharma and Garnet K.-L. Chan, 2012                      
Copyright (c) 2012, Garnet K.-L. Chan                                        
                                                                             
This program is integrated in Molpro with the permission of 
Sandeep Sharma and Garnet K.-L. Chan
*/


#ifndef SPIN_TIMER_HEADER_H
#define SPIN_TIMER_HEADER_H
#include <ctime>
#ifdef _OPENMP
#include <omp.h>
#endif
#include <iostream>
#include <cstdlib>
#include <cassert>
#include <execinfo.h>
#include <stdlib.h>
#include <stdio.h>
#include <sys/time.h>
#ifndef SERIAL
#include <boost/serialization/shared_ptr.hpp>
#include <boost/mpi/timer.hpp>
#include <include/communicate.h>
#endif

using namespace std;
namespace SpinAdapted{

class cumulTimer
{
 private:
  double localStart;
  double cumulativeSum;

 public:

 cumulTimer() : localStart(-1.0), cumulativeSum(0) {};

  void reset() {localStart = -1.0; cumulativeSum = 0.;}

#ifdef _OPENMP
  void start() {
    if(!omp_get_thread_num()) {
      struct timeval start;
      gettimeofday(&start, NULL);
      localStart = start.tv_sec + 1.e-6*start.tv_usec;
    }
  }
#else
  void start() {
    struct timeval start;
    gettimeofday(&start, NULL);
    localStart = start.tv_sec + 1.e-6*start.tv_usec;
  }
#endif


  void stop() 
  {
#ifdef _OPENMP
    if(!omp_get_thread_num()){
#endif


      struct timeval start;
      gettimeofday(&start, NULL);
      cumulativeSum = cumulativeSum + (start.tv_sec + 1.e-6*start.tv_usec) - localStart;
      if ((start.tv_sec + 1.e-6*start.tv_usec) - localStart < 0 ) 
	{
	  
	  cout << "local stop called without starting first"<<endl;
	  cout << localStart<<"  "<<(start.tv_sec + 1.e-6*start.tv_usec)<<endl;
	  throw 20;
	  assert(1==2);
	  abort();
	}


      localStart = 0;

#ifdef _OPENMP
    }
#endif
  }

  friend ostream& operator<<(ostream& os, const cumulTimer& t)
  {
    os << ((float)t.cumulativeSum);
    return os;
  }

};

class Timer
{
public:
  Timer(bool s) { if (s) start(); }
  Timer() { start(); }
#ifndef SERIAL
  void start() { t = boost::shared_ptr<boost::mpi::timer>(new boost::mpi::timer()); walltime = t->elapsed(); lastwalltime = walltime; wallstarttime = walltime; cputime = clock(); lastcputime = cputime; cpustarttime = cputime; }  
  double elapsedwalltime() { walltime = t->elapsed(); double elapsed = walltime - lastwalltime; lastwalltime = walltime; return elapsed; }
  double elapsedcputime() { cputime = clock(); double elapsed = double(cputime - lastcputime) / double(CLOCKS_PER_SEC); elapsed=accumclock(elapsed);lastcputime = cputime; return elapsed; }
  double totalwalltime() { return t->elapsed() - wallstarttime; }
  double totalcputime() { double total=double(clock() - cpustarttime) / double(CLOCKS_PER_SEC);total=accumclock(total);return total; }
#else
  void start() { walltime = time(NULL); lastwalltime = walltime; wallstarttime = walltime; cputime = clock(); lastcputime = cputime; cpustarttime = cputime; }  
  long elapsedwalltime() { walltime = time(NULL); time_t elapsed = walltime - lastwalltime; lastwalltime = walltime; return elapsed; }
  double elapsedcputime() { cputime = clock(); double elapsed = double(cputime - lastcputime) / double(CLOCKS_PER_SEC); lastcputime = cputime; return elapsed; }
  long totalwalltime() { return time(NULL) - wallstarttime; }
  double totalcputime() { return (double)(clock() - cpustarttime) / double(CLOCKS_PER_SEC); }
#endif
private:
#ifndef SERIAL
  boost::shared_ptr<boost::mpi::timer> t;
  double  walltime;
  double lastwalltime;
  double wallstarttime;
  double accumclock(double tcpu) 
  {
   double sum;
   all_reduce(calc, tcpu, sum, std::plus<double>());
   return sum;
  }
#else
  time_t walltime;
  time_t lastwalltime;
  time_t wallstarttime;
#endif
  clock_t cputime;
  clock_t lastcputime;
  clock_t cpustarttime;
};

void __GetMachineName(char* machineName);
void mcheck(const char* message);
void mdebugcheck(const char* message);
}
#endif