TruthTrackMatchingTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "TruthTrackMatchingTool.h"
#include "TrackAnalysisCollections.h"
#include "TrackMatchingLookup.h"
#include "OfflineObjectDecorHelper.h"
#include "TrackParametersHelper.h"
 
IDTPM::TruthTrackMatchingTool::TruthTrackMatchingTool(
    const std::string& name ) :
        asg::AsgTool( name ) { }
 
 
StatusCode IDTPM::TruthTrackMatchingTool::initialize()
{
  ATH_CHECK( asg::AsgTool::initialize() );
 
  ATH_MSG_DEBUG( "Initializing " << name() << "..." );
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode IDTPM::TruthTrackMatchingTool::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.testTruthVec( TrackAnalysisCollections::InRoI ),
      trkAnaColls.refTrackVec( TrackAnalysisCollections::InRoI ),
      trkAnaColls.matches() ) );
 
  ATH_MSG_DEBUG( trkAnaColls.printMatchInfo() );
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode IDTPM::TruthTrackMatchingTool::match(
    const std::vector< const xAOD::TruthParticle* >& vTest,
    const std::vector< const xAOD::TrackParticle* >& vRef,
    ITrackMatchingLookup& matches ) const
{
  ATH_MSG_DEBUG( "Doing Truth->Track matching via truthParticleLink" );
 
  for( const xAOD::TruthParticle* truth_particle : vTest ) {
 
    const xAOD::TrackParticle* matched_track_particle = nullptr;
    float prob = -99.;
 
    for( const xAOD::TrackParticle* track_particle : vRef ) {
 
      const xAOD::TruthParticle* truth_particle_tmp =
          getLinkedTruth( *track_particle, m_truthProbCut.value() );
 
      if( ( not truth_particle_tmp ) or
          ( truth_particle_tmp != truth_particle ) ) continue;
 
      float prob_tmp = getTruthMatchProb( *track_particle );
 
      if( prob_tmp > prob ) {
        matched_track_particle = track_particle;
        prob = prob_tmp;
      }
 
    } // loop over vRef
 
    if( not matched_track_particle ) continue;
 
    ATH_MSG_DEBUG( "Found matched track particle with pT = " <<
                   pT( *matched_track_particle ) <<
                   " and prob = " << prob );
 
    float dist = 1 - prob;
 
    ATH_CHECK( matches.update( *truth_particle, *matched_track_particle, dist ) );
 
  } // loop over vTest
 
  return StatusCode::SUCCESS;
}