TrackQualitySelectionTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "TrackQualitySelectionTool.h"
#include "TrackAnalysisCollections.h"
 
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/Service.h"
 
 
IDTPM::TrackQualitySelectionTool::TrackQualitySelectionTool( 
    const std::string& name ) :
  asg::AsgTool( name ) { }
 
 
StatusCode IDTPM::TrackQualitySelectionTool::initialize() {
 
  ATH_CHECK( asg::AsgTool::initialize() );
 
  ATH_MSG_INFO( "Initializing " << name() );
 
  ATH_CHECK( m_objSelectionTool.retrieve( EnableTool{ m_doObjSelection.value() } ) );
  ATH_CHECK( m_truthSelectionTool.retrieve( EnableTool{ not m_truthSelectionTool.empty() } ) );
 
  ATH_CHECK( m_offlineSelectionTool.retrieve(EnableTool{ not m_offlineSelectionTool.empty() } ));
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode IDTPM::TrackQualitySelectionTool::selectTracks(
    TrackAnalysisCollections& trkAnaColls ) {
 
  ATH_MSG_DEBUG( "Initially copying collections to FullScan vectors" );
 
  ITrackAnalysisDefinitionSvc* trkAnaDefSvc( nullptr );
  ISvcLocator* svcLoc = Gaudi::svcLocator();
  ATH_CHECK( svcLoc->service( "TrkAnaDefSvc" + trkAnaColls.anaTag(), trkAnaDefSvc ) );
 
  if( trkAnaDefSvc->useOffline() ) {
    ATH_CHECK( trkAnaColls.fillOfflTrackVec(
        trkAnaColls.offlTrackVec( TrackAnalysisCollections::FULL ),
        TrackAnalysisCollections::FS ) );
  }
 
  if( trkAnaDefSvc->useEFTrigger() ) {
    ATH_CHECK( trkAnaColls.fillTrigTrackVec(
        trkAnaColls.trigTrackVec( TrackAnalysisCollections::FULL ),
        TrackAnalysisCollections::FS ) );
  }
 
  if( trkAnaDefSvc->useTruth() ) {
    ATH_CHECK( trkAnaColls.fillTruthPartVec(
        trkAnaColls.truthPartVec( TrackAnalysisCollections::FULL ),
        TrackAnalysisCollections::FS ) );
    ATH_CHECK (m_truthSelectionTool->selectTracks( trkAnaColls ));
  }
 
  ATH_MSG_DEBUG( "Tracks after initial FullScan copy: " << 
      trkAnaColls.printInfo( TrackAnalysisCollections::FS ) );
 
  if( trkAnaDefSvc->useOffline() and m_doObjSelection.value() ) {
    ATH_CHECK( m_objSelectionTool->selectTracks( trkAnaColls ) );
  }
 
  if ( m_offlineSelectionTool ) {
    ATH_CHECK(m_offlineSelectionTool->selectTracks(trkAnaColls));
  }
 
  return StatusCode::SUCCESS;
}