ATLAS Offline Software
NewVrtSecInclusiveConfig.py
Go to the documentation of this file.
1 # Copyright (C) 2002-2022 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 BeamSpotConditions.BeamSpotConditionsConfig import BeamSpotCondAlgCfg
9 
10 from AthenaCommon.Logging import logging
11 mlog = logging.getLogger('Rec__NewVrtSecInclusiveConfig')
12 
13 
16 def SoftBFinderToolCfg(flags, name="SoftBFinderTool", **myargs):
17 
18  mlog.info("entering SoftBFinderTool configuration")
19  acc = ComponentAccumulator()
20  acc.merge(BeamSpotCondAlgCfg(flags))
21 
22  myargs.setdefault("VertexFitterTool", acc.popToolsAndMerge(TrkVKalVrtFitterCfg(flags)))
23  myargs.setdefault("ExtrapolatorName", acc.popToolsAndMerge(AtlasExtrapolatorCfg(flags)))
24  myargs.setdefault("CutPt" , 500.)
25  myargs.setdefault("CutBLayHits" , 1 )
26  myargs.setdefault("CutPixelHits" , 3 )
27  myargs.setdefault("CutSiHits" , 8 )
28  myargs.setdefault("CutTRTHits" , 10 )
29  myargs.setdefault("useVertexCleaning" , True)
30  myargs.setdefault("MultiWithOneTrkVrt" , True)
31  myargs.setdefault("removeTrkMatSignif" , -1.) # No additional material rejection
32  myargs.setdefault("AntiPileupSigRCut" , 2.0)
33  myargs.setdefault("TrkSigCut" , 2.0)
34  myargs.setdefault("SelVrtSigCut" , 3.0)
35  myargs.setdefault("v2tIniBDTCut" , -0.7)
36  myargs.setdefault("v2tFinBDTCut" , 0.0)
37  myargs.setdefault("cosSVPVCut" , 0.4)
38  myargs.setdefault("FastZSVCut" , 5.)
39  myargs.setdefault("VertexMergeCut" , 4.)
40  myargs.setdefault("MaxSVRadiusCut" , 50.)
41  SoftBFinder = CompFactory.Rec.NewVrtSecInclusiveTool(name,**myargs)
42  acc.setPrivateTools(SoftBFinder)
43  mlog.info("SoftBFinderTool created")
44 
45  return acc
46 
47 
50 def InclusiveBFinderToolCfg(flags, name="InclusiveBFinderTool", **myargs):
51 
52  mlog.info("entering InclusiveBFinderTool configuration")
53  acc = ComponentAccumulator()
54  acc.merge(BeamSpotCondAlgCfg(flags))
55 
56  myargs.setdefault("VertexFitterTool" , acc.popToolsAndMerge(TrkVKalVrtFitterCfg(flags)))
57  myargs.setdefault("ExtrapolatorName" , acc.popToolsAndMerge(AtlasExtrapolatorCfg(flags)))
58  myargs.setdefault("CutPt" , 500.)
59  myargs.setdefault("CutBLayHits" , 0)
60  myargs.setdefault("CutPixelHits" , 2)
61  myargs.setdefault("CutSiHits" , 8)
62  myargs.setdefault("CutTRTHits" , 10)
63  myargs.setdefault("useVertexCleaning" , True)
64  myargs.setdefault("MultiWithOneTrkVrt" , True)
65  myargs.setdefault("removeTrkMatSignif" , -1.) # No additional material rejection
66  myargs.setdefault("AntiPileupSigRCut" , 2.0)
67  myargs.setdefault("TrkSigCut" , 2.0)
68  myargs.setdefault("SelVrtSigCut" , 3.0)
69  myargs.setdefault("v2tIniBDTCut" , -0.7)
70  myargs.setdefault("v2tFinBDTCut" , -0.2)
71  myargs.setdefault("cosSVPVCut" , 0.5)
72  myargs.setdefault("FastZSVCut" , 8.)
73 
74  InclusiveBFinder = CompFactory.Rec.NewVrtSecInclusiveTool(name,**myargs)
75  acc.setPrivateTools(InclusiveBFinder)
76  mlog.info("InclusiveBFinderTool created")
77 
78  return acc
79 
80 
81 
84 def HighPtBFinderToolCfg(flags, name="HighPtBFinderTool", **myargs):
85 
86  mlog.info("entering HighPtBFinderTool configuration")
87  acc = ComponentAccumulator()
88  acc.merge(BeamSpotCondAlgCfg(flags))
89 
90  myargs.setdefault("VertexFitterTool" , acc.popToolsAndMerge(TrkVKalVrtFitterCfg(flags)))
91  myargs.setdefault("ExtrapolatorName" , acc.popToolsAndMerge(AtlasExtrapolatorCfg(flags)))
92  myargs.setdefault("CutPt" , 1000.)
93  myargs.setdefault("CutBLayHits" , 0)
94  myargs.setdefault("CutPixelHits" , 2)
95  myargs.setdefault("CutSiHits" , 8)
96  myargs.setdefault("CutTRTHits" , 10)
97  myargs.setdefault("useVertexCleaning" , True)
98  myargs.setdefault("MultiWithOneTrkVrt" , True)
99  myargs.setdefault("removeTrkMatSignif" , -1.) # No additional material rejection
100  myargs.setdefault("AntiPileupSigRCut" , 2.0)
101  myargs.setdefault("TrkSigCut" , 2.0)
102  myargs.setdefault("SelVrtSigCut" , 3.0)
103  myargs.setdefault("v2tIniBDTCut" ,-0.6)
104  myargs.setdefault("v2tFinBDTCut" , 0.2)
105  myargs.setdefault("cosSVPVCut" , 0.7)
106  myargs.setdefault("FastZSVCut" , 8.)
107 
108  HighPtBFinder = CompFactory.Rec.NewVrtSecInclusiveTool(name,**myargs)
109  acc.setPrivateTools(HighPtBFinder)
110  mlog.info("HighPtBFinderTool created")
111 
112  return acc
113 
114 
115 
118 def MaterialSVFinderToolCfg(flags, name="MaterialSVFinderTool", **myargs):
119 
120  mlog.info("entering MaterialSVFinderTool configuration")
121  acc = ComponentAccumulator()
122  acc.merge(BeamSpotCondAlgCfg(flags))
123 
124  myargs.setdefault("VertexFitterTool" , acc.popToolsAndMerge(TrkVKalVrtFitterCfg(flags)))
125  myargs.setdefault("ExtrapolatorName" , acc.popToolsAndMerge(AtlasExtrapolatorCfg(flags)))
126  myargs.setdefault("CutPt" , 500.)
127  myargs.setdefault("CutBLayHits" , 0)
128  myargs.setdefault("CutPixelHits" , 1)
129  myargs.setdefault("CutSiHits" , 8)
130  myargs.setdefault("CutTRTHits" , 10)
131  myargs.setdefault("useVertexCleaning" , False)
132  myargs.setdefault("MultiWithOneTrkVrt" , False)
133  myargs.setdefault("removeTrkMatSignif" , -1.) # No additional material rejection
134  myargs.setdefault("AntiPileupSigRCut" , 5.0)
135  myargs.setdefault("TrkSigCut" , 5.0)
136  myargs.setdefault("SelVrtSigCut" , 10.0)
137  myargs.setdefault("v2tIniBDTCut" ,-1.01) #Remove BDT selection
138  myargs.setdefault("v2tFinBDTCut" ,-1.01) #Remove BDT selection
139  myargs.setdefault("cosSVPVCut" , 0.)
140  myargs.setdefault("FastZSVCut" , 10.)
141  myargs.setdefault("VrtMassLimit" , 8000.)
142  myargs.setdefault("Vrt2TrMassLimit", 8000.)
143 
144  MaterialSVFinder = CompFactory.Rec.NewVrtSecInclusiveTool(name,**myargs)
145  acc.setPrivateTools(MaterialSVFinder)
146  mlog.info("MaterialSVFinderTool created")
147 
148  return acc
149 
150 
153 def DVFinderToolCfg(flags, name="DVFinderTool", **myargs):
154 
155  mlog.info("entering DVFinderTool configuration")
156  acc = ComponentAccumulator()
157  acc.merge(BeamSpotCondAlgCfg(flags))
158 
159  myargs.setdefault("VertexFitterTool" , acc.popToolsAndMerge(TrkVKalVrtFitterCfg(flags)))
160  myargs.setdefault("ExtrapolatorName" , acc.popToolsAndMerge(AtlasExtrapolatorCfg(flags)))
161  myargs.setdefault("CutPt" , 1000.)
162  myargs.setdefault("CutBLayHits" , 0)
163  myargs.setdefault("CutPixelHits" , 0)
164  myargs.setdefault("CutSiHits" , 7)
165  myargs.setdefault("CutTRTHits" , 15)
166  myargs.setdefault("useVertexCleaning" , False)
167  myargs.setdefault("MultiWithOneTrkVrt" , False)
168  myargs.setdefault("removeTrkMatSignif" , -1.) # No additional material rejection
169  myargs.setdefault("AntiPileupSigRCut" , 6.0)
170  myargs.setdefault("TrkSigCut" , 10.0)
171  myargs.setdefault("SelVrtSigCut" , 8.0)
172  myargs.setdefault("v2tIniBDTCut" ,-1.01) # BDT selection is disabled
173  myargs.setdefault("v2tFinBDTCut" ,-1.01) # BDT selection is disabled
174  myargs.setdefault("cosSVPVCut" , 0.)
175  myargs.setdefault("FastZSVCut" , 30.)
176  myargs.setdefault("VrtMassLimit" , 1000000.)
177  myargs.setdefault("Vrt2TrMassLimit", 1000000.)
178  myargs.setdefault("VertexMergeCut" , 10.)
179  myargs.setdefault("MaxSVRadiusCut" , 350.)
180  myargs.setdefault("CutD0Max" , 1000.) # Maximal track impact parameter
181  myargs.setdefault("CutD0Min" , 0.) # Minimal track impact parameter
182  myargs.setdefault("CutZVrt" , 100.)
183 
184  DVFinder = CompFactory.Rec.NewVrtSecInclusiveTool(name,**myargs)
185  acc.setPrivateTools(DVFinder)
186  mlog.info("DVFinderTool created")
187 
188  return acc
189 
190 
194 def V2TCalibrationToolCfg(flags, name="V2TCalibrationTool", **myargs):
195 
196  mlog.info("entering V2TCalibrationTool configuration")
197  acc = ComponentAccumulator()
198  acc.merge(BeamSpotCondAlgCfg(flags))
199 
200  myargs.setdefault("VertexFitterTool" , acc.popToolsAndMerge(TrkVKalVrtFitterCfg(flags)))
201  myargs.setdefault("ExtrapolatorName" , acc.popToolsAndMerge(AtlasExtrapolatorCfg(flags)))
202  myargs.setdefault("FillHist" , True)
203  myargs.setdefault("CutPt" , 400.)
204  myargs.setdefault("CutBLayHits" , 0)
205  myargs.setdefault("CutPixelHits" , 1)
206  myargs.setdefault("CutSiHits" , 8)
207  myargs.setdefault("CutTRTHits" , 10)
208  myargs.setdefault("useVertexCleaning" , False)
209  myargs.setdefault("MultiWithOneTrkVrt" , False)
210  myargs.setdefault("removeTrkMatSignif" , -1.) # No additional material rejection
211  myargs.setdefault("AntiPileupSigRCut" , 2.0)
212  myargs.setdefault("TrkSigCut" , 2.0)
213  myargs.setdefault("SelVrtSigCut" , 2.0)
214  myargs.setdefault("v2tIniBDTCut" , -1.01) #Remove BDT selection
215  myargs.setdefault("v2tFinBDTCut" , -1.01) #Remove BDT selection
216  myargs.setdefault("cosSVPVCut" , 0.)
217  myargs.setdefault("FastZSVCut" , 15.)
218  myargs.setdefault("VrtMassLimit" , 5500.)
219  myargs.setdefault("Vrt2TrMassLimit", 4000.)
220  myargs.setdefault("MaxSVRadiusCut" , 140.)
221  myargs.setdefault("CutD0Max" , 100.) # Maximal track impact parameter
222  myargs.setdefault("CutD0Min" , 0.) # Minimal track impact parameter
223  myargs.setdefault("CutZVrt" , 100.)
224 
225 
226  V2TCalibration = CompFactory.Rec.NewVrtSecInclusiveTool(name,**myargs)
227  acc.setPrivateTools(V2TCalibration)
228  mlog.info("V2TCalibrationTool created")
229 
230  return acc
231 
232 
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:16
python.AtlasExtrapolatorConfig.AtlasExtrapolatorCfg
def AtlasExtrapolatorCfg(flags, name='AtlasExtrapolator')
Definition: AtlasExtrapolatorConfig.py:63
NewVrtSecInclusiveConfig.MaterialSVFinderToolCfg
def MaterialSVFinderToolCfg(flags, name="MaterialSVFinderTool", **myargs)
Definition: NewVrtSecInclusiveConfig.py:118
python.TrkVKalVrtFitterConfig.TrkVKalVrtFitterCfg
def TrkVKalVrtFitterCfg(flags, name="TrkVKalVrtFitter", **kwargs)
Definition: TrkVKalVrtFitterConfig.py:7
NewVrtSecInclusiveConfig.DVFinderToolCfg
def DVFinderToolCfg(flags, name="DVFinderTool", **myargs)
Definition: NewVrtSecInclusiveConfig.py:153
NewVrtSecInclusiveConfig.HighPtBFinderToolCfg
def HighPtBFinderToolCfg(flags, name="HighPtBFinderTool", **myargs)
Definition: NewVrtSecInclusiveConfig.py:84
NewVrtSecInclusiveConfig.InclusiveBFinderToolCfg
def InclusiveBFinderToolCfg(flags, name="InclusiveBFinderTool", **myargs)
Definition: NewVrtSecInclusiveConfig.py:50
NewVrtSecInclusiveConfig.V2TCalibrationToolCfg
def V2TCalibrationToolCfg(flags, name="V2TCalibrationTool", **myargs)
Configuration for creation of calibration ntuples for 2-track vertex classification BDT.
Definition: NewVrtSecInclusiveConfig.py:194
Generated on Wed Jan 8 2025 21:14:52 for ATLAS Offline Software by   doxygen 1.8.18