10 from AthenaConfiguration.ComponentAccumulator
import ComponentAccumulator
11 from AthenaConfiguration.ComponentFactory
import CompFactory
12 from AthenaConfiguration.Enums
import MetadataCategory
13 from AthenaCommon.Logging
import logging
14 logPHYS = logging.getLogger(
'PHYS')
18 """Configure the derivation framework driving algorithm (kernel) for PHYS"""
21 from TrkConfig.VertexFindingFlags
import VertexSortingSetup
22 if flags.Tracking.PriVertex.sortingSetup
is VertexSortingSetup.GNNSorting:
23 from DerivationFrameworkPhys.GNNVertexConfig
import GNNVertexCfg
27 from DerivationFrameworkPhys.PhysCommonConfig
import PhysCommonAugmentationsCfg
30 TriggerListsHelper = kwargs[
'TriggerListsHelper']
35 nametag = name.replace(
'Kernel',
'')
37 'TrackParticleThinningToolName' : nametag+
"TrackParticleThinningTool",
38 'MuonTPThinningToolName' : nametag+
"MuonTPThinningTool",
39 'TauJetThinningToolName' : nametag+
"TauJetThinningTool",
40 'TauJets_MuonRMThinningToolName' : nametag+
"TauJets_MuonRMThinningTool",
41 'DiTauThinningToolName' : nametag+
"DiTauThinningTool",
42 'DiTauTPThinningToolName' : nametag+
"DiTauTPThinningTool",
43 'DiTauLowPtThinningToolName' : nametag+
"DiTauLowPtThinningTool",
44 'DiTauLowPtTPThinningToolName' : nametag+
"DiTauLowPtTPThinningTool",
47 if flags.Tau.TauEleRM_isAvailable:
48 thinningToolsArgs[
'TauJets_EleRMThinningToolName'] = nametag+
"TauJets_EleRMThinningTool"
50 from DerivationFrameworkPhys.PhysCommonThinningConfig
import PhysCommonThinningCfg
54 for key
in thinningToolsArgs:
55 thinningTools.append(acc.getPublicTool(thinningToolsArgs[key]))
58 DerivationKernel = CompFactory.DerivationFramework.DerivationKernel
63 def PHYSCoreCfg(flags, name_tag='PHYS', StreamName='StreamDAOD_PHYS', TriggerListsHelper=None, addExtraVariables=None):
65 if TriggerListsHelper
is None:
66 from DerivationFrameworkPhys.TriggerListsHelper
import TriggerListsHelper
67 TriggerListsHelper = TriggerListsHelper(flags)
72 from DerivationFrameworkHiggs.HiggsPhysContent
import HiggsAugmentationAlgsCfg
77 from IsolationSelection.IsolationSelectionConfig
import IsoCloseByAlgsCfg
81 from LeptonTaggers.LeptonTaggersConfig
import DecoratePLITAlgsCfg
87 from DerivationFrameworkMCTruth.HFClassificationCommonConfig
import HFClassificationCommonCfg
93 from OutputStreamAthenaPool.OutputStreamConfig
import OutputStreamCfg
94 from xAODMetaDataCnv.InfileMetaDataConfig
import SetupMetaDataForStreamCfg
95 from DerivationFrameworkCore.SlimmingHelper
import SlimmingHelper
97 PHYSSlimmingHelper =
SlimmingHelper(name_tag+
"SlimmingHelper", NamesAndTypes = flags.Input.TypedCollections, flags = flags)
98 PHYSSlimmingHelper.SmartCollections = [
"EventInfo",
103 "InDetTrackParticles",
105 "AntiKt4EMPFlowJets",
106 "BTagging_AntiKt4EMPFlow",
107 "AntiKt4EMPFlowJets_FTAG",
108 "MET_Baseline_AntiKt4EMTopo",
109 "MET_Baseline_AntiKt4EMPFlow",
114 "AntiKt10UFOCSSKSoftDropBeta100Zcut10Jets",
115 "AntiKtVR30Rmax4Rmin02PV0TrackJets",
117 if flags.Tau.TauEleRM_isAvailable:
118 PHYSSlimmingHelper.SmartCollections.append(
"TauJets_EleRM")
120 excludedVertexAuxData =
"-vxTrackAtVertex.-MvfFitInfo.-isInitialized.-VTAV"
122 StaticContent += [
"xAOD::VertexContainer#SoftBVrtClusterTool_Tight_Vertices"]
123 StaticContent += [
"xAOD::VertexAuxContainer#SoftBVrtClusterTool_Tight_VerticesAux." + excludedVertexAuxData]
124 StaticContent += [
"xAOD::VertexContainer#SoftBVrtClusterTool_Medium_Vertices"]
125 StaticContent += [
"xAOD::VertexAuxContainer#SoftBVrtClusterTool_Medium_VerticesAux." + excludedVertexAuxData]
126 StaticContent += [
"xAOD::VertexContainer#SoftBVrtClusterTool_Loose_Vertices"]
127 StaticContent += [
"xAOD::VertexAuxContainer#SoftBVrtClusterTool_Loose_VerticesAux." + excludedVertexAuxData]
129 PHYSSlimmingHelper.StaticContent = StaticContent
132 PHYSSlimmingHelper.ExtraVariables += [
"AntiKt4EMTopoJets.DFCommonJets_QGTagger_truthjet_nCharged.DFCommonJets_QGTagger_truthjet_pt.DFCommonJets_QGTagger_truthjet_eta.DFCommonJets_QGTagger_NTracks.DFCommonJets_QGTagger_TracksWidth.DFCommonJets_QGTagger_TracksC1.ConeExclBHadronsFinal.ConeExclCHadronsFinal.GhostBHadronsFinal.GhostCHadronsFinal.GhostBHadronsFinalCount.GhostBHadronsFinalPt.GhostCHadronsFinalCount.GhostCHadronsFinalPt.IsoFixedCone5PtPUsub",
133 "AntiKt4EMPFlowJets.QGTransformer_ConstScore.DFCommonJets_QGTagger_truthjet_nCharged.DFCommonJets_QGTagger_truthjet_pt.DFCommonJets_QGTagger_truthjet_eta.DFCommonJets_QGTagger_NTracks.DFCommonJets_QGTagger_TracksWidth.DFCommonJets_QGTagger_TracksC1.ConeExclBHadronsFinal.ConeExclCHadronsFinal.GhostBHadronsFinal.GhostCHadronsFinal.GhostBHadronsFinalCount.GhostBHadronsFinalPt.GhostCHadronsFinalCount.GhostCHadronsFinalPt.GhostPartons.isJvtHS.isJvtPU.IsoFixedCone5PtPUsub",
134 "TruthPrimaryVertices.t.x.y.z",
135 "InDetTrackParticles.TTVA_AMVFVertices.TTVA_AMVFWeights.eProbabilityHT.numberOfTRTHits.numberOfTRTOutliers",
136 "EventInfo.GenFiltHT.GenFiltMET.GenFiltHTinclNu.GenFiltPTZ.GenFiltFatJ.HF_Classification.HF_SimpleClassification",
137 "TauJets.dRmax.etOverPtLeadTrk",
138 "TauJets_MuonRM.dRmax.etOverPtLeadTrk",
139 "HLT_xAOD__TrigMissingETContainer_TrigEFMissingET.ex.ey",
140 "HLT_xAOD__TrigMissingETContainer_TrigEFMissingET_mht.ex.ey"]
142 if addExtraVariables:
143 PHYSSlimmingHelper.ExtraVariables += addExtraVariables
145 if flags.Tau.TauEleRM_isAvailable:
146 PHYSSlimmingHelper.ExtraVariables += [
"TauJets_EleRM.dRmax.etOverPtLeadTrk"]
149 from LeptonTaggers.LeptonTaggersConfig
import GetExtraPLITVariablesForDxAOD
153 PHYSSlimmingHelper.ExtraVariables += [
"AntiKt10UFOCSSKSoftDropBeta100Zcut10Jets.TopTransformer_ConstScore"]
158 from DerivationFrameworkMCTruth.MCTruthCommonConfig
import addTruth3ContentToSlimmerTool
160 PHYSSlimmingHelper.AllVariables += [
'TruthLHEParticles',
'TruthHFWithDecayParticles',
'TruthHFWithDecayVertices',
'TruthCharm',
'TruthPileupParticles',
'InTimeAntiKt4TruthJets',
'OutOfTimeAntiKt4TruthJets']
161 PHYSSlimmingHelper.ExtraVariables += [
"Electrons.TruthLink",
164 "AntiKt4TruthDressedWZJets.IsoFixedCone5Pt.HFHadronOriginID"]
166 from DerivationFrameworkMCTruth.MCTruthCommonConfig
import AddTauAndDownstreamParticlesCfg
168 PHYSSlimmingHelper.AllVariables += [
'TruthTausWithDecayParticles',
'TruthTausWithDecayVertices']
171 from DerivationFrameworkHiggs.HiggsPhysContent
import setupHiggsSlimmingVariables
175 PHYSSlimmingHelper.AllVariables += [
'AFPSiHitContainer',
'AFPToFHitContainer' ]
178 PHYSSlimmingHelper.IncludeTriggerNavigation =
False
179 PHYSSlimmingHelper.IncludeJetTriggerContent =
False
180 PHYSSlimmingHelper.IncludeMuonTriggerContent =
False
181 PHYSSlimmingHelper.IncludeEGammaTriggerContent =
False
182 PHYSSlimmingHelper.IncludeTauTriggerContent =
False
183 PHYSSlimmingHelper.IncludeEtMissTriggerContent =
False
184 PHYSSlimmingHelper.IncludeBJetTriggerContent =
False
185 PHYSSlimmingHelper.IncludeBPhysTriggerContent =
False
186 PHYSSlimmingHelper.IncludeMinBiasTriggerContent =
False
188 PHYSSlimmingHelper.IncludeBJetTriggerByYearContent =
True
192 if flags.Trigger.EDMVersion == 2:
193 from DerivationFrameworkPhys.TriggerMatchingCommonConfig
import AddRun2TriggerMatchingToSlimmingHelper
195 OutputContainerPrefix =
"TrigMatch_",
196 TriggerList = TriggerListsHelper.Run2TriggerNamesTau)
198 OutputContainerPrefix =
"TrigMatch_",
199 TriggerList = TriggerListsHelper.Run2TriggerNamesNoTau)
201 if flags.Trigger.EDMVersion == 3
or (flags.Trigger.EDMVersion == 2
and flags.Trigger.doEDMVersionConversion):
202 from TrigNavSlimmingMT.TrigNavSlimmingMTConfig
import AddRun3TrigNavSlimmingCollectionsToSlimmingHelper
206 from DerivationFrameworkPhys.TriggerMatchingCommonConfig
import getDataYear
209 from DerivationFrameworkPhys.TriggerMatchingCommonConfig
import AddjFexRoIsToSlimmingHelper
213 from DerivationFrameworkPhys.TriggerMatchingCommonConfig
import AddLegacyL1JetRoIsToSlimmingHelper
217 PHYSItemList = PHYSSlimmingHelper.GetItemList()
218 acc.merge(
OutputStreamCfg(flags,
"DAOD_"+name_tag, ItemList=PHYSItemList, AcceptAlgs=[name_tag+
"Kernel"]))
219 acc.merge(
SetupMetaDataForStreamCfg(flags,
"DAOD_"+name_tag, AcceptAlgs=[name_tag+
"Kernel"], createMetadata=[MetadataCategory.CutFlowMetaData, MetadataCategory.TruthMetaData]))
225 logPHYS.info(
'****************** STARTING PHYS *****************')
227 stream_name =
'StreamDAOD_PHYS'
234 from DerivationFrameworkPhys.TriggerListsHelper
import TriggerListsHelper
235 PHYSTriggerListsHelper = TriggerListsHelper(flags)
241 StreamName = stream_name,
242 TriggerListsHelper = PHYSTriggerListsHelper
248 StreamName = stream_name,
249 TriggerListsHelper = PHYSTriggerListsHelper