function_timer.h

#ifndef FUNCTION_TIMER_H
#define FUNCTION_TIMER_H
 
#include <chrono>
#include <iostream>
 
namespace quad_utils {
 
 
class FunctionTimer {
 public:
  FunctionTimer(const char* function_name) {
    function_name_ = const_cast<char*>(function_name);
    start_time_ = std::chrono::steady_clock::now();
  }
 
  double reportSilent() {
    stop_time_ = std::chrono::steady_clock::now();
    std::chrono::duration<double> elapsed =
        std::chrono::duration_cast<std::chrono::duration<double>>(stop_time_ -
                                                                  start_time_);
    double current_time = elapsed.count();
    return current_time;
  }
 
  double reportStatistics() {
    double current_time = reportSilent();
    printf("Time spent in %s = %.2es\n", function_name_, current_time);
    return current_time;
  }
 
  double reportStatistics(int n) {
    double avg_time = reportSilent() / n;
    printf("Average time spent in %s = %.2es\n", function_name_, avg_time);
    return avg_time;
  }
 
  void reportAndRestart() {
    reportStatistics();
    start_time_ = std::chrono::steady_clock::now();
  }
 
 private:
  std::chrono::time_point<std::chrono::steady_clock> start_time_;
 
  std::chrono::time_point<std::chrono::steady_clock> stop_time_;
 
  char* function_name_;
};
 
}  // namespace quad_utils
 
#endif  // FUNCTION_TIMER_H