Tool to apply DNN-based ID to forward electrons for Run 4 using ITk and HGTD variables. More...

#include <AsgForwardElectronSelectorTool.h>

Inheritance diagram for AsgForwardElectronSelectorTool:

Collaboration diagram for AsgForwardElectronSelectorTool:

Public Member Functions
virtual void	print () const
	Print the state of the tool.
ServiceHandle< StoreGateSvc > &	evtStore ()
	The standard `StoreGateSvc` (event store) Returns (kind of) a pointer to the `StoreGateSvc`.
const ServiceHandle< StoreGateSvc > &	detStore () const
	The standard `StoreGateSvc/DetectorStore` Returns (kind of) a pointer to the `StoreGateSvc`.
virtual StatusCode	sysInitialize () override
	Perform system initialization for an algorithm.
virtual StatusCode	sysStart () override
	Handle START transition.
virtual std::vector< Gaudi::DataHandle * >	inputHandles () const override
	Return this algorithm's input handles.
virtual std::vector< Gaudi::DataHandle * >	outputHandles () const override
	Return this algorithm's output handles.
Gaudi::Details::PropertyBase &	declareProperty (Gaudi::Property< T, V, H > &t)
void	updateVHKA (Gaudi::Details::PropertyBase &)
MsgStream &	msg () const
bool	msgLvl (const MSG::Level lvl) const
Additional helper functions, not directly mimicking Athena
template<class T>
const T *	getProperty (const std::string &name) const
	Get one of the tool's properties.
const std::string &	msg_level_name () const __attribute__((deprecated))
	A deprecated function for getting the message level's name.
const std::string &	getName (const void *ptr) const
	Get the name of an object that is / should be in the event store.
SG::sgkey_t	getKey (const void *ptr) const
	Get the (hashed) key of an object that is in the event store.

Protected Member Functions
void	renounceArray (SG::VarHandleKeyArray &handlesArray)
	remove all handles from I/O resolution
std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void >	renounce (T &h)
void	extraDeps_update_handler (Gaudi::Details::PropertyBase &ExtraDeps)
	Add StoreName to extra input/output deps as needed.

Private Types
typedef ServiceHandle< StoreGateSvc >	StoreGateSvc_t

Private Member Functions
virtual ASG_TOOL_CLASS2(AsgForwardElectronSelectorTool, IAsgElectronLikelihoodTool, IAsgSelectionTool) public	~AsgForwardElectronSelectorTool ()
	Standard constructor.
virtual StatusCode	initialize () override final
	Gaudi Service Interface method implementations.
virtual const asg::AcceptInfo &	getAcceptInfo () const override
	Method to get the plain AcceptInfo.
virtual asg::AcceptData	accept (const xAOD::IParticle *part) const override
	IAsgSelectionTool interface.
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
	The main accept method: the actual cuts are applied here.
virtual asg::AcceptData	accept (const EventContext &ctx, const xAOD::Electron *eg, double mu) const override
	accept method with pointer to electron when mu not in EventInfo for online
virtual asg::AcceptData	accept (const EventContext &ctx, const xAOD::Egamma *eg) const override
	accept method with pointer to egamma
virtual asg::AcceptData	accept (const EventContext &ctx, const xAOD::Egamma *eg, double mu) const override
	accept method with pointer to egammma when mu not in EventInfo for online
virtual double	calculate (const EventContext &ctx, const xAOD::IParticle *part) const override
	The main result method: the actual DNN is calculated here.
virtual double	calculate (const EventContext &ctx, const xAOD::Electron *eg) const override
	calculate method: for pointer to electron
virtual double	calculate (const EventContext &ctx, const xAOD::Electron *eg, double mu) const override
	calculate method: for pointer to electron when mu not in EventInfo for online
virtual double	calculate (const EventContext &ctx, const xAOD::Egamma *eg) const override
	calculate method: for pointer to egamma
virtual double	calculate (const EventContext &ctx, const xAOD::Egamma *eg, double mu) const override
	calculate method: for pointer to egamma when mu not in EventInfo for online
virtual std::vector< float >	calculateMultipleOutputs (const EventContext &ctx, const xAOD::Electron *eg, double mu=-99) const override
	The result method for multiple outputs: only one output is used so return vector of that output.
virtual std::string	getOperatingPointName () const override
	Method to get the operating point.
int	getEtaBin (double absEta) const
	Select the eta bin index for \|eta\|, three eta bins: Bin 0: 2.5 < \|eta\| <= 2.7 Bin 1: 2.7 < \|eta\| <= 3.2 Bin 2: 3.2 < \|eta\| <= 4.0 Returns -1 if out of range.
int	getPtBin (double calibPt) const
	Select the pT bin index, Bin 0: 5 GeV <= pT < 15 GeV Bin 1: 15 GeV <= pT < 20 GeV Bin 2: 20 GeV <= pT < 25 GeV Bin 3: 25 GeV <= pT < 30 GeV Bin 4: 30 GeV <= pT < 35 GeV Bin 5: 35 GeV <= pT < 40 GeV Bin 6: 40 GeV <= pT < 45 GeV Bin 7: 45 GeV <= pT < 50 GeV Bin 8: 50 GeV <= pT < 60 GeV Bin 9: 60 GeV <= pT < 80 GeV Bin 10: 80 GeV <= pT < 150 GeV Bin 11: 150 GeV <= pT < 500 GeV Returns -1 if outside valid range.
bool	getInputs (const xAOD::Electron *eg, double calibPt, int etaBin, std::vector< double > &inputs) const
	Get 25 input variables Applies LR decorrelation Returns false on failure.
bool	calculateWithCalibPt (const EventContext &ctx, const xAOD::Electron *eg, double &score, double &calibPt) const
	Runs calibration + LR + DNN in a single pass.
Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &)
	specialization for handling Gaudi::Property<SG::VarHandleKey>

Private Attributes
std::vector< std::unique_ptr< lwt::LightweightGraph > >	m_graphs
std::vector< std::string >	m_variables
	Input variable names.
ToolHandle< AsgForwardElectronCalibrationTool >	m_calibTool
	Handle to the calibration tool.
int	m_wpIndex {-1}
	Working point index: 0 = Loose 1 = Medium 2 = Tight.
Gaudi::Property< std::vector< std::string > >	m_modelFiles {this,"ModelFiles",{"","",""} ,"lwtnn JSON files, one per eta bin (in eta order)"}
	One lwtnn JSON file / DNN per eta bin.
Gaudi::Property< std::string >	m_workingPoint {this,"WorkingPoint","Loose","Working Point: Loose(90%), Medium (80%), or Tight (70%)"}
	Working point: Loose = 90% \| Medium = 80% \| Tight = 70%.
asg::AcceptInfo	m_acceptInfo
	AcceptInfo: defines the cut structure.
StoreGateSvc_t	m_evtStore
	Pointer to StoreGate (event store by default).
StoreGateSvc_t	m_detStore
	Pointer to StoreGate (detector store by default).
std::vector< SG::VarHandleKeyArray * >	m_vhka
bool	m_varHandleArraysDeclared

Detailed Description

Tool to apply DNN-based ID to forward electrons for Run 4 using ITk and HGTD variables.

Internal Note: https://cds.cern.ch/record/2922175

Workflow:

Call AsgForwardElectronCalibrationTool to get the calibrated pT in this forward region
Apply Linear Regression to decorrelate the 7 shower shape moments from pT
Run DNN via lwtnn (MinMax scaler already into JSON as a batch norm layer)
Apply pT- and eta-dependent working point cut on the DNN score

Author: Mathis Dubau (LAPP)

Date: Feb 2026

Based on ElectronDNNCalculator and AsgForwardElectronLikelihoodTool

Definition at line 44 of file AsgForwardElectronSelectorTool.h.

Member Typedef Documentation

◆ StoreGateSvc_t

typedef ServiceHandle<StoreGateSvc> AthCommonDataStore< AthCommonMsg< AlgTool > >::StoreGateSvc_t

privateinherited

Definition at line 388 of file AthCommonDataStore.h.

Member Function Documentation

◆ accept() [1/6]

asg::AcceptData AsgForwardElectronSelectorTool::accept	(	const EventContext &	ctx,
		const xAOD::Egamma *	part ) const

overrideprivatevirtual

accept method with pointer to egamma

Implements IAsgElectronLikelihoodTool.

Definition at line 309 of file AsgForwardElectronSelectorTool.cxx.

{
  return accept(ctx, eg, -99);
}

◆ accept() [2/6]

asg::AcceptData AsgForwardElectronSelectorTool::accept	(	const EventContext &	ctx,
		const xAOD::Egamma *	part,
		double	mu ) const

overrideprivatevirtual

accept method with pointer to egammma when mu not in EventInfo for online

Implements IAsgElectronLikelihoodTool.

Definition at line 316 of file AsgForwardElectronSelectorTool.cxx.

{
  const xAOD::Electron* el = dynamic_cast<const xAOD::Electron*>(eg);
  if (!el) {
    ATH_MSG_ERROR("Input is not an electron.");
    return asg::AcceptData(&m_acceptInfo);
  }
  return accept(ctx, el, mu);
}

◆ accept() [3/6]

asg::AcceptData AsgForwardElectronSelectorTool::accept	(	const EventContext &	ctx,
		const xAOD::Electron *	eg ) const

overrideprivatevirtual

The main accept method: the actual cuts are applied here.

Implements IAsgElectronLikelihoodTool.

Definition at line 279 of file AsgForwardElectronSelectorTool.cxx.

{
  return accept(ctx, eg, -99);
}

◆ accept() [4/6]

asg::AcceptData AsgForwardElectronSelectorTool::accept	(	const EventContext &	ctx,
		const xAOD::Electron *	part,
		double	mu ) const

overrideprivatevirtual

accept method with pointer to electron when mu not in EventInfo for online

Implements IAsgElectronLikelihoodTool.

Definition at line 286 of file AsgForwardElectronSelectorTool.cxx.

{
  asg::AcceptData acceptData(&m_acceptInfo);
 
  double score{-999.}, calibPt{-999.};
  if (!calculateWithCalibPt(ctx, eg, score, calibPt)) return acceptData;
 
  const xAOD::CaloCluster* cluster = eg->caloCluster();
  const int etaBin = getEtaBin(std::abs(cluster->eta()));
  const int ptBin  = getPtBin(calibPt);
  if (etaBin < 0 || ptBin < 0) return acceptData;
 
  acceptData.setCutResult("PassDNNScore",
                          score >= s_cuts[etaBin][ptBin][m_wpIndex]);
  return acceptData;
}

◆ accept() [5/6]

asg::AcceptData AsgForwardElectronSelectorTool::accept	(	const EventContext &	ctx,
		const xAOD::IParticle *	part ) const

overrideprivatevirtual

Implements IAsgElectronLikelihoodTool.

Definition at line 263 of file AsgForwardElectronSelectorTool.cxx.

{
  if (part->type() == xAOD::Type::Electron) {
    const xAOD::Electron* el = static_cast<const xAOD::Electron*>(part);
    return accept(ctx, el);
  }
  ATH_MSG_ERROR("Input is not an electron.");
  return asg::AcceptData(&m_acceptInfo);
}

◆ accept() [6/6]

asg::AcceptData AsgForwardElectronSelectorTool::accept ( const xAOD::IParticle * part ) const

overrideprivatevirtual

IAsgSelectionTool interface.

The main accept method: using the generic interface

Implements IAsgElectronLikelihoodTool.

Definition at line 256 of file AsgForwardElectronSelectorTool.cxx.

{
  // Backwards compatibility
  return accept(Gaudi::Hive::currentContext(), part);
}

◆ calculate() [1/5]

double AsgForwardElectronSelectorTool::calculate	(	const EventContext &	ctx,
		const xAOD::Egamma *	eg ) const

overrideprivatevirtual

calculate method: for pointer to egamma

Implements IAsgElectronLikelihoodTool.

Definition at line 368 of file AsgForwardElectronSelectorTool.cxx.

{
  return calculate(ctx, eg, -99);
}

◆ calculate() [2/5]

double AsgForwardElectronSelectorTool::calculate	(	const EventContext &	ctx,
		const xAOD::Egamma *	eg,
		double	mu ) const

overrideprivatevirtual

calculate method: for pointer to egamma when mu not in EventInfo for online

Implements IAsgElectronLikelihoodTool.

Definition at line 375 of file AsgForwardElectronSelectorTool.cxx.

{
  const xAOD::Electron* el = dynamic_cast<const xAOD::Electron*>(eg);
  if (!el) { ATH_MSG_ERROR("Input is not an electron."); return -999.; }
  return calculate(ctx, el, mu);
}

◆ calculate() [3/5]

double AsgForwardElectronSelectorTool::calculate	(	const EventContext &	ctx,
		const xAOD::Electron *	eg ) const

overrideprivatevirtual

calculate method: for pointer to electron

Implements IAsgElectronLikelihoodTool.

Definition at line 347 of file AsgForwardElectronSelectorTool.cxx.

{
  return calculate(ctx, eg, -99);
}

◆ calculate() [4/5]

double AsgForwardElectronSelectorTool::calculate	(	const EventContext &	ctx,
		const xAOD::Electron *	eg,
		double	mu ) const

overrideprivatevirtual

calculate method: for pointer to electron when mu not in EventInfo for online

Implements IAsgElectronLikelihoodTool.

Definition at line 354 of file AsgForwardElectronSelectorTool.cxx.

{
  double score{-999.}, calibPt{-999.};
  calculateWithCalibPt(ctx, eg, score, calibPt);
 
  return score;
}

◆ calculate() [5/5]

double AsgForwardElectronSelectorTool::calculate	(	const EventContext &	ctx,
		const xAOD::IParticle *	part ) const

overrideprivatevirtual

The main result method: the actual DNN is calculated here.

Implements IAsgElectronLikelihoodTool.

Definition at line 332 of file AsgForwardElectronSelectorTool.cxx.

{
  if (part->type() == xAOD::Type::Electron) {
    const xAOD::Electron* el = static_cast<const xAOD::Electron*>(part);
    return calculate(ctx, el);
  }
  ATH_MSG_ERROR("Input is not an electron.");
  return -999.;
}

◆ calculateMultipleOutputs()

std::vector< float > AsgForwardElectronSelectorTool::calculateMultipleOutputs	(	const EventContext &	ctx,
		const xAOD::Electron *	eg,
		double	mu = -99 ) const

overrideprivatevirtual

The result method for multiple outputs: only one output is used so return vector of that output.

Implements IAsgElectronLikelihoodTool.

Definition at line 443 of file AsgForwardElectronSelectorTool.cxx.

{
  return { static_cast<float>(calculate(ctx, eg, mu)) };
}

◆ calculateWithCalibPt()

bool AsgForwardElectronSelectorTool::calculateWithCalibPt	(	const EventContext &	ctx,
		const xAOD::Electron *	eg,
		double &	score,
		double &	calibPt ) const

private

Runs calibration + LR + DNN in a single pass.

Returns false on error; on success fills score and calibPt.

Definition at line 388 of file AsgForwardElectronSelectorTool.cxx.

{
  score = -999.; calibPt = -999.;
 
  if (!eg) {
    ATH_MSG_ERROR("Failed, no Electron object.");
    return false; 
  }
 
  const xAOD::CaloCluster* cluster = eg->caloCluster();
  if (!cluster) {
    ATH_MSG_WARNING("Failed, no cluster.");
    return false; 
  }
 
  // ITk electron must have a track
  const xAOD::TrackParticle* track = eg->trackParticle();
  if (!track) {
    ATH_MSG_WARNING("Failed, no track.");
    return false;
  }
 
  const double absEta = std::abs(cluster->eta());
  const int etaBin = getEtaBin(absEta);
  if (etaBin < 0) {
    ATH_MSG_WARNING("Electron |eta|=" << absEta
                    << " is outside allowed range.");
    return false;
  }
 
  // Step 1 - Usage of allready calibrated pT
  calibPt=eg->pt();
 
  // Step 2 - extract inputs with LR decorrelation
  std::vector<double> inputs;
  if (!getInputs(eg, calibPt, etaBin, inputs)) return false;
 
  // Step 3 - run DNN
  std::map<std::string, std::map<std::string, double>> inputMap;
  for (size_t i = 0; i < m_variables.size(); ++i)
    inputMap["node_0"][m_variables[i]] = inputs[i];
 
  score = m_graphs[etaBin]->compute(inputMap).begin()->second;
  // Round 6 digits due to lwtnn mismatch
  score = std::round(score * 1e6) / 1e6;
  return true;
}

◆ declareGaudiProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< AlgTool > >::declareGaudiProperty	(	Gaudi::Property< T, V, H > &	hndl,
		const SG::VarHandleKeyType &	)

inlineprivateinherited

specialization for handling Gaudi::Property<SG::VarHandleKey>

Definition at line 156 of file AthCommonDataStore.h.

  {
    return *AthCommonDataStore<PBASE>::declareProperty(hndl.name(),
                                                       hndl.value(), 
                                                       hndl.documentation());
 
  }

◆ declareProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty ( Gaudi::Property< T, V, H > & t )

inlineinherited

Definition at line 145 of file AthCommonDataStore.h.

                                                                     {
    typedef typename SG::HandleClassifier<T>::type htype;
    return AthCommonDataStore<PBASE>::declareGaudiProperty(t, htype());
  }

◆ detStore()

const ServiceHandle< StoreGateSvc > & AthCommonDataStore< AthCommonMsg< AlgTool > >::detStore ( ) const

inlineinherited

The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 95 of file AthCommonDataStore.h.

95{ return m_detStore; }

AthCommonDataStore< AthCommonMsg< AlgTool > >::m_detStore

StoreGateSvc_t m_detStore

Definition AthCommonDataStore.h:393

◆ evtStore()

ServiceHandle< StoreGateSvc > & AthCommonDataStore< AthCommonMsg< AlgTool > >::evtStore ( )

inlineinherited

The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 85 of file AthCommonDataStore.h.

85{ return m_evtStore; }

AthCommonDataStore< AthCommonMsg< AlgTool > >::m_evtStore

StoreGateSvc_t m_evtStore

Definition AthCommonDataStore.h:390

◆ extraDeps_update_handler()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::extraDeps_update_handler ( Gaudi::Details::PropertyBase & ExtraDeps )

protectedinherited

Add StoreName to extra input/output deps as needed.

use the logic of the VarHandleKey to parse the DataObjID keys supplied via the ExtraInputs and ExtraOuputs Properties to add the StoreName if it's not explicitly given

◆ getAcceptInfo()

const asg::AcceptInfo & AsgForwardElectronSelectorTool::getAcceptInfo ( ) const

overrideprivatevirtual

Method to get the plain AcceptInfo.

This is needed so that one can already get the AcceptInfo and query what cuts are defined before the first object is passed to the tool.

Implements IAsgSelectionTool.

Definition at line 247 of file AsgForwardElectronSelectorTool.cxx.

{
  return m_acceptInfo;
}

◆ getEtaBin()

int AsgForwardElectronSelectorTool::getEtaBin ( double absEta ) const

private

Select the eta bin index for |eta|, three eta bins: Bin 0: 2.5 < |eta| <= 2.7 Bin 1: 2.7 < |eta| <= 3.2 Bin 2: 3.2 < |eta| <= 4.0 Returns -1 if out of range.

Definition at line 463 of file AsgForwardElectronSelectorTool.cxx.

{
  // Convention: x1 < |eta| <= x2
  if (absEta > 2.5 && absEta <= 2.7) return 0;
  if (absEta > 2.7 && absEta <= 3.2) return 1;
  if (absEta > 3.2 && absEta <= 4.0) return 2;
  return -1;
}

◆ getInputs()

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

private

Get 25 input variables Applies LR decorrelation Returns false on failure.

Definition at line 488 of file AsgForwardElectronSelectorTool.cxx.

{
  inputs.clear();
  inputs.reserve(25);
 
  const xAOD::CaloCluster*   cluster = eg->caloCluster();
  const xAOD::TrackParticle* track   = eg->trackParticle();
 
  if (!cluster) { ATH_MSG_ERROR("No CaloCluster.");   return false; }
  if (!track)   { ATH_MSG_ERROR("No TrackParticle."); return false; }
 
  // x1, x2 = calo eta and phi
  inputs.push_back(static_cast<float>(cluster->eta()));
  inputs.push_back(static_cast<float>(cluster->phi()));
 
  // x3, x4 = track eta and phi
  inputs.push_back(static_cast<float>(track->eta()));
  inputs.push_back(static_cast<float>(track->phi()));
 
  // x5 = HGTD time
  static const SG::AuxElement::Accessor<uint8_t> accValid("hasValidTime");
  if (accValid.isAvailable(*track))
    {
      if (accValid(*track))
    {
      inputs.push_back(static_cast<float>(track->time()));
    }
      else
    {
      ATH_MSG_DEBUG("No valid time for the track while doing track->time()" );
      inputs.push_back(-99);
    }                                                                                                      
    }
  else
    {
      ATH_MSG_ERROR("No available time for the track" );
    }
  
  // x6, x7 = ITk hit counts
  inputs.push_back(static_cast<float>(eg->trackParticleSummaryIntValue(xAOD::numberOfPixelHits)));
  inputs.push_back(static_cast<float>(eg->trackParticleSummaryIntValue(xAOD::numberOfSCTHits)));
 
  // x8 to x14 = 7 calorimeter shower-shape moments
  auto getMoment = [&](xAOD::CaloCluster::MomentType type,
                       const char* name) -> float {
    double val{0.};
    if (!cluster->retrieveMoment(type, val))
      ATH_MSG_WARNING("Could not retrieve calo moment: " << name);
    return val;
  };
 
  // Retrieve raw moments
  const double rawMoments[7] = {
    getMoment(xAOD::CaloCluster::ENG_FRAC_MAX,    "ENG_FRAC_MAX"),
    getMoment(xAOD::CaloCluster::LONGITUDINAL,    "LONGITUDINAL"),
    getMoment(xAOD::CaloCluster::SECOND_LAMBDA,   "SECOND_LAMBDA"),
    getMoment(xAOD::CaloCluster::LATERAL,         "LATERAL"),
    getMoment(xAOD::CaloCluster::SECOND_R,        "SECOND_R"),
    getMoment(xAOD::CaloCluster::CENTER_LAMBDA,   "CENTER_LAMBDA"),
    getMoment(xAOD::CaloCluster::SECOND_ENG_DENS, "SECOND_ENG_DENS")
  };
 
  // Apply LR decorrelation: moment_decorr = moment_raw - (coeff * calibPt + intercept)
  for (int i = 0; i < 7; ++i)
    inputs.push_back(rawMoments[i]
                    - (s_lrCoeff[etaBin][i] * calibPt + s_lrIntercept[etaBin][i]));
 
 
  // x15 to x18 = track-calo matching
  auto getMatch = [&](xAOD::EgammaParameters::TrackCaloMatchType type,
                      const char* name) -> float {
    float val{0.f};
    if (!eg->trackCaloMatchValue(val, type))
      ATH_MSG_WARNING("Could not retrieve track-calo match: " << name);
    return val;
  };
 
  inputs.push_back(getMatch(xAOD::EgammaParameters::deltaEta2,
                            "delta_eta2"));
  inputs.push_back(getMatch(xAOD::EgammaParameters::deltaPhi2,
                            "delta_phi2"));
  inputs.push_back(getMatch(xAOD::EgammaParameters::deltaPhiRescaled2,
                            "delta_phi_rescaled2"));
  inputs.push_back(getMatch(xAOD::EgammaParameters::deltaPhiFromLastMeasurement,
                            "delta_phi_last"));
 
  // x19 to x25 = Derived calorimeter energy fractions
  // Prevent division by zero
  const double caloE   = cluster->e();
  const double inv_E   = (caloE != 0.) ? 1. / caloE : 0.;
 
  using CS = CaloSampling::CaloSample;
 
  // EM fractions: 
  //   f_i^EM = energyBE(i) / caloE  for i in {1,2,3}
  inputs.push_back(static_cast<float>(cluster->energyBE(1) * inv_E));
  inputs.push_back(static_cast<float>(cluster->energyBE(2) * inv_E));
  inputs.push_back(static_cast<float>(cluster->energyBE(3) * inv_E));
 
  // HAD fractions:
  //   f_0^HAD = HEC0 / caloE
  //   f_i^HAD = (HEC_i + FCAL_(i-1)) / caloE  for i in {1,2,3}
  inputs.push_back(static_cast<float>(
    cluster->eSample(static_cast<CS>(CaloSampling::HEC0)) * inv_E));
  inputs.push_back(static_cast<float>(
    (cluster->eSample(static_cast<CS>(CaloSampling::HEC1)) +
     cluster->eSample(static_cast<CS>(CaloSampling::FCAL0))) * inv_E));
  inputs.push_back(static_cast<float>(
    (cluster->eSample(static_cast<CS>(CaloSampling::HEC2)) +
     cluster->eSample(static_cast<CS>(CaloSampling::FCAL1))) * inv_E));
  inputs.push_back(static_cast<float>(
    (cluster->eSample(static_cast<CS>(CaloSampling::HEC3)) +
     cluster->eSample(static_cast<CS>(CaloSampling::FCAL2))) * inv_E));
 
  return true;
}

◆ getKey()

SG::sgkey_t asg::AsgTool::getKey ( const void * ptr ) const

inherited

Get the (hashed) key of an object that is in the event store.

This is a bit of a special one. StoreGateSvc and xAOD::Event both provide ways for getting the SG::sgkey_t key for an object that is in the store, based on a bare pointer. But they provide different interfaces for doing so.

In order to allow tools to efficiently perform this operation, they can use this helper function.

See also: asg::AsgTool::getName

Parameters

ptr	The bare pointer to the object that the event store should know about

Returns: The hashed key of the object in the store. If not found, an invalid (zero) key.

Definition at line 119 of file AsgTool.cxx.

                                                    {
 
#ifdef XAOD_STANDALONE
      // In case we use @c xAOD::Event, we have a direct function call
      // for this.
      return evtStore()->event()->getKey( ptr );
#else
      const SG::DataProxy* proxy = evtStore()->proxy( ptr );
      return ( proxy == nullptr ? 0 : proxy->sgkey() );
#endif // XAOD_STANDALONE
   }

◆ getName()

const std::string & asg::AsgTool::getName ( const void * ptr ) const

inherited

Get the name of an object that is / should be in the event store.

This is a bit of a special one. StoreGateSvc and xAOD::Event both provide ways for getting the std::string name for an object that is in the store, based on a bare pointer. But they provide different interfaces for doing so.

In order to allow tools to efficiently perform this operation, they can use this helper function.

See also: asg::AsgTool::getKey

Parameters

ptr	The bare pointer to the object that the event store should know about

Returns: The string name of the object in the store. If not found, an empty string.

Definition at line 106 of file AsgTool.cxx.

                                                            {
 
#ifdef XAOD_STANDALONE
      // In case we use @c xAOD::Event, we have a direct function call
      // for this.
      return evtStore()->event()->getName( ptr );
#else
      const SG::DataProxy* proxy = evtStore()->proxy( ptr );
      static const std::string dummy = "";
      return ( proxy == nullptr ? dummy : proxy->name() );
#endif // XAOD_STANDALONE
   }

◆ getOperatingPointName()

std::string AsgForwardElectronSelectorTool::getOperatingPointName ( ) const

overrideprivatevirtual

Method to get the operating point.

Implements IAsgElectronLikelihoodTool.

Definition at line 455 of file AsgForwardElectronSelectorTool.cxx.

{
  return m_workingPoint;
}

◆ getProperty()

template<class T>

const T * asg::AsgTool::getProperty ( const std::string & name ) const

inherited

Get one of the tool's properties.

◆ getPtBin()

int AsgForwardElectronSelectorTool::getPtBin ( double calibPt ) const

private

Select the pT bin index, Bin 0: 5 GeV <= pT < 15 GeV Bin 1: 15 GeV <= pT < 20 GeV Bin 2: 20 GeV <= pT < 25 GeV Bin 3: 25 GeV <= pT < 30 GeV Bin 4: 30 GeV <= pT < 35 GeV Bin 5: 35 GeV <= pT < 40 GeV Bin 6: 40 GeV <= pT < 45 GeV Bin 7: 45 GeV <= pT < 50 GeV Bin 8: 50 GeV <= pT < 60 GeV Bin 9: 60 GeV <= pT < 80 GeV Bin 10: 80 GeV <= pT < 150 GeV Bin 11: 150 GeV <= pT < 500 GeV Returns -1 if outside valid range.

Definition at line 475 of file AsgForwardElectronSelectorTool.cxx.

{
  constexpr int nBins = 12;
  for (int i = 0; i < nBins; ++i) {
    if (calibPt >= s_ptEdges[i] && calibPt < s_ptEdges[i + 1])
      return i;
  }
  return -1;
}

◆ initialize()

StatusCode AsgForwardElectronSelectorTool::initialize ( void )

finaloverrideprivatevirtual

Gaudi Service Interface method implementations.

Reimplemented from asg::AsgTool.

Definition at line 158 of file AsgForwardElectronSelectorTool.cxx.

{
  // Resolve working point index
  if      (m_workingPoint == "Loose")  m_wpIndex = 0;
  else if (m_workingPoint == "Medium") m_wpIndex = 1;
  else if (m_workingPoint == "Tight")  m_wpIndex = 2;
  else {
    ATH_MSG_ERROR("Unknown Working Point '" << m_workingPoint
                  << "'. Choose Loose, Medium, or Tight.");
    return StatusCode::FAILURE;
  }
 
  // Retrieve the calibration tool
  ATH_CHECK(m_calibTool.retrieve());
 
  // Sanity checks
  if (m_modelFiles.size() != 3) {
    ATH_MSG_ERROR("Exactly 3 model files expected (one per eta bin): "
                  "[2.5,2.7], [2.7,3.2], [3.2,4.0]). Only got "
                  << m_modelFiles.size());
    return StatusCode::FAILURE;
  }
 
  // Fixed variable names - must match the lwtnn JSON input
  m_variables = {
    // Calo eta and Phi
    "x1_calo_eta",          
    "x2_calo_phi",
    // ITk eta and Phi
    "x3_track_eta", 
    "x4_track_phi",
    // HGTD time
    "x5_time",
    // ITk hit counts
    "x6_pixels",            
    "x7_strips",
    // Shower shape moments 
    "x8_ENG_FRAC_MAX", 
    "x9_LONGITUDINAL",
    "x10_SECOND_LAMBDA",    
    "x11_LATERAL",
    "x12_SECOND_R",         
    "x13_CENTER_LAMBDA",
    "x14_SECOND_ENG_DENS",
    // Track-cluster matching variables
    "x15_delta_eta2",       
    "x16_delta_phi2",
    "x17_delta_phi_rescaled2", 
    "x18_delta_phi_last",
    // Calo energy fractions in each layer
    "x19_calo_frac_EM_1",
    "x20_calo_frac_EM_2",  
    "x21_calo_frac_EM_3",
    "x22_calo_frac_HAD_0",  
    "x23_calo_frac_HAD_1",
    "x24_calo_frac_HAD_2",  
    "x25_calo_frac_HAD_3"
  };
  // More informations on Table 3 in Internal Note: https://cds.cern.ch/record/2922175
 
  // Load one JSON per eta bin
  m_graphs.reserve(3);
  for (const auto& model : m_modelFiles) {
    const std::string path = PathResolverFindCalibFile(model);
    if (path.empty()) {
      ATH_MSG_ERROR("Could not locate: " << model);
      return StatusCode::FAILURE;
    }
    std::ifstream dnn_json(path);
    auto parsed = lwt::parse_json_graph(dnn_json);
    m_graphs.emplace_back(
      std::make_unique<lwt::LightweightGraph>(parsed));
    ATH_MSG_INFO("Loaded ID model for bin "
                 << m_graphs.size() - 1 << ": " << path);
  }
 
  // Register the single cut in AcceptInfo
  m_acceptInfo.addCut("PassDNNScore",
                      "Electron passes the DNN score cut for the configured WP");
 
  ATH_MSG_INFO("AsgForwardElectronSelectorTool initialised. "
               << "WorkingPoint=" << m_workingPoint);
  return StatusCode::SUCCESS;
}

◆ inputHandles()

virtual std::vector< Gaudi::DataHandle * > AthCommonDataStore< AthCommonMsg< AlgTool > >::inputHandles ( ) const

overridevirtualinherited

Return this algorithm's input handles.

We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.

◆ msg()

MsgStream & AthCommonMsg< AlgTool >::msg ( ) const

inlineinherited

Definition at line 24 of file AthCommonMsg.h.

                                {
    return this->msgStream();
  }

◆ msg_level_name()

const std::string & asg::AsgTool::msg_level_name ( ) const

inherited

A deprecated function for getting the message level's name.

Instead of using this, weirdly named function, user code should get the string name of the current minimum message level (in case they really need it...), with:

MSG::name( msg().level() )

This function's name doesn't follow the ATLAS coding rules, and as such will be removed in the not too distant future.

Returns: The string name of the current minimum message level that's printed

Definition at line 101 of file AsgTool.cxx.

                                                  {
 
      return MSG::name( msg().level() );
   }

◆ msgLvl()

bool AthCommonMsg< AlgTool >::msgLvl ( const MSG::Level lvl ) const

inlineinherited

Definition at line 30 of file AthCommonMsg.h.

                                               {
    return this->msgLevel(lvl);
  }

◆ outputHandles()

virtual std::vector< Gaudi::DataHandle * > AthCommonDataStore< AthCommonMsg< AlgTool > >::outputHandles ( ) const

overridevirtualinherited

Return this algorithm's output handles.

We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.

◆ print()

void asg::AsgTool::print ( ) const

virtualinherited

Print the state of the tool.

Implements asg::IAsgTool.

Reimplemented in AsgHelloTool, HI::HIPileupTool, JetBottomUpSoftDrop, JetConstituentsRetriever, JetDumper, JetFinder, JetFromPseudojet, JetModifiedMassDrop, JetPileupLabelingTool, JetPruner, JetPseudojetRetriever, JetReclusterer, JetReclusteringTool, JetRecTool, JetRecursiveSoftDrop, JetSoftDrop, JetSplitter, JetSubStructureMomentToolsBase, JetToolRunner, JetTrimmer, JetTruthLabelingTool, and KtDeltaRTool.

Definition at line 131 of file AsgTool.cxx.

                             {
 
      ATH_MSG_INFO( "AsgTool " << name() << " @ " << this );
      return;
   }

◆ renounce()

std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void > AthCommonDataStore< AthCommonMsg< AlgTool > >::renounce ( T & h )

inlineprotectedinherited

Definition at line 380 of file AthCommonDataStore.h.

  {
    h.renounce();
    PBASE::renounce (h);
  }

◆ renounceArray()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::renounceArray ( SG::VarHandleKeyArray & handlesArray )

inlineprotectedinherited

remove all handles from I/O resolution

Definition at line 364 of file AthCommonDataStore.h.

                                                          {
    handlesArray.renounce();
  }

◆ sysInitialize()

virtual StatusCode AthCommonDataStore< AthCommonMsg< AlgTool > >::sysInitialize ( )

overridevirtualinherited

Perform system initialization for an algorithm.

We override this to declare all the elements of handle key arrays at the end of initialization. See comments on updateVHKA.

Reimplemented in asg::AsgMetadataTool, AthCheckedComponent< AthAlgTool >, AthCheckedComponent<::AthAlgTool >, and DerivationFramework::CfAthAlgTool.

◆ sysStart()

virtual StatusCode AthCommonDataStore< AthCommonMsg< AlgTool > >::sysStart ( )

overridevirtualinherited

Handle START transition.

We override this in order to make sure that conditions handle keys can cache a pointer to the conditions container.

◆ updateVHKA()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::updateVHKA ( Gaudi::Details::PropertyBase & )

inlineinherited

Definition at line 308 of file AthCommonDataStore.h.

                                               {
    // debug() << "updateVHKA for property " << p.name() << " " << p.toString() 
    //         << "  size: " << m_vhka.size() << endmsg;
    for (auto &a : m_vhka) {
      std::vector<SG::VarHandleKey*> keys = a->keys();
      for (auto k : keys) {
        k->setOwner(this);
      }
    }
  }

◆ ~AsgForwardElectronSelectorTool()

AsgForwardElectronSelectorTool::~AsgForwardElectronSelectorTool ( )

privatevirtualdefault

Standard constructor.

Standard destructor

Member Data Documentation

◆ m_acceptInfo

asg::AcceptInfo AsgForwardElectronSelectorTool::m_acceptInfo

private

AcceptInfo: defines the cut structure.

Definition at line 186 of file AsgForwardElectronSelectorTool.h.

◆ m_calibTool

ToolHandle<AsgForwardElectronCalibrationTool> AsgForwardElectronSelectorTool::m_calibTool

private

Initial value:

{
      this, "CalibrationTool", "",
      "Handle to the AsgForwardElectronCalibrationTool"}

Handle to the calibration tool.

Definition at line 171 of file AsgForwardElectronSelectorTool.h.

                                                           {
      this, "CalibrationTool", "",
      "Handle to the AsgForwardElectronCalibrationTool"};

◆ m_detStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< AlgTool > >::m_detStore

privateinherited

Pointer to StoreGate (detector store by default).

Definition at line 393 of file AthCommonDataStore.h.

◆ m_evtStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< AlgTool > >::m_evtStore

privateinherited

Pointer to StoreGate (event store by default).

Definition at line 390 of file AthCommonDataStore.h.

◆ m_graphs

std::vector<std::unique_ptr<lwt::LightweightGraph> > AsgForwardElectronSelectorTool::m_graphs

private

Definition at line 164 of file AsgForwardElectronSelectorTool.h.

◆ m_modelFiles

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

private

One lwtnn JSON file / DNN per eta bin.

Definition at line 182 of file AsgForwardElectronSelectorTool.h.

182{this,"ModelFiles",{"","",""} ,"lwtnn JSON files, one per eta bin (in eta order)"};

◆ m_varHandleArraysDeclared

bool AthCommonDataStore< AthCommonMsg< AlgTool > >::m_varHandleArraysDeclared

privateinherited

Definition at line 399 of file AthCommonDataStore.h.

◆ m_variables

std::vector<std::string> AsgForwardElectronSelectorTool::m_variables

private

Input variable names.

Definition at line 167 of file AsgForwardElectronSelectorTool.h.

◆ m_vhka

std::vector<SG::VarHandleKeyArray*> AthCommonDataStore< AthCommonMsg< AlgTool > >::m_vhka

privateinherited

Definition at line 398 of file AthCommonDataStore.h.

◆ m_workingPoint

Gaudi::Property<std::string> AsgForwardElectronSelectorTool::m_workingPoint {this,"WorkingPoint","Loose","Working Point: Loose(90%), Medium (80%), or Tight (70%)"}

private

Working point: Loose = 90% | Medium = 80% | Tight = 70%.

Definition at line 183 of file AsgForwardElectronSelectorTool.h.

183{this,"WorkingPoint","Loose","Working Point: Loose(90%), Medium (80%), or Tight (70%)"};

◆ m_wpIndex

int AsgForwardElectronSelectorTool::m_wpIndex {-1}

private

Working point index: 0 = Loose 1 = Medium 2 = Tight.

Definition at line 179 of file AsgForwardElectronSelectorTool.h.

179{-1};

The documentation for this class was generated from the following files:

Public Member Functions

Protected Member Functions

Private Types

Private Member Functions

Private Attributes

Detailed Description

Member Typedef Documentation

◆ StoreGateSvc_t

Member Function Documentation

◆ accept() [1/6]

◆ accept() [2/6]

◆ accept() [3/6]

◆ accept() [4/6]

◆ accept() [5/6]

◆ accept() [6/6]

◆ calculate() [1/5]

◆ calculate() [2/5]

◆ calculate() [3/5]

◆ calculate() [4/5]

◆ calculate() [5/5]

◆ calculateMultipleOutputs()

◆ calculateWithCalibPt()

◆ declareGaudiProperty()

◆ declareProperty()

◆ detStore()

◆ evtStore()

◆ extraDeps_update_handler()

◆ getAcceptInfo()

◆ getEtaBin()

◆ getInputs()

◆ getKey()

◆ getName()

◆ getOperatingPointName()

◆ getProperty()

◆ getPtBin()

◆ initialize()

◆ inputHandles()

◆ msg()

◆ msg_level_name()

◆ msgLvl()

◆ outputHandles()

◆ print()

◆ renounce()

◆ renounceArray()

◆ sysInitialize()

◆ sysStart()

◆ updateVHKA()

◆ ~AsgForwardElectronSelectorTool()

Member Data Documentation

◆ m_acceptInfo

◆ m_calibTool

◆ m_detStore

◆ m_evtStore

◆ m_graphs

◆ m_modelFiles

◆ m_varHandleArraysDeclared

◆ m_variables

◆ m_vhka

◆ m_workingPoint

◆ m_wpIndex