EleMuSharedTrkOverlapTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2018 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ASSOCIATIONUTILS_SHAREDTRKOVERLAPTOOL_H
#define ASSOCIATIONUTILS_SHAREDTRKOVERLAPTOOL_H
 
// Framework includes
#include "AsgTools/AsgTool.h"
 
// EDM includes
#include "xAODEgamma/ElectronContainer.h"
#include "xAODMuon/MuonContainer.h"
#include "ColumnarEgamma/EgammaDef.h"
#include "ColumnarMuon/MuonDef.h"
 
// Columnar includes
#include "ColumnarCore/LinkColumn.h"
#include "ColumnarTracking/TrackDef.h"
#include "ColumnarVariant/VariantDef.h"
#include "ColumnarVariant/VariantLinkColumn.h"
 
// Local includes
#include "AssociationUtils/IOverlapTool.h"
#include "AssociationUtils/BaseOverlapTool.h"
#include "AssociationUtils/DeltaRMatcher.h"
 
namespace ADD_CMODE(ORUtils)
{

  class EleMuSharedTrkOverlapTool : public virtual IOverlapTool,
                                    public BaseOverlapTool
  {

      ASG_TOOL_CLASS(EleMuSharedTrkOverlapTool, IOverlapTool)
 
    public:

      EleMuSharedTrkOverlapTool(const std::string& name);

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

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

      virtual StatusCode initializeDerived() override;
 
    private:
 
      //
      // Configurable properties
      //

      bool m_removeCaloMuons;

      bool m_useDRMatching;
      float m_maxDR;
      bool m_useRapidity;
 
      //
      // Utilities
      //

      using MyTrackDef = columnar::VariantContainerId<columnar::Track0Def,columnar::Track0Def, columnar::Track1Def>;

      struct Accessors final : columnar::ColumnarTool<>
      {
        columnar::Track0Accessor<columnar::ObjectColumn> m_track0Acc {*this, "InDetTrackParticles"};
        columnar::Track1Accessor<columnar::ObjectColumn> m_track1Acc {*this, "InDetForwardTrackParticles"};
        columnar::Track2Accessor<columnar::ObjectColumn> m_track2Acc {*this, "GSFTrackParticles"};
        columnar::Particle1Accessor<std::vector<columnar::OptTrack2Id>> m_eleTrackAcc {*this, "trackParticleLinks"};
        columnar::Particle2Accessor<columnar::ObjectLink<MyTrackDef>> m_muonTrkAcc {*this, "inDetTrackParticleLink"};
        columnar::Particle2Accessor<columnar::RetypeColumn<xAOD::Muon::MuonType,std::uint16_t>> m_muonTypeAcc {*this, "muonType"};
        columnar::Track2Accessor<columnar::ObjectLink<MyTrackDef>> m_gsfOriginalTrackAcc {*this, "originalTrackParticle"};
        using ColumnarTool::ColumnarTool;
      };
      std::unique_ptr<Accessors> m_accessors {std::make_unique<Accessors> (this)};
 
      [[nodiscard]] columnar::ObjectLink<MyTrackDef> getOriginalTrackParticle(columnar::Particle1Id electron) const;

      std::unique_ptr<DeltaRMatcher> m_dRMatcher;
 
  }; // class EleMuSharedTrkOverlapTool
 
} // namespace ADD_CMODE(ORUtils)
 
#endif