ElectronLRTOverlapRemovalTool.h

Go to the documentation of this file.

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

//   ElectronLRTOverlapRemovalTool.h
//   Author: Jonathan Long, jonathan.long AT cern.ch
//
//   Implements overlap removal between standard and LRT electrons.
 
#ifndef LRTELECTRONANALYSISTOOLS_ELECTRONLRTOVERLAPREMOVALTOOL_H
#define LRTELECTRONANALYSISTOOLS_ELECTRONLRTOVERLAPREMOVALTOOL_H
 
#include "EgammaAnalysisInterfaces/IElectronLRTOverlapRemovalTool.h"
#include "EgammaAnalysisInterfaces/IAsgElectronLikelihoodTool.h"
 
#include <string>
#include <map>
 
#include "xAODTracking/TrackParticle.h"
#include "xAODEgamma/ElectronContainer.h"
#include "xAODEgamma/Electron.h"
 
#include <AsgTools/ToolHandle.h>
#include <AsgTools/AsgTool.h>
#include <AsgTools/PropertyWrapper.h>
 
namespace CP
{

    class ElectronLRTOverlapRemovalTool : virtual public CP::IElectronLRTOverlapRemovalTool, public asg::AsgTool
    {
 
    public:

 
        ElectronLRTOverlapRemovalTool(const std::string &name);
        virtual ~ElectronLRTOverlapRemovalTool() = default;
        ASG_TOOL_CLASS(ElectronLRTOverlapRemovalTool, CP::IElectronLRTOverlapRemovalTool)
        virtual StatusCode initialize();

        virtual void checkOverlap(const xAOD::ElectronContainer &promptCollection,
                                  const xAOD::ElectronContainer &lrtCollection,
                                  std::set<const xAOD::Electron *> &ElectronsToRemove) const;
 
    private:
        Gaudi::Property<int> m_strategy{this, "overlapStrategy", CP::IElectronLRTOverlapRemovalTool::defaultStrategy, "Overlap removal strategy to use (0 = default)"}; 
        Gaudi::Property<float> m_ORThreshold{this, "ORThreshold", 0.001, "Delta R threshold for matching in overlap removal."};                                         
        Gaudi::Property<bool> m_isDAOD{this, "isDAOD", true, "Switch for running on AOD (false) or DAOD (true)"}; 
        Gaudi::Property<std::string> m_IDWorkingPoint{this, "IDWorkingPoint", "DFCommonElectronsLHVeryLooseNoPix", "ID working point for checking if ID is passed"}; 
 
        ToolHandle<IAsgElectronLikelihoodTool> m_electronLLHToolVeryLooseNoPix{this, "ElectronLLHToolVeryLooseNoPix", "", "Electron LLH tool to use for the overlap removal"}; 
        ToolHandle<IAsgElectronLikelihoodTool> m_electronLLHToolLooseNoPix{this, "ElectronLLHToolLooseNoPix", "", "Electron LLH tool to use for the overlap removal"}; 
        ToolHandle<IAsgElectronLikelihoodTool> m_electronLLHToolMediumNoPix{this, "ElectronLLHToolMediumNoPix", "", "Electron LLH tool to use for the overlap removal"}; 
        ToolHandle<IAsgElectronLikelihoodTool> m_electronLLHToolTightNoPix{this, "ElectronLLHToolTightNoPix", "", "Electron LLH tool to use for the overlap removal"}; 
 
        bool electronPassesID(const xAOD::Electron *electron, const std::string& IDWorkingPoint) const;
 
    }; // end namespace CP
 
}
#endif // LRTELECTRONANALYSISTOOLS_ELECTRONLRTOVERLAPREMOVALTOOL_H