From 9e3902b04e893d81c203e89d5472410a46181d13 Mon Sep 17 00:00:00 2001
From: Jon Burr <jon.burr@cern.ch>
Date: Wed, 24 Nov 2021 11:10:50 +0100
Subject: [PATCH] Add AODFULL EDM output to DAOD_PHYSVAL
---
.../share/PHYSVAL.py | 3 ++
.../python/TrigSlimmingHelper.py | 37 +++++++++++++++++++
2 files changed, 40 insertions(+)
create mode 100644 PhysicsAnalysis/DerivationFramework/DerivationFrameworkTrigger/python/TrigSlimmingHelper.py
diff --git a/PhysicsAnalysis/DerivationFramework/DerivationFrameworkPhysicsValidation/share/PHYSVAL.py b/PhysicsAnalysis/DerivationFramework/DerivationFrameworkPhysicsValidation/share/PHYSVAL.py
index 140fe72f349..0d8a46caf52 100644
--- a/PhysicsAnalysis/DerivationFramework/DerivationFrameworkPhysicsValidation/share/PHYSVAL.py
+++ b/PhysicsAnalysis/DerivationFramework/DerivationFrameworkPhysicsValidation/share/PHYSVAL.py
@@ -25,6 +25,7 @@ from DerivationFrameworkJetEtMiss.METCommon import scheduleStandardMETContent
from TriggerMenuMT.TriggerAPI.TriggerAPI import TriggerAPI
from TriggerMenuMT.TriggerAPI.TriggerEnums import TriggerPeriod, TriggerType
from DerivationFrameworkTrigger.TriggerMatchingHelper import TriggerMatchingHelper
+from DerivationFrameworkTrigger.TrigSlimmingHelper import addTrigEDMSetToOutput
from TrkDetDescrSvc.AtlasTrackingGeometrySvc import AtlasTrackingGeometrySvc
#====================================================================
@@ -269,6 +270,8 @@ PHYSVALSlimmingHelper.IncludeEtMissTriggerContent = True
PHYSVALSlimmingHelper.IncludeBJetTriggerContent = True
PHYSVALSlimmingHelper.IncludeBPHYSVALTriggerContent = True
PHYSVALSlimmingHelper.IncludeMinBiasTriggerContent = True
+addTrigEDMSetToOutput(PHYSVALSlimmingHelper, "AODFULL")
+
# Add the jet containers to the stream (defined in JetCommon if import needed)
#addJetOutputs(PHYSVALSlimmingHelper,["PHYSVAL"])
diff --git a/PhysicsAnalysis/DerivationFramework/DerivationFrameworkTrigger/python/TrigSlimmingHelper.py b/PhysicsAnalysis/DerivationFramework/DerivationFrameworkTrigger/python/TrigSlimmingHelper.py
new file mode 100644
index 00000000000..587a638d54c
--- /dev/null
+++ b/PhysicsAnalysis/DerivationFramework/DerivationFrameworkTrigger/python/TrigSlimmingHelper.py
@@ -0,0 +1,37 @@
+# Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
+
+"""Helper functions for adding trigger EDM content to a derivation"""
+
+from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
+from AthenaConfiguration.AllConfigFlags import ConfigFlags
+from TrigEDMConfig.TriggerEDM import getTriggerEDMList
+
+
+def addTrigEDMSetToOutput(helper: SlimmingHelper, edmSet: str, edmVersion: int = None):
+ """Add a full trigger EDM set to the output slimming helper"""
+
+ # NB: I'm not sure that importing and using the ConfigFlags is a pattern that we want to
+ # encourage, but given that the derivations aren't built with CA's yet this is the best I can
+ # do...
+ if edmVersion is None:
+ edmVersion = ConfigFlags.Trigger.EDMVersion
+
+ edmList = getTriggerEDMList(edmSet, edmVersion)
+ # This list is a mapping from container type to a list of required container names
+ # This includes the Aux containers and their lists of aux variables.
+ # The SlimmingHelper however requires the list of *interface* (non-Aux) containers with
+ # the dynamic aux variables specified
+ # Collate that information here
+ interface_containers = []
+ for cont_type, cont_list in edmList.items():
+ for container in cont_list:
+ # For the next part we need to know all the container names and the associated aux items
+ # If we can assume the standard relation between interface and aux names (which is
+ # probably safe) then we just need to look at the aux names for this.
+ interface_name, aux, auxitems = container.partition("Aux.")
+ if aux:
+ # If the string doesn't contain 'Aux.' then aux will be the empty string
+ # i.e. we only enter this block if we're really looking at an aux container
+ interface_containers.append(f"{interface_name}.{auxitems}")
+
+ helper.ExtraVariables += interface_containers
--
GitLab