JetAssociationProducers/src/JetTracksAssociatorAtVertex.h

0001 // \class JetTracksAssociatorAtVertex JetTracksAssociatorAtVertex.cc
0002 //
0003 // Original Author:  Andrea Rizzi
0004 //         Created:  Wed Apr 12 11:12:49 CEST 2006
0005 // Accommodated for Jet Package by: Fedor Ratnikov Jul. 30, 2007
0006 //
0007 //
0008 #ifndef JetTracksAssociatorAtVertex_h
0009 #define JetTracksAssociatorAtVertex_h
0010
0011 #include "FWCore/Framework/interface/stream/EDProducer.h"
0012 #include "DataFormats/Common/interface/EDProductfwd.h"
0013 #include "FWCore/Utilities/interface/InputTag.h"
0014 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0015
0016 #include "RecoJets/JetAssociationAlgorithms/interface/JetTracksAssociationDRVertex.h"
0017 #include "RecoJets/JetAssociationAlgorithms/interface/JetTracksAssociationDRVertexAssigned.h"
0018
0019 class JetTracksAssociatorAtVertex : public edm::stream::EDProducer<> {
0020 public:
0021   JetTracksAssociatorAtVertex(const edm::ParameterSet&);
0022   ~JetTracksAssociatorAtVertex() override;
0023
0024   void produce(edm::Event&, const edm::EventSetup&) override;
0025
0026 private:
0027   edm::EDGetTokenT<edm::View<reco::Jet>> mJets;
0028   edm::EDGetTokenT<reco::TrackCollection> mTracks;
0029
0030   int mTrackQuality;
0031   JetTracksAssociationDRVertex mAssociator;
0032   JetTracksAssociationDRVertexAssigned mAssociatorAssigned;
0033   bool useAssigned;  /// if true, use the track/jet association with vertex assignment to tracks
0034   edm::EDGetTokenT<reco::VertexCollection> pvSrc;  /// if useAssigned, will read this PV collection.
0035 };
0036
0037 #endif