|
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
|
/*
* See header file for a description of this class.
*
* \author A. Vilela Pereira
*/
#include "DTVDriftSegmentCalibration.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "Geometry/DTGeometry/interface/DTGeometry.h"
#include "Geometry/Records/interface/MuonGeometryRecord.h"
#include "CalibMuon/DTCalibration/interface/DTCalibDBUtils.h"
#include "CalibMuon/DTCalibration/interface/DTSegmentSelector.h"
#include <string>
#include <sstream>
#include "TFile.h"
#include "TH1F.h"
#include "TH2F.h"
using namespace std;
using namespace edm;
DTVDriftSegmentCalibration::DTVDriftSegmentCalibration(const ParameterSet& pset)
: theRecHits4DToken_(consumes<DTRecSegment4DCollection>(pset.getParameter<InputTag>("recHits4DLabel"))),
//writeVDriftDB_(pset.getUntrackedParameter<bool>("writeVDriftDB", false)),
theCalibChamber_(pset.getUntrackedParameter<string>("calibChamber", "All")),
dtGeomToken_(esConsumes()) {
LogVerbatim("Calibration") << "[DTVDriftSegmentCalibration] Constructor called!";
edm::ConsumesCollector collector(consumesCollector());
select_ = new DTSegmentSelector(pset, collector);
// the root file which will contain the histos
string rootFileName = pset.getUntrackedParameter<string>("rootFileName", "DTVDriftHistos.root");
rootFile_ = new TFile(rootFileName.c_str(), "RECREATE");
rootFile_->cd();
}
void DTVDriftSegmentCalibration::beginJob() { TH1::SetDefaultSumw2(true); }
void DTVDriftSegmentCalibration::beginRun(const edm::Run& run, const edm::EventSetup& setup) {}
DTVDriftSegmentCalibration::~DTVDriftSegmentCalibration() {
rootFile_->Close();
LogVerbatim("Calibration") << "[DTVDriftSegmentCalibration] Destructor called!";
}
void DTVDriftSegmentCalibration::analyze(const Event& event, const EventSetup& eventSetup) {
rootFile_->cd();
// Get the DT Geometry
ESHandle<DTGeometry> dtGeom;
dtGeom = eventSetup.getHandle(dtGeomToken_);
// Get the rechit collection from the event
const Handle<DTRecSegment4DCollection>& all4DSegments = event.getHandle(theRecHits4DToken_);
DTChamberId chosenChamberId;
if (theCalibChamber_ != "All") {
stringstream linestr;
int selWheel, selStation, selSector;
linestr << theCalibChamber_;
linestr >> selWheel >> selStation >> selSector;
chosenChamberId = DTChamberId(selWheel, selStation, selSector);
LogVerbatim("Calibration") << " Chosen chamber: " << chosenChamberId << endl;
}
// Loop over segments by chamber
DTRecSegment4DCollection::id_iterator chamberIdIt;
for (chamberIdIt = all4DSegments->id_begin(); chamberIdIt != all4DSegments->id_end(); ++chamberIdIt) {
// Calibrate just the chosen chamber/s
if ((theCalibChamber_ != "All") && ((*chamberIdIt) != chosenChamberId))
continue;
// Book histos
if (theVDriftHistoMapTH1F_.find(*chamberIdIt) == theVDriftHistoMapTH1F_.end()) {
LogTrace("Calibration") << " Booking histos for Chamber: " << *chamberIdIt;
bookHistos(*chamberIdIt);
}
// Get the chamber from the setup
const DTChamber* chamber = dtGeom->chamber(*chamberIdIt);
// Get the range for the corresponding ChamberId
DTRecSegment4DCollection::range range = all4DSegments->get((*chamberIdIt));
// Loop over the rechits of this DetUnit
for (DTRecSegment4DCollection::const_iterator segment = range.first; segment != range.second; ++segment) {
LogTrace("Calibration") << "Segment local pos (in chamber RF): " << (*segment).localPosition()
<< "\nSegment global pos: " << chamber->toGlobal((*segment).localPosition());
if (!(*select_)(*segment, event, eventSetup))
continue;
// Fill v-drift values
if ((*segment).hasPhi()) {
//if( segment->phiSegment()->ist0Valid() ){
double segmentVDrift = segment->phiSegment()->vDrift();
if (segmentVDrift != 0.00) {
(theVDriftHistoMapTH1F_[*chamberIdIt])[0]->Fill(segmentVDrift);
(theVDriftHistoMapTH2F_[*chamberIdIt])[0]->Fill(segment->localPosition().x(), segmentVDrift);
(theVDriftHistoMapTH2F_[*chamberIdIt])[1]->Fill(segment->localPosition().y(), segmentVDrift);
}
}
// Probably not meaningful
if ((*segment).hasZed()) {
//if( segment->zSegment()->ist0Valid() ){
double segmentVDrift = segment->zSegment()->vDrift();
if (segmentVDrift != 0.00) {
(theVDriftHistoMapTH1F_[*chamberIdIt])[1]->Fill(segmentVDrift);
}
}
} // DTRecSegment4DCollection::const_iterator segment
} // DTRecSegment4DCollection::id_iterator chamberIdIt
} // DTVDriftSegmentCalibration::analyze
void DTVDriftSegmentCalibration::endJob() {
rootFile_->cd();
LogVerbatim("Calibration") << "[DTVDriftSegmentCalibration] Writing histos to file!" << endl;
for (ChamberHistosMapTH1F::const_iterator itChHistos = theVDriftHistoMapTH1F_.begin();
itChHistos != theVDriftHistoMapTH1F_.end();
++itChHistos) {
vector<TH1F*>::const_iterator itHistTH1F = (*itChHistos).second.begin();
vector<TH1F*>::const_iterator itHistTH1F_end = (*itChHistos).second.end();
for (; itHistTH1F != itHistTH1F_end; ++itHistTH1F)
(*itHistTH1F)->Write();
vector<TH2F*>::const_iterator itHistTH2F = theVDriftHistoMapTH2F_[(*itChHistos).first].begin();
vector<TH2F*>::const_iterator itHistTH2F_end = theVDriftHistoMapTH2F_[(*itChHistos).first].end();
for (; itHistTH2F != itHistTH2F_end; ++itHistTH2F)
(*itHistTH2F)->Write();
}
/*if(writeVDriftDB_){
// ...
}*/
}
// Book a set of histograms for a given Chamber
void DTVDriftSegmentCalibration::bookHistos(DTChamberId chId) {
// Compose the chamber name
std::string wheel = std::to_string(chId.wheel());
std::string station = std::to_string(chId.station());
std::string sector = std::to_string(chId.sector());
string chHistoName = "_W" + wheel + "_St" + station + "_Sec" + sector;
vector<TH1F*> histosTH1F;
histosTH1F.push_back(
new TH1F(("hRPhiVDriftCorr" + chHistoName).c_str(), "v-drift corr. from Phi segments", 200, -0.4, 0.4));
if (chId.station() != 4)
histosTH1F.push_back(
new TH1F(("hRZVDriftCorr" + chHistoName).c_str(), "v-drift corr. from Z segments", 200, -0.4, 0.4));
vector<TH2F*> histosTH2F;
histosTH2F.push_back(new TH2F(("hRPhiVDriftCorrVsSegmPosX" + chHistoName).c_str(),
"v-drift corr. vs. segment x position",
250,
-125.,
125.,
200,
-0.4,
0.4));
histosTH2F.push_back(new TH2F(("hRPhiVDriftCorrVsSegmPosY" + chHistoName).c_str(),
"v-drift corr. vs. segment y position",
250,
-125.,
125.,
200,
-0.4,
0.4));
//if(chId.station() != 4) ...
theVDriftHistoMapTH1F_[chId] = histosTH1F;
theVDriftHistoMapTH2F_[chId] = histosTH2F;
}
|