Loading [MathJax]/extensions/tex2jax.js
ATLAS Offline Software
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Modules Pages
NewVrtSecInclusiveConfig.py
Go to the documentation of this file.
1 # Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
2 # Author: Vadim Kostyukhin vadim.kostyukhin@cern.ch
3 
4 from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
5 from AthenaConfiguration.ComponentFactory import CompFactory
6 from TrkConfig.TrkVKalVrtFitterConfig import TrkVKalVrtFitterCfg
7 from TrkConfig.AtlasExtrapolatorConfig import AtlasExtrapolatorCfg
8 from TrackToVertex.TrackToVertexConfig import TrackToVertexCfg
9 from BeamSpotConditions.BeamSpotConditionsConfig import BeamSpotCondAlgCfg
10 
11 from AthenaCommon.Logging import logging
12 mlog = logging.getLogger('Rec__NewVrtSecInclusiveConfig')
13 
14 
17 def SoftBFinderToolCfg(flags, name="SoftBFinderTool", **myargs):
18 
19  mlog.info("entering SoftBFinderTool configuration")
20  acc = ComponentAccumulator()
21  acc.merge(BeamSpotCondAlgCfg(flags))
22 
23  myargs.setdefault("VertexFitterTool", acc.popToolsAndMerge(TrkVKalVrtFitterCfg(flags)))
24  myargs.setdefault("ExtrapolatorName", acc.popToolsAndMerge(AtlasExtrapolatorCfg(flags)))
25  myargs.setdefault("TrackToVertexTool", acc.popToolsAndMerge(TrackToVertexCfg(flags)))
26  myargs.setdefault("CutPt" , 500.)
27  myargs.setdefault("CutBLayHits" , 1 )
28  myargs.setdefault("CutPixelHits" , 3 )
29  myargs.setdefault("CutSiHits" , 8 )
30  myargs.setdefault("CutTRTHits" , 10 )
31  myargs.setdefault("useVertexCleaning" , True)
32  myargs.setdefault("MultiWithOneTrkVrt" , True)
33  myargs.setdefault("removeTrkMatSignif" , -1.) # No additional material rejection
34  myargs.setdefault("AntiPileupSigRCut" , 2.0)
35  myargs.setdefault("TrkSigCut" , 2.0)
36  myargs.setdefault("SelVrtSigCut" , 3.0)
37  myargs.setdefault("v2tIniBDTCut" , -0.7)
38  myargs.setdefault("v2tFinBDTCut" , 0.0)
39  myargs.setdefault("cosSVPVCut" , 0.4)
40  myargs.setdefault("FastZSVCut" , 5.)
41  myargs.setdefault("VertexMergeCut" , 4.)
42  myargs.setdefault("MaxSVRadiusCut" , 50.)
43  SoftBFinder = CompFactory.Rec.NewVrtSecInclusiveTool(name,**myargs)
44  acc.setPrivateTools(SoftBFinder)
45  mlog.info("SoftBFinderTool created")
46 
47  return acc
48 
49 
52 def InclusiveBFinderToolCfg(flags, name="InclusiveBFinderTool", **myargs):
53 
54  mlog.info("entering InclusiveBFinderTool configuration")
55  acc = ComponentAccumulator()
56  acc.merge(BeamSpotCondAlgCfg(flags))
57 
58  myargs.setdefault("VertexFitterTool" , acc.popToolsAndMerge(TrkVKalVrtFitterCfg(flags)))
59  myargs.setdefault("ExtrapolatorName" , acc.popToolsAndMerge(AtlasExtrapolatorCfg(flags)))
60  myargs.setdefault("TrackToVertexTool", acc.popToolsAndMerge(TrackToVertexCfg(flags)))
61  myargs.setdefault("CutPt" , 500.)
62  myargs.setdefault("CutBLayHits" , 0)
63  myargs.setdefault("CutPixelHits" , 2)
64  myargs.setdefault("CutSiHits" , 8)
65  myargs.setdefault("CutTRTHits" , 10)
66  myargs.setdefault("useVertexCleaning" , True)
67  myargs.setdefault("MultiWithOneTrkVrt" , True)
68  myargs.setdefault("removeTrkMatSignif" , -1.) # No additional material rejection
69  myargs.setdefault("AntiPileupSigRCut" , 2.0)
70  myargs.setdefault("TrkSigCut" , 2.0)
71  myargs.setdefault("SelVrtSigCut" , 3.0)
72  myargs.setdefault("v2tIniBDTCut" , -0.7)
73  myargs.setdefault("v2tFinBDTCut" , -0.2)
74  myargs.setdefault("cosSVPVCut" , 0.5)
75  myargs.setdefault("FastZSVCut" , 8.)
76 
77  InclusiveBFinder = CompFactory.Rec.NewVrtSecInclusiveTool(name,**myargs)
78  acc.setPrivateTools(InclusiveBFinder)
79  mlog.info("InclusiveBFinderTool created")
80 
81  return acc
82 
83 
84 
87 def HighPtBFinderToolCfg(flags, name="HighPtBFinderTool", **myargs):
88 
89  mlog.info("entering HighPtBFinderTool configuration")
90  acc = ComponentAccumulator()
91  acc.merge(BeamSpotCondAlgCfg(flags))
92 
93  myargs.setdefault("VertexFitterTool" , acc.popToolsAndMerge(TrkVKalVrtFitterCfg(flags)))
94  myargs.setdefault("ExtrapolatorName" , acc.popToolsAndMerge(AtlasExtrapolatorCfg(flags)))
95  myargs.setdefault("TrackToVertexTool", acc.popToolsAndMerge(TrackToVertexCfg(flags)))
96  myargs.setdefault("CutPt" , 1000.)
97  myargs.setdefault("CutBLayHits" , 0)
98  myargs.setdefault("CutPixelHits" , 2)
99  myargs.setdefault("CutSiHits" , 8)
100  myargs.setdefault("CutTRTHits" , 10)
101  myargs.setdefault("useVertexCleaning" , True)
102  myargs.setdefault("MultiWithOneTrkVrt" , True)
103  myargs.setdefault("removeTrkMatSignif" , -1.) # No additional material rejection
104  myargs.setdefault("AntiPileupSigRCut" , 2.0)
105  myargs.setdefault("TrkSigCut" , 2.0)
106  myargs.setdefault("SelVrtSigCut" , 3.0)
107  myargs.setdefault("v2tIniBDTCut" ,-0.6)
108  myargs.setdefault("v2tFinBDTCut" , 0.2)
109  myargs.setdefault("cosSVPVCut" , 0.7)
110  myargs.setdefault("FastZSVCut" , 8.)
111 
112  HighPtBFinder = CompFactory.Rec.NewVrtSecInclusiveTool(name,**myargs)
113  acc.setPrivateTools(HighPtBFinder)
114  mlog.info("HighPtBFinderTool created")
115 
116  return acc
117 
118 
119 
122 def MaterialSVFinderToolCfg(flags, name="MaterialSVFinderTool", **myargs):
123 
124  mlog.info("entering MaterialSVFinderTool configuration")
125  acc = ComponentAccumulator()
126  acc.merge(BeamSpotCondAlgCfg(flags))
127 
128  myargs.setdefault("VertexFitterTool" , acc.popToolsAndMerge(TrkVKalVrtFitterCfg(flags)))
129  myargs.setdefault("ExtrapolatorName" , acc.popToolsAndMerge(AtlasExtrapolatorCfg(flags)))
130  myargs.setdefault("TrackToVertexTool", acc.popToolsAndMerge(TrackToVertexCfg(flags)))
131  myargs.setdefault("CutPt" , 500.)
132  myargs.setdefault("CutBLayHits" , 0)
133  myargs.setdefault("CutPixelHits" , 1)
134  myargs.setdefault("CutSiHits" , 8)
135  myargs.setdefault("CutTRTHits" , 10)
136  myargs.setdefault("useVertexCleaning" , False)
137  myargs.setdefault("MultiWithOneTrkVrt" , False)
138  myargs.setdefault("removeTrkMatSignif" , -1.) # No additional material rejection
139  myargs.setdefault("AntiPileupSigRCut" , 5.0)
140  myargs.setdefault("TrkSigCut" , 5.0)
141  myargs.setdefault("SelVrtSigCut" , 10.0)
142  myargs.setdefault("v2tIniBDTCut" ,-1.01) #Remove BDT selection
143  myargs.setdefault("v2tFinBDTCut" ,-1.01) #Remove BDT selection
144  myargs.setdefault("cosSVPVCut" , 0.)
145  myargs.setdefault("FastZSVCut" , 10.)
146  myargs.setdefault("VrtMassLimit" , 8000.)
147  myargs.setdefault("Vrt2TrMassLimit", 8000.)
148 
149  MaterialSVFinder = CompFactory.Rec.NewVrtSecInclusiveTool(name,**myargs)
150  acc.setPrivateTools(MaterialSVFinder)
151  mlog.info("MaterialSVFinderTool created")
152 
153  return acc
154 
155 
158 def DVFinderToolCfg(flags, name="DVFinderTool", **myargs):
159 
160  mlog.info("entering DVFinderTool configuration")
161  acc = ComponentAccumulator()
162  acc.merge(BeamSpotCondAlgCfg(flags))
163 
164  myargs.setdefault("VertexFitterTool" , acc.popToolsAndMerge(TrkVKalVrtFitterCfg(flags)))
165  myargs.setdefault("ExtrapolatorName" , acc.popToolsAndMerge(AtlasExtrapolatorCfg(flags)))
166  myargs.setdefault("TrackToVertexTool", acc.popToolsAndMerge(TrackToVertexCfg(flags)))
167  myargs.setdefault("CutPt" , 1000.)
168  myargs.setdefault("CutBLayHits" , 0)
169  myargs.setdefault("CutPixelHits" , 0)
170  myargs.setdefault("CutSiHits" , 7)
171  myargs.setdefault("CutTRTHits" , 0)
172  myargs.setdefault("useVertexCleaning" , False)
173  myargs.setdefault("MultiWithOneTrkVrt" , False)
174  myargs.setdefault("removeTrkMatSignif" , -1.) # No additional material rejection
175  myargs.setdefault("AntiPileupSigRCut" , 6.0)
176  myargs.setdefault("TrkSigCut" , 10.0)
177  myargs.setdefault("SelVrtSigCut" , 8.0)
178  myargs.setdefault("v2tIniBDTCut" ,-1.01) # BDT selection is disabled
179  myargs.setdefault("v2tFinBDTCut" ,-1.01) # BDT selection is disabled
180  myargs.setdefault("cosSVPVCut" , 0.)
181  myargs.setdefault("FastZSVCut" , 30.)
182  myargs.setdefault("VrtMassLimit" , 1000000.)
183  myargs.setdefault("Vrt2TrMassLimit", 1000000.)
184  myargs.setdefault("VertexMergeCut" , 10.)
185  myargs.setdefault("MaxSVRadiusCut" , 350.)
186  myargs.setdefault("CutD0Max" , 1000.) # Maximal track impact parameter
187  myargs.setdefault("CutD0Min" , 0.) # Minimal track impact parameter
188  myargs.setdefault("MaxZVrt" , 100.)
189  myargs.setdefault("MinZVrt" , 0.)
190  myargs.setdefault("TwoTrkVtxFormingD0Cut", 1.0) # 2-track forming cut
191  myargs.setdefault("do2TrkIBLChecks", False) # Do not explicitly require IBL/BL hits
192 
193  DVFinder = CompFactory.Rec.NewVrtSecInclusiveTool(name,**myargs)
194  acc.setPrivateTools(DVFinder)
195  mlog.info("DVFinderTool created")
196 
197  return acc
198 
199 
203 def V2TCalibrationToolCfg(flags, name="V2TCalibrationTool", **myargs):
204 
205  mlog.info("entering V2TCalibrationTool configuration")
206  acc = ComponentAccumulator()
207  acc.merge(BeamSpotCondAlgCfg(flags))
208 
209  myargs.setdefault("VertexFitterTool" , acc.popToolsAndMerge(TrkVKalVrtFitterCfg(flags)))
210  myargs.setdefault("ExtrapolatorName" , acc.popToolsAndMerge(AtlasExtrapolatorCfg(flags)))
211  myargs.setdefault("TrackToVertexTool", acc.popToolsAndMerge(TrackToVertexCfg(flags)))
212  myargs.setdefault("FillHist" , True)
213  myargs.setdefault("CutPt" , 400.)
214  myargs.setdefault("CutBLayHits" , 0)
215  myargs.setdefault("CutPixelHits" , 1)
216  myargs.setdefault("CutSiHits" , 8)
217  myargs.setdefault("CutTRTHits" , 10)
218  myargs.setdefault("useVertexCleaning" , False)
219  myargs.setdefault("MultiWithOneTrkVrt" , False)
220  myargs.setdefault("removeTrkMatSignif" , -1.) # No additional material rejection
221  myargs.setdefault("AntiPileupSigRCut" , 2.0)
222  myargs.setdefault("TrkSigCut" , 2.0)
223  myargs.setdefault("SelVrtSigCut" , 2.0)
224  myargs.setdefault("v2tIniBDTCut" , -1.01) #Remove BDT selection
225  myargs.setdefault("v2tFinBDTCut" , -1.01) #Remove BDT selection
226  myargs.setdefault("cosSVPVCut" , 0.)
227  myargs.setdefault("FastZSVCut" , 15.)
228  myargs.setdefault("VrtMassLimit" , 5500.)
229  myargs.setdefault("Vrt2TrMassLimit", 4000.)
230  myargs.setdefault("MaxSVRadiusCut" , 140.)
231  myargs.setdefault("CutD0Max" , 100.) # Maximal track impact parameter
232  myargs.setdefault("CutD0Min" , 0.) # Minimal track impact parameter
233  myargs.setdefault("MaxZVrt" , 100.)
234 
235 
236  V2TCalibration = CompFactory.Rec.NewVrtSecInclusiveTool(name,**myargs)
237  acc.setPrivateTools(V2TCalibration)
238  mlog.info("V2TCalibrationTool created")
239 
240  return acc
241 
242 
python.JetAnalysisCommon.ComponentAccumulator
ComponentAccumulator
Definition: JetAnalysisCommon.py:302
python.BeamSpotConditionsConfig.BeamSpotCondAlgCfg
def BeamSpotCondAlgCfg(flags, name="BeamSpotCondAlg", **kwargs)
Definition: BeamSpotConditionsConfig.py:7
NewVrtSecInclusiveConfig.SoftBFinderToolCfg
def SoftBFinderToolCfg(flags, name="SoftBFinderTool", **myargs)
Definition: NewVrtSecInclusiveConfig.py:17
python.AtlasExtrapolatorConfig.AtlasExtrapolatorCfg
def AtlasExtrapolatorCfg(flags, name='AtlasExtrapolator')
Definition: AtlasExtrapolatorConfig.py:63
NewVrtSecInclusiveConfig.MaterialSVFinderToolCfg
def MaterialSVFinderToolCfg(flags, name="MaterialSVFinderTool", **myargs)
Definition: NewVrtSecInclusiveConfig.py:122
python.TrkVKalVrtFitterConfig.TrkVKalVrtFitterCfg
def TrkVKalVrtFitterCfg(flags, name="TrkVKalVrtFitter", **kwargs)
Definition: TrkVKalVrtFitterConfig.py:7
NewVrtSecInclusiveConfig.DVFinderToolCfg
def DVFinderToolCfg(flags, name="DVFinderTool", **myargs)
Definition: NewVrtSecInclusiveConfig.py:158
NewVrtSecInclusiveConfig.HighPtBFinderToolCfg
def HighPtBFinderToolCfg(flags, name="HighPtBFinderTool", **myargs)
Definition: NewVrtSecInclusiveConfig.py:87
TrackToVertexConfig.TrackToVertexCfg
def TrackToVertexCfg(flags, name="AtlasTrackToVertexTool", **kwargs)
Definition: TrackToVertexConfig.py:6
NewVrtSecInclusiveConfig.InclusiveBFinderToolCfg
def InclusiveBFinderToolCfg(flags, name="InclusiveBFinderTool", **myargs)
Definition: NewVrtSecInclusiveConfig.py:52
NewVrtSecInclusiveConfig.V2TCalibrationToolCfg
def V2TCalibrationToolCfg(flags, name="V2TCalibrationTool", **myargs)
Configuration for creation of calibration ntuples for 2-track vertex classification BDT.
Definition: NewVrtSecInclusiveConfig.py:203