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 >;
    const SG::ConstAccessor< elementLink_t > acc( decoName );
    return ( not acc.isAvailable( track ) );
  }
 
  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;
 
    const SG::ConstAccessor< elementLink_t > acc( decoName );
    elementLink_t eleLink = acc( track );
 
    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 );
  inline bool isUnlinkedTruth( const xAOD::TruthParticle& ) { return false; }; // dummy - to avoid compilation errors
 
  float getTruthMatchProb( const xAOD::TrackParticle& track );
  inline float getTruthMatchProb( const xAOD::TruthParticle& ) { return -1.; }; // dummy - to avoid compilation errors
 
  const xAOD::TruthParticle* getLinkedTruth( const xAOD::TrackParticle& track,
                                             const float truthProbCut=0. );
  inline const xAOD::TruthParticle* getLinkedTruth(
      const xAOD::TruthParticle&, const float ) { return nullptr; }; // dummy - to avoid compilation errors
 
  bool isFakeTruth( const xAOD::TrackParticle& track, const float truthProbCut=0., const bool unlinkedAsFakes=false );
  inline bool isFakeTruth( const xAOD::TruthParticle&, const float, const bool ) { return false; }; // dummy - to avoid compilation errors
 
  bool isReconstructable( const xAOD::TruthParticle& truth, const std::vector<unsigned int>& minSilHits, const std::vector<float>& etaBins);
  inline bool isReconstructable( const xAOD::TrackParticle&, const std::vector<unsigned int>& , const std::vector<float>& ) { return false; }; // dummy - to avoid compilation errors
 
} // namespace IDTPM
 
#endif // > ! INDETTRACKPERFMON_OFFLINEOBJECTDECORHELPER_H