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"

// 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;
private:
    typedef std::pair<uint32_t,EncodedEventId> SimTrackIdentifiers;
    enum SimToRecoDenomType {denomnone,denomsim,denomreco};

    reco::RecoToSimCollection associateRecoToSimImplementation() const;

    reco::SimToRecoCollection associateSimToRecoImplementation() const;

    std::vector< std::pair<edm::Ref<TrackingParticleCollection>,size_t> > associateTrack( const reco::Track* pTrack ) const;

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

    int getDoubleCount( trackingRecHit_iterator begin, trackingRecHit_iterator end, const TrackingParticle& associatedTrackingParticle ) const;

    std::vector< std::pair<SimTrackIdentifiers,size_t> > getAllSimTrackIdentifiers( const reco::Track* pTrack ) const;

    //
    // 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_;
}; // end of the QuickTrackAssociatorByHits class

#endif // end of ifndef QuickTrackAssociatorByHits_h