TauResolutionTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TAUANALYSISTOOLS_TAURESOLUTIONTOOL_H
#define TAUANALYSISTOOLS_TAURESOLUTIONTOOL_H
 
// Framework include(s):
#include "AsgTools/AsgTool.h"
#include "AsgTools/ToolHandle.h"
#include "AsgTools/AsgMetadataTool.h"
#include "AsgTools/PropertyWrapper.h"
#include "ColumnarMode/ColumnarMode.h"
 
#include "TauAnalysisTools/ITauResolutionTool.h"
 
#include "TGraph.h"
 
namespace ADD_CMODE(TauAnalysisTools)
{
 
class TauResolutionTool
  : public asg::AsgMetadataTool
  , public virtual TauAnalysisTools::ITauResolutionTool
{
  ASG_TOOL_CLASS( TauResolutionTool, TauAnalysisTools::ITauResolutionTool )
 
public:
  TauResolutionTool( const std::string& sName );
 
  ~TauResolutionTool();

  virtual StatusCode initialize();

  double getMvaEnergyResolution(const xAOD::TauJet& tau) const;

  bool getTESCompatibility(const xAOD::TauJet& tau) const; 
 
private:
 
  Gaudi::Property<std::string> m_calFileName{this, "WeightFileName", "tauRecTools/R22_preprod/CombinedTES_R22_Round2.5_v2.root"};

  enum Binning {DecayModeBinning = 5, EtaBinning = 5};

  const std::array<std::string, DecayModeBinning> m_decayModeNames = {"1p0n","1p1n","1pXn","3p0n","3pXn"}; 
  
  const std::array<std::string, EtaBinning> m_etaBinNames = {"0", "1", "2", "3", "4"}; 

  int getEtaIndex(float eta) const;

  xAOD::TauJetParameters::DecayMode getDecayMode(const xAOD::TauJet& tau) const;

  int getDecayModeIndex(xAOD::TauJetParameters::DecayMode decayMode) const;

  std::array<std::array<std::unique_ptr<TGraph>, EtaBinning>, DecayModeBinning> m_mvaRes; 
  
  std::array<std::array<double, EtaBinning>, DecayModeBinning> m_mvaResMaxEt; 

  bool isValid(const xAOD::TauJet& tau) const; 

  double getCaloResolution(double et, int decayModeIndex, int etaIndex) const;

  std::array<std::array<std::unique_ptr<TGraph>, EtaBinning>, DecayModeBinning> m_caloRes; 

  std::array<std::array<double, EtaBinning>, DecayModeBinning> m_caloResMaxEt; 
 
}; // class TauResolutionTool
 
} // namespace TauAnalysisTools
 
#endif // TAUANALYSISTOOLS_TAURESOLUTIONTOOL_H