TtSemiLeptonicEvent.h

Go to the documentation of this file.

#ifndef TopObjects_TtSemiLeptonicEvent_h
#define TopObjects_TtSemiLeptonicEvent_h

#include "AnalysisDataFormats/TopObjects/interface/TtEvent.h"

namespace TtSemiLepDaughter {
  static const std::string Nu = "Nu", Lep = "Lep", LepW = "LepW", LepB = "LepB", LepTop = "LepTop";
  static const std::string HadQ = "HadQ", HadP = "HadP", HadW = "HadW", HadB = "HadB", HadTop = "HadTop";
}  // namespace TtSemiLepDaughter

class TtSemiLeptonicEvent : public TtEvent {
public:
  TtSemiLeptonicEvent(){};
  ~TtSemiLeptonicEvent() override{};

  const reco::Candidate* hadronicDecayTop(const std::string& key, const unsigned& cmb = 0) const {
    return hadronicDecayTop(hypoClassKeyFromString(key), cmb);
  };
  const reco::Candidate* hadronicDecayTop(const HypoClassKey& key, const unsigned& cmb = 0) const {
    return !isHypoValid(key, cmb) ? nullptr : eventHypo(key, cmb).daughter(TtSemiLepDaughter::HadTop);
  };
  const reco::Candidate* hadronicDecayB(const std::string& key, const unsigned& cmb = 0) const {
    return hadronicDecayB(hypoClassKeyFromString(key), cmb);
  };
  const reco::Candidate* hadronicDecayB(const HypoClassKey& key, const unsigned& cmb = 0) const {
    return !isHypoValid(key, cmb) ? nullptr : hadronicDecayTop(key, cmb)->daughter(TtSemiLepDaughter::HadB);
  };
  const reco::Candidate* hadronicDecayW(const std::string& key, const unsigned& cmb = 0) const {
    return hadronicDecayW(hypoClassKeyFromString(key), cmb);
  };
  const reco::Candidate* hadronicDecayW(const HypoClassKey& key, const unsigned& cmb = 0) const {
    return !isHypoValid(key, cmb) ? nullptr : hadronicDecayTop(key, cmb)->daughter(TtSemiLepDaughter::HadW);
  };
  const reco::Candidate* hadronicDecayQuark(const std::string& key, const unsigned& cmb = 0) const {
    return hadronicDecayQuark(hypoClassKeyFromString(key), cmb);
  };
  const reco::Candidate* hadronicDecayQuark(const HypoClassKey& key, const unsigned& cmb = 0) const {
    return !isHypoValid(key, cmb) ? nullptr : hadronicDecayW(key, cmb)->daughter(TtSemiLepDaughter::HadP);
  };
  const reco::Candidate* hadronicDecayQuarkBar(const std::string& key, const unsigned& cmb = 0) const {
    return hadronicDecayQuarkBar(hypoClassKeyFromString(key), cmb);
  };
  const reco::Candidate* hadronicDecayQuarkBar(const HypoClassKey& key, const unsigned& cmb = 0) const {
    return !isHypoValid(key, cmb) ? nullptr : hadronicDecayW(key, cmb)->daughter(TtSemiLepDaughter::HadQ);
  };
  const reco::Candidate* leptonicDecayTop(const std::string& key, const unsigned& cmb = 0) const {
    return leptonicDecayTop(hypoClassKeyFromString(key), cmb);
  };
  const reco::Candidate* leptonicDecayTop(const HypoClassKey& key, const unsigned& cmb = 0) const {
    return !isHypoValid(key, cmb) ? nullptr : eventHypo(key, cmb).daughter(TtSemiLepDaughter::LepTop);
  };
  const reco::Candidate* leptonicDecayB(const std::string& key, const unsigned& cmb = 0) const {
    return leptonicDecayB(hypoClassKeyFromString(key), cmb);
  };
  const reco::Candidate* leptonicDecayB(const HypoClassKey& key, const unsigned& cmb = 0) const {
    return !isHypoValid(key, cmb) ? nullptr : leptonicDecayTop(key, cmb)->daughter(TtSemiLepDaughter::LepB);
  };
  const reco::Candidate* leptonicDecayW(const std::string& key, const unsigned& cmb = 0) const {
    return leptonicDecayW(hypoClassKeyFromString(key), cmb);
  };
  const reco::Candidate* leptonicDecayW(const HypoClassKey& key, const unsigned& cmb = 0) const {
    return !isHypoValid(key, cmb) ? nullptr : leptonicDecayTop(key, cmb)->daughter(TtSemiLepDaughter::LepW);
  };
  const reco::Candidate* singleNeutrino(const std::string& key, const unsigned& cmb = 0) const {
    return singleNeutrino(hypoClassKeyFromString(key), cmb);
  };
  const reco::Candidate* singleNeutrino(const HypoClassKey& key, const unsigned& cmb = 0) const {
    return !isHypoValid(key, cmb) ? nullptr : leptonicDecayW(key, cmb)->daughter(TtSemiLepDaughter::Nu);
  };
  const reco::Candidate* singleLepton(const std::string& key, const unsigned& cmb = 0) const {
    return singleLepton(hypoClassKeyFromString(key), cmb);
  };
  const reco::Candidate* singleLepton(const HypoClassKey& key, const unsigned& cmb = 0) const {
    return !isHypoValid(key, cmb) ? nullptr : leptonicDecayW(key, cmb)->daughter(TtSemiLepDaughter::Lep);
  };

  const reco::GenParticle* hadronicDecayTop() const {
    return (!genEvt_ ? nullptr : this->genEvent()->hadronicDecayTop());
  };
  const reco::GenParticle* hadronicDecayB() const { return (!genEvt_ ? nullptr : this->genEvent()->hadronicDecayB()); };
  const reco::GenParticle* hadronicDecayW() const { return (!genEvt_ ? nullptr : this->genEvent()->hadronicDecayW()); };
  const reco::GenParticle* hadronicDecayQuark() const {
    return (!genEvt_ ? nullptr : this->genEvent()->hadronicDecayQuark());
  };
  const reco::GenParticle* hadronicDecayQuarkBar() const {
    return (!genEvt_ ? nullptr : this->genEvent()->hadronicDecayQuarkBar());
  };
  const reco::GenParticle* leptonicDecayTop() const {
    return (!genEvt_ ? nullptr : this->genEvent()->leptonicDecayTop());
  };
  const reco::GenParticle* leptonicDecayB() const { return (!genEvt_ ? nullptr : this->genEvent()->leptonicDecayB()); };
  const reco::GenParticle* leptonicDecayW() const { return (!genEvt_ ? nullptr : this->genEvent()->leptonicDecayW()); };
  const reco::GenParticle* singleLepton() const { return (!genEvt_ ? nullptr : this->genEvent()->singleLepton()); };
  const reco::GenParticle* singleNeutrino() const { return (!genEvt_ ? nullptr : this->genEvent()->singleNeutrino()); };

  void print(const int verbosity = 1) const;

  const int numberOfRealNeutrinoSolutions(const HypoClassKey& key) const {
    return (numberOfRealNeutrinoSolutions_.find(key) == numberOfRealNeutrinoSolutions_.end()
                ? -999
                : numberOfRealNeutrinoSolutions_.find(key)->second);
  };
  const int numberOfRealNeutrinoSolutions(const std::string& key) const {
    return numberOfRealNeutrinoSolutions(hypoClassKeyFromString(key));
  };

  void setNumberOfRealNeutrinoSolutions(const HypoClassKey& key, const int& nr) {
    numberOfRealNeutrinoSolutions_[key] = nr;
  };

protected:
  std::map<HypoClassKey, int> numberOfRealNeutrinoSolutions_;
};

#endif