JetGhostMergingAlg.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
 
//
// includes
//
#include <JetAnalysisAlgorithms/JetGhostMergingAlg.h>
#include <AsgDataHandles/WriteDecorHandle.h>
#include <AsgDataHandles/ReadDecorHandle.h>
 
namespace CP
{
  JetGhostMergingAlg ::
  JetGhostMergingAlg (const std::string& name, 
                      ISvcLocator* pSvcLocator)
    : AnaAlgorithm (name, pSvcLocator)
  {
  }
 
  StatusCode JetGhostMergingAlg :: 
  initialize ()
  {
    // containers
    ATH_CHECK (m_jetLocation.initialize ());
 
    // decorators
    ATH_CHECK (m_mergedGhostContainer.initialize ());
 
    // accessors
    m_ghostTrackKeys.reserve(m_ghostTrackKeys.size() + m_inputGhostTrackNames.size());
    for (const auto& ghostName: m_inputGhostTrackNames) {
      std::string full = m_jetLocation.key() + "." + ghostName;
      m_ghostTrackKeys.emplace_back( this, ghostName, full, "");
      ATH_CHECK( m_ghostTrackKeys.back().initialize() );
    }
 
    return StatusCode::SUCCESS;
  }
 
  StatusCode JetGhostMergingAlg :: 
  execute ()
  {
    const EventContext &ctx = Gaudi::Hive::currentContext();
 
    SG::ReadHandle<xAOD::JetContainer> inputJets(m_jetLocation,ctx);
    if (!inputJets.isValid()) {
        ATH_MSG_FATAL("No jet collection with name " << m_jetLocation.key() << " found in StoreGate!");
        return StatusCode::FAILURE;
    }
 
    // define GhostTrackContainer type (GTC)
    using GTC = std::vector<ElementLink<DataVector<xAOD::IParticle> > >;
 
    // create the accessors for each ghost track collection
    std::vector<SG::ReadDecorHandle<xAOD::JetContainer, GTC> > ghostTrackAccs;
    ghostTrackAccs.reserve(m_ghostTrackKeys.size());
    for (const auto &key: m_ghostTrackKeys) {
      ghostTrackAccs.emplace_back(key, ctx);
    }
 
    SG::WriteDecorHandle<xAOD::JetContainer, GTC > mergedGhostDecor(m_mergedGhostContainer, ctx);
 
    for(const xAOD::Jet *jet: *inputJets) {
        std::vector<ElementLink<xAOD::IParticleContainer>> mergedGhosts;
        for (const auto& ghostTrackAcc: ghostTrackAccs) {
            const GTC &ghosts = ghostTrackAcc( *jet );
            mergedGhosts.insert(mergedGhosts.end(), ghosts.begin(), ghosts.end());
        }
        mergedGhostDecor(*jet) = std::move(mergedGhosts);
    }
    return StatusCode::SUCCESS;
  }
 
}