timers.hpp

Go to the documentation of this file.

#ifndef MLPACK_CORE_UTILITIES_TIMERS_HPP
#define MLPACK_CORE_UTILITIES_TIMERS_HPP

#include <map>
#include <string>
#include <chrono> // chrono library for cross platform timer calculation

#if defined(_WIN32)
 // uint64_t isn't defined on every windows.
  #if !defined(HAVE_UINT64_T)
    #if SIZEOF_UNSIGNED_LONG == 8
      typedef unsigned long uint64_t;
    #else
      typedef unsigned long long  uint64_t;
    #endif  // SIZEOF_UNSIGNED_LONG
  #endif  // HAVE_UINT64_T
#endif

namespace mlpack {

class Timer
{
 public:
  static void Start(const std::string& name);

  static void Stop(const std::string& name);

  static std::chrono::microseconds Get(const std::string& name);
};

class Timers
{
 public:
  Timers() { }

  std::map<std::string, std::chrono::microseconds>& GetAllTimers();

  std::chrono::microseconds GetTimer(const std::string& timerName);

  void PrintTimer(const std::string& timerName);

  void StartTimer(const std::string& timerName);

  void StopTimer(const std::string& timerName);

  bool GetState(std::string timerName);

 private:
  std::map<std::string, std::chrono::microseconds> timers;
  std::map<std::string, bool> timerState;
  std::map<std::string, std::chrono::high_resolution_clock::time_point>
      timerStartTime;

  std::chrono::high_resolution_clock::time_point GetTime();
};

} // namespace mlpack

#endif // MLPACK_CORE_UTILITIES_TIMERS_HPP