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();
    QuickTrackAssociatorByHits( const QuickTrackAssociatorByHits& otherAssociator );
    QuickTrackAssociatorByHits& operator=( const QuickTrackAssociatorByHits& otherAssociator );
        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;

        void prepareCluster2TPMap(const edm::Event* pEvent) const;

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()); }

    reco::RecoToSimCollection associateRecoToSimImplementation() const;

    reco::SimToRecoCollection associateSimToRecoImplementation() const;

    template<typename iter> std::vector< std::pair<edm::Ref<TrackingParticleCollection>,size_t> > associateTrack( iter begin, iter end ) const;
    template<typename iter> std::vector< std::pair<edm::Ref<TrackingParticleCollection>,size_t> > associateTrackByCluster( iter begin, iter end ) const;


    bool trackingParticleContainsIdentifier( const TrackingParticle* pTrackingParticle, const SimTrackIdentifiers& identifier ) const;

    template<typename iter> int getDoubleCount( iter begin, iter end, TrackingParticleRef associatedTrackingParticle ) const;

    template<typename iter> std::vector< std::pair<SimTrackIdentifiers,size_t> > getAllSimTrackIdentifiers( 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);
    }

    //
    // Members. Note that there are custom copy constructor and assignment operators, so if any members are added
    // those methods will need to be updated.
    //
    mutable TrackerHitAssociator* pHitAssociator_;
    const mutable edm::Event* pEventForWhichAssociatorIsValid_;
    void initialiseHitAssociator( const edm::Event* event ) const;

    edm::ParameterSet hitAssociatorParameters_;

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

    mutable edm::Handle<edm::View<reco::Track> >* pTrackCollectionHandle_;

    mutable const edm::RefToBaseVector<reco::Track>* pTrackCollection_;

    mutable edm::Handle<TrackingParticleCollection>* pTrackingParticleCollectionHandle_;

    mutable const edm::RefVector<TrackingParticleCollection>* pTrackingParticleCollection_;

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

#endif // end of ifndef QuickTrackAssociatorByHits_h