JetTrackMatch.h

Go to the documentation of this file.

#ifndef JetReco_JetTrackMatch_h
#define JetReco_JetTrackMatch_h
 
#include "DataFormats/Common/interface/AssociationMap.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
 
namespace reco {
  template <typename JetC>
  class JetTrackMatch {
  public:
    typedef edm::Ref<JetC> JetRef;
    typedef edm::Ref<reco::TrackCollection> TrackRef;
    typedef edm::AssociationMap<edm::OneToMany<JetC, reco::TrackCollection> > Map;
 
  private:
    Map mMap;
 
  public:
    JetTrackMatch() {}
    ~JetTrackMatch() {}
 
    void insert(const JetRef& fJet) { mMap.insert(fJet, TrackRef()); }
 
    void insert(const JetRef& fJet, const TrackRef& fTrack) { mMap.insert(fJet, fTrack); }
 
    std::vector<JetRef> allJets() const {
      std::vector<JetRef> result;
      typename Map::const_iterator it = mMap.begin();
      for (; it != mMap.end(); ++it) {
        result.push_back(it->key);
      }
      return result;
    }
    std::vector<TrackRef> getTracks(const JetRef& mJet) const {
      std::vector<TrackRef> result;
      reco::TrackRefVector tracks = mMap[mJet];
      int i = tracks.size();
      while (--i >= 0) {
        if (!tracks[i].isNull())
          result.push_back(tracks[i]);
      }
      return result;
    }
  };
}  // namespace reco
 
#endif