scoped_profiler.hpp

// Copyright (c) Sleipnir contributors
 
#pragma once
 
#include <concepts>
#include <utility>
 
#include "sleipnir/util/setup_profiler.hpp"
#include "sleipnir/util/solve_profiler.hpp"
 
namespace slp {
 
template <typename Profiler>
  requires std::same_as<Profiler, SetupProfiler> ||
           std::same_as<Profiler, SolveProfiler>
class ScopedProfiler {
 public:
  explicit ScopedProfiler(Profiler& profiler) noexcept : m_profiler{&profiler} {
    m_profiler->start();
  }
 
  ~ScopedProfiler() {
    if (m_active) {
      m_profiler->stop();
    }
  }
 
  ScopedProfiler(ScopedProfiler&& rhs) noexcept
      : m_profiler{std::move(rhs.m_profiler)}, m_active{rhs.m_active} {
    rhs.m_active = false;
  }
 
  ScopedProfiler(const ScopedProfiler&) = delete;
  ScopedProfiler& operator=(const ScopedProfiler&) = delete;
 
  void stop() {
    if (m_active) {
      m_profiler->stop();
      m_active = false;
    }
  }
 
  const std::chrono::duration<double>& current_duration() const {
    return m_profiler->current_duration();
  }
 
 private:
  Profiler* m_profiler;
  bool m_active = true;
};
 
}  // namespace slp