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 ),
  genEvtToken_( consumes<TtGenEvent>( edm::InputTag( "genEvt" ) ) )
{ }

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.getByToken(genEvtToken_, 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;
}