TrackTruthMatchingTool.cxx

Go to the documentation of this file.

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


#include "TrackTruthMatchingTool.h"
#include "TrackAnalysisCollections.h"
#include "TrackMatchingLookup.h"
#include "OfflineObjectDecorHelper.h"
#include "TrackParametersHelper.h"

#include <algorithm> // for std::find

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

StatusCode IDTPM::TrackTruthMatchingTool::initialize()
{
  ATH_CHECK( asg::AsgTool::initialize() );
 
  ATH_MSG_DEBUG( "Initializing " << name() << "..." );
 
  return StatusCode::SUCCESS;
}
 

StatusCode IDTPM::TrackTruthMatchingTool::match(
    TrackAnalysisCollections& trkAnaColls,
    const std::string& chainRoIName,
    const std::string& roiStr ) const
{
  bool doMatch = trkAnaColls.updateChainRois( chainRoIName, roiStr );

  if( not doMatch ) {
    ATH_MSG_WARNING( "Matching for " << chainRoIName <<
                     " was already done. Skipping" );
    return StatusCode::SUCCESS;
  }

  ATH_CHECK( match(
      trkAnaColls.testTrackVec( TrackAnalysisCollections::InRoI ),
      trkAnaColls.refTruthVec( TrackAnalysisCollections::InRoI ),
      trkAnaColls.matches() ) );
 
  ATH_MSG_DEBUG( trkAnaColls.printMatchInfo() );
 
  return StatusCode::SUCCESS;
}
 

StatusCode IDTPM::TrackTruthMatchingTool::match(
    const std::vector< const xAOD::TrackParticle* >& vTest,
    const std::vector< const xAOD::TruthParticle* >& vRef,
    ITrackMatchingLookup& matches ) const
{
  ATH_MSG_DEBUG( "Doing Track->Truth matching via truthParticleLink" );
 
  for( const xAOD::TrackParticle* track_particle : vTest ) {

    const xAOD::TruthParticle* truth_particle = getLinkedTruth(
                        *track_particle, m_truthProbCut.value() );

    if( not truth_particle ) continue;
 
    ATH_MSG_DEBUG( "Found matched truth particle with pT = " <<
                   pT( *truth_particle ) << " and prob = " <<
                   getTruthMatchProb( *track_particle ) );

    if( std::find( vRef.begin(), vRef.end(),
                   truth_particle ) == vRef.end() ) {
      ATH_MSG_DEBUG( "Truth particle is not in selected reference. Skipping." );
      continue;
    }

    float dist = 1 - getTruthMatchProb( *track_particle );

    ATH_CHECK( matches.update( *track_particle, *truth_particle, dist ) );
  } // loop over vTest
 
  return StatusCode::SUCCESS;
}