timings.h

Go to the documentation of this file.

 
#ifndef timings_h
#define timings_h
 
#include <iostream>
#include "arts.h"
#include "arts_omp.h"
 
 
class Timings {
 public:
  friend std::ostream &operator<<(std::ostream &os, const Timings &timer);
 
 
  Timings() : running(), times(), stamps(), names() {
    ntimers = 0;
    nrunning = 0;
    stamp = 0.0;
    total_time = 0.0;
  }
 
 
  Index add_timer(String name) {
    ntimers++;
    running.push_back(false);
    times.push_back(0.0);
    names.push_back(name);
    stamps.push_back(0.0);
 
    return (Index)ntimers - 1;
  }
 
 
  Numeric get_time() {
#ifdef OMP
    return omp_get_wtime();
#else
    return 0.0;
#endif
  }
 
 
  void mark(Index i) {
    if (i < ntimers) {
      // Stop timer if it is running and check if overall timer
      // must be stopped.
 
      if (running[i]) {
        Numeric t = get_time();
        times[i] += t - stamps[i];
        running[i] = false;
 
        nrunning--;
        if (nrunning == 0) total_time += get_time() - stamp;
 
      } else  // Continue timer if not running.
      {
        if (nrunning == 0) stamp = get_time();
 
        nrunning++;
 
        stamps[i] = get_time();
        running[i] = true;
      }
    }
  }
 
 private:
  Index ntimers, nrunning;
  Numeric total_time, stamp;
  std::vector<bool> running;
  std::vector<Numeric> times, stamps;
  std::vector<String> names;
};
 
#endif  // timer_h