TtFullLepHypGenMatch Class Reference

Inheritance diagram for TtFullLepHypGenMatch:

Public Member Functions
	TtFullLepHypGenMatch (const edm::ParameterSet &)
Public Member Functions inherited from TtFullLepHypothesis
	TtFullLepHypothesis (const edm::ParameterSet &)
	default constructor More...
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
	EDProducer (const EDProducer &)=delete
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final
const EDProducer &	operator= (const EDProducer &)=delete

Protected Attributes
edm::EDGetTokenT< TtGenEvent >	genEvtToken_
Protected Attributes inherited from TtFullLepHypothesis
std::unique_ptr< reco::ShallowClonePtrCandidate >	b_
std::unique_ptr< reco::ShallowClonePtrCandidate >	bBar_
edm::EDGetTokenT< std::vector< pat::Electron > >	elecsToken_
bool	getMatch_
std::string	jetCorrectionLevel_
edm::EDGetTokenT< std::vector< pat::Jet > >	jetsToken_
int	key_
	hypothesis key (to be set by the buildKey function) More...
std::unique_ptr< reco::ShallowClonePtrCandidate >	lepton_
std::unique_ptr< reco::ShallowClonePtrCandidate >	leptonBar_
edm::EDGetTokenT< std::vector< std::vector< int > > >	matchToken_
	input label for all necessary collections More...
edm::EDGetTokenT< std::vector< pat::MET > >	metsToken_
edm::EDGetTokenT< std::vector< pat::Muon > >	musToken_
std::unique_ptr< reco::ShallowClonePtrCandidate >	neutrino_
std::unique_ptr< reco::ShallowClonePtrCandidate >	neutrinoBar_
std::unique_ptr< reco::LeafCandidate >	recNu
	candidates needed for the genmatch hypothesis More...
std::unique_ptr< reco::LeafCandidate >	recNuBar

Private Member Functions
void	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) override
	build event hypothesis from the reco objects of a semi-leptonic event More...
void	buildKey () override
	build the event hypothesis key More...
void	buildMatchingNeutrinos (edm::Event &, const edm::Handle< std::vector< pat::MET > > &)
template<typename O >
int	findMatchingLepton (const reco::GenParticle *, const edm::Handle< std::vector< O > > &)

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
using	CacheTypes = CacheContexts< T... >
using	GlobalCache = typename CacheTypes::GlobalCache
using	HasAbility = AbilityChecker< T... >
using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
using	RunCache = typename CacheTypes::RunCache
using	RunContext = RunContextT< RunCache, GlobalCache >
using	RunSummaryCache = typename CacheTypes::RunSummaryCache
Protected Member Functions inherited from TtFullLepHypothesis
reco::CompositeCandidate	hypo ()
	return event hypothesis More...
bool	isValid (const int &idx, const edm::Handle< std::vector< pat::Jet > > &jets)
	check if index is in valid range of selected jets More...
int	key () const
	return key More...
template<typename C >
std::unique_ptr< reco::ShallowClonePtrCandidate >	makeCandidate (const edm::Handle< C > &handle, const int &idx)
	use one object in a collection to set a ShallowClonePtrCandidate More...
std::unique_ptr< reco::ShallowClonePtrCandidate >	makeCandidate (const edm::Handle< std::vector< pat::Jet > > &handle, const int &idx, const std::string &correctionLevel)
	use one object in a jet collection to set a ShallowClonePtrCandidate with proper jet corrections More...
void	produce (edm::Event &, const edm::EventSetup &) override
	produce the event hypothesis as CompositeCandidate and Key More...
void	resetCandidates ()
	reset candidate pointers before hypo build process More...

Detailed Description

Definition at line 7 of file TtFullLepHypGenMatch.cc.

Constructor & Destructor Documentation

TtFullLepHypGenMatch()

TtFullLepHypGenMatch::TtFullLepHypGenMatch ( const edm::ParameterSet &  cfg )

explicit

Definition at line 31 of file TtFullLepHypGenMatch.cc.

    : TtFullLepHypothesis(cfg), genEvtToken_(consumes<TtGenEvent>(edm::InputTag("genEvt"))) {}

Member Function Documentation

buildHypo()

void TtFullLepHypGenMatch::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  )

overrideprivatevirtual

build event hypothesis from the reco objects of a semi-leptonic event

Implements TtFullLepHypothesis.

Definition at line 34 of file TtFullLepHypGenMatch.cc.

References TtFullLepEvtPartons::B, TtFullLepHypothesis::b_, TtFullLepEvtPartons::BBar, TtFullLepHypothesis::bBar_, buildMatchingNeutrinos(), singleTopDQM_cfi::elecs, findMatchingLepton(), TtGenEvtProducer_cfi::genEvt, genEvtToken_, edm::Event::getByToken(), heavyIonCSV_trainingSettings::idx, TtFullLepHypothesis::isValid(), TtFullLepHypothesis::jetCorrectionLevel_, PDWG_EXODelayedJetMET_cff::jets, WDecay::kElec, WDecay::kMuon, TtFullLepHypothesis::lepton_, TtFullLepHypothesis::leptonBar_, TtFullLepHypothesis::makeCandidate(), match(), and singleTopDQM_cfi::mets.

                                                               {
  // -----------------------------------------------------
  // add jets
  // -----------------------------------------------------
  for (unsigned idx = 0; idx < match.size(); ++idx) {
    if (isValid(match[idx], jets)) {
      switch (idx) {
        case TtFullLepEvtPartons::B:
          b_ = makeCandidate(jets, match[idx], jetCorrectionLevel_);
          break;
        case TtFullLepEvtPartons::BBar:
          bBar_ = makeCandidate(jets, match[idx], jetCorrectionLevel_);
          break;
      }
    }
  }

  // -----------------------------------------------------
  // add leptons
  // -----------------------------------------------------
  // get genEvent
  edm::Handle<TtGenEvent> genEvt;
  evt.getByToken(genEvtToken_, genEvt);

  // push back fake indices if no leptons in genevent
  if (!genEvt->isFullLeptonic() || !genEvt->lepton() || !genEvt->leptonBar()) {
    match.push_back(-1);
    match.push_back(-1);
    match.push_back(-1);
    match.push_back(-1);
  } else if (genEvt->isFullLeptonic(WDecay::kElec, WDecay::kElec) && elecs->size() >= 2) {
    //search indices for electrons
    int iLepBar = findMatchingLepton(genEvt->leptonBar(), elecs);
    leptonBar_ = makeCandidate(elecs, iLepBar);
    match.push_back(iLepBar);
    int iLep = findMatchingLepton(genEvt->lepton(), elecs);
    lepton_ = makeCandidate(elecs, iLep);
    match.push_back(iLep);

    // fake indices for muons
    match.push_back(-1);
    match.push_back(-1);
  } else if (genEvt->isFullLeptonic(WDecay::kElec, WDecay::kMuon) && !elecs->empty() && !mus->empty()) {
    if (genEvt->leptonBar()->isElectron()) {
      // push back index for e+
      int iLepBar = findMatchingLepton(genEvt->leptonBar(), elecs);
      leptonBar_ = makeCandidate(elecs, iLepBar);
      match.push_back(iLepBar);
      // push back fake indices for e- and mu+
      match.push_back(-1);
      match.push_back(-1);
      // push back index for mu-
      int iLep = findMatchingLepton(genEvt->lepton(), mus);
      lepton_ = makeCandidate(mus, iLep);
      match.push_back(iLep);
    } else {
      // push back fake index for e+
      match.push_back(-1);
      // push back index for e-
      int iLepBar = findMatchingLepton(genEvt->leptonBar(), mus);
      leptonBar_ = makeCandidate(mus, iLepBar);
      match.push_back(iLepBar);
      // push back index for mu+
      int iLep = findMatchingLepton(genEvt->lepton(), elecs);
      lepton_ = makeCandidate(elecs, iLep);
      match.push_back(iLep);
      // push back fake index for mu-
      match.push_back(-1);
    }
  } else if (genEvt->isFullLeptonic(WDecay::kMuon, WDecay::kMuon) && mus->size() >= 2) {
    // fake indices for electrons
    match.push_back(-1);
    match.push_back(-1);

    //search indices for electrons
    int iLepBar = findMatchingLepton(genEvt->leptonBar(), mus);
    leptonBar_ = makeCandidate(mus, iLepBar);
    match.push_back(iLepBar);
    int iLep = findMatchingLepton(genEvt->lepton(), mus);
    lepton_ = makeCandidate(mus, iLep);
    match.push_back(iLep);
  } else {  //this 'else' should happen if at least one genlepton is a tau
    match.push_back(-1);
    match.push_back(-1);
    match.push_back(-1);
    match.push_back(-1);
  }

  // -----------------------------------------------------
  // add met and neutrinos
  // -----------------------------------------------------
  if (!mets->empty()) {
    //met_ = makeCandidate(mets, 0);
    buildMatchingNeutrinos(evt, mets);
  }
}

buildKey()

void TtFullLepHypGenMatch::buildKey ( )

inlineoverrideprivatevirtual

build the event hypothesis key

Implements TtFullLepHypothesis.

Definition at line 13 of file TtFullLepHypGenMatch.cc.

References TtFullLepHypothesis::key_, and TtEvent::kGenMatch.

{ key_ = TtFullLeptonicEvent::kGenMatch; };

buildMatchingNeutrinos()

void TtFullLepHypGenMatch::buildMatchingNeutrinos ( edm::Event &  evt, const edm::Handle< std::vector< pat::MET > > &  mets  )

private

Definition at line 152 of file TtFullLepHypGenMatch.cc.

References TtGenEvtProducer_cfi::genEvt, genEvtToken_, edm::Event::getByToken(), singleTopDQM_cfi::mets, TtFullLepHypothesis::recNu, TtFullLepHypothesis::recNuBar, and mathSSE::sqrt().

Referenced by buildHypo().

                                                                                                          {
  // get genEvent
  edm::Handle<TtGenEvent> genEvt;
  evt.getByToken(genEvtToken_, genEvt);

  if (genEvt->isTtBar() && genEvt->isFullLeptonic() && genEvt->neutrino() && genEvt->neutrinoBar()) {
    double momXNu = genEvt->neutrino()->px();
    double momYNu = genEvt->neutrino()->py();
    double momXNuBar = genEvt->neutrinoBar()->px();
    double momYNuBar = genEvt->neutrinoBar()->py();

    double momXMet = mets->at(0).px();
    double momYMet = mets->at(0).py();

    double momXNeutrino = 0.5 * (momXNu - momXNuBar + momXMet);
    double momYNeutrino = 0.5 * (momYNu - momYNuBar + momYMet);
    double momXNeutrinoBar = momXMet - momXNeutrino;
    double momYNeutrinoBar = momYMet - momYNeutrino;

    math::XYZTLorentzVector recNuFM(
        momXNeutrino, momYNeutrino, 0, sqrt(momXNeutrino * momXNeutrino + momYNeutrino * momYNeutrino));
    recNu = std::make_unique<reco::LeafCandidate>(0, recNuFM);

    math::XYZTLorentzVector recNuBarFM(momXNeutrinoBar,
                                       momYNeutrinoBar,
                                       0,
                                       sqrt(momXNeutrinoBar * momXNeutrinoBar + momYNeutrinoBar * momYNeutrinoBar));
    recNuBar = std::make_unique<reco::LeafCandidate>(0, recNuBarFM);
  }
}

findMatchingLepton()

template<typename O >

int TtFullLepHypGenMatch::findMatchingLepton ( const reco::GenParticle *  genLep, const edm::Handle< std::vector< O > > &  leps  )

private

Definition at line 138 of file TtFullLepHypGenMatch.cc.

References PbPb_ZMuSkimMuonDPG_cff::deltaR, HGC3DClusterGenMatchSelector_cfi::dR, reco::LeafCandidate::eta(), mps_fire::i, heavyIonCSV_trainingSettings::idx, TtSemiLepHitFitProducer_Electrons_cfi::leps, HLT_2023v12_cff::minDR, and reco::LeafCandidate::phi().

Referenced by buildHypo().

                                                                                   {
  int idx = -1;
  double minDR = -1;
  for (unsigned i = 0; i < leps->size(); ++i) {
    double dR = deltaR(genLep->eta(), genLep->phi(), (*leps)[i].eta(), (*leps)[i].phi());
    if (minDR < 0 || dR < minDR) {
      minDR = dR;
      idx = i;
    }
  }
  return idx;
}

Member Data Documentation

genEvtToken_

edm::EDGetTokenT<TtGenEvent> TtFullLepHypGenMatch::genEvtToken_

protected

Definition at line 28 of file TtFullLepHypGenMatch.cc.

Referenced by buildHypo(), and buildMatchingNeutrinos().