QuickTrackAssociatorByHitsImpl.h

Go to the documentation of this file.

#ifndef QuickTrackAssociatorByHitsImpl_h
#define QuickTrackAssociatorByHitsImpl_h

#include "DataFormats/TrackerRecHit2D/interface/OmniClusterRef.h"

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

// Forward declarations
class TrackerHitAssociator;

  inline bool clusterTPAssociationListGreater(std::pair<OmniClusterRef, TrackingParticleRef> i,std::pair<OmniClusterRef, TrackingParticleRef> j) { return (i.first.rawIndex()>j.first.rawIndex()); }

class QuickTrackAssociatorByHitsImpl : public reco::TrackToTrackingParticleAssociatorBaseImpl
{
public:
  typedef std::vector<std::pair<OmniClusterRef, TrackingParticleRef> > ClusterTPAssociationList;
  enum SimToRecoDenomType {denomnone,denomsim,denomreco};

  QuickTrackAssociatorByHitsImpl(std::shared_ptr<const TrackerHitAssociator> hitAssoc,
                                 std::shared_ptr<const ClusterTPAssociationList> clusterToTPMap,
                                 bool absoluteNumberOfHits,
                                 double qualitySimToReco,
                                 double puritySimToReco,
                                 double cutRecoToSim,
                                 bool threeHitTracksAreSpecial,
                                 SimToRecoDenomType simToRecoDenominator);
  
  virtual
    reco::RecoToSimCollection associateRecoToSim( const edm::Handle<edm::View<reco::Track> >& trackCollectionHandle,
                                                  const edm::Handle<TrackingParticleCollection>& trackingParticleCollectionHandle) const override;
  virtual
    reco::SimToRecoCollection associateSimToReco( const edm::Handle<edm::View<reco::Track> >& trackCollectionHandle,
                                                  const edm::Handle<TrackingParticleCollection>& trackingParticleCollectionHandle) const override;
  virtual
    reco::RecoToSimCollection associateRecoToSim( const edm::RefToBaseVector<reco::Track>& trackCollection,
                                                  const edm::RefVector<TrackingParticleCollection>& trackingParticleCollection) const override;
  
  virtual
    reco::SimToRecoCollection associateSimToReco( const edm::RefToBaseVector<reco::Track>& trackCollection,
                                                  const edm::RefVector<TrackingParticleCollection>& trackingParticleCollection) const override;
  
  //seed
  virtual
    reco::RecoToSimCollectionSeed associateRecoToSim(const edm::Handle<edm::View<TrajectorySeed> >&,
                                                     const edm::Handle<TrackingParticleCollection>&) const override;
  
  virtual
    reco::SimToRecoCollectionSeed associateSimToReco(const edm::Handle<edm::View<TrajectorySeed> >&,
                                                     const edm::Handle<TrackingParticleCollection>&) const override;
  
  
 private:
  typedef std::pair<uint32_t,EncodedEventId> SimTrackIdentifiers; 
  
  // - added by S. Sarkar
  static bool tpIntPairGreater(std::pair<edm::Ref<TrackingParticleCollection>,size_t> i, std::pair<edm::Ref<TrackingParticleCollection>,size_t> j) { return (i.first.key()>j.first.key()); }
  
  template<class T_TrackCollection, class T_TrackingParticleCollection, class T_hitOrClusterAssociator>
    reco::RecoToSimCollection associateRecoToSimImplementation( T_TrackCollection trackCollection, T_TrackingParticleCollection trackingParticleCollection, T_hitOrClusterAssociator hitOrClusterAssociator ) const;
  
  template<class T_TrackCollection, class T_TrackingParticleCollection, class T_hitOrClusterAssociator>
    reco::SimToRecoCollection associateSimToRecoImplementation( T_TrackCollection trackCollection, T_TrackingParticleCollection trackingParticleCollection, T_hitOrClusterAssociator hitOrClusterAssociator ) const;
  
  
  template<typename T_TPCollection,typename iter> std::vector< std::pair<edm::Ref<TrackingParticleCollection>,size_t> > associateTrack( const TrackerHitAssociator& hitAssociator, T_TPCollection trackingParticles, iter begin, iter end ) const;
  template<typename T_TPCollection,typename iter> std::vector< std::pair<edm::Ref<TrackingParticleCollection>,size_t> > associateTrack( const ClusterTPAssociationList& clusterToTPMap, T_TPCollection trackingParticles, iter begin, iter end ) const;
  
  
  bool trackingParticleContainsIdentifier( const TrackingParticle* pTrackingParticle, const SimTrackIdentifiers& identifier ) const;
  
  template<typename iter> int getDoubleCount( const TrackerHitAssociator& hitAssociator, iter begin, iter end, TrackingParticleRef associatedTrackingParticle ) const;
  template<typename iter> int getDoubleCount( const ClusterTPAssociationList& clusterToTPList, iter begin, iter end, TrackingParticleRef associatedTrackingParticle ) const;
  
  template<typename iter> std::vector< std::pair<SimTrackIdentifiers,size_t> > getAllSimTrackIdentifiers( const TrackerHitAssociator& hitAssociator, iter begin, iter end ) const;
  
  // Added by S. Sarkar
  template<typename iter> std::vector< OmniClusterRef> getMatchedClusters( iter begin, iter end ) const;
  
  const TrackingRecHit* getHitFromIter(trackingRecHit_iterator iter) const {
    return &(**iter);
  }
  
  const TrackingRecHit* getHitFromIter(TrackingRecHitCollection::const_iterator iter) const {
    return &(*iter);
  }
  
  //void prepareEitherHitAssociatorOrClusterToTPMap( const edm::Event* pEvent, std::unique_ptr<ClusterTPAssociationList>& pClusterToTPMap, std::unique_ptr<TrackerHitAssociator>& pHitAssociator ) const;
  
  std::shared_ptr<const TrackerHitAssociator> hitAssociator_;
  std::shared_ptr<const ClusterTPAssociationList> clusterToTPMap_;
  
  double qualitySimToReco_;
  double puritySimToReco_;
  double cutRecoToSim_;
  SimToRecoDenomType simToRecoDenominator_;
  bool threeHitTracksAreSpecial_;
  bool absoluteNumberOfHits_;
  
  // Added by S. Sarkar
  //bool useClusterTPAssociation_;
}; // end of the QuickTrackAssociatorByHitsImpl class

#endif // end of ifndef QuickTrackAssociatorByHitsImpl_h