3 from AthenaConfiguration.ComponentAccumulator
import ComponentAccumulator
4 from AthenaConfiguration.ComponentFactory
import CompFactory
10 if "VertexFitterTool" not in kwargs:
11 from TrkConfig.TrkVKalVrtFitterConfig
import TrkVKalVrtFitterCfg
12 kwargs.setdefault(
"VertexFitterTool", acc.popToolsAndMerge(
TrkVKalVrtFitterCfg(flags, IterationNumber = 100, allowUltraDisplaced =
True)))
15 if "Extrapolator" not in kwargs:
16 from TrkConfig.AtlasExtrapolatorConfig
import AtlasExtrapolatorCfg
20 kwargs.setdefault(
"etaCutMSTP", 2.5)
21 kwargs.setdefault(
"baseChi2Cut", 50)
23 kwargs.setdefault(
"doValidation",
False)
25 acc.setPrivateTools(CompFactory.Rec.MuSAVtxFitterTool(name,**kwargs))
32 kwargs.setdefault(
"MuSAVtxContainerName",
"MuSAVertices")
33 kwargs.setdefault(
"MuSAExtrapolatedTracksName",
"MuSAExtrapolatedTrackParticles")
34 kwargs.setdefault(
"MuonContainerName",
"Muons")
35 kwargs.setdefault(
"MSTPContainerName",
"MuonSpectrometerTrackParticles")
37 if "TrackToVertexTool" not in kwargs:
38 from TrackToVertex.TrackToVertexConfig
import TrackToVertexCfg
39 kwargs.setdefault(
"TrackToVertexTool", acc.popToolsAndMerge(
TrackToVertexCfg(flags)))
41 if "MuSAVtxToolName" not in kwargs:
42 from MuSAVtxFitter.MuSAVtxFitterConfig
import MuSAVtxFitterToolConfig
45 acc.addEventAlgo(CompFactory.Rec.MuSAVtxFitter(name, **kwargs))
58 from TrkConfig.TrkVKalVrtFitterConfig
import TrkVKalVrtFitterCfg
61 from TrkConfig.AtlasExtrapolatorConfig
import AtlasExtrapolatorCfg
64 from InDetConfig.InDetTrackSelectorToolConfig
import InDetTrackSelectorToolCfg
66 acc.addPublicTool(trackselect)
68 from InDetConfig.InDetConversionFinderToolsConfig
import BPHY_VertexPointEstimatorCfg
70 acc.addPublicTool(vpest)
73 jpsiFinderTool = CompFactory.Analysis.JpsiFinder(
74 name =
"JPsiFinderTool",
77 TrackAndTrack =
False,
79 invMassLower = Jpsi_lo,
80 invMassUpper = Jpsi_hi,
82 oppChargesOnly =
True,
83 atLeastOneComb =
True,
84 useCombinedMeasurement =
False,
85 muonCollectionKey =
"StdWithLRTMuons",
86 TrackParticleCollection =
"InDetWithLRTTrackParticles",
88 TrkVertexFitterTool = vkalvrt,
89 TrackSelectorTool = trackselect,
90 VertexPointEstimator = vpest,
94 acc.addPublicTool(jpsiFinderTool)
95 kwargs.setdefault(
"JpsiFinderTool", jpsiFinderTool)
97 kwargs.setdefault(
"MuonContainer",
"StdWithLRTMuons")
98 kwargs.setdefault(
"EventInfo",
"EventInfo")
99 kwargs.setdefault(
"JPsiMuonContainer",
"JPsiMuons")
100 kwargs.setdefault(
"JPsiVertexContainer",
"JPsiVertices")
102 acc.addEventAlgo(CompFactory.Rec.MuSAVtxJPsiValidationAlg(name, **kwargs))
110 kwargs.setdefault(
"MuSAVtxContainerName",
"JPsiMuSAVertices")
111 kwargs.setdefault(
"MuSAExtrapolatedTracksName",
"JPsiMuSAExtrapolatedTrackParticles")
112 kwargs.setdefault(
"MuonContainerName",
"JPsiMuons")
113 kwargs.setdefault(
"MSTPContainerName",
"MuonSpectrometerTrackParticles")
115 if "TrackToVertexTool" not in kwargs:
116 from TrackToVertex.TrackToVertexConfig
import TrackToVertexCfg
117 kwargs.setdefault(
"TrackToVertexTool", acc.popToolsAndMerge(
TrackToVertexCfg(flags)))
119 if "MuSAVtxToolName" not in kwargs:
120 from MuSAVtxFitter.MuSAVtxFitterConfig
import MuSAVtxFitterToolConfig
123 acc.addEventAlgo(CompFactory.Rec.MuSAVtxFitter(name, **kwargs))
129 if __name__ ==
"__main__":
130 from AthenaConfiguration.AllConfigFlags
import initConfigFlags
131 from AthenaConfiguration.TestDefaults
import defaultTestFiles
134 flags.Concurrency.NumThreads = 1
135 flags.Concurrency.NumConcurrentEvents = 1
136 flags.Input.Files = defaultTestFiles.AOD_RUN2_MC
142 from MuonCondTest.MdtCablingTester
import setupServicesCfg
144 from DerivationFrameworkLLP.LLPToolsConfig
import LRTMuonMergerAlg
145 from AthenaConfiguration.Enums
import LHCPeriod
147 OutputMuonLocation =
"StdWithLRTMuons",
148 CreateViewCollection =
False,
150 UseRun3WP = flags.GeoModel.Run == LHCPeriod.Run3))
152 cfg.printConfig(withDetails=
True, summariseProps=
True)
155 cfg.getService(
"MessageSvc").enableSuppression =
True
158 evtMax = flags.Exec.MaxEvents
if flags.Exec.MaxEvents > 0
else None
161 if not sc.isSuccess():
163 sys.exit(
"Execution failed")