solve_profiler.hpp

// Copyright (c) Sleipnir contributors
 
#pragma once
 
#include <chrono>
#include <string>
#include <string_view>
 
namespace slp {
 
class SolveProfiler {
 public:
  std::string name;
 
  SolveProfiler() = default;
 
  explicit SolveProfiler(std::string_view name) : name{name} {}
 
  void start() {
#ifndef SLEIPNIR_DISABLE_DIAGNOSTICS
    m_current_solve_start_time = std::chrono::steady_clock::now();
#endif
  }
 
  void stop() {
#ifndef SLEIPNIR_DISABLE_DIAGNOSTICS
    m_current_solve_stop_time = std::chrono::steady_clock::now();
    m_current_solve_duration =
        m_current_solve_stop_time - m_current_solve_start_time;
    m_total_solve_duration += m_current_solve_duration;
 
    ++m_num_solves;
    m_average_solve_duration =
        (m_num_solves - 1.0) / m_num_solves * m_average_solve_duration +
        1.0 / m_num_solves * m_current_solve_duration;
#endif
  }
 
  int num_solves() const { return m_num_solves; }
 
  const std::chrono::duration<double>& current_duration() const {
    return m_current_solve_duration;
  }
 
  const std::chrono::duration<double>& average_duration() const {
    return m_average_solve_duration;
  }
 
  const std::chrono::duration<double>& total_duration() const {
    return m_total_solve_duration;
  }
 
 private:
  std::chrono::steady_clock::time_point m_current_solve_start_time;
  std::chrono::steady_clock::time_point m_current_solve_stop_time;
  std::chrono::duration<double> m_current_solve_duration{0.0};
  std::chrono::duration<double> m_total_solve_duration{0.0};
 
  int m_num_solves = 0;
  std::chrono::duration<double> m_average_solve_duration{0.0};
};
 
}  // namespace slp