|
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
|
#include <memory>
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/ESWatcher.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/StreamID.h"
#include "DataFormats/FEDRawData/interface/FEDRawDataCollection.h"
#include "DataFormats/HGCalDigi/interface/HGCalElectronicsId.h"
#include "DataFormats/HGCalDigi/interface/HGCalDigiHost.h"
#include "DataFormats/HGCalDigi/interface/HGCalECONDPacketInfoHost.h"
#include "DataFormats/HGCalDigi/interface/HGCalFEDPacketInfoHost.h"
#include "DataFormats/HGCalDigi/interface/HGCalRawDataDefinitions.h"
#include "CondFormats/DataRecord/interface/HGCalElectronicsMappingRcd.h"
#include "CondFormats/HGCalObjects/interface/HGCalMappingModuleIndexer.h"
//#include "CondFormats/HGCalObjects/interface/HGCalMappingCellIndexer.h"
#include "CondFormats/DataRecord/interface/HGCalModuleConfigurationRcd.h"
#include "CondFormats/HGCalObjects/interface/HGCalConfiguration.h"
#include "oneapi/tbb/task_arena.h"
#include "oneapi/tbb.h"
#include "EventFilter/HGCalRawToDigi/interface/HGCalUnpacker.h"
class HGCalRawToDigi : public edm::stream::EDProducer<> {
public:
explicit HGCalRawToDigi(const edm::ParameterSet&);
uint16_t callUnpacker(unsigned fedId,
const FEDRawData& fed_data,
const HGCalMappingModuleIndexer& moduleIndexer,
const HGCalConfiguration& config,
hgcaldigi::HGCalDigiHost& digis,
hgcaldigi::HGCalECONDPacketInfoHost& econdPacketInfo);
static void fillDescriptions(edm::ConfigurationDescriptions&);
private:
void produce(edm::Event&, const edm::EventSetup&) override;
void beginRun(edm::Run const&, edm::EventSetup const&) override;
// input tokens
const edm::EDGetTokenT<FEDRawDataCollection> fedRawToken_;
// output tokens
const edm::EDPutTokenT<hgcaldigi::HGCalDigiHost> digisToken_;
const edm::EDPutTokenT<hgcaldigi::HGCalECONDPacketInfoHost> econdPacketInfoToken_;
const edm::EDPutTokenT<hgcaldigi::HGCalFEDPacketInfoHost> fedPacketInfoToken_;
// TODO @hqucms
// what else do we want to output?
// config tokens
// COMMENT @pfs : Cell indexer is for the moment commented as it may be used in the future for Si calib and SiPM-on-tile cells
//edm::ESGetToken<HGCalMappingCellIndexer, HGCalElectronicsMappingRcd> cellIndexToken_;
edm::ESGetToken<HGCalMappingModuleIndexer, HGCalElectronicsMappingRcd> moduleIndexToken_;
edm::ESGetToken<HGCalConfiguration, HGCalModuleConfigurationRcd> configToken_;
// TODO @hqucms
// how to implement this enabled eRx pattern? Can this be taken from the logical mapping?
// HGCalCondSerializableModuleInfo::ERxBitPatternMap erxEnableBits_;
// std::map<uint16_t, uint16_t> fed2slink_;
HGCalUnpacker unpacker_;
const bool doSerial_;
bool headersOnly_;
};
HGCalRawToDigi::HGCalRawToDigi(const edm::ParameterSet& iConfig)
: fedRawToken_(consumes<FEDRawDataCollection>(iConfig.getParameter<edm::InputTag>("src"))),
digisToken_(produces<hgcaldigi::HGCalDigiHost>()),
econdPacketInfoToken_(produces<hgcaldigi::HGCalECONDPacketInfoHost>()),
fedPacketInfoToken_(produces<hgcaldigi::HGCalFEDPacketInfoHost>()),
//cellIndexToken_(esConsumes()),
moduleIndexToken_(esConsumes()),
configToken_(esConsumes()),
doSerial_(iConfig.getParameter<bool>("doSerial")),
headersOnly_(iConfig.getParameter<bool>("headersOnly")) {}
void HGCalRawToDigi::beginRun(edm::Run const& iRun, edm::EventSetup const& iSetup) {
// TODO @hqucms
// init unpacker with proper configs
}
void HGCalRawToDigi::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
// retrieve logical mapping
const auto& moduleIndexer = iSetup.getData(moduleIndexToken_);
//const auto& cellIndexer = iSetup.getData(cellIndexToken_);
const auto& config = iSetup.getData(configToken_);
hgcaldigi::HGCalDigiHost digis(moduleIndexer.getMaxDataSize(), cms::alpakatools::host());
hgcaldigi::HGCalECONDPacketInfoHost econdPacketInfo(moduleIndexer.getMaxModuleSize(), cms::alpakatools::host());
hgcaldigi::HGCalFEDPacketInfoHost fedPacketInfo(moduleIndexer.fedCount(), cms::alpakatools::host());
// retrieve the FED raw data
const auto& raw_data = iEvent.get(fedRawToken_);
for (int32_t i = 0; i < digis.view().metadata().size(); i++) {
digis.view()[i].flags() = hgcal::DIGI_FLAG::NotAvailable;
}
//serial unpacking calls
if (doSerial_) {
for (unsigned fedId = 0; fedId < moduleIndexer.fedCount(); ++fedId) {
const auto& fed_data = raw_data.FEDData(fedId);
fedPacketInfo.view()[fedId].FEDPayload() = fed_data.size();
if (fed_data.size() == 0)
continue;
fedPacketInfo.view()[fedId].FEDUnpackingFlag() =
callUnpacker(fedId, fed_data, moduleIndexer, config, digis, econdPacketInfo);
}
}
//parallel unpacking calls
else {
oneapi::tbb::this_task_arena::isolate([&]() {
oneapi::tbb::parallel_for(0U, moduleIndexer.fedCount(), [&](unsigned fedId) {
const auto& fed_data = raw_data.FEDData(fedId);
fedPacketInfo.view()[fedId].FEDPayload() = fed_data.size();
if (fed_data.size() == 0)
return;
fedPacketInfo.view()[fedId].FEDUnpackingFlag() =
callUnpacker(fedId, fed_data, moduleIndexer, config, digis, econdPacketInfo);
return;
});
});
}
// put information to the event
iEvent.emplace(digisToken_, std::move(digis));
iEvent.emplace(econdPacketInfoToken_, std::move(econdPacketInfo));
iEvent.emplace(fedPacketInfoToken_, std::move(fedPacketInfo));
}
//
uint16_t HGCalRawToDigi::callUnpacker(unsigned fedId,
const FEDRawData& fed_data,
const HGCalMappingModuleIndexer& moduleIndexer,
const HGCalConfiguration& config,
hgcaldigi::HGCalDigiHost& digis,
hgcaldigi::HGCalECONDPacketInfoHost& econdPacketInfo) {
uint16_t status =
unpacker_.parseFEDData(fedId, fed_data, moduleIndexer, config, digis, econdPacketInfo, headersOnly_);
return status;
}
// fill descriptions
void HGCalRawToDigi::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
edm::ParameterSetDescription desc;
desc.add<edm::InputTag>("src", edm::InputTag("rawDataCollector"));
desc.add<std::vector<unsigned int> >("fedIds", {});
desc.add<bool>("doSerial", true)->setComment("do not attempt to paralleize unpacking of different FEDs");
desc.add<bool>("headersOnly", false)->setComment("unpack only headers");
descriptions.add("hgcalDigis", desc);
}
// define this as a plug-in
DEFINE_FWK_MODULE(HGCalRawToDigi);
|