PFMETAlgo.cc

Go to the documentation of this file.

#include "CommonTools/ParticleFlow/interface/PFMETAlgo.h"
 
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
 
#include "DataFormats/METReco/interface/MET.h"
#include "DataFormats/METReco/interface/METFwd.h"
#include "DataFormats/Math/interface/LorentzVector.h"
 
#include "FWCore/Framework/interface/EventSetup.h"
 
using namespace std;
using namespace edm;
using namespace reco;
using namespace math;
using namespace pf2pat;
 
PFMETAlgo::PFMETAlgo(const edm::ParameterSet& iConfig) {
  verbose_ = iConfig.getUntrackedParameter<bool>("verbose", false);
 
  hfCalibFactor_ = iConfig.getParameter<double>("hfCalibFactor");
}
 
PFMETAlgo::~PFMETAlgo() {}
 
reco::MET PFMETAlgo::produce(const reco::PFCandidateCollection& pfCandidates) {
  double sumEx = 0;
  double sumEy = 0;
  double sumEt = 0;
 
  for (unsigned i = 0; i < pfCandidates.size(); i++) {
    const reco::PFCandidate& cand = pfCandidates[i];
 
    double E = cand.energy();
 
    if (cand.particleId() == PFCandidate::h_HF || cand.particleId() == PFCandidate::egamma_HF)
      E *= hfCalibFactor_;
 
    double phi = cand.phi();
    double cosphi = cos(phi);
    double sinphi = sin(phi);
 
    double theta = cand.theta();
    double sintheta = sin(theta);
 
    double et = E * sintheta;
    double ex = et * cosphi;
    double ey = et * sinphi;
 
    sumEx += ex;
    sumEy += ey;
    sumEt += et;
  }
 
  double Et = sqrt(sumEx * sumEx + sumEy * sumEy);
  XYZTLorentzVector missingEt(-sumEx, -sumEy, 0, Et);
 
  if (verbose_) {
    cout << "PFMETAlgo: mEx, mEy, mEt = " << missingEt.X() << ", " << missingEt.Y() << ", " << missingEt.T() << endl;
  }
 
  XYZPoint vertex;  // dummy vertex
  return MET(sumEt, missingEt, vertex);
}