VertexQualitySelectionTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/


#include "VertexQualitySelectionTool.h"
#include "TrackAnalysisCollections.h"

#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/Service.h"
 

IDTPM::VertexQualitySelectionTool::VertexQualitySelectionTool( 
    const std::string& name ) :
  asg::AsgTool( name ) { }
 

StatusCode IDTPM::VertexQualitySelectionTool::initialize() {
 
  ATH_CHECK( asg::AsgTool::initialize() );
 
  ATH_MSG_DEBUG( "Initializing " << name() );

 
  return StatusCode::SUCCESS;
}
 

StatusCode IDTPM::VertexQualitySelectionTool::selectVertices(
    TrackAnalysisCollections& trkAnaColls ) {
 
  ATH_MSG_DEBUG( "Initially copying vertex 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.fillOfflVertexVec(
        trkAnaColls.offlVertexVec( TrackAnalysisCollections::FULL ),
        TrackAnalysisCollections::FS ) );
  }
 
  if( trkAnaDefSvc->useTrigger() or trkAnaDefSvc->useEFTrigger() ) {
    ATH_CHECK( trkAnaColls.fillTrigVertexVec(
        trkAnaColls.trigVertexVec( TrackAnalysisCollections::FULL ),
        TrackAnalysisCollections::FS ) );
  }
 
  if( trkAnaDefSvc->useTruth() ) {
    ATH_CHECK( trkAnaColls.fillTruthVertexVec(
        trkAnaColls.truthVertexVec( TrackAnalysisCollections::FULL ),
        TrackAnalysisCollections::FS ) );
  }

  ATH_MSG_DEBUG( "Vertices after initial FullScan copy: " << 
      trkAnaColls.printVertexInfo( IDTPM::TrackAnalysisCollections::FS ) );

 
  return StatusCode::SUCCESS;
}