InDetPRDtoTrackMapToolGangedPixels.h

Go to the documentation of this file.

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

  class InDetPRDtoTrackMapToolGangedPixels : public extends<AthAlgTool, Trk::IPRDtoTrackMapTool>
{
 
  public:
    InDetPRDtoTrackMapToolGangedPixels(const std::string&,const std::string&,const IInterface*);
 
    virtual ~InDetPRDtoTrackMapToolGangedPixels ();
    virtual StatusCode initialize() override;
    virtual StatusCode finalize  () override;
 
    virtual
    std::unique_ptr<Trk::PRDtoTrackMap> createPRDtoTrackMap() const override;
    virtual
    std::unique_ptr<Trk::PRDtoTrackMap> reduceToStorableMap(std::unique_ptr<Trk::PRDtoTrackMap> &&obj_in) const override;

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

    virtual std::vector< const Trk::PrepRawData* > getPrdsOnTrack(Trk::PRDtoTrackMap &virt_prd_to_track_map,
                                                                  const Trk::Track& track) const override;

  virtual Trk::IPRDtoTrackMapTool::TrackSet findConnectedTracks( Trk::PRDtoTrackMap &virt_prd_to_track_map,
                                                                const Trk::Track& track) const override;
 
  private:
 
    void ensureType(Trk::PRDtoTrackMap &virt_prd_to_track_map) const;
 
    SG::ReadHandleKey<PixelGangedClusterAmbiguities> m_pixelClusterAmbiguitiesMapName{this, "PixelClusterAmbiguitiesMapName", "PixelClusterAmbiguitiesMap"};

    BooleanProperty m_addTRToutliers{this, "addTRToutliers", false};
 
    class PRDtoTrackMap : public Trk::PRDtoTrackMap
    {
      friend class InDetPRDtoTrackMapToolGangedPixels;
    public:
      const std::type_info& getType() const override {return typeid(PRDtoTrackMap); }
 
      PRDtoTrackMap(const SG::ReadHandleKey<PixelGangedClusterAmbiguities> &key);
 
    protected:
      SG::ReadHandle<PixelGangedClusterAmbiguities> m_gangedAmbis;
    };
  };
 
 
 
}
#endif