CbmUtils.cxx

Go to the documentation of this file.

/* Copyright (C) 2010-2021 GSI/JINR-LIT, Darmstadt/Dubna
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Andrey Lebedev [committer], Florian Uhlig */
 
#include "CbmUtils.h"
 
#include "boost/filesystem.hpp"
 
#include <RtypesCore.h>  // for Int_t
#include <TAxis.h>       // for TAxis
#include <TCanvas.h>     // for TCanvas
#include <TH1.h>         // for TH1D, TH1
#include <TH2.h>         // for TH2, TH2D
#include <TSystem.h>     // for TSystem, gSystem
 
#include <cstddef>  // for size_t
#include <string>   // for operator+, allocator, operator!=, char_traits
#include <vector>   // for vector
 
using boost::filesystem::path;
using std::string;
using std::vector;
 
namespace Cbm
{
 
  CbmMCDataArray* InitOrFatalMc(const std::string& objName, const std::string& description)
  {
    CbmMCDataManager* mcManager = GetOrFatal<CbmMCDataManager>("MCDataManager", description);
    CbmMCDataArray* array       = mcManager->InitBranch(objName.c_str());
    if (array == nullptr) { LOG(fatal) << description << " No MCTrack!"; }
    return array;
  }
 
  void SaveCanvasAsImage(TCanvas* c, const string& dir, const string& option)
  {
    if (dir == "") return;
    SaveCanvasAsImageImpl("eps", c, dir, option);
    SaveCanvasAsImageImpl("png", c, dir, option);
    SaveCanvasAsImageImpl("gif", c, dir, option);
    SaveCanvasAsImageImpl("pdf", c, dir, option);
    SaveCanvasAsImageImpl("svg", c, dir, option);
  }
 
  void SaveCanvasAsImageImpl(const string& imageType, TCanvas* c, const string& dir, const string& option)
  {
    if (dir == "") return;
    if (option.find(imageType) != string::npos) {
      path fullPath  = path(dir + "/" + imageType + "/" + string(c->GetTitle()) + "." + imageType);
      string fullDir = fullPath.parent_path().string();
      if (fullDir.length() > 0) {
        gSystem->mkdir(fullDir.c_str(), true);  // create directory if it does not exist
      }
      c->SaveAs(fullPath.c_str());
    }
  }
 
  string FindAndReplace(const string& name, const string& oldSubstr, const string& newSubstr)
  {
    string newName = name;
    Int_t startPos = name.find(oldSubstr);
    newName.replace(startPos, oldSubstr.size(), newSubstr);
    return newName;
  }
 
  vector<string> Split(const string& name, char delimiter)
  {
    vector<string> result;
    std::size_t begin = 0;
    std::size_t end   = name.find_first_of(delimiter);
    while (end != string::npos) {
      string str = name.substr(begin, end - begin);
      if (str[0] == delimiter) str.erase(0, 1);
      result.push_back(str);
      begin = end;
      end   = name.find_first_of(delimiter, end + 1);
    }
    result.push_back(name.substr(begin + 1));
    return result;
  }
 
 
  TH1D* DivideH1(TH1* h1, TH1* h2, const string& histName, double scale, const string& titleYaxis)
  {
    int nBins    = h1->GetNbinsX();
    double min   = h1->GetXaxis()->GetXmin();
    double max   = h1->GetXaxis()->GetXmax();
    string hname = string(h1->GetName()) + "_divide";
    if (histName != "") hname = histName;
 
    TH1D* h3 = new TH1D(histName.c_str(), hname.c_str(), nBins, min, max);
    h3->GetXaxis()->SetTitle(h1->GetXaxis()->GetTitle());
    h3->GetYaxis()->SetTitle(titleYaxis.c_str());
    h1->Sumw2();
    h2->Sumw2();
    h3->Sumw2();
    h3->Divide(h1, h2, 1., 1., "B");
    h3->Scale(scale);
    return h3;
  }
 
  TH2D* DivideH2(TH2* h1, TH2* h2, const string& histName, double scale, const string& titleZaxis)
  {
    int nBinsX   = h1->GetNbinsX();
    double minX  = h1->GetXaxis()->GetXmin();
    double maxX  = h1->GetXaxis()->GetXmax();
    int nBinsY   = h1->GetNbinsY();
    double minY  = h1->GetYaxis()->GetXmin();
    double maxY  = h1->GetYaxis()->GetXmax();
    string hname = string(h1->GetName()) + "_divide";
    if (histName != "") hname = histName;
 
    TH2D* h3 = new TH2D(hname.c_str(), hname.c_str(), nBinsX, minX, maxX, nBinsY, minY, maxY);
    h3->GetXaxis()->SetTitle(h1->GetXaxis()->GetTitle());
    h3->GetYaxis()->SetTitle(h1->GetYaxis()->GetTitle());
    h3->GetZaxis()->SetTitle(titleZaxis.c_str());
    h1->Sumw2();
    h2->Sumw2();
    h3->Sumw2();
    h3->Divide(h1, h2, 1., 1., "B");
    h3->Scale(scale);
    return h3;
  }
 
}  // namespace Cbm