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 "HeterogeneousCore/AlpakaTest/interface/TestHostOnlyHelperClass.h"

#include "TestHelperClass.h"

namespace ALPAKA_ACCELERATOR_NAMESPACE {
  class TestAlpakaStreamSynchronizingProducer : public stream::SynchronizingEDProducer<> {
  public:
    TestAlpakaStreamSynchronizingProducer(edm::ParameterSet const& iConfig)
        : esTokenDevice_(esConsumes()),
          putToken_{produces()},
          putTokenMulti2_{produces()},
          putTokenMulti3_{produces()},
          helper_{iConfig, consumesCollector()},
          hostHelper_{iConfig, consumesCollector()},
          expectedInt_{iConfig.getParameter<int>("expectedInt")} {}

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

      int const value = hostHelper_.run(iEvent, iSetup);
      if (value != expectedInt_) {
        throw cms::Exception("Assert") << "Expected value " << expectedInt_ << ", but got " << value;
      }

      helper_.makeAsync(iEvent, iSetup);
    }

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

    static void fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
      edm::ParameterSetDescription desc;
      TestHelperClass::fillPSetDescription(desc);
      cms::alpakatest::TestHostOnlyHelperClass::fillPSetDescription(desc);
      desc.add<int>("expectedInt");
      descriptions.addWithDefaultLabel(desc);
    }

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

    TestHelperClass helper_;
    cms::alpakatest::TestHostOnlyHelperClass const hostHelper_;
    int const expectedInt_;
  };

}  // namespace ALPAKA_ACCELERATOR_NAMESPACE

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