forked from cms-analysis/EgammaAnalysis-TnPTreeProducer
-
Notifications
You must be signed in to change notification settings - Fork 2
/
egmElectronIDModules_cff.py
113 lines (91 loc) · 6.32 KB
/
egmElectronIDModules_cff.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
import FWCore.ParameterSet.Config as cms
###################################################################
## ID MODULES
###################################################################
from PhysicsTools.SelectorUtils.tools.vid_id_tools import *
from EgammaAnalysis.TnPTreeProducer.cmssw_version import isReleaseAbove
def setIDs(process, options):
switchOnVIDElectronIdProducer(process, DataFormat.AOD if options['useAOD'] else DataFormat.MiniAOD)
# define which IDs we want to produce
my_id_modules = [
'RecoEgamma.ElectronIdentification.Identification.cutBasedElectronID_Summer16_80X_V1_cff',
'RecoEgamma.ElectronIdentification.Identification.heepElectronID_HEEPV60_cff',
'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Spring16_GeneralPurpose_V1_cff',
'RecoEgamma.ElectronIdentification.Identification.cutBasedElectronID_Fall17_94X_V1_cff',
'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Fall17_noIso_V1_cff',
'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Fall17_iso_V1_cff',
'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Fall17_noIso_V2_cff',
'RecoEgamma.ElectronIdentification.Identification.mvaElectronID_Fall17_iso_V2_cff',
'RecoEgamma.ElectronIdentification.Identification.cutBasedElectronID_Fall17_94X_V2_cff'
]
### add only miniAOD supported IDs
if not options['useAOD'] :
my_id_modules.append( 'RecoEgamma.ElectronIdentification.Identification.cutBasedElectronHLTPreselecition_Summer16_V1_cff' )
my_id_modules.append( 'EgammaAnalysis.TnPTreeProducer.Identification.cutBasedDoubleElectronHLTPreselecition_Summer16_V1_cff')
for idmod in my_id_modules:
setupAllVIDIdsInModule(process, idmod, setupVIDElectronSelection)
process.egmGsfElectronIDs.physicsObjectSrc = cms.InputTag(options['ELECTRON_COLL'])
process.electronMVAValueMapProducer.src = cms.InputTag(options['ELECTRON_COLL'])
if not isReleaseAbove(10, 6): # only for CMSSW_10_2
process.electronRegressionValueMapProducer.srcMiniAOD = cms.InputTag(options['ELECTRON_COLL'])
#
# One tag module --> cut based tight 94X V2
#
process.tagEleCutBasedTight = cms.EDProducer('GsfElectronSelectorByValueMap' if options['useAOD'] else 'PatElectronSelectorByValueMap',
input = cms.InputTag("goodElectrons"),
cut = cms.string(options['ELECTRON_TAG_CUTS']),
selection = cms.InputTag("egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-tight"),
id_cut = cms.bool(True)
)
#
# Add many probe modules, use the PatElectronNm1Selector in case we want to check the effect of one cut
#
def addNewProbeModule(sequence, name, inputTag, cutNamesToMask=None):
if cutNamesToMask:
temp = cms.EDProducer('PatElectronNm1Selector',
input = cms.InputTag("goodElectrons"),
cut = cms.string(options['ELECTRON_CUTS']),
selection = cms.InputTag(inputTag),
cutNamesToMask = cutNamesToMask,
)
else:
temp = cms.EDProducer('GsfElectronSelectorByValueMap' if options['useAOD'] else 'PatElectronSelectorByValueMap',
input = cms.InputTag("goodElectrons"),
cut = cms.string(options['ELECTRON_CUTS']),
selection = cms.InputTag(inputTag),
id_cut = cms.bool(True)
)
setattr(process, 'probeEle%s' % name, temp)
sequence += temp
probeSequence = cms.Sequence()
if not options['useAOD']:
addNewProbeModule(probeSequence, 'HLTsafe', 'egmGsfElectronIDs:cutBasedElectronHLTPreselection-Summer16-V1')
addNewProbeModule(probeSequence, 'DoubleEleHLTsafe', 'egmGsfElectronIDs:cutBasedDoubleElectronHLTPreselection-Summer16-V1')
for wp in ['Veto', 'Loose', 'Medium', 'Tight']:
addNewProbeModule(probeSequence, 'CutBased%s80X' % wp, 'egmGsfElectronIDs:cutBasedElectronID-Summer16-80X-V1-%s' % wp.lower())
addNewProbeModule(probeSequence, 'CutBased%s94X' % wp, 'egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V1-%s' % wp.lower())
addNewProbeModule(probeSequence, 'CutBased%s94XV2' % wp, 'egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-%s' % wp.lower())
for wp in ['wp80', 'wp90']:
addNewProbeModule(probeSequence, 'MVA80X%s' %wp, 'egmGsfElectronIDs:mvaEleID-Spring16-GeneralPurpose-V1-%s' % wp)
for wp in ['wp80', 'wp90', 'wpLoose']:
addNewProbeModule(probeSequence, 'MVA94X%snoiso' %wp, 'egmGsfElectronIDs:mvaEleID-Fall17-noIso-V1-%s' % wp)
addNewProbeModule(probeSequence, 'MVA94X%siso' %wp, 'egmGsfElectronIDs:mvaEleID-Fall17-iso-V1-%s' % wp)
addNewProbeModule(probeSequence, 'MVA94X%snoisoV2' %wp, 'egmGsfElectronIDs:mvaEleID-Fall17-noIso-V2-%s' % wp)
addNewProbeModule(probeSequence, 'MVA94X%sisoV2' %wp, 'egmGsfElectronIDs:mvaEleID-Fall17-iso-V2-%s' % wp)
addNewProbeModule(probeSequence, 'MVA94XwpHZZisoV2', 'egmGsfElectronIDs:mvaEleID-Fall17-iso-V2-wpHZZ')
#
# For cut based 94X V2, also check partial cuts
#
allCuts = ["MinPt", "GsfEleSCEtaMultiRange", "GsfEleDEtaInSeed", "GsfEleDPhiIn", "GsfEleFull5x5SigmaIEtaIEta",
"GsfEleHadronicOverEMEnergyScaled", "GsfEleEInverseMinusPInverse", "GsfEleRelPFIsoScaled", "GsfEleConversionVeto", "GsfEleMissingHits"]
for cut in allCuts:
otherCuts = cms.vstring([i + 'Cut_0' for i in allCuts if i!=cut])
for wp in ['Veto', 'Loose', 'Medium', 'Tight']:
addNewProbeModule(probeSequence, 'CutBased%s94XV2%sCut' % (wp, cut), 'egmGsfElectronIDs:cutBasedElectronID-Fall17-94X-V2-%s' % wp.lower(), cutNamesToMask=otherCuts)
#
# Optional: SUSY variables (broken?)
#
if options['addSUSY'] :
from EgammaAnalysis.TnPTreeProducer.electronsExtrasSUSY_cff import workingPoints
for wp in workingPoints: addNewProbeModule(probeSequence, wp, 'susyEleVarHelper:pass' + wp)
return probeSequence