CMS 3D CMS Logo

List of all members | Public Member Functions | Private Member Functions | Private Attributes
ParticleLevelProducer Class Reference

#include <ParticleLevelProducer.h>

Inheritance diagram for ParticleLevelProducer:
edm::one::EDProducer< edm::one::SharedResources > edm::one::EDProducerBase edm::ProducerBase edm::EDConsumerBase edm::ProductRegistryHelper

Public Member Functions

 ParticleLevelProducer (const edm::ParameterSet &pset)
 
void produce (edm::Event &event, const edm::EventSetup &eventSetup) override
 
 ~ParticleLevelProducer () override
 
- Public Member Functions inherited from edm::one::EDProducer< edm::one::SharedResources >
 EDProducer ()=default
 
SerialTaskQueueglobalLuminosityBlocksQueue () final
 
SerialTaskQueueglobalRunsQueue () final
 
bool hasAbilityToProduceInLumis () const final
 
bool hasAbilityToProduceInRuns () const final
 
bool wantsGlobalLuminosityBlocks () const final
 
bool wantsGlobalRuns () const final
 
- Public Member Functions inherited from edm::one::EDProducerBase
 EDProducerBase ()
 
ModuleDescription const & moduleDescription () const
 
bool wantsStreamLuminosityBlocks () const
 
bool wantsStreamRuns () const
 
 ~EDProducerBase () override
 
- Public Member Functions inherited from edm::ProducerBase
void callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
 
std::vector< edm::ProductResolverIndex > const & indiciesForPutProducts (BranchType iBranchType) const
 
 ProducerBase ()
 
std::vector< edm::ProductResolverIndex > const & putTokenIndexToProductResolverIndex () const
 
void registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
 
std::function< void(BranchDescription const &)> registrationCallback () const
 used by the fwk to register list of products More...
 
void resolvePutIndicies (BranchType iBranchType, ModuleToResolverIndicies const &iIndicies, std::string const &moduleLabel)
 
 ~ProducerBase () noexcept(false) override
 
- Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfoconsumesInfo () const
 
void convertCurrentProcessAlias (std::string const &processName)
 Convert "@currentProcess" in InputTag process names to the actual current process name. More...
 
 EDConsumerBase ()
 
 EDConsumerBase (EDConsumerBase const &)=delete
 
 EDConsumerBase (EDConsumerBase &&)=default
 
ProductResolverIndexAndSkipBit indexFrom (EDGetToken, BranchType, TypeID const &) const
 
void itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
 
void itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
 
std::vector< ProductResolverIndexAndSkipBit > const & itemsToGetFrom (BranchType iType) const
 
void labelsForToken (EDGetToken iToken, Labels &oLabels) const
 
void modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const
 
EDConsumerBase const & operator= (EDConsumerBase const &)=delete
 
EDConsumerBaseoperator= (EDConsumerBase &&)=default
 
bool registeredToConsume (ProductResolverIndex, bool, BranchType) const
 
bool registeredToConsumeMany (TypeID const &, BranchType) const
 
ProductResolverIndexAndSkipBit uncheckedIndexFrom (EDGetToken) const
 
void updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
 
virtual ~EDConsumerBase () noexcept(false)
 

Private Member Functions

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
 

Private Attributes

Rivet::AnalysisHandler analysisHandler_
 
reco::Particle::Point genVertex_
 
Rivet::RivetAnalysisrivetAnalysis_
 
const edm::EDGetTokenT< edm::HepMCProductsrcToken_
 

Additional Inherited Members

- Public Types inherited from edm::one::EDProducerBase
typedef EDProducerBase ModuleType
 
- Public Types inherited from edm::ProducerBase
using ModuleToResolverIndicies = std::unordered_multimap< std::string, std::tuple< edm::TypeID const *, const char *, edm::ProductResolverIndex >>
 
typedef ProductRegistryHelper::TypeLabelList TypeLabelList
 
- Public Types inherited from edm::EDConsumerBase
typedef ProductLabels Labels
 
- Static Public Member Functions inherited from edm::one::EDProducerBase
static const std::string & baseType ()
 
static void fillDescriptions (ConfigurationDescriptions &descriptions)
 
static void prevalidate (ConfigurationDescriptions &descriptions)
 
- Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType > consumes (edm::InputTag const &tag)
 
EDGetToken consumes (const TypeToGet &id, edm::InputTag const &tag)
 
template<BranchType B>
EDGetToken consumes (TypeToGet const &id, edm::InputTag const &tag)
 
ConsumesCollector consumesCollector ()
 Use a ConsumesCollector to gather consumes information from helper functions. More...
 
template<typename ProductType , BranchType B = InEvent>
void consumesMany ()
 
void consumesMany (const TypeToGet &id)
 
template<BranchType B>
void consumesMany (const TypeToGet &id)
 
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType > mayConsume (edm::InputTag const &tag)
 
EDGetToken mayConsume (const TypeToGet &id, edm::InputTag const &tag)
 
template<BranchType B>
EDGetToken mayConsume (const TypeToGet &id, edm::InputTag const &tag)
 

Detailed Description

Definition at line 18 of file ParticleLevelProducer.h.

Constructor & Destructor Documentation

ParticleLevelProducer::ParticleLevelProducer ( const edm::ParameterSet pset)

Definition at line 19 of file ParticleLevelProducer.cc.

References analysisHandler_, genVertex_, and rivetAnalysis_.

19  :
20  srcToken_(consumes<edm::HepMCProduct>(pset.getParameter<edm::InputTag>("src"))),
22 {
23  usesResource("Rivet");
24 
26 
27  produces<reco::GenParticleCollection>("neutrinos");
28  produces<reco::GenParticleCollection>("photons");
29  produces<reco::GenJetCollection>("leptons");
30  produces<reco::GenJetCollection>("jets");
31  produces<reco::GenJetCollection>("fatjets");
32  produces<reco::GenParticleCollection>("consts");
33  produces<reco::GenParticleCollection>("tags");
34  produces<reco::METCollection>("mets");
35 
36  analysisHandler_.setIgnoreBeams(true);
37  analysisHandler_.addAnalysis(rivetAnalysis_);
38 }
T getParameter(std::string const &) const
math::XYZPoint Point
point in the space
Definition: Particle.h:25
Rivet::RivetAnalysis * rivetAnalysis_
const edm::EDGetTokenT< edm::HepMCProduct > srcToken_
Rivet::AnalysisHandler analysisHandler_
reco::Particle::Point genVertex_
ParticleLevelProducer::~ParticleLevelProducer ( )
inlineoverride

Definition at line 22 of file ParticleLevelProducer.h.

References addGenJet(), metsig::jet, nanoDQM_cfi::Jet, fwrapper::jets, and produce().

22 {}

Member Function Documentation

void ParticleLevelProducer::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 
)
private

Definition at line 40 of file ParticleLevelProducer.cc.

References reco::CompositePtrCandidate::addDaughter(), MillePedeFileConverter_cfg::e, genVertex_, match(), AlCaHLTBitMon_ParallelJobs::p, p4(), lumiQueryAPI::q, edm::refToPtr(), reco::Jet::setJetArea(), reco::LeafCandidate::setP4(), reco::LeafCandidate::setPdgId(), and reco::LeafCandidate::setVertex().

Referenced by produce(), and ~ParticleLevelProducer().

43 {
44  const auto pjet = jet.pseudojet();
45 
46  reco::GenJet genJet;
47  genJet.setP4(p4(jet));
48  genJet.setVertex(genVertex_);
49  if ( jet.bTagged() ) genJet.setPdgId(5);
50  else if ( jet.cTagged() ) genJet.setPdgId(4);
51  genJet.setJetArea(pjet.has_area() ? pjet.area() : 0);
52 
53  for ( auto const & p : jet.particles()) {
54  auto pp4 = p4(p);
55  bool match = false; int iMatch = -1;
56  for ( auto const & q : *consts ) {
57  ++iMatch;
58  if (q.p4() == pp4) { match = true; break; }
59  }
60  if (match){
61  genJet.addDaughter(edm::refToPtr(reco::GenParticleRef(constsRefHandle, iMatch)));
62  }
63  else {
64  consts->push_back(reco::GenParticle(p.charge(), pp4, genVertex_, p.pdgId(), 1, true));
65  genJet.addDaughter(edm::refToPtr(reco::GenParticleRef(constsRefHandle, ++iConstituent)));
66  }
67  }
68  for ( auto const & p : jet.tags()) {
69  // The tag particles are accessible as jet daughters, so scale down p4 for safety.
70  // p4 needs to be multiplied by 1e20 for fragmentation analysis.
71  auto pp4 = p4(p)*1e-20;
72  bool match = false; int iMatch = -1;
73  for ( auto const & q : *tags ) {
74  ++iMatch;
75  if (q.p4() == pp4) { match = true; break; }
76  }
77  if (match){
78  genJet.addDaughter(edm::refToPtr(reco::GenParticleRef(tagsRefHandle, iMatch)));
79  }
80  else {
81  tags->push_back(reco::GenParticle(p.charge(), p4(p)*1e-20, genVertex_, p.pdgId(), 2, true));
82  genJet.addDaughter(edm::refToPtr(reco::GenParticleRef(tagsRefHandle, ++iTag)));
83  }
84  }
85 
86  jets->push_back(genJet);
87 }
Ptr< typename C::value_type > refToPtr(Ref< C, typename C::value_type, refhelper::FindUsingAdvance< C, typename C::value_type > > const &ref)
Definition: RefToPtr.h:18
void setVertex(const Point &vertex) override
set vertex
virtual void setJetArea(float fArea)
set jet area
Definition: Jet.h:103
reco::Candidate::LorentzVector p4(const T &p) const
vector< PseudoJet > jets
Jets made from MC generator particles.
Definition: GenJet.h:24
void addDaughter(const CandidatePtr &)
add a daughter via a reference
std::pair< typename Association::data_type::first_type, double > match(Reference key, Association association, bool bestMatchByMaxValue)
Generic matching function.
Definition: Utils.h:10
void setPdgId(int pdgId) final
void setP4(const LorentzVector &p4) final
set 4-momentum
reco::Particle::Point genVertex_
template<typename T >
reco::Candidate::LorentzVector ParticleLevelProducer::p4 ( const T p) const
inlineprivate

Definition at line 30 of file ParticleLevelProducer.h.

Referenced by addGenJet(), Tau.Tau::dxy_approx(), Tau.Tau::dz(), Lepton.Lepton::p4WithFSR(), and produce().

31  {
32  return reco::Candidate::LorentzVector(p.px(), p.py(), p.pz(), p.energy());
33  }
math::XYZTLorentzVector LorentzVector
Lorentz vector.
Definition: Candidate.h:37
void ParticleLevelProducer::produce ( edm::Event event,
const edm::EventSetup eventSetup 
)
override

Definition at line 89 of file ParticleLevelProducer.cc.

References reco::CompositePtrCandidate::addDaughter(), addGenJet(), analysisHandler_, GetRecoTauVFromDQM_MC_cff::cl, DEFINE_FWK_MODULE, Rivet::RivetAnalysis::fatjets(), nano_cff::genEvent, genVertex_, metsig::jet, Rivet::RivetAnalysis::jets(), fwrapper::jets, AK4GenJetFlavourInfos_cfi::leptons, Rivet::RivetAnalysis::leptons(), Rivet::RivetAnalysis::met(), TtSemiLepSignalSelMVAComputer_cfi::mets, eostools::move(), TtSemiLepHypHitFit_cfi::neutrinos, Rivet::RivetAnalysis::neutrinos(), AlCaHLTBitMon_ParallelJobs::p, p4(), Rivet::RivetAnalysis::photons(), muons_cff::photons, funct::pow(), edm::refToPtr(), rivetAnalysis_, reco::LeafCandidate::setCharge(), reco::LeafCandidate::setP4(), reco::LeafCandidate::setPdgId(), reco::LeafCandidate::setVertex(), mathSSE::sqrt(), and srcToken_.

Referenced by ~ParticleLevelProducer().

90 {
91  using namespace Rivet;
93 
94  std::unique_ptr<reco::GenParticleCollection> neutrinos(new reco::GenParticleCollection);
95  std::unique_ptr<reco::GenParticleCollection> photons(new reco::GenParticleCollection);
96  std::unique_ptr<reco::GenJetCollection> leptons(new reco::GenJetCollection);
97  std::unique_ptr<reco::GenJetCollection> jets(new reco::GenJetCollection);
98  std::unique_ptr<reco::GenJetCollection> fatjets(new reco::GenJetCollection);
99  std::unique_ptr<reco::GenParticleCollection> consts(new reco::GenParticleCollection);
100  std::unique_ptr<reco::GenParticleCollection> tags(new reco::GenParticleCollection);
101  std::unique_ptr<reco::METCollection> mets(new reco::METCollection);
102  auto constsRefHandle = event.getRefBeforePut<reco::GenParticleCollection>("consts");
103  auto tagsRefHandle = event.getRefBeforePut<reco::GenParticleCollection>("tags");
104 
105  edm::Handle<HepMCProduct> srcHandle;
106  event.getByToken(srcToken_, srcHandle);
107 
108  const HepMC::GenEvent* genEvent = srcHandle->GetEvent();
109  analysisHandler_.analyze(*genEvent);
110 
111  // Convert into edm objects
112  // Prompt neutrinos
113  for ( auto const & p : rivetAnalysis_->neutrinos() ) {
114  neutrinos->push_back(reco::GenParticle(p.charge(), p4(p), genVertex_, p.pdgId(), 1, true));
115  }
116  std::sort(neutrinos->begin(), neutrinos->end(), GreaterByPt<reco::Candidate>());
117 
118  // Photons
119  for ( auto const & p : rivetAnalysis_->photons() ) {
120  photons->push_back(reco::GenParticle(p.charge(), p4(p), genVertex_, p.pdgId(), 1, true));
121  }
122  std::sort(photons->begin(), photons->end(), GreaterByPt<reco::Candidate>());
123 
124  // Prompt leptons
125  int iConstituent = -1;
126  int iTag = -1;
127  for ( auto const & lepton : rivetAnalysis_->leptons() ) {
128  reco::GenJet lepJet;
129  lepJet.setP4(p4(lepton));
130  lepJet.setVertex(genVertex_);
131  lepJet.setPdgId(lepton.pdgId());
132  lepJet.setCharge(lepton.charge());
133 
134  const auto& cl = lepton.constituentLepton();
135  consts->push_back(reco::GenParticle(cl.charge(), p4(cl), genVertex_, cl.pdgId(), 1, true));
136  lepJet.addDaughter(edm::refToPtr(reco::GenParticleRef(constsRefHandle, ++iConstituent)));
137 
138  for ( auto const & p : lepton.constituentPhotons()) {
139  // ghost taus (momentum scaled with 10e-20 in RivetAnalysis.h already)
140  if (p.abspid() == 15) {
141  tags->push_back(reco::GenParticle(p.charge(), p4(p), genVertex_, p.pdgId(), 2, true));
142  lepJet.addDaughter(edm::refToPtr(reco::GenParticleRef(tagsRefHandle, ++iTag)));
143  }
144  // photons
145  else {
146  consts->push_back(reco::GenParticle(p.charge(), p4(p), genVertex_, p.pdgId(), 1, true));
147  lepJet.addDaughter(edm::refToPtr(reco::GenParticleRef(constsRefHandle, ++iConstituent)));
148  }
149  }
150 
151  leptons->push_back(lepJet);
152  }
153  std::sort(leptons->begin(), leptons->end(), GreaterByPt<reco::GenJet>());
154 
155  // Jets with constituents and tag particles
156  for ( auto jet : rivetAnalysis_->jets() ) {
157  addGenJet(jet, jets, consts, constsRefHandle, iConstituent, tags, tagsRefHandle, iTag);
158  }
159  for ( auto jet : rivetAnalysis_->fatjets() ) {
160  addGenJet(jet, fatjets, consts, constsRefHandle, iConstituent, tags, tagsRefHandle, iTag);
161  }
162 
163  // MET
165  mets->push_back(reco::MET(metP4, genVertex_));
166 
167  event.put(std::move(neutrinos), "neutrinos");
168  event.put(std::move(photons), "photons");
169  event.put(std::move(leptons), "leptons");
170  event.put(std::move(jets), "jets");
171  event.put(std::move(fatjets), "fatjets");
172  event.put(std::move(consts), "consts");
173  event.put(std::move(tags), "tags");
174  event.put(std::move(mets), "mets");
175 }
std::vector< GenParticle > GenParticleCollection
collection of GenParticles
Vector3 met() const
Definition: RivetAnalysis.h:31
Ptr< typename C::value_type > refToPtr(Ref< C, typename C::value_type, refhelper::FindUsingAdvance< C, typename C::value_type > > const &ref)
Definition: RefToPtr.h:18
std::vector< GenJet > GenJetCollection
collection of GenJet objects
std::vector< reco::MET > METCollection
collection of MET objects
Definition: METCollection.h:23
Jets fatjets() const
Definition: RivetAnalysis.h:30
void setVertex(const Point &vertex) override
set vertex
ParticleVector neutrinos() const
Definition: RivetAnalysis.h:28
void setCharge(Charge q) final
set electric charge
Definition: LeafCandidate.h:93
Jets jets() const
Definition: RivetAnalysis.h:29
ParticleVector photons() const
Definition: RivetAnalysis.h:27
reco::Candidate::LorentzVector p4(const T &p) const
Definition: MET.h:42
T sqrt(T t)
Definition: SSEVec.h:18
vector< PseudoJet > jets
std::vector< DressedLepton > leptons() const
Definition: RivetAnalysis.h:26
Jets made from MC generator particles.
Definition: GenJet.h:24
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)
Rivet::RivetAnalysis * rivetAnalysis_
math::XYZTLorentzVector LorentzVector
Lorentz vector.
Definition: Candidate.h:37
const edm::EDGetTokenT< edm::HepMCProduct > srcToken_
Rivet::AnalysisHandler analysisHandler_
void addDaughter(const CandidatePtr &)
add a daughter via a reference
void setPdgId(int pdgId) final
void setP4(const LorentzVector &p4) final
set 4-momentum
Power< A, B >::type pow(const A &a, const B &b)
Definition: Power.h:40
def move(src, dest)
Definition: eostools.py:510
math::PtEtaPhiELorentzVectorF LorentzVector
reco::Particle::Point genVertex_

Member Data Documentation

Rivet::AnalysisHandler ParticleLevelProducer::analysisHandler_
private

Definition at line 40 of file ParticleLevelProducer.h.

Referenced by ParticleLevelProducer(), and produce().

reco::Particle::Point ParticleLevelProducer::genVertex_
private

Definition at line 37 of file ParticleLevelProducer.h.

Referenced by addGenJet(), ParticleLevelProducer(), and produce().

Rivet::RivetAnalysis* ParticleLevelProducer::rivetAnalysis_
private

Definition at line 39 of file ParticleLevelProducer.h.

Referenced by ParticleLevelProducer(), and produce().

const edm::EDGetTokenT<edm::HepMCProduct> ParticleLevelProducer::srcToken_
private

Definition at line 35 of file ParticleLevelProducer.h.

Referenced by produce().