#include <ParticleLevelProducer.h>

Inheritance diagram for ParticleLevelProducer:

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

SerialTaskQueue *	globalLuminosityBlocksQueue () final

SerialTaskQueue *	globalRunsQueue () 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< ConsumesInfo >	consumesInfo () 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

ESProxyIndex const *	esGetTokenIndices (edm::Transition iTrans) const

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

EDConsumerBase &	operator= (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)

void	updateLookup (eventsetup::ESRecordsToProxyIndices const &)

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::RivetAnalysis *	rivetAnalysis_

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

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 ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)

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_.

     : srcToken_(consumes<edm::HepMCProduct>(pset.getParameter<edm::InputTag>("src"))),
       rivetAnalysis_(new Rivet::RivetAnalysis(pset)) {
   usesResource("Rivet");
 
   genVertex_ = reco::Particle::Point(0, 0, 0);
 
   produces<reco::GenParticleCollection>("neutrinos");
   produces<reco::GenParticleCollection>("photons");
   produces<reco::GenJetCollection>("leptons");
   produces<reco::GenJetCollection>("jets");
   produces<reco::GenJetCollection>("fatjets");
   produces<reco::GenParticleCollection>("consts");
   produces<reco::GenParticleCollection>("tags");
   produces<reco::METCollection>("mets");
 
   analysisHandler_.setIgnoreBeams(true);
   analysisHandler_.addAnalysis(rivetAnalysis_);
 }

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 39 of file ParticleLevelProducer.cc.

References reco::CompositePtrCandidate::addDaughter(), edmIntegrityCheck::d, 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().

                                                  {
   const auto pjet = jet.pseudojet();
 
   reco::GenJet genJet;
   genJet.setP4(p4(jet));
   genJet.setVertex(genVertex_);
   if (jet.bTagged())
     genJet.setPdgId(5);
   else if (jet.cTagged())
     genJet.setPdgId(4);
   genJet.setJetArea(pjet.has_area() ? pjet.area() : 0);
 
   for (auto const& p : jet.particles()) {
     auto pp4 = p4(p);
     bool match = false;
     int iMatch = -1;
     for (auto const& q : *consts) {
       ++iMatch;
       if (q.p4() == pp4) {
         match = true;
         break;
       }
     }
     if (match) {
       genJet.addDaughter(edm::refToPtr(reco::GenParticleRef(constsRefHandle, iMatch)));
     } else {
       consts->push_back(reco::GenParticle(p.charge(), pp4, genVertex_, p.pdgId(), 1, true));
       genJet.addDaughter(edm::refToPtr(reco::GenParticleRef(constsRefHandle, ++iConstituent)));
     }
   }
   for (auto const& p : jet.tags()) {
     // The tag particles are accessible as jet daughters, so scale down p4 for safety.
     // p4 needs to be multiplied by 1e20 for fragmentation analysis.
     auto pp4 = p4(p) * 1e-20;
     bool match = false;
     int iMatch = -1;
     for (auto const& q : *tags) {
       ++iMatch;
       if (q.p4() == pp4) {
         match = true;
         break;
       }
     }
     if (match) {
       genJet.addDaughter(edm::refToPtr(reco::GenParticleRef(tagsRefHandle, iMatch)));
     } else {
       tags->push_back(reco::GenParticle(p.charge(), p4(p) * 1e-20, genVertex_, p.pdgId(), 2, true));
       genJet.addDaughter(edm::refToPtr(reco::GenParticleRef(tagsRefHandle, ++iTag)));
       // Also save lepton+neutrino daughters of tag particles
       int iTagMother = iTag;
       for (auto const& d : p.constituents()) {
         ++iTag;
         int d_status = (d.isStable()) ? 1 : 2;
         tags->push_back(reco::GenParticle(d.charge(), p4(d) * 1e-20, genVertex_, d.pid(), d_status, true));
         tags->at(iTag).addMother(reco::GenParticleRef(tagsRefHandle, iTagMother));
         tags->at(iTagMother).addDaughter(reco::GenParticleRef(tagsRefHandle, iTag));
         genJet.addDaughter(edm::refToPtr(reco::GenParticleRef(tagsRefHandle, iTag)));
       }
     }
   }
 
   jets->push_back(genJet);
 }

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().

   {
     return reco::Candidate::LorentzVector(p.px(), p.py(), p.pz(), p.energy());
   }

void ParticleLevelProducer::produce	(	edm::Event &	event,
		const edm::EventSetup &	eventSetup
	)

override

Definition at line 110 of file ParticleLevelProducer.cc.

References reco::CompositePtrCandidate::addDaughter(), addGenJet(), analysisHandler_, DEFINE_FWK_MODULE, Rivet::RivetAnalysis::fatjets(), genWeightsTable_cfi::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(), jetUpdater_cfi::sort, mathSSE::sqrt(), and srcToken_.

Referenced by ~ParticleLevelProducer().

                                                                                     {
   using namespace Rivet;
   typedef reco::Candidate::LorentzVector LorentzVector;
 
   std::unique_ptr<reco::GenParticleCollection> neutrinos(new reco::GenParticleCollection);
   std::unique_ptr<reco::GenParticleCollection> photons(new reco::GenParticleCollection);
   std::unique_ptr<reco::GenJetCollection> leptons(new reco::GenJetCollection);
   std::unique_ptr<reco::GenJetCollection> jets(new reco::GenJetCollection);
   std::unique_ptr<reco::GenJetCollection> fatjets(new reco::GenJetCollection);
   std::unique_ptr<reco::GenParticleCollection> consts(new reco::GenParticleCollection);
   std::unique_ptr<reco::GenParticleCollection> tags(new reco::GenParticleCollection);
   std::unique_ptr<reco::METCollection> mets(new reco::METCollection);
   auto constsRefHandle = event.getRefBeforePut<reco::GenParticleCollection>("consts");
   auto tagsRefHandle = event.getRefBeforePut<reco::GenParticleCollection>("tags");
 
   edm::Handle<HepMCProduct> srcHandle;
   event.getByToken(srcToken_, srcHandle);
 
   const HepMC::GenEvent* genEvent = srcHandle->GetEvent();
   analysisHandler_.analyze(*genEvent);
 
   // Convert into edm objects
   // Prompt neutrinos
   for (auto const& p : rivetAnalysis_->neutrinos()) {
     neutrinos->push_back(reco::GenParticle(p.charge(), p4(p), genVertex_, p.pdgId(), 1, true));
   }
   std::sort(neutrinos->begin(), neutrinos->end(), GreaterByPt<reco::Candidate>());
 
   // Photons
   for (auto const& p : rivetAnalysis_->photons()) {
     photons->push_back(reco::GenParticle(p.charge(), p4(p), genVertex_, p.pdgId(), 1, true));
   }
   std::sort(photons->begin(), photons->end(), GreaterByPt<reco::Candidate>());
 
   // Prompt leptons
   int iConstituent = -1;
   int iTag = -1;
   for (auto const& lepton : rivetAnalysis_->leptons()) {
     reco::GenJet lepJet;
     lepJet.setP4(p4(lepton));
     lepJet.setVertex(genVertex_);
     lepJet.setPdgId(lepton.pdgId());
     lepJet.setCharge(lepton.charge());
 
     for (auto const & p : lepton.constituents()) {
       // ghost taus (momentum scaled with 10e-20 in RivetAnalysis.h already)
       if (p.abspid() == 15) {
         tags->push_back(reco::GenParticle(p.charge(), p4(p), genVertex_, p.pdgId(), 2, true));
         lepJet.addDaughter(edm::refToPtr(reco::GenParticleRef(tagsRefHandle, ++iTag)));
       }
       // electrons, muons, photons
       else {
         consts->push_back(reco::GenParticle(p.charge(), p4(p), genVertex_, p.pdgId(), 1, true));
         lepJet.addDaughter(edm::refToPtr(reco::GenParticleRef(constsRefHandle, ++iConstituent)));
       }
     }
 
     leptons->push_back(lepJet);
   }
   std::sort(leptons->begin(), leptons->end(), GreaterByPt<reco::GenJet>());
 
   // Jets with constituents and tag particles
   for (auto jet : rivetAnalysis_->jets()) {
     addGenJet(jet, jets, consts, constsRefHandle, iConstituent, tags, tagsRefHandle, iTag);
   }
   for (auto jet : rivetAnalysis_->fatjets()) {
     addGenJet(jet, fatjets, consts, constsRefHandle, iConstituent, tags, tagsRefHandle, iTag);
   }
 
   // MET
   reco::Candidate::LorentzVector metP4(rivetAnalysis_->met().x(),
                                        rivetAnalysis_->met().y(),
                                        0.,
                                        sqrt(pow(rivetAnalysis_->met().x(), 2) + pow(rivetAnalysis_->met().y(), 2)));
   mets->push_back(reco::MET(metP4, genVertex_));
 
   event.put(std::move(neutrinos), "neutrinos");
   event.put(std::move(photons), "photons");
   event.put(std::move(leptons), "leptons");
   event.put(std::move(jets), "jets");
   event.put(std::move(fatjets), "fatjets");
   event.put(std::move(consts), "consts");
   event.put(std::move(tags), "tags");
   event.put(std::move(mets), "mets");
 }