11 from AthenaConfiguration.ComponentAccumulator
import ComponentAccumulator
12 from AthenaConfiguration.ComponentFactory
import CompFactory
13 from AthenaConfiguration.Enums
import MetadataCategory
17 """Configure the derivation framework driving algorithm (kernel) for TRIG9"""
21 augmentationTools = [ ]
24 from DerivationFrameworkPhys.PhysCommonConfig
import PhysCommonAugmentationsCfg
27 from TriggerMenuMT.TriggerAPI.TriggerAPI
import TriggerAPI
28 from TriggerMenuMT.TriggerAPI.TriggerEnums
import TriggerPeriod
31 allperiods = TriggerPeriod.future2e34
32 TriggerAPI.setConfigFlags(flags)
33 trig_all =
list(TriggerAPI.getAllHLT(allperiods).
keys())
36 TriggerListsHelper = kwargs[
'TriggerListsHelper']
37 trig_all += TriggerListsHelper.Run3TriggerNames
41 from DerivationFrameworkInDet.InDetToolsConfig
import TrackParticleThinningCfg
42 tp_thinning_expression =
"InDetTrackParticles.DFCommonTightPrimary && abs(DFCommonInDetTrackZ0AtPV)*sin(InDetTrackParticles.theta) < 3.0*mm && InDetTrackParticles.pt > 10*GeV"
45 name =
'TRIG9TrackParticleThinningTool',
46 StreamName = kwargs[
'StreamName'],
47 SelectionString = tp_thinning_expression,
48 InDetTrackParticlesKey =
"InDetTrackParticles"))
51 from DerivationFrameworkInDet.InDetToolsConfig
import MuonTrackParticleThinningCfg
54 name =
'TRIG9MuonTrackParticleThinningTool',
55 StreamName = kwargs[
'StreamName'],
57 InDetTrackParticlesKey =
"InDetTrackParticles"))
60 from DerivationFrameworkTau.TauCommonConfig
import TauThinningCfg
62 tau_thinning_expression =
"( TauJets.pt >= 20*GeV && abs(TauJets.eta) < 2.5 && abs(TauJets.charge)==1.0 && (TauJets.nTracks == 1 || TauJets.nTracks == 3) )"
66 name =
'TRIG9TauJetThinningTool',
67 StreamName = kwargs[
'StreamName'],
69 TauTracks =
"TauTracks",
70 TrackParticles =
"InDetTrackParticles",
71 TauNeutralPFOs =
"TauNeutralParticleFlowObjects",
72 TauSecondaryVertices =
"TauSecondaryVertices",
73 SelectionString = tau_thinning_expression))
77 TRIG9TrackParticleThinningTool,
78 TRIG9MuonTrackParticleThinningTool,
79 TRIG9TauJetThinningTool,
86 tauProngs13 =
"( abs(TauJets.charge)==1.0 && (TauJets.nTracks == 1 || TauJets.nTracks == 3) )"
88 e22 =
'(count( Electrons.pt > 22.0*GeV && abs(Electrons.eta) < 2.5 && Electrons.DFCommonElectronsLHLoose) >= 1)'
89 mu18 =
'(count( Muons.pt > 18.0*GeV && abs(Muons.eta) < 2.5 && Muons.DFCommonMuonPassPreselection) >= 1)'
90 tau20 =
'(count( TauJets.pt > 20.0*GeV && abs(TauJets.eta) < 2.5 && '+tauProngs13+
' && TauJets.DFTauRNNLoose ) >= 1)'
91 mutau =
'('+mu18+
' && '+tau20+
')'
92 etau =
'('+e22+
' && '+tau20+
')'
93 skim_expression =
'('+mutau+
') || ('+etau+
')'
95 EventSkimmingTool = CompFactory.DerivationFramework.xAODStringSkimmingTool(
96 name=
"EventSkimmingTool", expression=skim_expression,
98 acc.addPublicTool(EventSkimmingTool)
99 skimmingTools.append(EventSkimmingTool)
102 trig_keys = [
'mediumRNN',]
104 additional_triggers = [
105 "HLT_mu24_ivarmedium_L1MU14FCH",
106 "HLT_mu26_ivarmedium_L1MU14FCH",
107 "HLT_e26_lhtight_ivarloose_L1EM22VHI",
108 "HLT_e26_lhtight_ivarloose_L1eEM26M",
110 trig_keys += additional_triggers
111 triggers = [t
for t
in trig_all
for k
in trig_keys
if k
in t]
114 trig_veto = [
'HLT_g',
'HLT_e17_',
'HLT_e24_',
'HLT_mu14_',]
115 final_triggers = [t
for t
in triggers
for k
in trig_veto
if k
not in t]
119 print(
'TRIG9 list of triggers used for skimming:')
120 for trig
in final_triggers:
print(trig)
122 TriggerSkimmingTool = CompFactory.DerivationFramework.TriggerSkimmingTool
123 TRIG9TriggerSkimmingTool = TriggerSkimmingTool(name =
"TRIG9TriggerPreSkimmingTool",
125 TriggerListOR = final_triggers)
126 acc.addPublicTool(TRIG9TriggerSkimmingTool)
128 skimmingTools.append(TRIG9TriggerSkimmingTool)
130 DerivationKernel = CompFactory.DerivationFramework.DerivationKernel
132 SkimmingTools = skimmingTools,
133 ThinningTools = thinningTools,
134 AugmentationTools = augmentationTools))
147 from DerivationFrameworkPhys.TriggerListsHelper
import TriggerListsHelper
148 TRIG9TriggerListsHelper = TriggerListsHelper(flags)
151 acc.merge(
TRIG9KernelCfg(flags, name=
"TRIG9Kernel", StreamName =
'StreamDAOD_TRIG9', TriggerListsHelper = TRIG9TriggerListsHelper))
157 from OutputStreamAthenaPool.OutputStreamConfig
import OutputStreamCfg
158 from xAODMetaDataCnv.InfileMetaDataConfig
import SetupMetaDataForStreamCfg
159 from DerivationFrameworkCore.SlimmingHelper
import SlimmingHelper
161 TRIG9SlimmingHelper =
SlimmingHelper(
"TRIG9SlimmingHelper", NamesAndTypes = flags.Input.TypedCollections, flags = flags)
163 TRIG9SlimmingHelper.SmartCollections = [
"EventInfo",
168 "InDetTrackParticles",
169 "AntiKt4EMPFlowJets",
170 "MET_Baseline_AntiKt4EMPFlow",
171 "BTagging_AntiKt4EMPFlow",
175 TRIG9SlimmingHelper.StaticContent = [
176 "TrigRoiDescriptorCollection#HLT_eTAURoIs",
177 "TrigRoiDescriptorCollection#HLT_jTAURoIs",
178 "TrigRoiDescriptorCollection#HLT_cTAURoIs",
179 "TrigRoiDescriptorCollection#HLT_TAURoI",
180 "TrigRoiDescriptorCollection#HLT_Roi_Tau",
181 "TrigRoiDescriptorCollection#HLT_Roi_Tau_probe",
182 "TrigRoiDescriptorCollection#HLT_Roi_TauCore",
183 "TrigRoiDescriptorCollection#HLT_Roi_TauCore_probe",
184 "TrigRoiDescriptorCollection#HLT_Roi_TauIso",
185 "TrigRoiDescriptorCollection#HLT_Roi_TauIso_probe",
188 TRIG9SlimmingHelper.ExtraVariables += [
189 "TruthPrimaryVertices.t.x.y.z",
190 "PrimaryVertices.t.x.y.z.numberDoF.chiSquared.covariance.trackParticleLinks",
191 "EventInfo.hardScatterVertexLink.timeStampNSOffset",
194 TRIG9SlimmingHelper.AllVariables += [
'HLT_TrigTauRecMerged_MVA',
'HLT_tautrack_MVA']
198 from DerivationFrameworkMCTruth.MCTruthCommonConfig
import addTruth3ContentToSlimmerTool
200 TRIG9SlimmingHelper.AllVariables += [
'InTimeAntiKt4TruthJets',
'OutOfTimeAntiKt4TruthJets']
201 TRIG9SlimmingHelper.ExtraVariables += [
"Electrons.TruthLink",
205 from DerivationFrameworkMCTruth.MCTruthCommonConfig
import AddTauAndDownstreamParticlesCfg
207 TRIG9SlimmingHelper.AllVariables += [
'TruthTausWithDecayParticles',
'TruthTausWithDecayVertices']
213 TRIG9SlimmingHelper.IncludeTriggerNavigation =
True
214 TRIG9SlimmingHelper.IncludeAdditionalTriggerContent =
True
215 TRIG9SlimmingHelper.IncludeJetTriggerContent =
False
216 TRIG9SlimmingHelper.IncludeMuonTriggerContent =
False
217 TRIG9SlimmingHelper.IncludeEGammaTriggerContent =
False
218 TRIG9SlimmingHelper.IncludeTauTriggerContent =
False
219 TRIG9SlimmingHelper.IncludeEtMissTriggerContent =
False
220 TRIG9SlimmingHelper.IncludeBJetTriggerContent =
False
221 TRIG9SlimmingHelper.IncludeBPhysTriggerContent =
False
222 TRIG9SlimmingHelper.IncludeMinBiasTriggerContent =
False
226 if flags.Trigger.EDMVersion == 2:
227 from DerivationFrameworkPhys.TriggerMatchingCommonConfig
import AddRun2TriggerMatchingToSlimmingHelper
229 OutputContainerPrefix =
"TrigMatch_",
230 TriggerList = TRIG9TriggerListsHelper.Run2TriggerNamesTau)
232 OutputContainerPrefix =
"TrigMatch_",
233 TriggerList = TRIG9TriggerListsHelper.Run2TriggerNamesNoTau)
235 if flags.Trigger.EDMVersion == 3
or (flags.Trigger.EDMVersion == 2
and flags.Trigger.doEDMVersionConversion):
236 from TrigNavSlimmingMT.TrigNavSlimmingMTConfig
import AddRun3TrigNavSlimmingCollectionsToSlimmingHelper
240 TRIG9ItemList = TRIG9SlimmingHelper.GetItemList()
241 acc.merge(
OutputStreamCfg(flags,
"DAOD_TRIG9", ItemList=TRIG9ItemList, AcceptAlgs=[
"TRIG9Kernel"]))
242 acc.merge(
SetupMetaDataForStreamCfg(flags,
"DAOD_TRIG9", AcceptAlgs=[
"TRIG9Kernel"], createMetadata=[MetadataCategory.CutFlowMetaData,MetadataCategory.TruthMetaData]))