TopGenEvent.h

Go to the documentation of this file.

#ifndef TopObjects_TopGenEvent_h
#define TopObjects_TopGenEvent_h

#include "DataFormats/Candidate/interface/Candidate.h"
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"


namespace TopDecayID{
  static const int stable = 2;
  static const int unfrag = 3;
  static const int tID    = 6;
  static const int bID    = 5;
  static const int glueID = 21;
  static const int photID = 22;
  static const int ZID    = 23;
  static const int WID    = 24;
  static const int elecID = 11;
  static const int muonID = 13;
  static const int tauID  = 15;
}

namespace WDecay{
  enum LeptonType {kNone, kElec, kMuon, kTau};
}

class TopGenEvent {

 public:

  TopGenEvent(){};
  TopGenEvent(reco::GenParticleRefProd& decaySubset, reco::GenParticleRefProd& iniSubset);
  virtual ~TopGenEvent(){};

  const reco::GenParticleCollection& particles() const { return *parts_; }
  const reco::GenParticleCollection& initialPartons() const { return *initPartons_;}
  std::vector<const reco::GenParticle*> radiatedGluons(int pdgId) const;
  std::vector<const reco::GenParticle*> lightQuarks(bool includingBQuarks=false) const;
  int numberOfLeptons(bool fromWBoson=true) const;
  int numberOfLeptons(WDecay::LeptonType type, bool fromWBoson=true) const;
  int numberOfBQuarks(bool fromTopQuark=true) const;
  std::vector<const reco::GenParticle*> topSisters() const;
  const reco::GenParticle* daughterQuarkOfTop(bool invertCharge=false) const;
  const reco::GenParticle* daughterQuarkOfTopBar() const { return daughterQuarkOfTop(true); };
  const reco::GenParticle* daughterQuarkOfWPlus(bool invertQuarkCharge=false, bool invertBosonCharge=false) const;
  const reco::GenParticle* daughterQuarkOfWMinus() const { return daughterQuarkOfWPlus(false, true); };
  const reco::GenParticle* daughterQuarkBarOfWPlus() const { return daughterQuarkOfWPlus(true, false); };
  const reco::GenParticle* daughterQuarkBarOfWMinus() const { return daughterQuarkOfWPlus(true, true); };

  const reco::GenParticle* candidate(int id, unsigned int parentId=0) const;
  const reco::GenParticle* eMinus() const   { return candidate( TopDecayID::elecID, TopDecayID::WID );}
  const reco::GenParticle* ePlus() const    { return candidate(-TopDecayID::elecID, TopDecayID::WID );}
  const reco::GenParticle* muMinus() const  { return candidate( TopDecayID::muonID, TopDecayID::WID );}
  const reco::GenParticle* muPlus() const   { return candidate(-TopDecayID::muonID, TopDecayID::WID );}
  const reco::GenParticle* tauMinus() const { return candidate( TopDecayID::tauID, TopDecayID::WID  );}
  const reco::GenParticle* tauPlus() const  { return candidate(-TopDecayID::tauID, TopDecayID::WID  );}
  const reco::GenParticle* wMinus() const   { return candidate(-TopDecayID::WID, TopDecayID::tID    );}
  const reco::GenParticle* wPlus() const    { return candidate( TopDecayID::WID, TopDecayID::tID    );}
  const reco::GenParticle* b() const        { return candidate( TopDecayID::bID, TopDecayID::tID    );}
  const reco::GenParticle* bBar() const     { return candidate(-TopDecayID::bID, TopDecayID::tID    );}
  const reco::GenParticle* top() const      { return candidate( TopDecayID::tID    );}
  const reco::GenParticle* topBar() const   { return candidate(-TopDecayID::tID    );}

  void print() const;

 protected:

  reco::GenParticleRefProd parts_;       
  reco::GenParticleRefProd initPartons_; 
};

#endif