|
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
|
#include "FWCore/ParameterSet/interface/FileInPath.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/ESGetToken.h"
#include "HeterogeneousCore/AlpakaCore/interface/alpaka/ESProducer.h"
#include "HeterogeneousCore/AlpakaCore/interface/alpaka/ModuleFactory.h"
#include "HeterogeneousCore/AlpakaInterface/interface/config.h"
#include "HeterogeneousCore/AlpakaInterface/interface/host.h"
#include "HeterogeneousCore/AlpakaInterface/interface/memory.h"
#include "CondFormats/DataRecord/interface/HGCalElectronicsMappingRcd.h"
#include "CondFormats/HGCalObjects/interface/HGCalMappingCellIndexer.h"
#include "CondFormats/HGCalObjects/interface/HGCalMappingParameterHost.h"
#include "CondFormats/HGCalObjects/interface/alpaka/HGCalMappingParameterDevice.h"
#include "DataFormats/HGCalDigi/interface/HGCalElectronicsId.h"
#include "DataFormats/ForwardDetId/interface/HGCSiliconDetId.h"
#include "DataFormats/ForwardDetId/interface/HGCScintillatorDetId.h"
#include "Geometry/HGCalMapping/interface/HGCalMappingTools.h"
#include <string>
#include <iostream>
#include <fstream>
#include <sstream>
namespace ALPAKA_ACCELERATOR_NAMESPACE {
namespace hgcal {
class HGCalMappingCellESProducer : public ESProducer {
public:
//
HGCalMappingCellESProducer(const edm::ParameterSet& iConfig)
: ESProducer(iConfig), filelist_(iConfig.getParameter<std::vector<std::string> >("filelist")) {
auto cc = setWhatProduced(this);
cellIndexTkn_ = cc.consumes(iConfig.getParameter<edm::ESInputTag>("cellindexer"));
}
//
static void fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
edm::ParameterSetDescription desc;
desc.add<std::vector<std::string> >("filelist", std::vector<std::string>({}))
->setComment("list of files with the readout cells of each module");
desc.add<edm::ESInputTag>("cellindexer", edm::ESInputTag(""))->setComment("Dense cell index tool");
descriptions.addWithDefaultLabel(desc);
}
//
std::optional<HGCalMappingCellParamHost> produce(const HGCalElectronicsMappingRcd& iRecord) {
//get cell indexer
const HGCalMappingCellIndexer& cellIndexer = iRecord.get(cellIndexTkn_);
const uint32_t size = cellIndexer.maxDenseIndex(); // channel-level size
HGCalMappingCellParamHost cellParams(size, cms::alpakatools::host());
for (uint32_t i = 0; i < size; i++)
cellParams.view()[i].valid() = false;
//loop over cell types and then over cells
for (const auto& url : filelist_) {
::hgcal::mappingtools::HGCalEntityList pmap;
edm::FileInPath fip(url);
pmap.buildFrom(fip.fullPath());
auto& entities = pmap.getEntries();
for (auto row : entities) {
//identify special cases (Si vs SiPM, calib vs normal)
std::string typecode = pmap.getAttr("Typecode", row);
auto typeidx = cellIndexer.getEnumFromTypecode(typecode);
bool isSiPM = typecode.find("TM") != std::string::npos;
int rocpin = pmap.getIntAttr("ROCpin", row);
int celltype = pmap.getIntAttr("t", row);
int i1(0), i2(0), sensorcell(0);
bool isHD(false), iscalib(false);
uint32_t detid(0);
if (isSiPM) {
i1 = pmap.getIntAttr("iring", row);
i2 = pmap.getIntAttr("iphi", row);
} else {
i1 = pmap.getIntAttr("iu", row);
i2 = pmap.getIntAttr("iv", row);
isHD = {typecode.find("MH") != std::string::npos ? true : false};
sensorcell = pmap.getIntAttr("SiCell", row);
if (celltype == 0) {
iscalib = true;
std::string rocpinstr = pmap.getAttr("ROCpin", row);
rocpin = uint16_t(rocpinstr[rocpinstr.size() - 1]);
}
//detector id is initiated for a random sub-detector with Si wafers
//we only care about the first 10bits where the cell u-v are stored
DetId::Detector det(DetId::Detector::HGCalEE);
detid = 0x3ff & HGCSiliconDetId(det, 0, 0, 0, 0, 0, i1, i2).rawId();
}
//fill cell info in the appopriate dense index
int chip = pmap.getIntAttr("ROC", row);
int half = pmap.getIntAttr("HalfROC", row);
int seq = pmap.getIntAttr("Seq", row);
int idx = cellIndexer.denseIndex(typecode, chip, half, seq);
auto cell = cellParams.view()[idx];
cell.valid() = true;
cell.isHD() = isHD;
cell.iscalib() = iscalib;
cell.isSiPM() = isSiPM;
cell.typeidx() = typeidx;
cell.chip() = chip;
cell.half() = half;
cell.seq() = seq;
cell.rocpin() = rocpin;
cell.sensorcell() = sensorcell;
cell.triglink() = pmap.getIntAttr("TrLink", row);
cell.trigcell() = pmap.getIntAttr("TrCell", row);
cell.i1() = i1;
cell.i2() = i2;
cell.t() = celltype;
cell.trace() = pmap.getFloatAttr("trace", row);
cell.eleid() = HGCalElectronicsId(false, 0, 0, 0, chip * 2 + half, seq).raw();
cell.detid() = detid;
} //end loop over entities
} //end loop over cell types
return cellParams;
} // end of produce()
private:
edm::ESGetToken<HGCalMappingCellIndexer, HGCalElectronicsMappingRcd> cellIndexTkn_;
const std::vector<std::string> filelist_;
};
} // namespace hgcal
} // namespace ALPAKA_ACCELERATOR_NAMESPACE
DEFINE_FWK_EVENTSETUP_ALPAKA_MODULE(hgcal::HGCalMappingCellESProducer);
|