|
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
|
#include "EventFilter/L1ScoutingRawToDigi/plugins/ScBMTFRawToDigi.h"
ScBMTFRawToDigi::ScBMTFRawToDigi(const edm::ParameterSet& iConfig) {
using namespace edm;
srcInputTag_ = iConfig.getParameter<InputTag>("srcInputTag");
sourceIdList_ = iConfig.getParameter<std::vector<int>>("sourceIdList");
debug_ = iConfig.getUntrackedParameter<bool>("debug", false);
// initialize orbit buffer for BX 1->3564;
orbitBuffer_ = std::vector<std::vector<l1ScoutingRun3::BMTFStub>>(3565);
for (auto& bxVec : orbitBuffer_) {
bxVec.reserve(32);
}
nStubsOrbit_ = 0;
produces<l1ScoutingRun3::BMTFStubOrbitCollection>("BMTFStub").setBranchAlias("BMTFStubOrbitCollection");
rawToken_ = consumes<SDSRawDataCollection>(srcInputTag_);
}
ScBMTFRawToDigi::~ScBMTFRawToDigi() {}
void ScBMTFRawToDigi::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
using namespace edm;
Handle<SDSRawDataCollection> ScoutingRawDataCollection;
iEvent.getByToken(rawToken_, ScoutingRawDataCollection);
std::unique_ptr<l1ScoutingRun3::BMTFStubOrbitCollection> unpackedStubs(new l1ScoutingRun3::BMTFStubOrbitCollection);
for (const auto& sdsId : sourceIdList_) {
if ((sdsId < SDSNumbering::BmtfMinSDSID) || (sdsId > SDSNumbering::BmtfMaxSDSID))
edm::LogError("ScBMTFRawToDigi::produce")
<< "Provided a source ID outside the expected range: " << sdsId << ", expected range ["
<< SDSNumbering::BmtfMinSDSID << ", " << SDSNumbering::BmtfMaxSDSID;
const FEDRawData& sourceRawData = ScoutingRawDataCollection->FEDData(sdsId);
size_t orbitSize = sourceRawData.size();
if ((sourceRawData.size() == 0) && debug_) {
std::cout << "No raw data for BMTF FED " << sdsId << std::endl;
}
// unpack current orbit and store data into the orbitBufferr
unpackOrbit(sourceRawData.data(), orbitSize, sdsId);
}
// fill orbit collection and clear the Bx buffer vector
unpackedStubs->fillAndClear(orbitBuffer_, nStubsOrbit_);
// store collection in the event
iEvent.put(std::move(unpackedStubs), "BMTFStub");
}
void ScBMTFRawToDigi::unpackOrbit(const unsigned char* buf, size_t len, int sdsId) {
using namespace l1ScoutingRun3;
// reset counters
nStubsOrbit_ = 0;
size_t pos = 0;
while (pos < len) {
assert(pos + 4 <= len);
bmtf::block* bl = (bmtf::block*)(buf + pos);
unsigned bx = bl->bx;
unsigned orbit = (bl->orbit) & 0x7FFFFFFF;
unsigned sCount = (bl->header) & 0xff;
size_t pos_increment = 12 + sCount * 8;
assert(pos_increment <= len);
pos += 12; // header
if (debug_) {
std::cout << " BMTF #" << sdsId << " Orbit " << orbit << ", BX -> " << bx << ", nStubs -> " << sCount
<< std::endl;
}
// Unpack stubs for the current pair (BX, sector)
int32_t phi, phiB, tag, qual, eta, qeta, station, wheel, sector;
// map for station and wheel, to find chambers with 2 stubs
std::vector<std::vector<bool>> stwh_matrix(4, std::vector<bool>(5, false));
for (unsigned int i = 0; i < sCount; i++) {
uint64_t stub_raw = *(uint64_t*)(buf + pos);
pos += 8;
phi = ((stub_raw >> bmtf::shiftsStubs::phi) & bmtf::masksStubs::phi);
phiB = ((stub_raw >> bmtf::shiftsStubs::phiB) & bmtf::masksStubs::phiB);
qual = ((stub_raw >> bmtf::shiftsStubs::qual) & bmtf::masksStubs::qual);
eta = ((stub_raw >> bmtf::shiftsStubs::eta) & bmtf::masksStubs::eta);
qeta = ((stub_raw >> bmtf::shiftsStubs::qeta) & bmtf::masksStubs::qeta);
station = ((stub_raw >> bmtf::shiftsStubs::station) & bmtf::masksStubs::station) + 1;
wheel = ((stub_raw >> bmtf::shiftsStubs::wheel) & bmtf::masksStubs::wheel);
sector = sdsId - SDSNumbering::BmtfMinSDSID;
if (stwh_matrix[station - 1][wheel + 2] == false) {
tag = 1;
} else {
tag = 0;
}
stwh_matrix[station - 1][wheel + 2] = true;
phi = phi >= 2048 ? phi - 4096 : phi;
phiB = phiB >= 512 ? phiB - 1024 : phiB;
wheel = wheel >= 4 ? wheel - 8 : wheel;
BMTFStub stub(phi, phiB, qual, eta, qeta, station, wheel, sector, tag);
orbitBuffer_[bx].push_back(stub);
nStubsOrbit_++;
if (debug_) {
std::cout << "Stub " << i << ", raw: 0x" << std::hex << stub_raw << std::dec << std::endl;
std::cout << "\tPhi: " << phi << std::endl;
std::cout << "\tPhiB: " << phiB << std::endl;
std::cout << "\tQuality: " << qual << std::endl;
std::cout << "\tEta: " << eta << std::endl;
std::cout << "\tQEta: " << qeta << std::endl;
std::cout << "\tStation: " << station << std::endl;
std::cout << "\tWheel: " << wheel << std::endl;
std::cout << "\tSector: " << sector << std::endl;
std::cout << "\tTag: " << tag << std::endl;
}
}
} // end orbit while loop
}
void ScBMTFRawToDigi::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
edm::ParameterSetDescription desc;
desc.setUnknown();
descriptions.addDefault(desc);
}
DEFINE_FWK_MODULE(ScBMTFRawToDigi);
|