CbmHistManager.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file CbmHistManager.h

#ifndef CBMHISTMANAGER_H_
#define CBMHISTMANAGER_H_

#include "TObject.h"
#include <iostream>
#include <map>
#include <string>
#include <vector>
#include <cassert>

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 CbmHistManager : public TObject
{
public:

   CbmHistManager();

   virtual ~CbmHistManager();

   void Add(
         const string& name,
         TNamed* object) {
     //fMap.insert(std::make_pair<string, TNamed*>(name, object));
   }

   template<class T> void Create1(
         const string& name,
         const string& title,
         Int_t nofBins,
         Double_t minBin,
         Double_t maxBin) {
    T* h = new T(name.c_str(), title.c_str(), nofBins, minBin, maxBin);
    Add(name, h);
  }

   template<class T> void Create2(
         const string& name,
         const string& title,
         Int_t nofBinsX,
         Double_t minBinX,
         Double_t maxBinX,
         Int_t nofBinsY,
         Double_t minBinY,
         Double_t maxBinY) {
    T* h = new T(name.c_str(), title.c_str(), nofBinsX, minBinX, maxBinX, nofBinsY, minBinY, maxBinY);
    Add(name, h);
   }

   TH1* H1(
         const string& name) const {
      if (fMap.count(name) == 0) { // Temporarily used for debugging
        std::cout << "Error: CbmHistManager::H1(name): name=" << name << std::endl;
      }
      assert(fMap.count(name) != 0);
      return (TH1*) fMap.find(name)->second;
   }

   vector<TH1*> H1Vector(
         const string& pattern) const;

   TH2* H2(
         const string& name) const {
      if (fMap.count(name) == 0) { // Temporarily used for debugging
        std::cout << "Error: CbmHistManager::H2(name): name=" << name << std::endl;
      }
      assert(fMap.count(name) != 0);
      return (TH2*) fMap.find(name)->second;
   }

   vector<TH2*> H2Vector(
         const string& pattern) const;

   TGraph* G1(
         const string& name) const {
      if (fMap.count(name) == 0) { // Temporarily used for debugging
        std::cout << "Error: CbmHistManager::G1(name): name=" << name << std::endl;
      }
      assert(fMap.count(name) != 0);
      return (TGraph*) fMap.find(name)->second;
   }

   vector<TGraph*> G1Vector(
         const string& pattern) const;

   TGraph2D* G2(
         const string& name) const {
      if (fMap.count(name) == 0) { // Temporarily used for debugging
        std::cout << "Error: CbmHistManager::G2(name): name=" << name << std::endl;
      }
      assert(fMap.count(name) != 0);
      return (TGraph2D*) fMap.find(name)->second;
   }

   vector<TGraph2D*> G2Vector(
         const string& pattern) const;

   TProfile* P1(
         const string& name) const {
      if (fMap.count(name) == 0) { // Temporarily used for debugging
        std::cout << "Error: CbmHistManager::P1(name): name=" << name << std::endl;
      }
      assert(fMap.count(name) != 0);
      return (TProfile*) fMap.find(name)->second;
   }

   vector<TProfile*> P1Vector(
         const string& pattern) const;

   TProfile2D* P2(
         const string& name) const {
      if (fMap.count(name) == 0) { // Temporarily used for debugging
        std::cout << "Error: CbmHistManager::P2(name): name=" << name << std::endl;
      }
      assert(fMap.count(name) != 0);
      return (TProfile2D*) fMap.find(name)->second;
   }

   vector<TProfile2D*> P2Vector(
         const string& pattern) const;

   Bool_t Exists(
         const string& name) const {
      return (fMap.count(name) == 0) ? false : true;
   }

   void WriteToFile();

   void ReadFromFile(
         TFile* file);

   void Clear();

   void ShrinkEmptyBins(
         const string& histName);

   void ShrinkEmptyBinsByPattern(
         const string& pattern);

   void Scale(
         const string& histName,
         Double_t scale);

   void ScaleByPattern(
         const string& pattern,
         Double_t scale);

   void Rebin(
         const string& histName,
         Int_t ngroup);

   void RebinByPattern(
         const string& pattern,
         Int_t ngroup);

   string ToString() const;

   friend std::ostream& operator<<(std::ostream& strm, const CbmHistManager& histManager) {
      strm << histManager.ToString();
      return strm;
   }

private:
   template<class T> vector<T> ObjectVector(
         const string& pattern) const;

   // Map of histogram (graph) name to its pointer
   map<string, TNamed*> fMap;

   ClassDef(CbmHistManager, 1)
};

#endif /* CBMHISTMANAGER_H_ */