PatUtils/interface/SimpleJetTrackAssociator.h

0001 #include "DataFormats/TrackReco/interface/TrackFwd.h"
0002 #include "DataFormats/Math/interface/Vector3D.h"
0003 #include "DataFormats/Common/interface/View.h"
0004
0005 namespace helper {
0006   class SimpleJetTrackAssociator {
0007   public:
0008     SimpleJetTrackAssociator() : deltaR2_(0), nHits_(0), chi2nMax_(0) {}
0009     SimpleJetTrackAssociator(double deltaR, int32_t nHits, double chi2nMax)
0010         : deltaR2_(deltaR * deltaR), nHits_(nHits), chi2nMax_(chi2nMax) {}
0011
0012     // 100% FWLite compatible (but will make up transient refs)
0013     void associateTransient(const math::XYZVector &dir, const reco::TrackCollection &in, reco::TrackRefVector &out);
0014
0015     // more versatile, persistent refs, for when we're in full framework
0016     void associate(const math::XYZVector &dir, const edm::View<reco::Track> &in, reco::TrackRefVector &out);
0017
0018   private:
0019     double deltaR2_;
0020     int32_t nHits_;
0021     double chi2nMax_;
0022   };
0023 }  // namespace helper