CorrectedMETProducerT.h

Go to the documentation of this file.

#ifndef JetMETCorrections_Type1MET_CorrectedMETProducer_h
#define JetMETCorrections_Type1MET_CorrectedMETProducer_h

#include "FWCore/Framework/interface/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"

#include "JetMETCorrections/Type1MET/interface/METCorrectionAlgorithm.h"
#include "DataFormats/METReco/interface/CorrMETData.h"
#include "DataFormats/Candidate/interface/Candidate.h"

#include <vector>

namespace CorrectedMETProducer_namespace
{
  template <typename T>
  reco::Candidate::LorentzVector correctedP4(const T& rawMEt, const CorrMETData& correction)
  {
    double correctedMEtPx = rawMEt.px() + correction.mex;
    double correctedMEtPy = rawMEt.py() + correction.mey;
    double correctedMEtPt = sqrt(correctedMEtPx*correctedMEtPx + correctedMEtPy*correctedMEtPy);
    return reco::Candidate::LorentzVector(correctedMEtPx, correctedMEtPy, 0., correctedMEtPt);
  }

  template <typename T>
  double correctedSumEt(const T& rawMEt, const CorrMETData& correction)
  {
    return rawMEt.sumEt() + correction.sumet;
  }

  template <typename T>
  class CorrectedMETFactoryT
  {
    public:

     T operator()(const T&, const CorrMETData&) const
     {
       assert(0); // "place-holder" for template instantiations for concrete T types only, **not** to be called
     }
  };
}

template<typename T>
class CorrectedMETProducerT : public edm::EDProducer  
{
  typedef std::vector<T> METCollection;

 public:

  explicit CorrectedMETProducerT(const edm::ParameterSet& cfg)
    : moduleLabel_(cfg.getParameter<std::string>("@module_label")),
      algorithm_(0)
  {
    src_ = cfg.getParameter<edm::InputTag>("src");

    algorithm_ = new METCorrectionAlgorithm(cfg);

    produces<METCollection>("");
  }
  ~CorrectedMETProducerT()
  {
    delete algorithm_;
  }
    
 private:

  void produce(edm::Event& evt, const edm::EventSetup& es)
  {
    std::auto_ptr<METCollection> correctedMEtCollection(new METCollection);

    edm::Handle<METCollection> rawMEtCollection;
    evt.getByLabel(src_, rawMEtCollection);

    for ( typename METCollection::const_iterator rawMEt = rawMEtCollection->begin();
      rawMEt != rawMEtCollection->end(); ++rawMEt ) {
      CorrMETData correction = algorithm_->compMETCorrection(evt, es);
      
      static CorrectedMETProducer_namespace::CorrectedMETFactoryT<T> correctedMET_factory;
      T correctedMEt = correctedMET_factory(*rawMEt, correction);

      correctedMEtCollection->push_back(correctedMEt);
    }
      
//--- add collection of MET objects with Type 1 / Type 1 + 2 corrections applied to the event
    evt.put(correctedMEtCollection);
  }

  std::string moduleLabel_;

  edm::InputTag src_; // input collection

  METCorrectionAlgorithm* algorithm_; // algorithm for computing Type 1 / Type 1 + 2 MET corrections
};

#endif