/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_5_3_13_patch3/src/TopQuarkAnalysis/TopJetCombination/plugins/TtSemiLepHypGenMatch.cc

Go to the documentation of this file.

#include "DataFormats/Math/interface/deltaR.h"
#include "AnalysisDataFormats/TopObjects/interface/TtGenEvent.h"
#include "AnalysisDataFormats/TopObjects/interface/TtSemiLepEvtPartons.h"
#include "TopQuarkAnalysis/TopJetCombination/plugins/TtSemiLepHypGenMatch.h"

TtSemiLepHypGenMatch::TtSemiLepHypGenMatch(const edm::ParameterSet& cfg):
  TtSemiLepHypothesis( cfg )
{ }

TtSemiLepHypGenMatch::~TtSemiLepHypGenMatch() { }

void
TtSemiLepHypGenMatch::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)
{
  // -----------------------------------------------------
  // get genEvent (to distinguish between uds and c quarks
  // and for the lepton matching)
  // -----------------------------------------------------
  edm::Handle<TtGenEvent> genEvt;
  evt.getByLabel("genEvt", genEvt);  

  // -----------------------------------------------------
  // add jets
  // -----------------------------------------------------
  for(unsigned idx=0; idx<match.size(); ++idx){
    if( isValid(match[idx], jets) ){
      switch(idx){
      case TtSemiLepEvtPartons::LightQ:
        if( std::abs(genEvt->hadronicDecayQuark()->pdgId())==4 )
          setCandidate(jets, match[idx], lightQ_, jetCorrectionLevel("cQuark"));
        else
          setCandidate(jets, match[idx], lightQ_, jetCorrectionLevel("udsQuark"));
        break;
      case TtSemiLepEvtPartons::LightQBar:
        if( std::abs(genEvt->hadronicDecayQuarkBar()->pdgId())==4 )
          setCandidate(jets, match[idx], lightQBar_, jetCorrectionLevel("cQuark"));
        else
          setCandidate(jets, match[idx], lightQBar_, jetCorrectionLevel("udsQuark"));
        break;
      case TtSemiLepEvtPartons::HadB:
        setCandidate(jets, match[idx], hadronicB_, jetCorrectionLevel("bQuark")); break;
      case TtSemiLepEvtPartons::LepB: 
        setCandidate(jets, match[idx], leptonicB_, jetCorrectionLevel("bQuark")); break;
      }
    }
  }
 
  // -----------------------------------------------------
  // add lepton
  // -----------------------------------------------------
  int iLepton = findMatchingLepton(genEvt, leps);
  if( iLepton<0 )
    return;
  setCandidate(leps, iLepton, lepton_);
  match.push_back( iLepton );

  // -----------------------------------------------------
  // add neutrino
  // -----------------------------------------------------
  if( mets->empty() )
    return;
  if(neutrinoSolutionType_ == -1)
    setCandidate(mets, 0, neutrino_);
  else
    setNeutrino(mets, leps, iLepton, neutrinoSolutionType_);
}

int
TtSemiLepHypGenMatch::findMatchingLepton(const edm::Handle<TtGenEvent>& genEvt,
                                         const edm::Handle<edm::View<reco::RecoCandidate> >& leps)
{
  int genIdx=-1;

  // jump out with -1 when the collection is empty
  if( leps->empty() ) return genIdx;
  
  if( genEvt->isTtBar() && genEvt->isSemiLeptonic( leptonType( &(leps->front()) ) ) && genEvt->singleLepton() ){
    double minDR=-1;
    for(unsigned i=0; i<leps->size(); ++i){
      double dR = deltaR(genEvt->singleLepton()->eta(), genEvt->singleLepton()->phi(), (*leps)[i].eta(), (*leps)[i].phi());
      if(minDR<0 || dR<minDR){
        minDR=dR;
        genIdx=i;
      }
    }
  }
  return genIdx;
}