/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_6_2_5/src/TopQuarkAnalysis/TopPairBSM/src/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(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(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