/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_6_2_5/src/SimTracker/TrackAssociation/interface/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 );
        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;
        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;
        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;
        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;

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

        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