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_);

}