HistoContainer.h

#ifndef HISTOCONTAINER
#define HISTOCONTAINER

#include <TH1F.h>
#include <TH2F.h>
#include <TProfile.h>
#include <map>
#include <string>

class HistoContainer {

 public:
  HistoContainer();
  HistoContainer(int,std::string);
  ~HistoContainer();
    
  void Add(char *, char*, char*, int, int, float, float);
  void Add(char *, char*, char*, int, int, float, float, int, float, float);
  void Add(char *, char*, char*, int, int, float, float, float, float);

  void Fill(std::string, int, float);
  void Fill(std::string, int, float, float);
  
  void Fill2D(std::string, int, float, float);
  void Fill2D(std::string, int, float, float, float);

  int ncat(int n);
  int size() { return (int)names.size(); }
  int nbins(int, bool);
  std::string axisName(int, bool);
  float max(int, bool);
  float min(int, bool);
  std::string getName(int n) { return names[n]; }
  
  void Save();

  int getDimension(int);
  int getHistVal();
  void setHistVal(int);
  void setHistNam(std::string);
  void setScale(float);
  std::string ModifiedName(char*, int);
  float total_scale;

 private:
  int histVal;
  std::string histNam;
  std::vector<std::string> names;
  
  std::map<std::string, std::vector<TH1F> > h1;
  std::map<std::string, std::vector<TH2F> > h2;
  std::map<std::string, std::vector<TProfile> > hp;
};

#endif