|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/global/EDProducer.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "FWCore/Utilities/interface/EDPutToken.h"
#include "FWCore/Utilities/interface/StreamID.h"
#include "DataFormats/TrackingRecHitSoA/interface/TrackingRecHitsHost.h"
#include <memory>
#include <utility>
#include <vector>
namespace edmtest {
class TestWriteHostHitSoA : public edm::global::EDProducer<> {
public:
TestWriteHostHitSoA(edm::ParameterSet const&);
void produce(edm::StreamID, edm::Event&, edm::EventSetup const&) const override;
static void fillDescriptions(edm::ConfigurationDescriptions&);
using HitsOnHost = ::reco::TrackingRecHitHost;
private:
unsigned int hitSize_;
edm::EDPutTokenT<HitsOnHost> putToken_;
};
TestWriteHostHitSoA::TestWriteHostHitSoA(edm::ParameterSet const& iPSet)
: hitSize_(iPSet.getParameter<unsigned int>("hitSize")), putToken_(produces()) {}
void TestWriteHostHitSoA::produce(edm::StreamID, edm::Event& iEvent, edm::EventSetup const&) const {
HitsOnHost hits(cms::alpakatools::host(), hitSize_, 100);
auto hitsView = hits.view();
for (unsigned int i = 0; i < hitSize_; ++i) {
hitsView[i].xGlobal() = float(i);
}
iEvent.emplace(putToken_, std::move(hits));
}
void TestWriteHostHitSoA::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
edm::ParameterSetDescription desc;
desc.add<unsigned int>("hitSize", 1000);
descriptions.addDefault(desc);
}
} // namespace edmtest
using edmtest::TestWriteHostHitSoA;
DEFINE_FWK_MODULE(TestWriteHostHitSoA);
|