TruthTrackQualitySelectionTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "TruthTrackQualitySelectionTool.h"
#include <GaudiKernel/StatusCode.h>
#include "TrackAnalysisCollections.h"
 
IDTPM::TruthTrackQualitySelectionTool::TruthTrackQualitySelectionTool( const std::string& name )
  : asg::AsgTool( name ) { }
 
StatusCode IDTPM::TruthTrackQualitySelectionTool::initialize() {
  ATH_CHECK(not m_truthTool.empty());
  ATH_CHECK(m_truthTool.retrieve());
  return StatusCode::SUCCESS;
}
 
 
StatusCode IDTPM::TruthTrackQualitySelectionTool::selectTracks( TrackAnalysisCollections& trkAnaColls ) {
  std::vector< const xAOD::TruthParticle* > selected;
  for ( auto trk: trkAnaColls.truthPartVec(IDTPM::TrackAnalysisCollections::FS)) {
    if ( m_truthTool->accept(trk)) {
      selected.push_back(trk);
    }
  }
  ATH_MSG_DEBUG("Size before selection: " << trkAnaColls.truthPartVec(IDTPM::TrackAnalysisCollections::FS).size() << "\t Size after selection: " << selected.size());
  ATH_CHECK(trkAnaColls.fillTruthPartVec(selected, IDTPM::TrackAnalysisCollections::FS));
  return StatusCode::SUCCESS;
}