#include <TtSemiLepJetCombWMassDeltaTopMass.h>

Inheritance diagram for TtSemiLepJetCombWMassDeltaTopMass:

Public Member Functions
	TtSemiLepJetCombWMassDeltaTopMass (const edm::ParameterSet &)

	~TtSemiLepJetCombWMassDeltaTopMass () override

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

SerialTaskQueue *	globalLuminosityBlocksQueue ()

SerialTaskQueue *	globalRunsQueue ()

ModuleDescription const &	moduleDescription () const

	~EDProducer () 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	beginJob () override

void	endJob () override

bool	isValid (const int &idx, const edm::Handle< std::vector< pat::Jet > > &jets)

void	produce (edm::Event &evt, const edm::EventSetup &setup) override

Private Attributes
std::string	bTagAlgorithm_

edm::EDGetTokenT< std::vector< pat::Jet > >	jetsToken_

edm::EDGetTokenT< edm::View< reco::RecoCandidate > >	lepsToken_

double	maxBDiscLightJets_

int	maxNJets_

edm::EDGetTokenT< std::vector< pat::MET > >	metsToken_

double	minBDiscBJets_

int	neutrinoSolutionType_

bool	useBTagging_

double	wMass_

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	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::EDProducer
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

static bool	wantsGlobalLuminosityBlocks ()

static bool	wantsGlobalRuns ()

static bool	wantsStreamLuminosityBlocks ()

static bool	wantsStreamRuns ()

Protected Member Functions inherited from edm::ProducerBase
ProducesCollector	producesCollector ()

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

Constructor & Destructor Documentation

TtSemiLepJetCombWMassDeltaTopMass::TtSemiLepJetCombWMassDeltaTopMass ( const edm::ParameterSet & cfg )

explicit

Definition at line 7 of file TtSemiLepJetCombWMassDeltaTopMass.cc.

References Exception, and maxNJets_.

     : jetsToken_(consumes<std::vector<pat::Jet> >(cfg.getParameter<edm::InputTag>("jets"))),
       lepsToken_(consumes<edm::View<reco::RecoCandidate> >(cfg.getParameter<edm::InputTag>("leps"))),
       metsToken_(consumes<std::vector<pat::MET> >(cfg.getParameter<edm::InputTag>("mets"))),
       maxNJets_(cfg.getParameter<int>("maxNJets")),
       wMass_(cfg.getParameter<double>("wMass")),
       useBTagging_(cfg.getParameter<bool>("useBTagging")),
       bTagAlgorithm_(cfg.getParameter<std::string>("bTagAlgorithm")),
       minBDiscBJets_(cfg.getParameter<double>("minBDiscBJets")),
       maxBDiscLightJets_(cfg.getParameter<double>("maxBDiscLightJets")),
       neutrinoSolutionType_(cfg.getParameter<int>("neutrinoSolutionType")) {
   if (maxNJets_ < 4 && maxNJets_ != -1)
     throw cms::Exception("WrongConfig") << "Parameter maxNJets can not be set to " << maxNJets_ << ". \n"
                                         << "It has to be larger than 4 or can be set to -1 to take all jets.";
 
   produces<std::vector<std::vector<int> > >();
   produces<int>("NumberOfConsideredJets");
 }

TtSemiLepJetCombWMassDeltaTopMass::~TtSemiLepJetCombWMassDeltaTopMass ( )

override

Definition at line 26 of file TtSemiLepJetCombWMassDeltaTopMass.cc.

26 {}

Member Function Documentation

void TtSemiLepJetCombWMassDeltaTopMass::beginJob ( void )

inlineoverrideprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 19 of file TtSemiLepJetCombWMassDeltaTopMass.h.

References produce(), and singleTopDQM_cfi::setup.

19 {};

void TtSemiLepJetCombWMassDeltaTopMass::endJob ( void )

inlineoverrideprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 21 of file TtSemiLepJetCombWMassDeltaTopMass.h.

21 {};

bool TtSemiLepJetCombWMassDeltaTopMass::isValid	(	const int &	idx,
		const edm::Handle< std::vector< pat::Jet > > &	jets
	)

inlineprivate

Definition at line 23 of file TtSemiLepJetCombWMassDeltaTopMass.h.

References singleTopDQM_cfi::jets.

Referenced by ntupleDataFormat._Object::_checkIsValid(), produce(), and core.AutoHandle.AutoHandle::ReallyLoad().

                                                                           {
     return (0 <= idx && idx < (int)jets->size());
   };

void TtSemiLepJetCombWMassDeltaTopMass::produce	(	edm::Event &	evt,
		const edm::EventSetup &	setup
	)

overrideprivate

Definition at line 28 of file TtSemiLepJetCombWMassDeltaTopMass.cc.

References bTagAlgorithm_, MEzCalculator::Calculate(), Exception, edm::Event::getByToken(), TtSemiLepEvtPartons::HadB, mps_fire::i, training_settings::idx, isValid(), singleTopDQM_cfi::jets, jetsToken_, TtSemiLepEvtPartons::LepB, TtSemiLepHitFitProducer_Electrons_cfi::leps, lepsToken_, TtSemiLepEvtPartons::LightQ, TtSemiLepEvtPartons::LightQBar, match(), maxBDiscLightJets_, SUSYDQMAnalyzer_cfi::maxNJets, maxNJets_, singleTopDQM_cfi::mets, metsToken_, minBDiscBJets_, eostools::move(), neutrinoSolutionType_, edm::Event::put(), MEzCalculator::SetLepton(), MEzCalculator::SetMET(), mathSSE::sqrt(), useBTagging_, and wMass_.

Referenced by beginJob().

                                                                                          {
   std::unique_ptr<std::vector<std::vector<int> > > pOut(new std::vector<std::vector<int> >);
   std::unique_ptr<int> pJetsConsidered(new int);
 
   std::vector<int> match;
   for (unsigned int i = 0; i < 4; ++i)
     match.push_back(-1);
 
   // get jets
   edm::Handle<std::vector<pat::Jet> > jets;
   evt.getByToken(jetsToken_, jets);
 
   // get leptons
   edm::Handle<edm::View<reco::RecoCandidate> > leps;
   evt.getByToken(lepsToken_, leps);
 
   // get MET
   edm::Handle<std::vector<pat::MET> > mets;
   evt.getByToken(metsToken_, mets);
 
   // skip events without lepton candidate or less than 4 jets or no MET
   if (leps->empty() || jets->size() < 4 || mets->empty()) {
     pOut->push_back(match);
     evt.put(std::move(pOut));
     *pJetsConsidered = jets->size();
     evt.put(std::move(pJetsConsidered), "NumberOfConsideredJets");
     return;
   }
 
   unsigned maxNJets = maxNJets_;
   if (maxNJets_ == -1 || (int)jets->size() < maxNJets_)
     maxNJets = jets->size();
   *pJetsConsidered = maxNJets;
   evt.put(std::move(pJetsConsidered), "NumberOfConsideredJets");
 
   std::vector<bool> isBJet;
   std::vector<bool> isLJet;
   int cntBJets = 0;
   if (useBTagging_) {
     for (unsigned int idx = 0; idx < maxNJets; ++idx) {
       isBJet.push_back(((*jets)[idx].bDiscriminator(bTagAlgorithm_) > minBDiscBJets_));
       isLJet.push_back(((*jets)[idx].bDiscriminator(bTagAlgorithm_) < maxBDiscLightJets_));
       if ((*jets)[idx].bDiscriminator(bTagAlgorithm_) > minBDiscBJets_)
         cntBJets++;
     }
   }
 
   // -----------------------------------------------------
   // associate those jets that get closest to the W mass
   // with their invariant mass to the hadronic W boson
   // -----------------------------------------------------
   double wDist = -1.;
   std::vector<int> closestToWMassIndices;
   closestToWMassIndices.push_back(-1);
   closestToWMassIndices.push_back(-1);
   for (unsigned idx = 0; idx < maxNJets; ++idx) {
     if (useBTagging_ && (!isLJet[idx] || (cntBJets <= 2 && isBJet[idx])))
       continue;
     for (unsigned jdx = (idx + 1); jdx < maxNJets; ++jdx) {
       if (useBTagging_ &&
           (!isLJet[jdx] || (cntBJets <= 2 && isBJet[jdx]) || (cntBJets == 3 && isBJet[idx] && isBJet[jdx])))
         continue;
       reco::Particle::LorentzVector sum = (*jets)[idx].p4() + (*jets)[jdx].p4();
       if (wDist < 0. || wDist > fabs(sum.mass() - wMass_)) {
         wDist = fabs(sum.mass() - wMass_);
         closestToWMassIndices.clear();
         closestToWMassIndices.push_back(idx);
         closestToWMassIndices.push_back(jdx);
       }
     }
   }
 
   // -----------------------------------------------------
   // build a leptonic W boson from the lepton and the MET
   // -----------------------------------------------------
   double neutrino_pz = 0.;
   if (neutrinoSolutionType_ != -1) {
     MEzCalculator mez;
     mez.SetMET(*mets->begin());
     if (dynamic_cast<const reco::Muon*>(&(leps->front())))
       mez.SetLepton((*leps)[0], true);
     else if (dynamic_cast<const reco::GsfElectron*>(&(leps->front())))
       mez.SetLepton((*leps)[0], false);
     else
       throw cms::Exception("UnimplementedFeature")
           << "Type of lepton given together with MET for solving neutrino kinematics is neither muon nor electron.\n";
     neutrino_pz = mez.Calculate(neutrinoSolutionType_);
   }
   const math::XYZTLorentzVector neutrino(mets->begin()->px(),
                                          mets->begin()->py(),
                                          neutrino_pz,
                                          sqrt(mets->begin()->px() * mets->begin()->px() +
                                               mets->begin()->py() * mets->begin()->py() + neutrino_pz * neutrino_pz));
   const reco::Particle::LorentzVector lepW = neutrino + leps->front().p4();
 
   // -----------------------------------------------------
   // associate those jets to the hadronic and the leptonic
   // b quark that minimize the difference between
   // hadronic and leptonic top mass
   // -----------------------------------------------------
   double deltaTop = -1.;
   int hadB = -1;
   int lepB = -1;
   if (isValid(closestToWMassIndices[0], jets) && isValid(closestToWMassIndices[1], jets)) {
     const reco::Particle::LorentzVector hadW =
         (*jets)[closestToWMassIndices[0]].p4() + (*jets)[closestToWMassIndices[1]].p4();
     // find hadronic b candidate
     for (unsigned idx = 0; idx < maxNJets; ++idx) {
       if (useBTagging_ && !isBJet[idx])
         continue;
       // make sure it's not used up already from the hadronic W
       if ((int)idx != closestToWMassIndices[0] && (int)idx != closestToWMassIndices[1]) {
         reco::Particle::LorentzVector hadTop = hadW + (*jets)[idx].p4();
         // find leptonic b candidate
         for (unsigned jdx = 0; jdx < maxNJets; ++jdx) {
           if (useBTagging_ && !isBJet[jdx])
             continue;
           // make sure it's not used up already from the hadronic branch
           if ((int)jdx != closestToWMassIndices[0] && (int)jdx != closestToWMassIndices[1] && jdx != idx) {
             reco::Particle::LorentzVector lepTop = lepW + (*jets)[jdx].p4();
             if (deltaTop < 0. || deltaTop > fabs(hadTop.mass() - lepTop.mass())) {
               deltaTop = fabs(hadTop.mass() - lepTop.mass());
               hadB = idx;
               lepB = jdx;
             }
           }
         }
       }
     }
   }
 
   match[TtSemiLepEvtPartons::LightQ] = closestToWMassIndices[0];
   match[TtSemiLepEvtPartons::LightQBar] = closestToWMassIndices[1];
   match[TtSemiLepEvtPartons::HadB] = hadB;
   match[TtSemiLepEvtPartons::LepB] = lepB;
 
   pOut->push_back(match);
   evt.put(std::move(pOut));
 }