TrackTruthLookup.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef INDETPHYSVALMONITORING_TRACKTRUTHLOOKUP_H
#define INDETPHYSVALMONITORING_TRACKTRUTHLOOKUP_H 1
 
// STL include(s)
#include <unordered_map> /* std::unordered_map */
#include <vector> /* std::vector */
#include <stdexcept> /* std::out_of_range */
 
// xAOD include(s)
#include "xAODTracking/TrackParticle.h"
#include "xAODTracking/TrackParticleContainer.h"
#include "xAODTruth/TruthParticle.h"
#include "xAODTruth/TruthParticleContainer.h"
 
namespace IDPVM {
  class TrackTruthLookup {
  public:
    TrackTruthLookup() {
    }
 
    TrackTruthLookup(const xAOD::TrackParticleContainer* trackParticles,
                     const xAOD::TruthParticleContainer* truthParticles) {
      cache(trackParticles, truthParticles);
    }
 
    TrackTruthLookup(const xAOD::TrackParticleContainer* trackParticles,
                     const std::vector<const xAOD::TruthParticle*>* truthParticlesVec) {
      cache(trackParticles, truthParticlesVec);
    }
 
    ~TrackTruthLookup() {
    }
 
    // @{
    void cache(const xAOD::TrackParticleContainer* trackParticles,
               const std::vector<const xAOD::TruthParticle*>* truthParticlesVec);
 
    void cache(const xAOD::TrackParticleContainer* trackParticles,
               const xAOD::TruthParticleContainer* truthParticles);
    // @}
 
    std::vector<const xAOD::TrackParticle*> getTracks(const xAOD::TruthParticle* truth) const;
 
    const xAOD::TruthParticle* getTruth(const xAOD::TrackParticle* track) const;
 
    inline bool
    contains(const xAOD::TruthParticle* truth) const {
      return m_mapTruth.find(truth) != m_mapTruth.end();
    }
 
    inline bool
    contains(const xAOD::TrackParticle* track) const {
      return m_mapTrack.find(track) != m_mapTrack.end();
    }
 
    inline void
    clear() {
      m_mapTrack.clear();
      m_mapTruth.clear();
      return;
    }
 
  private:
    void cacheTracks(const xAOD::TrackParticleContainer* trackParticles);
    void cacheTruth(const xAOD::TruthParticleContainer* truthParticles);
    void cacheTruth(const std::vector<const xAOD::TruthParticle*>* truthParticlesVec);
    // Mapping of xAOD::TrackParticle to their unique associated
    // xAOD::TruthParticle. nullptr if none exist.
    std::unordered_map<const xAOD::TrackParticle*,
                       const xAOD::TruthParticle*> m_mapTrack;
 
    // Mapping of xAOD::TruthParticle to vector of their (possibly multiple
    // associated xAOD::TrackParticles. Empty if none exist.
    std::unordered_map<const xAOD::TruthParticle*,
                       std::vector<const xAOD::TrackParticle*> > m_mapTruth;
  };
} // namespace
 
#endif // > !INDETPHYSVALMONITORING_TRACKTRUTHLOOKUP_H