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