TtFullLeptonicEvent.h

Go to the documentation of this file.

#ifndef TopObjects_TtFullLeptonicEvent_h
#define TopObjects_TtFullLeptonicEvent_h

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

namespace TtFullLepDaughter {
  static const std::string Nu = "Nu", LepBar = "LepBar", WPlus = "WPlus", B = "B", Top = "Top";
  static const std::string NuBar = "NuBar", Lep = "Lep", WMinus = "WMinus", BBar = "BBar", TopBar = "TopBar";
}  // namespace TtFullLepDaughter

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

  const reco::Candidate* top(const std::string& key, const unsigned& cmb = 0) const {
    return top(hypoClassKeyFromString(key), cmb);
  };
  const reco::Candidate* top(const HypoClassKey& key, const unsigned& cmb = 0) const {
    return !isHypoValid(key, cmb) ? nullptr : eventHypo(key, cmb).daughter(TtFullLepDaughter::Top);
  };
  const reco::Candidate* b(const std::string& key, const unsigned& cmb = 0) const {
    return b(hypoClassKeyFromString(key), cmb);
  };
  const reco::Candidate* b(const HypoClassKey& key, const unsigned& cmb = 0) const {
    return !isHypoValid(key, cmb) ? nullptr : top(key, cmb)->daughter(TtFullLepDaughter::B);
  };
  const reco::Candidate* wPlus(const std::string& key, const unsigned& cmb = 0) const {
    return wPlus(hypoClassKeyFromString(key), cmb);
  };
  const reco::Candidate* wPlus(const HypoClassKey& key, const unsigned& cmb = 0) const {
    return !isHypoValid(key, cmb) ? nullptr : top(key, cmb)->daughter(TtFullLepDaughter::WPlus);
  };
  const reco::Candidate* leptonBar(const std::string& key, const unsigned& cmb = 0) const {
    return leptonBar(hypoClassKeyFromString(key), cmb);
  };
  const reco::Candidate* leptonBar(const HypoClassKey& key, const unsigned& cmb = 0) const {
    return !isHypoValid(key, cmb) ? nullptr : wPlus(key, cmb)->daughter(TtFullLepDaughter::LepBar);
  };
  const reco::Candidate* neutrino(const std::string& key, const unsigned& cmb = 0) const {
    return neutrino(hypoClassKeyFromString(key), cmb);
  };
  const reco::Candidate* neutrino(const HypoClassKey& key, const unsigned& cmb = 0) const {
    return !isHypoValid(key, cmb) ? nullptr : wPlus(key, cmb)->daughter(TtFullLepDaughter::Nu);
  };
  const reco::Candidate* topBar(const std::string& key, const unsigned& cmb = 0) const {
    return topBar(hypoClassKeyFromString(key), cmb);
  };
  const reco::Candidate* topBar(const HypoClassKey& key, const unsigned& cmb = 0) const {
    return !isHypoValid(key, cmb) ? nullptr : eventHypo(key, cmb).daughter(TtFullLepDaughter::TopBar);
  };
  const reco::Candidate* bBar(const std::string& key, const unsigned& cmb = 0) const {
    return bBar(hypoClassKeyFromString(key), cmb);
  };
  const reco::Candidate* bBar(const HypoClassKey& key, const unsigned& cmb = 0) const {
    return !isHypoValid(key, cmb) ? nullptr : topBar(key, cmb)->daughter(TtFullLepDaughter::BBar);
  };
  const reco::Candidate* wMinus(const std::string& key, const unsigned& cmb = 0) const {
    return wMinus(hypoClassKeyFromString(key), cmb);
  };
  const reco::Candidate* wMinus(const HypoClassKey& key, const unsigned& cmb = 0) const {
    return !isHypoValid(key, cmb) ? nullptr : topBar(key, cmb)->daughter(TtFullLepDaughter::WMinus);
  };
  const reco::Candidate* lepton(const std::string& key, const unsigned& cmb = 0) const {
    return lepton(hypoClassKeyFromString(key), cmb);
  };
  const reco::Candidate* lepton(const HypoClassKey& key, const unsigned& cmb = 0) const {
    return !isHypoValid(key, cmb) ? nullptr : wMinus(key, cmb)->daughter(TtFullLepDaughter::Lep);
  };
  const reco::Candidate* neutrinoBar(const std::string& key, const unsigned& cmb = 0) const {
    return neutrinoBar(hypoClassKeyFromString(key), cmb);
  };
  const reco::Candidate* neutrinoBar(const HypoClassKey& key, const unsigned& cmb = 0) const {
    return !isHypoValid(key, cmb) ? nullptr : wMinus(key, cmb)->daughter(TtFullLepDaughter::NuBar);
  };

  const reco::GenParticle* genTop() const { return (!genEvt_ ? nullptr : this->genEvent()->top()); };
  const reco::GenParticle* genB() const { return (!genEvt_ ? nullptr : this->genEvent()->b()); };
  const reco::GenParticle* genWPlus() const { return (!genEvt_ ? nullptr : this->genEvent()->wPlus()); };
  const reco::GenParticle* genLeptonBar() const { return (!genEvt_ ? nullptr : this->genEvent()->leptonBar()); };
  const reco::GenParticle* genNeutrino() const { return (!genEvt_ ? nullptr : this->genEvent()->neutrino()); };
  const reco::GenParticle* genTopBar() const { return (!genEvt_ ? nullptr : this->genEvent()->topBar()); };
  const reco::GenParticle* genBBar() const { return (!genEvt_ ? nullptr : this->genEvent()->bBar()); };
  const reco::GenParticle* genWMinus() const { return (!genEvt_ ? nullptr : this->genEvent()->wMinus()); };
  const reco::GenParticle* genLepton() const { return (!genEvt_ ? nullptr : this->genEvent()->lepton()); };
  const reco::GenParticle* genNeutrinoBar() const { return (!genEvt_ ? nullptr : this->genEvent()->neutrinoBar()); };

  double solWeight(const unsigned& cmb = 0) const { return (cmb < solWeight_.size() ? solWeight_[cmb] : -1.); }
  bool isWrongCharge() const { return wrongCharge_; }

  void setSolWeight(const std::vector<double>& val) { solWeight_ = val; };
  void setWrongCharge(const bool& val) { wrongCharge_ = val; };

  void print(const int verbosity = 1) const;

protected:
  std::vector<double> solWeight_;
  bool wrongCharge_;
};

#endif