ATLAS Offline Software
Loading...
Searching...
No Matches
python.HION5 Namespace Reference

Functions

 HION5SkimmingToolCfg (flags)
 HION5Thinning (flags)
 HION5KernelCfg (flags, name="HION5Kernel", **kwargs)
 HION5Cfg (flags)

Function Documentation

◆ HION5Cfg()

python.HION5.HION5Cfg ( flags)

Definition at line 160 of file HION5.py.

160def HION5Cfg(flags):
161 acc = ComponentAccumulator()
162
163 from DerivationFrameworkEGamma.PhotonsCPDetailedContent import PhotonsCPDetailedContent
164 from DerivationFrameworkEGamma.ElectronsCPDetailedContent import ExtraElectronShowerShapes,ExtraElectronGSFVar
165
166 from DerivationFrameworkPhys.TriggerListsHelper import TriggerListsHelper
167 HION5TriggerListsHelper = TriggerListsHelper(flags)
168
169 acc.merge(HION5KernelCfg(flags, name="HION5Kernel", StreamName="StreamDAOD_HION5", TriggerListsHelper = HION5TriggerListsHelper,))
170
171 # configure slimming
172 from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
173 from xAODMetaDataCnv.InfileMetaDataConfig import SetupMetaDataForStreamCfg
174 from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
175 from DerivationFrameworkHI import ListSlimming
176
177 HION5SlimmingHelper = SlimmingHelper("HION5SlimmingHelper", NamesAndTypes = flags.Input.TypedCollections, flags = flags)
178 HION5SlimmingHelper.AppendToDictionary = {'EventInfo':'xAOD::EventInfo','EventInfoAux':'xAOD::EventAuxInfo',
179 'TruthEvents':'xAOD::TruthEventContainer','TruthEventsAux':'xAOD::TruthEventAuxContainer',
180 'MET_Truth':'xAOD::MissingETContainer','MET_TruthAux':'xAOD::MissingETAuxContainer',
181 'TruthLHEParticles':'xAOD::TruthParticleContainer', 'TruthLHEParticlesAux':'xAOD::TruthParticleAuxContainer',
182 'TruthElectrons':'xAOD::TruthParticleContainer','TruthElectronsAux':'xAOD::TruthParticleAuxContainer',
183 'TruthMuons':'xAOD::TruthParticleContainer','TruthMuonsAux':'xAOD::TruthParticleAuxContainer',
184 'TruthPhotons':'xAOD::TruthParticleContainer','TruthPhotonsAux':'xAOD::TruthParticleAuxContainer',
185 'TruthTaus':'xAOD::TruthParticleContainer','TruthTausAux':'xAOD::TruthParticleAuxContainer',
186 'TruthNeutrinos':'xAOD::TruthParticleContainer','TruthNeutrinosAux':'xAOD::TruthParticleAuxContainer',
187 'TruthBSM':'xAOD::TruthParticleContainer','TruthBSMAux':'xAOD::TruthParticleAuxContainer',
188 'TruthBoson':'xAOD::TruthParticleContainer','TruthBosonAux':'xAOD::TruthParticleAuxContainer',
189 'TruthBottom':'xAOD::TruthParticleContainer','TruthBottomAux':'xAOD::TruthParticleAuxContainer',
190 'TruthTop':'xAOD::TruthParticleContainer','TruthTopAux':'xAOD::TruthParticleAuxContainer',
191 'TruthForwardProtons':'xAOD::TruthParticleContainer','TruthForwardProtonsAux':'xAOD::TruthParticleAuxContainer',
192 'BornLeptons':'xAOD::TruthParticleContainer','BornLeptonsAux':'xAOD::TruthParticleAuxContainer',
193 'TruthBosonsWithDecayParticles':'xAOD::TruthParticleContainer','TruthBosonsWithDecayParticlesAux':'xAOD::TruthParticleAuxContainer',
194 'TruthBosonsWithDecayVertices':'xAOD::TruthVertexContainer','TruthBosonsWithDecayVerticesAux':'xAOD::TruthVertexAuxContainer',
195 'TruthBSMWithDecayParticles':'xAOD::TruthParticleContainer','TruthBSMWithDecayParticlesAux':'xAOD::TruthParticleAuxContainer',
196 'TruthBSMWithDecayVertices':'xAOD::TruthVertexContainer','TruthBSMWithDecayVerticesAux':'xAOD::TruthVertexAuxContainer',
197 'AntiKt4TruthDressedWZJets':'xAOD::JetContainer','AntiKt4TruthDressedWZJetsAux':'xAOD::JetAuxContainer',
198 'AntiKt10TruthSoftDropBeta100Zcut10Jets':'xAOD::JetContainer','AntiKt10TruthSoftDropBeta100Zcut10JetsAux':'xAOD::JetAuxContainer',
199 'MET_Track1000':'xAOD::MissingETContainer', 'MET_Track1000Aux':'xAOD::MissingETAuxContainer',
200 'MET_Track2000':'xAOD::MissingETContainer', 'MET_Track2000Aux':'xAOD::MissingETAuxContainer',
201 'MET_Track3000':'xAOD::MissingETContainer', 'MET_Track3000Aux':'xAOD::MissingETAuxContainer',
202 'MET_Track4000':'xAOD::MissingETContainer', 'MET_Track4000Aux':'xAOD::MissingETAuxContainer',
203 'MET_Track5000':'xAOD::MissingETContainer', 'MET_Track5000Aux':'xAOD::MissingETAuxContainer',
204 }
205 # Build track MET with ptCut in MeV and HItight Tracks
206 from DerivationFrameworkHI.TrackMET_config import Cfg_METTrack
207 met_ptCutList = [1000,2000,3000,4000,5000]
208
209 for ptCut in met_ptCutList:
210 acc.merge(Cfg_METTrack(flags, ptCut))
211
212 AllVariables = []
213 AllVariables += ListSlimming.HION5AllVariables(flags.Input.RunNumbers[0])
214 AllVariables += ListSlimming.HION5ExtraContainersTrigger()
215
216 if flags.Input.isMC:
217 from DerivationFrameworkMCTruth.MCTruthCommonConfig import AddStandardTruthContentsCfg
218 acc.merge(AddStandardTruthContentsCfg(flags))
219 AllVariables += ListSlimming.HION5AllTruthVariables()
220
221 HION5SlimmingHelper.SmartCollections = ListSlimming.HION5SmartCollections()
222 HION5SlimmingHelper.ExtraVariables = ListSlimming.HION5ExtraVariables()
223 HION5SlimmingHelper.ExtraVariables += PhotonsCPDetailedContent
224 HION5SlimmingHelper.ExtraVariables += ExtraElectronShowerShapes
225 HION5SlimmingHelper.ExtraVariables += ExtraElectronGSFVar
226 HION5SlimmingHelper.AllVariables = AllVariables
227
228 # Add egamma trigger objects
229 HION5SlimmingHelper.IncludeEGammaTriggerContent = True
230 # Add muon trigger objects
231 HION5SlimmingHelper.IncludeMuonTriggerContent = True
232
233 HION5ItemList = HION5SlimmingHelper.GetItemList()
234
235 acc.merge(OutputStreamCfg(flags, "DAOD_HION5", ItemList=HION5ItemList, AcceptAlgs=["HION5Kernel"]))
236 acc.merge(SetupMetaDataForStreamCfg(flags, "DAOD_HION5", AcceptAlgs=["HION5Kernel"], createMetadata=[MetadataCategory.CutFlowMetaData]))
237
238 return acc

◆ HION5KernelCfg()

python.HION5.HION5KernelCfg ( flags,
name = "HION5Kernel",
** kwargs )
Configure the derivation framework driving algorithm (kernel)
for HION5

Definition at line 86 of file HION5.py.

86def HION5KernelCfg(flags, name="HION5Kernel", **kwargs):
87 """Configure the derivation framework driving algorithm (kernel)
88 for HION5"""
89 acc = ComponentAccumulator()
90
91 # Schedule extra jets collections
92 from JetRecConfig.StandardSmallRJets import AntiKt4PV0Track
93 from JetRecConfig.JetRecConfig import JetRecCfg
94
95 jetList = [AntiKt4PV0Track]
96 for jd in jetList:
97 acc.merge(JetRecCfg(flags, jd))
98
99 # Common augmentations
100 # cannot use PhysCommon sequence because
101 # - no triggers
102 # - no TauJets
103 # so we have to use a modified version here
104 from DerivationFrameworkMuons.MuonsCommonConfig import MuonsCommonCfg
105 from DerivationFrameworkEGamma.EGammaCommonConfig import EGammaCommonCfg
106
107 acc.merge(MuonsCommonCfg(flags))
108 acc.merge(EGammaCommonCfg(flags))
109
110 # jet cleaning
111 # standard way in PhysCommon is
112 # - calculate tau ID (needed for default jet OR)
113 # - decorate jets with overlap removal
114 # - do event cleaning
115 # but taus are missing in HI derivations so need to do differently
116
117 # NO JVT criteria in HI data (see pp config for details)
118
119 # Decorate if jet passes OR and save decoration DFCommonJets_passOR
120 # Use modified OR that does not check overlaps with taus
121 from AssociationUtils.AssociationUtilsConfig import OverlapRemovalToolCfg
122
123 outputLabel = "DFCommonJets_passOR_HI"
124 bJetLabel = "" # default
125 tauLabel = "" # workaround for missing taus
126 tauKey = "" # workaround for missing taus
127 orTool = acc.popToolsAndMerge(
128 OverlapRemovalToolCfg(
129 flags, outputLabel=outputLabel, bJetLabel=bJetLabel, doTaus=False
130 )
131 )
132 algOR = CompFactory.OverlapRemovalGenUseAlg(
133 "OverlapRemovalGenUseAlg",
134 OverlapLabel=outputLabel,
135 OverlapRemovalTool=orTool,
136 JetKey = 'AntiKt4HIJets',
137 TauKey=tauKey,
138 TauLabel=tauLabel,
139 BJetLabel=bJetLabel,
140 )
141 acc.addEventAlgo(algOR)
142
143 # skimming
144 skimmingTool = acc.getPrimaryAndMerge(HION5SkimmingToolCfg(flags))
145
146 # Thinning
147 thinningTool= acc.getPrimaryAndMerge(HION5Thinning(flags))
148
149 # setup the kernel
150 acc.addEventAlgo(
151 CompFactory.DerivationFramework.DerivationKernel(
152 name,
153 SkimmingTools = [skimmingTool],
154 ThinningTools = [thinningTool],
155 AugmentationTools = [],
156 ))
157
158 return acc
159

◆ HION5SkimmingToolCfg()

python.HION5.HION5SkimmingToolCfg ( flags)
Configure the example skimming tool

Definition at line 9 of file HION5.py.

9def HION5SkimmingToolCfg(flags):
10 """Configure the example skimming tool"""
11 acc = ComponentAccumulator()
12
13 ExtraData = []
14 ExtraData += ['xAOD::MuonContainer/Muons']
15 ExtraData += ['xAOD::ElectronContainer/Electrons']
16 ExtraData += ['xAOD::PhotonContainer/Photons']
17 ExtraData += ['xAOD::TrackParticleContainer/InDetTrackParticles']
18
19 acc.addSequence( seqAND("HION5Sequence") )
20 acc.getSequence("HION5Sequence").ExtraDataForDynamicConsumers = ExtraData
21 acc.getSequence("HION5Sequence").ProcessDynamicDataDependencies = True
22
23 from DerivationFrameworkHI import ListTriggers
24
25 triggers = ListTriggers.HION5SkimmingTriggers()
26
27 req_electrons = 'count( ( Electrons.pt > 15*GeV ) && ( abs(Electrons.eta) < 2.5) )>0'
28 req_muons = 'count( Muons.DFCommonMuonPassPreselection && (Muons.pt > 15*GeV) && ( abs(Muons.eta) < 2.7))>0'
29 req_photons = 'count( Photons.DFCommonPhotonsIsEMLoose && (Photons.pt > 30*GeV) ) > 0'
30 req_total = '(' + req_electrons + ' || ' + req_muons + ' || ' + req_photons + ')'
31
32 expression = ' ( ' +' || '.join(triggers) + ' ) && ' + req_total
33
34 from DerivationFrameworkTools.DerivationFrameworkToolsConfig import (
35 xAODStringSkimmingToolCfg)
36 acc.addPublicTool(acc.getPrimaryAndMerge(xAODStringSkimmingToolCfg(
37 flags, name = "HION5StringSkimmingTool", expression = expression)), primary = True)
38
39 return acc
40

◆ HION5Thinning()

python.HION5.HION5Thinning ( flags)

Definition at line 41 of file HION5.py.

41def HION5Thinning(flags):
42 from DerivationFrameworkInDet.InDetToolsConfig import TrackParticleThinningCfg,JetTrackParticleThinningCfg
43 acc = ComponentAccumulator()
44
45 # find collision type
46 from CoolConvUtilities.ParticleTypeUtil import getTypeForRun
47 info=getTypeForRun(flags.Input.RunNumbers[0])
48 isOxygenOxygenCollision = False
49 if (info.getBeam1Type() == 8) or (info.getBeam2Type() == 8):
50 isOxygenOxygenCollision = True
51
52 pTCut = 0.9
53 if isOxygenOxygenCollision:
54 pTCut = 0.5
55
56 track_thinning_expression = f"InDetTrackParticles.pt > {pTCut}*GeV"
57 TrackParticleThinningTool = acc.getPrimaryAndMerge(TrackParticleThinningCfg(
58 flags,
59 name = "PHYSTrackParticleThinningTool",
60 StreamName = "StreamDAOD_HION5",
61 SelectionString = track_thinning_expression,
62 InDetTrackParticlesKey = "InDetTrackParticles"))
63
64 AntiKt2HIJetsThinningTool = acc.getPrimaryAndMerge(JetTrackParticleThinningCfg(
65 flags,
66 name = "AntiKt2HIJetsThinningTool",
67 StreamName = "StreamDAOD_HION5",
68 JetKey = "AntiKt2HIJets",
69 SelectionString = "AntiKt2HIJets.pt > 15*GeV",
70 InDetTrackParticlesKey = "InDetTrackParticles"))
71
72 AntiKt4HIJetsThinningTool = acc.getPrimaryAndMerge(JetTrackParticleThinningCfg(
73 flags,
74 name = "AntiKt4HIJetsThinningTool",
75 StreamName = "StreamDAOD_HION5",
76 JetKey = "AntiKt4HIJets",
77 SelectionString = "AntiKt4HIJets.pt > 15*GeV",
78 InDetTrackParticlesKey = "InDetTrackParticles"))
79
80 acc.addPublicTool(TrackParticleThinningTool,primary = True)
81 acc.addPublicTool(AntiKt2HIJetsThinningTool)
82 acc.addPublicTool(AntiKt4HIJetsThinningTool)
83
84 return acc
85