source: git/external/ExRootAnalysis/ExRootResult.h@ 3e5f05c

#ifndef ExRootResult_h
#define ExRootResult_h

#include "Gtypes.h"
#include "Rtypes.h"
#include "TMath.h"

#include <map>
#include <set>

class TH1;
class TH2;
class THStack;
class TCanvas;
class TLegend;
class TProfile;
class TPaveText;
class TObjArray;
class TFolder;

class ExRootResult
{

public:
  ExRootResult();
  ~ExRootResult();

  void Reset();
  void Write(const char *fileName = "results.root");
  void Print(const char *format = "eps");

  TH1 *AddHist1D(const char *name, const char *title,
    const char *xlabel, const char *ylabel,
    Int_t nxbins, Axis_t xmin, Axis_t xmax,
    Int_t logx = 0, Int_t logy = 0);

  TH1 *AddHist1D(const char *name, const char *title,
    const char *xlabel, const char *ylabel,
    Int_t nxbins, const Float_t *bins,
    Int_t logx = 0, Int_t logy = 0);

  TProfile *AddProfile(const char *name, const char *title,
    const char *xlabel, const char *ylabel,
    Int_t nxbins, Axis_t xmin, Axis_t xmax,
    Int_t logx = 0, Int_t logy = 0);

  TH2 *AddHist2D(const char *name, const char *title,
    const char *xlabel, const char *ylabel,
    Int_t nxbins, Axis_t xmin, Axis_t xmax,
    Int_t nybins, Axis_t ymin, Axis_t ymax,
    Int_t logx = 0, Int_t logy = 0);

  THStack *AddHistStack(const char *name, const char *title);

  TLegend *AddLegend(Double_t x1, Double_t y1, Double_t x2, Double_t y2);

  TPaveText *AddComment(Double_t x1, Double_t y1, Double_t x2, Double_t y2);

  void Attach(TObject *plot, TObject *object);

  TCanvas *GetCanvas();

  void PrintPlot(TObject *plot, const char *sufix = "", const char *format = "eps");

  void SetFolder(TFolder *folder) { fFolder = folder; }

private:
  struct PlotSettings
  {
    Int_t logx;
    Int_t logy;
    TObjArray *attachments;
  };

  void CreateCanvas();

  TCanvas *fCanvas; //!

  std::set<TObject *> fPool; //!

  std::map<TObject *, PlotSettings> fPlotMap; //!

  TFolder *fFolder; //!
};

#endif /* ExRootResult_h */