|
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
|
// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/global/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "DataFormats/FEDRawData/interface/FEDRawDataCollection.h"
#include "DataFormats/FEDRawData/interface/FEDRawData.h"
#include "DataFormats/FEDRawData/interface/FEDNumbering.h"
#include "EventFilter/HcalRawToDigi/interface/HcalDCCHeader.h"
#include "EventFilter/HcalRawToDigi/interface/HcalFEDList.h"
class HcalCalibFEDSelector : public edm::global::EDProducer<> {
public:
HcalCalibFEDSelector(const edm::ParameterSet&);
private:
void produce(edm::StreamID, edm::Event&, const edm::EventSetup&) const override;
// ----------member data ---------------------------
edm::EDGetTokenT<FEDRawDataCollection> tok_fed_;
std::vector<int> extraFEDs_;
};
HcalCalibFEDSelector::HcalCalibFEDSelector(const edm::ParameterSet& iConfig) {
tok_fed_ = consumes<FEDRawDataCollection>(iConfig.getParameter<edm::InputTag>("rawInputLabel"));
extraFEDs_ = iConfig.getParameter<std::vector<int> >("extraFEDsToKeep");
produces<FEDRawDataCollection>();
}
void HcalCalibFEDSelector::produce(edm::StreamID, edm::Event& iEvent, const edm::EventSetup& iSetup) const {
auto producedData = std::make_unique<FEDRawDataCollection>();
edm::Handle<FEDRawDataCollection> rawIn;
iEvent.getByToken(tok_fed_, rawIn);
std::vector<int> selFEDs;
//--- Get the list of FEDs to be kept ---//
int calibType = -1;
for (int i = FEDNumbering::MINHCALFEDID; i <= FEDNumbering::MAXHCALFEDID; i++) {
const FEDRawData& fedData = rawIn->FEDData(i);
if (fedData.size() < 24)
continue; // FED is empty
int value = ((const HcalDCCHeader*)(fedData.data()))->getCalibType();
if (calibType < 0) {
calibType = value;
} else {
if (calibType != value)
edm::LogWarning("HcalCalibFEDSelector") << "Conflicting calibration types found: " << calibType << " vs. "
<< value << ". Staying with " << calibType;
}
}
HcalFEDList calibFeds(calibType);
selFEDs = calibFeds.getListOfFEDs();
for (unsigned int i = 0; i < extraFEDs_.size(); i++) {
bool duplicate = false;
for (unsigned int j = 0; j < selFEDs.size(); j++) {
if (extraFEDs_.at(i) == selFEDs.at(j)) {
duplicate = true;
break;
}
}
if (!duplicate)
selFEDs.push_back(extraFEDs_.at(i));
}
// Copying:
const FEDRawDataCollection* rdc = rawIn.product();
for (int j = 0; j <= FEDNumbering::lastFEDId(); ++j) {
bool rightFED = false;
for (uint32_t k = 0; k < selFEDs.size(); k++) {
if (j == selFEDs[k]) {
rightFED = true;
}
}
if (!rightFED)
continue;
const FEDRawData& fedData = rdc->FEDData(j);
size_t size = fedData.size();
if (size > 0) {
// this fed has data -- lets copy it
FEDRawData& fedDataProd = producedData->FEDData(j);
if (fedDataProd.size() != 0) {
continue;
}
fedDataProd.resize(size);
unsigned char* dataProd = fedDataProd.data();
const unsigned char* data = fedData.data();
// memcpy is at-least-as-fast as assignment and can be much faster
memcpy(dataProd, data, size);
}
}
iEvent.put(std::move(producedData));
}
DEFINE_FWK_MODULE(HcalCalibFEDSelector);
|