PhotonSingleBDTCalculator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/

 
#ifndef ELECTRONPHOTONSELECTORTOOLS_PHOTONSINGLEBDTCALCULATOR_H
#define ELECTRONPHOTONSELECTORTOOLS_PHOTONSINGLEBDTCALCULATOR_H
 
#include "AsgTools/PropertyWrapper.h"
#include "AsgTools/AsgTool.h"
#include <memory>
#include <string>
#include <vector>
 
namespace MVAUtils {
  class BDT;
}
 
class TFile;
class TTree;
 
namespace PhotonIDBDT {
 
class PhotonSingleBDTCalculator : public asg::AsgTool {
  ASG_TOOL_CLASS(PhotonSingleBDTCalculator, asg::IAsgTool)
 
public:
  PhotonSingleBDTCalculator(const std::string& name);
  virtual ~PhotonSingleBDTCalculator() override;
 
  virtual StatusCode initialize() override;

  StatusCode computeScore(const std::vector<float>& vars, float& score) const;
 
private:
  StatusCode loadBDT();
 
  Gaudi::Property<std::string> m_modelFile {
    this, "ModelFile", "", "ROOT file containing the BDT TTree"
  };
 
  Gaudi::Property<std::string> m_bdtTreeName {
    this, "BDTTreeName", "lgbm", "Name of the TTree containing the BDT (e.g. lgbm)"
  };
 
  // Private members
  std::unique_ptr<TFile> m_file;
  TTree* m_tree{nullptr};
  std::unique_ptr<MVAUtils::BDT> m_bdt;
 
};
 
} // namespace PhotonIDBDT
 
#endif