DiTauDiscriminantTool.h

Go to the documentation of this file.

// Dear emacs, this is -*- c++ -*-
 
/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef  DITAURECTOOLS_DITAUDISCRIMINANTTOOL_H
#define  DITAURECTOOLS_DITAUDISCRIMINANTTOOL_H
 
// Framework include(s):
#include "AsgTools/AsgTool.h"
 
// Core include(s):
 
// EDM include(s):
#include "xAODTau/DiTauJetContainer.h"
 
// Local include(s):
#include "DiTauRecTools/IDiTauToolBase.h"
 
// MVAUtils includes
#include "MVAUtils/BDT.h"
 
#include <string>
#include <map>
 
 
namespace DiTauRecTools
{
 
 
class DiTauDiscriminantTool
  : public DiTauRecTools::IDiTauToolBase
  , public asg::AsgTool
{
  ASG_TOOL_CLASS( DiTauDiscriminantTool,
                  DiTauRecTools::IDiTauToolBase )
 
public:
 
  DiTauDiscriminantTool( const std::string& name );
 
  virtual ~DiTauDiscriminantTool();
 
  // initialize the tool
  virtual StatusCode initialize() override;
 
  // get ID score depricated
  double getJetBDTScore(const xAOD::DiTauJet& xDiTau);
 
  // calculate and decorate BDTJetScore
  virtual StatusCode execute(const xAOD::DiTauJet& xDiTau) override;
 
  // get decay mode
  virtual std::string getDecayMode() override;
  
private:
 
  StatusCode parseWeightsFile();
 
  void setIDVariables(const xAOD::DiTauJet& xDiTau);
 
  // const xAOD::DiTauJetContainer* m_xDiTauContainer;
 
  // steering variables
  std::string m_sWeightsFile;
  std::string m_sBDTScoreName;
  
  //MVAUtils::BDT* m_bdt; //!
  std::unique_ptr<MVAUtils::BDT> m_bdt;
 
  std::map<TString, float*> m_mIDVariables; 
  std::map<TString, float*> m_mIDSpectators; 
 
  inline float& setVar(const TString& var) { return *(m_mIDVariables[var]); } 
 
  std::vector<std::string> m_vVarNames;
private:
  enum DecayMode{ HadHad, HadMu, HadEl, Default };
  std::string m_sDecayMode;
  DecayMode m_eDecayMode;
}; // class DiTauDiscriminantTool
 
}
#endif // DITAURECTOOLS_DITAUDISCRIMINANTTOOL_H