EleJetOverlapTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ASSOCIATIONUTILS_ELEJETOVERLAPTOOL_H
#define ASSOCIATIONUTILS_ELEJETOVERLAPTOOL_H
 
// Framework includes
#include "AsgTools/AsgTool.h"
#include "AsgDataHandles/ReadHandleKey.h"
 
// EDM includes
#include "xAODEgamma/ElectronContainer.h"
#include "xAODJet/JetContainer.h"
#include "ColumnarEgamma/EgammaDef.h"
#include "ColumnarJet/JetDef.h"
 
// Local includes
#include "AssociationUtils/IOverlapTool.h"
#include "AssociationUtils/BaseOverlapTool.h"
#include "AssociationUtils/DeltaRMatcher.h"
#include "AssociationUtils/FEMatcher.h"
 
namespace ADD_CMODE(ORUtils)
{

  class EleJetOverlapTool : public virtual IOverlapTool,
                            public BaseOverlapTool
  {

      ASG_TOOL_CLASS(EleJetOverlapTool, IOverlapTool)
 
    public:

      EleJetOverlapTool(const std::string& name);

      virtual StatusCode
      findOverlaps(columnar::Particle1Range cont1,
                   columnar::Particle2Range cont2,
                   columnar::EventContextId eventContext) const override;

      virtual StatusCode
      internalFindOverlaps(columnar::Particle1Range electrons,
                           columnar::Particle2Range jets) const;
 
    protected:

      virtual StatusCode initializeDerived() override;
 
    private:


      std::string m_bJetLabel;

      double m_maxElePtForBJetAwareOR;

      bool m_applyPtRatio;
      double m_eleJetPtRatio;

      float m_innerDR;
      float m_outerDR;

      bool m_useSlidingDR;
      double m_slidingDRC1;
      double m_slidingDRC2;
      double m_slidingDRMaxCone;

      bool m_useRapidity;

      bool m_useFEOR;

      struct Accessors final : columnar::ColumnarTool<>
      {
        columnar::Particle1Accessor<float> m_elePtAcc {*this, "pt"};
        columnar::Particle2Accessor<float> m_jetPtAcc {*this, "pt"};
        columnar::Particle2Accessor<char> m_bJetAcc;
        using ColumnarTool::ColumnarTool;
      };
      std::unique_ptr<Accessors> m_accessors {std::make_unique<Accessors> (this)};
 
      SG::ReadHandleKey<xAOD::MissingETAssociationMap> m_feMapKey{
        this, "FEAssociationMap", "FEAssociationMap", "Flow element association map"
      };



      std::unique_ptr<IParticleAssociator> m_dRMatchCone1;
      std::unique_ptr<IParticleAssociator> m_dRMatchCone2;
      // Flow element matcher
      std::unique_ptr<FEMatcher> m_FEMatcher;

 
  }; // class EleJetOverlapTool
 
} // namespace ADD_CMODE(ORUtils)
 
#endif