File indexing completed on 2024-07-03 04:17:52
0001 #ifndef GeneratorInterface_RivetInterface_ParticleLevelProducer_H
0002 #define GeneratorInterface_RivetInterface_ParticleLevelProducer_H
0003
0004 #include "FWCore/Framework/interface/Frameworkfwd.h"
0005 #include "FWCore/Framework/interface/one/EDProducer.h"
0006 #include "FWCore/Framework/interface/Event.h"
0007 #include "FWCore/Framework/interface/MakerMacros.h"
0008 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0009
0010 #include "DataFormats/Candidate/interface/Candidate.h"
0011 #include "SimDataFormats/GeneratorProducts/interface/HepMC3Product.h"
0012 #include "DataFormats/JetReco/interface/GenJetCollection.h"
0013 #include "DataFormats/HepMCCandidate/interface/GenParticleFwd.h"
0014
0015 #include "Rivet/AnalysisHandler.hh"
0016 #include "GeneratorInterface/RivetInterface/interface/RivetAnalysis.h"
0017
0018 class ParticleLevelProducer : public edm::one::EDProducer<edm::one::SharedResources> {
0019 public:
0020 ParticleLevelProducer(const edm::ParameterSet& pset);
0021 ~ParticleLevelProducer() override {}
0022 void produce(edm::Event& event, const edm::EventSetup& eventSetup) override;
0023
0024 private:
0025 void addGenJet(Rivet::Jet jet,
0026 std::unique_ptr<reco::GenJetCollection>& jets,
0027 std::unique_ptr<reco::GenParticleCollection>& consts,
0028 edm::RefProd<reco::GenParticleCollection>& constsRefHandle,
0029 int& iConstituent,
0030 std::unique_ptr<reco::GenParticleCollection>& tags,
0031 edm::RefProd<reco::GenParticleCollection>& tagsRefHandle,
0032 int& iTag);
0033
0034 template <typename T>
0035 reco::Candidate::LorentzVector p4(const T& p) const {
0036 return reco::Candidate::LorentzVector(p.px(), p.py(), p.pz(), p.energy());
0037 }
0038
0039 const edm::EDGetTokenT<edm::HepMC3Product> srcToken_;
0040 const edm::ParameterSet pset_;
0041
0042 reco::Particle::Point genVertex_;
0043
0044 Rivet::RivetAnalysis* rivetAnalysis_ = nullptr;
0045 std::unique_ptr<Rivet::AnalysisHandler> analysisHandler_;
0046 };
0047
0048 #endif
