7 from AthenaConfiguration.ComponentAccumulator
import ComponentAccumulator
8 from AthenaConfiguration.Enums
import Format, BeamType
13 if not flags.Detector.EnableTRT:
16 if flags.Input.Format
is Format.BS
or 'TRT_RDOs' in flags.Input.Collections:
21 if flags.Beam.Type
is BeamType.Cosmics:
22 from InDetConfig.InDetPrepRawDataFormationConfig
import (
23 InDetTRT_NoTime_RIO_MakerCfg)
26 from InDetConfig.InDetPrepRawDataFormationConfig
import (
27 InDetTRT_RIO_MakerCfg)
30 if flags.InDet.doSplitReco:
31 from InDetConfig.InDetPrepRawDataFormationConfig
import (
32 InDetTRT_RIO_MakerPUCfg)
38 if flags.InDet.doTRTGlobalOccupancy:
39 from InDetConfig.TRT_ElectronPidToolsConfig
import (
40 TRTOccupancyIncludeCfg)
46 if flags.InDet.doTruth
and (
47 flags.Beam.Type
is not BeamType.Cosmics
and
48 'PRD_MultiTruthTRT' not in flags.Input.Collections):
49 from InDetConfig.InDetTruthAlgsConfig
import (
50 InDetPRD_MultiTruthMakerTRTCfg)
52 if flags.InDet.doSplitReco:
53 from InDetConfig.InDetTruthAlgsConfig
import (
54 InDetPRD_MultiTruthMakerTRTPUCfg)
59 if __name__ ==
"__main__":
60 from AthenaConfiguration.AllConfigFlags
import initConfigFlags
63 from AthenaConfiguration.TestDefaults
import defaultTestFiles
64 flags.Input.Files = defaultTestFiles.RDO_RUN2
69 flags.Concurrency.NumThreads = numThreads
71 flags.Concurrency.NumConcurrentEvents = numThreads
76 from AthenaConfiguration.MainServicesConfig
import MainServicesCfg
79 from AthenaPoolCnvSvc.PoolReadConfig
import PoolReadCfg
82 from TRT_GeoModel.TRT_GeoModelConfig
import TRT_ReadoutGeometryCfg
85 from PixelGeoModel.PixelGeoModelConfig
import PixelReadoutGeometryCfg
86 from SCT_GeoModel.SCT_GeoModelConfig
import SCT_ReadoutGeometryCfg
92 iovsvc = top_acc.getService(
'IOVDbSvc')
93 iovsvc.OutputLevel = 5
95 top_acc.store(
open(
"test_TRTPrepocessing.pkl",
"wb"))