QuickTrackAssociatorByHits.h

Go to the documentation of this file.

#ifndef QuickTrackAssociatorByHits_h
#define QuickTrackAssociatorByHits_h

#include "SimTracker/TrackAssociation/interface/TrackAssociatorBase.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/TrackerRecHit2D/interface/OmniClusterRef.h"

#include "SimTracker/TrackerHitAssociation/interface/ClusterTPAssociationProducer.h"
//#include "SimTracker/TrackerHitAssociation/plugins/Cluster2TPMapProducer.h"
//#include "SimTracker/TrackerHitAssociation/plugins/TP2ClusterMapProducer.h"

// Forward declarations
class TrackerHitAssociator;

class QuickTrackAssociatorByHits : public TrackAssociatorBase
{
public:
    QuickTrackAssociatorByHits( const edm::ParameterSet& config );
    ~QuickTrackAssociatorByHits();
    virtual
    reco::RecoToSimCollection associateRecoToSim( edm::Handle<edm::View<reco::Track> >& trackCollectionHandle,
                                                  edm::Handle<TrackingParticleCollection>& trackingParticleCollectionHandle,
                                                  const edm::Event* pEvent=0,
                                                  const edm::EventSetup* pSetup=0 ) const override;
    virtual
    reco::SimToRecoCollection associateSimToReco( edm::Handle<edm::View<reco::Track> >& trackCollectionHandle,
                                                  edm::Handle<TrackingParticleCollection>& trackingParticleCollectionHandle,
                                                  const edm::Event* pEvent=0,
                                                  const edm::EventSetup* pSetup=0 ) const override;
    virtual
    reco::RecoToSimCollection associateRecoToSim( const edm::RefToBaseVector<reco::Track>& trackCollection,
                                                  const edm::RefVector<TrackingParticleCollection>& trackingParticleCollection,
                                                  const edm::Event* pEvent=0,
                                                  const edm::EventSetup* pSetup=0 ) const override;
    virtual
    reco::SimToRecoCollection associateSimToReco( const edm::RefToBaseVector<reco::Track>& trackCollection,
                                                  const edm::RefVector<TrackingParticleCollection>& trackingParticleCollection,
                                                  const edm::Event* pEvent=0,
                                                  const edm::EventSetup* pSetup=0 ) const override;

    //seed
    virtual
    reco::RecoToSimCollectionSeed associateRecoToSim(edm::Handle<edm::View<TrajectorySeed> >&,
                             edm::Handle<TrackingParticleCollection>&,
                             const edm::Event * event ,
                             const edm::EventSetup * setup ) const override;
    
    virtual
    reco::SimToRecoCollectionSeed associateSimToReco(edm::Handle<edm::View<TrajectorySeed> >&,
                             edm::Handle<TrackingParticleCollection>&,
                             const edm::Event * event ,
                             const edm::EventSetup * setup ) const override;


private:
    typedef std::pair<uint32_t,EncodedEventId> SimTrackIdentifiers; 
    enum SimToRecoDenomType {denomnone,denomsim,denomreco};

    // - added by S. Sarkar
    typedef std::vector<std::pair<OmniClusterRef, TrackingParticleRef> > ClusterTPAssociationList;
    static bool clusterTPAssociationListGreater(std::pair<OmniClusterRef, TrackingParticleRef> i,std::pair<OmniClusterRef, TrackingParticleRef> j) { return (i.first.rawIndex()>j.first.rawIndex()); }
    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;

    edm::ParameterSet hitAssociatorParameters_;

    bool absoluteNumberOfHits_;
    double qualitySimToReco_;
    double puritySimToReco_;
    double cutRecoToSim_;
    bool threeHitTracksAreSpecial_;
    SimToRecoDenomType simToRecoDenominator_;

    // Added by S. Sarkar
    bool useClusterTPAssociation_;
    edm::InputTag cluster2TPSrc_;
}; // end of the QuickTrackAssociatorByHits class

#endif // end of ifndef QuickTrackAssociatorByHits_h