|
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
|
#include "ShallowRechitClustersProducer.h"
#include "Geometry/TrackerGeometryBuilder/interface/StripGeomDetUnit.h"
#include "Geometry/CommonTopologies/interface/StripTopology.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "CalibTracker/SiStripCommon/interface/ShallowTools.h"
ShallowRechitClustersProducer::ShallowRechitClustersProducer(const edm::ParameterSet& iConfig)
: Suffix(iConfig.getParameter<std::string>("Suffix")),
Prefix(iConfig.getParameter<std::string>("Prefix")),
geomToken_(esConsumes<TrackerGeometry, TrackerDigiGeometryRecord>()),
clusters_token_(consumes<edmNew::DetSetVector<SiStripCluster>>(iConfig.getParameter<edm::InputTag>("Clusters"))) {
std::vector<edm::InputTag> rec_hits_tags = iConfig.getParameter<std::vector<edm::InputTag>>("InputTags");
for (const auto& itag : rec_hits_tags) {
rec_hits_tokens_.push_back(consumes<SiStripRecHit2DCollection>(itag));
}
produces<std::vector<float>>(Prefix + "strip" + Suffix);
produces<std::vector<float>>(Prefix + "merr" + Suffix);
produces<std::vector<float>>(Prefix + "localx" + Suffix);
produces<std::vector<float>>(Prefix + "localy" + Suffix);
produces<std::vector<float>>(Prefix + "localxerr" + Suffix);
produces<std::vector<float>>(Prefix + "localyerr" + Suffix);
produces<std::vector<float>>(Prefix + "globalx" + Suffix);
produces<std::vector<float>>(Prefix + "globaly" + Suffix);
produces<std::vector<float>>(Prefix + "globalz" + Suffix);
}
void ShallowRechitClustersProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
shallow::CLUSTERMAP clustermap = shallow::make_cluster_map(iEvent, clusters_token_);
int size = clustermap.size();
auto strip = std::make_unique<std::vector<float>>(size, -10000);
auto merr = std::make_unique<std::vector<float>>(size, -10000);
auto localx = std::make_unique<std::vector<float>>(size, -10000);
auto localy = std::make_unique<std::vector<float>>(size, -10000);
auto localxerr = std::make_unique<std::vector<float>>(size, -1);
auto localyerr = std::make_unique<std::vector<float>>(size, -1);
auto globalx = std::make_unique<std::vector<float>>(size, -10000);
auto globaly = std::make_unique<std::vector<float>>(size, -10000);
auto globalz = std::make_unique<std::vector<float>>(size, -10000);
edm::ESHandle<TrackerGeometry> theTrackerGeometry = iSetup.getHandle(geomToken_);
for (auto recHit_token : rec_hits_tokens_) {
edm::Handle<SiStripRecHit2DCollection> recHits;
iEvent.getByToken(recHit_token, recHits);
for (auto const& ds : *recHits) {
for (auto const& hit : ds) {
shallow::CLUSTERMAP::iterator cluster =
clustermap.find(std::make_pair(hit.geographicalId().rawId(), hit.cluster()->firstStrip()));
if (cluster != clustermap.end()) {
const StripGeomDetUnit* theStripDet =
dynamic_cast<const StripGeomDetUnit*>(theTrackerGeometry->idToDet(hit.geographicalId()));
unsigned int i = cluster->second;
strip->at(i) = theStripDet->specificTopology().strip(hit.localPosition());
merr->at(i) = sqrt(
theStripDet->specificTopology().measurementError(hit.localPosition(), hit.localPositionError()).uu());
localx->at(i) = hit.localPosition().x();
localy->at(i) = hit.localPosition().y();
localxerr->at(i) = sqrt(hit.localPositionError().xx());
localyerr->at(i) = sqrt(hit.localPositionError().yy());
globalx->at(i) = theStripDet->toGlobal(hit.localPosition()).x();
globaly->at(i) = theStripDet->toGlobal(hit.localPosition()).y();
globalz->at(i) = theStripDet->toGlobal(hit.localPosition()).z();
} else {
throw cms::Exception("cluster not found");
}
}
}
}
iEvent.put(std::move(strip), Prefix + "strip" + Suffix);
iEvent.put(std::move(merr), Prefix + "merr" + Suffix);
iEvent.put(std::move(localx), Prefix + "localx" + Suffix);
iEvent.put(std::move(localy), Prefix + "localy" + Suffix);
iEvent.put(std::move(localxerr), Prefix + "localxerr" + Suffix);
iEvent.put(std::move(localyerr), Prefix + "localyerr" + Suffix);
iEvent.put(std::move(globalx), Prefix + "globalx" + Suffix);
iEvent.put(std::move(globaly), Prefix + "globaly" + Suffix);
iEvent.put(std::move(globalz), Prefix + "globalz" + Suffix);
}
|