egammaTrackThinner.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
/********************************************************************
NAME:     egammaTrackThinner.cxx
PACKAGE:  offline/Reconstruction/egammaRec
********************************************************************/
 
// INCLUDE HEADER FILES:
 
#include "egammaTrackThinner.h"
 
#include "xAODTracking/TrackParticleContainer.h"
#include "xAODTracking/VertexContainer.h"
#include "xAODEgamma/ElectronContainer.h"
#include "xAODEgamma/PhotonContainer.h"
#include "xAODEgamma/EgammaxAODHelpers.h"
 
#include "StoreGate/ReadHandle.h"
#include "StoreGate/ThinningHandle.h"
 
//  END OF HEADER FILES INCLUDE
 
 
 
// INITIALIZE METHOD:
 
StatusCode egammaTrackThinner::initialize()
{
 
  // initialize data handles
  ATH_CHECK(m_InputElectronContainerKey.initialize(m_doThinning));
  ATH_CHECK(m_InputPhotonContainerKey.initialize(m_doThinning));
  bool includeFwdEl =
    m_doThinning && !m_InputForwardElectronContainerKey.empty();
  ATH_CHECK(m_InputForwardElectronContainerKey.initialize(includeFwdEl));
  ATH_CHECK(m_TrackParticlesKey.initialize(m_streamName, m_doThinning));
  ATH_CHECK(m_VertexKey.initialize(m_streamName, m_doThinning));
 
  return StatusCode::SUCCESS;
}
 
// FINALIZE METHOD:
 
StatusCode egammaTrackThinner::finalize() {
  return StatusCode::SUCCESS;
}
 
// ATHENA EXECUTE METHOD:
 
StatusCode egammaTrackThinner::execute (const EventContext& ctx) const {
 
  ATH_MSG_DEBUG("Executing egammaTrackThinner");
  if(!m_doThinning){
    ATH_MSG_DEBUG("Thinning not requested do nothing");
    return StatusCode::SUCCESS;
  }
 
  /*
   * GSF Track Particles
   * The vector that we'll use to filter the track particles:
   */
  std::vector<bool> keptTrackParticles;
  SG::ThinningHandle<xAOD::TrackParticleContainer> trackPC(m_TrackParticlesKey, ctx);
  // check is only used for serial running; remove when MT scheduler used
  ATH_CHECK(trackPC.isValid());
  ATH_MSG_DEBUG("Number of TrackParticles "<< trackPC->size());
  keptTrackParticles.resize( trackPC->size(), false );
 
  /*
   * GSF Vertices
   * The vector that we'll use to filter the GSF vertices:
   */
  std::vector<bool> keptVertices;
  SG::ThinningHandle<xAOD::VertexContainer> vertices(m_VertexKey, ctx);
  // check is only used for serial running; remove when MT scheduler used
  ATH_CHECK(vertices.isValid());
  ATH_MSG_DEBUG("Number of Vertices "<< vertices->size());
  keptVertices.resize( vertices->size(), false );
 
  /*
   * In Det Track Particles
   * TRT standalone tracks are now centrally thinned by ThinTRTStandaloneTrackAlg for e/gamma and taus
   */
 
  /*
   * Electron track particle Thinning
   */
  SG::ReadHandle<xAOD::ElectronContainer> electrons(m_InputElectronContainerKey, ctx);
  // check is only used for serial running; remove when MT scheduler used
  ATH_CHECK(electrons.isValid());
 
  //Loop over electrons
  auto el_itr = electrons->begin();
  auto el_end = electrons->end();
  for( ; el_itr != el_end; ++el_itr ) {
 
    auto trackParticleLinks = (*el_itr)->trackParticleLinks();
    for (const auto& link :  trackParticleLinks){
      if( ! link.isValid() ){
        continue;
      }
      ATH_MSG_DEBUG("Electrons : Keeping GSF Track Particle with index : "<< link.index() );
      keptTrackParticles[link.index() ] = true;
    }
  }
 
  /*
   * Photon Vertex and track particle  Thinning
   */
  SG::ReadHandle<xAOD::PhotonContainer> photons(m_InputPhotonContainerKey, ctx);
  // check is only used for serial running; remove when MT scheduler used
  ATH_CHECK(photons.isValid());
 
  //Loop over photons
  auto ph_itr = photons->begin();
  auto ph_end = photons->end();
  for( ; ph_itr != ph_end; ++ph_itr ) {
 
    auto vertexLinks= (*ph_itr) ->vertexLinks();
    for ( const auto& vxlink :  vertexLinks){
      if( ! vxlink.isValid() ){
        continue;
      }
      ATH_MSG_DEBUG("Photons : Keeping GSF Vertex with index : "<< vxlink.index() );
      keptVertices[vxlink.index()]=true;
      const xAOD::Vertex* vx = *(vxlink);
      if(!vx){
        continue;
      }
 
      auto trackParticleLinks = vx->trackParticleLinks();
      for ( const auto& link :  trackParticleLinks){
        if( ! link.isValid() ){
          continue;
        }
        ATH_MSG_DEBUG("Photons : Keeping GSF Track Particle with index : "<< link.index() );
        keptTrackParticles[link.index() ] = true;
      }
    }
  }
 
  if (!m_InputForwardElectronContainerKey.empty()) {
    SG::ReadHandle<xAOD::ElectronContainer>
      fwdelectrons(m_InputForwardElectronContainerKey, ctx);
    // check is only used for serial running; remove when MT scheduler used
    ATH_CHECK(fwdelectrons.isValid());
 
    //Loop over forward electrons
    for (const auto *el : *fwdelectrons) {
 
      auto trackParticleLinks = el->trackParticleLinks();
      for (const auto& link : trackParticleLinks){
    if (!link.isValid()) {
      continue;
    }
    ATH_MSG_DEBUG("ForwardElectrons : "
              "Keeping GSF Track Particle with index : "
              << link.index());
    keptTrackParticles[link.index()] = true;
      }
    }
  }
 
  //Do the Thinning
  ATH_MSG_DEBUG("Do the Thinning");
  trackPC.keep (keptTrackParticles);
  vertices.keep (keptVertices);
  ATH_MSG_DEBUG("completed successfully");
 
  //Return Gracefully
  return StatusCode::SUCCESS;
}