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