---

TtGenEvent.h

Go to the documentation of this file.

#ifndef TopObjects_TtGenEvent_h
#define TopObjects_TtGenEvent_h

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


namespace TtFullLepEvtPartons{
  enum { B, BBar, Lepton, LeptonBar };
}

namespace TtSemiLepEvtPartons{
  enum { LightQ, LightQBar, HadB, LepB, Lepton };
}

namespace TtFullHadEvtPartons{
  enum { LightQTop, LightQBarTop, B, LightQTopBar, LightQBarTopBar, BBar};
}


// ----------------------------------------------------------------------
// derived class for: 
//
//  * TtGenEvent
//
//  the structure holds reference information to the generator particles 
//  of the decay chains for each top quark and of the initial partons 
//  and provides access and administration;  the derived class contains 
//  a few additional getters with respect to its base class
// ----------------------------------------------------------------------

class TtGenEvent: public TopGenEvent {

 public:

  TtGenEvent() {};
  TtGenEvent(reco::GenParticleRefProd&, reco::GenParticleRefProd&);
  virtual ~TtGenEvent() {};

  bool isTtBar() const {return (top() && topBar());}
  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 invert=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 reco::GenParticle* lightQFromTop() const;
  const reco::GenParticle* lightQBarFromTop() const;
  const reco::GenParticle* lightQFromTopBar() const;
  const reco::GenParticle* lightQBarFromTopBar() const;

 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

---