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