CMS 3D CMS Logo
electrons_cff.py
Go to the documentation of this file.
1 import FWCore.ParameterSet.Config as cms
2 from PhysicsTools.NanoAOD.simpleCandidateFlatTableProducer_cfi import simpleCandidateFlatTableProducer
3 from PhysicsTools.NanoAOD.nano_eras_cff import *
4 from PhysicsTools.NanoAOD.common_cff import *
5 from math import ceil,log
6 
7 
8 electron_id_modules_WorkingPoints_nanoAOD = cms.PSet(
9  modules = cms.vstring(
10  'RecoEgamma.ElectronIdentification.Identification.heepElectronID_HEEPV70_cff',
11  # HZZ ID
12  'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Summer16UL_ID_ISO_cff',
13  'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Summer17UL_ID_ISO_cff',
14  'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Summer18UL_ID_ISO_cff',
15  # Fall17: need to include the modules too to make sure they are run
16  'RecoEgamma.ElectronIdentification.Identification.cutBasedElectronID_Fall17_94X_V2_cff',
17  'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Fall17_iso_V2_cff',
18  'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Fall17_noIso_V2_cff',
19  # Run3Winter22:
20  'RecoEgamma.ElectronIdentification.Identification.cutBasedElectronID_Winter22_122X_V1_cff',
21  'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_RunIIIWinter22_iso_V1_cff',
22  'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_RunIIIWinter22_noIso_V1_cff',
23  ),
24  WorkingPoints = cms.vstring(
25  "egmGsfElectronIDs:cutBasedElectronID-RunIIIWinter22-V1-veto",
26  "egmGsfElectronIDs:cutBasedElectronID-RunIIIWinter22-V1-loose",
27  "egmGsfElectronIDs:cutBasedElectronID-RunIIIWinter22-V1-medium",
28  "egmGsfElectronIDs:cutBasedElectronID-RunIIIWinter22-V1-tight",
29  )
30 )
31 
32 # Use Fall17-94X-V2 as default for Run 2
33 electron_id_modules_WorkingPoints_nanoAOD_Run2 = cms.PSet(
34  modules = cms.vstring(
35  'RecoEgamma.ElectronIdentification.Identification.cutBasedElectronID_Fall17_94X_V2_cff',
36  'RecoEgamma.ElectronIdentification.Identification.heepElectronID_HEEPV70_cff',
37  'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Fall17_iso_V2_cff',
38  'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Fall17_noIso_V2_cff',
39  # HZZ ID
40  'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Summer16UL_ID_ISO_cff',
41  'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Summer17UL_ID_ISO_cff',
42  'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Summer18UL_ID_ISO_cff',
43  ),
44  WorkingPoints = cms.vstring(
45  "egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-veto",
46  "egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-loose",
47  "egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-medium",
48  "egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-tight",
49  )
50 )
51 
52 # make Fall17 the default one in Run2
53 run2_egamma.toModify(electron_id_modules_WorkingPoints_nanoAOD,
54  modules=electron_id_modules_WorkingPoints_nanoAOD_Run2.modules).\
55  toModify(electron_id_modules_WorkingPoints_nanoAOD,
56  WorkingPoints=electron_id_modules_WorkingPoints_nanoAOD_Run2.WorkingPoints)
57 
58 def _get_bitmapVIDForEle_docstring(modules,WorkingPoints):
59  docstring=''
60  for modname in modules:
61  ids= __import__(modname, globals(), locals(), ['idName','cutFlow'])
62  for name in dir(ids):
63  _id = getattr(ids,name)
64  if hasattr(_id,'idName') and hasattr(_id,'cutFlow'):
65  if (len(WorkingPoints)>0 and _id.idName==WorkingPoints[0].split(':')[-1]):
66  docstring = 'VID compressed bitmap (%s), %d bits per cut'%(','.join([cut.cutName.value() for cut in _id.cutFlow]),int(ceil(log(len(WorkingPoints)+1,2))))
67  return docstring
68 
69 bitmapVIDForEle = cms.EDProducer("EleVIDNestedWPBitmapProducer",
70  src = cms.InputTag("slimmedElectrons"),
71  srcForID = cms.InputTag("reducedEgamma","reducedGedGsfElectrons"),
72  WorkingPoints = electron_id_modules_WorkingPoints_nanoAOD.WorkingPoints,
73 )
74 _bitmapVIDForEle_docstring = _get_bitmapVIDForEle_docstring(electron_id_modules_WorkingPoints_nanoAOD.modules,bitmapVIDForEle.WorkingPoints)
75 
76 bitmapVIDForEleFall17V2 = bitmapVIDForEle.clone(
77  WorkingPoints = electron_id_modules_WorkingPoints_nanoAOD_Run2.WorkingPoints
78  )
79 _bitmapVIDForEleFall17V2_docstring = _get_bitmapVIDForEle_docstring(electron_id_modules_WorkingPoints_nanoAOD.modules, bitmapVIDForEleFall17V2.WorkingPoints)
80 
81 bitmapVIDForEleHEEP = bitmapVIDForEle.clone(
82  WorkingPoints = cms.vstring("egmGsfElectronIDs:heepElectronID-HEEPV70"
83  )
84 )
85 _bitmapVIDForEleHEEP_docstring = _get_bitmapVIDForEle_docstring(electron_id_modules_WorkingPoints_nanoAOD.modules,bitmapVIDForEleHEEP.WorkingPoints)
86 
87 
88 
90 isoForEle = cms.EDProducer("EleIsoValueMapProducer",
91  src = cms.InputTag("slimmedElectrons"),
92  relative = cms.bool(False),
93  rho_MiniIso = cms.InputTag("fixedGridRhoFastjetAll"),
94  rho_PFIso = cms.InputTag("fixedGridRhoFastjetAll"),
95  EAFile_MiniIso = cms.FileInPath("RecoEgamma/ElectronIdentification/data/Run3_Winter22/effAreaElectrons_cone03_pfNeuHadronsAndPhotons_122X.txt"),
96  EAFile_PFIso = cms.FileInPath("RecoEgamma/ElectronIdentification/data/Run3_Winter22/effAreaElectrons_cone03_pfNeuHadronsAndPhotons_122X.txt"),
97 )
98 
99 isoForEleFall17V2 = isoForEle.clone(
100  EAFile_MiniIso = cms.FileInPath("RecoEgamma/ElectronIdentification/data/Fall17/effAreaElectrons_cone03_pfNeuHadronsAndPhotons_94X.txt"),
101  EAFile_PFIso = cms.FileInPath("RecoEgamma/ElectronIdentification/data/Fall17/effAreaElectrons_cone03_pfNeuHadronsAndPhotons_94X.txt"),
102 )
103 
104 
105 
107 ptRatioRelForEle = cms.EDProducer("ElectronJetVarProducer",
108  srcJet = cms.InputTag("updatedJetsPuppi"),
109  srcLep = cms.InputTag("slimmedElectrons"),
110  srcVtx = cms.InputTag("offlineSlimmedPrimaryVertices"),
111 )
112 
113 
114 
115 seedGainEle = cms.EDProducer("ElectronSeedGainProducer", src = cms.InputTag("slimmedElectrons"))
116 
117 
118 
119 seedLocationEle = cms.EDProducer("ElectronSeedLocationProducer", src = cms.InputTag("slimmedElectrons"))
120 
121 
123 import RecoEgamma.EgammaTools.calibratedEgammas_cff
124 
125 calibratedPatElectronsNano = RecoEgamma.EgammaTools.calibratedEgammas_cff.calibratedPatElectrons.clone(
126  produceCalibratedObjs = False,
127  src = "slimmedElectrons"
128 )
129 
130 (run2_egamma_2016 & tracker_apv_vfp30_2016).toModify(
131  calibratedPatElectronsNano,
132  correctionFile = "EgammaAnalysis/ElectronTools/data/ScalesSmearings/Run2016_UltraLegacy_preVFP_RunFineEtaR9Gain"
133 )
134 
135 (run2_egamma_2016 & ~tracker_apv_vfp30_2016).toModify(
136  calibratedPatElectronsNano,
137  correctionFile = "EgammaAnalysis/ElectronTools/data/ScalesSmearings/Run2016_UltraLegacy_postVFP_RunFineEtaR9Gain"
138 )
139 
140 run2_egamma_2017.toModify(
141  calibratedPatElectronsNano,
142  correctionFile = "EgammaAnalysis/ElectronTools/data/ScalesSmearings/Run2017_24Feb2020_runEtaR9Gain_v2"
143 )
144 
145 run2_egamma_2018.toModify(
146  calibratedPatElectronsNano,
147  correctionFile = "EgammaAnalysis/ElectronTools/data/ScalesSmearings/Run2018_29Sep2020_RunFineEtaR9Gain"
148 )
149 
150 
151 
153 slimmedElectronsWithUserData = cms.EDProducer("PATElectronUserDataEmbedder",
154  src = cms.InputTag("slimmedElectrons"),
155  parentSrcs = cms.VInputTag("reducedEgamma:reducedGedGsfElectrons"),
156  userFloats = cms.PSet(
157  mvaIso_Fall17V2 = cms.InputTag("electronMVAValueMapProducer:ElectronMVAEstimatorRun2Fall17IsoV2Values"),
158  mvaNoIso_Fall17V2 = cms.InputTag("electronMVAValueMapProducer:ElectronMVAEstimatorRun2Fall17NoIsoV2Values"),
159  mvaIso = cms.InputTag("electronMVAValueMapProducer:ElectronMVAEstimatorRun2RunIIIWinter22IsoV1Values"),
160  mvaNoIso = cms.InputTag("electronMVAValueMapProducer:ElectronMVAEstimatorRun2RunIIIWinter22NoIsoV1Values"),
161  mvaHZZIso = cms.InputTag("electronMVAValueMapProducer:ElectronMVAEstimatorRun2Summer18ULIdIsoValues"),
162 
163  miniIsoChg = cms.InputTag("isoForEle:miniIsoChg"),
164  miniIsoAll = cms.InputTag("isoForEle:miniIsoAll"),
165  PFIsoChg = cms.InputTag("isoForEle:PFIsoChg"),
166  PFIsoAll = cms.InputTag("isoForEle:PFIsoAll"),
167  PFIsoAll04 = cms.InputTag("isoForEle:PFIsoAll04"),
168 
169  miniIsoChg_Fall17V2 = cms.InputTag("isoForEleFall17V2:miniIsoChg"),
170  miniIsoAll_Fall17V2 = cms.InputTag("isoForEleFall17V2:miniIsoAll"),
171  PFIsoChg_Fall17V2 = cms.InputTag("isoForEleFall17V2:PFIsoChg"),
172  PFIsoAll_Fall17V2 = cms.InputTag("isoForEleFall17V2:PFIsoAll"),
173  PFIsoAll04_Fall17V2 = cms.InputTag("isoForEleFall17V2:PFIsoAll04"),
174 
175  ptRatio = cms.InputTag("ptRatioRelForEle:ptRatio"),
176  ptRel = cms.InputTag("ptRatioRelForEle:ptRel"),
177  jetNDauChargedMVASel = cms.InputTag("ptRatioRelForEle:jetNDauChargedMVASel"),
178  ),
179  userIntFromBools = cms.PSet(
180  mvaIso_Fall17V2_WP90 = cms.InputTag("egmGsfElectronIDs:mvaEleID-Fall17-iso-V2-wp90"),
181  mvaIso_Fall17V2_WP80 = cms.InputTag("egmGsfElectronIDs:mvaEleID-Fall17-iso-V2-wp80"),
182  mvaIso_Fall17V2_WPL = cms.InputTag("egmGsfElectronIDs:mvaEleID-Fall17-iso-V2-wpLoose"),
183  mvaIso_WP90 = cms.InputTag("egmGsfElectronIDs:mvaEleID-RunIIIWinter22-iso-V1-wp90"),
184  mvaIso_WP80 = cms.InputTag("egmGsfElectronIDs:mvaEleID-RunIIIWinter22-iso-V1-wp80"),
185  mvaNoIso_Fall17V2_WP90 = cms.InputTag("egmGsfElectronIDs:mvaEleID-Fall17-noIso-V2-wp90"),
186  mvaNoIso_Fall17V2_WP80 = cms.InputTag("egmGsfElectronIDs:mvaEleID-Fall17-noIso-V2-wp80"),
187  mvaNoIso_Fall17V2_WPL = cms.InputTag("egmGsfElectronIDs:mvaEleID-Fall17-noIso-V2-wpLoose"),
188  mvaNoIso_WP90 = cms.InputTag("egmGsfElectronIDs:mvaEleID-RunIIIWinter22-noIso-V1-wp90"),
189  mvaNoIso_WP80 = cms.InputTag("egmGsfElectronIDs:mvaEleID-RunIIIWinter22-noIso-V1-wp80"),
190 
191  cutBasedID_veto = cms.InputTag("egmGsfElectronIDs:cutBasedElectronID-RunIIIWinter22-V1-veto"),
192  cutBasedID_loose = cms.InputTag("egmGsfElectronIDs:cutBasedElectronID-RunIIIWinter22-V1-loose"),
193  cutBasedID_medium = cms.InputTag("egmGsfElectronIDs:cutBasedElectronID-RunIIIWinter22-V1-medium"),
194  cutBasedID_tight = cms.InputTag("egmGsfElectronIDs:cutBasedElectronID-RunIIIWinter22-V1-tight"),
195  cutBasedID_Fall17V2_veto = cms.InputTag("egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-veto"),
196  cutBasedID_Fall17V2_loose = cms.InputTag("egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-loose"),
197  cutBasedID_Fall17V2_medium = cms.InputTag("egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-medium"),
198  cutBasedID_Fall17V2_tight = cms.InputTag("egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-tight"),
199  cutBasedID_HEEP = cms.InputTag("egmGsfElectronIDs:heepElectronID-HEEPV70"),
200  ),
201  userInts = cms.PSet(
202  VIDNestedWPBitmap = cms.InputTag("bitmapVIDForEle"),
203  VIDNestedWPBitmap_Fall17V2 = cms.InputTag("bitmapVIDForEleFall17V2"),
204  VIDNestedWPBitmapHEEP = cms.InputTag("bitmapVIDForEleHEEP"),
205  seedGain = cms.InputTag("seedGainEle"),
206  seediEtaOriX = cms.InputTag("seedLocationEle:iEtaOriX"),
207  seediPhiOriY = cms.InputTag("seedLocationEle:iPhiOriY"),
208  ),
209  userCands = cms.PSet(
210  jetForLepJetVar = cms.InputTag("ptRatioRelForEle:jetForLepJetVar") # warning: Ptr is null if no match is found
211  ),
212 )
213 
214 # no need for the Run3 IDs in Run2
215 run2_egamma.toModify(slimmedElectronsWithUserData.userFloats,
216  mvaIso = None,
217  mvaNoIso = None,
218  miniIsoChg = None,
219  miniIsoAll = None,
220  PFIsoChg = None,
221  PFIsoAll = None,
222  PFIsoAll04 = None).\
223  toModify(slimmedElectronsWithUserData.userIntFromBools,
224  mvaIso_WP90 = None,
225  mvaIso_WP80 = None,
226  mvaNoIso_WP90 = None,
227  mvaNoIso_WP80 = None,
228  cutBasedID_veto = None,
229  cutBasedID_loose = None,
230  cutBasedID_medium = None,
231  cutBasedID_tight = None).\
232  toModify(slimmedElectronsWithUserData.userInts,
233  VIDNestedWPBitmap = None)
234 
235 run2_egamma.toModify(
236  slimmedElectronsWithUserData.userFloats,
237  ecalTrkEnergyErrPostCorrNew = cms.InputTag("calibratedPatElectronsNano","ecalTrkEnergyErrPostCorr"),
238  ecalTrkEnergyPreCorrNew = cms.InputTag("calibratedPatElectronsNano","ecalTrkEnergyPreCorr"),
239  ecalTrkEnergyPostCorrNew = cms.InputTag("calibratedPatElectronsNano","ecalTrkEnergyPostCorr"),
240  energyScaleUpNew = cms.InputTag("calibratedPatElectronsNano","energyScaleUp"),
241  energyScaleDownNew = cms.InputTag("calibratedPatElectronsNano","energyScaleDown"),
242  energySigmaUpNew = cms.InputTag("calibratedPatElectronsNano","energySigmaUp"),
243  energySigmaDownNew = cms.InputTag("calibratedPatElectronsNano","energySigmaDown")
244 )
245 
246 (run2_egamma_2016).toModify(
247  slimmedElectronsWithUserData.userFloats,
248  mvaHZZIso = "electronMVAValueMapProducer:ElectronMVAEstimatorRun2Summer16ULIdIsoValues"
249 )
250 (run2_egamma_2017).toModify(
251  slimmedElectronsWithUserData.userFloats,
252  mvaHZZIso = "electronMVAValueMapProducer:ElectronMVAEstimatorRun2Summer17ULIdIsoValues"
253 )
254 (run2_egamma_2018).toModify(
255  slimmedElectronsWithUserData.userFloats,
256  mvaHZZIso = "electronMVAValueMapProducer:ElectronMVAEstimatorRun2Summer18ULIdIsoValues"
257 )
258 
259 
260 
261 finalElectrons = cms.EDFilter("PATElectronRefSelector",
262  src = cms.InputTag("slimmedElectronsWithUserData"),
263  cut = cms.string("pt > 5 ")
264 )
265 
266 
267 
268 electronMVATTH= cms.EDProducer("EleBaseMVAValueMapProducer",
269  src = cms.InputTag("linkedObjects","electrons"),
270  weightFile = cms.FileInPath("PhysicsTools/NanoAOD/data/el_BDTG_2017.weights.xml"),
271  name = cms.string("electronMVATTH"),
272  isClassifier = cms.bool(True),
273  variablesOrder = cms.vstring(["LepGood_pt","LepGood_eta","LepGood_jetNDauChargedMVASel","LepGood_miniRelIsoCharged","LepGood_miniRelIsoNeutral","LepGood_jetPtRelv2","LepGood_jetDF","LepGood_jetPtRatio","LepGood_dxy","LepGood_sip3d","LepGood_dz","LepGood_mvaFall17V2noIso"]),
274  variables = cms.PSet(
275  LepGood_pt = cms.string("pt"),
276  LepGood_eta = cms.string("eta"),
277  LepGood_jetNDauChargedMVASel = cms.string("?userCand('jetForLepJetVar').isNonnull()?userFloat('jetNDauChargedMVASel'):0"),
278  # NB: only using Fall17V2 iso here
279  LepGood_miniRelIsoCharged = cms.string("userFloat('miniIsoChg_Fall17V2')/pt"),
280  LepGood_miniRelIsoNeutral = cms.string("(userFloat('miniIsoAll_Fall17V2')-userFloat('miniIsoChg_Fall17V2'))/pt"),
281  LepGood_jetPtRelv2 = cms.string("?userCand('jetForLepJetVar').isNonnull()?userFloat('ptRel'):0"),
282  LepGood_jetDF = cms.string("?userCand('jetForLepJetVar').isNonnull()?max(userCand('jetForLepJetVar').bDiscriminator('pfDeepFlavourJetTags:probbb')+userCand('jetForLepJetVar').bDiscriminator('pfDeepFlavourJetTags:probb')+userCand('jetForLepJetVar').bDiscriminator('pfDeepFlavourJetTags:problepb'),0.0):0.0"),
283  LepGood_jetPtRatio = cms.string("?userCand('jetForLepJetVar').isNonnull()?min(userFloat('ptRatio'),1.5):1.0/(1.0+userFloat('PFIsoAll04_Fall17V2')/pt)"),
284  LepGood_dxy = cms.string("log(abs(dB('PV2D')))"),
285  LepGood_sip3d = cms.string("abs(dB('PV3D')/edB('PV3D'))"),
286  LepGood_dz = cms.string("log(abs(dB('PVDZ')))"),
287  LepGood_mvaFall17V2noIso = cms.string("userFloat('mvaNoIso_Fall17V2')"),
288  )
289 )
290 run2_egamma_2016.toModify(
291  electronMVATTH,
292  weightFile = "PhysicsTools/NanoAOD/data/el_BDTG_2016.weights.xml",
293 )
294 
295 
296 
297 electronTable = simpleCandidateFlatTableProducer.clone(
298  src = cms.InputTag("linkedObjects","electrons"),
299  name= cms.string("Electron"),
300  doc = cms.string("slimmedElectrons after basic selection (" + finalElectrons.cut.value()+")"),
301  variables = cms.PSet(CandVars,
302  jetIdx = Var("?hasUserCand('jet')?userCand('jet').key():-1", int, doc="index of the associated jet (-1 if none)"),
303  photonIdx = Var("?overlaps('photons').size()>0?overlaps('photons')[0].key():-1", int, doc="index of the first associated photon (-1 if none)"),
304  svIdx = Var("?hasUserCand('vertex')?userCand('vertex').key():-1", int, doc="index of matching secondary vertex"),
305  energyErr = Var("p4Error('P4_COMBINATION')",float,doc="energy error of the cluster-track combination",precision=6),
306  dz = Var("dB('PVDZ')",float,doc="dz (with sign) wrt first PV, in cm",precision=10),
307  dzErr = Var("abs(edB('PVDZ'))",float,doc="dz uncertainty, in cm",precision=6),
308  dxy = Var("dB('PV2D')",float,doc="dxy (with sign) wrt first PV, in cm",precision=10),
309  dxyErr = Var("edB('PV2D')",float,doc="dxy uncertainty, in cm",precision=6),
310  ip3d = Var("abs(dB('PV3D'))",float,doc="3D impact parameter wrt first PV, in cm",precision=10),
311  sip3d = Var("abs(dB('PV3D')/edB('PV3D'))",float,doc="3D impact parameter significance wrt first PV, in cm",precision=10),
312  deltaEtaSC = Var("superCluster().eta()-eta()",float,doc="delta eta (SC,ele) with sign",precision=10),
313  r9 = Var("full5x5_r9()",float,doc="R9 of the supercluster, calculated with full 5x5 region",precision=10),
314  sieie = Var("full5x5_sigmaIetaIeta()",float,doc="sigma_IetaIeta of the supercluster, calculated with full 5x5 region",precision=10),
315  eInvMinusPInv = Var("(1-eSuperClusterOverP())/ecalEnergy()",float,doc="1/E_SC - 1/p_trk",precision=10),
316  scEtOverPt = Var("(superCluster().energy()/(pt*cosh(superCluster().eta())))-1",float,doc="(supercluster transverse energy)/pt-1",precision=8),
317 
318  mvaIso = Var("userFloat('mvaIso')",float,doc="MVA Iso ID score, Winter22V1"),
319  mvaIso_WP80 = Var("userInt('mvaIso_WP80')",bool,doc="MVA Iso ID WP80, Winter22V1"),
320  mvaIso_WP90 = Var("userInt('mvaIso_WP90')",bool,doc="MVA Iso ID WP90, Winter22V1"),
321  mvaNoIso = Var("userFloat('mvaNoIso')",float,doc="MVA noIso ID score, Winter22V1"),
322  mvaNoIso_WP80 = Var("userInt('mvaNoIso_WP80')",bool,doc="MVA noIso ID WP80, Winter22V1"),
323  mvaNoIso_WP90 = Var("userInt('mvaNoIso_WP90')",bool,doc="MVA noIso ID WP90, Winter22V1"),
324  mvaIso_Fall17V2 = Var("userFloat('mvaIso_Fall17V2')",float,doc="MVA Iso ID score, Fall17V2"),
325  mvaIso_Fall17V2_WP80 = Var("userInt('mvaIso_Fall17V2_WP80')",bool,doc="MVA Iso ID WP80, Fall17V2"),
326  mvaIso_Fall17V2_WP90 = Var("userInt('mvaIso_Fall17V2_WP90')",bool,doc="MVA Iso ID WP90, Fall17V2"),
327  mvaIso_Fall17V2_WPL = Var("userInt('mvaIso_Fall17V2_WPL')",bool,doc="MVA Iso ID loose WP, Fall17V2"),
328  mvaNoIso_Fall17V2 = Var("userFloat('mvaNoIso_Fall17V2')",float,doc="MVA noIso ID score, Fall17V2"),
329  mvaNoIso_Fall17V2_WP80 = Var("userInt('mvaNoIso_Fall17V2_WP80')",bool,doc="MVA noIso ID WP80, Fall17V2"),
330  mvaNoIso_Fall17V2_WP90 = Var("userInt('mvaNoIso_Fall17V2_WP90')",bool,doc="MVA noIso ID WP90, Fall17V2"),
331  mvaNoIso_Fall17V2_WPL = Var("userInt('mvaNoIso_Fall17V2_WPL')",bool,doc="MVA noIso ID loose WP, Fall17V2"),
332  mvaHZZIso = Var("userFloat('mvaHZZIso')", float,doc="HZZ MVA Iso ID score"),
333 
334  cutBased = Var("userInt('cutBasedID_veto')+userInt('cutBasedID_loose')+userInt('cutBasedID_medium')+userInt('cutBasedID_tight')",int,doc="cut-based ID RunIII Winter22 (0:fail, 1:veto, 2:loose, 3:medium, 4:tight)"),
335  cutBased_Fall17V2 = Var("userInt('cutBasedID_Fall17V2_veto')+userInt('cutBasedID_Fall17V2_loose')+userInt('cutBasedID_Fall17V2_medium')+userInt('cutBasedID_Fall17V2_tight')",int,doc="cut-based ID Fall17V2 (0:fail, 1:veto, 2:loose, 3:medium, 4:tight)"),
336  vidNestedWPBitmap = Var("userInt('VIDNestedWPBitmap')",int,doc=_bitmapVIDForEle_docstring),
337  vidNestedWPBitmap_Fall17V2 = Var("userInt('VIDNestedWPBitmap_Fall17V2')",int,doc=_bitmapVIDForEleFall17V2_docstring),
338  vidNestedWPBitmapHEEP = Var("userInt('VIDNestedWPBitmapHEEP')",int,doc=_bitmapVIDForEleHEEP_docstring),
339  cutBased_HEEP = Var("userInt('cutBasedID_HEEP')",bool,doc="cut-based HEEP ID"),
340  miniPFRelIso_chg = Var("userFloat('miniIsoChg')/pt",float,doc="mini PF relative isolation, charged component"),
341  miniPFRelIso_all = Var("userFloat('miniIsoAll')/pt",float,doc="mini PF relative isolation, total (with scaled rho*EA PU Winter22V1 corrections)"),
342  pfRelIso03_chg = Var("userFloat('PFIsoChg')/pt",float,doc="PF relative isolation dR=0.3, charged component"),
343  pfRelIso03_all = Var("userFloat('PFIsoAll')/pt",float,doc="PF relative isolation dR=0.3, total (with rho*EA PU Winter22V1 corrections)"),
344  miniPFRelIso_chg_Fall17V2 = Var("userFloat('miniIsoChg_Fall17V2')/pt",float,doc="mini PF relative isolation, charged component"),
345  miniPFRelIso_all_Fall17V2 = Var("userFloat('miniIsoAll_Fall17V2')/pt",float,doc="mini PF relative isolation, total (with scaled rho*EA PU Fall17V2 corrections)"),
346  pfRelIso03_chg_Fall17V2 = Var("userFloat('PFIsoChg_Fall17V2')/pt",float,doc="PF relative isolation dR=0.3 with 94 EffArea, charged component"),
347  pfRelIso03_all_Fall17V2 = Var("userFloat('PFIsoAll_Fall17V2')/pt",float,doc="PF relative isolation dR=0.3 with 94 EffArea, total (with rho*EA PU corrections)"),
348  jetRelIso = Var("?userCand('jetForLepJetVar').isNonnull()?(1./userFloat('ptRatio'))-1.:userFloat('PFIsoAll04')/pt",float,doc="Relative isolation in matched jet (1/ptRatio-1, pfRelIso04_all if no matched jet)",precision=8),
349  jetRelIso_Fall17V2 = Var("?userCand('jetForLepJetVar').isNonnull()?(1./userFloat('ptRatio'))-1.:userFloat('PFIsoAll04_Fall17V2')/pt",float,doc="Relative isolation in matched jet (1/ptRatio-1, pfRelIso04_all if no matched jet)",precision=8),
350  jetPtRelv2 = Var("?userCand('jetForLepJetVar').isNonnull()?userFloat('ptRel'):0",float,doc="Relative momentum of the lepton with respect to the closest jet after subtracting the lepton",precision=8),
351  dr03TkSumPt = Var("?pt>35?dr03TkSumPt():0",float,doc="Non-PF track isolation within a delta R cone of 0.3 with electron pt > 35 GeV",precision=8),
352  dr03TkSumPtHEEP = Var("?pt>35?dr03TkSumPtHEEP():0",float,doc="Non-PF track isolation within a delta R cone of 0.3 with electron pt > 35 GeV used in HEEP ID",precision=8),
353  dr03EcalRecHitSumEt = Var("?pt>35?dr03EcalRecHitSumEt():0",float,doc="Non-PF Ecal isolation within a delta R cone of 0.3 with electron pt > 35 GeV",precision=8),
354  dr03HcalDepth1TowerSumEt = Var("?pt>35?dr03HcalTowerSumEt(1):0",float,doc="Non-PF Hcal isolation within a delta R cone of 0.3 with electron pt > 35 GeV",precision=8),
355  hoe = Var("hadronicOverEm()",float,doc="H over E",precision=8),
356  tightCharge = Var("isGsfCtfScPixChargeConsistent() + isGsfScPixChargeConsistent()",int,doc="Tight charge criteria (0:none, 1:isGsfScPixChargeConsistent, 2:isGsfCtfScPixChargeConsistent)"),
357  convVeto = Var("passConversionVeto()",bool,doc="pass conversion veto"),
358  lostHits = Var("gsfTrack.hitPattern.numberOfLostHits('MISSING_INNER_HITS')","uint8",doc="number of missing inner hits"),
359  isPFcand = Var("pfCandidateRef().isNonnull()",bool,doc="electron is PF candidate"),
360  seedGain = Var("userInt('seedGain')","uint8",doc="Gain of the seed crystal"),
361  seediEtaOriX = Var("userInt('seediEtaOriX')","int8",doc="iEta or iX of seed crystal. iEta is barrel-only, iX is endcap-only. iEta runs from -85 to +85, with no crystal at iEta=0. iX runs from 1 to 100."),
362  seediPhiOriY = Var("userInt('seediPhiOriY')",int,doc="iPhi or iY of seed crystal. iPhi is barrel-only, iY is endcap-only. iPhi runs from 1 to 360. iY runs from 1 to 100."),
363  jetNDauCharged = Var("?userCand('jetForLepJetVar').isNonnull()?userFloat('jetNDauChargedMVASel'):0", "uint8", doc="number of charged daughters of the closest jet"),
364  ),
365  externalVariables = cms.PSet(
366  mvaTTH = ExtVar(cms.InputTag("electronMVATTH"),float, doc="TTH MVA lepton ID score",precision=14),
367  fsrPhotonIdx = ExtVar(cms.InputTag("leptonFSRphotons:eleFsrIndex"),int, doc="Index of the lowest-dR/ET2 among associated FSR photons"),
368  ),
369 )
370 
371 (run2_egamma).toModify(
372  # energy scale/smearing: only for Run2
373  electronTable.variables,
374  pt = Var("pt*userFloat('ecalTrkEnergyPostCorrNew')/userFloat('ecalTrkEnergyPreCorrNew')", float, precision=-1, doc="p_{T}"),
375  energyErr = Var("userFloat('ecalTrkEnergyErrPostCorrNew')", float, precision=6, doc="energy error of the cluster-track combination"),
376  eCorr = Var("userFloat('ecalTrkEnergyPostCorrNew')/userFloat('ecalTrkEnergyPreCorrNew')", float, doc="ratio of the calibrated energy/miniaod energy"),
377  scEtOverPt = Var("(superCluster().energy()/(pt*userFloat('ecalTrkEnergyPostCorrNew')/userFloat('ecalTrkEnergyPreCorrNew')*cosh(superCluster().eta())))-1",float,doc="(supercluster transverse energy)/pt-1",precision=8),
378  dEscaleUp=Var("userFloat('ecalTrkEnergyPostCorrNew')-userFloat('energyScaleUpNew')", float, doc="ecal energy scale shifted 1 sigma up(adding gain/stat/syst in quadrature)", precision=8),
379  dEscaleDown=Var("userFloat('ecalTrkEnergyPostCorrNew')-userFloat('energyScaleDownNew')", float, doc="ecal energy scale shifted 1 sigma down (adding gain/stat/syst in quadrature)", precision=8),
380  dEsigmaUp=Var("userFloat('ecalTrkEnergyPostCorrNew')-userFloat('energySigmaUpNew')", float, doc="ecal energy smearing value shifted 1 sigma up", precision=8),
381  dEsigmaDown=Var("userFloat('ecalTrkEnergyPostCorrNew')-userFloat('energySigmaDownNew')", float, doc="ecal energy smearing value shifted 1 sigma up", precision=8),
382 
383  # switch default IDs back to Fall17V2 in Run2, remove Winter22V1 and quadratic iso
384  cutBased = electronTable.variables.cutBased_Fall17V2,
385  cutBased_Fall17V2 = None,
386  mvaIso = electronTable.variables.mvaIso_Fall17V2,
387  mvaIso_Fall17V2 = None,
388  mvaIso_WP80 = electronTable.variables.mvaIso_Fall17V2_WP80,
389  mvaIso_Fall17V2_WP80 = None,
390  mvaIso_WP90 = electronTable.variables.mvaIso_Fall17V2_WP90,
391  mvaIso_Fall17V2_WP90 = None,
392  mvaIso_WPL = electronTable.variables.mvaIso_Fall17V2_WPL,
393  mvaIso_Fall17V2_WPL = None,
394  mvaNoIso = electronTable.variables.mvaNoIso_Fall17V2,
395  mvaNoIso_Fall17V2 = None,
396  mvaNoIso_WP80 = electronTable.variables.mvaNoIso_Fall17V2_WP80,
397  mvaNoIso_Fall17V2_WP80 = None,
398  mvaNoIso_WP90 = electronTable.variables.mvaNoIso_Fall17V2_WP90,
399  mvaNoIso_Fall17V2_WP90 = None,
400  mvaNoIso_WPL = electronTable.variables.mvaNoIso_Fall17V2_WPL,
401  mvaNoIso_Fall17V2_WPL = None,
402  vidNestedWPBitmap = electronTable.variables.cutBased_Fall17V2,
403  vidNestedWPBitmap_Fall17V2 = None,
404  miniPFRelIso_chg = electronTable.variables.miniPFRelIso_chg_Fall17V2,
405  miniPFRelIso_chg_Fall17V2 = None,
406  miniPFRelIso_all = electronTable.variables.miniPFRelIso_all_Fall17V2,
407  miniPFRelIso_all_Fall17V2 = None,
408  pfRelIso03_chg = electronTable.variables.pfRelIso03_chg_Fall17V2,
409  pfRelIso03_chg_Fall17V2 = None,
410  pfRelIso03_all = electronTable.variables.pfRelIso03_all_Fall17V2,
411  pfRelIso03_all_Fall17V2 = None,
412  jetRelIso = electronTable.variables.jetRelIso_Fall17V2,
413  jetRelIso_Fall17V2 = None,
414 )
415 
416 
418 tautaggerForMatching = cms.EDProducer("GenJetTauTaggerProducer",
419  src = cms.InputTag('particleLevel:leptons')
420 )
421 
422 matchingElecPhoton = cms.EDProducer("GenJetGenPartMerger",
423  srcJet =cms.InputTag("particleLevel:leptons"),
424  srcPart=cms.InputTag("particleLevel:photons"),
425  cut = cms.string("pt > 3"),
426  hasTauAnc=cms.InputTag("tautaggerForMatching"),
427 )
428 electronsMCMatchForTableAlt = cms.EDProducer("GenJetMatcherDRPtByDR", # cut on deltaR, deltaPt/Pt; pick best by deltaR
429  src = electronTable.src, # final reco collection
430  matched = cms.InputTag("matchingElecPhoton:merged"), # final mc-truth particle collection
431  mcPdgId = cms.vint32(11,22), # one or more PDG ID (11 = el, 22 = pho); absolute values (see below)
432  checkCharge = cms.bool(False), # True = require RECO and MC objects to have the same charge
433  mcStatus = cms.vint32(),
434  maxDeltaR = cms.double(0.3), # Minimum deltaR for the match
435  maxDPtRel = cms.double(0.5), # Minimum deltaPt/Pt for the match
436  resolveAmbiguities = cms.bool(True), # Forbid two RECO objects to match to the same GEN object
437  resolveByMatchQuality = cms.bool(True), # False = just match input in order; True = pick lowest deltaR pair first
438 )
439 electronsMCMatchForTable = cms.EDProducer("MCMatcher", # cut on deltaR, deltaPt/Pt; pick best by deltaR
440  src = electronTable.src, # final reco collection
441  matched = cms.InputTag("finalGenParticles"), # final mc-truth particle collection
442  mcPdgId = cms.vint32(11,22), # one or more PDG ID (11 = el, 22 = pho); absolute values (see below)
443  checkCharge = cms.bool(False), # True = require RECO and MC objects to have the same charge
444  mcStatus = cms.vint32(1), # PYTHIA status code (1 = stable, 2 = shower, 3 = hard scattering)
445  maxDeltaR = cms.double(0.3), # Minimum deltaR for the match
446  maxDPtRel = cms.double(0.5), # Minimum deltaPt/Pt for the match
447  resolveAmbiguities = cms.bool(True), # Forbid two RECO objects to match to the same GEN object
448  resolveByMatchQuality = cms.bool(True), # False = just match input in order; True = pick lowest deltaR pair first
449 )
450 #should be cloned from PhysicsTools/NanoAOD/python/candMcMatchTable_cfi.py
451 electronMCTable = cms.EDProducer("CandMCMatchTableProducer",
452  src = electronTable.src,
453  mcMapDressedLep = cms.InputTag("electronsMCMatchForTableAlt"),
454  mcMap = cms.InputTag("electronsMCMatchForTable"),
455  mapTauAnc = cms.InputTag("matchingElecPhoton:hasTauAnc"),
456  objName = electronTable.name,
457  objType = electronTable.name, #cms.string("Electron"),
458  branchName = cms.string("genPart"),
459  docString = cms.string("MC matching to status==1 electrons or photons"),
460  genparticles = cms.InputTag("finalGenParticles"),
461 )
462 
463 electronTask = cms.Task(bitmapVIDForEle,bitmapVIDForEleFall17V2,bitmapVIDForEleHEEP,isoForEle,isoForEleFall17V2,ptRatioRelForEle,seedGainEle,seedLocationEle,calibratedPatElectronsNano,slimmedElectronsWithUserData,finalElectrons)
464 electronTablesTask = cms.Task(electronMVATTH, electronTable)
465 electronMCTask = cms.Task(tautaggerForMatching, matchingElecPhoton, electronsMCMatchForTable, electronsMCMatchForTableAlt, electronMCTable)
466 
467 _electronTask_Run2 = electronTask.copy()
468 _electronTask_Run2.remove(bitmapVIDForEle)
469 _electronTask_Run2.remove(isoForEle)
470 _electronTask_Run2.add(calibratedPatElectronsNano)
471 run2_egamma.toReplaceWith(electronTask, _electronTask_Run2)
472 
473 # Revert back to AK4 CHS jets for Run2 inputs
474 run2_nanoAOD_ANY.toModify(
475  ptRatioRelForEle,srcJet="updatedJets")
reco::ceil
constexpr int32_t ceil(float num)
Definition: constexpr_cmath.h:7
common_cff.ExtVar
def ExtVar(tag, valtype, doc=None, precision=-1)
Definition: common_cff.py:27
common_cff.Var
def Var(expr, valtype, doc=None, precision=-1)
Definition: common_cff.py:16
nano_eras_cff
DeadROC_duringRun.dir
dir
Definition: DeadROC_duringRun.py:23
electrons_cff.int
int
Definition: electrons_cff.py:302
electrons_cff._get_bitmapVIDForEle_docstring
def _get_bitmapVIDForEle_docstring(modules, WorkingPoints)
Definition: electrons_cff.py:58
join
static std::string join(char **cmd)
Definition: RemoteFile.cc:19
common_cff
Generated for CMSSW Reference Manual by doxygen 1.8.14