|
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
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
|
#ifndef Fireworks_Core_FWHeatmapProxyBuilderTemplate_h
#define Fireworks_Core_FWHeatmapProxyBuilderTemplate_h
// -*- C++ -*-
//
// Package: Core
// Class : FWHeatmapProxyBuilderTemplate
//
/**\class FWHeatmapProxyBuilderTemplate FWHeatmapProxyBuilderTemplate.h Fireworks/Calo/interface/FWHeatmapProxyBuilderTemplate.h
Description: <one line class summary>
Usage:
<usage>
*/
//
// Original Author: Alex Mourtziapis
// Created: Wed Jan 23 14:50:00 EST 2019
//
// system include files
#include <cmath>
// user include files
#include "Fireworks/Core/interface/FWSimpleProxyBuilder.h"
#include "Fireworks/Core/interface/FWProxyBuilderConfiguration.h"
#include "Fireworks/Core/interface/FWEventItem.h"
#include "FWCore/Framework/interface/Event.h"
#include "DataFormats/HGCRecHit/interface/HGCRecHitCollections.h"
// forward declarations
template <typename T>
class FWHeatmapProxyBuilderTemplate : public FWSimpleProxyBuilder {
public:
FWHeatmapProxyBuilderTemplate() : FWSimpleProxyBuilder(typeid(T)) {}
//virtual ~FWHeatmapProxyBuilderTemplate();
// ---------- const member functions ---------------------
// ---------- static member functions --------------------
// ---------- member functions ---------------------------
protected:
std::unordered_map<DetId, const HGCRecHit*>* hitmap = new std::unordered_map<DetId, const HGCRecHit*>;
static constexpr uint8_t gradient_steps = 9;
static constexpr uint8_t gradient[3][gradient_steps] = {{static_cast<uint8_t>(0.2082 * 255),
static_cast<uint8_t>(0.0592 * 255),
static_cast<uint8_t>(0.0780 * 255),
static_cast<uint8_t>(0.0232 * 255),
static_cast<uint8_t>(0.1802 * 255),
static_cast<uint8_t>(0.5301 * 255),
static_cast<uint8_t>(0.8186 * 255),
static_cast<uint8_t>(0.9956 * 255),
static_cast<uint8_t>(0.9764 * 255)},
{static_cast<uint8_t>(0.1664 * 255),
static_cast<uint8_t>(0.3599 * 255),
static_cast<uint8_t>(0.5041 * 255),
static_cast<uint8_t>(0.6419 * 255),
static_cast<uint8_t>(0.7178 * 255),
static_cast<uint8_t>(0.7492 * 255),
static_cast<uint8_t>(0.7328 * 255),
static_cast<uint8_t>(0.7862 * 255),
static_cast<uint8_t>(0.9832 * 255)},
{static_cast<uint8_t>(0.5293 * 255),
static_cast<uint8_t>(0.8684 * 255),
static_cast<uint8_t>(0.8385 * 255),
static_cast<uint8_t>(0.7914 * 255),
static_cast<uint8_t>(0.6425 * 255),
static_cast<uint8_t>(0.4662 * 255),
static_cast<uint8_t>(0.3499 * 255),
static_cast<uint8_t>(0.1968 * 255),
static_cast<uint8_t>(0.0539 * 255)}};
const T& modelData(int index) { return *reinterpret_cast<const T*>(m_helper.offsetObject(item()->modelData(index))); }
void setItem(const FWEventItem* iItem) override {
FWProxyBuilderBase::setItem(iItem);
if (iItem) {
iItem->getConfig()->keepEntries(true);
iItem->getConfig()->assertParam("Layer", 0L, 0L, 52L);
iItem->getConfig()->assertParam("EnergyCutOff", 0.5, 0.2, 5.0);
iItem->getConfig()->assertParam("Heatmap", true);
iItem->getConfig()->assertParam("Z+", true);
iItem->getConfig()->assertParam("Z-", true);
}
}
void build(const FWEventItem* iItem, TEveElementList* product, const FWViewContext* vc) override {
if (item()->getConfig()->template value<bool>("Heatmap")) {
hitmap->clear();
edm::Handle<HGCRecHitCollection> recHitHandleEE;
edm::Handle<HGCRecHitCollection> recHitHandleFH;
edm::Handle<HGCRecHitCollection> recHitHandleBH;
const edm::EventBase* event = iItem->getEvent();
event->getByLabel(edm::InputTag("HGCalRecHit", "HGCEERecHits"), recHitHandleEE);
event->getByLabel(edm::InputTag("HGCalRecHit", "HGCHEFRecHits"), recHitHandleFH);
event->getByLabel(edm::InputTag("HGCalRecHit", "HGCHEBRecHits"), recHitHandleBH);
if (recHitHandleEE.isValid()) {
const auto& rechitsEE = *recHitHandleEE;
for (unsigned int i = 0; i < rechitsEE.size(); ++i) {
(*hitmap)[rechitsEE[i].detid().rawId()] = &rechitsEE[i];
}
}
if (recHitHandleFH.isValid()) {
const auto& rechitsFH = *recHitHandleFH;
for (unsigned int i = 0; i < rechitsFH.size(); ++i) {
(*hitmap)[rechitsFH[i].detid().rawId()] = &rechitsFH[i];
}
}
if (recHitHandleBH.isValid()) {
const auto& rechitsBH = *recHitHandleBH;
for (unsigned int i = 0; i < rechitsBH.size(); ++i) {
(*hitmap)[rechitsBH[i].detid().rawId()] = &rechitsBH[i];
}
}
}
FWSimpleProxyBuilder::build(iItem, product, vc);
}
using FWSimpleProxyBuilder::build;
void build(const void* iData, unsigned int iIndex, TEveElement& oItemHolder, const FWViewContext* context) override {
if (nullptr != iData) {
build(*reinterpret_cast<const T*>(iData), iIndex, oItemHolder, context);
}
}
using FWSimpleProxyBuilder::buildViewType;
void buildViewType(const void* iData,
unsigned int iIndex,
TEveElement& oItemHolder,
FWViewType::EType viewType,
const FWViewContext* context) override {
if (nullptr != iData) {
buildViewType(*reinterpret_cast<const T*>(iData), iIndex, oItemHolder, viewType, context);
}
}
/**iIndex is the index where iData is found in the container from which it came
iItemHolder is the object to which you add your own objects which inherit from TEveElement
*/
virtual void build(const T& iData, unsigned int iIndex, TEveElement& oItemHolder, const FWViewContext*) {
throw std::runtime_error(
"virtual build(const T&, unsigned int, TEveElement&, const FWViewContext*) not implemented by inherited "
"class.");
}
virtual void buildViewType(
const T& iData, unsigned int iIndex, TEveElement& oItemHolder, FWViewType::EType viewType, const FWViewContext*) {
throw std::runtime_error(
"virtual buildViewType(const T&, unsigned int, TEveElement&, FWViewType::EType, const FWViewContext*) not "
"implemented by inherited class");
};
public:
FWHeatmapProxyBuilderTemplate(const FWHeatmapProxyBuilderTemplate&) = delete; // stop default
const FWHeatmapProxyBuilderTemplate& operator=(const FWHeatmapProxyBuilderTemplate&) = delete; // stop default
private:
// ---------- member data --------------------------------
};
#endif
|