ElectronLRTMergingAlg.h

Go to the documentation of this file.

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

//   ElectronLRTMergingAlg
//   Author: Jonathan Long, jonathan.long@cern.ch
//   Algorithm invoking std+lrt electron merger, based on the
//   MuonLRTMergingAlg, author: Max Goblirsch-Kolb
 
#ifndef ASG_ANALYSIS_ALGORITHMS__ELECTRON_LRT_MERGING_ALGORITHM__H
#define ASG_ANALYSIS_ALGORITHMS__ELECTRON_LRT_MERGING_ALGORITHM__H
 
#include <AnaAlgorithm/AnaReentrantAlgorithm.h>
#include <xAODEgamma/ElectronContainer.h>
#include <EgammaAnalysisInterfaces/IElectronLRTOverlapRemovalTool.h>
#include <AsgTools/ToolHandle.h>
 
#include "AthContainers/ConstDataVector.h"
 
#include <AsgTools/PropertyWrapper.h>
#include <AsgDataHandles/WriteHandleKey.h>
#include <AsgDataHandles/ReadHandleKey.h>
#include <AsgDataHandles/WriteHandle.h>
#include <AsgDataHandles/ReadHandle.h>
#include <AsgDataHandles/WriteDecorHandleKey.h>
 
namespace CP
{
    class ElectronLRTMergingAlg final : public EL::AnaReentrantAlgorithm
    {
    public:
        ElectronLRTMergingAlg(const std::string &name,
                              ISvcLocator *pSvcLocator);
 
    public:
        StatusCode initialize() override;
 
    public:
        StatusCode execute(const EventContext &ctx) const override;
 
    private:

 
        Gaudi::Property<bool> m_createViewCollection{this, "CreateViewCollection", true, "Create a view to avoid deep copy"}; 
 
        Gaudi::Property<bool> m_doRemoval{this, "doRemoval", true, "Remove overlapping electrons from output collection."}; 
 
        Gaudi::Property<float> m_ORThreshold{this, "ORThreshold", 0.001, "Delta R threshold for matching in overlap removal."}; 

        Gaudi::Property<int> m_ORstrategy{this, "overlapStrategy", CP::IElectronLRTOverlapRemovalTool::defaultStrategy, "Overlap removal strategy to use (0 = default)"};
 
        Gaudi::Property<bool> m_isDAOD{this, "isDAOD", true, "Switch for running on AOD (false) or DAOD (true)"}; 
 
        SG::ReadHandleKey<xAOD::ElectronContainer> m_promptElectronLocation{this, "PromptElectronLocation", "Electrons", "Prompt electrons to merge"}; 
 
        SG::ReadHandleKey<xAOD::ElectronContainer> m_lrtElectronLocation{this, "LRTElectronLocation", "LRTElectrons", "LRT electrons to merge"}; 
 
        SG::WriteHandleKey<xAOD::ElectronContainer> m_outElectronLocation{this, "OutputCollectionName", "StdWithLRTElectrons", "Name for combined output collection"}; 
 
        ToolHandle<CP::IElectronLRTOverlapRemovalTool> m_overlapRemovalTool{this, "overlapRemovalTool", "", "Tool used to check overlaps between standard and LRT electrons"}; 



        void mergeElectron(const xAOD::ElectronContainer &electronCol,
                           xAOD::ElectronContainer *outputCol,
                           const std::set<const xAOD::Electron *> &LRTElectronsToRemove) const;

        void mergeElectron(const xAOD::ElectronContainer &electronCol,
                           ConstDataVector<xAOD::ElectronContainer> *outputCol,
                           const std::set<const xAOD::Electron *> &LRTElectronsToRemove) const;
 
      SG::WriteDecorHandleKey<xAOD::ElectronContainer> m_promptIsLRTKey
      { this, "PromptIsLRTKey", m_promptElectronLocation, "isLRT" };
      SG::WriteDecorHandleKey<xAOD::ElectronContainer> m_lrtIsLRTKey
      { this, "LRTIsLRTKey", m_lrtElectronLocation, "isLRT" };
    };
}
#endif