OfflineObjectDecorHelper.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "OfflineObjectDecorHelper.h"
 
 
namespace IDTPM {
 
  const xAOD::Electron* getLinkedElectron( const xAOD::TrackParticle& track,
                                           const std::string& quality ) {
    std::string decoName = "LinkedElectron_" + quality;
    return getLinkedObject< xAOD::ElectronContainer >( track, decoName );
  }
 
 
  const xAOD::Muon* getLinkedMuon( const xAOD::TrackParticle& track,
                                   const std::string& quality ) {
    std::string decoName = "LinkedMuon_" + quality;
    return getLinkedObject< xAOD::MuonContainer >( track, decoName );
  }
 
 
  const xAOD::TauJet* getLinkedTau( const xAOD::TrackParticle& track,
                                    const int requiredNtracks,
                                    const std::string& type,
                                    const std::string& quality ) {
    std::string decoName = "LinkedTau" + type +
        std::to_string( requiredNtracks ) + "_" + quality;
    return getLinkedObject< xAOD::TauJetContainer >( track, decoName );
  }
 
 
  bool isUnlinkedTruth( const xAOD::TrackParticle& track ) {
    return isUnlinkedObject< xAOD::TruthParticleContainer >(
        track, "truthParticleLink" );
  }
 
 
  float getTruthMatchProb( const xAOD::TrackParticle& track ) {
    return ( track.isAvailable< float >( "truthMatchProbability" ) ?
        track.auxdata< float >( "truthMatchProbability" ) : -1. );
  }
 
 
  const xAOD::TruthParticle* getLinkedTruth( const xAOD::TrackParticle& track,
                                             const float truthProbCut ) {
    float prob = getTruthMatchProb( track );
    if( prob < truthProbCut ) return nullptr;
 
    return getLinkedObject< xAOD::TruthParticleContainer >(
        track, "truthParticleLink" );
  }
 
} // namespace IDTPM