diff --git a/Control/AthenaConfiguration/python/Enums.py b/Control/AthenaConfiguration/python/Enums.py index 083e7d01e2b53ba67e24bb556d53da57dbe51ffd..0a24c9483071ac33960aa86c6bc1a5528db706af 100644 --- a/Control/AthenaConfiguration/python/Enums.py +++ b/Control/AthenaConfiguration/python/Enums.py @@ -50,6 +50,7 @@ class ProductionStep(FlagEnum): Overlay = 'Overlay' FastChain = 'FastChain' Digitization = 'Digitization' + PileUpPretracking = 'PileUpPretracking' Reconstruction = 'Reconstruction' Derivation = 'Derivation' diff --git a/Database/AthenaPOOL/OutputStreamAthenaPool/python/OutputStreamConfig.py b/Database/AthenaPOOL/OutputStreamAthenaPool/python/OutputStreamConfig.py index 9e3f14f7f6310fac808671718ae97632ecf0aaf9..11d1e06df3276b55cdd074e30aad52ed0ec1d3ff 100644 --- a/Database/AthenaPOOL/OutputStreamAthenaPool/python/OutputStreamConfig.py +++ b/Database/AthenaPOOL/OutputStreamAthenaPool/python/OutputStreamConfig.py @@ -13,7 +13,7 @@ def OutputStreamCfg(flags, streamName, ItemList=[], MetadataItemList=[], disableEventTag=False, trigNavThinningSvc=None, takeItemsFromInput=False, AcceptAlgs=[], HelperTools=[]): eventInfoKey = "EventInfo" - if flags.Common.ProductionStep == ProductionStep.PileUpPresampling: + if flags.Common.ProductionStep in [ProductionStep.PileUpPresampling, ProductionStep.PileUpPretracking]: eventInfoKey = f"{flags.Overlay.BkgPrefix}EventInfo" msg = logging.getLogger("OutputStreamCfg") diff --git a/Event/xAOD/xAODMetaDataCnv/python/InfileMetaDataConfig.py b/Event/xAOD/xAODMetaDataCnv/python/InfileMetaDataConfig.py index b50cc62fc20af9039e05f48fd760e4a6a2db64c9..1ce83ae80d5d2f62fced30dd2c96cd461f5ea8b2 100644 --- a/Event/xAOD/xAODMetaDataCnv/python/InfileMetaDataConfig.py +++ b/Event/xAOD/xAODMetaDataCnv/python/InfileMetaDataConfig.py @@ -125,7 +125,7 @@ def propagateMetaData(flags, streamName="", category=None, *args, **kwargs): Key=outputStreamName(streamName), DataHeaderKey=outputStreamName(streamName), EventInfoKey=f"{flags.Overlay.BkgPrefix}EventInfo" - if flags.Common.ProductionStep == ProductionStep.PileUpPresampling + if flags.Common.ProductionStep in [ProductionStep.PileUpPresampling, ProductionStep.PileUpPretracking] else "EventInfo", ) tools.mdItems += [