/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_6_2_5/src/DataFormats/JetReco/interface/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;
    }
  };
}

#endif