HistogramPool.h

#ifndef __FRAMEWORK_HISTOGRAM_POOL_H__
#define __FRAMEWORK_HISTOGRAM_POOL_H__
 
//----------------//
//   C++ StdLib   //
//----------------//
#include <functional>
#include <unordered_map>
 
//----------//
//   ROOT   //
//----------//
#include "Framework/Configure/Parameters.h"
#include "Framework/Exception/Exception.h"
#include "TDirectory.h"
#include "TH1F.h"
#include "TH2F.h"
 
namespace framework {
 
class HistogramPool {
 private:
  double the_weight_{1.};
  std::unordered_map<std::string, TH1*> histograms_;
  std::function<TDirectory*()> get_directory_;
 
  void insert(const std::string& name, std::function<TH1*()> factory,
              bool weighted);
 
 public:
  HistogramPool(std::function<TDirectory*()> get_directory)
      : get_directory_{get_directory} {}
 
  void setWeight(double w) { the_weight_ = w; }
 
  TH1* get(const std::string& name);
 
  void create(const config::Parameters& p);
 
  void create(const std::string& name, const std::string& x_label,
              const int& bins, const double& xmin, const double& xmax,
              bool weighted = false);
 
  void create(const std::string& name, const std::string& x_label,
              const std::vector<double>& bins, bool weighted = false);
 
  void create(const std::string& name, const std::string& x_label,
              const std::vector<std::string>& categories,
              bool weighted = false);
 
  void create(const std::string& name, const std::string& x_label,
              const int& xbins, const double& xmin, const double& xmax,
              const std::string& y_label, const int& ybins, const double& ymin,
              const double& ymax, bool weighted = false);
 
  void create(const std::string& name, const std::string& x_label,
              const std::vector<double>& xbins, const std::string& y_label,
              const int& ybins, const double& ymin, const double& ymax,
              bool weighted = false);
 
  void create(const std::string& name, const std::string& x_label,
              const std::vector<std::string>& xcategories,
              const std::string& y_label, const int& ybins, const double& ymin,
              const double& ymax, bool weighted = false);
 
  void create(const std::string& name, const std::string& x_label,
              const int& xbins, const double& xmin, const double& xmax,
              const std::string& y_label, const std::vector<double>& ybins,
              bool weighted = false);
 
  void create(const std::string& name, const std::string& x_label,
              const std::vector<double>& xbins, const std::string& y_label,
              const std::vector<double>& ybins, bool weighted = false);
 
  void create(const std::string& name, const std::string& x_label,
              const std::vector<std::string>& xcategories,
              const std::string& y_label, const std::vector<double>& ybins,
              bool weighted = false);
 
  void create(const std::string& name, const std::string& x_label,
              const int& xbins, const double& xmin, const double& xmax,
              const std::string& y_label,
              const std::vector<std::string>& ycategories,
              bool weighted = false);
 
  void create(const std::string& name, const std::string& x_label,
              const std::vector<double>& xbins, const std::string& y_label,
              const std::vector<std::string>& ycategories,
              bool weighted = false);
 
  void create(const std::string& name, const std::string& x_label,
              const std::vector<std::string>& xcategories,
              const std::string& y_label,
              const std::vector<std::string>& ycategories,
              bool weighted = false);
 
  template <typename T>
  void fill(const std::string& name, const T& val) {
    auto hist = dynamic_cast<TH1F*>(this->get(name));
    if (hist) {
      hist->Fill(val, the_weight_);
    } else {
      // the `get` method handles checking if the histogram exists
      // the only way that hist would be null at this point is if its not a TH1F
      EXCEPTION_RAISE(
          "BadHistSize",
          "Attempting to 1D fill a histogram that is not actually 1D.");
    }
  }
 
  template <typename T>
  void fillw(const std::string& name, const T& val, double w) {
    auto hist = dynamic_cast<TH1F*>(this->get(name));
    if (hist) {
      hist->Fill(val, w);
    } else {
      // the `get` method handles checking if the histogram exists
      // the only way that hist would be null at this point is if its not a TH1F
      EXCEPTION_RAISE(
          "BadHistSize",
          "Attempting to 1D fill a histogram that is not actually 1D.");
    }
  }
 
  template <typename Tx, typename Ty>
  void fill(const std::string& name, const Tx& valx, const Ty& valy) {
    auto hist = dynamic_cast<TH2F*>(this->get(name));
    if (hist) {
      hist->Fill(valx, valy, the_weight_);
    } else {
      // the `get` method handles checking if the histogram exists
      // the only way that hist would be null at this point is if its not a TH2F
      EXCEPTION_RAISE(
          "BadHistSize",
          "Attempting to 2D fill a histogram that is not actually 2D.");
    }
  }
 
  template <typename Tx, typename Ty>
  void fillw(const std::string& name, const Tx& valx, const Ty& valy,
             double w) {
    auto hist = dynamic_cast<TH2F*>(this->get(name));
    if (hist) {
      hist->Fill(valx, valy, w);
    } else {
      // the `get` method handles checking if the histogram exists
      // the only way that hist would be null at this point is if its not a TH2F
      EXCEPTION_RAISE(
          "BadHistSize",
          "Attempting to 2D fill a histogram that is not actually 2D.");
    }
  }
};
}  // namespace framework
 
#endif  // __FRAMEWORK_HISTOGRAM_POOL_H__