|
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
|
#ifndef DQM_EcalMonitorTasks_GpuTask_H
#define DQM_EcalMonitorTasks_GpuTask_H
#include "DQWorkerTask.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"
#include "DataFormats/EcalDigi/interface/EcalDigiCollections.h"
namespace ecaldqm {
class GpuTask : public DQWorkerTask {
public:
GpuTask();
~GpuTask() override = default;
void addDependencies(DependencySet&) override;
bool filterRunType(short const*) override;
void beginEvent(edm::Event const&, edm::EventSetup const&, bool const&, bool&) override;
bool analyze(void const*, Collections) override;
template <typename DigiCollection>
void runOnCpuDigis(DigiCollection const&, Collections);
template <typename DigiCollection>
void runOnGpuDigis(DigiCollection const&, Collections);
void runOnCpuUncalibRecHits(EcalUncalibratedRecHitCollection const&, Collections);
void runOnGpuUncalibRecHits(EcalUncalibratedRecHitCollection const&, Collections);
void runOnCpuRecHits(EcalRecHitCollection const&, Collections);
void runOnGpuRecHits(EcalRecHitCollection const&, Collections);
private:
void setParams(edm::ParameterSet const&) override;
bool runGpuTask_;
bool enableDigi_;
bool enableUncalib_;
bool enableRecHit_;
bool digi1D_;
bool digi2D_;
bool uncalib1D_;
bool uncalib2D_;
bool rechit1D_;
bool rechit2D_;
std::vector<int> uncalibOOTAmps_;
EBDigiCollection const* EBCpuDigis_;
EEDigiCollection const* EECpuDigis_;
EcalUncalibratedRecHitCollection const* EBCpuUncalibRecHits_;
EcalUncalibratedRecHitCollection const* EECpuUncalibRecHits_;
EcalRecHitCollection const* EBCpuRecHits_;
EcalRecHitCollection const* EECpuRecHits_;
};
inline bool GpuTask::analyze(void const* collection_data, Collections collection) {
switch (collection) {
case kEBCpuDigi:
if (collection_data && runGpuTask_ && enableDigi_)
runOnCpuDigis(*static_cast<EBDigiCollection const*>(collection_data), collection);
return enableDigi_;
break;
case kEECpuDigi:
if (collection_data && runGpuTask_ && enableDigi_)
runOnCpuDigis(*static_cast<EEDigiCollection const*>(collection_data), collection);
return enableDigi_;
break;
case kEBGpuDigi:
if (collection_data && runGpuTask_ && enableDigi_)
runOnGpuDigis(*static_cast<EBDigiCollection const*>(collection_data), collection);
return enableDigi_;
break;
case kEEGpuDigi:
if (collection_data && runGpuTask_ && enableDigi_)
runOnGpuDigis(*static_cast<EEDigiCollection const*>(collection_data), collection);
return enableDigi_;
break;
case kEBCpuUncalibRecHit:
case kEECpuUncalibRecHit:
if (collection_data && runGpuTask_ && enableUncalib_)
runOnCpuUncalibRecHits(*static_cast<EcalUncalibratedRecHitCollection const*>(collection_data), collection);
return enableUncalib_;
break;
case kEBGpuUncalibRecHit:
case kEEGpuUncalibRecHit:
if (collection_data && runGpuTask_ && enableUncalib_)
runOnGpuUncalibRecHits(*static_cast<EcalUncalibratedRecHitCollection const*>(collection_data), collection);
return enableUncalib_;
break;
case kEBCpuRecHit:
case kEECpuRecHit:
if (collection_data && runGpuTask_ && enableRecHit_)
runOnCpuRecHits(*static_cast<EcalRecHitCollection const*>(collection_data), collection);
return enableRecHit_;
break;
case kEBGpuRecHit:
case kEEGpuRecHit:
if (collection_data && runGpuTask_ && enableRecHit_)
runOnGpuRecHits(*static_cast<EcalRecHitCollection const*>(collection_data), collection);
return enableRecHit_;
break;
default:
break;
}
return false;
}
} // namespace ecaldqm
#endif
|