TauPi0ScoreCalculator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TAURECTOOLS_TAUPI0SCORECALCULATOR_H
#define TAURECTOOLS_TAUPI0SCORECALCULATOR_H
 
#include "tauRecTools/TauRecToolBase.h"
#include "tauRecTools/BDTHelper.h"
 
#include "xAODPFlow/PFO.h"
#include "AsgTools/PropertyWrapper.h"
 
#include <string>

 
class TauPi0ScoreCalculator : public TauRecToolBase {
 
public:
 
  ASG_TOOL_CLASS2(TauPi0ScoreCalculator, TauRecToolBase, ITauToolBase)
  
  TauPi0ScoreCalculator(const std::string& name);
  virtual ~TauPi0ScoreCalculator() = default;
 
  virtual StatusCode initialize() override;
  virtual StatusCode executePi0nPFO(xAOD::TauJet& pTau, xAOD::PFOContainer& pNeutralPFOContainer) const override;
 
private:
  
  Gaudi::Property<std::string> m_weightfile{this, "BDTWeightFile", ""}; 

  float calculateScore(const xAOD::PFO* neutralPFO) const;
 
  std::unique_ptr<tauRecTools::BDTHelper> m_mvaBDT = nullptr;
};
 
#endif  // TAURECTOOLS_TAUPI0SCORECALCULATOR_H