TtFullHadHypGenMatch.cc

Go to the documentation of this file.

#include "TopQuarkAnalysis/TopJetCombination/plugins/TtFullHadHypGenMatch.h"
#include "AnalysisDataFormats/TopObjects/interface/TtFullHadEvtPartons.h"

TtFullHadHypGenMatch::TtFullHadHypGenMatch(const edm::ParameterSet& cfg):
  TtFullHadHypothesis( cfg ),
  genEvtToken_( consumes<TtGenEvent>( edm::InputTag( "genEvt" ) ) )
{
}

TtFullHadHypGenMatch::~TtFullHadHypGenMatch() { }

void
TtFullHadHypGenMatch::buildHypo(edm::Event& evt,
                    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)
  // -----------------------------------------------------
  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 TtFullHadEvtPartons::LightQ:
    if( std::abs(genEvt->daughterQuarkOfWPlus()->pdgId())==4 )
      setCandidate(jets, match[idx], lightQ_   , jetCorrectionLevel("cQuark"));
    else
      setCandidate(jets, match[idx], lightQ_   , jetCorrectionLevel("udsQuark"));
    break;
      case TtFullHadEvtPartons::LightQBar:
    if( std::abs(genEvt->daughterQuarkBarOfWPlus()->pdgId())==4 )
      setCandidate(jets, match[idx], lightQBar_, jetCorrectionLevel("cQuark"));
    else
      setCandidate(jets, match[idx], lightQBar_, jetCorrectionLevel("udsQuark"));
    break;
      case TtFullHadEvtPartons::B:
    setCandidate(jets, match[idx], b_          , jetCorrectionLevel("bQuark")); break;
      case TtFullHadEvtPartons::LightP:
    if( std::abs(genEvt->daughterQuarkOfWMinus()->pdgId())==4 )
      setCandidate(jets, match[idx], lightP_   , jetCorrectionLevel("cQuark"));
    else
      setCandidate(jets, match[idx], lightP_   , jetCorrectionLevel("udsQuark"));
    break;
      case TtFullHadEvtPartons::LightPBar:
    if( std::abs(genEvt->daughterQuarkBarOfWMinus()->pdgId())==4 )
      setCandidate(jets, match[idx], lightPBar_, jetCorrectionLevel("cQuark"));
    else
      setCandidate(jets, match[idx], lightPBar_, jetCorrectionLevel("udsQuark"));
    break;
      case TtFullHadEvtPartons::BBar:
    setCandidate(jets, match[idx], bBar_       , jetCorrectionLevel("bQuark")); break;
      }
    }
  }
}