TtFullHadHypKinFit.cc

Go to the documentation of this file.

#include "TopQuarkAnalysis/TopJetCombination/plugins/TtFullHadHypKinFit.h"


TtFullHadHypKinFit::TtFullHadHypKinFit(const edm::ParameterSet& cfg):
  TtFullHadHypothesis( cfg ),
  statusToken_   (consumes<std::vector<int> >          (cfg.getParameter<edm::InputTag>("status"      ))),
  lightQToken_   (consumes<std::vector<pat::Particle> >(cfg.getParameter<edm::InputTag>("lightQTag"   ))),
  lightQBarToken_(consumes<std::vector<pat::Particle> >(cfg.getParameter<edm::InputTag>("lightQBarTag"))),
  bToken_        (consumes<std::vector<pat::Particle> >(cfg.getParameter<edm::InputTag>("bTag"        ))),
  bBarToken_     (consumes<std::vector<pat::Particle> >(cfg.getParameter<edm::InputTag>("bBarTag"     ))),
  lightPToken_   (consumes<std::vector<pat::Particle> >(cfg.getParameter<edm::InputTag>("lightPTag"   ))),
  lightPBarToken_(consumes<std::vector<pat::Particle> >(cfg.getParameter<edm::InputTag>("lightPBarTag")))
{
}

TtFullHadHypKinFit::~TtFullHadHypKinFit() { }

void
TtFullHadHypKinFit::buildHypo(edm::Event& evt,
                  const edm::Handle<std::vector<pat::Jet> >& jets,
                  std::vector<int>& match, const unsigned int iComb)
{
  edm::Handle<std::vector<int> > status;
  evt.getByToken(statusToken_, status);
  if( (*status)[iComb] != 0 ){
    // create empty hypothesis if kinematic fit did not converge
    return;
  }

  edm::Handle<std::vector<pat::Particle> > lightQ;
  edm::Handle<std::vector<pat::Particle> > lightQBar;
  edm::Handle<std::vector<pat::Particle> > b;
  edm::Handle<std::vector<pat::Particle> > bBar;
  edm::Handle<std::vector<pat::Particle> > lightP;
  edm::Handle<std::vector<pat::Particle> > lightPBar;

  evt.getByToken(lightQToken_   , lightQ   );
  evt.getByToken(lightQBarToken_, lightQBar);
  evt.getByToken(bToken_        , b        );
  evt.getByToken(bBarToken_     , bBar     );
  evt.getByToken(lightPToken_   , lightP   );
  evt.getByToken(lightPBarToken_, lightPBar);

  // -----------------------------------------------------
  // add jets
  // -----------------------------------------------------
  if( !( lightQ->empty() || lightQBar->empty() || b->empty() || bBar->empty() ||
     lightP->empty() || lightPBar->empty() ) ) {
    setCandidate(lightQ   , iComb, lightQ_   );
    setCandidate(lightQBar, iComb, lightQBar_);
    setCandidate(b        , iComb, b_        );
    setCandidate(bBar     , iComb, bBar_     );
    setCandidate(lightP   , iComb, lightP_   );
    setCandidate(lightPBar, iComb, lightPBar_);
  }
}