CachedGetAssocTruth.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef IDPVM_CachedGetAssocTruth_h
#define IDPVM_CachedGetAssocTruth_h

 
#include <unordered_map>
#include <string>
#include "xAODTruth/TruthParticle.h"
#include "xAODTracking/TrackParticle.h"

namespace IDPVM {
  class CachedGetAssocTruth {
  public:
    CachedGetAssocTruth() = default;
    CachedGetAssocTruth& operator = (CachedGetAssocTruth&&) = default;
    CachedGetAssocTruth(CachedGetAssocTruth&&) = default;
    ~CachedGetAssocTruth() = default;
    CachedGetAssocTruth& operator = (const CachedGetAssocTruth&) = delete;
    CachedGetAssocTruth(const CachedGetAssocTruth&) = delete;
    void clear();
    const xAOD::TruthParticle* operator () (const xAOD::TrackParticle* trackParticle);
    const xAOD::TruthParticle* getTruth(const xAOD::TrackParticle* const trackParticle);
 
    static void neededTrackParticleDecorations(std::vector<std::string> &decorations);
  private:
    std::unordered_map<const xAOD::TrackParticle*, const xAOD::TruthParticle*> m_cache;
    static const std::string s_trackParticleLinkDecorationName;
  };
}// end of namespace
#endif