CbmAnaLmvmDrawStudy.h

Go to the documentation of this file.

 
#ifndef CBM_ANA_LMVM_DRAW_STUDY
#define CBM_ANA_LMVM_DRAW_STUDY
 
#include <string>
#include <vector>
 
#include "CbmLmvmHist.h"
 
#include "TObject.h"
 
class TH1;
class TH2D;
class TH1D;
class TFile;
class TCanvas;
class CbmHistManager;
class CbmAnaPTree;
 
class CbmAnaLmvmDrawStudy : public TObject {
 
public:
  CbmAnaLmvmDrawStudy()
    : TObject()
    , fCanvas()
    , fNofStudies(0)
    , fStudyNames()
    , fHM()
    , fMeanFiles()
    , fOutputDir("") {
    ;
  }
 
  virtual ~CbmAnaLmvmDrawStudy() { ; }
 
  void DrawFromFile(const std::vector<std::string>& fileNames,
                    const std::vector<std::string>& fileNamesMean,
                    const std::vector<std::string>& studyNames,
                    const std::string& outputDir = "");
 
private:
  std::vector<TCanvas*>
    fCanvas;  // store all pointers to TCanvas -> save to images
  Int_t fNofStudies;
  std::vector<std::string> fStudyNames;
 
  std::vector<CbmHistManager*>
    fHM;  // store pointers to histogram manager for different simulations
 
  std::vector<std::string> fMeanFiles;  // Files mean
 
  std::string fOutputDir;  // output directory for figures and .json file
 
  TCanvas* CreateCanvas(const std::string& name,
                        const std::string& title,
                        int width,
                        int height);
 
  void DrawTextOnHist(const std::string& text,
                      Double_t x1,
                      Double_t y1,
                      Double_t x2,
                      Double_t y2);
 
  TH1D* H1(int studyNum, const std::string& name);
 
  TH2D* H2(int studyNum, const std::string& name);
 
  void SaveCanvasToImage();
 
  void SetAnalysisStepLabels(TH1* h);
 
  void DrawMinv();
  void DrawNofBgTracks();
  void DrawBgSourceTracks();
  void DrawBgSourcePairsStep(int step);
  void DrawBgSourcePairs();
  void DrawBgSourceMinv();
  void DrawDistributions(const std::string& canvasName,
                         const std::string& histName,
                         int step,
                         int sourceType);
 
  ClassDef(CbmAnaLmvmDrawStudy, 1);
};
 
#endif