TauCombinedTES.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TAURECTOOLS_TAUCOMBINEDTES_H
#define TAURECTOOLS_TAUCOMBINEDTES_H
 
#include "tauRecTools/TauRecToolBase.h"
 
#include "AsgTools/PropertyWrapper.h"
 
#include "xAODTau/TauJet.h"
 
#include "TH1F.h"
#include "TF1.h"
#include "TGraph.h"
 
#include <array>
#include <string>
 
class TauCombinedTES : public TauRecToolBase {
public:
  ASG_TOOL_CLASS2( TauCombinedTES, TauRecToolBase, ITauToolBase )
 
  TauCombinedTES(const std::string& name="TauCombinedTES");
    
  virtual StatusCode initialize() override;
      
  virtual StatusCode execute(xAOD::TauJet& xTau) const override;

  bool getTESCompatibility(const xAOD::TauJet& tau) const;
  
  double getMvaEnergyResolution(const xAOD::TauJet& tau) const;
 
private:
    
  Gaudi::Property<bool> m_addCalibrationResultVariables{this, "addCalibrationResultVariables", false};
  Gaudi::Property<std::string> m_calFileName{this, "WeightFileName", ""};
  Gaudi::Property<bool> m_useMvaResolution{this, "useMvaResolution", false};  
 
  struct Variables
  {
    double pt_constituent{0.0};
    double pt_tauRecCalibrated{0.0};
    double pt_weighted{0.0};
    double weight{-1111.0};
    double sigma_compatibility{-1111.};
    double sigma_tauRec{-1111.0};
    double sigma_constituent{-1111.0};
    double corrcoeff{-1111.0};
  };

  TLorentzVector getCombinedP4(const xAOD::TauJet& tau, Variables& variables) const;

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

  int getEtaIndex(float eta) const;

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

  double getCorrelation(int decayModeIndex, int etaIndex) const;
  
  double getCaloResolution(double et, int decayModeIndex, int etaIndex) const;
  
  double getPanTauResolution(double et, int decayModeIndex, int etaIndex) const;
   
  double getCaloCalEt(double et, int decayModeIndex, int etaIndex) const;
 
  double getPanTauCalEt(double panTauEt, int decayModeIndex, int etaIndex) const;
 
  double getWeight(double caloSigma, double panTauSigma, double correlatioon) const;

  double getCompatibilitySigma(double caloSigma, double panTauSigma, double correlation) const;

  double getCombinedEt(double caloEt, double et_substructure,
               xAOD::TauJetParameters::DecayMode decayMode, float eta,
                       Variables& variables) const;
   
  double getNsigmaCompatibility(double caloEt, int decayModeIndex) const;
  
  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"}; 
  
  std::array<std::array<std::unique_ptr<TGraph>, EtaBinning>, DecayModeBinning> m_caloRelBias; 

  std::array<std::array<double, EtaBinning>, DecayModeBinning> m_caloRelBiasMaxEt; 

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

  std::array<std::array<double, EtaBinning>, DecayModeBinning> m_caloResMaxEt; 

  std::array<std::array<std::unique_ptr<TGraph>, EtaBinning>, DecayModeBinning> m_panTauRelBias; 
  
  std::array<std::array<double, EtaBinning>, DecayModeBinning> m_panTauRelBiasMaxEt; 

  std::array<std::array<std::unique_ptr<TGraph>, EtaBinning>, DecayModeBinning> m_panTauRes; 
  
  std::array<std::array<double, EtaBinning>, DecayModeBinning> m_panTauResMaxEt; 

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

  std::array<std::unique_ptr<TF1>, DecayModeBinning> m_nSigmaCompatibility; 
};
 
#endif // TAURECTOOLS_TAUCOMBINEDTES_H