AsgForwardElectronCalibrationTool.h

Go to the documentation of this file.

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

 
 
#include "AsgTools/PropertyWrapper.h"
#include "AsgTools/AsgTool.h"
#include "xAODEgamma/ElectronFwd.h"
 
#include "EgammaAnalysisInterfaces/IForwardElectronCalib.h"
 
#include "lwtnn/LightweightGraph.hh"
 
#include <vector>
#include <string>
#include <memory>
 
 
class EventContext;
 
 
class AsgForwardElectronCalibrationTool 
  : public asg::AsgTool
  , virtual public IForwardElectronCalib
{
  ASG_TOOL_CLASS1(AsgForwardElectronCalibrationTool,
          IForwardElectronCalib)
 
 
 
public:

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

  virtual StatusCode initialize() override;
 

  double calibrate(const EventContext& ctx,
                   const xAOD::Electron* eg) const override;
 
 
 
 
 
private:

  int getEtaBin(double absEta) const;

  bool getInputs(const xAOD::Electron* eg,
                 std::vector<float>& inputs) const;

  double softplus(double x) const;

  double unscalePt(double x) const;
 
 

  Gaudi::Property<double> m_pTMin {this,"pTMin", 10000, "Lower bound of pT Min scaling [MeV]"};
  Gaudi::Property<double> m_pTMax {this,"pTMax", 255000, "Lower bound of pT Max scaling [MeV]"};

  Gaudi::Property<std::vector<std::string>> m_modelFiles {this,"ModelFiles",{"","",""} , "lwtnn JSON files, one per eta bin (in eta order)"};
  std::vector<std::unique_ptr<lwt::LightweightGraph>> m_graphs;

  std::vector<std::string> m_variables;
 
};
 
#endif