CMS 3D CMS Logo
Public Member Functions | Private Member Functions | Private Attributes

TtSemiLepHypWMassMaxSumPt Class Reference

#include <TtSemiLepHypWMassMaxSumPt.h>

Inheritance diagram for TtSemiLepHypWMassMaxSumPt:
TtSemiLepHypothesis edm::EDProducer edm::ProducerBase edm::ProductRegistryHelper

List of all members.

Public Member Functions

 TtSemiLepHypWMassMaxSumPt (const edm::ParameterSet &)
 ~TtSemiLepHypWMassMaxSumPt ()

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
virtual void buildKey ()
 build the event hypothesis key

Private Attributes

std::string bTagAlgorithm_
double maxBDiscLightJets_
int maxNJets_
double minBDiscBJets_
int neutrinoSolutionType_
bool useBTagging_
double wMass_

Detailed Description

Definition at line 7 of file TtSemiLepHypWMassMaxSumPt.h.

Constructor & Destructor Documentation

TtSemiLepHypWMassMaxSumPt::TtSemiLepHypWMassMaxSumPt ( const edm::ParameterSet &  cfg) [explicit]

Definition at line 4 of file TtSemiLepHypWMassMaxSumPt.cc.

References 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.";
}
TtSemiLepHypWMassMaxSumPt::~TtSemiLepHypWMassMaxSumPt ( )

Definition at line 20 of file TtSemiLepHypWMassMaxSumPt.cc.

{ }

Member Function Documentation

void TtSemiLepHypWMassMaxSumPt::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 
) [private, virtual]

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

Implements TtSemiLepHypothesis.

Definition at line 23 of file TtSemiLepHypWMassMaxSumPt.cc.

References bTagAlgorithm_, TtSemiLepDaughter::HadB, TtSemiLepHypothesis::hadronicB_, i, UserOptions_cff::idx, TtSemiLepHypothesis::isValid(), TtSemiLepHypothesis::jetCorrectionLevel(), analyzePatCleaning_cfg::jets, TtSemiLepDaughter::LepB, TtSemiLepEvtPartons::Lepton, TtSemiLepHypothesis::lepton_, TtSemiLepHypothesis::leptonicB_, TtFullHadDaughter::LightQ, TtSemiLepHypothesis::lightQ_, TtFullHadDaughter::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(unsigned int i=0; i<5; ++i)
    match.push_back(-1);

  // -----------------------------------------------------
  // 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(int idx=0; idx<maxNJets; ++idx){
    if(useBTagging_ && (!isLJet[idx] || (cntBJets<=2 && isBJet[idx]))) continue;
    for(int 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);
      }
    }
  }

  // -----------------------------------------------------
  // associate those jets with maximum pt of the vectorial 
  // sum to the hadronic decay chain
  // -----------------------------------------------------
  double maxPt=-1.;
  int hadB=-1;
  if( isValid(closestToWMassIndices[0], jets) && isValid(closestToWMassIndices[1], jets)) {
    for(int idx=0; idx<maxNJets; ++idx){
      if(useBTagging_ && !isBJet[idx]) continue;
      // make sure it's not used up already from the hadronic W
      if( idx!=closestToWMassIndices[0] && idx!=closestToWMassIndices[1] ){
        reco::Particle::LorentzVector sum = 
          (*jets)[closestToWMassIndices[0]].p4()+
          (*jets)[closestToWMassIndices[1]].p4()+
          (*jets)[idx].p4();
        if( maxPt<0. || maxPt<sum.pt() ){
          maxPt=sum.pt();
          hadB=idx;
        }
      }
    }
  }
  
  // -----------------------------------------------------
  // associate the remaining jet with maximum pt of the   
  // vectorial sum with the leading lepton with the 
  // leptonic b quark
  // -----------------------------------------------------
  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( idx!=closestToWMassIndices[0] && idx!=closestToWMassIndices[1] && idx!=hadB) {
      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];
  }
  
  if( isValid(hadB, jets) ){
    setCandidate(jets, hadB, hadronicB_, jetCorrectionLevel("bQuark"));
    match[TtSemiLepEvtPartons::HadB] = hadB;
  }

  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_);
}
virtual void TtSemiLepHypWMassMaxSumPt::buildKey ( ) [inline, private, virtual]

build the event hypothesis key

Implements TtSemiLepHypothesis.

Definition at line 17 of file TtSemiLepHypWMassMaxSumPt.h.

References TtSemiLepHypothesis::key_, and TtEvent::kWMassMaxSumPt.

{ key_= TtSemiLeptonicEvent::kWMassMaxSumPt; };

Member Data Documentation

std::string TtSemiLepHypWMassMaxSumPt::bTagAlgorithm_ [private]

Definition at line 30 of file TtSemiLepHypWMassMaxSumPt.h.

Referenced by buildHypo().

double TtSemiLepHypWMassMaxSumPt::maxBDiscLightJets_ [private]

Definition at line 32 of file TtSemiLepHypWMassMaxSumPt.h.

Referenced by buildHypo().

int TtSemiLepHypWMassMaxSumPt::maxNJets_ [private]

Definition at line 27 of file TtSemiLepHypWMassMaxSumPt.h.

Referenced by buildHypo(), and TtSemiLepHypWMassMaxSumPt().

double TtSemiLepHypWMassMaxSumPt::minBDiscBJets_ [private]

Definition at line 31 of file TtSemiLepHypWMassMaxSumPt.h.

Referenced by buildHypo().

int TtSemiLepHypWMassMaxSumPt::neutrinoSolutionType_ [private]

Definition at line 33 of file TtSemiLepHypWMassMaxSumPt.h.

Referenced by buildHypo().

bool TtSemiLepHypWMassMaxSumPt::useBTagging_ [private]

Definition at line 29 of file TtSemiLepHypWMassMaxSumPt.h.

Referenced by buildHypo().

double TtSemiLepHypWMassMaxSumPt::wMass_ [private]

Definition at line 28 of file TtSemiLepHypWMassMaxSumPt.h.

Referenced by buildHypo().

Generated for CMSSW Reference Manual by doxygen 1.7.3