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 TopDecayID

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

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