Histograms.h

#ifndef __FRAMEWORK_HISTOGRAM_POOL_H__
#define __FRAMEWORK_HISTOGRAM_POOL_H__
 
//----------------//
//   C++ StdLib   //
//----------------//
#include <iostream>
#include <memory>
#include <unordered_map>
 
//----------//
//   ROOT   //
//----------//
#include "TH1F.h"
#include "TH2F.h"
 
namespace framework {
 
class HistogramPool {
 private:
  std::unordered_map<std::string, TH1*> histograms_;
 
  HistogramPool();
 
 public:
  HistogramPool(HistogramPool const&) = delete;
 
  void operator=(HistogramPool const&) = delete;
 
  static HistogramPool& getInstance();
 
  void insert(const std::string& name, TH1* hist) { histograms_[name] = hist; }
 
  TH1* get(const std::string& name);
 
};  // HistogramPool
 
class HistogramHelper {
 private:
  double theWeight_{1.};
 
  std::string name_;
 
 public:
  HistogramHelper(const std::string& name) : name_(name) {}
 
  void setWeight(double w) { theWeight_ = w; }
 
  void create(const std::string& name, const std::string& xLabel,
              const double& bins, const double& xmin, const double& xmax);
 
  void create(const std::string& name, const std::string& xLabel,
              const std::vector<double>& bins);
 
  void create(const std::string& name, const std::string& xLabel,
              const double& xbins, const double& xmin, const double& xmax,
              const std::string& yLabel, const double& ybins,
              const double& ymin, const double& ymax);
 
  void create(const std::string& name, const std::string& xLabel,
              const std::vector<double>& xbins, const std::string& yLabel,
              const std::vector<double>& ybins);
 
  void fill(const std::string& name, const double& val) {
    auto hist = dynamic_cast<TH1F*>(this->get(name));
    if (hist) {
      hist->Fill(val, theWeight_);
    }
  }
 
  void fill(const std::string& name, const double& valx, const double& valy) {
    auto hist = dynamic_cast<TH2F*>(this->get(name));
    if (hist) {
      hist->Fill(valx, valy, theWeight_);
    }
  }
 
  TH1* get(const std::string& name) {
    return HistogramPool::getInstance().get(name_ + "_" + name);
  }
};
}  // namespace framework
 
#endif  // __FRAMEWORK_HISTOGRAM_POOL_H__