Newer
Older
# Copyright (C) 2002-2018 CERN for the benefit of the ATLAS collaboration
# commont content for Hbb DAODs
# will impact content of HIGG5D1, HIGG5D2, HIGG5D3, HIGG2D4
def getHIGG5Common() :
return [
"egammaClusters.rawE.phi_sampl.calM",
"Muons.clusterLink.EnergyLoss.energyLossType",
"TauJets.IsTruthMatched.truthJetLink.truthParticleLink.ptDetectorAxis.etaDetectorAxis.phiDetectorAxis.mDetectorAxis",
("AntiKt4EMTopoJets.TrackWidthPt500.GhostTrackCount.Jvt.JvtJvfcorr.JvtRpt"
".JetEMScaleMomentum_pt.JetEMScaleMomentum_eta.JetEMScaleMomentum_phi.JetEMScaleMomentum_m.DetectorEta"
".DFCommonJets_Calib_pt.DFCommonJets_Calib_eta.DFCommonJets_Calib_phi.DFCommonJets_Calib_m"),
("AntiKtVR30Rmax4Rmin02TrackJets.-JetConstitScaleMomentum_pt.-JetConstitScaleMomentum_eta.-JetConstitScaleMomentum_phi.-JetConstitScaleMomentum_m"
".-constituentLinks.-constituentWeight.-ConstituentScale"),
"AntiKt10LCTopoJets.GhostVR30Rmax4Rmin02TrackJet",
("AntiKt10TrackCaloClusterTrimmedPtFrac5SmallR20Jets.pt.eta.phi.m.constituentLinks.JetConstitScaleMomentum_pt.JetConstitScaleMomentum_eta.JetConstitScaleMomentum_phi.JetConstitScaleMomentum_m"
".Angularity.Aplanarity.DetectorEta.ECF1.ECF2.ECF3.FoxWolfram0.FoxWolfram2.GhostMuonSegmentCount.GhostTrackCount.KtDR.Parent.PlanarFlow.Qw.Split12.Split23.Tau1_wta.Tau2_wta.Tau3_wta.ZCut12"),
"BTagging_AntiKtVR30Rmax4Rmin02Track.MV2c10_discriminant",
"BTagging_AntiKt4EMTopo.MV2cl100_discriminant",
"CaloCalTopoClusters.CENTER_MAG.calE.calEta.calM.calPhi.calPt.e_sampl.etaCalo.eta_sampl.phiCalo.phi_sampl.rawE.rawEta.rawM.rawPhi",
"TauChargedParticleFlowObjects.bdtPi0Score.e.eta.m.phi.pt.rapidity"
]
#Truth 3
# def getHIGG5CommonTruthContainers() :
# return ["TruthElectrons", "TruthMuons", "TruthTaus", "TruthPhotons", "TruthNeutrinos", "TruthTop", "TruthBSM", "TruthBoson"]
def getHIGG5CommonTruth() :
return [
"AntiKt4EMTopoJets.ConeTruthLabelID",
"TruthEvents.PDFID1.PDFID2.PDGID1.PDGID2.Q.X1.X2.XF1.XF2.weights.crossSection.crossSectionError.truthParticleLinks",
"TruthVertices.barcode.x.y.z.t.id.incomingParticleLinks.outgoingParticleLinks",
# "TruthParticles.px.py.pz.e.m.decayVtxLink.prodVtxLink.barcode.pdgId.status.TopHadronOriginFlag.classifierParticleOrigin.classifierParticleType.classifierParticleOutCome.dressedPhoton.polarizationTheta.polarizationPhi",
("TruthParticles.px.py.pz.e.m.decayVtxLink.prodVtxLink.barcode.pdgId.status.TopHadronOriginFlag"
".classifierParticleOrigin.classifierParticleType.classifierParticleOutCome"
".dressedPhoton.polarizationPhi.polarizationTheta"
".truthOrigin.truthParticleLink.truthType"),
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
"MuonTruthParticles.barcode.decayVtxLink.e.m.pdgId.prodVtxLink.px.py.pz.recoMuonLink.status.truthOrigin.truthParticleLink.truthType"
]
def filterContentList(pattern, content_list) :
result=[]
import re
pat=re.compile(pattern)
for elm in content_list :
head=elm.split('.',2)[0]
if pat.match(head) :
result.append(elm)
return result
# --- common thinning tools
def getTruthThinningTool(tool_prefix, thinning_helper) :
from DerivationFrameworkCore.DerivationFrameworkMaster import DerivationFrameworkIsMonteCarlo
if not DerivationFrameworkIsMonteCarlo :
return None
# MC truth thinning (not for data)
truth_cond_WZH = "((abs(TruthParticles.pdgId) >= 23) && (abs(TruthParticles.pdgId) <= 25))" # W, Z and Higgs
truth_cond_Lepton = "((abs(TruthParticles.pdgId) >= 11) && (abs(TruthParticles.pdgId) <= 16))" # Leptons
truth_cond_Quark = "((abs(TruthParticles.pdgId) == 6) || (abs(TruthParticles.pdgId) == 5))" # Top quark and Bottom quark
truth_cond_Photon = "((abs(TruthParticles.pdgId) == 22) && (TruthParticles.pt > 1*GeV))" # Photon
truth_expression = '('+truth_cond_WZH+' || '+truth_cond_Lepton +' || '+truth_cond_Quark +' || '+truth_cond_Photon+')'
from DerivationFrameworkMCTruth.DerivationFrameworkMCTruthConf import DerivationFramework__GenericTruthThinning
MCThinningTool = DerivationFramework__GenericTruthThinning(
name = tool_prefix + "MCThinningTool",
ThinningService = thinning_helper.ThinningSvc(),
ParticleSelectionString = truth_expression,
PreserveDescendants = False,
PreserveGeneratorDescendants = True,
PreserveAncestors = True)
from AthenaCommon.AppMgr import ToolSvc
ToolSvc += MCThinningTool
return MCThinningTool
def getInDetTrackParticleThinning(tool_prefix, thinning_helper, **kwargs) :
kwargs.setdefault('name', tool_prefix + 'TPThinningTool')
kwargs.setdefault('ThinningService', thinning_helper.ThinningSvc())
kwargs.setdefault('SelectionString', '( abs(InDetTrackParticles.d0) < 2 ) && ( abs(DFCommonInDetTrackZ0AtPV*sin(InDetTrackParticles.theta)) < 3 )')
kwargs.setdefault('InDetTrackParticlesKey', 'InDetTrackParticles')
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TrackParticleThinning
thinning_tool = DerivationFramework__TrackParticleThinning( name = kwargs.pop('name'), **kwargs)
from AthenaCommon.AppMgr import ToolSvc
ToolSvc += thinning_tool
return thinning_tool
def getMuonTrackParticleThinning(tool_prefix, thinning_helper) :
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__MuonTrackParticleThinning
thinning_tool = DerivationFramework__MuonTrackParticleThinning(name = tool_prefix + "MuonTPThinningTool",
ThinningService = thinning_helper.ThinningSvc(),
MuonKey = "Muons",
InDetTrackParticlesKey = "InDetTrackParticles")
from AthenaCommon.AppMgr import ToolSvc
ToolSvc += thinning_tool
return thinning_tool
def getEgammaTrackParticleThinning(tool_prefix, thinning_helper, **kwargs) :
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__EgammaTrackParticleThinning
kwargs.setdefault( 'name', tool_prefix + "ElectronTPThinningTool")
kwargs.setdefault( 'ThinningService', thinning_helper.ThinningSvc())
kwargs.setdefault( 'SGKey', 'Electrons')
kwargs.setdefault( 'InDetTrackParticlesKey','InDetTrackParticles')
kwargs.setdefault( 'BestMatchOnly', True)
thinning_tool = DerivationFramework__EgammaTrackParticleThinning(name = kwargs.pop('name'),
**kwargs)
from AthenaCommon.AppMgr import ToolSvc
ToolSvc+= thinning_tool
return thinning_tool
def getElectronTrackParticleThinning(tool_prefix, thinning_helper, **kwargs) :
kwargs.setdefault( 'name', tool_prefix + "ElectronTPThinningTool")
kwargs.setdefault( 'SGKey','Electrons')
return getEgammaTrackParticleThinning(tool_prefix, thinning_helper,**kwargs)
def getPhotonTrackParticleThinning(tool_prefix, thinning_helper, **kwargs) :
kwargs.setdefault( 'name', tool_prefix + "PhotonTPThinningTool")
kwargs.setdefault( 'SGKey','Photons')
return getEgammaTrackParticleThinning(tool_prefix, thinning_helper,**kwargs)
def getJetTrackParticleThinning(tool_prefix, thinning_helper, **kwargs) :
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__JetTrackParticleThinning
kwargs.setdefault( 'name', tool_prefix + "JetTPThinningTool")
kwargs.setdefault( 'ThinningService', thinning_helper.ThinningSvc())
kwargs.setdefault( 'JetKey', 'AntiKt4EMTopoJets')
kwargs.setdefault( 'InDetTrackParticlesKey','InDetTrackParticles')
thinning_tool = DerivationFramework__JetTrackParticleThinning(name = kwargs.pop('name'),
**kwargs)
from AthenaCommon.AppMgr import ToolSvc
ToolSvc+= thinning_tool
return thinning_tool
def getAntiKt4EMTopoTrackParticleThinning(tool_prefix, thinning_helper, **kwargs) :
kwargs.setdefault( 'name',tool_prefix + 'AntiKt4EMTopoJetTPThinningTool')
kwargs.setdefault( 'JetKey','AntiKt4EMTopoJets')
kwargs.setdefault( 'SelectionString','(AntiKt4EMTopoJets.DFCommonJets_Calib_pt > 15*GeV)')
return getJetTrackParticleThinning(tool_prefix, thinning_helper, **kwargs)
#PFlow
def getAntiKt4EMPFlowTrackParticleThinning(tool_prefix, thinning_helper, **kwargs) :
kwargs.setdefault( 'name',tool_prefix + 'AntiKt4EMPFlowJetTPThinningTool')
kwargs.setdefault( 'JetKey','AntiKt4EMPFlowJets')
kwargs.setdefault( 'SelectionString','(AntiKt4EMPFlowJets.pt > 15*GeV)')
return getJetTrackParticleThinning(tool_prefix, thinning_helper, **kwargs)
def getAntiKt10LCTopoTrackParticleThinning(tool_prefix, thinning_helper, **kwargs) :
kwargs.setdefault( 'name', tool_prefix + 'AntiKt10LCTopoJetTPThinningTool')
kwargs.setdefault( 'JetKey', 'AntiKt10LCTopoJets')
kwargs.setdefault( 'SelectionString','(AntiKt10LCTopoJets.pt > 100*GeV && abs(AntiKt10LCTopoJets.eta)<2.6)')
kwargs.setdefault( 'ApplyAnd', False)
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
return getJetTrackParticleThinning(tool_prefix, thinning_helper, **kwargs)
def getTCCTrackParticleThinning(tool_prefix, thinning_helper) :
# Tracks and CaloClusters associated with TCCs
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TCCTrackParticleThinning
thinning_tool = DerivationFramework__TCCTrackParticleThinning(name = tool_prefix + "TCCTPThinningTool",
ThinningService = thinning_helper.ThinningSvc(),
JetKey = "AntiKt10TrackCaloClusterJets",
TCCKey = "TrackCaloClustersCombinedAndNeutral",
InDetTrackParticlesKey = "InDetTrackParticles",
CaloCalTopoClustersKey = "CaloCalTopoClusters",
ThinOriginCorrectedClusters = True,
SelectionString = "AntiKt10TrackCaloClusterJets.pt>100*GeV && abs(AntiKt10TrackCaloClusterJets.eta)<2.6",
OriginCaloCalTopoClustersKey = "LCOriginTopoClusters")
from AthenaCommon.AppMgr import ToolSvc
ToolSvc+= thinning_tool
return thinning_tool
def getTauTrackParticleThinning(tool_prefix, thinning_helper) :
# Tracks associated with taus
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TauTrackParticleThinning
thinning_tool = DerivationFramework__TauTrackParticleThinning(name = tool_prefix + "TauTPThinningTool",
ThinningService = thinning_helper.ThinningSvc(),
TauKey = "TauJets",
ConeSize = 0.6,
InDetTrackParticlesKey = "InDetTrackParticles")
from AthenaCommon.AppMgr import ToolSvc
ToolSvc+= thinning_tool
return thinning_tool
def getTauCaloClusterThinning(tool_prefix, thinning_helper) :
# calo cluster thinning
from DerivationFrameworkCalo.DerivationFrameworkCaloConf import DerivationFramework__CaloClusterThinning
thinning_tool = DerivationFramework__CaloClusterThinning(name = tool_prefix + "TauCCThinningTool",
ThinningService = thinning_helper.ThinningSvc(),
SGKey = "TauJets",
TopoClCollectionSGKey = "CaloCalTopoClusters")
from AthenaCommon.AppMgr import ToolSvc
ToolSvc+= thinning_tool
return thinning_tool
def getJetCaloClusterThinning(tool_prefix, thinning_helper, **kwargs) :
if 'SGKey' not in kwargs :
raise Exception('getJetCaloClusterThinning called without providing the keyword argument SGKey')
from DerivationFrameworkCalo.DerivationFrameworkCaloConf import DerivationFramework__JetCaloClusterThinning
kwargs.setdefault('name', tool_prefix + 'FatjetCCThinningTool')
kwargs.setdefault('ThinningService', thinning_helper.ThinningSvc())
kwargs.setdefault('TopoClCollectionSGKey','CaloCalTopoClusters')
kwargs.setdefault('ApplyAnd', False)
kwargs.setdefault('AdditionalClustersKey', ["EMOriginTopoClusters","LCOriginTopoClusters","CaloCalTopoClusters"])
thinning_tool = DerivationFramework__JetCaloClusterThinning(name = kwargs.pop('name'),
AdditionalClustersKey = kwargs.pop('AdditionalClustersKey'),
**kwargs)
from AthenaCommon.AppMgr import ToolSvc
ToolSvc+= thinning_tool
return thinning_tool
def getAntiKt10LCTopoCaloClusterThinning(tool_prefix, thinning_helper, **kwargs) :
kwargs.setdefault('name', tool_prefix + 'AntiKt10LCTopoCaloClusterThinningTool')
kwargs.setdefault('SGKey', 'AntiKt10LCTopoJets')
kwargs.setdefault('SelectionString', '(AntiKt10LCTopoJets.pt > 100*GeV && abs(AntiKt10LCTopoJets.eta)<2.6)')
kwargs.setdefault('AdditionalClustersKey', ["EMOriginTopoClusters","LCOriginTopoClusters","CaloCalTopoClusters"])
return getJetCaloClusterThinning(tool_prefix,
thinning_helper,
AdditionalClustersKey = kwargs.pop('AdditionalClustersKey'),
**kwargs)
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
def getAntiKt10LCTopoTrimmedPtFrac5SmallR20Thinning(tool_prefix, thinning_helper) :
from DerivationFrameworkTools.DerivationFrameworkToolsConf import DerivationFramework__GenericObjectThinning
thinning_tool = DerivationFramework__GenericObjectThinning( name = tool_prefix + "LargeRJetThinningTool",
ThinningService = thinning_helper.ThinningSvc(),
ContainerName = "AntiKt10LCTopoTrimmedPtFrac5SmallR20Jets",
SelectionString = "(AntiKt10LCTopoTrimmedPtFrac5SmallR20Jets.pt > 100*GeV && abs(AntiKt10LCTopoTrimmedPtFrac5SmallR20Jets.eta)<2.6)",
ApplyAnd = False)
from AthenaCommon.AppMgr import ToolSvc
ToolSvc+= thinning_tool
return thinning_tool
def getAntiKt10TrackCaloClusterTrimmedPtFrac5SmallR20Thinning(tool_prefix, thinning_helper) :
from DerivationFrameworkTools.DerivationFrameworkToolsConf import DerivationFramework__GenericObjectThinning
thinning_tool =DerivationFramework__GenericObjectThinning( name = tool_prefix + "TCCJetThinningTool",
ThinningService = thinning_helper.ThinningSvc(),
ContainerName = "AntiKt10TrackCaloClusterTrimmedPtFrac5SmallR20Jets",
SelectionString = "(AntiKt10TrackCaloClusterTrimmedPtFrac5SmallR20Jets.pt > 100*GeV && abs(AntiKt10TrackCaloClusterTrimmedPtFrac5SmallR20Jets.eta)<2.6)",
ApplyAnd = False)
from AthenaCommon.AppMgr import ToolSvc
ToolSvc+= thinning_tool
return thinning_tool
def addTrimmedTruthWZJets(sequence, output_group) :
from DerivationFrameworkJetEtMiss.JetCommon import addTrimmedJets
addTrimmedJets("AntiKt", 1.0, "TruthWZ", rclus=0.2, ptfrac=0.05, mods="groomed", includePreTools=False, algseq=sequence,outputGroup=output_group)
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
def addJetOutputs(slimhelper,contentlist,smartlist=[],vetolist=[]):
'''
copy from DerivationFrameworkJetEtMiss.JetCommon, which only adds the jet to
all variables if the jet content is not already listed in the ExtraVariables
'''
outputlist = []
from AthenaCommon import Logging
dfjetlog = Logging.logging.getLogger('JetCommon')
from DerivationFrameworkJetEtMiss.JetCommon import OutputJets
for content in contentlist:
if content in OutputJets.keys():
for item in OutputJets[content]:
if item in vetolist: continue
outputlist.append(item)
else:
outputlist.append(content)
for item in outputlist:
if not slimhelper.AppendToDictionary.has_key(item):
slimhelper.AppendToDictionary[item]='xAOD::JetContainer'
slimhelper.AppendToDictionary[item+"Aux"]='xAOD::JetAuxContainer'
if item in smartlist:
dfjetlog.info( "Add smart jet collection "+item )
slimhelper.SmartCollections.append(item)
else :
head=item + '.'
add_item=True
for var in slimhelper.ExtraVariables :
if len(var) > len(head) :
print 'DEBUG HIGG5Commong.addJetOutputs %s == %s ' %(var[0:len(head)], head)
if var[0:len(head)] == head :
dfjetlog.info( "Add specialised content for jet collection "+item )
add_item=False
break
if add_item :
dfjetlog.info( "Add full jet collection "+item )
slimhelper.AllVariables.append(item)