4from AthenaConfiguration.ComponentAccumulator
import ComponentAccumulator
5from AthenaConfiguration.ComponentFactory
import CompFactory
6from TrkConfig.TrkVKalVrtFitterConfig
import TrkVKalVrtFitterCfg
7from TrkConfig.AtlasExtrapolatorConfig
import AtlasExtrapolatorCfg
8from TrackToVertex.TrackToVertexConfig
import TrackToVertexCfg
9from BeamSpotConditions.BeamSpotConditionsConfig
import BeamSpotCondAlgCfg
11from AthenaCommon.Logging
import logging
12mlog = logging.getLogger(
'Rec__NewVrtSecInclusiveConfig')
19 mlog.info(
"entering SoftBFinderTool configuration")
20 acc = ComponentAccumulator()
21 acc.merge(BeamSpotCondAlgCfg(flags))
24 iniV2T_cosSVPVCut = myargs.pop(
"IniV2T_cosSVPVCut", 0.4)
25 iniV2T_v2tBDTCut = myargs.pop(
"IniV2T_v2tBDTCut", -0.7)
26 finV2T_cosSVPVCut = myargs.pop(
"FinV2T_cosSVPVCut", 0.4)
27 finV2T_v2tBDTCut = myargs.pop(
"FinV2T_v2tBDTCut", 0.)
32 iniV2Targs.setdefault(
"Vrt2TrPtMin" , 1000.)
33 iniV2Targs.setdefault(
"useVertexCleaning" ,
True)
34 iniV2Targs.setdefault(
"cosSVPVCut" , iniV2T_cosSVPVCut)
35 iniV2Targs.setdefault(
"v2tBDTCut" , iniV2T_v2tBDTCut)
36 iniV2TSelector = CompFactory.Rec.TwoTrackVrtBDTSelector(
"IniV2TSelector",**iniV2Targs)
39 finV2Targs.setdefault(
"Vrt2TrPtMin" , 1000.)
40 finV2Targs.setdefault(
"useVertexCleaning" ,
True)
41 finV2Targs.setdefault(
"cosSVPVCut" , finV2T_cosSVPVCut)
42 finV2Targs.setdefault(
"v2tBDTCut" , finV2T_v2tBDTCut)
43 finV2TSelector = CompFactory.Rec.TwoTrackVrtBDTSelector(
"FinV2TSelector",**finV2Targs)
46 myargs.setdefault(
"CutPt" , 500.)
47 myargs.setdefault(
"CutBLayHits" , 1 )
48 myargs.setdefault(
"CutPixelHits" , 3 )
49 myargs.setdefault(
"CutSiHits" , 8 )
50 myargs.setdefault(
"CutTRTHits" , 10 )
51 myargs.setdefault(
"AntiPileupSigRCut" , 2.0)
52 myargs.setdefault(
"TrkSigCut" , 2.0)
54 myargs.setdefault(
"FastZSVCut" , 5.)
55 myargs.setdefault(
"MultiWithOneTrkVrt" ,
True)
56 myargs.setdefault(
"removeTrkMatSignif" , -1.)
57 myargs.setdefault(
"SelVrtSigCut" , 2.5)
59 myargs.setdefault(
"VertexMergeCut" , 4.)
60 myargs.setdefault(
"MaxSVRadiusCut" , 50.)
62 myargs.setdefault(
"TwoTrkVtxSelectorIni",iniV2TSelector)
63 myargs.setdefault(
"TwoTrkVtxSelectorFinal",finV2TSelector)
64 myargs.setdefault(
"VertexFitterTool", acc.popToolsAndMerge(TrkVKalVrtFitterCfg(flags)))
65 myargs.setdefault(
"ExtrapolatorName", acc.popToolsAndMerge(AtlasExtrapolatorCfg(flags)))
66 myargs.setdefault(
"TrackToVertexTool", acc.popToolsAndMerge(TrackToVertexCfg(flags)))
68 SoftBFinder = CompFactory.Rec.NewVrtSecInclusiveTool(name,**myargs)
69 acc.setPrivateTools(SoftBFinder)
70 mlog.info(
"SoftBFinderTool created")
79 mlog.info(
"entering InclusiveBFinderTool configuration")
80 acc = ComponentAccumulator()
81 acc.merge(BeamSpotCondAlgCfg(flags))
85 iniV2Targs.setdefault(
"Vrt2TrPtMin" , 1000.)
86 iniV2Targs.setdefault(
"cosSVPVCut" , 0.5)
87 iniV2Targs.setdefault(
"v2tBDTCut" , -0.7)
88 iniV2TSelector = CompFactory.Rec.TwoTrackVrtBDTSelector(
"IniV2TSelector",**iniV2Targs)
91 finV2Targs.setdefault(
"Vrt2TrPtMin" , 1000.)
92 finV2Targs.setdefault(
"cosSVPVCut" , 0.5)
93 finV2Targs.setdefault(
"v2tBDTCut" , -0.2)
94 finV2TSelector = CompFactory.Rec.TwoTrackVrtBDTSelector(
"FinV2TSelector",**finV2Targs)
97 myargs.setdefault(
"CutPt" , 500.)
98 myargs.setdefault(
"CutBLayHits" , 0 )
99 myargs.setdefault(
"CutPixelHits", 2 )
100 myargs.setdefault(
"CutSiHits" , 8 )
101 myargs.setdefault(
"CutTRTHits" , 10 )
102 myargs.setdefault(
"AntiPileupSigRCut" , 2.0)
103 myargs.setdefault(
"TrkSigCut" , 2.0)
105 myargs.setdefault(
"FastZSVCut" , 8.0)
106 myargs.setdefault(
"MultiWithOneTrkVrt" ,
True)
107 myargs.setdefault(
"removeTrkMatSignif" , -1.)
108 myargs.setdefault(
"SelVrtSigCut" , 3.0)
112 myargs.setdefault(
"TwoTrkVtxSelectorIni", iniV2TSelector)
113 myargs.setdefault(
"TwoTrkVtxSelectorFinal",finV2TSelector)
114 myargs.setdefault(
"VertexFitterTool", acc.popToolsAndMerge(TrkVKalVrtFitterCfg(flags)))
115 myargs.setdefault(
"ExtrapolatorName", acc.popToolsAndMerge(AtlasExtrapolatorCfg(flags)))
116 myargs.setdefault(
"TrackToVertexTool", acc.popToolsAndMerge(TrackToVertexCfg(flags)))
119 InclusiveBFinder = CompFactory.Rec.NewVrtSecInclusiveTool(name,**myargs)
120 acc.setPrivateTools(InclusiveBFinder)
121 mlog.info(
"InclusiveBFinderTool created")
131 mlog.info(
"entering HighPtBFinderTool configuration")
132 acc = ComponentAccumulator()
133 acc.merge(BeamSpotCondAlgCfg(flags))
137 iniV2Targs.setdefault(
"Vrt2TrPtMin" , 1000.)
138 iniV2Targs.setdefault(
"cosSVPVCut" , 0.7)
139 iniV2Targs.setdefault(
"v2tBDTCut" , -0.6)
140 iniV2TSelector = CompFactory.Rec.TwoTrackVrtBDTSelector(
"IniV2TSelector",**iniV2Targs)
143 finV2Targs.setdefault(
"Vrt2TrPtMin" , 1000.)
144 finV2Targs.setdefault(
"cosSVPVCut" , 0.7)
145 finV2Targs.setdefault(
"v2tBDTCut" , -0.2)
146 finV2TSelector = CompFactory.Rec.TwoTrackVrtBDTSelector(
"FinV2TSelector",**finV2Targs)
149 myargs.setdefault(
"CutPt" , 1000.)
150 myargs.setdefault(
"CutBLayHits" , 0 )
151 myargs.setdefault(
"CutPixelHits", 2 )
152 myargs.setdefault(
"CutSiHits" , 8 )
153 myargs.setdefault(
"CutTRTHits" , 10 )
154 myargs.setdefault(
"AntiPileupSigRCut", 2.0)
155 myargs.setdefault(
"TrkSigCut" , 2.0)
157 myargs.setdefault(
"FastZSVCut" , 8.0)
158 myargs.setdefault(
"MultiWithOneTrkVrt" ,
True)
159 myargs.setdefault(
"removeTrkMatSignif" , -1.)
160 myargs.setdefault(
"SelVrtSigCut" , 3.0)
163 myargs.setdefault(
"TwoTrkVtxSelectorIni", iniV2TSelector)
164 myargs.setdefault(
"TwoTrkVtxSelectorFinal",finV2TSelector)
165 myargs.setdefault(
"VertexFitterTool", acc.popToolsAndMerge(TrkVKalVrtFitterCfg(flags)))
166 myargs.setdefault(
"ExtrapolatorName", acc.popToolsAndMerge(AtlasExtrapolatorCfg(flags)))
167 myargs.setdefault(
"TrackToVertexTool", acc.popToolsAndMerge(TrackToVertexCfg(flags)))
169 HighPtBFinder = CompFactory.Rec.NewVrtSecInclusiveTool(name,**myargs)
170 acc.setPrivateTools(HighPtBFinder)
171 mlog.info(
"HighPtBFinderTool created")
181 mlog.info(
"entering MaterialSVFinderTool configuration")
182 acc = ComponentAccumulator()
183 acc.merge(BeamSpotCondAlgCfg(flags))
187 iniV2Targs.setdefault(
"Vrt2TrPtMin" , 1000.)
188 iniV2Targs.setdefault(
"cosSVPVCut" , 0.)
189 iniV2Targs.setdefault(
"useVertexCleaning",
False)
190 iniV2Targs.setdefault(
"v2tBDTCut" , -1.01)
191 iniV2Targs.setdefault(
"Vrt2TrMassLimit", 8000.)
192 iniV2TSelector = CompFactory.Rec.TwoTrackVrtBDTSelector(
"IniV2TSelector",**iniV2Targs)
195 finV2Targs.setdefault(
"Vrt2TrPtMin" , 1000.)
196 finV2Targs.setdefault(
"cosSVPVCut" , 0.)
197 finV2Targs.setdefault(
"useVertexCleaning",
False)
198 finV2Targs.setdefault(
"v2tBDTCut" , -1.01)
199 finV2Targs.setdefault(
"Vrt2TrMassLimit", 8000.)
200 finV2TSelector = CompFactory.Rec.TwoTrackVrtBDTSelector(
"FinV2TSelector",**finV2Targs)
203 myargs.setdefault(
"CutPt" , 500.)
204 myargs.setdefault(
"CutBLayHits" , 0 )
205 myargs.setdefault(
"CutPixelHits" , 1 )
206 myargs.setdefault(
"CutSiHits" , 8 )
207 myargs.setdefault(
"CutTRTHits" , 10 )
208 myargs.setdefault(
"AntiPileupSigRCut", 5.0)
209 myargs.setdefault(
"TrkSigCut" , 2.0)
211 myargs.setdefault(
"FastZSVCut" , 10.0)
212 myargs.setdefault(
"MultiWithOneTrkVrt" ,
False)
213 myargs.setdefault(
"removeTrkMatSignif" , -1.)
214 myargs.setdefault(
"SelVrtSigCut" , 10.0)
215 myargs.setdefault(
"VrtMassLimit", 8000.)
218 myargs.setdefault(
"TwoTrkVtxSelectorIni", iniV2TSelector)
219 myargs.setdefault(
"TwoTrkVtxSelectorFinal",finV2TSelector)
220 myargs.setdefault(
"VertexFitterTool", acc.popToolsAndMerge(TrkVKalVrtFitterCfg(flags)))
221 myargs.setdefault(
"ExtrapolatorName", acc.popToolsAndMerge(AtlasExtrapolatorCfg(flags)))
222 myargs.setdefault(
"TrackToVertexTool", acc.popToolsAndMerge(TrackToVertexCfg(flags)))
224 MaterialSVFinder = CompFactory.Rec.NewVrtSecInclusiveTool(name,**myargs)
225 acc.setPrivateTools(MaterialSVFinder)
226 mlog.info(
"MaterialSVFinderTool created")
235 mlog.info(
"entering KsFinderTool configuration")
236 acc = ComponentAccumulator()
237 acc.merge(BeamSpotCondAlgCfg(flags))
241 iniV2Targs.setdefault(
"Vrt2TrPtMin" , 1000.)
242 iniV2Targs.setdefault(
"cosSVPVCut" , 0.)
243 iniV2Targs.setdefault(
"v2tBDTCut" , -1.01)
244 iniV2Targs.setdefault(
"MaxSVRadiusCut", 350.)
245 iniV2Targs.setdefault(
"Vrt2TrMassLimit", 1000.)
246 iniV2Targs.setdefault(
"useVertexCleaning" ,
False)
247 iniV2TSelector = CompFactory.Rec.TwoTrackVrtBDTSelector(
"IniV2TSelector",**iniV2Targs)
250 myargs.setdefault(
"CutPt" , 1000.)
251 myargs.setdefault(
"CutBLayHits" , 0 )
252 myargs.setdefault(
"CutPixelHits", 0 )
253 myargs.setdefault(
"CutSiHits" , 8 )
254 myargs.setdefault(
"CutTRTHits" , 0 )
255 myargs.setdefault(
"AntiPileupSigRCut", 6.0)
256 myargs.setdefault(
"TrkSigCut" , 10.0)
257 myargs.setdefault(
"CutD0Max" , 1000.)
258 myargs.setdefault(
"CutD0Min" , 0.)
259 myargs.setdefault(
"MaxZVrt" , 100.)
260 myargs.setdefault(
"MinZVrt" , 0.)
262 myargs.setdefault(
"FastZSVCut" , 30.0)
263 myargs.setdefault(
"MultiWithOneTrkVrt" ,
False)
264 myargs.setdefault(
"removeTrkMatSignif" , -1.)
265 myargs.setdefault(
"SelVrtSigCut" , 8.0)
266 myargs.setdefault(
"VertexMergeCut" , 10.)
267 myargs.setdefault(
"VrtMassLimit" , 800000.)
268 myargs.setdefault(
"MaxSVRadiusCut" , 350.)
271 myargs.setdefault(
"TwoTrkVtxSelectorIni" ,iniV2TSelector)
272 myargs.setdefault(
"TwoTrkVtxSelectorFinal",iniV2TSelector)
273 myargs.setdefault(
"VertexFitterTool", acc.popToolsAndMerge(TrkVKalVrtFitterCfg(flags)))
274 myargs.setdefault(
"ExtrapolatorName", acc.popToolsAndMerge(AtlasExtrapolatorCfg(flags)))
275 myargs.setdefault(
"TrackToVertexTool", acc.popToolsAndMerge(TrackToVertexCfg(flags)))
277 KsFinder = CompFactory.Rec.NewVrtSecInclusiveTool(name,**myargs)
278 acc.setPrivateTools(KsFinder)
279 mlog.info(
"KsFinderTool created")
288 mlog.info(
"entering DVFinderTool configuration")
289 acc = ComponentAccumulator()
290 acc.merge(BeamSpotCondAlgCfg(flags))
294 iniV2Targs.setdefault(
"Vrt2TrPtMin" , 2000.)
295 iniV2Targs.setdefault(
"cosSVPVCut" , 0.)
296 iniV2Targs.setdefault(
"v2tBDTCut" , -1.01)
297 iniV2Targs.setdefault(
"MaxSVRadiusCut", 350.)
298 iniV2Targs.setdefault(
"Vrt2TrMassLimit", 1000000.)
299 iniV2Targs.setdefault(
"useVertexCleaning" ,
False)
300 iniV2TSelector = CompFactory.Rec.TwoTrackVrtBDTSelector(
"IniV2TSelector",**iniV2Targs)
303 finV2Targs.setdefault(
"Vrt2TrPtMin" , 2000.)
304 finV2Targs.setdefault(
"cosSVPVCut" , 0.)
305 finV2Targs.setdefault(
"v2tBDTCut" , -1.01)
306 finV2Targs.setdefault(
"MaxSVRadiusCut", 350.)
307 finV2Targs.setdefault(
"Vrt2TrMassLimit", 1000000.)
308 finV2Targs.setdefault(
"useVertexCleaning" ,
False)
309 finV2TSelector = CompFactory.Rec.TwoTrackVrtBDTSelector(
"FinV2TSelector",**finV2Targs)
312 myargs.setdefault(
"CutPt" , 1000.)
313 myargs.setdefault(
"CutBLayHits" , 0 )
314 myargs.setdefault(
"CutPixelHits", 0 )
315 myargs.setdefault(
"CutSiHits" , 7 )
316 myargs.setdefault(
"CutTRTHits" , 0 )
317 myargs.setdefault(
"AntiPileupSigRCut", 6.0)
318 myargs.setdefault(
"TrkSigCut" , 10.0)
319 myargs.setdefault(
"CutD0Max" , 1000.)
320 myargs.setdefault(
"CutD0Min" , 0.)
321 myargs.setdefault(
"MaxZVrt" , 100.)
322 myargs.setdefault(
"MinZVrt" , 0.)
324 myargs.setdefault(
"FastZSVCut" , 30.0)
325 myargs.setdefault(
"MultiWithOneTrkVrt" ,
False)
326 myargs.setdefault(
"removeTrkMatSignif" , -1.)
327 myargs.setdefault(
"SelVrtSigCut" , 8.0)
328 myargs.setdefault(
"VertexMergeCut" , 10.)
329 myargs.setdefault(
"VrtMassLimit" , 1000000.)
330 myargs.setdefault(
"MaxSVRadiusCut" , 350.)
333 myargs.setdefault(
"TwoTrkVtxSelectorIni" ,iniV2TSelector)
334 myargs.setdefault(
"TwoTrkVtxSelectorFinal",finV2TSelector)
335 myargs.setdefault(
"VertexFitterTool", acc.popToolsAndMerge(TrkVKalVrtFitterCfg(flags)))
336 myargs.setdefault(
"ExtrapolatorName", acc.popToolsAndMerge(AtlasExtrapolatorCfg(flags)))
337 myargs.setdefault(
"TrackToVertexTool", acc.popToolsAndMerge(TrackToVertexCfg(flags)))
339 DVFinder = CompFactory.Rec.NewVrtSecInclusiveTool(name,**myargs)
340 acc.setPrivateTools(DVFinder)
341 mlog.info(
"DVFinderTool created")
351 mlog.info(
"entering V2TCalibrationTool configuration")
352 acc = ComponentAccumulator()
353 acc.merge(BeamSpotCondAlgCfg(flags))
357 iniV2Targs.setdefault(
"Vrt2TrPtMin" , 2000.)
358 iniV2Targs.setdefault(
"cosSVPVCut" , 0.)
359 iniV2Targs.setdefault(
"v2tBDTCut" , -1.01)
360 iniV2Targs.setdefault(
"Vrt2TrMassLimit", 4000.)
361 iniV2Targs.setdefault(
"useVertexCleaning" ,
False)
362 iniV2TSelector = CompFactory.Rec.TwoTrackVrtBDTSelector(
"IniV2TSelector",**iniV2Targs)
365 finV2Targs.setdefault(
"Vrt2TrPtMin" , 2000.)
366 finV2Targs.setdefault(
"cosSVPVCut" , 0.)
367 finV2Targs.setdefault(
"v2tBDTCut" , -1.01)
368 finV2Targs.setdefault(
"Vrt2TrMassLimit", 4000.)
369 finV2Targs.setdefault(
"useVertexCleaning" ,
False)
370 finV2TSelector = CompFactory.Rec.TwoTrackVrtBDTSelector(
"FinV2TSelector",**finV2Targs)
372 myargs.setdefault(
"FillHist" ,
True)
374 myargs.setdefault(
"CutPt" , 400.)
375 myargs.setdefault(
"CutBLayHits" , 0 )
376 myargs.setdefault(
"CutPixelHits", 1 )
377 myargs.setdefault(
"CutSiHits" , 8 )
378 myargs.setdefault(
"CutTRTHits" , 10 )
379 myargs.setdefault(
"AntiPileupSigRCut", 2.0)
380 myargs.setdefault(
"TrkSigCut" , 2.0)
381 myargs.setdefault(
"CutD0Max" , 100.)
382 myargs.setdefault(
"CutD0Min" , 0.)
383 myargs.setdefault(
"MaxZVrt" , 100.)
384 myargs.setdefault(
"MinZVrt" , 0.)
386 myargs.setdefault(
"FastZSVCut" , 15.0)
387 myargs.setdefault(
"MultiWithOneTrkVrt" ,
False)
388 myargs.setdefault(
"removeTrkMatSignif" , -1.)
389 myargs.setdefault(
"SelVrtSigCut" , 2.0)
390 myargs.setdefault(
"VertexMergeCut" , 10.)
391 myargs.setdefault(
"VrtMassLimit" , 5500.)
392 myargs.setdefault(
"MaxSVRadiusCut" , 140.)
395 myargs.setdefault(
"TwoTrkVtxSelectorIni" ,iniV2TSelector)
396 myargs.setdefault(
"TwoTrkVtxSelectorFinal",finV2TSelector)
397 myargs.setdefault(
"VertexFitterTool", acc.popToolsAndMerge(TrkVKalVrtFitterCfg(flags)))
398 myargs.setdefault(
"ExtrapolatorName", acc.popToolsAndMerge(AtlasExtrapolatorCfg(flags)))
399 myargs.setdefault(
"TrackToVertexTool", acc.popToolsAndMerge(TrackToVertexCfg(flags)))
401 V2TCalibration = CompFactory.Rec.NewVrtSecInclusiveTool(name,**myargs)
402 acc.setPrivateTools(V2TCalibration)
403 mlog.info(
"V2TCalibrationTool created")
HighPtBFinderToolCfg(flags, name="HighPtBFinderTool", **myargs)
KsFinderToolCfg(flags, name="KsFinderTool", **myargs)
InclusiveBFinderToolCfg(flags, name="InclusiveBFinderTool", **myargs)
SoftBFinderToolCfg(flags, name="SoftBFinderTool", **myargs)
DVFinderToolCfg(flags, name="DVFinderTool", **myargs)
MaterialSVFinderToolCfg(flags, name="MaterialSVFinderTool", **myargs)
V2TCalibrationToolCfg(flags, name="V2TCalibrationTool", **myargs)
Configuration for creation of calibration ntuples for 2-track vertex classification BDT.