|
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
|
#include "FWCore/Framework/interface/one/EDAnalyzer.h"
#include "CondCore/DBOutputService/interface/PoolDBOutputService.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESTransientHandle.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CondFormats/GeometryObjects/interface/RecoIdealGeometry.h"
#include "Geometry/Records/interface/DTRecoGeometryRcd.h"
#include "Geometry/Records/interface/MuonGeometryRecord.h"
#include "Geometry/Records/interface/MuonNumberingRecord.h"
#include "Geometry/MuonNumbering/interface/MuonGeometryConstants.h"
#include "Geometry/DTGeometryBuilder/interface/DTGeometryParsFromDD.h"
#include "DetectorDescription/Core/interface/DDCompactView.h"
#include "Geometry/Records/interface/IdealGeometryRecord.h"
#include "DetectorDescription/DDCMS/interface/DDCompactView.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
class DTRecoIdealDBLoader : public edm::one::EDAnalyzer<edm::one::WatchRuns> {
public:
DTRecoIdealDBLoader(edm::ParameterSet const&);
void beginRun(edm::Run const& iEvent, edm::EventSetup const&) override;
void analyze(edm::Event const& iEvent, edm::EventSetup const&) override {}
void endRun(edm::Run const& iEvent, edm::EventSetup const&) override {}
private:
bool fromDD4hep_;
edm::ESGetToken<cms::DDCompactView, IdealGeometryRecord> dd4HepCompactViewToken_;
edm::ESGetToken<DDCompactView, IdealGeometryRecord> compactViewToken_;
edm::ESGetToken<MuonGeometryConstants, IdealGeometryRecord> muonGeomConstantsToken_;
};
DTRecoIdealDBLoader::DTRecoIdealDBLoader(const edm::ParameterSet& iC) {
fromDD4hep_ = iC.getUntrackedParameter<bool>("fromDD4hep", false);
dd4HepCompactViewToken_ = esConsumes<edm::Transition::BeginRun>();
compactViewToken_ = esConsumes<edm::Transition::BeginRun>();
muonGeomConstantsToken_ = esConsumes<edm::Transition::BeginRun>();
}
void DTRecoIdealDBLoader::beginRun(const edm::Run&, edm::EventSetup const& es) {
RecoIdealGeometry rig;
edm::Service<cond::service::PoolDBOutputService> mydbservice;
if (!mydbservice.isAvailable()) {
edm::LogError("DTRecoIdealDBLoader") << "PoolDBOutputService unavailable";
return;
}
auto pMNDC = es.getHandle(muonGeomConstantsToken_);
DTGeometryParsFromDD dtgp;
if (fromDD4hep_) {
auto pDD = es.getTransientHandle(dd4HepCompactViewToken_);
const cms::DDCompactView& cpv = *pDD;
dtgp.build(&cpv, *pMNDC, rig);
} else {
auto pDD = es.getTransientHandle(compactViewToken_);
const DDCompactView& cpv = *pDD;
dtgp.build(&cpv, *pMNDC, rig);
}
if (mydbservice->isNewTagRequest("DTRecoGeometryRcd")) {
mydbservice->createOneIOV(rig, mydbservice->beginOfTime(), "DTRecoGeometryRcd");
} else {
edm::LogError("DTRecoIdealDBLoader") << "DTRecoGeometryRcd Tag is already present.";
}
}
DEFINE_FWK_MODULE(DTRecoIdealDBLoader);
|