|
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
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
|
#ifndef CalibTracker_SiPixelerrorEstimation_SiPixelErrorEstimation_h
#define CalibTracker_SiPixelerrorEstimation_SiPixelErrorEstimation_h
#include <memory>
// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/one/EDAnalyzer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "DataFormats/TrackingRecHit/interface/TrackingRecHit.h"
#include "DataFormats/TrajectorySeed/interface/TrajectorySeed.h"
#include "TrackingTools/MaterialEffects/interface/PropagatorWithMaterial.h"
#include "TrackingTools/KalmanUpdators/interface/KFUpdator.h"
#include "TrackingTools/KalmanUpdators/interface/Chi2MeasurementEstimator.h"
#include "TrackingTools/TrackFitters/interface/KFTrajectoryFitter.h"
#include "TrackingTools/TrackFitters/interface/KFTrajectorySmoother.h"
#include "TrackingTools/TrajectoryState/interface/TrajectoryStateTransform.h"
#include "TrackingTools/TrajectoryState/interface/TrajectoryStateOnSurface.h"
#include "RecoTracker/TransientTrackingRecHit/interface/TkTransientTrackingRecHitBuilder.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
//--- for SimHit association
#include "SimDataFormats/TrackingHit/interface/PSimHit.h"
#include "SimTracker/TrackerHitAssociation/interface/TrackerHitAssociator.h"
#include "Geometry/CommonTopologies/interface/PixelTopology.h"
#include "Geometry/CommonDetUnit/interface/PixelGeomDetUnit.h"
#include "Geometry/CommonDetUnit/interface/GeomDetType.h"
#include "Geometry/CommonDetUnit/interface/GeomDet.h"
#include "Geometry/CommonDetUnit/interface/GluedGeomDet.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
#include "Geometry/TrackerNumberingBuilder/interface/GeometricDet.h"
#include "Geometry/CommonDetUnit/interface/PixelGeomDetType.h"
#include "DataFormats/TrackerRecHit2D/interface/SiPixelRecHitCollection.h"
#include "SimDataFormats/Track/interface/SimTrackContainer.h"
#include <string>
#include <TROOT.h>
#include <TTree.h>
#include <TFile.h>
#include <TH1F.h>
#include <TProfile.h>
class TTree;
class TFile;
class SiPixelErrorEstimation : public edm::one::EDAnalyzer<> {
public:
explicit SiPixelErrorEstimation(const edm::ParameterSet&);
~SiPixelErrorEstimation() override;
void beginJob() override;
void analyze(const edm::Event&, const edm::EventSetup&) override;
void endJob() override;
void computeAnglesFromDetPosition(const SiPixelCluster& cl, const GeomDetUnit& det, float& alpha, float& beta);
private:
edm::ParameterSet conf_;
edm::EDGetTokenT<std::vector<Trajectory>> tTrajectory;
edm::EDGetTokenT<SiPixelRecHitCollection> tPixRecHitCollection;
edm::EDGetTokenT<edm::SimTrackContainer> tSimTrackContainer;
edm::EDGetTokenT<reco::TrackCollection> tTrackCollection;
edm::ESGetToken<TrackerTopology, TrackerTopologyRcd> trackerTopoToken_;
edm::ESGetToken<TrackerGeometry, TrackerDigiGeometryRecord> trackerGeomToken_;
edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> magneticFieldToken_;
std::string outputFile_;
std::string src_;
bool checkType_; // do we check that the simHit associated with recHit is of the expected particle type ?
int genType_; // the type of particle that the simHit associated with recHits should be
bool include_trk_hits_; // if set to false, take only hits directly from detector modules (don't ntuplize hits from tracks)
// variables that go in the output ttree_track_hits_
float rechitx; // x position of hit
float rechity; // y position of hit
float rechitz; // z position of hit
float rechiterrx; // x position error of hit (error not squared)
float rechiterry; // y position error of hit (error not squared)
float rechitresx; // difference between reconstructed hit x position and 'true' x position
float rechitresy; // difference between reconstructed hit y position and 'true' y position
float rechitpullx; // x residual divideded by error
float rechitpully; // y residual divideded by error
float strip_rechitx; // x position of hit
float strip_rechity; // y position of hit
float strip_rechitz; // z position of hit
float strip_rechiterrx; // x position error of hit (error not squared)
float strip_rechiterry; // y position error of hit (error not squared)
float strip_rechitresx; // difference between reconstructed hit x position and 'true' x position
float strip_rechitresx2;
float strip_rechitresy; // difference between reconstructed hit y position and 'true' y position
float strip_rechitpullx; // x residual divideded by error
float strip_rechitpully; // y residual divideded by error
int strip_is_stereo;
int strip_hit_type; // matched=0, 1D=1 or 2D=2
int detector_type; //IB1=1, IB2=2, OB1=3, OB2=4
float strip_trk_pt;
float strip_cotalpha;
float strip_cotbeta;
float strip_locbx;
float strip_locby;
float strip_locbz;
float strip_charge;
int strip_size;
int strip_edge;
int strip_nsimhit; // number of simhits associated with a rechit
int strip_pidhit; // PID of the particle that produced the simHit associated with the recHit
int strip_simproc; // procces type
int strip_subdet_id; // enum SubDetector { UNKNOWN=0, TIB=3, TID=4, TOB=5, TEC=6 };
int strip_tib_layer;
int strip_tib_module;
int strip_tib_order;
int strip_tib_side;
int strip_tib_is_double_side;
int strip_tib_is_z_plus_side;
int strip_tib_is_z_minus_side;
int strip_tib_layer_number;
int strip_tib_string_number;
int strip_tib_module_number;
int strip_tib_is_internal_string;
int strip_tib_is_external_string;
int strip_tib_is_rphi;
int strip_tib_is_stereo;
int strip_tob_layer;
int strip_tob_module;
//int strip_tob_order ;
int strip_tob_side;
int strip_tob_is_double_side;
int strip_tob_is_z_plus_side;
int strip_tob_is_z_minus_side;
int strip_tob_layer_number;
int strip_tob_rod_number;
int strip_tob_module_number;
int strip_tob_is_rphi;
int strip_tob_is_stereo;
float strip_prob;
int strip_qbin;
int strip_nprm;
int strip_pidhit1;
int strip_simproc1;
int strip_pidhit2;
int strip_simproc2;
int strip_pidhit3;
int strip_simproc3;
int strip_pidhit4;
int strip_simproc4;
int strip_pidhit5;
int strip_simproc5;
int strip_split;
float strip_clst_err_x;
float strip_clst_err_y;
int npix; // number of pixel in the cluster
int nxpix; // size of cluster (number of pixels) along x direction
int nypix; // size of cluster (number of pixels) along y direction
float charge; // total charge in cluster
int edgex; // edgex = 1 if the cluster is at the x edge of the module
int edgey; // edgey = 1 if the cluster is at the y edge of the module
int bigx; // bigx = 1 if the cluster contains at least one big pixel in x
int bigy; // bigy = 1 if the cluster contains at least one big pixel in y
float alpha; // track angle in the xz plane of the module local coordinate system
float beta; // track angle in the yz plane of the module local coordinate system
float trk_alpha; // reconstructed track angle in the xz plane of the module local coordinate system
float trk_beta; // reconstructed track angle in the yz plane of the module local coordinate system
float phi; // polar track angle
float eta; // pseudo-rapidity (function of theta, the azimuthal angle)
int subdetId;
int layer;
int ladder;
int mod;
int side;
int disk;
int blade;
int panel;
int plaq;
int half; // half = 1 if the barrel module is half size and 0 if it is full size (only defined for barrel)
int flipped; // flipped = 1 if the module is flipped and 0 if non-flipped (only defined for barrel)
int nsimhit; // number of simhits associated with a rechit
int pidhit; // PID of the particle that produced the simHit associated with the recHit
int simproc; // procces tye
float simhitx; // true x position of hit
float simhity; // true y position of hit
int evt;
int run;
float hit_probx;
float hit_proby;
float hit_cprob0;
float hit_cprob1;
float hit_cprob2;
int pixel_split;
float pixel_clst_err_x;
float pixel_clst_err_y;
// variables that go in the output ttree_sll_hits_
int all_subdetid;
int all_layer;
int all_ladder;
int all_mod;
int all_side;
int all_disk;
int all_blade;
int all_panel;
int all_plaq;
int all_half;
int all_flipped;
int all_cols;
int all_rows;
float all_rechitx;
float all_rechity;
float all_rechitz;
float all_simhitx;
float all_simhity;
float all_rechiterrx;
float all_rechiterry;
float all_rechitresx;
float all_rechitresy;
float all_rechitpullx;
float all_rechitpully;
int all_npix;
int all_nxpix;
int all_nypix;
int all_edgex;
int all_edgey;
int all_bigx;
int all_bigy;
float all_alpha;
float all_beta;
float all_simphi;
float all_simtheta;
int all_nsimhit;
int all_pidhit;
int all_simproc;
float all_vtxr;
float all_vtxz;
float all_simpx;
float all_simpy;
float all_simpz;
float all_eloss;
int all_trkid;
float all_x1;
float all_x2;
float all_y1;
float all_y2;
float all_z1;
float all_z2;
float all_row1;
float all_row2;
float all_col1;
float all_col2;
float all_gx1;
float all_gx2;
float all_gy1;
float all_gy2;
float all_gz1;
float all_gz2;
float all_simtrketa;
float all_simtrkphi;
float all_clust_row;
float all_clust_col;
float all_clust_x;
float all_clust_y;
float all_clust_q;
int all_clust_maxpixcol;
int all_clust_maxpixrow;
int all_clust_minpixcol;
int all_clust_minpixrow;
int all_clust_geoid;
float all_clust_alpha;
float all_clust_beta;
static const int maxpix = 10000;
float all_pixrow[maxpix];
float all_pixcol[maxpix];
float all_pixadc[maxpix];
// Just added
float all_pixx[maxpix];
float all_pixy[maxpix];
float all_pixgx[maxpix];
float all_pixgy[maxpix];
float all_pixgz[maxpix];
float all_hit_probx;
float all_hit_proby;
float all_hit_cprob0;
float all_hit_cprob1;
float all_hit_cprob2;
int all_pixel_split;
float all_pixel_clst_err_x;
float all_pixel_clst_err_y;
// ----------------------------------
TFile* tfile_;
TTree* ttree_all_hits_;
TTree* ttree_track_hits_;
TTree* ttree_track_hits_strip_;
TrackerHitAssociator::Config trackerHitAssociatorConfig_;
};
#endif
|