PlotManager.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
 */
 
#ifndef PLOTMANAGER_H_
#define PLOTMANAGER_H_
 
#include <string>
#include <unordered_map>
 
class TFile;
class TH1;
class TDirectory;
 
namespace EL {
  class Worker;
}
 
namespace top {
  class PlotManager {
  public:
    explicit PlotManager(const std::string& name, TFile* outputFile = nullptr, EL::Worker* wk = nullptr);
 
    PlotManager(const std::string& sample, const std::string& channel, const std::string& syst, TFile* outputFile);
 
    void addHist(const std::string& hname, const std::string& title, int bins, double start, double end) const;
 
    void addHist(const std::string& hname, const std::string& title, int bins, double* binArray) const;
 
    void addHist(const std::string& hname, const std::string& title,
                 int xbins, double xstart, double xend,
                 int ybins, double ystart, double yend) const;
 
    void addHist(const std::string& hname, const std::string& title,
                 int xbins, double* xbinArray,
                 int ybins, double* ybinArray) const;
 
    void addHist(const std::string& hname, const std::string& title,
                 int xbins, double* xbinArray,
                 int ybins, double ystart, double yend) const;
 
    TH1* hist(const std::string& name) const;
 
    void scaleHistograms(double sf = 1.) const;
  private:
    void checkDuplicate(const std::string& hname) const;
 
    EL::Worker* m_wk;
 
    std::string m_name;
 
    mutable std::unordered_map<std::string, TH1*> m_histograms;
 
    mutable TDirectory* m_plotDir;
  };
}
 
#endif