TauCombinedTES.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TAURECTOOLS_TAUCOMBINEDTES_H
#define TAURECTOOLS_TAUCOMBINEDTES_H
 
#include "tauRecTools/TauRecToolBase.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:
    
  bool m_addCalibrationResultVariables;
  
  std::string m_calFileName;
 
  bool m_useMvaResolution;
  
  struct Variables
  {
    double pt_constituent{0.0};
    double pt_tauRecCalibrated{0.0};
    double pt_weighted{0.0};
    double weight{-1111.0};
    //double sigma_combined{-1111.};
    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 getCombinedSigma(double caloSigma, double panTauSigma, double correlation) 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