#include <TtSemiLepHypMaxSumPtWMass.h>

Inheritance diagram for TtSemiLepHypMaxSumPtWMass:

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

	~TtSemiLepHypMaxSumPtWMass ()

Public Member Functions inherited from TtSemiLepHypothesis
	TtSemiLepHypothesis (const edm::ParameterSet &)
	default constructor More...

	~TtSemiLepHypothesis ()
	default destructor More...

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

virtual	~EDProducer ()

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

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

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

virtual	~ProducerBase ()

Private Member Functions
virtual void	buildHypo (edm::Event &, const edm::Handle< edm::View< reco::RecoCandidate > > &, const edm::Handle< std::vector< pat::MET > > &, const edm::Handle< std::vector< pat::Jet > > &, std::vector< int > &, const unsigned int iComb)
	build event hypothesis from the reco objects of a semi-leptonic event More...

virtual void	buildKey ()
	build the event hypothesis key More...

Private Attributes
std::string	bTagAlgorithm_

double	maxBDiscLightJets_

int	maxNJets_

double	minBDiscBJets_

int	neutrinoSolutionType_

bool	useBTagging_

double	wMass_

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

typedef WorkerT< EDProducer >	WorkerType

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)

Protected Member Functions inherited from TtSemiLepHypothesis
reco::CompositeCandidate	hypo ()
	return event hypothesis More...

bool	isValid (const int &idx, const edm::Handle< std::vector< pat::Jet > > &jets)
	check if index is in valid range of selected jets More...

std::string	jetCorrectionLevel (const std::string &quarkType)
	helper function to construct the proper correction level string for corresponding quarkType More...

int	key () const
	return key More...

WDecay::LeptonType	leptonType (const reco::RecoCandidate *cand)
	determine lepton type of reco candidate and return a corresponding WDecay::LeptonType; the type is kNone if it is whether a muon nor an electron More...

virtual void	produce (edm::Event &, const edm::EventSetup &)
	produce the event hypothesis as CompositeCandidate and Key More...

void	resetCandidates ()
	reset candidate pointers before hypo build process More...

template<typename C >
void	setCandidate (const edm::Handle< C > &handle, const int &idx, reco::ShallowClonePtrCandidate *&clone)
	use one object in a collection to set a ShallowClonePtrCandidate More...

void	setCandidate (const edm::Handle< std::vector< pat::Jet > > &handle, const int &idx, reco::ShallowClonePtrCandidate *&clone, const std::string &correctionLevel)
	use one object in a jet collection to set a ShallowClonePtrCandidate with proper jet corrections More...

void	setNeutrino (const edm::Handle< std::vector< pat::MET > > &met, const edm::Handle< edm::View< reco::RecoCandidate > > &leps, const int &idx, const int &type)
	set neutrino, using mW = 80.4 to calculate the neutrino pz More...

Protected Member Functions inherited from edm::EDProducer
CurrentProcessingContext const *	currentContext () const

Protected Member Functions inherited from edm::ProducerBase
template<class TProducer , class TMethod >
void	callWhenNewProductsRegistered (TProducer *iProd, TMethod iMethod)

Protected Attributes inherited from TtSemiLepHypothesis
bool	getMatch_

reco::ShallowClonePtrCandidate *	hadronicB_

std::string	jetCorrectionLevel_

edm::InputTag	jets_
	input label for all necessary collections More...

int	key_
	hypothesis key (to be set by the buildKey function) More...

edm::InputTag	leps_

reco::ShallowClonePtrCandidate *	lepton_

reco::ShallowClonePtrCandidate *	leptonicB_

reco::ShallowClonePtrCandidate *	lightQ_

reco::ShallowClonePtrCandidate *	lightQBar_

edm::InputTag	match_

edm::InputTag	mets_

reco::ShallowClonePtrCandidate *	neutrino_

int	numberOfRealNeutrinoSolutions_

Detailed Description

Definition at line 7 of file TtSemiLepHypMaxSumPtWMass.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 4 of file TtSemiLepHypMaxSumPtWMass.cc.

References edm::hlt::Exception, and maxNJets_.

                                                                               :
   TtSemiLepHypothesis( cfg ),
   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.";
 }

TtSemiLepHypMaxSumPtWMass::~TtSemiLepHypMaxSumPtWMass ( )

Definition at line 20 of file TtSemiLepHypMaxSumPtWMass.cc.

20 { }

Member Function Documentation

void TtSemiLepHypMaxSumPtWMass::buildHypo	(	edm::Event &	evt,
		const edm::Handle< edm::View< reco::RecoCandidate > > &	leps,
		const edm::Handle< std::vector< pat::MET > > &	mets,
		const edm::Handle< std::vector< pat::Jet > > &	jets,
		std::vector< int > &	match,
		const unsigned int	iComb
	)

privatevirtual

build event hypothesis from the reco objects of a semi-leptonic event

Implements TtSemiLepHypothesis.

Definition at line 23 of file TtSemiLepHypMaxSumPtWMass.cc.

References bTagAlgorithm_, spr::find(), TtSemiLepEvtPartons::HadB, TtSemiLepHypothesis::hadronicB_, i, TtSemiLepHypothesis::isValid(), TtSemiLepHypothesis::jetCorrectionLevel(), analyzePatCleaning_cfg::jets, TtSemiLepEvtPartons::LepB, TtSemiLepEvtPartons::Lepton, TtSemiLepHypothesis::lepton_, TtSemiLepHypothesis::leptonicB_, TtSemiLepEvtPartons::LightQ, TtSemiLepHypothesis::lightQ_, TtSemiLepEvtPartons::LightQBar, TtSemiLepHypothesis::lightQBar_, maxBDiscLightJets_, maxNJets_, minBDiscBJets_, TtSemiLepHypothesis::neutrino_, neutrinoSolutionType_, TtSemiLepHypothesis::setCandidate(), TtSemiLepHypothesis::setNeutrino(), useBTagging_, and wMass_.

 {
   if(leps->empty() || mets->empty() || jets->size()<4){
     // create empty hypothesis
     return;
   }
 
   int maxNJets = maxNJets_;
   if(maxNJets_ == -1 || (int)jets->size() < maxNJets_) maxNJets = jets->size();
 
   std::vector<bool> isBJet;
   std::vector<bool> isLJet;
   int cntBJets = 0;
   if(useBTagging_) {
     for(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++;
     }
   }
 
   match.clear();
   for(int i=0; i<5; ++i)
     match.push_back(-1);
 
   // -----------------------------------------------------
   // associate those jets with maximum pt of the vectorial 
   // sum to the hadronic decay chain
   // -----------------------------------------------------
   double maxPt=-1.;
   std::vector<int> maxPtIndices;
   maxPtIndices.push_back(-1);
   maxPtIndices.push_back(-1);
   maxPtIndices.push_back(-1);
   for(int idx=0; idx<maxNJets; ++idx){
     if(useBTagging_ && (!isLJet[idx] || (cntBJets<=2 && isBJet[idx]))) continue;
     for(int jdx=(idx+1); jdx<maxNJets; ++jdx){
       if(jdx==idx || (useBTagging_ && (!isLJet[jdx] || (cntBJets<=2 && isBJet[jdx]) || (cntBJets==3 && isBJet[idx] && isBJet[jdx])))) continue;
       for(int kdx=0; kdx<maxNJets; ++kdx){
     if(kdx==idx || kdx==jdx || (useBTagging_ && !isBJet[kdx])) continue;
     reco::Particle::LorentzVector sum = 
       (*jets)[idx].p4()+
       (*jets)[jdx].p4()+
       (*jets)[kdx].p4();
     if( maxPt<0. || maxPt<sum.pt() ){
       maxPt=sum.pt();
       maxPtIndices.clear();
       maxPtIndices.push_back(idx);
       maxPtIndices.push_back(jdx);
       maxPtIndices.push_back(kdx);
     }
       }
     }
   }
 
   // -----------------------------------------------------
   // associate those jets that get closest to the W mass
   // with their invariant mass to the W boson
   // -----------------------------------------------------
   double wDist =-1.;
   std::vector<int> closestToWMassIndices;
   closestToWMassIndices.push_back(-1);
   closestToWMassIndices.push_back(-1);
   if( isValid(maxPtIndices[0], jets) && isValid(maxPtIndices[1], jets) && isValid(maxPtIndices[2], jets)) {
     for(unsigned idx=0; idx<maxPtIndices.size(); ++idx){  
       for(unsigned jdx=0; jdx<maxPtIndices.size(); ++jdx){  
     if( jdx==idx || maxPtIndices[idx]>maxPtIndices[jdx] || (useBTagging_ && (!isLJet[maxPtIndices[idx]] || !isLJet[maxPtIndices[jdx]] || (cntBJets<=2 && isBJet[maxPtIndices[idx]]) || (cntBJets<=2 && isBJet[maxPtIndices[jdx]]) || (cntBJets==3 && isBJet[maxPtIndices[idx]] && isBJet[maxPtIndices[jdx]])))) continue;
     reco::Particle::LorentzVector sum = 
       (*jets)[maxPtIndices[idx]].p4()+
       (*jets)[maxPtIndices[jdx]].p4();
     if( wDist<0. || wDist>fabs(sum.mass()-wMass_) ){
       wDist=fabs(sum.mass()-wMass_);
       closestToWMassIndices.clear();
       closestToWMassIndices.push_back(maxPtIndices[idx]);
       closestToWMassIndices.push_back(maxPtIndices[jdx]);
     }
       }
     }
   }
 
   // -----------------------------------------------------
   // associate the remaining jet with maximum pt of the   
   // vectorial sum with the leading lepton with the 
   // leptonic decay chain
   // -----------------------------------------------------
   maxPt=-1.;
   int lepB=-1;
   for(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( std::find(maxPtIndices.begin(), maxPtIndices.end(), idx) == maxPtIndices.end() ){
       reco::Particle::LorentzVector sum = 
     (*jets)[idx].p4()+(*leps)[ 0 ].p4();
       if( maxPt<0. || maxPt<sum.pt() ){
     maxPt=sum.pt();
     lepB=idx;
       }
     }
   }
 
   // -----------------------------------------------------
   // add jets
   // -----------------------------------------------------
   if( isValid(closestToWMassIndices[0], jets) ){
     setCandidate(jets, closestToWMassIndices[0], lightQ_, jetCorrectionLevel("wQuarkMix"));
     match[TtSemiLepEvtPartons::LightQ] = closestToWMassIndices[0];
   }
 
   if( isValid(closestToWMassIndices[1], jets) ){
     setCandidate(jets, closestToWMassIndices[1], lightQBar_, jetCorrectionLevel("wQuarkMix"));
     match[TtSemiLepEvtPartons::LightQBar] = closestToWMassIndices[1];
   }
 
   for(unsigned idx=0; idx<maxPtIndices.size(); ++idx){
     // if this idx is not yet contained in the list of W mass candidates...
     if( std::find( closestToWMassIndices.begin(), closestToWMassIndices.end(), maxPtIndices[idx]) == closestToWMassIndices.end() ){
       // ...and if it is valid
       if( isValid(maxPtIndices[idx], jets) ){
     setCandidate(jets, maxPtIndices[idx], hadronicB_, jetCorrectionLevel("bQuark"));
     match[TtSemiLepEvtPartons::HadB] = maxPtIndices[idx];
     break; // there should be no other cadidates!
       }
     }
   }
 
   if( isValid(lepB, jets) ){
     setCandidate(jets, lepB, leptonicB_, jetCorrectionLevel("bQuark"));
     match[TtSemiLepEvtPartons::LepB] = lepB;
   }
 
   // -----------------------------------------------------
   // add lepton
   // -----------------------------------------------------
   setCandidate(leps, 0, lepton_);
   match[TtSemiLepEvtPartons::Lepton] = 0;
   
   // -----------------------------------------------------
   // add neutrino
   // -----------------------------------------------------
   if(neutrinoSolutionType_ == -1)
     setCandidate(mets, 0, neutrino_);
   else
     setNeutrino(mets, leps, 0, neutrinoSolutionType_);
 }