HcalCPURecHitsProducer Class Reference

Inheritance diagram for HcalCPURecHitsProducer:

Public Member Functions
	HcalCPURecHitsProducer (edm::ParameterSet const &ps)
	~HcalCPURecHitsProducer () override=default
Public Member Functions inherited from edm::stream::EDProducer< edm::ExternalWork >
	EDProducer ()=default
	EDProducer (const EDProducer &)=delete
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final
const EDProducer &	operator= (const EDProducer &)=delete

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &)

Private Types
using	IProductType = cms::cuda::Product< hcal::RecHitCollection< calo::common::DevStoragePolicy > >
using	OProductType = hcal::RecHitCollection< calo::common::VecStoragePolicy< calo::common::CUDAHostAllocatorAlias > >

Private Member Functions
void	acquire (edm::Event const &, edm::EventSetup const &, edm::WaitingTaskWithArenaHolder) override
void	produce (edm::Event &, edm::EventSetup const &) override

Private Attributes
const bool	produceLegacy_
const bool	produceSoA_
const edm::EDPutTokenT< HBHERecHitCollection >	recHitsLegacyTokenOut_
const edm::EDGetTokenT< IProductType >	recHitsM0TokenIn_
const edm::EDPutTokenT< OProductType >	recHitsM0TokenOut_
OProductType	tmpRecHits_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer< edm::ExternalWork >
using	CacheTypes = CacheContexts< T... >
using	GlobalCache = typename CacheTypes::GlobalCache
using	HasAbility = AbilityChecker< T... >
using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
using	RunCache = typename CacheTypes::RunCache
using	RunContext = RunContextT< RunCache, GlobalCache >
using	RunSummaryCache = typename CacheTypes::RunSummaryCache

Detailed Description

Definition at line 15 of file HcalCPURecHitsProducer.cc.

Member Typedef Documentation

IProductType

using HcalCPURecHitsProducer::IProductType = cms::cuda::Product<hcal::RecHitCollection<calo::common::DevStoragePolicy> >

private

Definition at line 30 of file HcalCPURecHitsProducer.cc.

OProductType

using HcalCPURecHitsProducer::OProductType = hcal::RecHitCollection<calo::common::VecStoragePolicy<calo::common::CUDAHostAllocatorAlias> >

private

Definition at line 33 of file HcalCPURecHitsProducer.cc.

Constructor & Destructor Documentation

HcalCPURecHitsProducer()

HcalCPURecHitsProducer::HcalCPURecHitsProducer ( edm::ParameterSet const &  ps )

explicit

Definition at line 53 of file HcalCPURecHitsProducer.cc.

References edm::ParameterSet::getParameter().

    : produceSoA_{ps.getParameter<bool>("produceSoA")},
      produceLegacy_{ps.getParameter<bool>("produceLegacy")},
      recHitsM0TokenIn_{consumes<IProductType>(ps.getParameter<edm::InputTag>("recHitsM0LabelIn"))},
      recHitsM0TokenOut_{produceSoA_ ? produces<OProductType>(ps.getParameter<std::string>("recHitsM0LabelOut"))
                                     : edm::EDPutTokenT<OProductType>{}},  // empty token if disabled
      recHitsLegacyTokenOut_{produceLegacy_
                                 ? produces<HBHERecHitCollection>(ps.getParameter<std::string>("recHitsLegacyLabelOut"))
                                 : edm::EDPutTokenT<HBHERecHitCollection>{}}  // empty token if disabled
{}

~HcalCPURecHitsProducer()

HcalCPURecHitsProducer::~HcalCPURecHitsProducer ( )

overridedefault

Member Function Documentation

acquire()

void HcalCPURecHitsProducer::acquire ( edm::Event const &  event, edm::EventSetup const &  setup, edm::WaitingTaskWithArenaHolder  taskHolder  )

overrideprivate

Definition at line 64 of file HcalCPURecHitsProducer.cc.

References hcal::RecHitCollection< StoragePolicy >::chi2, gather_cfg::cout, cudaCheck, mps_fire::dest, hcal::RecHitCollection< StoragePolicy >::did, hcal::RecHitCollection< StoragePolicy >::energy, hcal::RecHitCollection< StoragePolicy >::energyM0, eostools::move(), FastTrackerRecHitMaskProducer_cfi::recHits, recHitsM0TokenIn_, hcal::RecHitCollection< StoragePolicy >::resize(), TrackRefitter_38T_cff::src, hcal::RecHitCollection< StoragePolicy >::timeM0, tmpRecHits_, and relativeConstraints::value.

                                                                               {
  // retrieve data/ctx
  auto const& recHitsProduct = event.get(recHitsM0TokenIn_);
  cms::cuda::ScopedContextAcquire ctx{recHitsProduct, std::move(taskHolder)};
  auto const& recHits = ctx.get(recHitsProduct);

  // resize tmp buffers
  tmpRecHits_.resize(recHits.size);

#ifdef HCAL_MAHI_CPUDEBUG
  std::cout << "num rec Hits = " << recHits.size << std::endl;
#endif

  // do not try to copy the rechits if they are empty
  if (recHits.size == 0) {
    return;
  }

  auto lambdaToTransfer = [&ctx](auto& dest, auto* src) {
    using vector_type = typename std::remove_reference<decltype(dest)>::type;
    using src_data_type = typename std::remove_pointer<decltype(src)>::type;
    using type = typename vector_type::value_type;
    static_assert(std::is_same<src_data_type, type>::value && "Dest and Src data types do not match");
    cudaCheck(cudaMemcpyAsync(dest.data(), src, dest.size() * sizeof(type), cudaMemcpyDeviceToHost, ctx.stream()));
  };

  lambdaToTransfer(tmpRecHits_.energy, recHits.energy.get());
  lambdaToTransfer(tmpRecHits_.chi2, recHits.chi2.get());
  lambdaToTransfer(tmpRecHits_.energyM0, recHits.energyM0.get());
  lambdaToTransfer(tmpRecHits_.timeM0, recHits.timeM0.get());
  lambdaToTransfer(tmpRecHits_.did, recHits.did.get());
}

fillDescriptions()

void HcalCPURecHitsProducer::fillDescriptions ( edm::ConfigurationDescriptions &  confDesc )

static

Definition at line 41 of file HcalCPURecHitsProducer.cc.

References edm::ConfigurationDescriptions::addWithDefaultLabel(), submitPVResolutionJobs::desc, and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                                    {
  edm::ParameterSetDescription desc;

  desc.add<edm::InputTag>("recHitsM0LabelIn", edm::InputTag{"hbheRecHitProducerGPU"});
  desc.add<std::string>("recHitsM0LabelOut", "");
  desc.add<std::string>("recHitsLegacyLabelOut", "");
  desc.add<bool>("produceSoA", true);
  desc.add<bool>("produceLegacy", true);

  confDesc.addWithDefaultLabel(desc);
}

produce()

void HcalCPURecHitsProducer::produce ( edm::Event &  event, edm::EventSetup const &  setup  )

overrideprivate

Definition at line 99 of file HcalCPURecHitsProducer.cc.

References hcal::RecHitCollection< StoragePolicy >::chi2, hcal::RecHitCollection< StoragePolicy >::did, hcal::RecHitCollection< StoragePolicy >::energy, hcal::RecHitCollection< StoragePolicy >::energyM0, mps_fire::i, eostools::move(), produceLegacy_, produceSoA_, recHitsLegacyTokenOut_, recHitsM0TokenOut_, hcal::RecHitCollection< StoragePolicy >::resize(), and tmpRecHits_.

                                                                                {
  if (produceLegacy_) {
    // populate the legacy collection
    auto recHitsLegacy = std::make_unique<HBHERecHitCollection>();
    // did not set size with ctor as there is no setter for did
    recHitsLegacy->reserve(tmpRecHits_.did.size());
    for (uint32_t i = 0; i < tmpRecHits_.did.size(); i++) {
      // skip bad channels
      if (tmpRecHits_.chi2[i] < 0)
        continue;

      // build a legacy rechit with the computed detid and MAHI energy
      recHitsLegacy->emplace_back(HcalDetId{tmpRecHits_.did[i]},
                                  tmpRecHits_.energy[i],
                                  0  // timeRising
      );
      // update the legacy rechit with the Chi2 and M0 values
      recHitsLegacy->back().setChiSquared(tmpRecHits_.chi2[i]);
      recHitsLegacy->back().setRawEnergy(tmpRecHits_.energyM0[i]);
    }

    // put the legacy collection
    event.put(recHitsLegacyTokenOut_, std::move(recHitsLegacy));
  }

  if (produceSoA_) {
    // put the SoA collection
    event.emplace(recHitsM0TokenOut_, std::move(tmpRecHits_));
  }
  // clear the temporary collection for the next event
  tmpRecHits_.resize(0);
}

Member Data Documentation

produceLegacy_

const bool HcalCPURecHitsProducer::produceLegacy_

private

Definition at line 28 of file HcalCPURecHitsProducer.cc.

Referenced by produce().

produceSoA_

const bool HcalCPURecHitsProducer::produceSoA_

private

Definition at line 27 of file HcalCPURecHitsProducer.cc.

Referenced by produce().

recHitsLegacyTokenOut_

const edm::EDPutTokenT<HBHERecHitCollection> HcalCPURecHitsProducer::recHitsLegacyTokenOut_

private

Definition at line 35 of file HcalCPURecHitsProducer.cc.

Referenced by produce().

recHitsM0TokenIn_

const edm::EDGetTokenT<IProductType> HcalCPURecHitsProducer::recHitsM0TokenIn_

private

Definition at line 31 of file HcalCPURecHitsProducer.cc.

Referenced by acquire().

recHitsM0TokenOut_

const edm::EDPutTokenT<OProductType> HcalCPURecHitsProducer::recHitsM0TokenOut_

private

Definition at line 34 of file HcalCPURecHitsProducer.cc.

Referenced by produce().

tmpRecHits_

OProductType HcalCPURecHitsProducer::tmpRecHits_

private

Definition at line 38 of file HcalCPURecHitsProducer.cc.

Referenced by acquire(), and produce().