TrackSoA/test/TestWriteHostTrackSoA.cc

0001 #include "FWCore/Framework/interface/Event.h"
0002 #include "FWCore/Framework/interface/Frameworkfwd.h"
0003 #include "FWCore/Framework/interface/global/EDProducer.h"
0004 #include "FWCore/Framework/interface/MakerMacros.h"
0005 #include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
0006 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0007 #include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
0008 #include "FWCore/Utilities/interface/EDPutToken.h"
0009 #include "FWCore/Utilities/interface/StreamID.h"
0010
0011 #include "DataFormats/TrackSoA/interface/TracksHost.h"
0012
0013 #include <memory>
0014 #include <utility>
0015 #include <vector>
0016
0017 namespace edmtest {
0018
0019   class TestWriteHostTrackSoA : public edm::global::EDProducer<> {
0020   public:
0021     TestWriteHostTrackSoA(edm::ParameterSet const&);
0022     void produce(edm::StreamID, edm::Event&, edm::EventSetup const&) const override;
0023     static void fillDescriptions(edm::ConfigurationDescriptions&);
0024
0025   private:
0026     unsigned int trackSize_;
0027     edm::EDPutTokenT<::reco::TracksHost> putToken_;
0028   };
0029
0030   TestWriteHostTrackSoA::TestWriteHostTrackSoA(edm::ParameterSet const& iPSet)
0031       : trackSize_(iPSet.getParameter<unsigned int>("trackSize")), putToken_(produces()) {}
0032
0033   void TestWriteHostTrackSoA::produce(edm::StreamID, edm::Event& iEvent, edm::EventSetup const&) const {
0034     ::reco::TracksHost tracks({{int(trackSize_), int(4 * trackSize_)}}, cms::alpakatools::host());
0035     auto tracksView = tracks.view();
0036     for (unsigned int i = 0; i < trackSize_; ++i) {
0037       tracksView[i].eta() = float(i);
0038     }
0039     iEvent.emplace(putToken_, std::move(tracks));
0040   }
0041
0042   void TestWriteHostTrackSoA::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
0043     edm::ParameterSetDescription desc;
0044     desc.add<unsigned int>("trackSize", 1000);
0045     descriptions.addDefault(desc);
0046   }
0047 }  // namespace edmtest
0048
0049 using edmtest::TestWriteHostTrackSoA;
0050 DEFINE_FWK_MODULE(TestWriteHostTrackSoA);