stats.h

 
#pragma once
 
#include <atomic>
#include <iostream>
#include <string>
 
#include "smats/util/timer.h"
 
namespace smats {
 
class Stats {
 public:
  explicit Stats(bool enabled, std::string class_name, std::string name_time = "Time spent in Operations");
  Stats(const Stats &other) = default;
  Stats &operator=(const Stats &other) = default;
  virtual ~Stats() = default;
 
  Stats &operator+=(const Stats &other);
  Stats operator+(const Stats &other) const;
 
  [[nodiscard]] bool enabled() const { return enabled_; }
  [[nodiscard]] Timer &m_timer() { return timer_; }
  [[nodiscard]] const Timer &timer() const { return timer_; }
  [[nodiscard]] virtual std::string to_segment_string() const;
  [[nodiscard]] virtual std::string to_string() const;
 
 protected:
  bool enabled_;                 
  std::string class_name_;       
  std::string operations_name_;  
 
 private:
  Timer timer_;  
};
 
class IterationStats : public Stats {
 public:
  explicit IterationStats(bool enabled, std::string class_name, std::string name_time = "Time spent in Operations",
                          std::string iterations_name = "Total # of Iterations");
  IterationStats(const IterationStats &other);
  IterationStats &operator=(const IterationStats &other);
 
  IterationStats &operator+=(const IterationStats &other);
  IterationStats operator+(const IterationStats &other) const;
 
  [[nodiscard]] std::string to_segment_string() const override;
 
  [[nodiscard]] std::string to_string() const override;
 
  void increase();
 
  [[nodiscard]] unsigned int iterations() const { return iterations_.load(); }
 
  void operator++();
  void operator++(int);
 
 private:
  std::atomic<unsigned int> iterations_;  
  std::string iterations_name_;           
};
 
std::ostream &operator<<(std::ostream &os, const Stats &stats);
std::ostream &operator<<(std::ostream &os, const IterationStats &stats);
 
}  // namespace smats