CbmSimulationReport.cxx

Go to the documentation of this file.

 
#include "CbmSimulationReport.h"
 
#include "CbmDrawHist.h"     // for DrawH1, kLinear, DrawH2, HistScale
#include "CbmHistManager.h"  // for CbmHistManager
 
#include <RtypesCore.h>  // for UInt_t
#include <TFile.h>       // for TFile
#include <TH1.h>         // for TH1
#include <TH2.h>         // for TH2
 
#include <assert.h>  // for assert
#include <fstream>   // for string, ofstream
#include <string>    // for operator+
#include <vector>    // for vector, __vector_base<>::value_type
 
using std::ofstream;
using std::string;
using std::vector;
 
CbmSimulationReport::CbmSimulationReport() : CbmReport(), fHM(nullptr) {}
 
CbmSimulationReport::~CbmSimulationReport() {}
 
void CbmSimulationReport::Create(CbmHistManager* histManager,
                                 const string& outputDir) {
  assert(histManager != nullptr);
  fHM = histManager;
  SetOutputDir(outputDir);
  CreateReports();
}
 
void CbmSimulationReport::Create(const string& fileName,
                                 const string& outputDir) {
  assert(fHM == nullptr);
  fHM         = new CbmHistManager();
  TFile* file = new TFile(fileName.c_str());
  fHM->ReadFromFile(file);
  SetOutputDir(outputDir);
  CreateReports();
  //   delete fHM;
  //   delete file;
}
 
void CbmSimulationReport::DrawH1ByPattern(const string& histNamePattern) {
  vector<TH1*> histos = HM()->H1Vector(histNamePattern);
  UInt_t nofHistos    = histos.size();
  if (nofHistos < 1) return;
  for (UInt_t iHist = 0; iHist < nofHistos; iHist++) {
    TH1* hist         = histos[iHist];
    string canvasName = GetReportName() + hist->GetName();
    //      TCanvas* canvas = CreateCanvas(canvasName.c_str(), canvasName.c_str(), 800, 500);
    CreateCanvas(canvasName.c_str(), canvasName.c_str(), 800, 500);
    DrawH1(hist, kLinear, kLinear);
  }
}
 
void CbmSimulationReport::DrawH1ByPattern(
  const string& histNamePattern,
  string (*labelFormatter)(const string&, const CbmHistManager*)) {
  vector<TH1*> histos = HM()->H1Vector(histNamePattern);
  UInt_t nofHistos    = histos.size();
  if (nofHistos < 1) return;
  string canvasName = GetReportName() + histos[0]->GetName();
  //   TCanvas* canvas = CreateCanvas(canvasName.c_str(), canvasName.c_str(), 600, 500);
  CreateCanvas(canvasName.c_str(), canvasName.c_str(), 600, 500);
 
  vector<string> labels(nofHistos);
  for (UInt_t iHist = 0; iHist < nofHistos; iHist++) {
    string name   = histos[iHist]->GetName();
    labels[iHist] = labelFormatter(name, HM());
  }
 
  DrawH1(histos, labels, kLinear, kLinear, true, 0.3, 0.3, 0.85, 0.6, "PE1");
}
 
void CbmSimulationReport::DrawH2ByPattern(const string& histNamePattern,
                                          HistScale logx,
                                          HistScale logy,
                                          HistScale logz,
                                          const string& drawOpt) {
  vector<TH2*> histos = HM()->H2Vector(histNamePattern);
  UInt_t nofHistos    = histos.size();
  if (nofHistos < 1) return;
  for (UInt_t iHist = 0; iHist < nofHistos; iHist++) {
    TH2* hist         = histos[iHist];
    string canvasName = GetReportName() + hist->GetName();
    //        TCanvas* canvas = CreateCanvas(canvasName.c_str(), canvasName.c_str(), 800, 500);
    CreateCanvas(canvasName.c_str(), canvasName.c_str(), 800, 500);
    DrawH2(hist, logx, logy, logz, drawOpt);
  }
}
 
ClassImp(CbmSimulationReport)