BmnHistManager.h

Go to the documentation of this file.

 
#ifndef BMNHISTMANAGER_H_
#define BMNHISTMANAGER_H_
 
#include "TObject.h"
#include <iostream>
#include <map>
#include "TString.h"
#include <vector>
#include <cassert>
#include <utility>
#include <functional>
 
class TFile;
class TNamed;
class TH1;
class TH2;
class TGraph;
class TGraph2D;
class TProfile;
class TProfile2D;
 
using std::map;
using std::make_pair;
using std::string;
using std::vector;
 
class BmnHistManager : public TObject
{
public:
 
   BmnHistManager();
 
   virtual ~BmnHistManager();
 
   void Add(const TString& name, TNamed* object) {
         std::pair<TString, TNamed*> newpair = std::make_pair(name, object);
      fMap.insert(newpair);
   }
 
   template<class T> void Create1(
         const TString& name,
         const TString& title,
         Int_t nofBins,
         Double_t minBin,
         Double_t maxBin) {
    T* h = new T(name.Data(), title.Data(), nofBins, minBin, maxBin);
    Add(name, h);
  }
 
   template<class T> void Create2(
         const TString& name,
         const TString& title,
         Int_t nofBinsX,
         Double_t minBinX,
         Double_t maxBinX,
         Int_t nofBinsY,
         Double_t minBinY,
         Double_t maxBinY) {
    T* h = new T(name.Data(), title.Data(), nofBinsX, minBinX, maxBinX, nofBinsY, minBinY, maxBinY);
    Add(name, h);
   }
 
   TH1* H1(const TString& name) const {
      if (fMap.count(name) == 0) { // Temporarily used for debugging
        std::cout << "Error: BmnHistManager::H1(name): name=" << name << std::endl;
      }
      assert(fMap.count(name) != 0);
      return (TH1*) fMap.find(name)->second;
   }
 
   TH2* H2(const TString& name) const {
      if (fMap.count(name) == 0) { // Temporarily used for debugging
        std::cout << "Error: BmnHistManager::H2(name): name=" << name << std::endl;
      }
      assert(fMap.count(name) != 0);
      return (TH2*) fMap.find(name)->second;
   }
 
   Bool_t Exists(const TString& name) const {
      return (fMap.count(name) == 0) ? false : true;
   }
 
   void WriteToFile();
 
   void ResetHists();
 
 
   void ReadFromFile(
         TFile* file);
 
   void Clear();
 
   void ShrinkEmptyBinsH1(const TString& histName);
 
 
   void ShrinkEmptyBinsH2(const TString& histName);
 
   void Scale(const TString& histName, Double_t scale);
   
   void NormalizeToIntegral(const TString& histName);
 
   void Rebin(const TString& histName, Int_t ngroup);
 
   TString ToString() const;
 
   friend std::ostream& operator<<(std::ostream& strm, const BmnHistManager& histManager) {
      strm << histManager.ToString();
      return strm;
   }
 
private:
 
   // Map of histogram (graph) name to its pointer
   map<TString, TNamed*> fMap;
 
   ClassDef(BmnHistManager, 1)
};
 
#endif /* BmnHISTMANAGER_H_ */