3 from AthenaConfiguration.AccumulatorCache
import AccumulatorCache
4 from AthenaConfiguration.ComponentAccumulator
import ComponentAccumulator
5 from AthenaConfiguration.ComponentFactory
import CompFactory
6 from AthenaConfiguration.Enums
import LHCPeriod
13 materialTagBase =
'AtlasLayerMat_v'
23 atlasMaterialTag = materialTagBase +
str(version) + sub_version
24 cfolder = CoolDataBaseFolder +
'<tag>TagInfoMajor/' + \
25 atlasMaterialTag +
'_/GeoAtlas</tag>'
27 if flags.GeoModel.Run < LHCPeriod.Run4:
29 from IOVDbSvc.IOVDbSvcConfig
import addFoldersSplitOnline
31 flags,
'GLOBAL', [cfolder], [cfolder],
32 splitMC=
True, className=
'Trk::LayerMaterialMap'))
36 atlasMaterialTag = flags.ITk.trackingGeometry.materialTag + \
37 str(flags.ITk.trackingGeometry.version)
38 if flags.ITk.trackingGeometry.loadLocalDbForMaterialMaps:
39 DataBaseName = flags.ITk.trackingGeometry.localDatabaseName
40 from IOVDbSvc.IOVDbSvcConfig
import addFolders
42 "/GLOBAL/TrackingGeo/LayerMaterialITK",
45 className=
'Trk::LayerMaterialMap'))
47 materialFileTag = atlasMaterialTag +
'_'+ \
48 flags.GeoModel.AtlasVersion
49 from IOVDbSvc.IOVDbSvcConfig
import addFolders
51 "/GLOBAL/TrackingGeo/LayerMaterialITK",
55 className=
"Trk::LayerMaterialMap"))
63 kwargs.setdefault(
"WorldDimension", [])
64 kwargs.setdefault(
"WorldMaterialProperties", [])
66 if "TrackingVolumeArrayCreator" not in kwargs:
67 from TrackingGeometryCondAlg.TrkDetDescrToolsConfig
import (
68 TrackingVolumeArrayCreatorCfg)
69 TrackingVolumeArrayCreator = result.popToolsAndMerge(
71 result.addPublicTool(TrackingVolumeArrayCreator)
72 kwargs.setdefault(
"TrackingVolumeArrayCreator",
73 TrackingVolumeArrayCreator)
75 if "TrackingVolumeHelper" not in kwargs:
76 from TrackingGeometryCondAlg.TrkDetDescrToolsConfig
import (
77 TrackingVolumeHelperCfg)
78 TrackingVolumeHelper = result.popToolsAndMerge(
80 result.addPublicTool(TrackingVolumeHelper)
81 kwargs.setdefault(
"TrackingVolumeHelper", TrackingVolumeHelper)
85 if flags.Detector.GeometryID:
86 from TrackingGeometryCondAlg.InDetTrackingGeometryConfig
import (
87 InDetTrackingGeometryBuilderCfg)
89 "InDetTrackingGeometryBuilder", result.popToolsAndMerge(
92 elif flags.Detector.GeometryITk:
93 from TrackingGeometryCondAlg.InDetTrackingGeometryConfig
import (
94 ITkTrackingGeometryBuilderCfg)
96 "InDetTrackingGeometryBuilder", result.popToolsAndMerge(
100 if flags.Detector.GeometryHGTD
and useCond:
101 from TrackingGeometryCondAlg.HGTD_TrackingGeometryConfig
import (
102 HGTD_TrackingGeometryBuilderCfg)
104 "HGTD_TrackingGeometryBuilder", result.popToolsAndMerge(
107 if flags.Detector.GeometryCalo:
108 from TrackingGeometryCondAlg.CaloTrackingGeometryConfig
import (
109 CaloTrackingGeometryBuilderCfg)
111 "CaloTrackingGeometryBuilder", result.popToolsAndMerge(
114 if flags.Detector.GeometryMuon:
115 from TrackingGeometryCondAlg.MuonTrackingGeometryConfig
import (
116 MuonTrackingGeometryBuilderCfg)
118 "MuonTrackingGeometryBuilder", result.popToolsAndMerge(
121 geometryBuilder = CompFactory.Trk.GeometryBuilderCond(name, **kwargs) \
123 CompFactory.Trk.GeometryBuilder(name, **kwargs)
124 result.setPrivateTools(geometryBuilder)
129 doMaterialValidation=False,
132 Sets up the Tracking Geometry Conditions Algorithm
136 kwargs.setdefault(
"GeometryBuilder", result.popToolsAndMerge(
140 atlas_geometry_processors = []
142 if flags.TrackingGeometry.MaterialSource ==
'COOL':
143 if flags.GeoModel.Run < LHCPeriod.Run4:
144 CoolDataBaseFolder =
'/GLOBAL/TrackingGeo/LayerMaterialV2'
146 CoolDataBaseFolder =
'/GLOBAL/TrackingGeo/LayerMaterialITK'
148 from TrackingGeometryCondAlg.TrkDetDescrToolsConfig
import (
149 LayerMaterialProviderCfg)
150 atlas_geometry_processors += [ result.popToolsAndMerge(
152 LayerMaterialMapName = CoolDataBaseFolder,
153 LayerMaterialMapKey = CoolDataBaseFolder)) ]
158 elif flags.TrackingGeometry.MaterialSource ==
'Input':
159 from TrackingGeometryCondAlg.TrkDetDescrToolsConfig
import (
160 InputLayerMaterialProviderCfg)
161 atlas_geometry_processors += [ result.popToolsAndMerge(
164 if doMaterialValidation:
165 from TrackingGeometryCondAlg.TrkDetDescrToolsConfig
import (
166 LayerMaterialInspectorCfg)
167 atlas_geometry_processors += [ result.popToolsAndMerge(
170 from GaudiKernel.GaudiHandles
import PrivateToolHandleArray
171 kwargs.setdefault(
"GeometryProcessors", PrivateToolHandleArray(
172 atlas_geometry_processors))
173 kwargs.setdefault(
"TrackingGeometryWriteKey",
'AtlasTrackingGeometry')
175 result.addCondAlgo(CompFactory.Trk.TrackingGeometryCondAlg(name, **kwargs),
181 if __name__ ==
'__main__':
182 from AthenaConfiguration.AllConfigFlags
import initConfigFlags
183 from AthenaConfiguration.TestDefaults
import defaultTestFiles
185 flags.Input.Files = defaultTestFiles.RAW_RUN2
190 f =
open(
'TrackingGeometryCondAlgCfg.pkl',
'wb')