---

JetTrackAssociator.h

Go to the documentation of this file.

#ifndef JetProducers_JetTrackAssociator_h
#define JetProducers_JetTrackAssociator_h


#include "FWCore/Utilities/interface/Exception.h"
#include "DataFormats/JetReco/interface/JetTrackMatch.h"


template <typename JetC>
class JetTrackAssociator {
 public:
  typedef edm::Ref<JetC> JetRef;
  typedef reco::TrackCollection TrackC;
  typedef edm::Ref<TrackC> TrackRef;
  typedef reco::JetTrackMatch<JetC> MatchMap;
  JetTrackAssociator () {}
  virtual ~JetTrackAssociator () {}

  virtual bool associate (const JetRef& fJet, const TrackRef& fTrack) const = 0;

  virtual bool goodTrack (const TrackRef& fTrack) const {return true;}

  void buildMap (const edm::Handle<JetC>& fJets, const edm::Handle<TrackC>& fTracks, MatchMap* fMap) const {
    if (!fMap) {
      throw cms::Exception ("JetTrackAssociator") << " Invalid supplied matching map pointer" << std::endl;
    }
    for (unsigned iJet = 0; iJet < fJets->size(); ++iJet) {
      typename MatchMap::JetRef jetRef (fJets, iJet);  
      bool orphanJet = true;
      for (unsigned iTrack = 0; iTrack < fTracks->size(); ++iTrack) {
        typename MatchMap::TrackRef trackRef (fTracks, iTrack); 
        if (goodTrack (trackRef) && associate (jetRef, trackRef)) {
          fMap->insert (jetRef, trackRef);
          orphanJet = false;
        }
      }
      if (orphanJet) fMap->insert (jetRef);
    }
  }
};

#endif

---