20 from AthenaConfiguration.AllConfigFlags
import initConfigFlags
24 from AthenaConfiguration.TestDefaults
import defaultTestFiles
26 flags.Exec.MaxEvents = args.maxEvents
27 if not args.inputFileList:
28 flags.Input.Files = defaultTestFiles.ESD_RUN3_MC
30 flags.Input.Files = args.inputFileList
32 from AthenaConfiguration.Enums
import ProductionStep
34 flags.Common.ProductionStep = ProductionStep.Reconstruction
36 flags.Detector.GeometryMuon =
False
37 flags.Detector.EnableAFP =
False
38 flags.Detector.EnableLucid =
False
39 flags.Detector.EnableZDC =
False
40 flags.Input.isMC =
True
43 flags.Output.AODFileName = args.outputAODFile
46 flags.Egamma.Keys.Output.CaloClusters =
"new_egammaClusters"
47 flags.Egamma.Keys.Output.Electrons =
"new_Electrons"
48 flags.Egamma.Keys.Output.Photons =
"new_Photons"
51 from AthenaConfiguration.DetectorConfigFlags
import setupDetectorFlags
54 flags,
None, use_metadata=
True, toggle_geometry=
True, keep_beampipe=
True
59 from AthenaConfiguration.MainServicesConfig
import MainServicesCfg
63 from AtlasGeoModel.GeoModelConfig
import GeoModelCfg
67 from AthenaPoolCnvSvc.PoolReadConfig
import PoolReadCfg
71 if flags.Detector.EnablePixel:
72 from PixelGeoModel.PixelGeoModelConfig
import PixelReadoutGeometryCfg
75 if flags.Detector.EnableSCT:
76 from SCT_GeoModel.SCT_GeoModelConfig
import SCT_ReadoutGeometryCfg
79 if flags.Detector.EnableTRT:
80 from TRT_GeoModel.TRT_GeoModelConfig
import TRT_ReadoutGeometryCfg
84 if flags.Detector.EnableLAr:
85 from LArBadChannelTool.LArBadChannelConfig
import LArBadFebCfg
90 from DigitizationConfig.DigitizationSteering
import DigitizationMessageSvcCfg
97 from TrkEventCnvTools.TrkEventCnvToolsConfig
import TrkEventCnvSuperToolCfg
102 from CaloRec.CaloTopoClusterConfig
import CaloTopoClusterCfg
106 from egammaAlgs.egammaTopoClusterCopierConfig
import egammaTopoClusterCopierCfg
110 from egammaAlgs.egammaRecBuilderConfig
import egammaRecBuilderCfg
114 from egammaAlgs.egammaSuperClusterBuilderConfig
import (
115 electronSuperClusterBuilderCfg,
116 photonSuperClusterBuilderCfg,
122 from egammaAlgs.xAODEgammaBuilderConfig
import xAODEgammaBuilderCfg
126 from egammaConfig.egammaOutputConfig
import egammaOutputCfg
130 from AthenaConfiguration.Utils
import setupLoggingLevels
134 if args.doCopyOriginalCollections:
135 from OutputStreamAthenaPool.OutputStreamConfig
import addToAOD
138 "xAOD::PhotonContainer#Photons",
139 "xAOD::PhotonAuxContainer#Photons"
140 f
"Aux.{flags.Egamma.Keys.Output.PhotonsSuppAOD}",
141 "xAOD::ElectronContainer#Electrons",
142 "xAOD::ElectronAuxContainer#Electrons"
143 f
"Aux.{flags.Egamma.Keys.Output.ElectronsSuppAOD}",
148 statusCode = acc.run()
152 if __name__ ==
"__main__":
156 from argparse
import ArgumentParser
158 parser = ArgumentParser(
"egammaFromESD")
164 help=
"The number of events to run. -1 runs all events.",
167 "-i",
"--inputFileList", nargs=
"*", help=
"list of input ESD files"
170 "-o",
"--outputAODFile", default=
"myAOD.pool.root", help=
"Output file name"
173 "--doCopyOriginalCollections",
176 help=
"store original electron and photon collections",
178 args = parser.parse_args()
180 statusCode =
_run(args)
181 assert statusCode
is not None,
"Issue while running"
182 sys.exit(
not statusCode.isSuccess())