InDetPRD_AssociationToolGangedPixels.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef INDET_PRD_ASSOCIATIONTOOL_H
#define INDET_PRD_ASSOCIATIONTOOL_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ServiceHandle.h"
#include "StoreGate/ReadHandleKey.h"
#include "TrkToolInterfaces/IPRD_AssociationTool.h"
#include "InDetPrepRawData/PixelGangedClusterAmbiguities.h"
 
class AtlasDetectorID;
class Identifier;
 
namespace InDet {

  class InDetPRD_AssociationToolGangedPixels : public extends<AthAlgTool, Trk::IPRD_AssociationTool>
  {
 
  public:
    typedef Trk::IPRD_AssociationTool::Maps Maps;
 
    InDetPRD_AssociationToolGangedPixels(const std::string&,const std::string&,const IInterface*);
    virtual ~InDetPRD_AssociationToolGangedPixels ();
    virtual StatusCode initialize() override;
    virtual StatusCode finalize  () override;

    virtual StatusCode addPRDs(const Trk::Track& track) override;

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

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

    virtual StatusCode removePRDs(const Trk::Track& track) override;

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

    virtual bool isUsed(const Trk::PrepRawData& prd) const override;

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

    virtual bool isShared(const Trk::PrepRawData& prd) const override;

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

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

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

    virtual void reset() override;
 
  private:
    Maps m_maps;
 
    SG::ReadHandleKey<PixelGangedClusterAmbiguities> m_pixelClusterAmbiguitiesMapName{this, "PixelClusterAmbiguitiesMapName", "PixelClusterAmbiguitiesMap"};

    BooleanProperty m_addTRToutliers{this, "addTRToutliers", false};
 
};
 
inline bool
InDet::InDetPRD_AssociationToolGangedPixels::isUsed(const Maps& maps,
                                                    const Trk::PrepRawData& prd) const
{
  return (maps.m_prepRawDataTrackMap.count(&prd)>0);
}
 
inline bool
InDet::InDetPRD_AssociationToolGangedPixels::isUsed(const Trk::PrepRawData& prd) const
{
  return isUsed (m_maps, prd);
}
 
inline bool
InDet::InDetPRD_AssociationToolGangedPixels::isShared(const Maps& maps,
                                                      const Trk::PrepRawData& prd) const
{
  return (maps.m_prepRawDataTrackMap.count(&prd)>1);
}
 
inline bool
InDet::InDetPRD_AssociationToolGangedPixels::isShared(const Trk::PrepRawData& prd) const
{
  return isShared (m_maps, prd);
}
 
 
}
#endif