JetTracksAssociationDR.h

Go to the documentation of this file.

// \class JetTracksAssociationDR
// Associate jets with tracks by simple "delta R" criteria
// Fedor Ratnikov (UMd)

#ifndef RecoJets_JetAssociationAlgorithms_JetTracksAssociationDR_h
#define RecoJets_JetAssociationAlgorithms_JetTracksAssociationDR_h

#include "DataFormats/JetReco/interface/Jet.h"
#include "DataFormats/JetReco/interface/JetTracksAssociation.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"

class MagneticField;
class Propagator;

class JetTracksAssociationDR {
  
 public:

  // ---------- Con(de)structors ----------
  
  explicit JetTracksAssociationDR( double dr_threshold );
  
  virtual ~JetTracksAssociationDR();
  
  // ---------- Typedefs ----------
  
  typedef reco::JetTracksAssociation::Container Association;
  
  typedef edm::Handle< edm::View<reco::Jet> > Jets;
  
  typedef edm::Handle< reco::TrackCollection > Tracks;
  
  // Jet reference
  typedef edm::RefToBase<reco::Jet> JetRef;
  
  // Collection of jet references
  typedef std::vector<JetRef> JetRefs;
  
  // Collection of track references
  typedef std::vector<reco::TrackRef> TrackRefs;

  // Track Quality
  typedef reco::TrackBase::TrackQuality TrackQuality;

  // ---------- Public interface ----------
  
  // Associate tracks to jets
  void associateTracksToJets( Association*,
                  const JetRefs&,
                  const TrackRefs& );
  
  virtual void associateTracksToJet( reco::TrackRefVector&,
                     const reco::Jet&,
                     const TrackRefs& ) = 0;

  // Takes Handle as input and creates collection of edm::Refs
  static void createJetRefs( JetRefs&, 
                 const Jets& );
  
  // Takes Handle as input and creates collection of edm::Refs
  static void createTrackRefs( TrackRefs&,
                   const Tracks&, 
                   const TrackQuality& );
  
 protected:
  
  JetTracksAssociationDR() {}
  
  double mDeltaR2Threshold;
  
};

#endif // RecoJets_JetAssociationAlgorithms_JetTracksAssociationDR_h