ATLAS Offline Software
TrkVertexFitterUtilsConfig.py
Go to the documentation of this file.
1 # Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
2 # Configuration of TrkVertexFitterUtils package
3 from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
4 from AthenaConfiguration.ComponentFactory import CompFactory
5 
6 def FullLinearizedTrackFactoryCfg(flags, name='FullLinearizedTrkFactory', **kwargs):
7  from MagFieldServices.MagFieldServicesConfig import (
8  AtlasFieldCacheCondAlgCfg)
9  acc = AtlasFieldCacheCondAlgCfg(flags) # To produce AtlasFieldCacheCondObj
10 
11  if 'Extrapolator' not in kwargs:
12  from TrkConfig.AtlasExtrapolatorConfig import InDetExtrapolatorCfg
13  kwargs.setdefault("Extrapolator", acc.popToolsAndMerge(
14  InDetExtrapolatorCfg(flags)))
15 
16  acc.setPrivateTools(
17  CompFactory.Trk.FullLinearizedTrackFactory(name, **kwargs))
18  return acc
19 
20 def AtlasFullLinearizedTrackFactoryCfg(flags, name='AtlasFullLinearizedTrkFactory', **kwargs):
21  from MagFieldServices.MagFieldServicesConfig import (
22  AtlasFieldCacheCondAlgCfg)
23  acc = AtlasFieldCacheCondAlgCfg(flags) # To produce AtlasFieldCacheCondObj
24 
25  if 'Extrapolator' not in kwargs:
26  from TrkConfig.AtlasExtrapolatorConfig import AtlasExtrapolatorCfg
27  kwargs.setdefault("Extrapolator", acc.popToolsAndMerge(
28  AtlasExtrapolatorCfg(flags)))
29 
30  acc.setPrivateTools(
31  CompFactory.Trk.FullLinearizedTrackFactory(name, **kwargs))
32  return acc
33 
34 def DetAnnealingMakerCfg(flags, name='DetAnnealingMaker', **kwargs):
35  acc = ComponentAccumulator()
36  kwargs.setdefault("SetOfTemperatures", [ 64., 16., 4., 2., 1.5, 1. ])
37  acc.setPrivateTools(CompFactory.Trk.DetAnnealingMaker(name, **kwargs))
38  return acc
39 
40 def SecVtxDetAnnealingMakerCfg(flags, name='SecVtxDetAnnealingMaker', **kwargs):
41  kwargs.setdefault("SetOfTemperatures", [ 1.0 ])
42  return DetAnnealingMakerCfg(flags, name, **kwargs)
43 
44 def TauDetAnnealingMakerCfg(flags, name='tauRec_TauDetAnnealingMaker', **kwargs):
45  kwargs.setdefault("SetOfTemperatures", [ 64, 32, 16, 8, 4, 2, 1 ])
46  return DetAnnealingMakerCfg(flags, name, **kwargs)
47 
48 def TrackToVertexIPEstimatorCfg(flags, name='TrackToVertexIPEstimator', **kwargs):
49  acc = ComponentAccumulator()
50 
51  if 'Extrapolator' not in kwargs:
52  from TrkConfig.AtlasExtrapolatorConfig import InDetExtrapolatorCfg
53  kwargs.setdefault("Extrapolator", acc.popToolsAndMerge(
54  InDetExtrapolatorCfg(flags)))
55 
56  if 'LinearizedTrackFactory' not in kwargs:
57  kwargs.setdefault("LinearizedTrackFactory", acc.popToolsAndMerge(
59 
60  acc.setPrivateTools(CompFactory.Trk.TrackToVertexIPEstimator(name, **kwargs))
61  return acc
62 
63 def AtlasTrackToVertexIPEstimatorCfg(flags, name='AtlasTrackToVertexIPEstimator', **kwargs):
64  acc = ComponentAccumulator()
65 
66  if 'Extrapolator' not in kwargs:
67  from TrkConfig.AtlasExtrapolatorConfig import AtlasExtrapolatorCfg
68  kwargs.setdefault("Extrapolator", acc.popToolsAndMerge(
69  AtlasExtrapolatorCfg(flags)))
70 
71  if 'LinearizedTrackFactory' not in kwargs:
72  kwargs.setdefault("LinearizedTrackFactory", acc.popToolsAndMerge(
74 
75  acc.setPrivateTools(CompFactory.Trk.TrackToVertexIPEstimator(name, **kwargs))
76  return acc
77 
78 def ImpactPoint3dEstimatorCfg(flags, name='ImpactPoint3dEstimator', **kwargs):
79  from MagFieldServices.MagFieldServicesConfig import (
80  AtlasFieldCacheCondAlgCfg)
81  acc = AtlasFieldCacheCondAlgCfg(flags) # To produce AtlasFieldCacheCondObj
82 
83  if "Extrapolator" not in kwargs:
84  from TrkConfig.AtlasExtrapolatorConfig import InDetExtrapolatorCfg
85  Extrapolator = acc.popToolsAndMerge(InDetExtrapolatorCfg(flags))
86  acc.addPublicTool(Extrapolator)
87  kwargs.setdefault("Extrapolator", Extrapolator)
88 
89  acc.setPrivateTools(CompFactory.Trk.ImpactPoint3dEstimator(name, **kwargs))
90  return acc
91 
92 def AtlasImpactPoint3dEstimatorCfg(flags, name='AtlasImpactPoint3dEstimator', **kwargs):
93  acc = ComponentAccumulator()
94 
95  if "Extrapolator" not in kwargs:
96  from TrkConfig.AtlasExtrapolatorConfig import AtlasExtrapolatorCfg
97  Extrapolator = acc.popToolsAndMerge(AtlasExtrapolatorCfg(flags))
98  acc.addPublicTool(Extrapolator)
99  kwargs.setdefault("Extrapolator", Extrapolator)
100 
101  acc.setPrivateTools(acc.popToolsAndMerge(
102  ImpactPoint3dEstimatorCfg(flags, name, **kwargs)))
103  return acc
104 
105 def KalmanVertexUpdatorCfg(flags, name='KalmanVertexUpdator', **kwargs):
106 
107  acc = ComponentAccumulator()
108  acc.setPrivateTools(CompFactory.Trk.KalmanVertexUpdator(name, **kwargs))
109  return acc
110 
111 def KalmanVertexTrackUpdatorCfg(flags, name='KalmanVertexTrackUpdator', **kwargs):
112  acc = ComponentAccumulator()
113  acc.setPrivateTools(CompFactory.Trk.KalmanVertexTrackUpdator(name, **kwargs))
114  return acc
python.JetAnalysisCommon.ComponentAccumulator
ComponentAccumulator
Definition: JetAnalysisCommon.py:302
python.TrkVertexFitterUtilsConfig.AtlasTrackToVertexIPEstimatorCfg
def AtlasTrackToVertexIPEstimatorCfg(flags, name='AtlasTrackToVertexIPEstimator', **kwargs)
Definition: TrkVertexFitterUtilsConfig.py:63
python.TrkVertexFitterUtilsConfig.SecVtxDetAnnealingMakerCfg
def SecVtxDetAnnealingMakerCfg(flags, name='SecVtxDetAnnealingMaker', **kwargs)
Definition: TrkVertexFitterUtilsConfig.py:40
python.TrkVertexFitterUtilsConfig.ImpactPoint3dEstimatorCfg
def ImpactPoint3dEstimatorCfg(flags, name='ImpactPoint3dEstimator', **kwargs)
Definition: TrkVertexFitterUtilsConfig.py:78
python.TrkVertexFitterUtilsConfig.TrackToVertexIPEstimatorCfg
def TrackToVertexIPEstimatorCfg(flags, name='TrackToVertexIPEstimator', **kwargs)
Definition: TrkVertexFitterUtilsConfig.py:48
python.TrkVertexFitterUtilsConfig.KalmanVertexTrackUpdatorCfg
def KalmanVertexTrackUpdatorCfg(flags, name='KalmanVertexTrackUpdator', **kwargs)
Definition: TrkVertexFitterUtilsConfig.py:111
python.TrkVertexFitterUtilsConfig.KalmanVertexUpdatorCfg
def KalmanVertexUpdatorCfg(flags, name='KalmanVertexUpdator', **kwargs)
Definition: TrkVertexFitterUtilsConfig.py:105
python.AtlasExtrapolatorConfig.AtlasExtrapolatorCfg
def AtlasExtrapolatorCfg(flags, name='AtlasExtrapolator')
Definition: AtlasExtrapolatorConfig.py:63
python.TrkVertexFitterUtilsConfig.DetAnnealingMakerCfg
def DetAnnealingMakerCfg(flags, name='DetAnnealingMaker', **kwargs)
Definition: TrkVertexFitterUtilsConfig.py:34
python.TrkVertexFitterUtilsConfig.AtlasFullLinearizedTrackFactoryCfg
def AtlasFullLinearizedTrackFactoryCfg(flags, name='AtlasFullLinearizedTrkFactory', **kwargs)
Definition: TrkVertexFitterUtilsConfig.py:20
python.TrkVertexFitterUtilsConfig.TauDetAnnealingMakerCfg
def TauDetAnnealingMakerCfg(flags, name='tauRec_TauDetAnnealingMaker', **kwargs)
Definition: TrkVertexFitterUtilsConfig.py:44
python.TrkVertexFitterUtilsConfig.AtlasImpactPoint3dEstimatorCfg
def AtlasImpactPoint3dEstimatorCfg(flags, name='AtlasImpactPoint3dEstimator', **kwargs)
Definition: TrkVertexFitterUtilsConfig.py:92
python.TrkVertexFitterUtilsConfig.FullLinearizedTrackFactoryCfg
def FullLinearizedTrackFactoryCfg(flags, name='FullLinearizedTrkFactory', **kwargs)
Definition: TrkVertexFitterUtilsConfig.py:6
python.MagFieldServicesConfig.AtlasFieldCacheCondAlgCfg
def AtlasFieldCacheCondAlgCfg(flags, **kwargs)
Definition: MagFieldServicesConfig.py:8
python.AtlasExtrapolatorConfig.InDetExtrapolatorCfg
def InDetExtrapolatorCfg(flags, name='InDetExtrapolator', **kwargs)
Definition: AtlasExtrapolatorConfig.py:315