SimTestHisto.h

Go to the documentation of this file.

/* -*- C++ -*- */
 
/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef G4ATLASTESTS_SIMTESTHISTO_H
#define G4ATLASTESTS_SIMTESTHISTO_H
 
#include "AthenaKernel/errorcheck.h"
#include "GaudiKernel/ITHistSvc.h"
#include "GaudiKernel/ServiceHandle.h"
 
#include "TH1.h"
#include "TH2.h"
class TProfile;
 
class SimTestHisto
{
 
public:
 
  SimTestHisto() = default;
  ~SimTestHisto() = default;
 
protected:
  std::string m_path{"/truth/"};
  ServiceHandle<ITHistSvc> m_histSvc{"THistSvc", "SimTestHisto"};
};
 
// note: var should be of type "TH1*" even if it is filled with a TProfile*
#define _TPROFILE(var,name,nbin,xmin,xmax)             \
  if (!m_histSvc->exists(m_path+name)) {               \
    var = new TProfile(name,name,nbin,xmin,xmax);          \
    CHECK(m_histSvc->regHist(m_path+name,var));        \
  } else {                             \
    CHECK(m_histSvc->getHist(m_path+name, var));           \
  }
 
#define _TH1D(var,name,nbin,xmin,xmax)                 \
  if (!m_histSvc->exists(m_path+name)) {               \
    var = new TH1D(name,name,nbin,xmin,xmax);              \
    var->StatOverflows();                      \
    CHECK(m_histSvc->regHist(m_path+name,var));            \
  } else {                             \
    CHECK(m_histSvc->getHist(m_path+name,var));        \
  }
 
#define _TH1D_NOCHECK(var,name,nbin,xmin,xmax)                 \
  if (!m_histSvc->exists(m_path+name)) {               \
    var = new TH1D(name,name,nbin,xmin,xmax);              \
    var->StatOverflows();                      \
    if(m_histSvc->regHist(m_path+name,var).isFailure())        \
  std::cout<<"Cannot register histogram "<<name<<std::endl;    \
  } else {                             \
    if(m_histSvc->getHist(m_path+name,var).isFailure())       \
      std::cout<<"Cannot get histogram "<<name<<std::endl;     \
  }
 
#define _TH1D_WEIGHTED(var,name,nbin,xmin,xmax) \
  _TH1D(var,name,nbin,xmin,xmax);       \
  var->Sumw2();
 
#define _TH2D_NOCHECK(var,name,nbinx,xmin,xmax,nbiny,ymin,ymax)        \
  if (!m_histSvc->exists(m_path+name)) {               \
    var = new TH2D(name,name,nbinx,xmin,xmax,nbiny,ymin,ymax); \
    if(m_histSvc->regHist(m_path+name,var).isFailure())        \
      std::cout<<"Cannot register histogram "<<name<<std::endl;\
  } else {                             \
    if(m_histSvc->getHist(m_path+name,var).isFailure())       \
      std::cout<<"Cannot get histogram "<<name<<std::endl;     \
  }
 
#define _TH2D(var,name,nbinx,xmin,xmax,nbiny,ymin,ymax)        \
  if (!m_histSvc->exists(m_path+name)) {               \
    var = new TH2D(name,name,nbinx,xmin,xmax,nbiny,ymin,ymax); \
    CHECK(m_histSvc->regHist(m_path+name,var));                \
  } else {                             \
    CHECK(m_histSvc->getHist(m_path+name,var));        \
  }
 
#define _TH2D_WEIGHTED(var,name,nbinx,xmin,xmax,nbiny,ymin,ymax)    \
  _TH2D(var,name,nbinx,xmin,xmax,nbiny,ymin,ymax);          \
  var->Sumw2();
 
#define _SET_TITLE(var,title,xaxis,yaxis)               \
  var->SetXTitle(xaxis);                        \
  var->SetYTitle(yaxis);                        \
  var->SetTitle((std::string(var->GetName())+" : "+title).c_str());
 
//#define _SET_LOGX(var)
//TAxis *axis = var->GetXaxis();
//int bins = axis->GetNbin//s();
 
// Axis_t from = axis->GetXmin();
// Axis_t to = axis->GetXmax();
// Axis_t width = (to - from) / bins;
// Axis_t *new_bins = new Axis_t[bins + 1];
 
// for (int i = 0; i <= bins; i++) {
//   new_bins[i] = TMath::Power(10, from + i * width);
//
// }
// axis->Set(bins, new_bins);
// delete new_bins;
 
#endif //G4ATLASTESTS_SIMTESTHISTO_H