CommonTools/UtilAlgos/interface/CollectionAdder.h - master - githubmirror/cmssw

CommonTools/UtilAlgos/interface/CollectionAdder.h

40 строк1 KB

Christopher Jones

Modernized module in PhysicsTools/HepMCCandAlgos

03 дек 2021, 23:47

03 дек 2021, 23:47967b8d9

#ifndef PhysicsTools_UtilAlgos_CollectionAdder_h
#define PhysicsTools_UtilAlgos_CollectionAdder_h
/* \class CollectionAdder<C>
 *
 * \author Luca Lista, INFN
 *
 * \version $Id: CollectionAdder.h,v 1.3 2010/02/20 20:55:17 wmtan Exp $
 */
#include "FWCore/Framework/interface/global/EDProducer.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/transform.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/Framework/interface/Event.h"
#include "DataFormats/Common/interface/Handle.h"

template <typename C>
class CollectionAdder : public edm::global::EDProducer<> {
public:
  typedef C collection;
  CollectionAdder(const edm::ParameterSet& cfg)
      : srcTokens_(edm::vector_transform(cfg.template getParameter<std::vector<edm::InputTag>>("src"),
                                         [this](edm::InputTag const& tag) { return consumes<collection>(tag); })) {
    produces<collection>();
  }

private:
  std::vector<edm::EDGetTokenT<collection>> srcTokens_;
  void produce(edm::StreamID, edm::Event& evt, const edm::EventSetup&) const override {
    std::unique_ptr<collection> coll(new collection);
    typename collection::Filler filler(*coll);
    for (size_t i = 0; i < srcTokens_.size(); ++i) {
      edm::Handle<collection> src;
      evt.getByToken(srcTokens_[i], src);
      *coll += *src;
    }
    evt.put(std::move(coll));
  }
};

#endif