UniqueMerger.h

Go to the documentation of this file.

#ifndef UtilAlgos_UniqueMerger_h
#define UtilAlgos_UniqueMerger_h
 
#include "FWCore/Framework/interface/global/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/transform.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "DataFormats/Common/interface/CloneTrait.h"
#include <vector>
 
template <typename InputCollection,
          typename OutputCollection = InputCollection,
          typename P = typename edm::clonehelper::CloneTrait<InputCollection>::type>
class UniqueMerger : public edm::global::EDProducer<> {
public:
  typedef std::set<typename OutputCollection::value_type> set_type;
  explicit UniqueMerger(const edm::ParameterSet&);
  ~UniqueMerger() override;
 
private:
  void produce(edm::StreamID, edm::Event&, const edm::EventSetup&) const override;
  typedef std::vector<edm::EDGetTokenT<InputCollection> > vtoken;
  vtoken srcToken_;
};
 
template <typename InputCollection, typename OutputCollection, typename P>
UniqueMerger<InputCollection, OutputCollection, P>::UniqueMerger(const edm::ParameterSet& par)
    : srcToken_(edm::vector_transform(par.template getParameter<std::vector<edm::InputTag> >("src"),
                                      [this](edm::InputTag const& tag) { return consumes<InputCollection>(tag); })) {
  produces<OutputCollection>();
}
 
template <typename InputCollection, typename OutputCollection, typename P>
UniqueMerger<InputCollection, OutputCollection, P>::~UniqueMerger() {}
 
template <typename InputCollection, typename OutputCollection, typename P>
void UniqueMerger<InputCollection, OutputCollection, P>::produce(edm::StreamID,
                                                              edm::Event& evt,
                                                              const edm::EventSetup&) const {
  set_type coll_set;
  for (typename vtoken::const_iterator s = srcToken_.begin(); s != srcToken_.end(); ++s) {
    edm::Handle<InputCollection> h;
    evt.getByToken(*s, h);
    for (typename InputCollection::const_iterator c = h->begin(); c != h->end(); ++c) {
      coll_set.emplace(P::clone(*c));
    }
  }
  std::unique_ptr<OutputCollection> coll(new OutputCollection(coll_set.size()));
  std::copy(coll_set.begin(), coll_set.end(), coll->begin());
  evt.put(std::move(coll));
}
 
#endif