TrackToTrackingParticleAssociatorBaseImpl.h

Go to the documentation of this file.

#ifndef SimDataFormats_Associations_TrackToTrackingParticleAssociatorBaseImpl_h
#define SimDataFormats_Associations_TrackToTrackingParticleAssociatorBaseImpl_h

#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/RecoCandidate/interface/TrackAssociation.h"

#include "DataFormats/TrackCandidate/interface/TrackCandidateCollection.h"
#include "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"

namespace reco {
  typedef edm::AssociationMap<
      edm::OneToManyWithQualityGeneric<TrackingParticleCollection, edm::View<TrajectorySeed>, double>>
      SimToRecoCollectionSeed;
  typedef edm::AssociationMap<
      edm::OneToManyWithQualityGeneric<edm::View<TrajectorySeed>, TrackingParticleCollection, double>>
      RecoToSimCollectionSeed;

  typedef edm::AssociationMap<
      edm::OneToManyWithQualityGeneric<TrackingParticleCollection, TrackCandidateCollection, double>>
      SimToRecoCollectionTCandidate;
  typedef edm::AssociationMap<
      edm::OneToManyWithQualityGeneric<TrajectorySeedCollection, TrackCandidateCollection, double>>
      RecoToSimCollectionTCandidate;

  class TrackToTrackingParticleAssociatorBaseImpl {
  public:
    TrackToTrackingParticleAssociatorBaseImpl();
    virtual ~TrackToTrackingParticleAssociatorBaseImpl();

    virtual reco::RecoToSimCollection associateRecoToSim(const edm::Handle<edm::View<reco::Track>> &tCH,
                                                         const edm::Handle<TrackingParticleCollection> &tPCH) const;

    virtual reco::SimToRecoCollection associateSimToReco(const edm::Handle<edm::View<reco::Track>> &tCH,
                                                         const edm::Handle<TrackingParticleCollection> &tPCH) const;

    virtual reco::RecoToSimCollection associateRecoToSim(
        const edm::RefToBaseVector<reco::Track> &tc, const edm::RefVector<TrackingParticleCollection> &tpc) const = 0;
    virtual reco::SimToRecoCollection associateSimToReco(
        const edm::RefToBaseVector<reco::Track> &tc, const edm::RefVector<TrackingParticleCollection> &tpc) const = 0;

    // TrajectorySeed
    virtual reco::RecoToSimCollectionSeed associateRecoToSim(const edm::Handle<edm::View<TrajectorySeed>> &,
                                                             const edm::Handle<TrackingParticleCollection> &) const;

    virtual reco::SimToRecoCollectionSeed associateSimToReco(const edm::Handle<edm::View<TrajectorySeed>> &,
                                                             const edm::Handle<TrackingParticleCollection> &) const;

    // TrackCandidate
    virtual reco::RecoToSimCollectionTCandidate associateRecoToSim(
        const edm::Handle<TrackCandidateCollection> &, const edm::Handle<TrackingParticleCollection> &) const;

    virtual reco::SimToRecoCollectionTCandidate associateSimToReco(
        const edm::Handle<TrackCandidateCollection> &, const edm::Handle<TrackingParticleCollection> &) const;
  };
}  // namespace reco

#endif