NanoAOD/python/electrons_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002 from PhysicsTools.NanoAOD.simpleCandidateFlatTableProducer_cfi import simpleCandidateFlatTableProducer
0003 from PhysicsTools.NanoAOD.nano_eras_cff import *
0004 from PhysicsTools.NanoAOD.common_cff import *
0005 from math import ceil,log
0006
0007 ############################FOR bitmapVIDForEle main defn#############################
0008 electron_id_modules_WorkingPoints_nanoAOD = cms.PSet(
0009     modules = cms.vstring(
0010         'RecoEgamma.ElectronIdentification.Identification.heepElectronID_HEEPV70_cff',
0011         # HZZ ID
0012         'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Summer16UL_ID_ISO_cff',
0013         'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Summer17UL_ID_ISO_cff',
0014         'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Summer18UL_ID_ISO_cff',
0015         # Fall17: need to include the modules too to make sure they are run
0016         'RecoEgamma.ElectronIdentification.Identification.cutBasedElectronID_Fall17_94X_V2_cff',
0017         'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Fall17_iso_V2_cff',
0018         'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Fall17_noIso_V2_cff',
0019         # Run3Winter22:
0020         'RecoEgamma.ElectronIdentification.Identification.cutBasedElectronID_Winter22_122X_V1_cff',
0021         'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_RunIIIWinter22_iso_V1_cff',
0022         'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_RunIIIWinter22_noIso_V1_cff',
0023     ),
0024     WorkingPoints = cms.vstring(
0025         "egmGsfElectronIDs:cutBasedElectronID-RunIIIWinter22-V1-veto",
0026         "egmGsfElectronIDs:cutBasedElectronID-RunIIIWinter22-V1-loose",
0027         "egmGsfElectronIDs:cutBasedElectronID-RunIIIWinter22-V1-medium",
0028         "egmGsfElectronIDs:cutBasedElectronID-RunIIIWinter22-V1-tight",
0029     )
0030 )
0031
0032 # Use Fall17-94X-V2 as default for Run 2
0033 electron_id_modules_WorkingPoints_nanoAOD_Run2 = cms.PSet(
0034     modules = cms.vstring(
0035         'RecoEgamma.ElectronIdentification.Identification.cutBasedElectronID_Fall17_94X_V2_cff',
0036         'RecoEgamma.ElectronIdentification.Identification.heepElectronID_HEEPV70_cff',
0037         'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Fall17_iso_V2_cff',
0038         'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Fall17_noIso_V2_cff',
0039         # HZZ ID
0040         'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Summer16UL_ID_ISO_cff',
0041         'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Summer17UL_ID_ISO_cff',
0042         'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Summer18UL_ID_ISO_cff',
0043     ),
0044     WorkingPoints = cms.vstring(
0045         "egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-veto",
0046         "egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-loose",
0047         "egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-medium",
0048         "egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-tight",
0049     )
0050 )
0051
0052 # make Fall17 the default one in Run2
0053 run2_egamma.toModify(electron_id_modules_WorkingPoints_nanoAOD,
0054                           modules=electron_id_modules_WorkingPoints_nanoAOD_Run2.modules).\
0055         toModify(electron_id_modules_WorkingPoints_nanoAOD,
0056                  WorkingPoints=electron_id_modules_WorkingPoints_nanoAOD_Run2.WorkingPoints)
0057
0058 def _get_bitmapVIDForEle_docstring(modules,WorkingPoints):
0059     docstring=''
0060     for modname in modules:
0061         ids= __import__(modname, globals(), locals(), ['idName','cutFlow'])
0062         for name in dir(ids):
0063             _id = getattr(ids,name)
0064             if hasattr(_id,'idName') and hasattr(_id,'cutFlow'):
0065                 if (len(WorkingPoints)>0 and _id.idName==WorkingPoints[0].split(':')[-1]):
0066                     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))))
0067     return docstring
0068
0069 bitmapVIDForEle = cms.EDProducer("EleVIDNestedWPBitmapProducer",
0070     src = cms.InputTag("slimmedElectrons"),
0071     srcForID = cms.InputTag("reducedEgamma","reducedGedGsfElectrons"),
0072     WorkingPoints = electron_id_modules_WorkingPoints_nanoAOD.WorkingPoints,
0073 )
0074 _bitmapVIDForEle_docstring = _get_bitmapVIDForEle_docstring(electron_id_modules_WorkingPoints_nanoAOD.modules,bitmapVIDForEle.WorkingPoints)
0075
0076 bitmapVIDForEleFall17V2 = bitmapVIDForEle.clone(
0077     WorkingPoints = electron_id_modules_WorkingPoints_nanoAOD_Run2.WorkingPoints
0078     )
0079 _bitmapVIDForEleFall17V2_docstring = _get_bitmapVIDForEle_docstring(electron_id_modules_WorkingPoints_nanoAOD.modules, bitmapVIDForEleFall17V2.WorkingPoints)
0080
0081 bitmapVIDForEleHEEP = bitmapVIDForEle.clone(
0082     WorkingPoints = cms.vstring("egmGsfElectronIDs:heepElectronID-HEEPV70"
0083     )
0084 )
0085 _bitmapVIDForEleHEEP_docstring = _get_bitmapVIDForEle_docstring(electron_id_modules_WorkingPoints_nanoAOD.modules,bitmapVIDForEleHEEP.WorkingPoints)
0086 ############################for bitmapVIDForEle defn end#############################
0087
0088 #######################ISO ELE defn(in principle should be an import####################
0089 ##PhysicsTools/NanoAOD/python/EleIsoValueMapProducer_cfi.py
0090 isoForEle = cms.EDProducer("EleIsoValueMapProducer",
0091     src = cms.InputTag("slimmedElectrons"),
0092     relative = cms.bool(False),
0093     rho_MiniIso = cms.InputTag("fixedGridRhoFastjetAll"),
0094     rho_PFIso = cms.InputTag("fixedGridRhoFastjetAll"),
0095     EAFile_MiniIso = cms.FileInPath("RecoEgamma/ElectronIdentification/data/Run3_Winter22/effAreaElectrons_cone03_pfNeuHadronsAndPhotons_122X.txt"),
0096     EAFile_PFIso = cms.FileInPath("RecoEgamma/ElectronIdentification/data/Run3_Winter22/effAreaElectrons_cone03_pfNeuHadronsAndPhotons_122X.txt"),
0097 )
0098
0099 isoForEleFall17V2 = isoForEle.clone(
0100     EAFile_MiniIso = cms.FileInPath("RecoEgamma/ElectronIdentification/data/Fall17/effAreaElectrons_cone03_pfNeuHadronsAndPhotons_94X.txt"),
0101     EAFile_PFIso = cms.FileInPath("RecoEgamma/ElectronIdentification/data/Fall17/effAreaElectrons_cone03_pfNeuHadronsAndPhotons_94X.txt"),
0102 )
0103 #######################################ISO ELE end#####################################
0104
0105 ######################################ptRatioForEle#####################################
0106 ###import from hysicsTools/NanoAOD/pythonElectronJetVarProducer_cfi.py
0107 ptRatioRelForEle = cms.EDProducer("ElectronJetVarProducer",
0108     srcJet = cms.InputTag("updatedJetsPuppi"),
0109     srcLep = cms.InputTag("slimmedElectrons"),
0110     srcVtx = cms.InputTag("offlineSlimmedPrimaryVertices"),
0111 )
0112 ######################################ptRatioForEle#####################################
0113
0114 #############3###################seedGailEle#############################
0115 seedGainEle = cms.EDProducer("ElectronSeedGainProducer", src = cms.InputTag("slimmedElectrons"))
0116 ############################################seed gainELE
0117
0118 ############################calibratedPatElectrons##############
0119 ##this is a special one, so we leave the era modifications here#####
0120 import RecoEgamma.EgammaTools.calibratedEgammas_cff
0121
0122 calibratedPatElectronsNano = RecoEgamma.EgammaTools.calibratedEgammas_cff.calibratedPatElectrons.clone(
0123     produceCalibratedObjs = False,
0124     src = "slimmedElectrons"
0125 )
0126
0127 (run2_egamma_2016 & tracker_apv_vfp30_2016).toModify(
0128     calibratedPatElectronsNano,
0129     correctionFile = "EgammaAnalysis/ElectronTools/data/ScalesSmearings/Run2016_UltraLegacy_preVFP_RunFineEtaR9Gain"
0130 )
0131
0132 (run2_egamma_2016 & ~tracker_apv_vfp30_2016).toModify(
0133     calibratedPatElectronsNano,
0134     correctionFile = "EgammaAnalysis/ElectronTools/data/ScalesSmearings/Run2016_UltraLegacy_postVFP_RunFineEtaR9Gain"
0135 )
0136
0137 run2_egamma_2017.toModify(
0138     calibratedPatElectronsNano,
0139     correctionFile = "EgammaAnalysis/ElectronTools/data/ScalesSmearings/Run2017_24Feb2020_runEtaR9Gain_v2"
0140 )
0141
0142 run2_egamma_2018.toModify(
0143     calibratedPatElectronsNano,
0144     correctionFile = "EgammaAnalysis/ElectronTools/data/ScalesSmearings/Run2018_29Sep2020_RunFineEtaR9Gain"
0145 )
0146 ##############################end calibratedPatElectronsNano############################33
0147
0148 #####################Start slimmedElectronsWithUserData###############################3
0149 ##import from PhysicsTools/PatAlgos/python/electronsWithUserData_cfi.py
0150 slimmedElectronsWithUserData = cms.EDProducer("PATElectronUserDataEmbedder",
0151     src = cms.InputTag("slimmedElectrons"),
0152     parentSrcs = cms.VInputTag("reducedEgamma:reducedGedGsfElectrons"),
0153     userFloats = cms.PSet(
0154         mvaIso_Fall17V2 = cms.InputTag("electronMVAValueMapProducer:ElectronMVAEstimatorRun2Fall17IsoV2Values"),
0155         mvaNoIso_Fall17V2 = cms.InputTag("electronMVAValueMapProducer:ElectronMVAEstimatorRun2Fall17NoIsoV2Values"),
0156         mvaIso = cms.InputTag("electronMVAValueMapProducer:ElectronMVAEstimatorRun2RunIIIWinter22IsoV1Values"),
0157         mvaNoIso = cms.InputTag("electronMVAValueMapProducer:ElectronMVAEstimatorRun2RunIIIWinter22NoIsoV1Values"),
0158         mvaHZZIso = cms.InputTag("electronMVAValueMapProducer:ElectronMVAEstimatorRun2Summer18ULIdIsoValues"),
0159
0160         miniIsoChg = cms.InputTag("isoForEle:miniIsoChg"),
0161         miniIsoAll = cms.InputTag("isoForEle:miniIsoAll"),
0162         PFIsoChg = cms.InputTag("isoForEle:PFIsoChg"),
0163         PFIsoAll = cms.InputTag("isoForEle:PFIsoAll"),
0164         PFIsoAll04 = cms.InputTag("isoForEle:PFIsoAll04"),
0165
0166         miniIsoChg_Fall17V2 = cms.InputTag("isoForEleFall17V2:miniIsoChg"),
0167         miniIsoAll_Fall17V2 = cms.InputTag("isoForEleFall17V2:miniIsoAll"),
0168         PFIsoChg_Fall17V2 = cms.InputTag("isoForEleFall17V2:PFIsoChg"),
0169         PFIsoAll_Fall17V2 = cms.InputTag("isoForEleFall17V2:PFIsoAll"),
0170         PFIsoAll04_Fall17V2 = cms.InputTag("isoForEleFall17V2:PFIsoAll04"),
0171
0172         ptRatio = cms.InputTag("ptRatioRelForEle:ptRatio"),
0173         ptRel = cms.InputTag("ptRatioRelForEle:ptRel"),
0174         jetNDauChargedMVASel = cms.InputTag("ptRatioRelForEle:jetNDauChargedMVASel"),
0175     ),
0176     userIntFromBools = cms.PSet(
0177         mvaIso_Fall17V2_WP90 = cms.InputTag("egmGsfElectronIDs:mvaEleID-Fall17-iso-V2-wp90"),
0178         mvaIso_Fall17V2_WP80 = cms.InputTag("egmGsfElectronIDs:mvaEleID-Fall17-iso-V2-wp80"),
0179         mvaIso_Fall17V2_WPL = cms.InputTag("egmGsfElectronIDs:mvaEleID-Fall17-iso-V2-wpLoose"),
0180         mvaIso_WP90 = cms.InputTag("egmGsfElectronIDs:mvaEleID-RunIIIWinter22-iso-V1-wp90"),
0181         mvaIso_WP80 = cms.InputTag("egmGsfElectronIDs:mvaEleID-RunIIIWinter22-iso-V1-wp80"),
0182         mvaNoIso_Fall17V2_WP90 = cms.InputTag("egmGsfElectronIDs:mvaEleID-Fall17-noIso-V2-wp90"),
0183         mvaNoIso_Fall17V2_WP80 = cms.InputTag("egmGsfElectronIDs:mvaEleID-Fall17-noIso-V2-wp80"),
0184         mvaNoIso_Fall17V2_WPL = cms.InputTag("egmGsfElectronIDs:mvaEleID-Fall17-noIso-V2-wpLoose"),
0185         mvaNoIso_WP90 = cms.InputTag("egmGsfElectronIDs:mvaEleID-RunIIIWinter22-noIso-V1-wp90"),
0186         mvaNoIso_WP80 = cms.InputTag("egmGsfElectronIDs:mvaEleID-RunIIIWinter22-noIso-V1-wp80"),
0187
0188         cutBasedID_veto = cms.InputTag("egmGsfElectronIDs:cutBasedElectronID-RunIIIWinter22-V1-veto"),
0189         cutBasedID_loose = cms.InputTag("egmGsfElectronIDs:cutBasedElectronID-RunIIIWinter22-V1-loose"),
0190         cutBasedID_medium = cms.InputTag("egmGsfElectronIDs:cutBasedElectronID-RunIIIWinter22-V1-medium"),
0191         cutBasedID_tight = cms.InputTag("egmGsfElectronIDs:cutBasedElectronID-RunIIIWinter22-V1-tight"),
0192         cutBasedID_Fall17V2_veto = cms.InputTag("egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-veto"),
0193         cutBasedID_Fall17V2_loose = cms.InputTag("egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-loose"),
0194         cutBasedID_Fall17V2_medium = cms.InputTag("egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-medium"),
0195         cutBasedID_Fall17V2_tight = cms.InputTag("egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-tight"),
0196         cutBasedID_HEEP = cms.InputTag("egmGsfElectronIDs:heepElectronID-HEEPV70"),
0197     ),
0198     userInts = cms.PSet(
0199         VIDNestedWPBitmap = cms.InputTag("bitmapVIDForEle"),
0200         VIDNestedWPBitmap_Fall17V2 = cms.InputTag("bitmapVIDForEleFall17V2"),
0201         VIDNestedWPBitmapHEEP = cms.InputTag("bitmapVIDForEleHEEP"),
0202         seedGain = cms.InputTag("seedGainEle"),
0203
0204     ),
0205     userCands = cms.PSet(
0206         jetForLepJetVar = cms.InputTag("ptRatioRelForEle:jetForLepJetVar") # warning: Ptr is null if no match is found
0207     ),
0208 )
0209
0210 # no need for the Run3 IDs in Run2
0211 run2_egamma.toModify(slimmedElectronsWithUserData.userFloats,
0212                      mvaIso = None,
0213                      mvaNoIso = None,
0214                      miniIsoChg = None,
0215                      miniIsoAll = None,
0216                      PFIsoChg = None,
0217                      PFIsoAll = None,
0218                      PFIsoAll04 = None).\
0219         toModify(slimmedElectronsWithUserData.userIntFromBools,
0220                  mvaIso_WP90 = None,
0221                  mvaIso_WP80 = None,
0222                  mvaNoIso_WP90 = None,
0223                  mvaNoIso_WP80 = None,
0224                  cutBasedID_veto = None,
0225                  cutBasedID_loose = None,
0226                  cutBasedID_medium = None,
0227                  cutBasedID_tight = None).\
0228         toModify(slimmedElectronsWithUserData.userInts,
0229                  VIDNestedWPBitmap = None)
0230
0231 run2_egamma.toModify(
0232     slimmedElectronsWithUserData.userFloats,
0233     ecalTrkEnergyErrPostCorrNew = cms.InputTag("calibratedPatElectronsNano","ecalTrkEnergyErrPostCorr"),
0234     ecalTrkEnergyPreCorrNew     = cms.InputTag("calibratedPatElectronsNano","ecalTrkEnergyPreCorr"),
0235     ecalTrkEnergyPostCorrNew    = cms.InputTag("calibratedPatElectronsNano","ecalTrkEnergyPostCorr"),
0236     energyScaleUpNew            = cms.InputTag("calibratedPatElectronsNano","energyScaleUp"),
0237     energyScaleDownNew          = cms.InputTag("calibratedPatElectronsNano","energyScaleDown"),
0238     energySigmaUpNew            = cms.InputTag("calibratedPatElectronsNano","energySigmaUp"),
0239     energySigmaDownNew          = cms.InputTag("calibratedPatElectronsNano","energySigmaDown")
0240 )
0241
0242 (run2_egamma_2016).toModify(
0243     slimmedElectronsWithUserData.userFloats,
0244     mvaHZZIso = "electronMVAValueMapProducer:ElectronMVAEstimatorRun2Summer16ULIdIsoValues"
0245 )
0246 (run2_egamma_2017).toModify(
0247     slimmedElectronsWithUserData.userFloats,
0248     mvaHZZIso = "electronMVAValueMapProducer:ElectronMVAEstimatorRun2Summer17ULIdIsoValues"
0249 )
0250 (run2_egamma_2018).toModify(
0251     slimmedElectronsWithUserData.userFloats,
0252     mvaHZZIso = "electronMVAValueMapProducer:ElectronMVAEstimatorRun2Summer18ULIdIsoValues"
0253 )
0254 #################################################END slimmedElectrons with user data#####################
0255
0256 #################################################finalElectrons#####################
0257 finalElectrons = cms.EDFilter("PATElectronRefSelector",
0258     src = cms.InputTag("slimmedElectronsWithUserData"),
0259     cut = cms.string("pt > 5 ")
0260 )
0261 #################################################finalElectrons#####################
0262
0263 ################################################electronMVATTH#####################
0264 electronMVATTH= cms.EDProducer("EleBaseMVAValueMapProducer",
0265     src = cms.InputTag("linkedObjects","electrons"),
0266     weightFile =  cms.FileInPath("PhysicsTools/NanoAOD/data/el_BDTG_2017.weights.xml"),
0267     name = cms.string("electronMVATTH"),
0268     isClassifier = cms.bool(True),
0269     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"]),
0270     variables = cms.PSet(
0271         LepGood_pt = cms.string("pt"),
0272         LepGood_eta = cms.string("eta"),
0273         LepGood_jetNDauChargedMVASel = cms.string("?userCand('jetForLepJetVar').isNonnull()?userFloat('jetNDauChargedMVASel'):0"),
0274         # NB: only using Fall17V2 iso here
0275         LepGood_miniRelIsoCharged = cms.string("userFloat('miniIsoChg_Fall17V2')/pt"),
0276         LepGood_miniRelIsoNeutral = cms.string("(userFloat('miniIsoAll_Fall17V2')-userFloat('miniIsoChg_Fall17V2'))/pt"),
0277         LepGood_jetPtRelv2 = cms.string("?userCand('jetForLepJetVar').isNonnull()?userFloat('ptRel'):0"),
0278         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"),
0279         LepGood_jetPtRatio = cms.string("?userCand('jetForLepJetVar').isNonnull()?min(userFloat('ptRatio'),1.5):1.0/(1.0+userFloat('PFIsoAll04_Fall17V2')/pt)"),
0280         LepGood_dxy = cms.string("log(abs(dB('PV2D')))"),
0281         LepGood_sip3d = cms.string("abs(dB('PV3D')/edB('PV3D'))"),
0282         LepGood_dz = cms.string("log(abs(dB('PVDZ')))"),
0283         LepGood_mvaFall17V2noIso = cms.string("userFloat('mvaNoIso_Fall17V2')"),
0284     )
0285 )
0286 run2_egamma_2016.toModify(
0287     electronMVATTH,
0288     weightFile = "PhysicsTools/NanoAOD/data/el_BDTG_2016.weights.xml",
0289 )
0290 ################################################electronMVATTH end#####################
0291
0292 ################################################electronTable defn #####################
0293 electronTable = simpleCandidateFlatTableProducer.clone(
0294     src = cms.InputTag("linkedObjects","electrons"),
0295     name= cms.string("Electron"),
0296     doc = cms.string("slimmedElectrons after basic selection (" + finalElectrons.cut.value()+")"),
0297     variables = cms.PSet(CandVars,
0298         jetIdx = Var("?hasUserCand('jet')?userCand('jet').key():-1", "int16", doc="index of the associated jet (-1 if none)"),
0299         photonIdx = Var("?overlaps('photons').size()>0?overlaps('photons')[0].key():-1", "int16", doc="index of the first associated photon (-1 if none)"),
0300         svIdx = Var("?hasUserCand('vertex')?userCand('vertex').key():-1", "int16", doc="index of matching secondary vertex"),
0301         energyErr = Var("p4Error('P4_COMBINATION')",float,doc="energy error of the cluster-track combination",precision=6),
0302         dz = Var("dB('PVDZ')",float,doc="dz (with sign) wrt first PV, in cm",precision=10),
0303         dzErr = Var("abs(edB('PVDZ'))",float,doc="dz uncertainty, in cm",precision=6),
0304         dxy = Var("dB('PV2D')",float,doc="dxy (with sign) wrt first PV, in cm",precision=10),
0305         dxyErr = Var("edB('PV2D')",float,doc="dxy uncertainty, in cm",precision=6),
0306         ip3d = Var("abs(dB('PV3D'))",float,doc="3D impact parameter wrt first PV, in cm",precision=10),
0307         sip3d = Var("abs(dB('PV3D')/edB('PV3D'))",float,doc="3D impact parameter significance wrt first PV, in cm",precision=10),
0308         deltaEtaSC = Var("superCluster().eta()-eta()",float,doc="delta eta (SC,ele) with sign",precision=10),
0309         r9 = Var("full5x5_r9()",float,doc="R9 of the supercluster, calculated with full 5x5 region",precision=10),
0310         sieie = Var("full5x5_sigmaIetaIeta()",float,doc="sigma_IetaIeta of the supercluster, calculated with full 5x5 region",precision=10),
0311         eInvMinusPInv = Var("(1-eSuperClusterOverP())/ecalEnergy()",float,doc="1/E_SC - 1/p_trk",precision=10),
0312         scEtOverPt = Var("(superCluster().energy()/(pt*cosh(superCluster().eta())))-1",float,doc="(supercluster transverse energy)/pt-1",precision=8),
0313
0314         mvaIso = Var("userFloat('mvaIso')",float,doc="MVA Iso ID score, Winter22V1"),
0315         mvaIso_WP80 = Var("userInt('mvaIso_WP80')",bool,doc="MVA Iso ID WP80, Winter22V1"),
0316         mvaIso_WP90 = Var("userInt('mvaIso_WP90')",bool,doc="MVA Iso ID WP90, Winter22V1"),
0317         mvaNoIso = Var("userFloat('mvaNoIso')",float,doc="MVA noIso ID score, Winter22V1"),
0318         mvaNoIso_WP80 = Var("userInt('mvaNoIso_WP80')",bool,doc="MVA noIso ID WP80, Winter22V1"),
0319         mvaNoIso_WP90 = Var("userInt('mvaNoIso_WP90')",bool,doc="MVA noIso ID WP90, Winter22V1"),
0320         mvaIso_Fall17V2 = Var("userFloat('mvaIso_Fall17V2')",float,doc="MVA Iso ID score, Fall17V2"),
0321         mvaIso_Fall17V2_WP80 = Var("userInt('mvaIso_Fall17V2_WP80')",bool,doc="MVA Iso ID WP80, Fall17V2"),
0322         mvaIso_Fall17V2_WP90 = Var("userInt('mvaIso_Fall17V2_WP90')",bool,doc="MVA Iso ID WP90, Fall17V2"),
0323         mvaIso_Fall17V2_WPL = Var("userInt('mvaIso_Fall17V2_WPL')",bool,doc="MVA Iso ID loose WP, Fall17V2"),
0324         mvaNoIso_Fall17V2 = Var("userFloat('mvaNoIso_Fall17V2')",float,doc="MVA noIso ID score, Fall17V2"),
0325         mvaNoIso_Fall17V2_WP80 = Var("userInt('mvaNoIso_Fall17V2_WP80')",bool,doc="MVA noIso ID WP80, Fall17V2"),
0326         mvaNoIso_Fall17V2_WP90 = Var("userInt('mvaNoIso_Fall17V2_WP90')",bool,doc="MVA noIso ID WP90, Fall17V2"),
0327         mvaNoIso_Fall17V2_WPL = Var("userInt('mvaNoIso_Fall17V2_WPL')",bool,doc="MVA noIso ID loose WP, Fall17V2"),
0328         mvaHZZIso = Var("userFloat('mvaHZZIso')", float,doc="HZZ MVA Iso ID score"),
0329
0330         cutBased = Var("userInt('cutBasedID_veto')+userInt('cutBasedID_loose')+userInt('cutBasedID_medium')+userInt('cutBasedID_tight')", "uint8", doc="cut-based ID RunIII Winter22 (0:fail, 1:veto, 2:loose, 3:medium, 4:tight)"),
0331         cutBased_Fall17V2 = Var("userInt('cutBasedID_Fall17V2_veto')+userInt('cutBasedID_Fall17V2_loose')+userInt('cutBasedID_Fall17V2_medium')+userInt('cutBasedID_Fall17V2_tight')", "uint8", doc="cut-based ID Fall17V2 (0:fail, 1:veto, 2:loose, 3:medium, 4:tight)"),
0332         vidNestedWPBitmap = Var("userInt('VIDNestedWPBitmap')", int, doc=_bitmapVIDForEle_docstring),
0333         vidNestedWPBitmap_Fall17V2 = Var("userInt('VIDNestedWPBitmap_Fall17V2')", int, doc=_bitmapVIDForEleFall17V2_docstring),
0334         vidNestedWPBitmapHEEP = Var("userInt('VIDNestedWPBitmapHEEP')", int, doc=_bitmapVIDForEleHEEP_docstring),
0335         cutBased_HEEP = Var("userInt('cutBasedID_HEEP')",bool,doc="cut-based HEEP ID"),
0336         miniPFRelIso_chg = Var("userFloat('miniIsoChg')/pt",float,doc="mini PF relative isolation, charged component"),
0337         miniPFRelIso_all = Var("userFloat('miniIsoAll')/pt",float,doc="mini PF relative isolation, total (with scaled rho*EA PU Winter22V1 corrections)"),
0338         pfRelIso03_chg = Var("userFloat('PFIsoChg')/pt",float,doc="PF relative isolation dR=0.3, charged component"),
0339         pfRelIso03_all = Var("userFloat('PFIsoAll')/pt",float,doc="PF relative isolation dR=0.3, total (with rho*EA PU Winter22V1 corrections)"),
0340         miniPFRelIso_chg_Fall17V2 = Var("userFloat('miniIsoChg_Fall17V2')/pt",float,doc="mini PF relative isolation, charged component"),
0341         miniPFRelIso_all_Fall17V2 = Var("userFloat('miniIsoAll_Fall17V2')/pt",float,doc="mini PF relative isolation, total (with scaled rho*EA PU Fall17V2 corrections)"),
0342         pfRelIso03_chg_Fall17V2 = Var("userFloat('PFIsoChg_Fall17V2')/pt",float,doc="PF relative isolation dR=0.3 with 94 EffArea, charged component"),
0343         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)"),
0344         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),
0345         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),
0346         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),
0347         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),
0348         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),
0349         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),
0350         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),
0351         hoe = Var("hadronicOverEm()",float,doc="H over E",precision=8),
0352         tightCharge = Var("isGsfCtfScPixChargeConsistent() + isGsfScPixChargeConsistent()", "uint8", doc="Tight charge criteria (0:none, 1:isGsfScPixChargeConsistent, 2:isGsfCtfScPixChargeConsistent)"),
0353         convVeto = Var("passConversionVeto()",bool,doc="pass conversion veto"),
0354         lostHits = Var("gsfTrack.hitPattern.numberOfLostHits('MISSING_INNER_HITS')","uint8",doc="number of missing inner hits"),
0355         isPFcand = Var("pfCandidateRef().isNonnull()",bool,doc="electron is PF candidate"),
0356         seedGain = Var("userInt('seedGain')","uint8",doc="Gain of the seed crystal"),
0357         seediEtaOriX = Var("superCluster().seedCrysIEtaOrIx","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."),
0358         seediPhiOriY = Var("superCluster().seedCrysIPhiOrIy",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."),
0359         jetNDauCharged = Var("?userCand('jetForLepJetVar').isNonnull()?userFloat('jetNDauChargedMVASel'):0", "uint8", doc="number of charged daughters of the closest jet"),
0360     ),
0361     externalVariables = cms.PSet(
0362         mvaTTH = ExtVar(cms.InputTag("electronMVATTH"),float, doc="TTH MVA lepton ID score",precision=14),
0363         fsrPhotonIdx = ExtVar(cms.InputTag("leptonFSRphotons:eleFsrIndex"), "int16", doc="Index of the lowest-dR/ET2 among associated FSR photons"),
0364     ),
0365 )
0366
0367 (run2_egamma).toModify(
0368         # energy scale/smearing: only for Run2
0369         electronTable.variables,
0370         pt = Var("pt*userFloat('ecalTrkEnergyPostCorrNew')/userFloat('ecalTrkEnergyPreCorrNew')", float, precision=-1, doc="p_{T}"),
0371         energyErr = Var("userFloat('ecalTrkEnergyErrPostCorrNew')", float, precision=6, doc="energy error of the cluster-track combination"),
0372         eCorr = Var("userFloat('ecalTrkEnergyPostCorrNew')/userFloat('ecalTrkEnergyPreCorrNew')", float, doc="ratio of the calibrated energy/miniaod energy"),
0373         scEtOverPt = Var("(superCluster().energy()/(pt*userFloat('ecalTrkEnergyPostCorrNew')/userFloat('ecalTrkEnergyPreCorrNew')*cosh(superCluster().eta())))-1",float,doc="(supercluster transverse energy)/pt-1",precision=8),
0374         dEscaleUp=Var("userFloat('ecalTrkEnergyPostCorrNew')-userFloat('energyScaleUpNew')", float,  doc="ecal energy scale shifted 1 sigma up(adding gain/stat/syst in quadrature)", precision=8),
0375         dEscaleDown=Var("userFloat('ecalTrkEnergyPostCorrNew')-userFloat('energyScaleDownNew')", float,  doc="ecal energy scale shifted 1 sigma down (adding gain/stat/syst in quadrature)", precision=8),
0376         dEsigmaUp=Var("userFloat('ecalTrkEnergyPostCorrNew')-userFloat('energySigmaUpNew')", float, doc="ecal energy smearing value shifted 1 sigma up", precision=8),
0377         dEsigmaDown=Var("userFloat('ecalTrkEnergyPostCorrNew')-userFloat('energySigmaDownNew')", float,  doc="ecal energy smearing value shifted 1 sigma up", precision=8),
0378
0379         # switch default IDs back to Fall17V2 in Run2, remove Winter22V1 and quadratic iso
0380         cutBased = electronTable.variables.cutBased_Fall17V2,
0381         cutBased_Fall17V2 = None,
0382         mvaIso = electronTable.variables.mvaIso_Fall17V2,
0383         mvaIso_Fall17V2 = None,
0384         mvaIso_WP80 = electronTable.variables.mvaIso_Fall17V2_WP80,
0385         mvaIso_Fall17V2_WP80 = None,
0386         mvaIso_WP90 = electronTable.variables.mvaIso_Fall17V2_WP90,
0387         mvaIso_Fall17V2_WP90 = None,
0388         mvaIso_WPL = electronTable.variables.mvaIso_Fall17V2_WPL,
0389         mvaIso_Fall17V2_WPL = None,
0390         mvaNoIso = electronTable.variables.mvaNoIso_Fall17V2,
0391         mvaNoIso_Fall17V2 = None,
0392         mvaNoIso_WP80 = electronTable.variables.mvaNoIso_Fall17V2_WP80,
0393         mvaNoIso_Fall17V2_WP80 = None,
0394         mvaNoIso_WP90 = electronTable.variables.mvaNoIso_Fall17V2_WP90,
0395         mvaNoIso_Fall17V2_WP90 = None,
0396         mvaNoIso_WPL = electronTable.variables.mvaNoIso_Fall17V2_WPL,
0397         mvaNoIso_Fall17V2_WPL = None,
0398         vidNestedWPBitmap = electronTable.variables.cutBased_Fall17V2,
0399         vidNestedWPBitmap_Fall17V2 = None,
0400         miniPFRelIso_chg = electronTable.variables.miniPFRelIso_chg_Fall17V2,
0401         miniPFRelIso_chg_Fall17V2 = None,
0402         miniPFRelIso_all = electronTable.variables.miniPFRelIso_all_Fall17V2,
0403         miniPFRelIso_all_Fall17V2 = None,
0404         pfRelIso03_chg = electronTable.variables.pfRelIso03_chg_Fall17V2,
0405         pfRelIso03_chg_Fall17V2 = None,
0406         pfRelIso03_all = electronTable.variables.pfRelIso03_all_Fall17V2,
0407         pfRelIso03_all_Fall17V2 = None,
0408         jetRelIso = electronTable.variables.jetRelIso_Fall17V2,
0409         jetRelIso_Fall17V2 = None,
0410 )
0411
0412 #############electron Table END#####################
0413 # Depends on particlelevel producer run in particlelevel_cff
0414 tautaggerForMatching = cms.EDProducer("GenJetTauTaggerProducer",
0415                                       src = cms.InputTag('particleLevel:leptons')
0416 )
0417  ##PhysicsTools/NanoAOD/plugins/GenJetGenPartMerger.cc##this class misses fillDescription#TODO
0418 matchingElecPhoton = cms.EDProducer("GenJetGenPartMerger",
0419                                     srcJet =cms.InputTag("particleLevel:leptons"),
0420                                     srcPart=cms.InputTag("particleLevel:photons"),
0421                                     cut = cms.string("pt > 3"),
0422                                     hasTauAnc=cms.InputTag("tautaggerForMatching"),
0423 )
0424 electronsMCMatchForTableAlt = cms.EDProducer("GenJetMatcherDRPtByDR",  # cut on deltaR, deltaPt/Pt; pick best by deltaR
0425     src         = electronTable.src,                 # final reco collection
0426     matched     = cms.InputTag("matchingElecPhoton:merged"), # final mc-truth particle collection
0427     mcPdgId     = cms.vint32(11,22),                 # one or more PDG ID (11 = el, 22 = pho); absolute values (see below)
0428     checkCharge = cms.bool(False),              # True = require RECO and MC objects to have the same charge
0429     mcStatus    = cms.vint32(),
0430     maxDeltaR   = cms.double(0.3),              # Minimum deltaR for the match
0431     maxDPtRel   = cms.double(0.5),              # Minimum deltaPt/Pt for the match
0432     resolveAmbiguities    = cms.bool(True),     # Forbid two RECO objects to match to the same GEN object
0433     resolveByMatchQuality = cms.bool(True),    # False = just match input in order; True = pick lowest deltaR pair first
0434 )
0435 electronsMCMatchForTable = cms.EDProducer("MCMatcher",  # cut on deltaR, deltaPt/Pt; pick best by deltaR
0436     src         = electronTable.src,                 # final reco collection
0437     matched     = cms.InputTag("finalGenParticles"), # final mc-truth particle collection
0438     mcPdgId     = cms.vint32(11,22),                 # one or more PDG ID (11 = el, 22 = pho); absolute values (see below)
0439     checkCharge = cms.bool(False),              # True = require RECO and MC objects to have the same charge
0440     mcStatus    = cms.vint32(1),                # PYTHIA status code (1 = stable, 2 = shower, 3 = hard scattering)
0441     maxDeltaR   = cms.double(0.3),              # Minimum deltaR for the match
0442     maxDPtRel   = cms.double(0.5),              # Minimum deltaPt/Pt for the match
0443     resolveAmbiguities    = cms.bool(True),     # Forbid two RECO objects to match to the same GEN object
0444     resolveByMatchQuality = cms.bool(True),    # False = just match input in order; True = pick lowest deltaR pair first
0445 )
0446 #should be cloned from PhysicsTools/NanoAOD/python/candMcMatchTable_cfi.py
0447 electronMCTable = cms.EDProducer("CandMCMatchTableProducer",
0448     src     = electronTable.src,
0449     mcMapDressedLep = cms.InputTag("electronsMCMatchForTableAlt"),
0450     mcMap   = cms.InputTag("electronsMCMatchForTable"),
0451     mapTauAnc = cms.InputTag("matchingElecPhoton:hasTauAnc"),
0452     objName = electronTable.name,
0453     objType = electronTable.name, #cms.string("Electron"),
0454     branchName = cms.string("genPart"),
0455     docString = cms.string("MC matching to status==1 electrons or photons"),
0456     genparticles     = cms.InputTag("finalGenParticles"),
0457 )
0458
0459 electronTask = cms.Task(bitmapVIDForEle,bitmapVIDForEleFall17V2,bitmapVIDForEleHEEP,isoForEle,isoForEleFall17V2,ptRatioRelForEle,seedGainEle,calibratedPatElectronsNano,slimmedElectronsWithUserData,finalElectrons)
0460 electronTablesTask = cms.Task(electronMVATTH, electronTable)
0461 electronMCTask = cms.Task(tautaggerForMatching, matchingElecPhoton, electronsMCMatchForTable, electronsMCMatchForTableAlt, electronMCTable)
0462
0463 _electronTask_Run2 = electronTask.copy()
0464 _electronTask_Run2.remove(bitmapVIDForEle)
0465 _electronTask_Run2.remove(isoForEle)
0466 _electronTask_Run2.add(calibratedPatElectronsNano)
0467 run2_egamma.toReplaceWith(electronTask, _electronTask_Run2)
0468
0469 # Revert back to AK4 CHS jets for Run2 inputs
0470 run2_nanoAOD_ANY.toModify(
0471     ptRatioRelForEle,srcJet="updatedJets")