PFClusterCollectionMerger.cc

Go to the documentation of this file.

#ifndef __PFClusterCollectionMerger__
#define __PFClusterCollectionMerger__

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/global/EDProducer.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "DataFormats/ParticleFlowReco/interface/PFCluster.h"
#include "DataFormats/ParticleFlowReco/interface/PFClusterFwd.h"

class PFClusterCollectionMerger : public edm::global::EDProducer<> {
public: 
  PFClusterCollectionMerger(const edm::ParameterSet& conf) {
    const std::vector<edm::InputTag>& inputs = 
      conf.getParameter<std::vector<edm::InputTag> >("inputs");
    for( const auto& input : inputs ) {
      _inputs.push_back(consumes<reco::PFClusterCollection>( input ));
    }
    produces<reco::PFClusterCollection>();
  }

  virtual void produce(edm::StreamID, edm::Event& e, const edm::EventSetup& es) const {
    std::auto_ptr<reco::PFClusterCollection> output;
    output.reset(new reco::PFClusterCollection);
    for( const auto& input : _inputs ) {
      edm::Handle<reco::PFClusterCollection> handle;
      e.getByToken(input,handle);
      output->insert(output->end(),handle->begin(),handle->end());
    }
    e.put(output);
  }
private:
  std::vector<edm::EDGetTokenT<reco::PFClusterCollection> > _inputs;
};

DEFINE_FWK_MODULE(PFClusterCollectionMerger);

#endif