TrackToGenParticleAssociator.h

Go to the documentation of this file.

#ifndef SimDataFormats_Associations_TrackToGenParticleAssociator_h
#define SimDataFormats_Associations_TrackToGenParticleAssociator_h

#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
#include "DataFormats/HepMCCandidate/interface/GenParticleFwd.h"
#include "DataFormats/Math/interface/LorentzVector.h"
#include "DataFormats/RecoCandidate/interface/TrackAssociation.h"
#include "SimDataFormats/Track/interface/SimTrackContainer.h"
#include "SimDataFormats/Vertex/interface/SimVertexContainer.h"

#include "SimDataFormats/Associations/interface/TrackToGenParticleAssociatorBaseImpl.h"

#include <memory>

// Note that the Association Map is filled with -ch2 and not chi2 because it is
// ordered using std::greater: the track with the lowest association chi2 will
// be the first in the output map.

namespace reco {
  typedef edm::AssociationMap<
      edm::OneToManyWithQualityGeneric<reco::GenParticleCollection, edm::View<reco::Track>, double>>
      GenToRecoCollection;
  typedef edm::AssociationMap<
      edm::OneToManyWithQualityGeneric<edm::View<reco::Track>, reco::GenParticleCollection, double>>
      RecoToGenCollection;

  class TrackToGenParticleAssociator {
  public:
    TrackToGenParticleAssociator() = default;
#ifndef __GCCXML__
    TrackToGenParticleAssociator(std::unique_ptr<reco::TrackToGenParticleAssociatorBaseImpl>);
#endif
    ~TrackToGenParticleAssociator() = default;
    TrackToGenParticleAssociator(TrackToGenParticleAssociator &&) = default;
    TrackToGenParticleAssociator &operator=(TrackToGenParticleAssociator &&) = default;

    reco::RecoToGenCollection associateRecoToGen(const edm::RefToBaseVector<reco::Track> &tracks,
                                                 const edm::RefVector<reco::GenParticleCollection> &gens) const {
      return m_impl->associateRecoToGen(tracks, gens);
    }
    reco::GenToRecoCollection associateGenToReco(const edm::RefToBaseVector<reco::Track> &tracks,
                                                 const edm::RefVector<reco::GenParticleCollection> &gens) const {
      return m_impl->associateGenToReco(tracks, gens);
    }

    reco::RecoToGenCollection associateRecoToGen(const edm::Handle<edm::View<reco::Track>> &tCH,
                                                 const edm::Handle<reco::GenParticleCollection> &tPCH) const {
      return m_impl->associateRecoToGen(tCH, tPCH);
    }

    reco::GenToRecoCollection associateGenToReco(const edm::Handle<edm::View<reco::Track>> &tCH,
                                                 const edm::Handle<reco::GenParticleCollection> &tPCH) const {
      return m_impl->associateGenToReco(tCH, tPCH);
    }

  private:
    TrackToGenParticleAssociator(const TrackToGenParticleAssociator &) = delete;  // stop default

    const TrackToGenParticleAssociator &operator=(const TrackToGenParticleAssociator &) = delete;  // stop default

    std::unique_ptr<TrackToGenParticleAssociatorBaseImpl> m_impl;
  };
}  // namespace reco

#endif