AsgForwardElectronSelectorTool.h

Go to the documentation of this file.

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

 
 
#include "AsgTools/AsgTool.h"
#include "EgammaAnalysisInterfaces/IAsgElectronLikelihoodTool.h"
#include "ElectronPhotonSelectorTools/AsgForwardElectronCalibrationTool.h"
#include "PATCore/AcceptInfo.h"
#include "PATCore/AcceptData.h"
#include "xAODEgamma/ElectronFwd.h"
 
#include "lwtnn/LightweightGraph.hh"
 
#include <vector>
#include <string>
#include <memory>
 
 
class EventContext;
 
 
class AsgForwardElectronSelectorTool
  : public asg::AsgTool
  , virtual public IAsgElectronLikelihoodTool
{
  ASG_TOOL_CLASS2(AsgForwardElectronSelectorTool,
                  IAsgElectronLikelihoodTool,
                  IAsgSelectionTool)
 
 
 
public:

  AsgForwardElectronSelectorTool(const std::string& myname);
  virtual ~AsgForwardElectronSelectorTool();
 

  virtual StatusCode initialize() override final;
 

  virtual const asg::AcceptInfo& getAcceptInfo() const override;
 
 
 


  virtual asg::AcceptData accept(const xAOD::IParticle* part) const override;
  virtual asg::AcceptData accept(const EventContext& ctx,
                                 const xAOD::IParticle* part) const override;

  virtual asg::AcceptData accept(const EventContext& ctx,
                                 const xAOD::Electron* eg) const override;
  virtual asg::AcceptData accept(const EventContext& ctx,
                                 const xAOD::Electron* eg,
                                 double mu) const override;
  virtual asg::AcceptData accept(const EventContext& ctx,
                                 const xAOD::Egamma* eg) const override;
  virtual asg::AcceptData accept(const EventContext& ctx,
                                 const xAOD::Egamma* eg,
                                 double mu) const override;

  virtual double calculate(const EventContext& ctx,
                           const xAOD::IParticle* part) const override;
  virtual double calculate(const EventContext& ctx,
                           const xAOD::Electron* eg) const override;
  virtual double calculate(const EventContext& ctx,
                           const xAOD::Electron* eg,
                           double mu) const override;
  virtual double calculate(const EventContext& ctx,
                           const xAOD::Egamma* eg) const override;
  virtual double calculate(const EventContext& ctx,
                           const xAOD::Egamma* eg,
                           double mu) const override;
 
 

  virtual std::vector<float> calculateMultipleOutputs(
      const EventContext& ctx,
      const xAOD::Electron* eg,
      double mu = -99) const override;
 
 
  virtual std::string getOperatingPointName() const override;
 
 
 
 
 
  private:

  int getEtaBin(double absEta) const;

  int getPtBin(double calibPt) const;

  bool getInputs(const xAOD::Electron* eg,
                 double calibPt,
                 int etaBin,
                 std::vector<double>& inputs) const;

  bool calculateWithCalibPt(const EventContext& ctx,
                            const xAOD::Electron* eg,
                            double& score,
                            double& calibPt) const;
 
  std::vector<std::unique_ptr<lwt::LightweightGraph>> m_graphs;

  std::vector<std::string> m_variables;
 

  ToolHandle<AsgForwardElectronCalibrationTool> m_calibTool{
      this, "CalibrationTool", "",
      "Handle to the AsgForwardElectronCalibrationTool"};

  int m_wpIndex{-1};

  Gaudi::Property<std::vector<std::string>> m_modelFiles {this,"ModelFiles",{"","",""} ,"lwtnn JSON files, one per eta bin (in eta order)"};                                         
  Gaudi::Property<std::string> m_workingPoint {this,"WorkingPoint","Loose","Working Point: Loose(90%), Medium (80%), or Tight (70%)"};
  
  asg::AcceptInfo m_acceptInfo;
 
};
 
#endif