TtSemiLepJetComb.h

Go to the documentation of this file.

#ifndef TtSemiLepJetComb_h
#define TtSemiLepJetComb_h

#include <vector>
#include <string>

#include "TMath.h"

#include "DataFormats/PatCandidates/interface/Jet.h"
#include "DataFormats/PatCandidates/interface/MET.h"

namespace JetComb {
  enum DecayType { kHad, kLep };
  enum VarType { kMass, kPt, kEta, kPhi, kTheta };
  enum CompType { kDeltaM, kDeltaR, kDeltaPhi, kDeltaTheta };
  enum BTagAlgo {
    kTrackCountHighEff,
    kTrackCountHighPur,
    kSoftMuon,
    kSoftMuonByPt,
    kSofMuonByIP3d,
    kSoftElec,
    kBProbability,
    kProbability,
    kSimpleSecondVtx,
    kCombSecondVtx,
    kCombSecondVtxMVA
  };
  enum Operator { kAdd, kMult };
}  // namespace JetComb

class TtSemiLepJetComb {
public:
  TtSemiLepJetComb();
  TtSemiLepJetComb(const std::vector<pat::Jet>&,
                   const std::vector<int>&,
                   const math::XYZTLorentzVector&,
                   const pat::MET&);
  ~TtSemiLepJetComb();

  double topVar(JetComb::DecayType decay, JetComb::VarType var) const;
  double wBosonVar(JetComb::DecayType decay, JetComb::VarType var) const;
  double bQuarkVar(JetComb::DecayType decay, JetComb::VarType var) const;
  double lightQVar(bool qbar, JetComb::VarType var) const;
  double leptonVar(JetComb::VarType var) const;
  double neutrinoVar(JetComb::VarType var) const;

  double compareHadTopLepTop(JetComb::CompType comp) const;
  double compareHadWLepW(JetComb::CompType comp) const;
  double compareHadBLepB(JetComb::CompType comp) const;
  double compareLightQuarks(JetComb::CompType comp) const;
  double compareLeptonNeutrino(JetComb::CompType comp) const;
  double compareTopW(JetComb::DecayType dec1, JetComb::DecayType dec2, JetComb::CompType comp) const;
  double compareTopB(JetComb::DecayType dec1, JetComb::DecayType dec2, JetComb::CompType comp) const;
  double compareWB(JetComb::DecayType dec1, JetComb::DecayType dec2, JetComb::CompType comp) const;
  double compareTopLepton(JetComb::DecayType decay, JetComb::CompType comp) const;
  double compareTopNeutrino(JetComb::DecayType decay, JetComb::CompType comp) const;
  double compareWLepton(JetComb::DecayType decay, JetComb::CompType comp) const;
  double compareWNeutrino(JetComb::DecayType decay, JetComb::CompType comp) const;
  double compareBLepton(JetComb::DecayType decay, JetComb::CompType comp) const;
  double compareBNeutrino(JetComb::DecayType decay, JetComb::CompType comp) const;

  double relativePtHadronicTop() const;
  double bOverLightQPt() const;

  double bTag(JetComb::DecayType decay, JetComb::BTagAlgo algo) const { return bTag(bQuark(decay), algo); };
  double combinedBTags(JetComb::BTagAlgo algo, JetComb::Operator op) const;
  double combinedBTagsForLightQuarks(JetComb::BTagAlgo algo, JetComb::Operator op) const;

  double addUserVar(std::string key, double value) { return userVariables_[key] = value; };
  double userVar(const std::string& key) const {
    return (userVariables_.find(key) != userVariables_.end() ? userVariables_.find(key)->second : -9999.);
  };

private:
  void deduceMothers();
  double bTag(const pat::Jet& jet, JetComb::BTagAlgo algo) const;
  double lightQVar(JetComb::VarType var) const { return lightQVar(false, var); };
  const pat::Jet& lightQ(bool qbar = false) const { return (qbar ? hadQBarJet_ : hadQJet_); }
  const pat::Jet& bQuark(JetComb::DecayType decay) const { return (decay == JetComb::kHad ? hadBJet_ : lepBJet_); }
  const math::XYZTLorentzVector& wBoson(JetComb::DecayType decay) const {
    return (decay == JetComb::kHad ? hadW_ : lepW_);
  }
  const math::XYZTLorentzVector& top(JetComb::DecayType decay) const {
    return (decay == JetComb::kHad ? hadTop_ : lepTop_);
  }

private:
  pat::Jet hadQJet_;
  pat::Jet hadQBarJet_;
  pat::Jet hadBJet_;
  pat::Jet lepBJet_;
  pat::MET neutrino_;
  math::XYZTLorentzVector lepton_;
  math::XYZTLorentzVector hadTop_;
  math::XYZTLorentzVector hadW_;
  math::XYZTLorentzVector lepTop_;
  math::XYZTLorentzVector lepW_;
  std::map<std::string, double> userVariables_;
};

#endif