TestAlpakaStreamSynchronizingProducer.cc

Go to the documentation of this file.

#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "HeterogeneousCore/AlpakaCore/interface/alpaka/stream/SynchronizingEDProducer.h"
#include "HeterogeneousCore/AlpakaInterface/interface/config.h"

#include "TestHelperClass.h"

namespace ALPAKA_ACCELERATOR_NAMESPACE {
  class TestAlpakaStreamSynchronizingProducer : public stream::SynchronizingEDProducer<> {
  public:
    TestAlpakaStreamSynchronizingProducer(edm::ParameterSet const& iConfig)
        : esTokenDevice_(esConsumes()), putToken_{produces()}, helper_{iConfig, consumesCollector()} {}

    void acquire(device::Event const& iEvent, device::EventSetup const& iSetup) override {
      [[maybe_unused]] auto const& esData = iSetup.getData(esTokenDevice_);

      helper_.makeAsync(iEvent, iSetup);
    }

    void produce(device::Event& iEvent, device::EventSetup const& iSetup) override {
      iEvent.emplace(putToken_, helper_.moveFrom());
    }

    static void fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
      edm::ParameterSetDescription desc;
      desc.add<edm::InputTag>("source");
      descriptions.addWithDefaultLabel(desc);
    }

  private:
    const device::ESGetToken<AlpakaESTestDataDDevice, AlpakaESTestRecordD> esTokenDevice_;
    const edm::EDPutTokenT<portabletest::TestHostCollection> putToken_;

    TestHelperClass helper_;
  };

}  // namespace ALPAKA_ACCELERATOR_NAMESPACE

#include "HeterogeneousCore/AlpakaCore/interface/MakerMacros.h"
DEFINE_FWK_ALPAKA_MODULE(TestAlpakaStreamSynchronizingProducer);