TestHelperClass.h

#ifndef HeterogeneousCore_AlpakaTest_plugins_alpaka_TestHelperClass_h
#define HeterogeneousCore_AlpakaTest_plugins_alpaka_TestHelperClass_h

#include <optional>

#include "DataFormats/PortableTestObjects/interface/TestHostCollection.h"
#include "DataFormats/PortableTestObjects/interface/alpaka/TestDeviceCollection.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "HeterogeneousCore/AlpakaCore/interface/alpaka/EDPutToken.h"
#include "HeterogeneousCore/AlpakaCore/interface/alpaka/ESGetToken.h"
#include "HeterogeneousCore/AlpakaCore/interface/alpaka/Event.h"
#include "HeterogeneousCore/AlpakaCore/interface/alpaka/EventSetup.h"
#include "HeterogeneousCore/AlpakaInterface/interface/config.h"
#include "HeterogeneousCore/AlpakaTest/interface/AlpakaESTestRecords.h"
#include "HeterogeneousCore/AlpakaTest/interface/ESTestData.h"
#include "HeterogeneousCore/AlpakaTest/interface/alpaka/AlpakaESTestData.h"

namespace ALPAKA_ACCELERATOR_NAMESPACE {
  class TestHelperClass {
  public:
    TestHelperClass(edm::ParameterSet const& iConfig, edm::ConsumesCollector iC);

    static void fillPSetDescription(edm::ParameterSetDescription& iDesc);

    void makeAsync(device::Event const& iEvent, device::EventSetup const& iSetup);

    portabletest::TestHostCollection moveFrom() {
      auto product = std::move(*hostProduct_);
      hostProduct_.reset();
      return product;
    }

    portabletest::TestHostMultiCollection2 moveFromMulti2() {
      auto product = std::move(*hostProductMulti2_);
      hostProductMulti2_.reset();
      return product;
    }

    portabletest::TestHostMultiCollection3 moveFromMulti3() {
      auto product = std::move(*hostProductMulti3_);
      hostProductMulti3_.reset();
      return product;
    }

  private:
    const device::EDGetToken<portabletest::TestDeviceCollection> getToken_;
    const device::EDGetToken<portabletest::TestDeviceMultiCollection2> getTokenMulti2_;
    const device::EDGetToken<portabletest::TestDeviceMultiCollection3> getTokenMulti3_;
    const edm::ESGetToken<cms::alpakatest::ESTestDataA, AlpakaESTestRecordA> esTokenHost_;
    const device::ESGetToken<AlpakaESTestDataCDevice, AlpakaESTestRecordC> esTokenDevice_;

    // hold the output product between acquire() and produce()
    std::optional<portabletest::TestHostCollection> hostProduct_;
    std::optional<portabletest::TestHostMultiCollection2> hostProductMulti2_;
    std::optional<portabletest::TestHostMultiCollection3> hostProductMulti3_;
  };
}  // namespace ALPAKA_ACCELERATOR_NAMESPACE

#endif