HGTD_AnalysisAlgBase.h

Go to the documentation of this file.

 
#ifndef HGTD_ANALYSIS_ANALYSISALGBASE_H
#define HGTD_ANALYSIS_ANALYSISALGBASE_H
 
#include "AthenaBaseComps/AthAlgorithm.h"
 
// Athena includes
#include "GaudiKernel/ITHistSvc.h"
#include "GaudiKernel/ServiceHandle.h"
 
// ROOT includes
#include "TEfficiency.h"
#include "TH1.h"
 
// stl includes
#include <map>
#include <string>
#include <string_view>
#include <format>
 
class HGTD_AnalysisAlgBase : public AthAlgorithm {
 
public:
  HGTD_AnalysisAlgBase(const std::string& name, ISvcLocator* svc_locator);
  virtual ~HGTD_AnalysisAlgBase();
 
  virtual StatusCode initialize();
 
  Gaudi::Property<std::string> m_directory_name{
      this, "DirectoryName", "/HGTD_ANA/", "The output directory name"};

  template <typename T = TH1F, typename... Ts>
  void bookSubdir(const std::string& trk_sel_name, const std::string& time_wp,
                  const std::string& hist_name, const std::string& title,
                  Ts... args) {
    // check if hist exists already and warn for duplication
    // doesn't do anything then...
    const std::string name = trk_sel_name + "/" + time_wp + "/" + hist_name;
    if (m_histos.contains(name)) {
      ATH_MSG_WARNING("You are duplicating histogram: "
                      << name << ", this is not a good idea");
      return;
    }
    auto * ptrT = new T(hist_name.c_str(), title.c_str(), args...);
    m_histos.emplace(name, ptrT);
    ptrT->Sumw2();
    if (not m_hist_svc->regHist(m_directory_name + name, ptrT).isSuccess()) {
      ATH_MSG_WARNING("Failed to book " << name);
    }
  }
 
  template <typename T, typename... Ts>
  void book(const std::string& name, const std::string& title, Ts... args) {
    // check if hist exists already and warn for duplication
    // doesn't do anything then...
    if (m_histos.contains(name)) {
      ATH_MSG_WARNING("You are duplicating histogram: "
                      << name << ", this is not a good idea");
      return;
    }
    auto * ptrT = new T(name.c_str(), title.c_str(), args...);
    m_histos.emplace(name, ptrT);
    ptrT->Sumw2();
    if (not m_hist_svc->regHist(m_directory_name + name,ptrT).isSuccess()) {
      ATH_MSG_WARNING("Failed to book " << name);
    }
  }
 
  template <typename... Ts>
  void bookEffSubdir(const std::string& trk_sel_name,
                     const std::string& time_wp, const std::string& hist_name,
                     const std::string& title, Ts... args) {
    // check if hist exists already and warn for duplication
    // doesn't do anything then...
    const std::string name = trk_sel_name + "/" + time_wp + "/" + hist_name;
    if (m_histos.contains(name)) {
      ATH_MSG_WARNING("You are duplicating histogram: "
                      << name << ", this is not a good idea");
      return;
    }
    auto * ptrT = new TEfficiency(hist_name.c_str(), title.c_str(), args...);
    m_histos.emplace(name, ptrT);
    if (not m_hist_svc->regEfficiency(m_directory_name + name, ptrT).isSuccess()) {
      ATH_MSG_WARNING("Failed to book " << name);
    }
  }
 
  template <typename... Ts>
  void bookEff(const std::string& name, const std::string& title, Ts... args) {
    // check if hist exists already and warn for duplication
    // doesn't do anything then...
    if (m_histos.contains(name)) {
      ATH_MSG_WARNING("You are duplicating histogram: "
                      << name << ", this is not a good idea");
      return;
    }
    auto * ptrT = new TEfficiency(name.c_str(), title.c_str(), args...);
    m_histos.emplace(name,ptrT);
    if (not m_hist_svc->regEfficiency(m_directory_name + name,ptrT).isSuccess()) {
      ATH_MSG_WARNING("Failed to book " << name);
    }
  }

  template <typename T, typename... Ts>
  void fill(std::string_view name, Ts... args) {
    auto it = m_histos.find(name);
    if (it == m_histos.end() or it->second == nullptr) {
      ATH_MSG_WARNING(
          "[HistogramHandler::fill] ERROR: you are attempting to fill "
          "a histogram with name "
          << name << " which doesn't exist!");
      return;
    }
    auto * ptrT = dynamic_cast<T*>(it->second);
    if (!ptrT)[[unlikely]]{
      ATH_MSG_WARNING("Cast failed for "<< name);
      return;
    }
    ptrT->Fill(args...);
  }
 
  template <typename T, typename... Ts>
  void fillSubdir(std::string_view trk_sel_name, std::string_view time_wp,
                  std::string_view hist_name, Ts... args) {
    const std::string name = std::format("{}/{}/{}", trk_sel_name, time_wp, hist_name);
    auto it = m_histos.find(name);
    if (it == m_histos.end() or it->second == nullptr) {
      ATH_MSG_WARNING(
          "[HistogramHandler::fill] ERROR: you are attempting to fill "
          "a histogram with name "
          << name << " which doesn't exist!");
      return;
    }
    auto * ptrT = dynamic_cast<T*>(it->second);
    if (!ptrT)[[unlikely]]{
      ATH_MSG_WARNING("Cast failed for "<< name);
      return;
    }
    ptrT->Fill(args...);
  }
 
  template <typename... Ts> 
  void fillEff(std::string_view name, Ts... args) {
    const auto it = m_histos.find(name);
    if (it == m_histos.end() or it->second == nullptr)[[unlikely]] {
      ATH_MSG_WARNING(
          "[HistogramHandler::fillEff] ERROR: you are attempting to fill "
          "a histogram with name "
          << name << " which doesn't exist!");
      return;
    }
    auto * ptrT = dynamic_cast<TEfficiency*>(it->second);
    if (!ptrT)[[unlikely]]{
      ATH_MSG_WARNING("Cast failed for "<< name);
      return;
    }
    ptrT->Fill(args...);
  }
 
  template <typename... Ts>
  void fillEffSubDir(std::string_view trk_sel_name,
                     std::string_view wp_name, std::string_view hist_name,
                     Ts... args) {
    const std::string name = std::format("{}/{}/{}", trk_sel_name, wp_name, hist_name);
    auto it = m_histos.find(name) ;
    if (it == m_histos.end() or it->second == nullptr) {
      ATH_MSG_WARNING(
          "[HistogramHandler::fill] ERROR: you are attempting to fill "
          "a histogram with name "
          << name << " which doesn't exist!");
      return;
    }
    auto * ptrT = dynamic_cast<TEfficiency*>(it->second);
    if (!ptrT)[[unlikely]]{
      ATH_MSG_WARNING("Cast failed for "<< name);
      return;
    }
    ptrT->Fill(args...);
  }
 
protected:
  ServiceHandle<ITHistSvc> m_hist_svc{this, "THistSvc", "THistSvc"};
 
private:
  std::map<std::string, TObject*, std::less<>> m_histos;
};
 
#endif // HGTD_ANALYSIS_ANALYSISALGBASE_H