#include <HITowerWeightTool.h>

Inheritance diagram for HITowerWeightTool:

Collaboration diagram for HITowerWeightTool:

Public Member Functions
	HITowerWeightTool (const std::string &type, const std::string &name, const IInterface *parent)

virtual	~HITowerWeightTool ()=default

virtual StatusCode	initialize () override

virtual float	getEtaPhiResponse (float eta, float phi, int runIndex) const override

virtual float	getEtaPhiOffset (float eta, float phi, int runIndex) const override

virtual float	getWeight (float eta, float phi, int sampling) const override

virtual float	getWeightEta (float eta, float phi, int sampling) const override

virtual float	getWeightPhi (float eta, float phi, int sampling) const override

virtual float	getWeightMag (float eta, float phi, int sampling) const override

virtual int	getRunIndex (const EventContext &ctx) const override

Private Attributes
Gaudi::Property< bool >	m_applycorrection {this, "ApplyCorrection", true , "If false, unit weigts are applied"}

Gaudi::Property< std::vector< int > >	m_defaultRunNumbers {this, "DefaultRunNumbers", {226000,287931,338037,367384,440101} , "List of run numbers that will be used if the event run number is not found in the InputFile"}

Gaudi::Property< std::string >	m_inputFile {this, "InputFile", "cluster.geo.HIJING_2018.root","File containing cluster geometric moments."}

Gaudi::Property< std::string >	m_configDir {this, "ConfigDir", "HIJetCorrection/","Directory containing configuration file."}

TH3F *	m_h3W

TH3F *	m_h3Eta

TH3F *	m_h3Phi

TH3F *	m_h3Mag

TH3F *	m_h3EtaPhiResponse

TH3F *	m_h3EtaPhiOffset

std::map< unsigned int, int >	m_runMap

Detailed Description

Definition at line 22 of file HITowerWeightTool.h.

Constructor & Destructor Documentation

◆ HITowerWeightTool()

HITowerWeightTool::HITowerWeightTool	(	const std::string &	type,
		const std::string &	name,
		const IInterface *	parent
	)

Definition at line 8 of file HITowerWeightTool.cxx.

                                                                                                            :
     base_class(type, name, parent),
     m_h3W(nullptr),
     m_h3Eta(nullptr),
     m_h3Phi(nullptr),
     m_h3Mag(nullptr),
     m_h3EtaPhiResponse(nullptr),
     m_h3EtaPhiOffset(nullptr)
 {
 }

◆ ~HITowerWeightTool()

virtual HITowerWeightTool::~HITowerWeightTool ( )

virtualdefault

Member Function Documentation

◆ getEtaPhiOffset()

float HITowerWeightTool::getEtaPhiOffset	(	float	eta,
		float	phi,
		int	runIndex
	)		const

overridevirtual

Definition at line 48 of file HITowerWeightTool.cxx.

 {
   if(runIndex<=0) return 0;
  
   int eb=std::as_const(m_h3EtaPhiOffset)->GetXaxis()->FindFixBin(eta);
   int pb=std::as_const(m_h3EtaPhiOffset)->GetYaxis()->FindFixBin(phi);
   return m_h3EtaPhiOffset->GetBinContent(eb,pb,runIndex)*std::cosh(eta);
 }

◆ getEtaPhiResponse()

float HITowerWeightTool::getEtaPhiResponse	(	float	eta,
		float	phi,
		int	runIndex
	)		const

overridevirtual

Definition at line 38 of file HITowerWeightTool.cxx.

 {
   if(runIndex<=0)  return 1;
  
   int eb=std::as_const(m_h3EtaPhiResponse)->GetXaxis()->FindFixBin(eta);
   int pb=std::as_const(m_h3EtaPhiResponse)->GetYaxis()->FindFixBin(phi);
   return m_h3EtaPhiResponse->GetBinContent(eb,pb,runIndex);
 }

◆ getRunIndex()

int HITowerWeightTool::getRunIndex ( const EventContext & ctx ) const

overridevirtual

Definition at line 58 of file HITowerWeightTool.cxx.

 {
   if (!m_applycorrection){
     ATH_MSG_DEBUG("Using unit weights and doing no eta-phi correction.");
     return 0;
   }
  
   unsigned int run_number=ctx.eventID().run_number();
  
   auto itr=m_runMap.find(run_number);
   if(itr==m_runMap.end())
   {
     //trying generic run numbers <=> no run dependence
     for(auto run_number : m_defaultRunNumbers)
     {
       auto itrg=m_runMap.find(run_number);
       if(itrg!=m_runMap.end())
       {
         ATH_MSG_DEBUG("Using run " << run_number << " generic calibration for eta-phi correction.");
         return itrg->second;
       }
     }
  
     if(m_defaultRunNumbers.empty())
     {
       ATH_MSG_WARNING("No calibration for " << run_number << " is avaliable and no generic run numbers were set. Doing no eta-phi correction.");
     }
     else
     {
       std::string str_defaultRunNumbers="";
       for(auto run_number : m_defaultRunNumbers)
       {
         str_defaultRunNumbers+=std::to_string(run_number)+", ";
       }
       str_defaultRunNumbers.resize(str_defaultRunNumbers.length()-2);
  
       ATH_MSG_WARNING("No calibration for " << run_number << " is avaliable; no generic calibration for runs "<<str_defaultRunNumbers<<". Doing no eta-phi correction.");
     }
  
     return 0;
   }
   else 
   {
     return itr->second;
   }
 }

◆ getWeight()

float HITowerWeightTool::getWeight	(	float	eta,
		float	phi,
		int	sampling
	)		const

overridevirtual

Definition at line 20 of file HITowerWeightTool.cxx.

 {
   return m_h3W->GetBinContent(m_h3W->FindFixBin(eta,phi,sample));
 }

◆ getWeightEta()

float HITowerWeightTool::getWeightEta	(	float	eta,
		float	phi,
		int	sampling
	)		const

overridevirtual

Definition at line 24 of file HITowerWeightTool.cxx.

 {
   return m_h3Eta->GetBinContent(m_h3Eta->FindFixBin(eta,phi,sample));
 }

◆ getWeightMag()

float HITowerWeightTool::getWeightMag	(	float	eta,
		float	phi,
		int	sampling
	)		const

overridevirtual

Definition at line 32 of file HITowerWeightTool.cxx.

 {
   return m_h3Mag->GetBinContent(m_h3Mag->FindFixBin(eta,phi,sample));
 }

◆ getWeightPhi()

float HITowerWeightTool::getWeightPhi	(	float	eta,
		float	phi,
		int	sampling
	)		const

overridevirtual

Definition at line 28 of file HITowerWeightTool.cxx.

 {
   return m_h3Phi->GetBinContent(m_h3Phi->FindFixBin(eta,phi,sample));
 }

◆ initialize()

StatusCode HITowerWeightTool::initialize ( )

overridevirtual

Definition at line 106 of file HITowerWeightTool.cxx.

 {
   std::string local_path = static_cast<std::string>(m_configDir)+m_inputFile;
   std::string full_path=PathResolverFindCalibFile(local_path);
   ATH_MSG_INFO("Reading input file "<< m_inputFile << " from " << full_path);
   TFile* f=TFile::Open(full_path.c_str());
   if(f==nullptr)
   {
     ATH_MSG_FATAL("Cannot read config file " << full_path );
     return StatusCode::FAILURE;
   }
  
   m_h3W=(TH3F*)f->GetObjectChecked("h3_w","TH3F");
   if(m_h3W==nullptr)
   {
     ATH_MSG_FATAL("Cannot find TH3F m_h3W in config file " << full_path );
     return StatusCode::FAILURE;
   }
  
   m_h3Eta=(TH3F*)f->GetObjectChecked("h3_eta","TH3F");
   if(m_h3Eta==nullptr)
   {
     ATH_MSG_FATAL("Cannot find TH3F m_h3Eta in config file " << full_path );
     return StatusCode::FAILURE;
   }
  
   m_h3Phi=(TH3F*)f->GetObjectChecked("h3_phi","TH3F");
   if(m_h3Phi==nullptr)
   {
     ATH_MSG_FATAL("Cannot find TH3F m_h3Phi in config file " << full_path );
     return StatusCode::FAILURE;
   }
  
   m_h3Mag=(TH3F*)f->GetObjectChecked("h3_R","TH3F");
   if(m_h3Mag==nullptr)
   {
     ATH_MSG_FATAL("Cannot find TH3F m_h3Mag in config file " << full_path );
     return StatusCode::FAILURE;
   }
  
   m_h3W->SetDirectory(0);
   m_h3Eta->SetDirectory(0);
   m_h3Phi->SetDirectory(0);
   m_h3Mag->SetDirectory(0);
  
   if(m_applycorrection)
   {
     ATH_MSG_DEBUG("ApplyCorrection is set to true, now loading h3_eta_phi_response, h3_eta_phi_offset, and h1_run_index.");
     
     m_h3EtaPhiResponse=(TH3F*)f->GetObjectChecked("h3_eta_phi_response","TH3F");
     if(m_h3EtaPhiResponse==nullptr)
     {
       ATH_MSG_FATAL("Cannot find TH3F h3_eta_phi_response in config file " << full_path );
       return StatusCode::FAILURE;
     }
     m_h3EtaPhiResponse->SetDirectory(0);
     m_h3EtaPhiOffset=(TH3F*)f->GetObjectChecked("h3_eta_phi_offset","TH3F");
     if(m_h3EtaPhiOffset==nullptr)
     {
       ATH_MSG_FATAL("Cannot find TH3F h3_eta_phi_offset in config file " << full_path );
       return StatusCode::FAILURE;
     }
     m_h3EtaPhiOffset->SetDirectory(0);
  
     TH1I* h1_run_index=(TH1I*)f->GetObjectChecked("h1_run_index","TH1I");
     if(h1_run_index==nullptr)
     {
       ATH_MSG_FATAL("Cannot find TH3F h1_run_index in config file " << full_path );
       return StatusCode::FAILURE;
     }
     for(int xbin=1; xbin<=h1_run_index->GetNbinsX(); xbin++) {
       m_runMap.emplace_hint(m_runMap.end(),std::make_pair(h1_run_index->GetBinContent(xbin),xbin));
     }
   }
   else
   {
     ATH_MSG_DEBUG("ApplyCorrection is set to false, not loading h3_eta_phi_response, h3_eta_phi_offset, and h1_run_index.");
   }
  
   f->Close();
   return StatusCode::SUCCESS;
 }

Member Data Documentation

◆ m_applycorrection

Gaudi::Property<bool> HITowerWeightTool::m_applycorrection {this, "ApplyCorrection", true , "If false, unit weigts are applied"}

private

Definition at line 38 of file HITowerWeightTool.h.

◆ m_configDir

Gaudi::Property<std::string> HITowerWeightTool::m_configDir {this, "ConfigDir", "HIJetCorrection/","Directory containing configuration file."}

private

Definition at line 42 of file HITowerWeightTool.h.

◆ m_defaultRunNumbers

Gaudi::Property<std::vector<int> > HITowerWeightTool::m_defaultRunNumbers {this, "DefaultRunNumbers", {226000,287931,338037,367384,440101} , "List of run numbers that will be used if the event run number is not found in the InputFile"}

private

Definition at line 40 of file HITowerWeightTool.h.

◆ m_h3Eta

TH3F* HITowerWeightTool::m_h3Eta

private

Definition at line 45 of file HITowerWeightTool.h.

◆ m_h3EtaPhiOffset

TH3F* HITowerWeightTool::m_h3EtaPhiOffset

private

Definition at line 49 of file HITowerWeightTool.h.

◆ m_h3EtaPhiResponse

TH3F* HITowerWeightTool::m_h3EtaPhiResponse

private

Definition at line 48 of file HITowerWeightTool.h.

◆ m_h3Mag

TH3F* HITowerWeightTool::m_h3Mag

private

Definition at line 47 of file HITowerWeightTool.h.

◆ m_h3Phi

TH3F* HITowerWeightTool::m_h3Phi

private

Definition at line 46 of file HITowerWeightTool.h.

◆ m_h3W

TH3F* HITowerWeightTool::m_h3W

private

Definition at line 44 of file HITowerWeightTool.h.

◆ m_inputFile

Gaudi::Property<std::string> HITowerWeightTool::m_inputFile {this, "InputFile", "cluster.geo.HIJING_2018.root","File containing cluster geometric moments."}

private

Definition at line 41 of file HITowerWeightTool.h.

◆ m_runMap

std::map<unsigned int, int> HITowerWeightTool::m_runMap

private

Definition at line 50 of file HITowerWeightTool.h.

The documentation for this class was generated from the following files:

Public Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ HITowerWeightTool()

◆ ~HITowerWeightTool()

Member Function Documentation

◆ getEtaPhiOffset()

◆ getEtaPhiResponse()

◆ getRunIndex()

◆ getWeight()

◆ getWeightEta()

◆ getWeightMag()

◆ getWeightPhi()

◆ initialize()

Member Data Documentation

◆ m_applycorrection

◆ m_configDir

◆ m_defaultRunNumbers

◆ m_h3Eta

◆ m_h3EtaPhiOffset

◆ m_h3EtaPhiResponse

◆ m_h3Mag

◆ m_h3Phi

◆ m_h3W

◆ m_inputFile

◆ m_runMap