JetVertexAssociation/interface/JetVertexAssociation.h

0001 #ifndef JetVertexAssociation_h
0002 #define JetVertexAssociation_h
0003
0004 #include "FWCore/Framework/interface/global/EDProducer.h"
0005 #include "DataFormats/JetReco/interface/CaloJetCollection.h"
0006 #include "DataFormats/TrackReco/interface/Track.h"
0007 #include "DataFormats/VertexReco/interface/VertexFwd.h"
0008
0009 #include "JetMETCorrections/JetVertexAssociation/interface/JetVertexMain.h"
0010
0011 #include <memory>
0012 #include <string>
0013 #include <iostream>
0014 #include <iomanip>
0015 #include <cmath>
0016 #include <vector>
0017
0018 namespace edm {
0019   class ParameterSet;
0020   class Event;
0021   class EventSetup;
0022 }  // namespace edm
0023
0024 namespace cms {
0025
0026   class JetVertexAssociation : public edm::global::EDProducer<> {
0027   public:
0028     JetVertexAssociation(const edm::ParameterSet& ps);
0029
0030     void produce(edm::StreamID, edm::Event& e, const edm::EventSetup& c) const override;
0031
0032   private:
0033     typedef std::vector<double> ResultCollection1;
0034     typedef std::vector<bool> ResultCollection2;
0035
0036     JetVertexMain m_algo;
0037     edm::EDGetTokenT<reco::CaloJetCollection> jet_token;
0038     edm::EDGetTokenT<reco::TrackCollection> track_token;
0039     edm::EDGetTokenT<reco::VertexCollection> vertex_token;
0040   };
0041 }  // namespace cms
0042
0043 #endif