MinMETProducerT.h

Go to the documentation of this file.

#ifndef RecoMET_METProducers_MinMETProducerT_h
#define RecoMET_METProducers_MinMETProducerT_h

#include "FWCore/Framework/interface/stream/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 "FWCore/Utilities/interface/Exception.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"

#include <vector>

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

 public:

  explicit MinMETProducerT(const edm::ParameterSet& cfg)    : moduleLabel_(cfg.getParameter<std::string>("@module_label"))
  {
    src_ = cfg.getParameter<vInputTag>("src");

    for ( vInputTag::const_iterator src_i = src_.begin();
      src_i != src_.end(); ++src_i ) {
      src_token_.push_back( consumes<METCollection>(*src_i) );
    }
    produces<METCollection>();
  }
  ~MinMETProducerT() override {}

 private:

  void produce(edm::Event& evt, const edm::EventSetup& es) override
  {
    auto outputMETs = std::make_unique<METCollection>();

    // check that all MET collections given as input have the same number of entries
    int numMEtObjects = -1;
    //    for ( vInputTag::const_iterator src_i = src_.begin();
    //        src_i != src_.end(); ++src_i ) {
    for ( typename vInputToken::const_iterator src_i = src_token_.begin();
      src_i != src_token_.end(); ++src_i ) {
      edm::Handle<METCollection> inputMETs;
      //      evt.getByLabel(*src_i, inputMETs);
      evt.getByToken(*src_i, inputMETs);
      if ( numMEtObjects == -1 ) numMEtObjects = inputMETs->size();
      else if ( numMEtObjects != (int)inputMETs->size() )
    throw cms::Exception("MinMETProducer::produce")
      << "Mismatch in number of input MET objects !!\n";
    }

    for ( int iMEtObject = 0; iMEtObject < numMEtObjects; ++iMEtObject ) {
      const T* minMET = nullptr;
      //      for ( vInputTag::const_iterator src_i = src_.begin();
      //        src_i != src_.end(); ++src_i ) {
      for ( typename vInputToken::const_iterator src_i = src_token_.begin();
        src_i != src_token_.end(); ++src_i ) {
    edm::Handle<METCollection> inputMETs;
    //  evt.getByLabel(*src_i, inputMETs);
    evt.getByToken(*src_i, inputMETs);
    const T& inputMET = inputMETs->at(iMEtObject);
    if ( minMET == nullptr || inputMET.pt() < minMET->pt() ) minMET = &inputMET;
      }
      assert(minMET);
      outputMETs->push_back(T(*minMET));
    }

    evt.put(std::move(outputMETs));
  }

  std::string moduleLabel_;

  typedef std::vector<edm::InputTag> vInputTag;
  vInputTag src_;
  typedef std::vector<edm::EDGetTokenT<METCollection> > vInputToken;
  vInputToken src_token_;
};

#include "DataFormats/METReco/interface/CaloMET.h"
#include "DataFormats/METReco/interface/PFMET.h"

namespace reco
{
  typedef MinMETProducerT<reco::CaloMET> MinCaloMETProducer;
  typedef MinMETProducerT<reco::PFMET> MinPFMETProducer;
}

#endif