ThreadTimer.h

Go to the documentation of this file.

#ifndef GALOIS_RUNTIME_THREADTIMER_H
#define GALOIS_RUNTIME_THREADTIMER_H

#include <ctime>

#include "galois/config.h"
#include "galois/substrate/PerThreadStorage.h"

namespace galois::runtime {

class ThreadTimer {
  timespec start_;
  timespec stop_;
  uint64_t nsec_{0};

public:
  ThreadTimer() = default;

  void start() { clock_gettime(CLOCK_THREAD_CPUTIME_ID, &start_); }

  void stop() {
    clock_gettime(CLOCK_THREAD_CPUTIME_ID, &stop_);
    nsec_ += (stop_.tv_nsec - start_.tv_nsec);
    nsec_ += ((stop_.tv_sec - start_.tv_sec) * 1000000000);
  }

  uint64_t get_nsec() const { return nsec_; }

  uint64_t get_sec() const { return (nsec_ / 1000000000); }

  uint64_t get_msec() const { return (nsec_ / 1000000); }
};

class ThreadTimers {
protected:
  substrate::PerThreadStorage<ThreadTimer> timers_;

  void reportTimes(const char* category, const char* region);
};

template <bool enabled>
class PerThreadTimer : private ThreadTimers {
  const char* const region_;
  const char* const category_;

  void reportTimes() { reportTimes(category_, region_); }

public:
  PerThreadTimer(const char* const region, const char* const category)
      : region_(region), category_(category) {}

  PerThreadTimer(const PerThreadTimer&) = delete;
  PerThreadTimer(PerThreadTimer&&)      = delete;
  PerThreadTimer& operator=(const PerThreadTimer&) = delete;
  PerThreadTimer& operator=(PerThreadTimer&&) = delete;

  ~PerThreadTimer() { reportTimes(); }

  void start() { timers_.getLocal()->start(); }

  void stop() { timers_.getLocal()->stop(); }
};

template <>
class PerThreadTimer<false> {

public:
  PerThreadTimer(const char* const, const char* const) {}

  PerThreadTimer(const PerThreadTimer&) = delete;
  PerThreadTimer(PerThreadTimer&&)      = delete;
  PerThreadTimer& operator=(const PerThreadTimer&) = delete;
  PerThreadTimer& operator=(PerThreadTimer&&) = delete;

  ~PerThreadTimer() = default;

  void start() const {}

  void stop() const {}
};

} // end namespace galois::runtime

#endif