histogramPlotter.h

#include "TH1.h"
#include <vector>
#include <map>
#include <string>
#include "plots.h"
#include "TColor.h"
#include "TPaveText.h"

typedef struct datasetInfo datasetInfo;

class HistogramPlotter{
  //A few things that govern the appearance of the plots.
  std::string lumiStr_;
  std::string outputFolder_; //Where the plots will be saved. Make sure to inclue the final /...
  std::string postfix_; //Will be appended to name of saved files. Defaults to aPostfix. Set with setter.
  
  //Orders of various things and information regarding plotting.
  std::vector<std::string> plotOrder_;
  std::vector<std::string> legOrder_;
  std::map<std::string,datasetInfo> dsetMap_;
  std::vector<std::string> extensions_; //Will default to saving root and png files.
  //Labels for the plot.
  TPaveText * labelOne_;
  TPaveText * labelTwo_;
  TPaveText * labelThree_;


 public:
 //Constructor
  HistogramPlotter(std::vector<std::string>, std::vector<std::string>, std::map<std::string,datasetInfo>);
  ~HistogramPlotter();
  //methods to set various bits of information in the class. This is so that it doesn't have to set in the constructor. Defaults to current stuff, but can be changed outside.
  void setLabelTextSize(float size);
  void setLabelOne(std::string label){labelOne_->SetLabel(label.c_str());};
  void setLabelTwo(std::string label){labelTwo_->SetLabel(label.c_str());};
  void setLabelThree(std::string label){labelThree_->SetLabel(label.c_str());};
  void setLumiStr(std::string lumiStr){lumiStr_ = lumiStr;};
  void setPostfix(std::string postfix){postfix_ = postfix;};
  void setOutputFolder(std::string output){outputFolder_ = output;};
  void changeExtensions(std::vector<std::string> extentions){extensions_ = extentions;};
  //Actual plotting commands
  void plotHistos(std::map<std::string, std::map<std::string, Plots*> >);
  void plotCutFlows(std::map<std::string, TH1F*>);
  void makePlot(std::map<std::string, TH1F*>,std::string);
  void makePlot(std::map<std::string, TH1F*>,std::string,std::string); //Adds a subLabel to the plot.
  void makePlot(std::map<std::string, TH1F*>,std::string,std::vector<std::string>); //Adds x-axis bin labels to the plot.
  void makePlot(std::map<std::string, TH1F*>,std::string,std::string,std::vector<std::string>); //Adds a subLabel AND bin labels to the plot. Might get confusing later. May come up with another name.
};

struct datasetInfo{
  int colour;
  std::string legLabel;
  std::string legType;
};