PRD_AssociationTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRK_PRD_ASSOCIATIONTOOL_H
#define TRK_PRD_ASSOCIATIONTOOL_H
 
#include "TrkToolInterfaces/IPRD_AssociationTool.h"
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ServiceHandle.h"
 
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
#include "StoreGate/ReadHandleKey.h"
#include "TrkEventUtils/PRDtoTrackMap.h"
#include <set>
#include <map>
 
namespace Trk {
 
  class Track;
  class PrepRawData;

  class PRD_AssociationTool final
      : public extends<AthAlgTool, IPRD_AssociationTool> {
 
   public:
    PRD_AssociationTool(const std::string&,const std::string&,const IInterface*);
    virtual ~PRD_AssociationTool()=default;
    virtual StatusCode initialize() override;

    virtual StatusCode addPRDs(const Track& track) override;

  virtual StatusCode addPRDs(Maps& maps, const Track& track) const override;

    virtual StatusCode removePRDs(Maps& maps, const Track& track) const override;

    virtual StatusCode removePRDs(const Track& track) override;

    virtual bool isUsed(const Maps& maps, const PrepRawData& prd) const override final;

    virtual bool isUsed(const PrepRawData& prd) const override final;

    virtual bool isShared(const Maps& maps, const PrepRawData& prd) const override final;

    virtual bool isShared(const PrepRawData& prd) const override final;

    virtual std::vector< const PrepRawData* > getPrdsOnTrack(const Track& track) const override;
 
    // getPrdsOnTrack with state passed explicitly.
    virtual std::vector< const PrepRawData* > getPrdsOnTrack(const Maps& maps,
                                                             const Track& track) const override;

    virtual IPRD_AssociationTool::TrackSet findConnectedTracks( const Track& track) const override;
 
    // findConnectedTracks with state passed explicitly.
    virtual IPRD_AssociationTool::TrackSet findConnectedTracks( const Maps& maps, const Track& track) const override;

    virtual IPRD_AssociationTool::PrepRawDataTrackMapRange onTracks(const PrepRawData& prd) const override;
 
    // onTracks with explicit state
    virtual IPRD_AssociationTool::PrepRawDataTrackMapRange onTracks(const Maps& maps,
                                                                    const PrepRawData& prd) const override;

    virtual void reset() override;
 
  private:
    // Holds the associations.
    Maps m_maps;
    SG::ReadHandleKey<Trk::PRDtoTrackMap> m_prdToTrackMap {this,"PRDtoTrackMap",""};
 
    ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc {this, "MuonIdHelperSvc", "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};
  };
 
inline bool Trk::PRD_AssociationTool::isUsed(const Maps& maps,
                                             const PrepRawData& prd) const
{
  return (maps.m_prepRawDataTrackMap.count(&prd)>0);
}
 
inline bool Trk::PRD_AssociationTool::isUsed(const PrepRawData& prd) const
{
  if (!m_prdToTrackMap.key().empty()) {return SG::ReadHandle<Trk::PRDtoTrackMap>(m_prdToTrackMap)->isUsed(prd);}
  return isUsed (m_maps, prd);
}
 
inline bool Trk::PRD_AssociationTool::isShared(const Maps& maps,
                                               const PrepRawData& prd) const
{
  return (maps.m_prepRawDataTrackMap.count(&prd)>1);
}
 
inline bool Trk::PRD_AssociationTool::isShared(const PrepRawData& prd) const
{
  if (!m_prdToTrackMap.key().empty()) {return SG::ReadHandle<Trk::PRDtoTrackMap>(m_prdToTrackMap)->isShared(prd);}
  return isShared (m_maps, prd);
}
 
}
#endif