TtFullLepHypKinSolution.cc

Go to the documentation of this file.

#include "DataFormats/PatCandidates/interface/Particle.h"
#include "TopQuarkAnalysis/TopJetCombination/plugins/TtFullLepHypKinSolution.h"

TtFullLepHypKinSolution::TtFullLepHypKinSolution(const edm::ParameterSet& cfg)
    : TtFullLepHypothesis(cfg),
      nusToken_(consumes<std::vector<reco::LeafCandidate> >(cfg.getParameter<edm::InputTag>("Neutrinos"))),
      nuBarsToken_(consumes<std::vector<reco::LeafCandidate> >(cfg.getParameter<edm::InputTag>("NeutrinoBars"))),
      solWeightToken_(consumes<std::vector<double> >(cfg.getParameter<edm::InputTag>("solutionWeight"))) {}

TtFullLepHypKinSolution::~TtFullLepHypKinSolution() {}

void TtFullLepHypKinSolution::buildHypo(edm::Event& evt,
                                        const edm::Handle<std::vector<pat::Electron> >& elecs,
                                        const edm::Handle<std::vector<pat::Muon> >& mus,
                                        const edm::Handle<std::vector<pat::Jet> >& jets,
                                        const edm::Handle<std::vector<pat::MET> >& mets,
                                        std::vector<int>& match,
                                        const unsigned int iComb) {
  edm::Handle<std::vector<double> > solWeight;
  //   edm::Handle<std::vector<std::vector<int> > >   idcsVec;
  edm::Handle<std::vector<reco::LeafCandidate> > nus;
  edm::Handle<std::vector<reco::LeafCandidate> > nuBars;

  evt.getByToken(solWeightToken_, solWeight);
  //   evt.getByToken(particleIdcsToken_, idcsVec  );
  evt.getByToken(nusToken_, nus);
  evt.getByToken(nuBarsToken_, nuBars);

  if ((*solWeight)[iComb] < 0) {
    // create empty hypothesis if no solution exists
    return;
  }

  // -----------------------------------------------------
  // add jets
  // -----------------------------------------------------
  if (!jets->empty()) {
    setCandidate(jets, match[0], b_, jetCorrectionLevel_);
    setCandidate(jets, match[1], bBar_, jetCorrectionLevel_);
  }
  // -----------------------------------------------------
  // add leptons
  // -----------------------------------------------------
  if (!elecs->empty() && match[2] >= 0)
    setCandidate(elecs, match[2], leptonBar_);

  if (!elecs->empty() && match[3] >= 0)
    setCandidate(elecs, match[3], lepton_);

  if (!mus->empty() && match[4] >= 0 && match[2] < 0)
    setCandidate(mus, match[4], leptonBar_);

  // this 'else' happens if you have a wrong charge electron-muon-
  // solution so the indices are (b-idx, bbar-idx, 0, -1, 0, -1)
  // so the mu^+ is stored as l^-
  else if (!mus->empty() && match[4] >= 0)
    setCandidate(mus, match[4], lepton_);

  if (!mus->empty() && match[5] >= 0 && match[3] < 0)
    setCandidate(mus, match[5], lepton_);

  // this 'else' happens if you have a wrong charge electron-muon-
  // solution so the indices are (b-idx, bbar-idx, -1, 0, -1, 0)
  // so the mu^- is stored as l^+
  else if (!mus->empty() && match[5] >= 0)
    setCandidate(mus, match[5], leptonBar_);

  // -----------------------------------------------------
  // add neutrinos
  // -----------------------------------------------------
  if (!nus->empty())
    setCandidate(nus, iComb, neutrino_);

  if (!nuBars->empty())
    setCandidate(nuBars, iComb, neutrinoBar_);
}