1 import FWCore.ParameterSet.Config
as cms
5 from math
import ceil,log
8 photon_id_modules_WorkingPoints_nanoAOD = cms.PSet(
10 'RecoEgamma.PhotonIdentification.Identification.cutBasedPhotonID_Fall17_94X_V1_TrueVtx_cff',
11 'RecoEgamma.PhotonIdentification.Identification.cutBasedPhotonID_Fall17_94X_V2_cff',
12 'RecoEgamma.PhotonIdentification.Identification.mvaPhotonID_Fall17_94X_V1p1_cff',
13 'RecoEgamma.PhotonIdentification.Identification.mvaPhotonID_Fall17_94X_V2_cff',
14 'RecoEgamma.PhotonIdentification.Identification.cutBasedPhotonID_Spring16_V2p2_cff',
15 'RecoEgamma.PhotonIdentification.Identification.mvaPhotonID_Spring16_nonTrig_V1_cff',
17 WorkingPoints = cms.vstring(
18 "egmPhotonIDs:cutBasedPhotonID-Fall17-94X-V2-loose",
19 "egmPhotonIDs:cutBasedPhotonID-Fall17-94X-V2-medium",
20 "egmPhotonIDs:cutBasedPhotonID-Fall17-94X-V2-tight",
23 photon_id_modules_WorkingPoints_nanoAOD_Spring16V2p2 = cms.PSet(
24 modules = photon_id_modules_WorkingPoints_nanoAOD.modules,
25 WorkingPoints = cms.vstring(
26 "egmPhotonIDs:cutBasedPhotonID-Spring16-V2p2-loose",
27 "egmPhotonIDs:cutBasedPhotonID-Spring16-V2p2-medium",
28 "egmPhotonIDs:cutBasedPhotonID-Spring16-V2p2-tight",
33 pset = id_modules_working_points_pset
35 for modname
in pset.modules:
36 ids = __import__(modname, globals(), locals(), [
'idName',
'cutFlow'])
38 _id = getattr(ids,name)
39 if hasattr(_id,
'idName')
and hasattr(_id,
'cutFlow'):
40 if (len(pset.WorkingPoints)>0
and _id.idName == pset.WorkingPoints[0].
split(
':')[-1]):
41 cut_names =
','.
join([cut.cutName.value()
for cut
in _id.cutFlow])
42 n_bits_per_cut =
int(
ceil(
log(len(pset.WorkingPoints)+1,2)))
43 return 'VID compressed bitmap (%s), %d bits per cut'%(cut_names, n_bits_per_cut)
44 raise ValueError(
"Something is wrong in the photon ID modules parameter set!")
46 bitmapVIDForPho = cms.EDProducer(
"PhoVIDNestedWPBitmapProducer",
47 src = cms.InputTag(
"slimmedPhotons"),
48 WorkingPoints = photon_id_modules_WorkingPoints_nanoAOD.WorkingPoints,
51 bitmapVIDForPhoSpring16V2p2 = cms.EDProducer(
"PhoVIDNestedWPBitmapProducer",
52 src = cms.InputTag(
"slimmedPhotons"),
53 WorkingPoints = photon_id_modules_WorkingPoints_nanoAOD_Spring16V2p2.WorkingPoints,
56 isoForPho = cms.EDProducer(
"PhoIsoValueMapProducer",
57 src = cms.InputTag(
"slimmedPhotons"),
58 relative = cms.bool(
False),
59 rho_PFIso = cms.InputTag(
"fixedGridRhoFastjetAll"),
60 mapIsoChg = cms.InputTag(
"photonIDValueMapProducer:phoChargedIsolation"),
61 mapIsoNeu = cms.InputTag(
"photonIDValueMapProducer:phoNeutralHadronIsolation"),
62 mapIsoPho = cms.InputTag(
"photonIDValueMapProducer:phoPhotonIsolation"),
63 EAFile_PFIso_Chg = cms.FileInPath(
"RecoEgamma/PhotonIdentification/data/Fall17/effAreaPhotons_cone03_pfChargedHadrons_90percentBased_V2.txt"),
64 EAFile_PFIso_Neu = cms.FileInPath(
"RecoEgamma/PhotonIdentification/data/Fall17/effAreaPhotons_cone03_pfNeutralHadrons_90percentBased_V2.txt"),
65 EAFile_PFIso_Pho = cms.FileInPath(
"RecoEgamma/PhotonIdentification/data/Fall17/effAreaPhotons_cone03_pfPhotons_90percentBased_V2.txt"),
67 for modifier
in run2_miniAOD_80XLegacy, run2_nanoAOD_94X2016:
68 modifier.toModify(isoForPho,
69 EAFile_PFIso_Chg = cms.FileInPath(
"RecoEgamma/PhotonIdentification/data/Spring16/effAreaPhotons_cone03_pfChargedHadrons_90percentBased.txt"),
70 EAFile_PFIso_Neu = cms.FileInPath(
"RecoEgamma/PhotonIdentification/data/Spring16/effAreaPhotons_cone03_pfNeutralHadrons_90percentBased.txt"),
71 EAFile_PFIso_Pho = cms.FileInPath(
"RecoEgamma/PhotonIdentification/data/Spring16/effAreaPhotons_cone03_pfPhotons_90percentBased.txt"),
74 seedGainPho = cms.EDProducer(
"PhotonSeedGainProducer", src = cms.InputTag(
"slimmedPhotons"))
76 import RecoEgamma.EgammaTools.calibratedEgammas_cff
78 calibratedPatPhotonsNano = RecoEgamma.EgammaTools.calibratedEgammas_cff.calibratedPatPhotons.clone(
79 produceCalibratedObjs =
False,
80 correctionFile = cms.string(
"EgammaAnalysis/ElectronTools/data/ScalesSmearings/Run2016_UltraLegacy_preVFP_RunFineEtaR9Gain"),
83 (run2_egamma_2016 & tracker_apv_vfp30_2016).toModify(calibratedPatPhotonsNano,
84 correctionFile = cms.string(
"EgammaAnalysis/ElectronTools/data/ScalesSmearings/Run2016_UltraLegacy_preVFP_RunFineEtaR9Gain")
87 (run2_egamma_2016 & ~tracker_apv_vfp30_2016).toModify(calibratedPatPhotonsNano,
88 correctionFile = cms.string(
"EgammaAnalysis/ElectronTools/data/ScalesSmearings/Run2016_UltraLegacy_postVFP_RunFineEtaR9Gain"),
91 run2_egamma_2017.toModify(calibratedPatPhotonsNano,
92 correctionFile = cms.string(
"EgammaAnalysis/ElectronTools/data/ScalesSmearings/Run2017_24Feb2020_runEtaR9Gain_v2")
95 run2_egamma_2018.toModify(calibratedPatPhotonsNano,
96 correctionFile = cms.string(
"EgammaAnalysis/ElectronTools/data/ScalesSmearings/Run2018_29Sep2020_RunFineEtaR9Gain")
99 run2_nanoAOD_102Xv1.toModify(calibratedPatPhotonsNano,
100 correctionFile = cms.string(
"EgammaAnalysis/ElectronTools/data/ScalesSmearings/Run2018_Step2Closure_CoarseEtaR9Gain_v2")
103 run2_nanoAOD_94XMiniAODv1.toModify(calibratedPatPhotonsNano,
104 correctionFile = cms.string(
"EgammaAnalysis/ElectronTools/data/ScalesSmearings/Run2017_17Nov2017_v1_ele_unc")
107 run2_nanoAOD_94XMiniAODv2.toModify(calibratedPatPhotonsNano,
108 correctionFile = cms.string(
"EgammaAnalysis/ElectronTools/data/ScalesSmearings/Run2017_17Nov2017_v1_ele_unc")
112 run2_miniAOD_80XLegacy.toModify(calibratedPatPhotonsNano,
113 correctionFile = cms.string(
"EgammaAnalysis/ElectronTools/data/ScalesSmearings/Legacy2016_07Aug2017_FineEtaR9_v3_ele_unc")
117 slimmedPhotonsWithUserData = cms.EDProducer(
"PATPhotonUserDataEmbedder",
118 src = cms.InputTag(
"slimmedPhotons"),
119 userFloats = cms.PSet(
120 mvaID = cms.InputTag(
"photonMVAValueMapProducer:PhotonMVAEstimatorRunIIFall17v2Values"),
121 mvaID_Fall17V1p1 = cms.InputTag(
"photonMVAValueMapProducer:PhotonMVAEstimatorRunIIFall17v1p1Values"),
122 mvaID_Spring16nonTrigV1 = cms.InputTag(
"photonMVAValueMapProducer:PhotonMVAEstimatorRun2Spring16NonTrigV1Values"),
123 PFIsoChg = cms.InputTag(
"isoForPho:PFIsoChg"),
124 PFIsoAll = cms.InputTag(
"isoForPho:PFIsoAll"),
126 userIntFromBools = cms.PSet(
127 cutbasedID_loose = cms.InputTag(
"egmPhotonIDs:cutBasedPhotonID-Fall17-94X-V2-loose"),
128 cutbasedID_medium = cms.InputTag(
"egmPhotonIDs:cutBasedPhotonID-Fall17-94X-V2-medium"),
129 cutbasedID_tight = cms.InputTag(
"egmPhotonIDs:cutBasedPhotonID-Fall17-94X-V2-tight"),
130 mvaID_WP90 = cms.InputTag(
"egmPhotonIDs:mvaPhoID-RunIIFall17-v2-wp90"),
131 mvaID_WP80 = cms.InputTag(
"egmPhotonIDs:mvaPhoID-RunIIFall17-v2-wp80"),
132 cutbasedIDV1_loose = cms.InputTag(
"egmPhotonIDs:cutBasedPhotonID-Fall17-94X-V1-loose"),
133 cutbasedIDV1_medium = cms.InputTag(
"egmPhotonIDs:cutBasedPhotonID-Fall17-94X-V1-medium"),
134 cutbasedIDV1_tight = cms.InputTag(
"egmPhotonIDs:cutBasedPhotonID-Fall17-94X-V1-tight"),
135 cutID_Spring16_loose = cms.InputTag(
"egmPhotonIDs:cutBasedPhotonID-Spring16-V2p2-loose"),
136 cutID_Spring16_medium = cms.InputTag(
"egmPhotonIDs:cutBasedPhotonID-Spring16-V2p2-medium"),
137 cutID_Spring16_tight = cms.InputTag(
"egmPhotonIDs:cutBasedPhotonID-Spring16-V2p2-tight"),
138 mvaID_Spring16nonTrigV1_WP90 = cms.InputTag(
"egmPhotonIDs:mvaPhoID-Spring16-nonTrig-V1-wp90"),
139 mvaID_Spring16nonTrigV1_WP80 = cms.InputTag(
"egmPhotonIDs:mvaPhoID-Spring16-nonTrig-V1-wp80"),
142 VIDNestedWPBitmap = cms.InputTag(
"bitmapVIDForPho"),
143 seedGain = cms.InputTag(
"seedGainPho"),
148 for modifier
in run2_egamma_2016, run2_egamma_2017, run2_egamma_2018:
149 modifier.toModify(slimmedPhotonsWithUserData.userFloats,
150 ecalEnergyErrPostCorrNew = cms.InputTag(
"calibratedPatPhotonsNano",
"ecalEnergyErrPostCorr"),
151 ecalEnergyPreCorrNew = cms.InputTag(
"calibratedPatPhotonsNano",
"ecalEnergyPreCorr"),
152 ecalEnergyPostCorrNew = cms.InputTag(
"calibratedPatPhotonsNano",
"ecalEnergyPostCorr"),
153 energyScaleUpNew = cms.InputTag(
"calibratedPatPhotonsNano",
"energyScaleUp"),
154 energyScaleDownNew = cms.InputTag(
"calibratedPatPhotonsNano",
"energyScaleDown"),
155 energySigmaUpNew = cms.InputTag(
"calibratedPatPhotonsNano",
"energySigmaUp"),
156 energySigmaDownNew = cms.InputTag(
"calibratedPatPhotonsNano",
"energySigmaDown"),
160 run2_nanoAOD_94X2016.toModify(slimmedPhotonsWithUserData.userFloats,
161 ecalEnergyErrPostCorrNew =
None,
162 ecalEnergyPreCorrNew =
None,
163 ecalEnergyPostCorrNew =
None,
164 energyScaleUpNew =
None,
165 energyScaleDownNew =
None,
166 energySigmaUpNew =
None,
167 energySigmaDownNew =
None
171 finalPhotons = cms.EDFilter(
"PATPhotonRefSelector",
172 src = cms.InputTag(
"slimmedPhotonsWithUserData"),
173 cut = cms.string(
"pt > 5 ")
176 photonTable = cms.EDProducer(
"SimpleCandidateFlatTableProducer",
177 src = cms.InputTag(
"linkedObjects",
"photons"),
178 cut = cms.string(
""),
179 name= cms.string(
"Photon"),
180 doc = cms.string(
"slimmedPhotons after basic selection (" + finalPhotons.cut.value()+
")"),
181 singleton = cms.bool(
False),
182 extension = cms.bool(
False),
183 variables = cms.PSet(CandVars,
184 jetIdx =
Var(
"?hasUserCand('jet')?userCand('jet').key():-1", int, doc=
"index of the associated jet (-1 if none)"),
185 electronIdx =
Var(
"?hasUserCand('electron')?userCand('electron').key():-1", int, doc=
"index of the associated electron (-1 if none)"),
186 energyErr =
Var(
"getCorrectedEnergyError('regression2')",float,doc=
"energy error of the cluster from regression",precision=6),
187 r9 =
Var(
"full5x5_r9()",float,doc=
"R9 of the supercluster, calculated with full 5x5 region",precision=10),
188 sieie =
Var(
"full5x5_sigmaIetaIeta()",float,doc=
"sigma_IetaIeta of the supercluster, calculated with full 5x5 region",precision=10),
190 "userInt('cutbasedID_loose')+userInt('cutbasedID_medium')+userInt('cutbasedID_tight')",
192 doc=
"cut-based ID bitmap, Fall17V2, (0:fail, 1:loose, 2:medium, 3:tight)"
194 cutBased_Fall17V1Bitmap =
Var(
195 "userInt('cutbasedIDV1_loose')+2*userInt('cutbasedIDV1_medium')+4*userInt('cutbasedIDV1_tight')",
197 doc=
"cut-based ID bitmap, Fall17V1, 2^(0:loose, 1:medium, 2:tight).",
200 "userInt('VIDNestedWPBitmap')",
204 electronVeto =
Var(
"passElectronVeto()",bool,doc=
"pass electron veto"),
205 pixelSeed =
Var(
"hasPixelSeed()",bool,doc=
"has pixel seed"),
206 mvaID =
Var(
"userFloat('mvaID')",float,doc=
"MVA ID score, Fall17V2",precision=10),
207 mvaID_Fall17V1p1 =
Var(
"userFloat('mvaID_Fall17V1p1')",float,doc=
"MVA ID score, Fall17V1p1",precision=10),
208 mvaID_WP90 =
Var(
"userInt('mvaID_WP90')",bool,doc=
"MVA ID WP90, Fall17V2"),
209 mvaID_WP80 =
Var(
"userInt('mvaID_WP80')",bool,doc=
"MVA ID WP80, Fall17V2"),
210 pfRelIso03_chg =
Var(
"userFloat('PFIsoChg')/pt",float,doc=
"PF relative isolation dR=0.3, charged component (with rho*EA PU corrections)"),
211 pfRelIso03_all =
Var(
"userFloat('PFIsoAll')/pt",float,doc=
"PF relative isolation dR=0.3, total (with rho*EA PU corrections)"),
212 hoe =
Var(
"hadronicOverEm()",float,doc=
"H over E",precision=8),
213 isScEtaEB =
Var(
"abs(superCluster().eta()) < 1.4442",bool,doc=
"is supercluster eta within barrel acceptance"),
214 isScEtaEE =
Var(
"abs(superCluster().eta()) > 1.566 && abs(superCluster().eta()) < 2.5",bool,doc=
"is supercluster eta within endcap acceptance"),
215 seedGain =
Var(
"userInt('seedGain')",
"uint8",doc=
"Gain of the seed crystal"),
221 for modifier
in run2_egamma_2016, run2_egamma_2017, run2_egamma_2018 :
222 modifier.toModify(photonTable.variables,
223 pt =
Var(
"pt*userFloat('ecalEnergyPostCorrNew')/userFloat('ecalEnergyPreCorrNew')", float, precision=-1, doc=
"p_{T}"),
224 energyErr =
Var(
"userFloat('ecalEnergyErrPostCorrNew')",float,doc=
"energy error of the cluster from regression",precision=6),
225 eCorr =
Var(
"userFloat('ecalEnergyPostCorrNew')/userFloat('ecalEnergyPreCorrNew')",float,doc=
"ratio of the calibrated energy/miniaod energy"),
229 for modifier
in run2_nanoAOD_94X2016,:
230 modifier.toModify(photonTable.variables,
231 pt =
Var(
"pt*userFloat('ecalEnergyPostCorr')/userFloat('ecalEnergyPreCorr')", float, precision=-1, doc=
"p_{T}"),
232 energyErr =
Var(
"userFloat('ecalEnergyErrPostCorr')",float,doc=
"energy error of the cluster from regression",precision=6),
233 eCorr =
Var(
"userFloat('ecalEnergyPostCorr')/userFloat('ecalEnergyPreCorr')",float,doc=
"ratio of the calibrated energy/miniaod energy"),
237 for modifier
in run2_nanoAOD_94X2016, run2_miniAOD_80XLegacy:
238 modifier.toModify(slimmedPhotonsWithUserData.userInts,
239 VIDNestedWPBitmap_Spring16V2p2 = cms.InputTag(
"bitmapVIDForPhoSpring16V2p2"),
242 for modifier
in run2_nanoAOD_94X2016, run2_miniAOD_80XLegacy:
243 modifier.toModify(photonTable.variables,
244 cutBased_Spring16V2p2 =
Var(
245 "userInt('cutID_Spring16_loose')+userInt('cutID_Spring16_medium')+userInt('cutID_Spring16_tight')",
247 doc=
"cut-based ID bitmap, Spring16V2p2, (0:fail, 1:loose, 2:medium, 3:tight)"
249 vidNestedWPBitmap_Spring16V2p2 =
Var(
250 "userInt('VIDNestedWPBitmap_Spring16V2p2')",
254 mvaID_Spring16nonTrigV1 =
Var(
255 "userFloat('mvaID_Spring16nonTrigV1')",
257 doc=
"MVA ID score, Spring16nonTrigV1",
262 photonsMCMatchForTable = cms.EDProducer(
"MCMatcher",
263 src = photonTable.src,
264 matched = cms.InputTag(
"finalGenParticles"),
265 mcPdgId = cms.vint32(11,22),
266 checkCharge = cms.bool(
False),
267 mcStatus = cms.vint32(1),
268 maxDeltaR = cms.double(0.3),
269 maxDPtRel = cms.double(0.5),
270 resolveAmbiguities = cms.bool(
True),
271 resolveByMatchQuality = cms.bool(
True),
274 photonMCTable = cms.EDProducer(
"CandMCMatchTableProducer",
275 src = photonTable.src,
276 mcMap = cms.InputTag(
"photonsMCMatchForTable"),
277 objName = photonTable.name,
278 objType = photonTable.name,
279 branchName = cms.string(
"genPart"),
280 docString = cms.string(
"MC matching to status==1 photons or electrons"),
283 from RecoEgamma.EgammaTools.egammaObjectModificationsInMiniAOD_cff
import egamma8XObjectUpdateModifier,egamma9X105XUpdateModifier,prependEgamma8XObjectUpdateModifier
285 slimmedPhotonsTo106X = cms.EDProducer(
"ModifiedPhotonProducer",
286 src = cms.InputTag(
"slimmedPhotons"),
287 modifierConfig = cms.PSet( modifications = cms.VPSet(egamma9X105XUpdateModifier) )
290 run2_miniAOD_80XLegacy.toModify( slimmedPhotonsTo106X.modifierConfig.modifications, prependEgamma8XObjectUpdateModifier )
292 for modifier
in run2_miniAOD_80XLegacy,run2_nanoAOD_94XMiniAODv1,run2_nanoAOD_94XMiniAODv2,run2_nanoAOD_94X2016 ,run2_nanoAOD_102Xv1:
293 modifier.toModify(bitmapVIDForPho, src =
"slimmedPhotonsTo106X")
294 modifier.toModify(bitmapVIDForPhoSpring16V2p2, src =
"slimmedPhotonsTo106X")
295 modifier.toModify(isoForPho, src =
"slimmedPhotonsTo106X")
296 modifier.toModify(calibratedPatPhotonsNano, src =
"slimmedPhotonsTo106X")
297 modifier.toModify(slimmedPhotonsWithUserData, src =
"slimmedPhotonsTo106X")
298 modifier.toModify(seedGainPho, src =
"slimmedPhotonsTo106X")
302 for modifier
in run2_egamma_2016, run2_egamma_2017, run2_egamma_2018:
303 modifier.toModify(photonTable.variables,
304 dEscaleUp=
Var(
"userFloat('ecalEnergyPostCorrNew') - userFloat('energyScaleUpNew')", float, doc=
"ecal energy scale shifted 1 sigma up (adding gain/stat/syst in quadrature)", precision=8),
305 dEscaleDown=
Var(
"userFloat('ecalEnergyPostCorrNew') - userFloat('energyScaleDownNew')", float, doc=
"ecal energy scale shifted 1 sigma down (adding gain/stat/syst in quadrature)", precision=8),
306 dEsigmaUp=
Var(
"userFloat('ecalEnergyPostCorrNew') - userFloat('energySigmaUpNew')", float, doc=
"ecal energy smearing value shifted 1 sigma up", precision=8),
307 dEsigmaDown=
Var(
"userFloat('ecalEnergyPostCorrNew') - userFloat('energySigmaDownNew')", float, doc=
"ecal energy smearing value shifted 1 sigma up", precision=8),
310 for modifier
in run2_nanoAOD_94X2016,:
311 modifier.toModify(photonTable.variables,
312 dEscaleUp=
Var(
"userFloat('ecalEnergyPostCorr') - userFloat('energyScaleUp')", float, doc=
"ecal energy scale shifted 1 sigma up (adding gain/stat/syst in quadrature)", precision=8),
313 dEscaleDown=
Var(
"userFloat('ecalEnergyPostCorr') - userFloat('energyScaleDown')", float, doc=
"ecal energy scale shifted 1 sigma down (adding gain/stat/syst in quadrature)", precision=8),
314 dEsigmaUp=
Var(
"userFloat('ecalEnergyPostCorr') - userFloat('energySigmaUp')", float, doc=
"ecal energy smearing value shifted 1 sigma up", precision=8),
315 dEsigmaDown=
Var(
"userFloat('ecalEnergyPostCorr') - userFloat('energySigmaDown')", float, doc=
"ecal energy smearing value shifted 1 sigma up", precision=8),
318 photonTask = cms.Task(bitmapVIDForPho, isoForPho, seedGainPho, calibratedPatPhotonsNano, slimmedPhotonsWithUserData, finalPhotons)
319 photonTablesTask = cms.Task(photonTable)
320 photonMCTask = cms.Task(photonsMCMatchForTable, photonMCTable)
322 from RecoEgamma.EgammaIsolationAlgos.egmPhotonIsolationMiniAOD_cff
import egmPhotonIsolation
323 from RecoEgamma.PhotonIdentification.photonIDValueMapProducer_cff
import photonIDValueMapProducer
325 _withUpdatePho_Task = cms.Task(egmPhotonIsolation,photonIDValueMapProducer,slimmedPhotonsTo106X)
326 _withUpdatePho_Task.add(photonTask.copy())
328 for modifier
in run2_nanoAOD_94XMiniAODv2,run2_nanoAOD_94X2016 ,run2_nanoAOD_102Xv1,run2_nanoAOD_94XMiniAODv1:
329 modifier.toReplaceWith(photonTask, _withUpdatePho_Task)
331 for modifier
in run2_miniAOD_80XLegacy, run2_nanoAOD_94X2016:
332 _withSpring16V2p2_Task = cms.Task(bitmapVIDForPhoSpring16V2p2)
333 _withSpring16V2p2_Task.add(_withUpdatePho_Task.copy())
334 modifier.toReplaceWith(photonTask, _withSpring16V2p2_Task)