PhotonBDTCalculator.h

Go to the documentation of this file.

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

 
#ifndef ELECTRONPHOTONSELECTORTOOLS_PHOTONBDTCALCULATOR_H
#define ELECTRONPHOTONSELECTORTOOLS_PHOTONBDTCALCULATOR_H
 
#include "AsgTools/AsgTool.h"
#include "AsgTools/ToolHandle.h"
#include "EgammaAnalysisInterfaces/IPhotonObservableTool.h"
#include "xAODEgamma/Photon.h"
#include "PhotonSingleBDTCalculator.h"
#include <string>
 
namespace PhotonIDBDT {
 
class PhotonBDTCalculator : public asg::AsgTool, virtual public IPhotonObservableTool
{
ASG_TOOL_CLASS2(PhotonBDTCalculator, IPhotonObservableTool, asg::IAsgTool)
 
public:
  PhotonBDTCalculator(const std::string& name);
  virtual ~PhotonBDTCalculator() override;
 
  virtual StatusCode initialize() override;

  using IPhotonObservableTool::evaluate;
  virtual double evaluate(const xAOD::Photon* photon) const override;

  StatusCode decorate(const xAOD::Photon& ph) const;

  StatusCode getScore(const xAOD::Photon& ph, float& score) const;
 
private:
  // Helpers to compute input variables for converted and unconverted photons
  StatusCode fillVariablesConv(const xAOD::Photon& ph, std::vector<float>& vars) const;
  StatusCode fillVariablesUnconv(const xAOD::Photon& ph, std::vector<float>& vars) const;
 
  // Reserve size for input features vector for converted and unconverted photons
  Gaudi::Property<unsigned> m_reserveVarsConv   {this, "ReserveVarsConv", 16, "Reserve size for input vector (converted photons)"};
  Gaudi::Property<unsigned> m_reserveVarsUnconv {this, "ReserveVarsUnconv", 16, "Reserve size for input vector (unconverted photons)"};
 
  // Sub-tools for the evaluation of the BDT score for converted and unconverted photons. 
  ToolHandle<PhotonSingleBDTCalculator> m_toolConv {this, "ToolConv", "", "BDT calculator for converted photons"};
  ToolHandle<PhotonSingleBDTCalculator> m_toolUnconv {this, "ToolUnconv", "", "BDT calculator for unconverted photons"};
 
  // Properties
  Gaudi::Property<std::string> m_decorationName {this, "DecorationName", "PhotonBDTScore", "Aux decoration name for BDT score"};
  Gaudi::Property<bool> m_excludeTRT {this, "ExcludeTRT", true, "Exclude TRT conversions (for Run 3 conversion definition)"};
  Gaudi::Property<bool> m_forceRecompute {this, "ForceRecompute", false, "If true, recompute even if decoration exists"};
 
  bool isConverted(const xAOD::Photon& ph) const;
};
 
} // namespace PhotonIDBDT
 
#endif