|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
#ifndef GeneratorInterface_RivetInterface_ParticleLevelProducer_H
#define GeneratorInterface_RivetInterface_ParticleLevelProducer_H
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/one/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "SimDataFormats/GeneratorProducts/interface/HepMC3Product.h"
#include "DataFormats/JetReco/interface/GenJetCollection.h"
#include "DataFormats/HepMCCandidate/interface/GenParticleFwd.h"
#include "Rivet/AnalysisHandler.hh"
#include "GeneratorInterface/RivetInterface/interface/RivetAnalysis.h"
class ParticleLevelProducer : public edm::one::EDProducer<edm::one::SharedResources> {
public:
ParticleLevelProducer(const edm::ParameterSet& pset);
~ParticleLevelProducer() override {}
void produce(edm::Event& event, const edm::EventSetup& eventSetup) override;
private:
void addGenJet(Rivet::Jet jet,
std::unique_ptr<reco::GenJetCollection>& jets,
std::unique_ptr<reco::GenParticleCollection>& consts,
edm::RefProd<reco::GenParticleCollection>& constsRefHandle,
int& iConstituent,
std::unique_ptr<reco::GenParticleCollection>& tags,
edm::RefProd<reco::GenParticleCollection>& tagsRefHandle,
int& iTag);
template <typename T>
reco::Candidate::LorentzVector p4(const T& p) const {
return reco::Candidate::LorentzVector(p.px(), p.py(), p.pz(), p.energy());
}
const edm::EDGetTokenT<edm::HepMC3Product> srcToken_;
const edm::ParameterSet pset_;
reco::Particle::Point genVertex_;
Rivet::RivetAnalysis* rivetAnalysis_ = nullptr;
std::unique_ptr<Rivet::AnalysisHandler> analysisHandler_;
};
#endif
|