CbmSimulationReport.cxx

Go to the documentation of this file.

/* Copyright (C) 2011-2021 GSI/JINR-LIT, Darmstadt/Dubna
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Semen Lebedev, Andrey Lebedev [committer], Florian Uhlig */

#include "CbmSimulationReport.h"
 
#include "CbmDrawHist.h"     // for DrawH1, kLinear, DrawH2, HistScale
#include "CbmHistManager.h"  // for CbmHistManager
 
#include <RtypesCore.h>  // for UInt_t
#include <TDirectory.h>  // for TDirectory, gDirectory
#include <TFile.h>       // for TFile, gFile
#include <TH1.h>         // for TH1
#include <TH2.h>         // for TH2
 
#include <cassert>  // 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* oldFile     = gFile;
  TDirectory* oldDir = gDirectory;
 
  TFile* file = new TFile(fileName.c_str());
  fHM->ReadFromFile(file);
  SetOutputDir(outputDir);
  CreateReports();
  //   delete fHM;
  //   delete file;
 
  // shouldn't the file be closed ????
  //  file->Close();
  gFile      = oldFile;
  gDirectory = oldDir;
}
 
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)