#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

virtual	~ParticleLevelProducer ()

Public Member Functions inherited from edm::one::EDProducer< edm::one::SharedResources >
	EDProducer ()=default

Public Member Functions inherited from edm::one::EDProducerBase
	EDProducerBase ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducerBase ()

Public Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

virtual	~ProducerBase ()

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

	EDConsumerBase ()

ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesDependentUpon (std::string const &iProcessName, std::string const &iModuleLabel, bool iPrint, std::vector< char const * > &oModuleLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

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

                                                                        :
   srcToken_(consumes<edm::HepMCProduct>(pset.getParameter<edm::InputTag>("src"))),
   rivetAnalysis_(new Rivet::RivetAnalysis(pset))
 {
   usesResource();
 
   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_.addAnalysis(rivetAnalysis_);
 }

virtual ParticleLevelProducer::~ParticleLevelProducer ( )

inlinevirtual

Definition at line 22 of file ParticleLevelProducer.h.

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(), alignCSCRings::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().

 {
   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)));
     }
   }
 
   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
	)

overridevirtual

Implements edm::one::EDProducerBase.

Definition at line 88 of file ParticleLevelProducer.cc.

References reco::CompositePtrCandidate::addDaughter(), addGenJet(), analysisHandler_, GetRecoTauVFromDQM_MC_cff::cl, Rivet::RivetAnalysis::fatjets(), MCTruth::genEvent, genVertex_, metsig::jet, Rivet::RivetAnalysis::jets(), fwrapper::jets, Rivet::RivetAnalysis::leptons(), HLT_25ns10e33_v2_cff::leptons, Rivet::RivetAnalysis::met(), eostools::move(), Rivet::RivetAnalysis::neutrinos(), AlCaHLTBitMon_ParallelJobs::p, p4(), interactiveExample::photons, Rivet::RivetAnalysis::photons(), funct::pow(), edm::refToPtr(), rivetAnalysis_, reco::LeafCandidate::setCharge(), reco::LeafCandidate::setP4(), reco::LeafCandidate::setPdgId(), reco::LeafCandidate::setVertex(), mathSSE::sqrt(), srcToken_, and o2o::tags.

 {
   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;
   for ( auto const & lepton : rivetAnalysis_->leptons() ) {
     reco::GenJet lepJet;
     lepJet.setP4(p4(lepton));
     lepJet.setVertex(genVertex_);
     lepJet.setPdgId(lepton.pdgId());
     lepJet.setCharge(lepton.charge());
     
     const auto cl = lepton.constituentLepton();
     consts->push_back(reco::GenParticle(cl.charge(), p4(cl), genVertex_, cl.pdgId(), 1, true));
     lepJet.addDaughter(edm::refToPtr(reco::GenParticleRef(constsRefHandle, ++iConstituent)));
     
     for ( auto const & p : lepton.constituentPhotons()) {
       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
   int iTag = -1;
   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");
 }