forked from cms-sw/cmssw
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathnano_mu_local_reco_cff.py
244 lines (209 loc) · 16.4 KB
/
nano_mu_local_reco_cff.py
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
import FWCore.ParameterSet.Config as cms
from DPGAnalysis.MuonTools.dtSegmentFlatTableProducer_cfi import dtSegmentFlatTableProducer
from PhysicsTools.NanoAOD.common_cff import *
from DPGAnalysis.MuonTools.common_cff import *
dtSegmentFlatTableProducer.name = "dtSegment"
dtSegmentFlatTableProducer.src = "dt4DSegments"
dtSegmentFlatTableProducer.doc = "DT segment information"
dtSegmentFlatTableProducer.variables = cms.PSet(
seg4D_hasPhi = Var("hasPhi()", bool, doc = "has segment phi view - bool"),
seg4D_hasZed = Var("hasZed()", bool, doc = "has segment zed view - bool"),
seg4D_posLoc_x = Var("localPosition().x()", float, doc = "position x in local coordinates - cm"),
seg4D_posLoc_y = Var("localPosition().y()", float, doc = "position y in local coordinates - cm"),
seg4D_posLoc_z = Var("localPosition().z()", float, doc = "position z in local coordinates - cm"),
seg4D_dirLoc_x = Var("localDirection().x()", float, doc = "direction x in local coordinates"),
seg4D_dirLoc_y = Var("localDirection().y()", float, doc = "direction y in local coordinates"),
seg4D_dirLoc_z = Var("localDirection().z()", float, doc = "direction z in local coordinates"),
seg2D_phi_t0 = Var(f"? hasPhi() ? phiSegment().t0() : {defaults.FLOAT}", float, doc = "t0 from segments with phi view - ns"),
seg2D_phi_nHits = Var(f"? hasPhi() ? phiSegment().specificRecHits().size() : 0", "int8", doc = "# hits in phi view - [0:8] range"),
seg2D_phi_vDrift = Var(f"? hasPhi() ? phiSegment().vDrift() : {defaults.FLOAT_POS}", float, doc = "v_drift from segments with phi view"),
seg2D_phi_normChi2 = Var(f"? hasPhi() ? (phiSegment().chi2() / phiSegment().degreesOfFreedom()) : {defaults.FLOAT_POS}", float, doc = "chi2/n.d.o.f. from segments with phi view"),
seg2D_z_t0 = Var(f"? hasZed() ? zSegment().t0() : {defaults.FLOAT}", float, doc = "t0 from segments with z view - ns"),
seg2D_z_nHits = Var(f"? hasZed() ? zSegment().specificRecHits().size() : 0", "int8", doc = "# hits in z view - [0:4] range"),
seg2D_z_normChi2 = Var(f"? hasZed() ? (zSegment().chi2() / zSegment().degreesOfFreedom()) : {defaults.FLOAT_POS}", float, doc = "chi2/n.d.o.f. from segments with z view"),
)
dtSegmentFlatTableProducer.detIdVariables = cms.PSet(
wheel = DetIdVar("wheel()", "int8", doc = "wheel - [-2:2] range"),
sector = DetIdVar("sector()", "int8", doc = "sector - [1:14] range"
"<br />sector 13 used for the second MB4 of sector 4"
"<br />sector 14 used for the second MB4 of sector 10"),
station = DetIdVar("station()", "int8", doc = "station - [1:4] range")
)
dtSegmentFlatTableProducer.globalPosVariables = cms.PSet(
seg4D_posGlb_phi = GlobGeomVar("phi().value()", doc = "position phi in global coordinates - radians [-pi:pi]"),
seg4D_posGlb_eta = GlobGeomVar("eta()", doc = "position eta in global coordinates"),
)
dtSegmentFlatTableProducer.globalDirVariables = cms.PSet(
seg4D_dirGlb_phi = GlobGeomVar("phi().value()", doc = "direction phi in global coordinates - radians [-pi:pi]"),
seg4D_dirGlb_eta = GlobGeomVar("eta()", doc = "direction eta in global coordinates"),
)
from DPGAnalysis.MuonTools.muDTSegmentExtTableProducer_cfi import muDTSegmentExtTableProducer
from DPGAnalysis.MuonTools.rpcRecHitFlatTableProducer_cfi import rpcRecHitFlatTableProducer
rpcRecHitFlatTableProducer.name = "rpcRecHit"
rpcRecHitFlatTableProducer.src = "rpcRecHits"
rpcRecHitFlatTableProducer.doc = "RPC rec-hit information"
rpcRecHitFlatTableProducer.variables = cms.PSet(
bx = Var("BunchX()", int, doc="bunch crossing number"),
time = Var("time()", float, doc = "time information in ns"),
firstClusterStrip = Var("firstClusterStrip()", "int8", doc = "lowest-numbered strip in the cluster"),
clusterSize = Var("clusterSize()", "int8", doc = "number of strips in the cluster"),
coordX = Var("localPosition().x()", float, doc = "position x in local coordinates - cm"),
coordY = Var("localPosition().y()", float, doc = "position y in local coordinates - cm"),
coordZ = Var("localPosition().z()", float, doc = "position z in local coordinates - cm"),
)
rpcRecHitFlatTableProducer.detIdVariables = cms.PSet(
region = DetIdVar("region()", "int8", doc = "0: barrel, +-1: endcap"),
ring = DetIdVar("ring()", "int8", doc = "ring id:"
"<br />wheel number in barrel (from -2 to +2)"
"<br />ring number in endcap (from 1 to 3)"),
station = DetIdVar("station()", "int8", doc = "chambers at same R in barrel, chambers at same Z ion endcap"),
layer = DetIdVar("layer()", "int8", doc = "layer id:"
"<br />in station 1 and 2 for barrel, we have two layers of chambers:"
"<br />layer 1 is the inner chamber and layer 2 is the outer chamber"),
sector = DetIdVar("sector()", "int8", doc = "group of chambers at same phi"),
subsector = DetIdVar("subsector()", "int8", doc = "Some sectors are divided along the phi direction in subsectors "
"(from 1 to 4 in Barrel, from 1 to 6 in Endcap)"),
roll = DetIdVar("roll()", "int8", doc = "roll id (also known as eta partition):"
"<br />each chamber is divided along the strip direction"),
rawId = DetIdVar("rawId()", "uint", doc = "unique detector unit ID")
)
dtrpcPointFlatTableProducer = rpcRecHitFlatTableProducer.clone(name = 'dtrpcPointProducer', src = cms.InputTag('rpcPointProducer','RPCDTExtrapolatedPoints'), doc = "DT extrapolated point on RPC")
dtrpcPointFlatTableProducer.variables = cms.PSet(
coordX = Var("localPosition().x()", float, doc = "position x in local coordinates - cm"),
coordY = Var("localPosition().y()", float, doc = "position y in local coordinates - cm"),
coordZ = Var("localPosition().z()", float, doc = "position z in local coordinates - cm"),
)
dtrpcPointFlatTableProducer.detIdVariables = cms.PSet(
region = DetIdVar("region()", "int8", doc = "0: barrel, +-1: endcap"),
ring = DetIdVar("ring()", "int8", doc = "ring id:"
"<br />wheel number in barrel (from -2 to +2)"
"<br />ring number in endcap (from 1 to 3)"),
station = DetIdVar("station()", "int8", doc = "chambers at same R in barrel, chambers at same Z ion endcap"),
layer = DetIdVar("layer()", "int8", doc = "layer id:"
"<br />in station 1 and 2 for barrel, we have two layers of chambers:"
"<br />layer 1 is the inner chamber and layer 2 is the outer chamber"),
sector = DetIdVar("sector()", "int8", doc = "group of chambers at same phi"),
subsector = DetIdVar("subsector()", "int8", doc = "Some sectors are divided along the phi direction in subsectors "
"(from 1 to 4 in Barrel, from 1 to 6 in Endcap)"),
roll = DetIdVar("roll()", "int8", doc = "roll id (also known as eta partition):"
"<br />each chamber is divided along the strip direction"),
rawId = DetIdVar("rawId()", "uint", doc = "unique detector unit ID")
)
cscrpcPointFlatTableProducer = rpcRecHitFlatTableProducer.clone(name = 'cscToRpc',
src = cms.InputTag('rpcPointProducer','RPCCSCExtrapolatedPoints'),
doc = "CSC segment extrapolated on RPC")
cscrpcPointFlatTableProducer.variables = cms.PSet(
coordX = Var("localPosition().x()", float, doc = "position x in local coordinates - cm"),
coordY = Var("localPosition().y()", float, doc = "position y in local coordinates - cm"),
coordZ = Var("localPosition().z()", float, doc = "position z in local coordinates - cm"),
)
cscrpcPointFlatTableProducer.detIdVariables = cms.PSet(
region = DetIdVar("region()", "int8", doc = "0: barrel, +-1: endcap"),
ring = DetIdVar("ring()", "int8", doc = "ring id:"
"<br />wheel number in barrel (from -2 to +2)"
"<br />ring number in endcap (from 1 to 3)"),
station = DetIdVar("station()", "int8", doc = "chambers at same R in barrel, chambers at same Z ion endcap"),
layer = DetIdVar("layer()", "int8", doc = "layer id:"
"<br />in station 1 and 2 for barrel, we have two layers of chambers:"
"<br />layer 1 is the inner chamber and layer 2 is the outer chamber"),
sector = DetIdVar("sector()", "int8", doc = "group of chambers at same phi"),
subsector = DetIdVar("subsector()", "int8", doc = "Some sectors are divided along the phi direction in subsectors "
"(from 1 to 4 in Barrel, from 1 to 6 in Endcap)"),
roll = DetIdVar("roll()", "int8", doc = "roll id (also known as eta partition):"
"<br />each chamber is divided along the strip direction"),
rawId = DetIdVar("rawId()", "uint", doc = "unique detector unit ID")
)
dtrpcPointFlatTableProducer = rpcRecHitFlatTableProducer.clone(name = 'dtToRpc',
src = cms.InputTag('rpcPointProducer','RPCDTExtrapolatedPoints'),
doc = "DT segment extrapolated on RPC")
dtrpcPointFlatTableProducer.variables = cms.PSet(
coordX = Var("localPosition().x()", float, doc = "position x in local coordinates - cm"),
coordY = Var("localPosition().y()", float, doc = "position y in local coordinates - cm"),
coordZ = Var("localPosition().z()", float, doc = "position z in local coordinates - cm"),
)
dtrpcPointFlatTableProducer.detIdVariables = cms.PSet(
region = DetIdVar("region()", "int8", doc = "0: barrel, +-1: endcap"),
ring = DetIdVar("ring()", "int8", doc = "ring id:"
"<br />wheel number in barrel (from -2 to +2)"
"<br />ring number in endcap (from 1 to 3)"),
station = DetIdVar("station()", "int8", doc = "chambers at same R in barrel, chambers at same Z ion endcap"),
layer = DetIdVar("layer()", "int8", doc = "layer id:"
"<br />in station 1 and 2 for barrel, we have two layers of chambers:"
"<br />layer 1 is the inner chamber and layer 2 is the outer chamber"),
sector = DetIdVar("sector()", "int8", doc = "group of chambers at same phi"),
subsector = DetIdVar("subsector()", "int8", doc = "Some sectors are divided along the phi direction in subsectors "
"(from 1 to 4 in Barrel, from 1 to 6 in Endcap)"),
roll = DetIdVar("roll()", "int8", doc = "roll id (also known as eta partition):"
"<br />each chamber is divided along the strip direction"),
rawId = DetIdVar("rawId()", "uint", doc = "unique detector unit ID")
)
from DPGAnalysis.MuonTools.gemRecHitFlatTableProducer_cfi import gemRecHitFlatTableProducer
gemRecHitFlatTableProducer.name = "gemRecHit"
gemRecHitFlatTableProducer.src = "gemRecHits"
gemRecHitFlatTableProducer.doc = "GEM rec-hit information"
gemRecHitFlatTableProducer.variables = cms.PSet(
bx = Var("BunchX()", int, doc="bunch crossing number"),
clusterSize = Var("clusterSize()", "int8", doc = "number of strips in the cluster"), loc_x = Var("localPosition().x()", float, doc = "hit position x in local coordinates - cm"),
firstClusterStrip = Var("firstClusterStrip()", "int8", doc = "lowest-numbered strip in the cluster"),
loc_phi = Var("localPosition().phi().value()", float, doc = "hit position phi in local coordinates - rad"),
loc_y = Var("localPosition().y()", float, doc = "hit position y in local coordinates - cm"),
loc_z = Var("localPosition().z()", float, doc = "hit position z in local coordinates - cm"),
)
gemRecHitFlatTableProducer.detIdVariables = cms.PSet(
roll = DetIdVar("roll()", "int8", doc = "roll id, also known as eta partition:"
"<br />(partitions numbered from 1 to 8)"),
region = DetIdVar("region()", "int8", doc = "GE11 region where the hit is reconstructed"
"<br />(int, positive endcap: +1, negative endcap: -1)"),
chamber = DetIdVar("chamber()", "int8", doc = "GE11 superchamber where the hit is reconstructed"
"<br />(chambers numbered from 0 to 35)"),
layer = DetIdVar("layer()", "int8", doc = "GE11 layer where the hit is reconstructed"
"<br />(layer1: 1, layer2: 2)")
)
gemRecHitFlatTableProducer.globalPosVariables = cms.PSet(
g_r = GlobGeomVar("perp()", doc = "hit position r in global coordinates - cm"),
g_phi = GlobGeomVar("phi().value()", doc = "hit position phi in global coordinates - radians [-pi:pi]"),
g_x = GlobGeomVar("x()", doc = "hit position x in global coordinates - cm"),
g_y = GlobGeomVar("y()", doc = "hit position y in global coordinates - cm"),
g_z = GlobGeomVar("z()", doc = "hit position z in global coordinates - cm")
)
from DPGAnalysis.MuonTools.gemSegmentFlatTableProducer_cfi import gemSegmentFlatTableProducer
gemSegmentFlatTableProducer.name = "gemSegment"
gemSegmentFlatTableProducer.src = "gemSegments"
gemSegmentFlatTableProducer.doc = "GEM segment information"
gemSegmentFlatTableProducer.variables = cms.PSet(
chi2 = Var("chi2()", int, doc = "chi2 from segment fit"),
bx = Var("bunchX()", int, doc="bunch crossing number"),
posLoc_x = Var("localPosition().x()", float, doc = "position x in local coordinates - cm"),
posLoc_y = Var("localPosition().y()", float, doc = "position y in local coordinates - cm"),
posLoc_z = Var("localPosition().z()", float, doc = "position z in local coordinates - cm"),
dirLoc_x = Var("localDirection().x()", float, doc = "direction x in local coordinates"),
dirLoc_y = Var("localDirection().y()", float, doc = "direction y in local coordinates"),
dirLoc_z = Var("localDirection().z()", float, doc = "direction z in local coordinates"),
)
gemSegmentFlatTableProducer.detIdVariables = cms.PSet(
region = DetIdVar("region()", "int8", doc = "GE11 region where the hit is reconstructed"
"<br />(int, positive endcap: +1, negative endcap: -1)"),
ring = DetIdVar("ring()", "int8", doc = ""),
station = DetIdVar("station()", "int8", doc = "GEM station <br />(always 1 for GE1/1)"),
chamber = DetIdVar("chamber()", "int8", doc = "GE11 superchamber where the hit is reconstructed"
"<br />(chambers numbered from 0 to 35)")
)
gemSegmentFlatTableProducer.globalPosVariables = cms.PSet(
posGlb_x = GlobGeomVar("x()", doc = "position x in global coordinates - cm"),
posGlb_y = GlobGeomVar("y()", doc = "position y in global coordinates - cm"),
posGlb_z = GlobGeomVar("z()", doc = "position z in global coordinates - cm"),
posGlb_phi = GlobGeomVar("phi().value()", doc = "position phi in global coordinates - radians [-pi:pi]"),
posGlb_eta = GlobGeomVar("eta()", doc = "position eta in global coordinates"),
)
gemSegmentFlatTableProducer.globalDirVariables = cms.PSet(
dirGlb_phi = GlobGeomVar("phi().value()", doc = "direction phi in global coordinates - radians [-pi:pi]"),
dirGlb_eta = GlobGeomVar("eta()", doc = "direction eta in global coordinates"),
)
muLocalRecoProducers = cms.Sequence(rpcRecHitFlatTableProducer
+ dtrpcPointFlatTableProducer
+ cscrpcPointFlatTableProducer
+ gemRecHitFlatTableProducer
+ dtSegmentFlatTableProducer
+ muDTSegmentExtTableProducer
+ gemSegmentFlatTableProducer
)