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