AltMuJetOverlapTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ASSOCIATIONUTILS_ALTMUJETOVERLAPTOOL_H
#define ASSOCIATIONUTILS_ALTMUJETOVERLAPTOOL_H
 
// Framework includes
#include "AsgDataHandles/ReadHandleKey.h"
 
// EDM includes
#include "xAODMuon/MuonContainer.h"
#include "xAODJet/JetContainer.h"
#include "xAODTracking/VertexContainer.h"
 
// Columnar includes
#include "ColumnarCore/ObjectColumn.h"
#include "ColumnarCore/OptObjectId.h"
#include "ColumnarCore/VectorColumn.h"
#include "ColumnarMuon/MuonDef.h"
#include "ColumnarJet/JetDef.h"
#include "ColumnarTracking/TrackDef.h"
 
// Local includes
#include "AssociationUtils/IOverlapTool.h"
#include "AssociationUtils/BaseOverlapTool.h"
#include "AssociationUtils/IObjectAssociator.h"
 
namespace ORUtils
{

  class AltMuJetOverlapTool : public virtual IOverlapTool,
                              public BaseOverlapTool
  {

      ASG_TOOL_CLASS(AltMuJetOverlapTool, IOverlapTool)
 
    public:

      AltMuJetOverlapTool(const std::string& name);

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

      virtual StatusCode
      internalFindOverlaps(columnar::Particle1Range muons,
                           columnar::Particle2Range jets,
                           columnar::EventContextId eventContext) const;
 
    protected:

      virtual StatusCode initializeDerived() override;

      int getPrimVtxIndex(columnar::EventContextId eventContext) const;

      int getNumTracks(columnar::Particle2Id jet, columnar::EventContextId eventContext) const;
 
    private:
 
      //
      // Configurable properties
      //

      std::string m_bJetLabel;

      int m_numJetTrk;
      float m_muJetPtRatio;

      float m_innerDR;
      double m_slidingDRC1;
      double m_slidingDRC2;
      double m_slidingDRMaxCone;
      bool m_useRapidity;
      std::string m_PVContName;

      struct Accessors final : columnar::ColumnarTool<>
      {
        columnar::Track0Accessor<columnar::ObjectColumn> m_track0Acc {*this, "InDetTrackParticles"};
        columnar::Track1Accessor<columnar::ObjectColumn> m_track1Acc {*this, "InDetForwardTrackParticles"};
        columnar::VertexAccessor<columnar::ObjectColumn> m_vtxContainerAcc;
        columnar::VertexAccessor<columnar::RetypeColumn<xAOD::VxType::VertexType,short>> m_vertexTypeAcc {*this, "vertexType"};
        columnar::Particle1Accessor<float> m_muonPtAcc {*this, "pt"};
        columnar::Particle2Accessor<float> m_jetPtAcc {*this, "pt"};
        columnar::Particle2Accessor< std::vector<int> > m_numTrkPt500Acc {*this, "NumTrkPt500"};
        columnar::Particle2Accessor<char> m_bJetAcc;
        using ColumnarTool::ColumnarTool;
      };
      std::unique_ptr<Accessors> m_accessors {std::make_unique<Accessors> (this)};
 
      //
      // Utilities
      //

      std::unique_ptr<IParticleAssociator> m_dRMatchCone1;
      std::unique_ptr<IParticleAssociator> m_dRMatchCone2;
 
  }; // class AltMuJetOverlapTool
 
} // namespace ORUtils
 
#endif