AdaptiveVertexFinder/interface/TTHelpers.h

0001 #ifndef TTHelper_s
0002 #define TTHelper_s
0003
0004 #include "DataFormats/Candidate/interface/Candidate.h"
0005
0006 #include "TrackingTools/TransientTrack/interface/TransientTrack.h"
0007 #include "TrackingTools/TransientTrack/interface/TransientTrackBuilder.h"
0008
0009 namespace tthelpers {
0010   inline reco::TransientTrack buildTT(edm::Handle<reco::TrackCollection> &tracks,
0011                                       edm::ESHandle<TransientTrackBuilder> &trackbuilder,
0012                                       unsigned int k) {
0013     reco::TrackRef ref(tracks, k);
0014     return trackbuilder->build(ref);
0015   }
0016   inline reco::TransientTrack buildTT(edm::Handle<edm::View<reco::Candidate> > &tracks,
0017                                       edm::ESHandle<TransientTrackBuilder> &trackbuilder,
0018                                       unsigned int k) {
0019     if ((*tracks)[k].bestTrack() == nullptr)
0020       return reco::TransientTrack();
0021     return trackbuilder->build(tracks->ptrAt(k));
0022   }
0023 }  // namespace tthelpers
0024 #endif