TrackCollectionMerger.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
// TrackCollectionMerger.h, (c) ATLAS Detector software
#ifndef TrackCollectionMerger_H
#define TrackCollectionMerger_H
 
#include "AthContainers/ConstDataVector.h"
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "GaudiKernel/ToolHandle.h"
 
#include "TrkEventUtils/PRDtoTrackMap.h"
#include "TrkToolInterfaces/IPRDtoTrackMapTool.h"
#include "TrkTrack/TrackCollection.h"
 
#include "InDetPrepRawData/PixelClusterContainer.h"
#include "InDetPrepRawData/SCT_ClusterContainer.h"
#include "InDetPrepRawData/TRT_DriftCircleContainer.h"
 
#include "StoreGate/WriteHandleKey.h"
#include "StoreGate/ReadHandleKeyArray.h"
 
#include <string>
#include <map>
 
 
namespace Trk {
 
  class TrackCollectionMerger final: public AthReentrantAlgorithm
    {
      
    public:
      
 
 
      TrackCollectionMerger(const std::string &name, ISvcLocator *pSvcLocator);
      virtual ~TrackCollectionMerger() {}
      virtual StatusCode initialize() override final;
      virtual StatusCode execute(const EventContext& ctx) const override final;
      virtual StatusCode finalize() override final;
 
    protected:
 
 
      
      SG::ReadHandleKeyArray<TrackCollection> m_tracklocation {
    this, "TracksLocation", {}};
      SG::ReadHandleKey<TrackCollection> m_overlayTrackLocation{
        this,
        "OverlayTracksLocation",
        "",
        "Pileup Track Collection"
      }; 
     // note that these handles are not used directly, they are needed here
      // just to ensure that these collections get loaded into SG
      SG::ReadHandleKey<InDet::TRT_DriftCircleContainer> m_pileupTRT{
        this,
        "OverlayTRTClusters",
        "Bkg_TRT_DriftCircles",
        "Pileup Drift Circles"
      }; 
      SG::ReadHandleKey<InDet::PixelClusterContainer> m_pileupPixel{
        this,
        "OverlayPixelClusters",
        "Bkg_PixelClusters",
        "Pileup Pixel Clusters"
      };
      SG::ReadHandleKey<InDet::SCT_ClusterContainer> m_pileupSCT{
        this,
        "OverlaySCTClusters",
        "Bkg_SCT_Clusters",
        "Pileup SCT Clusters"
      }; 
      SG::WriteHandleKey<ConstDataVector<TrackCollection>> m_outtracklocation
    {this, "OutputTracksLocation", "CombinedInDetTracks"};
 
      SG::WriteHandleKey<Trk::PRDtoTrackMap> m_assoMapName{
        this,
        "AssociationMapName",
        ""
      }; 
 
      ToolHandle<Trk::IPRDtoTrackMapTool> m_assoTool{
        this,
        "AssociationTool",
        "InDet::InDetPRDtoTrackMapToolGangedPixels"
      };
 
 
 
      StatusCode mergeTrack(const TrackCollection* trackCol,
                            Trk::PRDtoTrackMap* pPrdToTrackMap,
                            ConstDataVector<TrackCollection>* outputCol) const;
 
    private:
 
      //doing track overlay: needed to initialize the background PRD containers
      BooleanProperty m_doTrackOverlay{this, "DoTrackOverlay", false};
 
    };
    
}
#endif // TrackCollectionMerger_H