SoftElectronDecoratorAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ELECTRON_DECORATOR_ALG_HH
#define ELECTRON_DECORATOR_ALG_HH
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "AthContainers/AuxElement.h"
#include "StoreGate/WriteDecorHandleKey.h"
#include "StoreGate/ReadDecorHandleKey.h"
 
#include "xAODEgamma/ElectronContainer.h"
#include "xAODTracking/VertexContainer.h"
 
namespace FlavorTagDiscriminants {
 
  class SoftElectronDecoratorAlg: public AthReentrantAlgorithm {
  public:
    SoftElectronDecoratorAlg(const std::string& name,
                          ISvcLocator* pSvcLocator );
 
    virtual StatusCode initialize() override;
    virtual StatusCode execute(const EventContext& ) const override;
 
  private:
 
    // Input Containers
    SG::ReadHandleKey< xAOD::ElectronContainer > m_ElectronContainerKey {
      this, "electronContainer", "Electrons",
        "Key for the input electron collection"};
    SG::ReadHandleKey< xAOD::VertexContainer > m_VertexContainerKey {
      this, "vertexContainer", "PrimaryVertices",
        "Key for the input electron collection"};
 
    // Decorators for electrons
 
    SG::WriteDecorHandleKey< xAOD::ElectronContainer > m_dec_electron_et {
      this, "ftag_et", m_ElectronContainerKey, "ftag_et",
        "Transverse energy of the electron"};
 
    SG::WriteDecorHandleKey< xAOD::ElectronContainer > m_dec_electron_deltaPOverP {
      this, "ftag_deltaPOverP", m_ElectronContainerKey, "ftag_deltaPOverP",
        "Momentum lost by the electron track between the perigee and the last measurement point divided by the momentum at the perigee"};
 
    SG::WriteDecorHandleKey< xAOD::ElectronContainer > m_dec_electron_isoPtOverPt {
      this, "ftag_ptVarCone30OverPt", m_ElectronContainerKey, "ftag_ptVarCone30OverPt",
        "Ratio of isolated pt to pt"};
 
    SG::WriteDecorHandleKey< xAOD::ElectronContainer > m_dec_electron_energyOverP {
      this, "ftag_energyOverP", m_ElectronContainerKey, "ftag_energyOverP",
        "Ratio of cluster energy energy to track momentum"};
 
    SG::WriteDecorHandleKey< xAOD::ElectronContainer > m_dec_electron_z0 {
      this, "ftag_z0AlongBeamspot", m_ElectronContainerKey, "ftag_z0AlongBeamspot",
        "Electron z0 along beamspot"};
 
    SG::WriteDecorHandleKey< xAOD::ElectronContainer > m_dec_electron_z0_significance {
      this, "ftag_z0AlongBeamspotSignificance", m_ElectronContainerKey, "ftag_z0AlongBeamspotSignificance",
        "Electron z0 significance along beamspot"};
 
    SG::ReadDecorHandleKey< xAOD::ElectronContainer > m_pt_varcone30_key {
      this, "ptVarCone30Key", m_ElectronContainerKey,
      "ptvarcone30_Nonprompt_All_MaxWeightTTVALooseCone_pt1000",
        "Isolation pT variable cone 30"};
 
    const xAOD::Vertex* primary(const xAOD::VertexContainer& vertices) const;
  };
}
 
#endif