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