MEzCalculator.h

Go to the documentation of this file.

#ifndef TopTools_MEzCalculator_h
#define TopTools_MEzCalculator_h

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

#include "TLorentzVector.h"

class MEzCalculator {
public:
  MEzCalculator();
  ~MEzCalculator();
  void SetMET(const pat::MET& MET) { MET_ = MET; };
  void SetMET(const TLorentzVector& MET) {
    pat::Particle::LorentzVector p(MET.Px(), MET.Py(), MET.Pz(), MET.E());
    MET_.setP4(p);
  }
  void SetLepton(const pat::Particle& lepton, bool isMuon = true) {
    lepton_ = lepton;
    isMuon_ = isMuon;
  };
  void SetLepton(const TLorentzVector& lepton) {
    pat::Particle::LorentzVector p(lepton.Px(), lepton.Py(), lepton.Pz(), lepton.E());
    lepton_.setP4(p);
  }
  double Calculate(int type = 1);
  bool IsComplex() const { return isComplex_; };
  void Print() {
    std::cout << " METzCalculator: pxmu = " << lepton_.px() << " pzmu= " << lepton_.pz() << std::endl;
    std::cout << " METzCalculator: pxnu = " << MET_.px() << " pynu= " << MET_.py() << std::endl;
  }

private:
  bool isComplex_;
  pat::Particle lepton_;
  pat::MET MET_;
  bool isMuon_;
};

#endif