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_DEBUG( "Initializing " << name() );
 
  ATH_CHECK( m_offlineSelectionTool.retrieve( EnableTool{ m_doOfflSelection.value() } ) );
  ATH_CHECK( m_truthSelectionTool.retrieve( EnableTool{ m_doTruthSelection.value() } ) );
  ATH_CHECK( m_objSelectionTool.retrieve( EnableTool{ m_doObjSelection.value() } ) );
 
  return StatusCode::SUCCESS;
}
 

StatusCode IDTPM::TrackQualitySelectionTool::selectTracks(
    TrackAnalysisCollections& trkAnaColls ) {
 
  ATH_MSG_DEBUG( "Initially copying collections to FullScan vectors" );
 
  ISvcLocator* svcLoc = Gaudi::svcLocator();
  SmartIF< ITrackAnalysisDefinitionSvc > trkAnaDefSvc(
      svcLoc->service( "TrkAnaDefSvc" + trkAnaColls.anaTag() ) );
  ATH_CHECK( trkAnaDefSvc.isValid() );

  if( trkAnaDefSvc->useOffline() ) {
    ATH_CHECK( trkAnaColls.fillOfflTrackVec(
        trkAnaColls.offlTrackVec( TrackAnalysisCollections::FULL ),
        TrackAnalysisCollections::FS ) );
  }
 
  if( trkAnaDefSvc->useTrigger() or 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_MSG_DEBUG( "Tracks after initial FullScan copy: " << 
      trkAnaColls.printInfo( TrackAnalysisCollections::FS, false ) );

  if( trkAnaDefSvc->useOffline() and m_doOfflSelection.value() ) {
    ATH_CHECK( m_offlineSelectionTool->selectTracks( trkAnaColls ) );
  }

  if( trkAnaDefSvc->useTruth() and m_doTruthSelection.value() ) {
    ATH_CHECK( m_truthSelectionTool->selectTracks( trkAnaColls ) );
  }


  if( trkAnaDefSvc->useOffline() and m_doObjSelection.value() ) {
    ATH_CHECK( m_objSelectionTool->selectTracks( trkAnaColls ) );
  }

  ATH_MSG_DEBUG( "Tracks after full quality selection: " << 
      trkAnaColls.printInfo( TrackAnalysisCollections::FS, false ) );
 
  return StatusCode::SUCCESS;
}