Stats.h

#pragma once
 
#include <atomic>
#include <iosfwd>
#include <string>
 
#include "delpi/util/Timer.h"
 
namespace delpi {

class Stats {
 private:
  Timer timer_;  
 
 protected:
  bool enabled_;            
  std::string class_name_;  
  std::string operations_name_;  
 
 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]] const std::string &class_name() const { return class_name_; }
  [[nodiscard]] virtual std::string ToSegmentString() const;
  [[nodiscard]] virtual std::string ToString() const;
};

class IterationStats : public Stats {
 private:
  std::atomic<unsigned int> iterations_;  
  std::string iterations_name_;           
 
 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 ToSegmentString() const override;
 
  [[nodiscard]] std::string ToString() const override;

  void Increase();

  [[nodiscard]] unsigned int iterations() const { return iterations_.load(); }
  void SetIterations(const unsigned int iterations) { iterations_ = iterations; }
 
  void operator++();
  void operator++(int);
};
 
std::ostream &operator<<(std::ostream &os, const Stats &stats);
std::ostream &operator<<(std::ostream &os, const IterationStats &stats);
 
}  // namespace delpi
 
#ifdef DELPI_INCLUDE_FMT
 
#include "delpi/util/logging.h"
 
OSTREAM_FORMATTER(delpi::Stats);
OSTREAM_FORMATTER(delpi::IterationStats);
 
#endif