|
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
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
|
// -*- C++ -*-
//
// Package: CondTools/SiStrip
// Class: SiStripGainPayloadCopyAndExclude
//
/*
*\class SiStripGainPayloadCopyAndExclude SiStripGainPayloadCopyAndExclude.cc CondTools/SiStrip/plugins/SiStripGainPayloadCopyAndExclude.cc
Description: This module is meant to copy the content of a SiStrip APV Gain payload (either G1 or G2) from a local sqlite file (that should be feeded to the Event Setup via the SiStripApvGain3Rcd and put in another local sqlite file, excepted for the modules specified in the excludedModules parameter. If the doReverse parameter is true, the opposite action is performed.
Implementation: The implemenation takes advantage of the convenience record SiStripApvGain3Rcd in the EventSetup to be able to hold at the same time two instances of the Strip Gains in the same job.
*/
//
// Original Author: Marco Musich
// Created: Fri, 08 Jun 2018 08:28:01 GMT
//
//
// system include files
#include <memory>
#include <iostream>
// user include files
#include "CLHEP/Random/RandGauss.h"
#include "CalibFormats/SiStripObjects/interface/SiStripGain.h"
#include "CalibTracker/Records/interface/SiStripGainRcd.h"
#include "CommonTools/TrackerMap/interface/TrackerMap.h"
#include "CondCore/DBOutputService/interface/PoolDBOutputService.h"
#include "CondFormats/DataRecord/interface/SiStripApvGainRcd.h"
#include "CondFormats/SiStripObjects/interface/SiStripApvGain.h"
#include "CondFormats/SiStripObjects/interface/SiStripSummary.h"
#include "DataFormats/SiStripDetId/interface/SiStripDetId.h"
#include "DataFormats/SiStripDetId/interface/StripSubdetector.h"
#include "DataFormats/TrackerCommon/interface/TrackerTopology.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/one/EDAnalyzer.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "Geometry/Records/interface/TrackerTopologyRcd.h"
//
// class declaration
//
class SiStripGainPayloadCopyAndExclude : public edm::one::EDAnalyzer<edm::one::SharedResources> {
public:
explicit SiStripGainPayloadCopyAndExclude(const edm::ParameterSet&);
~SiStripGainPayloadCopyAndExclude() override = default;
static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
private:
void analyze(const edm::Event&, const edm::EventSetup&) override;
std::unique_ptr<SiStripApvGain> getNewObject(const std::map<std::pair<uint32_t, int>, float>& theMap);
// ----------member data ---------------------------
const edm::ESGetToken<SiStripGain, SiStripGainRcd> m_gainToken;
const edm::ESGetToken<SiStripApvGain, SiStripApvGain3Rcd> m_gain3Token;
std::vector<unsigned int> m_excludedMods;
const std::string m_Record;
const uint32_t m_gainType;
const bool m_reverseSelect;
};
//
// constructors and destructor
//
SiStripGainPayloadCopyAndExclude::SiStripGainPayloadCopyAndExclude(const edm::ParameterSet& iConfig)
: m_gainToken{esConsumes()},
m_gain3Token{esConsumes()},
m_excludedMods{iConfig.getUntrackedParameter<std::vector<unsigned int>>("excludedModules")},
m_Record{iConfig.getUntrackedParameter<std::string>("record", "SiStripApvGainRcd")},
m_gainType{iConfig.getUntrackedParameter<uint32_t>("gainType", 1)},
m_reverseSelect{iConfig.getUntrackedParameter<bool>("reverseSelection", false)} {
usesResource(cond::service::PoolDBOutputService::kSharedResource);
//now do what ever initialization is needed
sort(m_excludedMods.begin(), m_excludedMods.end());
edm::LogInfo("ExcludedModules") << "Selected module list";
for (std::vector<unsigned int>::const_iterator mod = m_excludedMods.begin(); mod != m_excludedMods.end(); mod++) {
edm::LogVerbatim("ExcludedModules") << *mod;
}
}
//
// member functions
//
// ------------ method called for each event ------------
void SiStripGainPayloadCopyAndExclude::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup) {
using namespace edm;
// gain to be validated
edm::ESHandle<SiStripApvGain> gNew = iSetup.getHandle(m_gain3Token);
edm::ESHandle<SiStripGain> gOld = iSetup.getHandle(m_gainToken);
std::map<std::pair<uint32_t, int>, float> theMap, oldPayloadMap;
std::vector<uint32_t> detid;
gNew->getDetIds(detid);
for (const auto& d : detid) {
SiStripApvGain::Range range_new = gNew->getRange(d);
SiStripApvGain::Range range_old = gOld->getRange(d, m_gainType);
float nAPV = 0;
for (int it = 0; it < range_new.second - range_new.first; it++) {
nAPV += 1;
float Gain = gNew->getApvGain(it, range_new);
float patchGain = gOld->getApvGain(it, range_old);
std::pair<uint32_t, int> index = std::make_pair(d, nAPV);
oldPayloadMap[index] = Gain;
bool found(false);
for (const auto& mod : m_excludedMods) {
if (d == mod) {
edm::LogInfo("ModuleFound") << " module " << mod << " found! Excluded... " << std::endl;
found = true;
break;
}
}
if (m_reverseSelect)
found = (!found);
if (!found) {
theMap[index] = Gain;
} else {
theMap[index] = patchGain;
}
} // loop over APVs
} // loop over DetIds
std::unique_ptr<SiStripApvGain> theAPVGains = this->getNewObject(theMap);
// write out the APVGains record
edm::Service<cond::service::PoolDBOutputService> poolDbService;
if (poolDbService.isAvailable())
poolDbService->writeOneIOV(theAPVGains.get(), poolDbService->currentTime(), m_Record);
else
throw std::runtime_error("PoolDBService required.");
}
//********************************************************************************//
std::unique_ptr<SiStripApvGain> SiStripGainPayloadCopyAndExclude::getNewObject(
const std::map<std::pair<uint32_t, int>, float>& theMap) {
std::unique_ptr<SiStripApvGain> obj = std::make_unique<SiStripApvGain>();
std::vector<float> theSiStripVector;
uint32_t PreviousDetId = 0;
for (const auto& element : theMap) {
uint32_t DetId = element.first.first;
if (DetId != PreviousDetId) {
if (!theSiStripVector.empty()) {
SiStripApvGain::Range range(theSiStripVector.begin(), theSiStripVector.end());
if (!obj->put(PreviousDetId, range))
printf("Bug to put detId = %i\n", PreviousDetId);
}
theSiStripVector.clear();
PreviousDetId = DetId;
}
theSiStripVector.push_back(element.second);
edm::LogInfo("SiStripGainPayloadCopyAndExclude")
<< " DetId: " << DetId << " APV: " << element.first.second << " Gain: " << element.second << std::endl;
}
if (!theSiStripVector.empty()) {
SiStripApvGain::Range range(theSiStripVector.begin(), theSiStripVector.end());
if (!obj->put(PreviousDetId, range))
printf("Bug to put detId = %i\n", PreviousDetId);
}
return obj;
}
// ------------ method fills 'descriptions' with the allowed parameters for the module ------------
void SiStripGainPayloadCopyAndExclude::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
edm::ParameterSetDescription desc;
desc.addUntracked<std::vector<unsigned int>>("excludedModules", {});
desc.addUntracked<std::string>("record", "SiStripApvGainRcd");
desc.addUntracked<uint32_t>("gainType", 1);
desc.addUntracked<bool>("reverseSelection", false);
descriptions.addWithDefaultLabel(desc);
}
//define this as a plug-in
DEFINE_FWK_MODULE(SiStripGainPayloadCopyAndExclude);
|