OfflineObjectDecorHelper.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef INDETTRACKPERFMON_OFFLINEOBJECTDECORHELPER_H
#define INDETTRACKPERFMON_OFFLINEOBJECTDECORHELPER_H
 
#include "xAODTracking/TrackParticle.h"
#include "xAODEgamma/ElectronContainer.h"
#include "xAODMuon/MuonContainer.h"
#include "xAODTau/TauJetContainer.h"
#include "xAODTruth/TruthParticleContainer.h"
 
#include <string>
 
 
namespace IDTPM {
 
  template < typename container_t >
  bool isUnlinkedObject( const xAOD::TrackParticle& track,
                         const std::string& decoName ) {
    using elementLink_t = ElementLink< container_t >;
    return ( not track.isAvailable< elementLink_t >( decoName ) );
  }
 
  template < typename container_t >
  typename container_t::const_value_type getLinkedObject(
      const xAOD::TrackParticle& track,
      const std::string& decoName )
  {
    using elementLink_t = ElementLink< container_t >;
 
    if( isUnlinkedObject< container_t >( track, decoName ) ) return nullptr;
 
    elementLink_t eleLink = track.auxdata< elementLink_t >( decoName );
 
    if( not eleLink.isValid() ) return nullptr;
 
    return *eleLink;
  }
 
  const xAOD::Electron* getLinkedElectron( const xAOD::TrackParticle& track,
                                           const std::string& quality="All" );
 
  const xAOD::Muon* getLinkedMuon( const xAOD::TrackParticle& track,
                                   const std::string& quality="All" );
 
  const xAOD::TauJet* getLinkedTau( const xAOD::TrackParticle& track,
                                    const int requiredNtracks,
                                    const std::string& type="RNN",
                                    const std::string& quality="" );
 
  bool isUnlinkedTruth( const xAOD::TrackParticle& track );
 
  float getTruthMatchProb( const xAOD::TrackParticle& track );
 
  const xAOD::TruthParticle* getLinkedTruth( const xAOD::TrackParticle& track,
                                             const float truthProbCut=0. );
 
} // namespace IDTPM
 
#endif // > ! INDETTRACKPERFMON_OFFLINEOBJECTDECORHELPER_H