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};
}

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