#include <TtSemiLepJetCombGeom.h>

Inheritance diagram for TtSemiLepJetCombGeom:

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

	~TtSemiLepJetCombGeom () override

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

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)

virtual	~ProducerBase () noexcept(false)

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

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

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

virtual	~EDConsumerBase () noexcept(false)

Private Member Functions
void	beginJob () override

double	distance (const math::XYZTLorentzVector &, const math::XYZTLorentzVector &)

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_

double	minBDiscBJets_

bool	useBTagging_

bool	useDeltaR_

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)

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 10 of file TtSemiLepJetCombGeom.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 9 of file TtSemiLepJetCombGeom.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"             ))),
   maxNJets_         (cfg.getParameter<int>          ("maxNJets"         )),
   useDeltaR_        (cfg.getParameter<bool>         ("useDeltaR"        )),
   useBTagging_      (cfg.getParameter<bool>         ("useBTagging"      )),
   bTagAlgorithm_    (cfg.getParameter<std::string>  ("bTagAlgorithm"    )),
   minBDiscBJets_    (cfg.getParameter<double>       ("minBDiscBJets"    )),
   maxBDiscLightJets_(cfg.getParameter<double>       ("maxBDiscLightJets"))
 {
   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");
 }

TtSemiLepJetCombGeom::~TtSemiLepJetCombGeom ( )

override

Definition at line 28 of file TtSemiLepJetCombGeom.cc.

29 {

30 }

Member Function Documentation

void TtSemiLepJetCombGeom::beginJob ( void )

inlineoverrideprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 19 of file TtSemiLepJetCombGeom.h.

References produce(), and GeneralSetup::setup().

19 {};

double TtSemiLepJetCombGeom::distance	(	const math::XYZTLorentzVector &	v1,
		const math::XYZTLorentzVector &	v2
	)

private

Definition at line 147 of file TtSemiLepJetCombGeom.cc.

References HiRegitMuonDetachedTripletStep_cff::DeltaR, and useDeltaR_.

Referenced by isValid(), and produce().

 {
   // calculate the distance between two lorentz vectors
   // using DeltaR or DeltaTheta
   if(useDeltaR_) return ROOT::Math::VectorUtil::DeltaR(v1, v2);
   return fabs(v1.theta() - v2.theta());
 }

void TtSemiLepJetCombGeom::endJob ( void )

inlineoverrideprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 21 of file TtSemiLepJetCombGeom.h.

21 {};

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

inlineprivate

Definition at line 23 of file TtSemiLepJetCombGeom.h.

References distance(), and fwrapper::jets.

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

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

training_settings.idx

idx

Definition: training_settings.py:16

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

overrideprivate

Definition at line 33 of file TtSemiLepJetCombGeom.cc.

References bTagAlgorithm_, distance(), edm::Event::getByToken(), TtSemiLepEvtPartons::HadB, mps_fire::i, training_settings::idx, isValid(), fwrapper::jets, jetsToken_, TtSemiLepEvtPartons::LepB, TtSemiLepHitFitProducer_Electrons_cfi::leps, lepsToken_, TtSemiLepEvtPartons::LightQ, TtSemiLepEvtPartons::LightQBar, match(), maxBDiscLightJets_, TtFullLepEvtBuilder_cfi::maxNJets, maxNJets_, minBDiscBJets_, eostools::move(), p4, edm::Event::put(), and useBTagging_.

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);
 
   // skip events without lepton candidate or less than 4 jets
   if(leps->empty() || jets->size() < 4){
     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 two jets to the hadronic W boson that
   // have the smallest distance to each other
   // -----------------------------------------------------
   double minDist=-1.;
   int lightQ   =-1;
   int lightQBar=-1;
   for(unsigned int idx=0; idx<maxNJets; ++idx){
     if(useBTagging_ && (!isLJet[idx] || (cntBJets<=2 && isBJet[idx]))) continue;
     for(unsigned int jdx=(idx+1); jdx<maxNJets; ++jdx){
       if(useBTagging_ && (!isLJet[jdx] || (cntBJets<=2 && isBJet[jdx]) || (cntBJets==3 && isBJet[idx] && isBJet[jdx]))) continue;
       double dist = distance((*jets)[idx].p4(), (*jets)[jdx].p4());
       if( minDist<0. || dist<minDist ){
     minDist=dist;
     lightQ   =idx;
     lightQBar=jdx;
       }
     }
   }
 
   reco::Particle::LorentzVector wHad;
   if( isValid(lightQ, jets) && isValid(lightQBar, jets) )
     wHad = (*jets)[lightQ].p4() + (*jets)[lightQBar].p4();
 
   // -----------------------------------------------------
   // associate to the hadronic b quark the remaining jet
   // that has the smallest distance to the hadronic W
   // -----------------------------------------------------
   minDist=-1.;
   int hadB=-1;
   if( isValid(lightQ, jets) && isValid(lightQBar, jets) ) {
     for(unsigned int 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!=lightQ && (int)idx!=lightQBar ) {
     double dist = distance((*jets)[idx].p4(), wHad);
     if( minDist<0. || dist<minDist ){
       minDist=dist;
       hadB=idx;
     }
       }
     }
   }
 
   // -----------------------------------------------------
   // associate to the leptonic b quark the remaining jet
   // that has the smallest distance to the leading lepton
   // -----------------------------------------------------
   minDist=-1.;
   int lepB=-1;
   for(unsigned int idx=0; idx<maxNJets; ++idx){
     if(useBTagging_ && !isBJet[idx]) continue;
     // make sure it's not used up already from the hadronic decay chain
     if( (int)idx!=lightQ && (int)idx!=lightQBar && (int)idx!=hadB ){
       double dist = distance((*jets)[idx].p4(), (*leps)[0].p4());
       if( minDist<0. || dist<minDist ){
     minDist=dist;
     lepB=idx;
       }
     }
   }
 
   match[TtSemiLepEvtPartons::LightQ   ] = lightQ;
   match[TtSemiLepEvtPartons::LightQBar] = lightQBar;
   match[TtSemiLepEvtPartons::HadB     ] = hadB;
   match[TtSemiLepEvtPartons::LepB     ] = lepB;
 
   pOut->push_back( match );
   evt.put(std::move(pOut));
 }

Member Data Documentation

std::string TtSemiLepJetCombGeom::bTagAlgorithm_

private

Definition at line 31 of file TtSemiLepJetCombGeom.h.

Referenced by produce().

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

private

Definition at line 26 of file TtSemiLepJetCombGeom.h.

Referenced by produce().

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

private

Definition at line 27 of file TtSemiLepJetCombGeom.h.

Referenced by produce().

double TtSemiLepJetCombGeom::maxBDiscLightJets_

private

Definition at line 33 of file TtSemiLepJetCombGeom.h.

Referenced by produce().

int TtSemiLepJetCombGeom::maxNJets_

private

Definition at line 28 of file TtSemiLepJetCombGeom.h.

Referenced by produce(), and TtSemiLepJetCombGeom().

double TtSemiLepJetCombGeom::minBDiscBJets_

private

Definition at line 32 of file TtSemiLepJetCombGeom.h.

Referenced by produce().

bool TtSemiLepJetCombGeom::useBTagging_

private

Definition at line 30 of file TtSemiLepJetCombGeom.h.

Referenced by produce().

bool TtSemiLepJetCombGeom::useDeltaR_

private

Definition at line 29 of file TtSemiLepJetCombGeom.h.

Referenced by distance().

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation