File indexing completed on 2023-03-17 10:50:11
0001 #ifndef JetReco_JetTrackMatch_h
0002 #define JetReco_JetTrackMatch_h
0003
0004
0005
0006
0007
0008
0009
0010
0011
0012
0013
0014
0015
0016
0017
0018
0019
0020
0021
0022
0023
0024
0025
0026
0027
0028
0029
0030
0031
0032
0033 #include "DataFormats/Common/interface/AssociationMap.h"
0034 #include "DataFormats/TrackReco/interface/Track.h"
0035 #include "DataFormats/TrackReco/interface/TrackFwd.h"
0036
0037 namespace reco {
0038 template <typename JetC>
0039 class JetTrackMatch {
0040 public:
0041 typedef edm::Ref<JetC> JetRef;
0042 typedef edm::Ref<reco::TrackCollection> TrackRef;
0043 typedef edm::AssociationMap<edm::OneToMany<JetC, reco::TrackCollection> > Map;
0044
0045 private:
0046 Map mMap;
0047
0048 public:
0049 JetTrackMatch() {}
0050 ~JetTrackMatch() {}
0051
0052
0053 void insert(const JetRef& fJet) { mMap.insert(fJet, TrackRef()); }
0054
0055
0056 void insert(const JetRef& fJet, const TrackRef& fTrack) { mMap.insert(fJet, fTrack); }
0057
0058
0059 std::vector<JetRef> allJets() const {
0060 std::vector<JetRef> result;
0061 typename Map::const_iterator it = mMap.begin();
0062 for (; it != mMap.end(); ++it) {
0063 result.push_back(it->key);
0064 }
0065 return result;
0066 }
0067
0068 std::vector<TrackRef> getTracks(const JetRef& mJet) const {
0069 std::vector<TrackRef> result;
0070 reco::TrackRefVector tracks = mMap[mJet];
0071 int i = tracks.size();
0072 while (--i >= 0) {
0073 if (!tracks[i].isNull())
0074 result.push_back(tracks[i]);
0075 }
0076 return result;
0077 }
0078 };
0079 }
0080
0081 #endif