13 from AthenaConfiguration.Enums
import LHCPeriod
14 from JetRecConfig.StandardJetConstits
import inputsFromContext
16 from AthenaCommon
import Logging
17 jetcaliblog = Logging.logging.getLogger(
'JetCalibToolsConfig')
19 all = [
'getJetCalibTool']
25 "T0":(
"JES_MC15cRecommendation_PFlow_Aug2016_rel21.config",
"00-04-77",
"JetArea_Residual_EtaJES"),
27 "AnalysisLatest":(
"PreRec_R22_PFlow_ResPU_EtaJES_GSC_February23_230215.config",
"00-04-82",
"JetArea_Residual_EtaJES_GSC_Insitu"),
28 "TrigLS2":(
"JES_MC16Recommendation_Consolidated_PFlow_Apr2019_Rel21_Trigger.config",
"00-04-82",
"JetArea_Residual_EtaJES_GSC_Insitu"),
29 "Trigger":(
"JES_MC16Recommendation_Consolidated_PFlow_30May2022_Rel22_Trigger.config",
"00-04-82",
"JetArea_Residual_EtaJES_GSC_Insitu"),
30 "TrigR22Prerec":(
"PreRec_R22_PFlow_ResPU_EtaJES_GSC_February23_230215.config",
"00-04-82",
"JetArea_Residual_EtaJES_GSC_Insitu"),
31 "TrigHIUPC" : (
"JES_MC16Recommendation_LowMu1718_MCJES_GSC_nTrkOn_PFlow_Sep2023_Rel21.config",
"00-04-82",
"EtaJES_GSC"),
35 "T0":(
"JES_MC15cRecommendation_May2016_rel21.config",
"00-04-77",
"JetArea_Residual_EtaJES"),
37 "AnalysisLatest":(
"JES_MC16Recommendation_Consolidated_EMTopo_Apr2019_Rel21.config",
"00-04-82",
"JetArea_Residual_EtaJES_GSC_Insitu"),
38 "TrigRun2":(
"JES_MC15cRecommendation_May2016_Trigger.config",
"00-04-77",
"JetArea_EtaJES_GSC_Insitu"),
39 "TrigRun2GSC":(
"JES_data2016_data2015_Recommendation_Dec2016_rel21.config",
"00-04-77",
"JetArea_EtaJES_GSC_Insitu"),
40 "TrigLS2":(
"JES_MC16Recommendation_Consolidated_EMTopo_Apr2019_Rel21_Trigger.config",
"00-04-82",
"JetArea_Residual_EtaJES_GSC_Insitu"),
41 "Trigger":(
"JES_MC16Recommendation_Consolidated_EMTopo_Apr2019_Rel21_Trigger.config",
"00-04-82",
"JetArea_Residual_EtaJES_GSC_Insitu"),
42 "HLLHC":(
"HLLHC/JES_MC16_HLLHC_Aug2021_rel21.config",
"00-04-82",
"JetArea_Residual_EtaJES"),
46 "T0":(
"Consolidated_R22_CSSKUFO_ResPU_EtaJES_GNNC_20231103.config",
"00-04-83",
"Residual_EtaJES_GNNC"),
50 "RScanLatest":(
"JES_MC16Recommendation_Rscan2LC_22Feb2018_rel21.config",
"00-04-81",
"JetArea_Residual_EtaJES_GSC_Insitu")
54 "RScanLatest":(
"JES_MC16Recommendation_Rscan6LC_22Feb2018_rel21.config",
"00-04-81",
"JetArea_Residual_EtaJES_GSC_Insitu")
58 "CombinedMass": (
"JES_MC16recommendation_FatJet_Trimmed_JMS_comb_17Oct2018.config",
"00-04-82",
"EtaJES_JMS"),
59 "CaloMass": (
"JES_MC16recommendation_FatJet_Trimmed_JMS_calo_12Oct2018.config",
"00-04-82",
"EtaJES_JMS"),
60 "TAMass": (
"JES_MC16recommendation_FatJet_Trimmed_JMS_TA_12Oct2018.config",
"00-04-82",
"EtaJES_JMS"),
61 "SoftDrop": (
"JES_MC20PreRecommendation_R10_UFO_CSSK_SoftDrop_JMS_R21Insitu_10Mar2023.config",
"00-04-82",
"EtaJES_JMS"),
62 "TrigUngroomed": (
"JES_Full2012dataset_Rscan_June2014.config",
"00-04-77",
"JetArea_EtaJES"),
63 "TrigTrimmed": (
"JES_MC15recommendation_FatJet_June2015_PtFrac4.config",
"00-04-82",
"EtaJES_JMS"),
64 "TrigSoftDrop": (
"JES_MC16recommendation_R10_UFO_CSSK_SoftDrop_JMS_01April2020.config",
"00-04-82",
"EtaJES_JMS"),
65 "TrigHLTSoftDrop":(
"JES_JMS_MC21_R10_PFlow_CSSK_SoftDrop_Dec2023_Trigger.config",
"00-04-83",
"EtaJES_JMS"),
66 "LargeRDNN": (
"JES_JMS_MC20dnnc_R10_UFO_CSSK_SoftDrop_Jun2023.config",
"00-04-83",
"LargeRDNN"),
71 "AntiKt4EMPFlow":
"JES_MC16Recommendation_AFII_PFlow_Apr2019_Rel21.config",
72 "AntiKt4GPFlow":
"JES_MC16Recommendation_AFII_PFlow_Apr2019_Rel21.config",
73 "AntiKt4EMTopo":
"JES_MC16Recommendation_AFII_EMTopo_Apr2019_Rel21.config",
74 "AntiKt4LCTopo":
"JES_MC16Recommendation_AFII_EMTopo_Apr2019_Rel21.config",
79 "AntiKt4EMPFlow":pflowcontexts,
80 "AntiKt4EMPFlow_noElectrons":pflowcontexts,
81 "AntiKt4EMPFlow_noMuons":pflowcontexts,
82 "AntiKt4EMPFlow_noLeptons":pflowcontexts,
83 "AntiKt4EMPFlow_tauSeedEleRM":pflowcontexts,
84 "AntiKt4EMPFlowByVertex":pflowcontexts,
85 "AntiKt4GPFlow":pflowcontexts,
86 "AntiKt4UFOCSSK":ufocontexts,
87 "AntiKt4EMTopo":topocontexts,
88 "AntiKt4LCTopo":topocontexts,
89 "AntiKt10LCTopo":fatjetcontexts,
91 "AntiKt10LCTopoTrimmedPtFrac5SmallR20":fatjetcontexts,
92 "AntiKt10LCTopoTrimmedPtFrac4SmallR20":fatjetcontexts,
94 "AntiKt10UFOCSSKSoftDropBeta100Zcut10":fatjetcontexts,
96 "AntiKt10UFOCSSK_noElectronsSoftDropBeta100Zcut10":fatjetcontexts,
97 "AntiKt10UFOCSSK_noMuonsSoftDropBeta100Zcut10":fatjetcontexts,
98 "AntiKt10UFOCSSK_noLeptonsSoftDropBeta100Zcut10":fatjetcontexts,
100 "AntiKt10EMPFlowCSSKSoftDropBeta100Zcut10":fatjetcontexts,
102 "AntiKt2LCTopo":rscanlc2,
103 "AntiKt6LCTopo":rscanlc6,
107 hasInSitu = [
"AntiKt4LCTopo",
"AntiKt4EMTopo",
"AntiKt4EMPFlow",
"AntiKt4EMPFlow_noElectrons",
"AntiKt4EMPFlow_noMuons",
"AntiKt4EMPFlow_noLeptons",
"AntiKt4EMPFlow_tauSeedEleRM",
"TrigAntiKt4EMTopo"]
113 def getJetCalibTool(jetdef, context, data_type, calibseq = "", rhoname = "", pvname = "PrimaryVertices", gscdepth = "auto"):
115 jetcollection = jetdef.basename
118 if data_type
not in [
'data',
'mc',
'afii']:
119 jetcaliblog.error(
"JetCalibConfig accepts data_type values: 'data', 'mc', 'afii'")
120 raise ValueError(
"Unsupported data_type provided: '{0}".
format(data_type))
122 if jetcollection.endswith(
"Jets"):
123 jetcaliblog.error(
"Jet collection '{0}'should be specified w/o 'Jets' in the name.".
format(jetcollection))
124 raise ValueError(
"Bad jet collection formatting in getJetCalibTool.")
126 jetcaliblog.debug(
"Preparing calibration for {0}, in context {1} on sample type {2}".
format(jetcollection,context,data_type))
128 jetcontexts = calibcontexts[jetcollection]
130 if jetcollection ==
"AntiKt4EMTopo" and context ==
"T0":
131 if jetdef._cflags.GeoModel.Run >= LHCPeriod.Run4:
135 configfile, calibarea, calibseq_def = jetcontexts[context]
136 _calibseq = calibseq
if calibseq
else calibseq_def
141 if _calibseq.endswith(
"Insitu"):
142 if data_type ==
'data':
143 if jetcollection
not in hasInSitu:
144 raise ValueError(
"In situ calibration does not exist for {0}, context {1}".
format(jetcollection,context))
146 raise ValueError(
"In situ calibration requested for MC on {0}, context {1}".
format(jetcollection,context))
147 _configfile = configfile
148 if data_type ==
"afii" and jetcollection
in af2configs.keys():
149 _configfile = af2configs[jetcollection]
150 _data_type = data_type
157 if "Residual" in _calibseq
or "GSC" in _calibseq
and gscdepth!=
"EM3" or "LargeRDNN" in _calibseq:
160 if context ==
"TrigHIUPC":
163 _jetcollection = jetcollection
164 if "PFlow" in jetcollection
and (context==
"TrigSoftDrop" or context==
"TrigHLTSoftDrop"):
165 _jetcollection = jetcollection.replace(
"EMPFlow",
"UFO")
166 if "_noElectrons" in jetcollection :
167 _jetcollection = _jetcollection.replace(
"_noElectrons",
"")
168 if "_noMuons" in jetcollection :
169 _jetcollection = _jetcollection.replace(
"_noMuons",
"")
170 if "_noLeptons" in jetcollection :
171 _jetcollection = _jetcollection.replace(
"_noLeptons",
"")
172 if "_tauSeedEleRM" in jetcollection :
173 _jetcollection = _jetcollection.replace(
"_tauSeedEleRM",
"")
175 if "ByVertex" in jetcollection:
176 _jetcollection = jetcollection.replace(
"ByVertex",
"")
177 return defineJetCalibTool(_jetcollection, context, _configfile, calibarea, _calibseq, _data_type, rhoname, _pvname, gscdepth)
178 except KeyError
as e:
179 jetcaliblog.error(
"Context '{0}' not found for jet collection '{1}'".
format(context,jetcollection))
180 jetcaliblog.error(
"Options are '{0}".
format(
','.
join(jetcontexts.keys())))
185 def defineJetCalibTool(jetcollection, context, configfile, calibarea, calibseq, data_type, rhoname, pvname, gscdepth):
187 calibseqshort =
''.
join([ step[0]
for step
in calibseq.split(
'_') ])
188 toolname =
"jetcalib_{0}_{1}_{2}".
format(jetcollection,calibseqshort,context)
190 from AthenaConfiguration.ComponentFactory
import CompFactory
191 jct = CompFactory.JetCalibrationTool(toolname,
192 JetCollection = jetcollection,
193 ConfigFile = configfile,
194 CalibArea = calibarea,
195 CalibSequence = calibseq,
196 IsData = (data_type ==
"data"),
198 PrimaryVerticesContainerName = pvname,
208 cfg, calibarea, calibseq = calibcontexts[jetdef.basename][calibcontext]
213 prereqs.append(
"mod:ConstitFourMom")
214 if "JetArea" in calibseq:
215 if calibcontext.startswith(
"Trig"): prereqs.append(
"input:HLT_EventDensity")
216 elif pvname ==
"PrimaryVertices_initial": prereqs.append(
"input:EventDensityCustomVtxGNN")
217 elif pvname !=
"PrimaryVertices": prereqs.append(
"input:EventDensityCustomVtx")
219 if "GSC" in calibseq:
220 prereqs += [
"mod:CaloEnergies"]
221 if calibcontext !=
"TrigRun2":
222 prereqs += [
"mod:TrackMoments",
224 if "GNNC" in calibseq:
225 prereqs += [
"mod:CaloQuality",
"mod:TrackMoments"]
226 if "CombinedMass" in calibcontext:
227 prereqs += [
"mod:TrackSumMoments"]
228 if "LargeRDNN" in calibseq:
229 prereqs += [
"mod:CaloEnergiesLargeR",
"mod:ConstitFrac",
"mod:groomMRatio",
"mod:Width",
230 "mod:nsubjettiness",
"mod:nsubjettinessR",
"mod:ktsplitter",
"mod:ecorr",
231 "mod:ecorrR",
"mod:qw"]
232 jetcaliblog.debug(
"Prereqs for calibseq '{0}': {1}".
format(calibseq,
str(prereqs)))
239 rhoname =
"EleRM_Kt4LCTopoOriginEventShape" if jetdef.context ==
"EleRM" else "auto"
240 pvname =
"PrimaryVertices"
242 calibspecs = modspec.split(
':')
244 calibcontext, data_type = calibspecs[:2]
245 if len(calibspecs)>2: calibseq = calibspecs[2]
246 if len(calibspecs)>3: rhoname = calibspecs[3]
247 if len(calibspecs)>4: pvname = calibspecs[4]
248 if len(calibspecs)>5: gscdepth = calibspecs[5]
250 return calibcontext, data_type, calibseq, rhoname, pvname, gscdepth
255 return getJetCalibTool(jetdef,calibcontext,data_type,calibseq,rhoname,pvname,gscdepth)