TtGenEvent.h

Go to the documentation of this file.

#ifndef TopObjects_TtGenEvent_h
#define TopObjects_TtGenEvent_h

#include "CommonTools/CandUtils/interface/pdgIdUtils.h"
#include "AnalysisDataFormats/TopObjects/interface/TopGenEvent.h"

class TtGenEvent : public TopGenEvent {
public:
  TtGenEvent() {}
  TtGenEvent(reco::GenParticleRefProd& decaySubset, reco::GenParticleRefProd& initSubset);
  ~TtGenEvent() override {}

  bool isTtBar() const { return (top() && topBar()); }
  bool fromGluonFusion() const;
  bool fromQuarkAnnihilation() const;
  bool isFullHadronic(bool excludeTauLeptons = false) const {
    return isTtBar() ? isNumberOfLeptons(excludeTauLeptons, 0) : false;
  }
  bool isSemiLeptonic(bool excludeTauLeptons = false) const {
    return isTtBar() ? isNumberOfLeptons(excludeTauLeptons, 1) : false;
  }
  bool isFullLeptonic(bool excludeTauLeptons = false) const {
    return isTtBar() ? isNumberOfLeptons(excludeTauLeptons, 2) : false;
  }

  WDecay::LeptonType semiLeptonicChannel() const;
  bool isSemiLeptonic(WDecay::LeptonType typeA) const { return semiLeptonicChannel() == typeA ? true : false; };
  bool isSemiLeptonic(WDecay::LeptonType typeA, WDecay::LeptonType typeB) const {
    return (semiLeptonicChannel() == typeA || semiLeptonicChannel() == typeB) ? true : false;
  };
  // return decay channel (as a std::pair of LeptonType's); all leptons including taus are allowed
  std::pair<WDecay::LeptonType, WDecay::LeptonType> fullLeptonicChannel() const;
  bool isFullLeptonic(WDecay::LeptonType typeA, WDecay::LeptonType typeB) const;

  const reco::GenParticle* singleLepton(bool excludeTauLeptons = false) const;
  const reco::GenParticle* singleNeutrino(bool excludeTauLeptons = false) const;
  const reco::GenParticle* leptonicDecayW(bool excludeTauLeptons = false) const;
  const reco::GenParticle* leptonicDecayB(bool excludeTauLeptons = false) const;
  const reco::GenParticle* leptonicDecayTop(bool excludeTauLeptons = false) const;
  const reco::GenParticle* hadronicDecayW(bool excludeTauLeptons = false) const;
  const reco::GenParticle* hadronicDecayB(bool excludeTauLeptons = false) const;
  const reco::GenParticle* hadronicDecayTop(bool excludeTauLeptons = false) const;
  const reco::GenParticle* hadronicDecayQuark(bool invertFlavor = false) const;
  const reco::GenParticle* hadronicDecayQuarkBar() const { return hadronicDecayQuark(true); };
  std::vector<const reco::GenParticle*> leptonicDecayTopRadiation(bool excludeTauLeptons = false) const;
  std::vector<const reco::GenParticle*> hadronicDecayTopRadiation(bool excludeTauLeptons = false) const;
  const reco::GenParticle* lepton(bool excludeTauLeptons = false) const;
  const reco::GenParticle* leptonBar(bool excludeTauLeptons = false) const;
  const reco::GenParticle* neutrino(bool excludeTauLeptons = false) const;
  const reco::GenParticle* neutrinoBar(bool excludeTauLeptons = false) const;

  const math::XYZTLorentzVector* topPair() const { return isTtBar() ? &topPair_ : nullptr; };

protected:
  math::XYZTLorentzVector topPair_;

private:
  bool isNumberOfLeptons(bool excludeTauLeptons, int nlep) const {
    return excludeTauLeptons ? (numberOfLeptons() - numberOfLeptons(WDecay::kTau)) == nlep : numberOfLeptons() == nlep;
  }
};

inline bool TtGenEvent::isFullLeptonic(WDecay::LeptonType typeA, WDecay::LeptonType typeB) const {
  return ((fullLeptonicChannel().first == typeA && fullLeptonicChannel().second == typeB) ||
          (fullLeptonicChannel().first == typeB && fullLeptonicChannel().second == typeA));
}

#endif