TrackCollectionMerger.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/

//   Implementation file for class Trk::TrackCollectionMerger

// Version 1.0 11/26/2007 Thomas Koffas
#include "GaudiKernel/MsgStream.h"
#include "TrkPrepRawData/PrepRawData.h"
#include "TrkTrackCollectionMerger/TrackCollectionMerger.h"

// Constructor
Trk::TrackCollectionMerger::TrackCollectionMerger(const std::string& name,
                                                  ISvcLocator* pSvcLocator)
  : AthReentrantAlgorithm(name, pSvcLocator) {}

// Initialisation
StatusCode
Trk::TrackCollectionMerger::initialize()
{
 
  ATH_MSG_DEBUG("Initializing TrackCollectionMerger");
  ATH_CHECK(m_tracklocation.initialize());
  ATH_CHECK(m_pileupTRT.initialize(m_doTrackOverlay));
  ATH_CHECK(m_pileupPixel.initialize(m_doTrackOverlay));
  ATH_CHECK(m_pileupSCT.initialize(m_doTrackOverlay));
  ATH_CHECK(m_outtracklocation.initialize());
  ATH_CHECK(m_assoTool.retrieve(DisableTool{m_assoTool.name().empty()}));
  ATH_CHECK(m_assoMapName.initialize(!m_assoMapName.key().empty()));
  return StatusCode::SUCCESS;
}

// Execute
StatusCode 
Trk::TrackCollectionMerger::execute(const EventContext& ctx) const{
 
  auto outputCol = std::make_unique<ConstDataVector<TrackCollection>>(SG::VIEW_ELEMENTS);
  ATH_MSG_DEBUG("Number of Track collections " << m_tracklocation.size());
  
  // pre-loop to reserve enough space in the output collection
  std::vector<const TrackCollection*> trackCollections;
  trackCollections.reserve(m_tracklocation.size());
  size_t ttNumber = 0;
  for (const auto& tcname : m_tracklocation){
    SG::ReadHandle<TrackCollection> trackCol (tcname, ctx);
    trackCollections.push_back(trackCol.cptr());
    ttNumber += trackCol->size();
  }
  std::unique_ptr<Trk::PRDtoTrackMap> pPrdToTrackMap(m_assoTool ? m_assoTool->createPRDtoTrackMap() : nullptr);
  // reserve the right number of entries for the output collection
  outputCol->reserve(ttNumber);
  // merging loop
  for (auto& tciter : trackCollections) {
    // merge them in
    if (mergeTrack(tciter, pPrdToTrackMap.get(), outputCol.get()).isFailure()) {
      ATH_MSG_ERROR("Failed to merge tracks! ");
    }
  }
  ATH_MSG_DEBUG("Size of combined tracks " << outputCol->size());
 
  auto h_write = SG::makeHandle(m_outtracklocation, ctx);
  ATH_CHECK(h_write.record(std::move(outputCol)));      
  //
  if (!m_assoMapName.key().empty()) {
     SG::WriteHandle<Trk::PRDtoTrackMap> write_handle(m_assoMapName, ctx);
     if (write_handle.record( m_assoTool->reduceToStorableMap(std::move(pPrdToTrackMap))).isFailure()) {
        ATH_MSG_FATAL("Failed to add PRD to track association map.");
     }
  }
  //Print common event information
  ATH_MSG_DEBUG("Done !");  
  return StatusCode::SUCCESS;
}

// Finalize
StatusCode
Trk::TrackCollectionMerger::finalize()
{
  return StatusCode::SUCCESS;
}

// Merge track collections and remove duplicates
StatusCode
Trk::TrackCollectionMerger::mergeTrack(const TrackCollection* trackCol,
                                       Trk::PRDtoTrackMap* pPrdToTrackMap,
                                       ConstDataVector<TrackCollection>* outputCol) const
{
  // loop over tracks, accept them and add them into association tool
  if (trackCol && !trackCol->empty()) {
    ATH_MSG_DEBUG("Size of track collection " << trackCol->size());
    if (not pPrdToTrackMap)
      ATH_MSG_WARNING("No valid PRD to Track Map; was the association tool name missing?");
    // loop over tracks
    for (const auto* const rf : *trackCol) {
      outputCol->push_back(rf);
      // add tracks into PRD tool
      if (m_assoTool and m_assoTool->addPRDs(*pPrdToTrackMap, *rf).isFailure())
        ATH_MSG_WARNING("Failed to add PRDs to map");
    }
  }
 
  return StatusCode::SUCCESS;
}