ClassifierMerger.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/global/EDProducer.h"
 
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
 
#include <vector>
#include <memory>
 
namespace {
  class ClassifierMerger final : public edm::global::EDProducer<> {
  public:
    explicit ClassifierMerger(const edm::ParameterSet& conf) {
      for (auto const& it : conf.getParameter<std::vector<std::string>>("inputClassifiers")) {
        srcMVAs.push_back(consumes<MVACollection>(edm::InputTag(it, "MVAValues")));
        srcQuals.push_back(consumes<QualityMaskCollection>(edm::InputTag(it, "QualityMasks")));
      }
 
      produces<MVACollection>("MVAValues");
      produces<QualityMaskCollection>("QualityMasks");
    }
 
    static void fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
      edm::ParameterSetDescription desc;
      desc.add<std::vector<std::string>>("inputClassifiers", std::vector<std::string>());
      descriptions.add("ClassifierMerger", desc);
    }
 
  private:
    using MVACollection = std::vector<float>;
    using QualityMaskCollection = std::vector<unsigned char>;
 
    void produce(edm::StreamID, edm::Event& evt, const edm::EventSetup&) const override {
      // get Master
      edm::Handle<MVACollection> hmva;
      evt.getByToken(srcMVAs[0], hmva);
      auto size = (*hmva).size();
 
      edm::Handle<QualityMaskCollection> hqual;
      evt.getByToken(srcQuals[0], hqual);
 
      // products
      auto mvas = std::make_unique<MVACollection>(*hmva);
      auto quals = std::make_unique<QualityMaskCollection>(*hqual);
 
      for (auto i = 1U; i < srcQuals.size(); ++i) {
        evt.getByToken(srcQuals[i], hqual);
        auto const& iq = *hqual;
        assert(iq.size() == size);
        for (auto j = 0U; j != size; ++j)
          (*quals)[j] |= iq[j];
      }
 
      evt.put(std::move(mvas), "MVAValues");
      evt.put(std::move(quals), "QualityMasks");
    }
 
    std::vector<edm::EDGetTokenT<MVACollection>> srcMVAs;
    std::vector<edm::EDGetTokenT<QualityMaskCollection>> srcQuals;
  };
}  // namespace
 
#include "FWCore/PluginManager/interface/ModuleDef.h"
#include "FWCore/Framework/interface/MakerMacros.h"
 
DEFINE_FWK_MODULE(ClassifierMerger);