METzCalculator.h

Go to the documentation of this file.

#ifndef METzCalculator_h
#define METzCalculator_h

//#include "FWCore/Framework/interface/Event.h"
//#include "FWCore/ParameterSet/interface/ParameterSet.h"

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

#include "TLorentzVector.h"

class METzCalculator {
public:
  METzCalculator();
  //METzCalculator(const edm::ParameterSEt& iConf);
  virtual ~METzCalculator();
  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 SetMuon(const pat::Particle& lepton) { lepton_ = lepton; };
  void SetMuon(const TLorentzVector& lepton) {
    pat::Particle::LorentzVector p(lepton.Px(), lepton.Py(), lepton.Pz(), lepton.E());
    lepton_.setP4(p);
  }
  double Calculate(int type = 0);
  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_;
};

#endif