#include <PseudoTopProducer.h>

Inheritance diagram for PseudoTopProducer:

Public Member Functions
void	produce (edm::Event &event, const edm::EventSetup &eventSetup) override

	PseudoTopProducer (const edm::ParameterSet &pset)

Public Member Functions inherited from edm::EDProducer
	EDProducer ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducer ()

Public Member Functions inherited from edm::ProducerBase
	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 (const std::string &iProcessName, std::vector< const char * > &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 Types
typedef fastjet::JetDefinition	JetDef

typedef reco::Particle::LorentzVector	LorentzVector

Private Member Functions
reco::GenParticleRef	buildGenParticle (const reco::Candidate *p, reco::GenParticleRefProd &refHandle, std::auto_ptr< reco::GenParticleCollection > &outColl) const

const reco::Candidate *	getLast (const reco::Candidate *p)

bool	isBHadron (const reco::Candidate *p) const

bool	isBHadron (const unsigned int pdgId) const

bool	isFromHadron (const reco::Candidate *p) const

Private Attributes
edm::EDGetTokenT< edm::View < reco::Candidate > >	finalStateToken_

std::shared_ptr< JetDef >	fjJetDef_

std::shared_ptr< JetDef >	fjLepDef_

edm::EDGetTokenT< edm::View < reco::Candidate > >	genParticleToken_

reco::Particle::Point	genVertex_

const double	jetMaxEta_

const double	jetMinPt_

const double	leptonMaxEta_

const double	leptonMinPt_

const double	tMass_

const double	wMass_

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Static Public Member Functions inherited from edm::EDProducer
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

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

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 13 of file PseudoTopProducer.h.

Member Typedef Documentation

typedef fastjet::JetDefinition PseudoTopProducer::JetDef

private

Definition at line 36 of file PseudoTopProducer.h.

typedef reco::Particle::LorentzVector PseudoTopProducer::LorentzVector

private

Definition at line 28 of file PseudoTopProducer.h.

Constructor & Destructor Documentation

PseudoTopProducer::PseudoTopProducer ( const edm::ParameterSet & pset )

Definition at line 12 of file PseudoTopProducer.cc.

References finalStateToken_, fjJetDef_, fjLepDef_, genParticleToken_, genVertex_, and edm::ParameterSet::getParameter().

                                                                :
   leptonMinPt_(pset.getParameter<double>("leptonMinPt")),
   leptonMaxEta_(pset.getParameter<double>("leptonMaxEta")),
   jetMinPt_(pset.getParameter<double>("jetMinPt")),
   jetMaxEta_(pset.getParameter<double>("jetMaxEta")),
   wMass_(pset.getParameter<double>("wMass")),
   tMass_(pset.getParameter<double>("tMass"))
 {
   finalStateToken_ = consumes<edm::View<reco::Candidate> >(pset.getParameter<edm::InputTag>("finalStates"));
   genParticleToken_ = consumes<edm::View<reco::Candidate> >(pset.getParameter<edm::InputTag>("genParticles"));
 
   const double leptonConeSize = pset.getParameter<double>("leptonConeSize");
   const double jetConeSize = pset.getParameter<double>("jetConeSize");
   fjLepDef_ = std::shared_ptr<JetDef>(new JetDef(fastjet::antikt_algorithm, leptonConeSize));
   fjJetDef_ = std::shared_ptr<JetDef>(new JetDef(fastjet::antikt_algorithm, jetConeSize));
 
   genVertex_ = reco::Particle::Point(0,0,0);
 
   produces<reco::GenParticleCollection>("neutrinos");
   produces<reco::GenJetCollection>("leptons");
   produces<reco::GenJetCollection>("jets");
 
   produces<reco::GenParticleCollection>();
 
 }

Member Function Documentation

reco::GenParticleRef PseudoTopProducer::buildGenParticle	(	const reco::Candidate *	p,
		reco::GenParticleRefProd &	refHandle,
		std::auto_ptr< reco::GenParticleCollection > &	outColl
	)		const

private

Definition at line 563 of file PseudoTopProducer.cc.

References reco::CompositeRefCandidateT< D >::clearDaughters(), reco::CompositeRefCandidateT< D >::clearMothers(), edm::RefProd< T >::id(), reco::CompositeRefCandidateT< D >::resetDaughters(), and reco::CompositeRefCandidateT< D >::resetMothers().

 {
   reco::GenParticle pOut(*dynamic_cast<const reco::GenParticle*>(p));
   pOut.clearMothers();
   pOut.clearDaughters();
   pOut.resetMothers(refHandle.id());
   pOut.resetDaughters(refHandle.id());
 
   outColl->push_back(pOut);
 
   return reco::GenParticleRef(refHandle, outColl->size()-1);
 }

const reco::Candidate * PseudoTopProducer::getLast ( const reco::Candidate * p )

private

Definition at line 504 of file PseudoTopProducer.cc.

References reco::Candidate::daughter(), i, gen::n, reco::Candidate::numberOfDaughters(), AlCaHLTBitMon_ParallelJobs::p, and reco::Candidate::pdgId().

 {
   for ( size_t i=0, n=p->numberOfDaughters(); i<n; ++i )
   {
     const reco::Candidate* dau = p->daughter(i);
     if ( p->pdgId() == dau->pdgId() ) return getLast(dau);
   }
   return p;
 }

bool PseudoTopProducer::isBHadron ( const reco::Candidate * p ) const

private

Definition at line 528 of file PseudoTopProducer.cc.

References funct::abs(), reco::Candidate::daughter(), i, gen::n, reco::Candidate::numberOfDaughters(), and reco::Candidate::pdgId().

Referenced by produce().

 {
   const unsigned int absPdgId = abs(p->pdgId());
   if ( !isBHadron(absPdgId) ) return false;
 
   // Do not consider this particle if it has B hadron daughter
   // For example, B* -> B0 + photon; then we drop B* and take B0 only
   for ( int i=0, n=p->numberOfDaughters(); i<n; ++i )
   {
     const reco::Candidate* dau = p->daughter(i);
     if ( isBHadron(abs(dau->pdgId())) ) return false;
   }
 
   return true;
 }

bool PseudoTopProducer::isBHadron ( const unsigned int pdgId ) const

private

Definition at line 544 of file PseudoTopProducer.cc.

 {
   if ( absPdgId <= 100 ) return false; // Fundamental particles and MC internals
   if ( absPdgId >= 1000000000 ) return false; // Nuclei, +-10LZZZAAAI
 
   // General form of PDG ID is 7 digit form
   // +- n nr nL nq1 nq2 nq3 nJ
   //const int nJ = absPdgId % 10; // Spin
   const int nq3 = (absPdgId / 10) % 10;
   const int nq2 = (absPdgId / 100) % 10;
   const int nq1 = (absPdgId / 1000) % 10;
 
   if ( nq3 == 0 ) return false; // Diquarks
   if ( nq1 == 0 and nq2 == 5 ) return true; // B mesons
   if ( nq1 == 5 ) return true; // B baryons
 
   return false;
 }

bool PseudoTopProducer::isFromHadron ( const reco::Candidate * p ) const

private

Definition at line 514 of file PseudoTopProducer.cc.

References funct::abs(), i, reco::Candidate::mother(), gen::n, reco::Candidate::numberOfMothers(), benchmark_cfg::pdgId, and reco::Candidate::pdgId().

Referenced by produce().

 {
   for ( size_t i=0, n=p->numberOfMothers(); i<n; ++i )
   {
     const reco::Candidate* mother = p->mother(i);
     if ( mother->numberOfMothers() == 0 ) continue; // Skip incident beam
     const int pdgId = abs(mother->pdgId());
 
     if ( pdgId > 100 ) return true;
     else if ( isFromHadron(mother) ) return true;
   }
   return false;
 }

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

overridevirtual

Build jets

Implements edm::EDProducer.

Definition at line 38 of file PseudoTopProducer.cc.

References funct::abs(), reco::CompositeRefCandidateT< D >::addDaughter(), RazorAnalyzer::bJet1, RazorAnalyzer::bJet2, reco::Candidate::charge(), alignCSCRings::e, reco::Candidate::energy(), finalStateToken_, fjJetDef_, fjLepDef_, genParticleToken_, genVertex_, i, cmsHarvester::index, isBHadron(), isFromHadron(), edm::detail::isnan(), edm::Ptr< T >::isNonnull(), edm::Ptr< T >::isNull(), j, jetMaxEta_, jetMinPt_, fwrapper::jets, leptonMaxEta_, leptonMinPt_, HLT_25ns14e33_v1_cff::leptons, visualization-live-secondInstance_cfg::m, reco::Candidate::mother(), gen::n, reco::Candidate::numberOfMothers(), or, reco::Candidate::p(), AlCaHLTBitMon_ParallelJobs::p, reco::Candidate::p4(), reco::Candidate::pdgId(), pseudoTop_cfi::pseudoTop, reco::Candidate::pt(), reco::Candidate::px(), reco::Candidate::py(), reco::Candidate::pz(), pileupReCalc_HLTpaths::scale, python.multivaluedict::sort(), reco::Candidate::status(), ntuplemaker::status, tMass_, reco::Candidate::vertex(), w2, wMass_, and reco::writeSpecific().

 {
   edm::Handle<edm::View<reco::Candidate> > finalStateHandle;
   event.getByToken(finalStateToken_, finalStateHandle);
 
   edm::Handle<edm::View<reco::Candidate> > genParticleHandle;
   event.getByToken(genParticleToken_, genParticleHandle);
 
   std::auto_ptr<reco::GenParticleCollection> neutrinos(new reco::GenParticleCollection);
   std::auto_ptr<reco::GenJetCollection> leptons(new reco::GenJetCollection);
   std::auto_ptr<reco::GenJetCollection> jets(new reco::GenJetCollection);
   auto neutrinosRefHandle = event.getRefBeforePut<reco::GenParticleCollection>("neutrinos");
   auto leptonsRefHandle = event.getRefBeforePut<reco::GenJetCollection>("leptons");
   auto jetsRefHandle = event.getRefBeforePut<reco::GenJetCollection>("jets");
 
   std::auto_ptr<reco::GenParticleCollection> pseudoTop(new reco::GenParticleCollection);
   auto pseudoTopRefHandle = event.getRefBeforePut<reco::GenParticleCollection>();
 
   // Collect unstable B-hadrons
   std::set<size_t> bHadronIdxs;
   for ( size_t i=0, n=genParticleHandle->size(); i<n; ++i )
   {
     const reco::Candidate& p = genParticleHandle->at(i);
     const int status = p.status();
     if ( status == 1 ) continue;
 
     // Collect B-hadrons, to be used in b tagging
     if ( isBHadron(&p) ) bHadronIdxs.insert(i);
   }
 
   // Collect stable leptons and neutrinos
   size_t nStables = 0;
   std::vector<size_t> leptonIdxs;
   std::set<size_t> neutrinoIdxs;
   for ( size_t i=0, n=finalStateHandle->size(); i<n; ++i )
   {
     const reco::Candidate& p = finalStateHandle->at(i);
     const int absPdgId = abs(p.pdgId());
     if ( p.status() != 1 ) continue;
 
     ++nStables;
     if ( p.numberOfMothers() == 0 ) continue; // Skip orphans (if exists)
     if ( p.mother()->status() == 4 ) continue; // Treat particle as hadronic if directly from the incident beam (protect orphans in MINIAOD)
     if ( isFromHadron(&p) ) continue;
     switch ( absPdgId )
     {
       case 11: case 13: // Leptons
       case 22: // Photons
         leptonIdxs.push_back(i);
         break;
       case 12: case 14: case 16:
         neutrinoIdxs.insert(i);
         neutrinos->push_back(reco::GenParticle(p.charge(), p.p4(), p.vertex(), p.pdgId(), p.status(), true));
         break;
     }
   }
 
   // Sort neutrinos by pT.
   std::sort(neutrinos->begin(), neutrinos->end(), GreaterByPt<reco::Candidate>());
 
   // Make dressed leptons with anti-kt(0.1) algorithm
   std::vector<fastjet::PseudoJet> fjLepInputs;
   fjLepInputs.reserve(leptonIdxs.size());
   for ( auto index : leptonIdxs )
   {
     const reco::Candidate& p = finalStateHandle->at(index);
     if ( std::isnan(p.pt()) or p.pt() <= 0 ) continue;
 
     fjLepInputs.push_back(fastjet::PseudoJet(p.px(), p.py(), p.pz(), p.energy()));
     fjLepInputs.back().set_user_index(index);
   }
 
   fastjet::ClusterSequence fjLepClusterSeq(fjLepInputs, *fjLepDef_);
   std::vector<fastjet::PseudoJet> fjLepJets = fastjet::sorted_by_pt(fjLepClusterSeq.inclusive_jets(leptonMinPt_));
 
   leptons->reserve(fjLepJets.size());
   std::set<size_t> lepDauIdxs; // keep lepton constituents to remove from GenJet construction
   for ( auto& fjJet : fjLepJets )
   {
     if ( abs(fjJet.eta()) > leptonMaxEta_ ) continue;
 
     // Get jet constituents from fastJet
     const std::vector<fastjet::PseudoJet> fjConstituents = fastjet::sorted_by_pt(fjJet.constituents());
     // Convert to CandidatePtr
     std::vector<reco::CandidatePtr> constituents;
     reco::CandidatePtr lepCand;
     for ( auto& fjConstituent : fjConstituents )
     {
       const size_t index = fjConstituent.user_index();
       reco::CandidatePtr cand = finalStateHandle->ptrAt(index);
       const int absPdgId = abs(cand->pdgId());
       if ( absPdgId == 11 or absPdgId == 13 )
       {
         if ( lepCand.isNonnull() and lepCand->pt() > cand->pt() ) continue; // Choose one with highest pt
         lepCand = cand;
       }
       constituents.push_back(cand);
     }
     if ( lepCand.isNull() ) continue;
     if ( lepCand->pt() < fjJet.pt()/2 ) continue; // Central lepton must be the major component
 
     const LorentzVector jetP4(fjJet.px(), fjJet.py(), fjJet.pz(), fjJet.E());
     reco::GenJet lepJet;
     reco::writeSpecific(lepJet, jetP4, genVertex_, constituents, eventSetup);
 
     lepJet.setPdgId(lepCand->pdgId());
     lepJet.setCharge(lepCand->charge());
 
     const double jetArea = fjJet.has_area() ? fjJet.area() : 0;
     lepJet.setJetArea(jetArea);
 
     leptons->push_back(lepJet);
 
     // Keep constituent indices to be used in the next step.
     for ( auto& fjConstituent : fjConstituents )
     {
       lepDauIdxs.insert(fjConstituent.user_index());
     }
   }
 
   // Now proceed to jets.
   // Jets: anti-kt excluding the e, mu, nu, and photons in selected leptons.
   std::vector<fastjet::PseudoJet> fjJetInputs;
   fjJetInputs.reserve(nStables);
   for ( size_t i=0, n=finalStateHandle->size(); i<n; ++i )
   {
     const reco::Candidate& p = finalStateHandle->at(i);
     if ( p.status() != 1 ) continue;
     if ( std::isnan(p.pt()) or p.pt() <= 0 ) continue;
 
     if ( neutrinoIdxs.find(i) != neutrinoIdxs.end() ) continue;
     if ( lepDauIdxs.find(i) != lepDauIdxs.end() ) continue;
 
     fjJetInputs.push_back(fastjet::PseudoJet(p.px(), p.py(), p.pz(), p.energy()));
     fjJetInputs.back().set_user_index(i);
   }
   for ( auto index : bHadronIdxs )
   {
     const reco::Candidate& p = genParticleHandle->at(index);
     if ( std::isnan(p.pt()) or p.pt() <= 0 ) continue;
 
     const double scale = 1e-20/p.p();
     fjJetInputs.push_back(fastjet::PseudoJet(p.px()*scale, p.py()*scale, p.pz()*scale, p.energy()*scale));
     fjJetInputs.back().set_user_index(index);
   }
 
   fastjet::ClusterSequence fjJetClusterSeq(fjJetInputs, *fjJetDef_);
   std::vector<fastjet::PseudoJet> fjJets = fastjet::sorted_by_pt(fjJetClusterSeq.inclusive_jets(jetMinPt_));
 
   jets->reserve(fjJets.size());
   std::vector<size_t> bjetIdxs, ljetIdxs;
   for ( auto& fjJet : fjJets )
   {
     if ( abs(fjJet.eta()) > jetMaxEta_ ) continue;
 
     // Get jet constituents from fastJet
     const std::vector<fastjet::PseudoJet> fjConstituents = fastjet::sorted_by_pt(fjJet.constituents());
     // Convert to CandidatePtr
     std::vector<reco::CandidatePtr> constituents;
     bool hasBHadron = false;
     for ( size_t j=0, m=fjConstituents.size(); j<m; ++j )
     { 
       const size_t index = fjConstituents[j].user_index();
       if ( bHadronIdxs.find(index) != bHadronIdxs.end() ) hasBHadron = true;
       reco::CandidatePtr cand = finalStateHandle->ptrAt(index);
       constituents.push_back(cand);
     }
     
     const LorentzVector jetP4(fjJet.px(), fjJet.py(), fjJet.pz(), fjJet.E());
     reco::GenJet genJet;
     reco::writeSpecific(genJet, jetP4, genVertex_, constituents, eventSetup);
 
     const double jetArea = fjJet.has_area() ? fjJet.area() : 0;
     genJet.setJetArea(jetArea);
     if ( hasBHadron )
     {
       genJet.setPdgId(5);
       bjetIdxs.push_back(jets->size());
     }
     else
     {
       ljetIdxs.push_back(jets->size());
     }
 
     jets->push_back(genJet);
   }
 
   // Every building blocks are ready. Continue to pseudo-W and pseudo-top combination
   // NOTE : A C++ trick, use do-while instead of long-nested if-statements.
   do
   {
     // Note : we will do dilepton or semilepton channel only
     const size_t nLepton = leptons->size();
 
     // Collect leptonic-decaying W's
     std::vector<std::pair<int, int> > wCandIdxs;
     for ( auto lep = leptons->begin(); lep != leptons->end(); ++lep )
     {
       const size_t iLep = lep-leptons->begin();
       for ( auto nu = neutrinos->begin(); nu != neutrinos->end(); ++nu )
       {
         //if ( abs(lep->pdgId()+nu->pdgId()) != 1 ) continue; // Enforce to conserve flavour, this reduces combinatorial bkg
         const double m = (lep->p4()+nu->p4()).mass();
         if ( m > 300 ) continue; // Raw mass cut, reduce combinatorial background
 
         const size_t iNu = nu-neutrinos->begin();
         wCandIdxs.push_back(make_pair(iLep, iNu));
       }
     }
 
     if      ( nLepton == 0 or wCandIdxs.empty() ) break; // Skip full hadronic channel
     else if ( nLepton == 1 and wCandIdxs.size() >= 1 ) // Semi-leptonic channel
     {
       double dm = 1e9; // Note: this will be re-used later.
       int bestLepIdx = -1, bestNuIdx = -1;
       for ( auto wCandIdx : wCandIdxs )
       {
         const int lepIdx = wCandIdx.first;
         const int nuIdx  = wCandIdx.second;
         const LorentzVector& lepLVec = leptons->at(lepIdx).p4();
         const LorentzVector& nuLVec = neutrinos->at(nuIdx).p4();
         const double mW = (lepLVec + nuLVec).mass();
         if ( mW > 300 ) continue;
 
         const double dmNew = abs(mW-wMass_);
         if ( dmNew < dm )
         {
           dm = dmNew;
           bestLepIdx = lepIdx;
           bestNuIdx = nuIdx;
         }
       }
       if ( bestLepIdx < 0 ) break; // Actually this should never happen
       const auto& lepton = leptons->at(bestLepIdx);
       const auto& neutrino = neutrinos->at(bestNuIdx);
       const LorentzVector w1LVec = lepton.p4()+neutrino.p4();
 
       // Continue to hadronic W
       // We also collect b jets to be used in the next step.
       dm = 1e9; // Reset dM to very large value.
       int bestJ1Idx = -1, bestJ2Idx = -1;
       for ( auto j1Idx = ljetIdxs.begin(); j1Idx != ljetIdxs.end(); ++j1Idx )
       {
         const auto& j1 = jets->at(*j1Idx);
         for ( auto j2Idx = j1Idx+1; j2Idx != ljetIdxs.end(); ++j2Idx )
         {
           const auto& j2 = jets->at(*j2Idx);
           const double mW = (j1.p4()+j2.p4()).mass();
           if ( mW > 300 ) continue;
 
           const double dmNew = abs(mW-wMass_);
           if ( dmNew < dm )
           {
             dm = dmNew;
             bestJ1Idx = *j1Idx;
             bestJ2Idx = *j2Idx;
           }
         }
       }
       if ( bestJ1Idx < 0 ) break;
       const auto& wJet1 = jets->at(bestJ1Idx);
       const auto& wJet2 = jets->at(bestJ2Idx);
       const LorentzVector w2LVec = wJet1.p4() + wJet2.p4();
 
       // Now we have leptonic W and hadronic W.
       // Contiue to top quarks
       dm = 1e9; // Reset once again for top combination.
       int bestB1Idx = -1, bestB2Idx = -1;
       if ( bjetIdxs.size() < 2 ) break;
       for ( auto b1Idx : bjetIdxs )
       {
         const double t1Mass = (w1LVec + jets->at(b1Idx).p4()).mass();
         if ( t1Mass > 300 ) continue;
         for ( auto b2Idx : bjetIdxs )
         {
           if ( b1Idx == b2Idx ) continue;
           const double t2Mass = (w2LVec + jets->at(b2Idx).p4()).mass();
           if ( t2Mass > 300 ) continue;
 
           const double dmNew = abs(t1Mass-tMass_) + abs(t2Mass-tMass_);
           if ( dmNew < dm )
           {
             dm = dmNew;
             bestB1Idx = b1Idx;
             bestB2Idx = b2Idx;
           }
         }
       }
       if ( bestB1Idx < 0 ) break;
       const auto& bJet1 = jets->at(bestB1Idx);
       const auto& bJet2 = jets->at(bestB2Idx);
       const LorentzVector t1LVec = w1LVec + bJet1.p4();
       const LorentzVector t2LVec = w2LVec + bJet2.p4();
 
       // Put all of them into candidate collection
       if ( true ) // Trick to restrict variables' scope to avoid collision
       {
         const int lepQ = lepton.charge();
 
         reco::GenParticle t1(lepQ*2/3, t1LVec, genVertex_, lepQ*6, 3, false);
         reco::GenParticle w1(lepQ, w1LVec, genVertex_, lepQ*24, 3, true);
         reco::GenParticle b1(0, bJet1.p4(), genVertex_, lepQ*5, 1, true);
         reco::GenParticle l1(lepQ, lepton.p4(), genVertex_, lepton.pdgId(), 1, true);
         reco::GenParticle n1(0, neutrino.p4(), genVertex_, neutrino.pdgId(), 1, true);
 
         reco::GenParticle t2(-lepQ*2/3, t2LVec, genVertex_, -lepQ*6, 3, false);
         reco::GenParticle w2(-lepQ, w2LVec, genVertex_, -lepQ*24, 3, true);
         reco::GenParticle b2(0, bJet2.p4(), genVertex_, -lepQ*5, 1, true);
         reco::GenParticle qA(0, wJet1.p4(), genVertex_, 1, 1, true);
         reco::GenParticle qB(0, wJet2.p4(), genVertex_, 2, 1, true);
 
         pseudoTop->push_back(t1);
         pseudoTop->push_back(w1);
         pseudoTop->push_back(b1);
         pseudoTop->push_back(l1);
         pseudoTop->push_back(n1);
 
         pseudoTop->push_back(t2);
         pseudoTop->push_back(w2);
         pseudoTop->push_back(b2);
         pseudoTop->push_back(qA);
         pseudoTop->push_back(qB);
       }
     }
     else if ( nLepton == 2 and wCandIdxs.size() >= 2 ) // Dilepton channel
     {
       double dm = 1e9;
       int bestLep1Idx = -1, bestLep2Idx = -1, bestNu1Idx = -1, bestNu2Idx = -1;
       for ( auto i : wCandIdxs )
       {
         const auto& lepton1 = leptons->at(i.first);
         if ( lepton1.charge() < 0 ) continue;
         const auto& neutrino1 = neutrinos->at(i.second);
         const double mW1 = (lepton1.p4()+neutrino1.p4()).mass();
         if ( mW1 > 300 ) continue;
         for ( auto j : wCandIdxs )
         {
           if ( i == j ) continue;
           const auto& lepton2 = leptons->at(j.first);
           if ( lepton2.charge() > 0 ) continue;
           const auto& neutrino2 = neutrinos->at(j.second);
           const double mW2 = (lepton2.p4()+neutrino2.p4()).mass();
           if ( mW2 > 300 ) continue;
 
           const double dmNew = abs(mW1-wMass_) + abs(mW2-wMass_);
           if ( dmNew < dm )
           {
             dm = dmNew;
             bestLep1Idx = i.first;
             bestLep2Idx = j.first;
             bestNu1Idx = i.second;
             bestNu2Idx = j.second;
           }
         }
       }
       if ( bestLep1Idx < 0 ) break;
       const auto& lepton1 = leptons->at(bestLep1Idx);
       const auto& lepton2 = leptons->at(bestLep2Idx);
       const auto& neutrino1 = neutrinos->at(bestNu1Idx);
       const auto& neutrino2 = neutrinos->at(bestNu2Idx);
       const LorentzVector w1LVec = lepton1.p4()+neutrino1.p4();
       const LorentzVector w2LVec = lepton2.p4()+neutrino2.p4();
 
       // Contiue to top quarks
       dm = 1e9; // Reset once again for top combination.
       int bestB1Idx = -1, bestB2Idx = -1;
       if ( bjetIdxs.size() < 2 ) break;
       for ( auto b1Idx : bjetIdxs )
       {
         const double t1Mass = (w1LVec + jets->at(b1Idx).p4()).mass();
         if ( t1Mass > 300 ) continue;
         for ( auto b2Idx : bjetIdxs )
         {
           if ( b1Idx == b2Idx ) continue;
           const double t2Mass = (w2LVec + jets->at(b2Idx).p4()).mass();
           if ( t2Mass > 300 ) continue;
 
           const double dmNew = abs(t1Mass-tMass_) + abs(t2Mass-tMass_);
           if ( dmNew < dm )
           {
             dm = dmNew;
             bestB1Idx = b1Idx;
             bestB2Idx = b2Idx;
           }
         }
       }
       if ( bestB1Idx < 0 ) break;
       const auto& bJet1 = jets->at(bestB1Idx);
       const auto& bJet2 = jets->at(bestB2Idx);
       const LorentzVector t1LVec = w1LVec + bJet1.p4();
       const LorentzVector t2LVec = w2LVec + bJet2.p4();
 
       // Put all of them into candidate collection
       if ( true ) // Trick to restrict variables' scope to avoid collision
       {
         const int lep1Q = lepton1.charge();
         const int lep2Q = lepton2.charge();
 
         reco::GenParticle t1(lep1Q*2/3, t1LVec, genVertex_, lep1Q*6, 3, true);
         reco::GenParticle w1(lep1Q, w1LVec, genVertex_, lep1Q*24, 3, true);
         reco::GenParticle b1(0, bJet1.p4(), genVertex_, lep1Q*5, 1, true);
         reco::GenParticle l1(lep1Q, lepton1.p4(), genVertex_, lepton1.pdgId(), 1, true);
         reco::GenParticle n1(0, neutrino1.p4(), genVertex_, neutrino1.pdgId(), 1, true);
 
         reco::GenParticle t2(lep2Q*2/3, t2LVec, genVertex_, lep2Q*6, 3, true);
         reco::GenParticle w2(lep2Q, w2LVec, genVertex_, lep2Q*24, 3, true);
         reco::GenParticle b2(0, bJet2.p4(), genVertex_, lep2Q*5, 1, true);
         reco::GenParticle l2(0, lepton2.p4(), genVertex_, lepton2.pdgId(), 1, true);
         reco::GenParticle n2(0, neutrino2.p4(), genVertex_, neutrino2.pdgId(), 1, true);
 
         pseudoTop->push_back(t1);
         pseudoTop->push_back(w1);
         pseudoTop->push_back(b1);
         pseudoTop->push_back(l1);
         pseudoTop->push_back(n1);
 
         pseudoTop->push_back(t2);
         pseudoTop->push_back(w2);
         pseudoTop->push_back(b2);
         pseudoTop->push_back(l2);
         pseudoTop->push_back(n2);
       }
     }
 
     if ( pseudoTop->size() == 10 ) // If pseudtop decay tree is completed
     {
       // t->W+b
       pseudoTop->at(0).addDaughter(reco::GenParticleRef(pseudoTopRefHandle, 1)); // t->W
       pseudoTop->at(0).addDaughter(reco::GenParticleRef(pseudoTopRefHandle, 2)); // t->b
       pseudoTop->at(1).addMother(reco::GenParticleRef(pseudoTopRefHandle, 0)); // t->W
       pseudoTop->at(2).addMother(reco::GenParticleRef(pseudoTopRefHandle, 0)); // t->b
 
       // W->lv or W->jj
       pseudoTop->at(1).addDaughter(reco::GenParticleRef(pseudoTopRefHandle, 3));
       pseudoTop->at(1).addDaughter(reco::GenParticleRef(pseudoTopRefHandle, 4));
       pseudoTop->at(3).addMother(reco::GenParticleRef(pseudoTopRefHandle, 1));
       pseudoTop->at(4).addMother(reco::GenParticleRef(pseudoTopRefHandle, 1));
 
       // tbar->W-b
       pseudoTop->at(5).addDaughter(reco::GenParticleRef(pseudoTopRefHandle, 6));
       pseudoTop->at(5).addDaughter(reco::GenParticleRef(pseudoTopRefHandle, 7));
       pseudoTop->at(6).addMother(reco::GenParticleRef(pseudoTopRefHandle, 5));
       pseudoTop->at(7).addMother(reco::GenParticleRef(pseudoTopRefHandle, 5));
 
       // W->jj
       pseudoTop->at(6).addDaughter(reco::GenParticleRef(pseudoTopRefHandle, 8));
       pseudoTop->at(6).addDaughter(reco::GenParticleRef(pseudoTopRefHandle, 9));
       pseudoTop->at(8).addMother(reco::GenParticleRef(pseudoTopRefHandle, 6));
       pseudoTop->at(9).addMother(reco::GenParticleRef(pseudoTopRefHandle, 6));
     }
   } while (false);
 
   event.put(neutrinos, "neutrinos");
   event.put(leptons, "leptons");
   event.put(jets, "jets");
 
   event.put(pseudoTop);
 }

Member Data Documentation

edm::EDGetTokenT<edm::View<reco::Candidate> > PseudoTopProducer::finalStateToken_

private

Definition at line 31 of file PseudoTopProducer.h.

Referenced by produce(), and PseudoTopProducer().

std::shared_ptr<JetDef> PseudoTopProducer::fjJetDef_

private

Definition at line 37 of file PseudoTopProducer.h.

Referenced by produce(), and PseudoTopProducer().

std::shared_ptr<JetDef> PseudoTopProducer::fjLepDef_

private

Definition at line 37 of file PseudoTopProducer.h.

Referenced by produce(), and PseudoTopProducer().

edm::EDGetTokenT<edm::View<reco::Candidate> > PseudoTopProducer::genParticleToken_

private

Definition at line 32 of file PseudoTopProducer.h.

Referenced by produce(), and PseudoTopProducer().

reco::Particle::Point PseudoTopProducer::genVertex_

private

Definition at line 38 of file PseudoTopProducer.h.

Referenced by produce(), and PseudoTopProducer().

const double PseudoTopProducer::jetMaxEta_

private

Definition at line 33 of file PseudoTopProducer.h.

Referenced by produce().

const double PseudoTopProducer::jetMinPt_

private

Definition at line 33 of file PseudoTopProducer.h.

Referenced by produce().

const double PseudoTopProducer::leptonMaxEta_

private

Definition at line 33 of file PseudoTopProducer.h.

Referenced by produce().

const double PseudoTopProducer::leptonMinPt_

private

Definition at line 33 of file PseudoTopProducer.h.

Referenced by produce().

const double PseudoTopProducer::tMass_

private

Definition at line 34 of file PseudoTopProducer.h.

Referenced by produce().

const double PseudoTopProducer::wMass_

private

Definition at line 34 of file PseudoTopProducer.h.

Referenced by produce().

Public Member Functions

Private Types

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Member Typedef Documentation

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation