HGCalRecProducers/plugins/HEBRecHitFromSoA.cc

0001 #include <memory>
0002
0003 #include "FWCore/Framework/interface/stream/EDProducer.h"
0004 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0005 #include "DataFormats/HGCRecHit/interface/HGCRecHit.h"
0006 #include "DataFormats/HGCRecHit/interface/HGCRecHitCollections.h"
0007 #include "DataFormats/ForwardDetId/interface/HGCSiliconDetId.h"
0008 #include "FWCore/Utilities/interface/EDGetToken.h"
0009 #include "FWCore/Utilities/interface/EDPutToken.h"
0010
0011 #include "CUDADataFormats/HGCal/interface/HGCRecHitCPUProduct.h"
0012 #include "CUDADataFormats/HGCal/interface/ConstHGCRecHitSoA.h"
0013
0014 class HEBRecHitFromSoA : public edm::stream::EDProducer<> {
0015 public:
0016   explicit HEBRecHitFromSoA(const edm::ParameterSet& ps);
0017   ~HEBRecHitFromSoA() override;
0018
0019   void produce(edm::Event&, const edm::EventSetup&) override;
0020   void convert_soa_data_to_collection_(uint32_t, HGCRecHitCollection&, ConstHGCRecHitSoA*);
0021
0022 private:
0023   std::unique_ptr<HGChefRecHitCollection> rechits_;
0024   edm::EDGetTokenT<HGCRecHitCPUProduct> recHitSoAToken_;
0025   edm::EDPutTokenT<HGChefRecHitCollection> recHitCollectionToken_;
0026 };
0027
0028 HEBRecHitFromSoA::HEBRecHitFromSoA(const edm::ParameterSet& ps) {
0029   recHitSoAToken_ = consumes<HGCRecHitCPUProduct>(ps.getParameter<edm::InputTag>("HEBRecHitSoATok"));
0030   recHitCollectionToken_ = produces<HGChebRecHitCollection>();
0031 }
0032
0033 HEBRecHitFromSoA::~HEBRecHitFromSoA() {}
0034
0035 void HEBRecHitFromSoA::produce(edm::Event& event, const edm::EventSetup& setup) {
0036   const HGCRecHitCPUProduct& recHits = event.get(recHitSoAToken_);
0037   ConstHGCRecHitSoA recHitsSoA = recHits.get();
0038   rechits_ = std::make_unique<HGCRecHitCollection>();
0039   convert_soa_data_to_collection_(recHits.nHits(), *rechits_, &recHitsSoA);
0040   event.put(std::move(rechits_));
0041 }
0042
0043 void HEBRecHitFromSoA::convert_soa_data_to_collection_(uint32_t nhits,
0044                                                        HGCRecHitCollection& rechits,
0045                                                        ConstHGCRecHitSoA* h_calibSoA) {
0046   rechits.reserve(nhits);
0047   for (uint i = 0; i < nhits; ++i) {
0048     DetId id_converted(h_calibSoA->id_[i]);
0049     rechits.emplace_back(id_converted,
0050                          h_calibSoA->energy_[i],
0051                          h_calibSoA->time_[i],
0052                          0,
0053                          h_calibSoA->flagBits_[i],
0054                          h_calibSoA->son_[i],
0055                          h_calibSoA->timeError_[i]);
0056   }
0057 }
0058
0059 #include "FWCore/Framework/interface/MakerMacros.h"
0060 DEFINE_FWK_MODULE(HEBRecHitFromSoA);