HistContainer.h

Go to the documentation of this file.

#ifndef HistContainer_h
#define HistContainer_h

#include <map>
#include <memory>
#include <string>

#include "TH1.h"
#include "TDirectoryFile.h"


namespace tvx {

typedef std::map<std::string, std::unique_ptr<TH1> >  HistMap;


class HistContainer : public TDirectoryFile
{
public:

   HistContainer(const std::string name, TDirectory* motherDir=nullptr, const std::string option="");

   void Add(TH1* hist);

   const HistMap& GetHists() const;

   const TH1& operator[](const std::string& hist_name) const;

   const TH1* FindHist(const std::string& hist_name) const;

   virtual void FillDerivedHists();

   void SaveAllAs(std::string prefix="./", std::string img_format="png");

protected:

   TH1* h(const std::string& hist_name) const;

private:

   HistMap fHs;
};


inline const HistMap& HistContainer::GetHists() const
{
   return fHs;
}


inline const TH1& HistContainer::operator[](const std::string& hist_name) const
{
   return *(fHs.at(hist_name));
}


inline const TH1* HistContainer::FindHist(const std::string& hist_name) const
{
   return h(hist_name);
}


inline void HistContainer::FillDerivedHists() {}


inline TH1* HistContainer::h(const std::string& hist_name) const
{
   auto iter = fHs.find(hist_name);
   return ( iter != fHs.end() ) ? iter->second.get() : nullptr;
}

}

#endif