TauDecayModeNNClassifier.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TAURECTOOLS_TAUDECAYMODENNCLASSIFIER_H
#define TAURECTOOLS_TAUDECAYMODENNCLASSIFIER_H
 
// base class include(s)
#include "tauRecTools/TauRecToolBase.h"
 
// xAOD include(s)
#include "xAODTau/TauJet.h"
 
// lwtnn include(s)
#include "lwtnn/LightweightGraph.hh"
#include "lwtnn/parse_json.hh"
#include "lwtnn/Exceptions.hh"
 
// standard library include(s)
#include <memory>
#include <vector>
#include <set>
#include <map>
 
class TauDecayModeNNClassifier : public TauRecToolBase
{
public:
  ASG_TOOL_CLASS2(TauDecayModeNNClassifier, TauRecToolBase, ITauToolBase)
 
  explicit TauDecayModeNNClassifier(const std::string &name = "TauDecayModeNNClassifier");
  virtual ~TauDecayModeNNClassifier();
 
  virtual StatusCode initialize() override;
  virtual StatusCode execute(xAOD::TauJet &xTau) const override;
 
private:
  std::string m_outputName;      
  std::string m_probPrefix;      
  std::string m_weightFile;      
  std::size_t m_maxTauTracks;  
  std::size_t m_maxNeutralPFOs;  
  std::size_t m_maxShotPFOs;     
  std::size_t m_maxConvTracks;   
  float m_neutralPFOPtCut;       
  bool m_ensureTrackConsistency; 
  bool m_decorateProb;           
 
  virtual StatusCode getInputs(const xAOD::TauJet &xTau,
                               std::map<std::string, std::map<std::string, std::vector<double>>> &inputSeqMap) const;
  std::unique_ptr<const lwt::LightweightGraph> m_lwtGraph; 
};
 
namespace tauRecTools
{
  class TauDecayModeNNVariable
  {
  public:
    TauDecayModeNNVariable() = delete;
    static const std::size_t nClasses = 5;
    static const std::set<std::string> sCommonP4Vars;
    static const std::set<std::string> sTrackIPVars;
    static const std::set<std::string> sNeutralPFOVars;
    static const std::array<std::string, nClasses> sModeNames;
    static float deltaPhi(const TLorentzVector &p4, const TLorentzVector &p4_tau);
    static float deltaEta(const TLorentzVector &p4, const TLorentzVector &p4_tau);
    static float deltaPhiECal(const TLorentzVector &p4, const std::pair<float, bool> &tau_phiTrkECal);
    static float deltaEtaECal(const TLorentzVector &p4, const std::pair<float, bool> &tau_etaTrkECal);
    template <typename T>
    static T pfoAttr(const xAOD::PFO *pfo, const xAOD::PFODetails::PFOAttributes &attr);
    static float ptSubRatio(const xAOD::PFO *pfo);
    static float energyFracEM2(const xAOD::PFO *pfo, float energy_em2);
  };
 
  class TauDecayModeNNHelper
  {
  public:
    TauDecayModeNNHelper() = delete;
    static float Log10Robust(const float val, const float min_val = 0.);
    template <typename T>
    static void sortAndKeep(std::vector<T> &vec, const std::size_t n_obj);
    template <typename T>
    static void initMapKeys(std::map<std::string, T> &empty_map, const std::set<std::string> &keys);
  };
} // namespace tauRecTools
 
#endif // TAURECTOOLS_TAUDECAYMODENNCLASSIFIER_H