AssembleProducts_module.cc

Go to the documentation of this file.

#include "art/Framework/Core/EDProducer.h"
#include "art/Framework/Core/ModuleMacros.h"
#include "art/Framework/Principal/Event.h"
#include "art/Framework/Principal/SubRun.h"
#include "canvas/Utilities/InputTag.h"
#include "fhiclcpp/types/TupleAs.h"

#include <atomic>

using art::InputTag;
using fhicl::Name;
using fhicl::TupleAs;
using std::string;

namespace {

  struct Config {
    TupleAs<InputTag(string)> particlesTag{Name("particlesTag")};
  };

  class AssembleProducts : public art::EDProducer {
  public:
    using Parameters = EDProducer::Table<Config>;
    explicit AssembleProducts(Parameters const& config)
      : EDProducer{config}, particlesTag_{config().particlesTag()}
    {
      produces<unsigned, art::InSubRun>("seenParticles");
    }

  private:
    void
    produce(art::Event& e) override
    {
      auto const& particles =
        e.getValidHandle<std::vector<double>>(particlesTag_);
      seenParticles_ += particles->size();
    }

    void
    endSubRun(art::SubRun& sr) override
    {
      sr.put(std::make_unique<unsigned>(seenParticles_),
             "seenParticles",
             art::subRunFragment());
      seenParticles_ = 0u;
    }

    InputTag const particlesTag_;
    std::atomic<unsigned> seenParticles_{};

  }; // AssembleProducts
} // namespace

DEFINE_ART_MODULE(AssembleProducts)