10 SiTrackCollection = None,
11 ExtendedTrackCollection = None,
12 ExtendedTracksMap = None):
16 from InDetConfig.TRT_TrackExtensionAlgConfig
import TRT_TrackExtensionAlgCfg
18 InputTracksLocation = SiTrackCollection,
19 ExtendedTracksLocation = ExtendedTracksMap)
21 from InDetConfig.InDetExtensionProcessorConfig
import (
22 InDetExtensionProcessorCfg)
24 TrackName = SiTrackCollection,
25 NewTrackName = ExtendedTrackCollection,
26 ExtensionMap = ExtendedTracksMap))
28 if flags.Tracking.doTruth:
29 from InDetConfig.TrackTruthConfig
import InDetTrackTruthCfg
32 Tracks = ExtendedTrackCollection,
33 DetailedTruth = ExtendedTrackCollection+
"DetailedTruth",
34 TracksTruth = ExtendedTrackCollection+
"TruthCollection"))
40 SiTrackCollection = None,
41 ExtendedTrackCollection = None,
42 ExtendedTracksMap = None):
46 from InDetConfig.TRT_TrackExtensionAlgConfig
import (
47 TRT_Phase_TrackExtensionAlgCfg)
49 InputTracksLocation = SiTrackCollection,
50 ExtendedTracksLocation = ExtendedTracksMap)
52 from InDetConfig.InDetExtensionProcessorConfig
import (
53 InDetExtensionProcessorCfg)
55 name =
"InDetExtensionProcessorPhase",
56 TrackName = SiTrackCollection,
57 NewTrackName = ExtendedTrackCollection,
58 ExtensionMap = ExtendedTracksMap))
64 if __name__ ==
"__main__":
65 from AthenaConfiguration.AllConfigFlags
import initConfigFlags
69 flags.Concurrency.NumThreads=numThreads
70 flags.Concurrency.NumConcurrentEvents=numThreads
72 flags.Detector.GeometryPixel =
True
73 flags.Detector.GeometrySCT =
True
74 flags.Detector.GeometryTRT =
True
77 flags.Detector.EnableCalo =
False
79 flags.Tracking.doTRTExtension =
True
81 from AthenaConfiguration.TestDefaults
import defaultTestFiles
82 flags.Input.Files = defaultTestFiles.RDO_RUN2
86 from AthenaConfiguration.MainServicesConfig
import MainServicesCfg
89 from AthenaPoolCnvSvc.PoolReadConfig
import PoolReadCfg
92 from BeamSpotConditions.BeamSpotConditionsConfig
import BeamSpotCondAlgCfg
95 if "EventInfo" not in flags.Input.Collections:
96 from xAODEventInfoCnv.xAODEventInfoCnvConfig
import EventInfoCnvAlgCfg
100 from xAODTruthCnv.xAODTruthCnvConfig
import GEN_AOD2xAODCfg
103 from PixelGeoModel.PixelGeoModelConfig
import PixelReadoutGeometryCfg
104 from SCT_GeoModel.SCT_GeoModelConfig
import SCT_ReadoutGeometryCfg
108 from TRT_GeoModel.TRT_GeoModelConfig
import TRT_ReadoutGeometryCfg
111 from BeamPipeGeoModel.BeamPipeGMConfig
import BeamPipeGeometryCfg
114 InputCollections = []
116 ResolvedTracks =
'ResolvedTracks'
117 InDetSpSeededTracksKey =
'SiSPSeededTracks'
118 ExtendedTrackCollection =
'ExtendedTracks'
119 ExtendedTracksMap =
'ExtendedTracksMap'
124 from InDetConfig.TRTPreProcessing
import TRTPreProcessingCfg
128 from InDetConfig.TRTSegmentFindingConfig
import TRTSegmentFindingCfg
132 from InDetConfig.SiliconPreProcessing
import InDetRecPreProcessingSiliconCfg
136 from InDetConfig.TrackingSiPatternConfig
import TrackingSiPatternCfg
139 InputCollections = InputCollections,
140 ResolvedTrackCollectionKey = ResolvedTracks,
141 SiSPSeededTrackCollectionKey = InDetSpSeededTracksKey))
146 SiTrackCollection = ResolvedTracks,
147 ExtendedTrackCollection = ExtendedTrackCollection,
148 ExtendedTracksMap = ExtendedTracksMap))
151 iovsvc = top_acc.getService(
'IOVDbSvc')
154 top_acc.printConfig()
156 top_acc.store(
open(
"test_TRTExtensionConfig.pkl",
"wb"))