TrackToTrackingParticleAssociator.h

Go to the documentation of this file.

#ifndef SimDataFormats_Associations_TrackToTrackingParticleAssociator_h
#define SimDataFormats_Associations_TrackToTrackingParticleAssociator_h
// -*- C++ -*-
//
// Package:     SimDataFormats/Associations
// Class  :     TrackToTrackingParticleAssociator
//
//
// Original Author:  Christopher Jones
//         Created:  Tue, 30 Dec 2014 20:47:00 GMT
//

// system include files
#include <memory>

// user include files
#include "DataFormats/RecoCandidate/interface/TrackAssociation.h"

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

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

// forward declarations

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 TrackToTrackingParticleAssociator {
  public:
#ifndef __GCCXML__
    TrackToTrackingParticleAssociator(std::unique_ptr<reco::TrackToTrackingParticleAssociatorBaseImpl>);
#endif
    TrackToTrackingParticleAssociator() = default;
    TrackToTrackingParticleAssociator(TrackToTrackingParticleAssociator &&) = default;
    TrackToTrackingParticleAssociator &operator=(TrackToTrackingParticleAssociator &&) = default;
    ~TrackToTrackingParticleAssociator() = default;

    // ---------- const member functions ---------------------
    reco::RecoToSimCollection associateRecoToSim(const edm::Handle<edm::View<reco::Track>> &tCH,
                                                 const edm::Handle<TrackingParticleCollection> &tPCH) const {
      return m_impl->associateRecoToSim(tCH, tPCH);
    }

    reco::SimToRecoCollection associateSimToReco(const edm::Handle<edm::View<reco::Track>> &tCH,
                                                 const edm::Handle<TrackingParticleCollection> &tPCH) const {
      return m_impl->associateSimToReco(tCH, tPCH);
    }

    reco::RecoToSimCollection associateRecoToSim(const edm::RefToBaseVector<reco::Track> &tc,
                                                 const edm::RefVector<TrackingParticleCollection> &tpc) const {
      return m_impl->associateRecoToSim(tc, tpc);
    }

    reco::SimToRecoCollection associateSimToReco(const edm::RefToBaseVector<reco::Track> &tc,
                                                 const edm::RefVector<TrackingParticleCollection> &tpc) const {
      return m_impl->associateSimToReco(tc, tpc);
    }

    // TrajectorySeed
    reco::RecoToSimCollectionSeed associateRecoToSim(const edm::Handle<edm::View<TrajectorySeed>> &ts,
                                                     const edm::Handle<TrackingParticleCollection> &tpc) const {
      return m_impl->associateRecoToSim(ts, tpc);
    }

    reco::SimToRecoCollectionSeed associateSimToReco(const edm::Handle<edm::View<TrajectorySeed>> &ts,
                                                     const edm::Handle<TrackingParticleCollection> &tpc) const {
      return m_impl->associateSimToReco(ts, tpc);
    }

    // TrackCandidate
    reco::RecoToSimCollectionTCandidate associateRecoToSim(const edm::Handle<TrackCandidateCollection> &tc,
                                                           const edm::Handle<TrackingParticleCollection> &tpc) const {
      return m_impl->associateRecoToSim(tc, tpc);
    }

    reco::SimToRecoCollectionTCandidate associateSimToReco(const edm::Handle<TrackCandidateCollection> &tc,
                                                           const edm::Handle<TrackingParticleCollection> &tpc) const {
      return m_impl->associateSimToReco(tc, tpc);
    }

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

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

    // ---------- member data --------------------------------
    std::unique_ptr<TrackToTrackingParticleAssociatorBaseImpl> m_impl;
  };
}  // namespace reco

#endif