Stats.h

 
#pragma once
 
#include <atomic>
#include <iosfwd>
#include <string>
 
#include "dlinear/util/Timer.h"
 
namespace dlinear {
 
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]] 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 operator++();
  void operator++(int);
};
 
std::ostream &operator<<(std::ostream &os, const Stats &stats);
std::ostream &operator<<(std::ostream &os, const IterationStats &stats);
 
}  // namespace dlinear