|
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
|
#include <string>
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/one/EDAnalyzer.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "Utilities/General/interface/ClassName.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/Utilities/interface/ESGetToken.h"
#include "CondCore/DBOutputService/interface/PoolDBOutputService.h"
#include "CondFormats/Alignment/interface/Alignments.h"
#include "CondFormats/Alignment/interface/AlignTransform.h"
#include "CondFormats/AlignmentRecord/interface/EBAlignmentRcd.h"
#include "CondFormats/AlignmentRecord/interface/EEAlignmentRcd.h"
#include "CondFormats/AlignmentRecord/interface/ESAlignmentRcd.h"
class CaloAlignmentRcdWrite : public edm::one::EDAnalyzer<> {
public:
explicit CaloAlignmentRcdWrite(const edm::ParameterSet& /*iConfig*/)
: ebToken_{esConsumes<Alignments, EBAlignmentRcd>(edm::ESInputTag{})},
eeToken_{esConsumes<Alignments, EEAlignmentRcd>(edm::ESInputTag{})},
esToken_{esConsumes<Alignments, ESAlignmentRcd>(edm::ESInputTag{})},
nEventCalls_(0) {}
~CaloAlignmentRcdWrite() override {}
template <typename T>
void writeAlignments(const edm::EventSetup& evtSetup, edm::ESGetToken<Alignments, T>& token);
void beginJob() override {}
void analyze(edm::Event const& iEvent, edm::EventSetup const&) override;
void endJob() override {}
private:
edm::ESGetToken<Alignments, EBAlignmentRcd> ebToken_;
edm::ESGetToken<Alignments, EEAlignmentRcd> eeToken_;
edm::ESGetToken<Alignments, ESAlignmentRcd> esToken_;
unsigned int nEventCalls_;
};
template <typename T>
void CaloAlignmentRcdWrite::writeAlignments(const edm::EventSetup& evtSetup, edm::ESGetToken<Alignments, T>& token) {
const auto& alignmentsES = evtSetup.getData(token);
std::string recordName = Demangle(typeid(T).name())();
edm::LogInfo("CaloAlignmentRcdWrite") << "Uploading alignments to the database: " << recordName;
edm::Service<cond::service::PoolDBOutputService> poolDbService;
if (!poolDbService.isAvailable())
throw cms::Exception("NotAvailable") << "PoolDBOutputService not available";
const Alignments alignments(alignmentsES);
poolDbService->writeOneIOV<Alignments>(alignments, poolDbService->currentTime(), recordName);
}
void CaloAlignmentRcdWrite::analyze(const edm::Event& /*evt*/, const edm::EventSetup& evtSetup) {
if (nEventCalls_ > 0) {
edm::LogInfo("CaloAlignmentRcdWrite") << "Writing to DB to be done only once, "
<< "set 'untracked PSet maxEvents = {untracked int32 input = 1}'."
<< "(Your writing should be fine.)";
return;
}
writeAlignments<EBAlignmentRcd>(evtSetup, ebToken_);
writeAlignments<EEAlignmentRcd>(evtSetup, eeToken_);
writeAlignments<ESAlignmentRcd>(evtSetup, esToken_);
edm::LogInfo("CaloAlignmentRcdWrite") << "done!";
nEventCalls_++;
}
DEFINE_FWK_MODULE(CaloAlignmentRcdWrite);
|