TrackAssociatorByHitsImpl.h

Go to the documentation of this file.

#ifndef SimTracker_TrackAssociatorProducers_TrackAssociatorByHitsImpl_h
#define SimTracker_TrackAssociatorProducers_TrackAssociatorByHitsImpl_h

#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/Common/interface/Ref.h"
#include "SimTracker/TrackerHitAssociation/interface/TrackerHitAssociator.h"

//reco track
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackingRecHit/interface/TrackingRecHit.h"
//TrackingParticle
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticle.h"
#include "SimDataFormats/EncodedEventId/interface/EncodedEventId.h"
#include "SimDataFormats/Associations/interface/TrackToTrackingParticleAssociatorBaseImpl.h"

class TrackerTopology;

namespace edm {
  class EDProductGetter;
}

class TrackAssociatorByHitsImpl : public reco::TrackToTrackingParticleAssociatorBaseImpl {
public:
  enum SimToRecoDenomType { denomnone, denomsim, denomreco };

  typedef std::pair<TrackingParticleRef, TrackPSimHitRef> SimHitTPPair;
  typedef std::vector<SimHitTPPair> SimHitTPAssociationList;

  TrackAssociatorByHitsImpl(edm::EDProductGetter const& productGetter,
                            std::unique_ptr<TrackerHitAssociator> iAssociate,
                            TrackerTopology const* iTopo,
                            SimHitTPAssociationList const* iSimHitsTPAssoc,
                            SimToRecoDenomType iSimToRecoDenominator,
                            double iQuality_SimToReco,
                            double iPurity_SimToReco,
                            double iCut_RecoToSim,
                            bool iUsePixels,
                            bool iUseGrouped,
                            bool iUseSplitting,
                            bool ThreeHitTracksAreSpecial,
                            bool AbsoluteNumberOfHits);

  /* Associate SimTracks to RecoTracks By Hits */


  reco::RecoToSimCollection associateRecoToSim(const edm::RefToBaseVector<reco::Track>&,
                                               const edm::RefVector<TrackingParticleCollection>&) const override;

  reco::SimToRecoCollection associateSimToReco(const edm::RefToBaseVector<reco::Track>&,
                                               const edm::RefVector<TrackingParticleCollection>&) const override;


  reco::RecoToSimCollection associateRecoToSim(const edm::Handle<edm::View<reco::Track> >& tCH,
                                               const edm::Handle<TrackingParticleCollection>& tPCH) const override {
    return TrackToTrackingParticleAssociatorBaseImpl::associateRecoToSim(tCH, tPCH);
  }


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

  //seed

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

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

private:
  template <typename iter>
  void getMatchedIds(std::vector<SimHitIdpr>&, std::vector<SimHitIdpr>&, int&, iter, iter, TrackerHitAssociator*) const;

  int getShared(std::vector<SimHitIdpr>&, std::vector<SimHitIdpr>&, TrackingParticle const&) const;

  template <typename iter>
  int getDoubleCount(iter, iter, TrackerHitAssociator*, TrackingParticle const&) const;

  // ----- member data
  edm::EDProductGetter const* productGetter_;
  std::unique_ptr<TrackerHitAssociator> associate;
  TrackerTopology const* tTopo;
  SimHitTPAssociationList const* simHitsTPAssoc;

  SimToRecoDenomType SimToRecoDenominator;
  const double quality_SimToReco;
  const double purity_SimToReco;
  const double cut_RecoToSim;
  const bool UsePixels;
  const bool UseGrouped;
  const bool UseSplitting;
  const bool ThreeHitTracksAreSpecial;
  const bool AbsoluteNumberOfHits;

  const TrackingRecHit* getHitPtr(edm::OwnVector<TrackingRecHit>::const_iterator iter) const { return &*iter; }
  const TrackingRecHit* getHitPtr(trackingRecHit_iterator iter) const { return &**iter; }

  //edm::InputTag _simHitTpMapTag;
};

#endif