Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2023-06-23 01:38:41

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/HepMCProduct.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::HepMCProduct> 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