|
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
|
#include "Geometry/EcalAlgo/interface/WriteESAlignments.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CondCore/DBOutputService/interface/PoolDBOutputService.h"
#include "CondCore/CondDB/interface/Serialization.h"
#include "Geometry/EcalAlgo/interface/EcalPreshowerGeometry.h"
typedef WriteESAlignments WEA;
const unsigned int WEA::k_nA = EcalPreshowerGeometry::numberOfAlignments();
WEA::WriteESAlignments(edm::ConsumesCollector&& cc)
: geometryToken_{cc.esConsumes<CaloGeometry, CaloGeometryRecord>(edm::ESInputTag{})},
alignmentToken_{cc.esConsumes<Alignments, ESAlignmentRcd>(edm::ESInputTag{})} {}
void WEA::writeAlignments(const edm::EventSetup& eventSetup,
const WEA::DVec& alphaVec,
const WEA::DVec& betaVec,
const WEA::DVec& gammaVec,
const WEA::DVec& xtranslVec,
const WEA::DVec& ytranslVec,
const WEA::DVec& ztranslVec) {
assert(alphaVec.size() == k_nA);
assert(betaVec.size() == k_nA);
assert(gammaVec.size() == k_nA);
assert(xtranslVec.size() == k_nA);
assert(ytranslVec.size() == k_nA);
assert(ztranslVec.size() == k_nA);
Alignments ali;
AliVec& vali(ali.m_align);
convert(eventSetup, alphaVec, betaVec, gammaVec, xtranslVec, ytranslVec, ztranslVec, vali);
write(ali);
}
void WEA::write(const Alignments& ali) {
edm::LogVerbatim("WriteESAlignments") << "Uploading ES alignments to the database";
edm::Service<cond::service::PoolDBOutputService> poolDbService;
if (!poolDbService.isAvailable())
throw cms::Exception("NotAvailable") << "PoolDBOutputService not available";
poolDbService->writeOneIOV<Alignments>(ali, poolDbService->currentTime(), "ESAlignmentRcd");
}
void WEA::convert(const edm::EventSetup& eS,
const WEA::DVec& a,
const WEA::DVec& b,
const WEA::DVec& g,
const WEA::DVec& x,
const WEA::DVec& y,
const WEA::DVec& z,
WEA::AliVec& va) {
const auto& pG = eS.getData(geometryToken_);
const CaloSubdetectorGeometry* geom(pG.getSubdetectorGeometry(DetId::Ecal, EcalPreshower));
const auto& pA = eS.getData(alignmentToken_);
const AliVec& vaPrev(pA.m_align);
va.reserve(k_nA);
for (unsigned int i(0); i != k_nA; ++i) {
// ordering of i is: left, right, left, right,...
// starting at ES- rear, then ES- front,
// then ES+ front, then ES+ rear
const ESDetId id(EcalPreshowerGeometry::detIdFromLocalAlignmentIndex(i));
double zPlanePrev(geom->getGeometry(id)->getPosition().z());
const double zAlignPrev(vaPrev[i].translation().z());
const Trl q_I(0, 0, zPlanePrev - zAlignPrev);
const Trl& s_p(vaPrev[i].translation());
const Trl t_n(x[i], y[i], z[i]);
const Rot G_p(vaPrev[i].rotation());
const double gamma(g[i]);
const double alpha(a[i]);
const double beta(b[i]);
const Rot L_n( // New rotation in local frame!
Rot(Rot(Rot().rotateZ(-gamma)).rotateX(-alpha)).rotateY(-beta));
const Rot InvL_n(L_n.inverse());
const Rot G_n(InvL_n * G_p);
const Trl s_n(t_n + s_p + q_I - InvL_n * q_I);
edm::LogVerbatim("WriteESAlignments") << "For i = " << i << ", q_I=" << q_I;
edm::LogVerbatim("WriteESAlignments") << "For i = " << i << ", s_p=" << s_p;
edm::LogVerbatim("WriteESAlignments") << "For i = " << i << ", alpha = " << 1000. * alpha << " mr";
edm::LogVerbatim("WriteESAlignments") << "For i = " << i << ", beta = " << 1000. * beta << " mr";
edm::LogVerbatim("WriteESAlignments") << "For i = " << i << ", gamma = " << 1000. * gamma << " mr";
edm::LogVerbatim("WriteESAlignments")
<< " For i = " << i << ", L_n = " << L_n << " Euler angles=" << InvL_n.eulerAngles() << "\n";
edm::LogVerbatim("WriteESAlignments") << "For i = " << i << ", t_n=" << t_n;
edm::LogVerbatim("WriteESAlignments")
<< "For i = " << i << ", G_p=" << G_p << " Euler angles=" << G_p.eulerAngles() << "\n";
edm::LogVerbatim("WriteESAlignments")
<< " For i = " << i << ", InvL_n = " << InvL_n << " Euler angles=" << InvL_n.eulerAngles() << "\n";
edm::LogVerbatim("WriteESAlignments")
<< " For i =" << i << ", G_n = " << G_n << " Euler angles=" << G_n.eulerAngles() << "\n";
edm::LogVerbatim("WriteESAlignments") << " For i =" << i << ", s_n = " << s_n;
edm::LogVerbatim("WriteESAlignments") << "++++++++++++++++++++++++++\n\n";
va.emplace_back(AlignTransform(s_n, G_n, id));
}
}
|