DiTauDiscriminantTool.h

Go to the documentation of this file.

// Dear emacs, this is -*- c++ -*-
 
/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef  DITAURECTOOLS_DITAUDISCRIMINANTTOOL_H
#define  DITAURECTOOLS_DITAUDISCRIMINANTTOOL_H

 
// Framework include(s):
#include "AsgTools/AsgTool.h"
#include "AsgTools/PropertyWrapper.h"
 
// Local include(s):
#include "DiTauRecTools/IDiTauToolBase.h"
#include "xAODTau/DiTauJet.h"
#include "tauRecTools/BDTHelper.h"
 
#include <string>
#include <map>
#include <memory>
 
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) const override;
  
private:
 
  Gaudi::Property<std::string> m_sWeightsFile{this, "WeightsFile", "tauRecTools/R22_preprod/DiTau_JetBDT_winter2024.weights.root"};
  Gaudi::Property<std::string> m_sBDTScoreName{this, "BDTScoreName", "JetBDT"};
 
  StatusCode parseWeightsFile();
 
  std::map<TString, float> setIDVariables(const xAOD::DiTauJet& xDiTau) const;
 
  std::unique_ptr<tauRecTools::BDTHelper> m_mvaBDT = nullptr;
 
  std::vector<std::string> m_vVarNames;
 
}; // class DiTauDiscriminantTool
 
}
#endif // DITAURECTOOLS_DITAUDISCRIMINANTTOOL_H