CMS 3D CMS Logo

All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
taus_cff.py
Go to the documentation of this file.
1 import FWCore.ParameterSet.Config as cms
4 from PhysicsTools.JetMCAlgos.TauGenJets_cfi import tauGenJets
5 from PhysicsTools.JetMCAlgos.TauGenJetsDecayModeSelectorAllHadrons_cfi import tauGenJetsSelectorAllHadrons
6 
7 ##################### Updated tau collection with MVA-based tau-Ids rerun #######
8 # Used only in some eras
10 
11 ##################### User floats producers, selectors ##########################
12 
13 
14 finalTaus = cms.EDFilter("PATTauRefSelector",
15  src = cms.InputTag("slimmedTausUpdated"),
16  cut = cms.string("pt > 18 && tauID('decayModeFindingNewDMs') && (tauID('byLooseCombinedIsolationDeltaBetaCorr3Hits') || tauID('byVLooseIsolationMVArun2v1DBoldDMwLT2015') || tauID('byVLooseIsolationMVArun2v1DBnewDMwLT') || tauID('byVLooseIsolationMVArun2v1DBdR03oldDMwLT') || tauID('byVVLooseIsolationMVArun2v1DBoldDMwLT') || tauID('byVVLooseIsolationMVArun2v1DBoldDMwLT2017v2') || tauID('byVVLooseIsolationMVArun2v1DBnewDMwLT2017v2') || tauID('byVVLooseIsolationMVArun2v1DBdR03oldDMwLT2017v2') || tauID('byVVVLooseDeepTau2017v2p1VSjet'))")
17 )
18 
19 for era in [run2_nanoAOD_94XMiniAODv1,]:
20  era.toModify(finalTaus,
21  cut = cms.string("pt > 18 && tauID('decayModeFindingNewDMs') && (tauID('byLooseCombinedIsolationDeltaBetaCorr3Hits') || tauID('byVLooseIsolationMVArun2v1DBoldDMwLT') || tauID('byVLooseIsolationMVArun2v1DBnewDMwLT') || tauID('byVLooseIsolationMVArun2v1DBdR03oldDMwLT') || tauID('byVVLooseIsolationMVArun2v1DBoldDMwLT2017v1') || tauID('byVVLooseIsolationMVArun2v1DBoldDMwLT2017v2') || tauID('byVVLooseIsolationMVArun2v1DBnewDMwLT2017v2') || tauID('byVVLooseIsolationMVArun2v1DBdR03oldDMwLT2017v2') || tauID('byVVVLooseDeepTau2017v2p1VSjet'))")
22  )
23 run2_miniAOD_80XLegacy.toModify(finalTaus,
24  src = cms.InputTag("slimmedTaus"),
25  cut = cms.string("pt > 18 && tauID('decayModeFindingNewDMs') && (tauID('byLooseCombinedIsolationDeltaBetaCorr3Hits') || tauID('byVLooseIsolationMVArun2v1DBoldDMwLT') || tauID('byVLooseIsolationMVArun2v1DBnewDMwLT') || tauID('byVLooseIsolationMVArun2v1DBdR03oldDMwLT'))")
26  )
27 
28 ##################### Tables for final output and docs ##########################
29 def _tauIdWPMask(pattern, choices, doc=""):
30  return Var(" + ".join(["%d * tauID('%s')" % (pow(2,i), pattern % c) for (i,c) in enumerate(choices)]), "uint8",
31  doc=doc+": bitmask "+", ".join(["%d = %s" % (pow(2,i),c) for (i,c) in enumerate(choices)]))
32 def _tauId2WPMask(pattern,doc):
33  return _tauIdWPMask(pattern,choices=("Loose","Tight"),doc=doc)
34 def _tauId3WPMask(pattern,doc):
35  return _tauIdWPMask(pattern,choices=("Loose","Medium","Tight"),doc=doc)
36 def _tauId4WPMask(pattern,doc):
37  return _tauIdWPMask(pattern, choices=("VLoose", "Loose", "Medium", "Tight"), doc=doc)
38 def _tauId5WPMask(pattern,doc):
39  return _tauIdWPMask(pattern,choices=("VLoose","Loose","Medium","Tight","VTight"),doc=doc)
40 def _tauId6WPMask(pattern,doc):
41  return _tauIdWPMask(pattern,choices=("VLoose","Loose","Medium","Tight","VTight","VVTight"),doc=doc)
42 def _tauId7WPMask(pattern,doc):
43  return _tauIdWPMask(pattern,choices=("VVLoose","VLoose","Loose","Medium","Tight","VTight","VVTight"),doc=doc)
44 def _tauId8WPMask(pattern,doc):
45  return _tauIdWPMask(pattern,choices=("VVVLoose","VVLoose","VLoose","Loose","Medium","Tight","VTight","VVTight"),doc=doc)
46 
47 tauTable = cms.EDProducer("SimpleCandidateFlatTableProducer",
48  src = cms.InputTag("linkedObjects","taus"),
49  cut = cms.string(""), #we should not filter on cross linked collections
50  name= cms.string("Tau"),
51  doc = cms.string("slimmedTaus after basic selection (" + finalTaus.cut.value()+")"),
52  singleton = cms.bool(False), # the number of entries is variable
53  extension = cms.bool(False), # this is the main table for the taus
54  variables = cms.PSet() # PSet defined below in era dependent way
55 )
56 _tauVarsBase = cms.PSet(P4Vars,
57  charge = Var("charge", int, doc="electric charge"),
58  jetIdx = Var("?hasUserCand('jet')?userCand('jet').key():-1", int, doc="index of the associated jet (-1 if none)"),
59  decayMode = Var("decayMode()",int),
60  idDecayMode = Var("tauID('decayModeFinding')", bool),
61  idDecayModeNewDMs = Var("tauID('decayModeFindingNewDMs')", bool),
62 
63  leadTkPtOverTauPt = Var("leadChargedHadrCand.pt/pt ",float, doc="pt of the leading track divided by tau pt",precision=10),
64  leadTkDeltaEta = Var("leadChargedHadrCand.eta - eta ",float, doc="eta of the leading track, minus tau eta",precision=8),
65  leadTkDeltaPhi = Var("deltaPhi(leadChargedHadrCand.phi, phi) ",float, doc="phi of the leading track, minus tau phi",precision=8),
66 
67  dxy = Var("leadChargedHadrCand().dxy()",float, doc="d_{xy} of lead track with respect to PV, in cm (with sign)",precision=10),
68  dz = Var("leadChargedHadrCand().dz()",float, doc="d_{z} of lead track with respect to PV, in cm (with sign)",precision=14),
69 
70  # these are too many, we may have to suppress some
71  rawIso = Var( "tauID('byCombinedIsolationDeltaBetaCorrRaw3Hits')", float, doc = "combined isolation (deltaBeta corrections)", precision=10),
72  rawIsodR03 = Var( "(tauID('chargedIsoPtSumdR03')+max(0.,tauID('neutralIsoPtSumdR03')-0.072*tauID('puCorrPtSum')))", float, doc = "combined isolation (deltaBeta corrections, dR=0.3)", precision=10),
73  chargedIso = Var( "tauID('chargedIsoPtSum')", float, doc = "charged isolation", precision=10),
74  neutralIso = Var( "tauID('neutralIsoPtSum')", float, doc = "neutral (photon) isolation", precision=10),
75  puCorr = Var( "tauID('puCorrPtSum')", float, doc = "pileup correction", precision=10),
76  photonsOutsideSignalCone = Var( "tauID('photonPtSumOutsideSignalCone')", float, doc = "sum of photons outside signal cone", precision=10),
77 
78  idAntiMu = _tauId2WPMask("againstMuon%s3", doc= "Anti-muon discriminator V3: "),
79  idAntiEleDeadECal = Var("tauID('againstElectronDeadECALForNano')", bool, doc = "Anti-electron dead-ECal discriminator"),
80 
81 # isoCI3hit = Var( "tauID("byCombinedIsolationDeltaBetaCorrRaw3Hits")" doc="byCombinedIsolationDeltaBetaCorrRaw3Hits raw output discriminator"),
82 # photonOutsideSigCone = Var( "tauID("photonPtSumOutsideSignalCone")" doc="photonPtSumOutsideSignalCone raw output discriminator"),
83 
84 )
85 
86 _mvaIsoVars2015 = cms.PSet(
87  rawMVAnewDM = Var( "tauID('byIsolationMVArun2v1DBnewDMwLTraw')",float, doc="byIsolationMVArun2v1DBoldDMwLT raw output discriminator (2015)",precision=10),
88  rawMVAoldDM = Var( "tauID('byIsolationMVArun2v1DBoldDMwLTraw')",float, doc="byIsolationMVArun2v1DBoldDMwLT raw output discriminator (2015)",precision=10),
89  rawMVAoldDMdR03 = Var( "tauID('byIsolationMVArun2v1DBdR03oldDMwLTraw')",float, doc="byIsolationMVArun2v1DBoldDMwLT raw output discriminator (2015)",precision=10),
90  idMVAnewDM = _tauId6WPMask( "by%sIsolationMVArun2v1DBnewDMwLT", doc="IsolationMVArun2v1DBnewDMwLT ID working point (2015)"),
91  idMVAoldDM = _tauId6WPMask( "by%sIsolationMVArun2v1DBoldDMwLT", doc="IsolationMVArun2v1DBoldDMwLT ID working point (2015)"),
92  idMVAoldDMdR03 = _tauId6WPMask( "by%sIsolationMVArun2v1DBdR03oldDMwLT", doc="IsolationMVArun2v1DBoldDMdR0p3wLT ID working point (2015)")
93 )
94 _mvaIsoVars2015Reduced = cms.PSet(
95  rawMVAoldDM = Var( "tauID('byIsolationMVArun2v1DBoldDMwLTraw2015')",float, doc="byIsolationMVArun2v1DBoldDMwLT raw output discriminator (2015)",precision=10),
96  idMVAoldDM = _tauId6WPMask( "by%sIsolationMVArun2v1DBoldDMwLT2015", doc="IsolationMVArun2v1DBoldDMwLT ID working point (2015)"),
97 )
98 _mvaIsoVars2017v1 = cms.PSet(
99  rawMVAoldDM2017v1 = Var( "tauID('byIsolationMVArun2v1DBoldDMwLTraw')",float, doc="byIsolationMVArun2v1DBoldDMwLT raw output discriminator (2017v1)",precision=10),
100  idMVAoldDM2017v1 = _tauId7WPMask( "by%sIsolationMVArun2v1DBoldDMwLT", doc="IsolationMVArun2v1DBoldDMwLT ID working point (2017v1)")
101 )
102 _mvaIsoVars2017v2 = cms.PSet(
103  rawMVAnewDM2017v2 = Var( "tauID('byIsolationMVArun2v1DBnewDMwLTraw2017v2')",float, doc="byIsolationMVArun2v1DBnewDMwLT raw output discriminator (2017v2)",precision=10),
104  rawMVAoldDM2017v2 = Var( "tauID('byIsolationMVArun2v1DBoldDMwLTraw2017v2')",float, doc="byIsolationMVArun2v1DBoldDMwLT raw output discriminator (2017v2)",precision=10),
105  rawMVAoldDMdR032017v2 = Var( "tauID('byIsolationMVArun2v1DBdR03oldDMwLTraw2017v2')",float, doc="byIsolationMVArun2v1DBdR03oldDMwLT raw output discriminator (2017v2)",precision=10),
106  idMVAnewDM2017v2 = _tauId7WPMask( "by%sIsolationMVArun2v1DBnewDMwLT2017v2", doc="IsolationMVArun2v1DBnewDMwLT ID working point (2017v2)"),
107  idMVAoldDM2017v2 = _tauId7WPMask( "by%sIsolationMVArun2v1DBoldDMwLT2017v2", doc="IsolationMVArun2v1DBoldDMwLT ID working point (2017v2)"),
108  idMVAoldDMdR032017v2 = _tauId7WPMask( "by%sIsolationMVArun2v1DBdR03oldDMwLT2017v2", doc="IsolationMVArun2v1DBoldDMdR0p3wLT ID working point (2017v2)")
109 )
110 _mvaAntiEVars = cms.PSet(
111  rawAntiEle2018 = Var("tauID('againstElectronMVA6Raw2018')", float, doc= "Anti-electron MVA discriminator V6 raw output discriminator (2018)", precision=10),
112  rawAntiEleCat2018 = Var("tauID('againstElectronMVA6category2018')", int, doc="Anti-electron MVA discriminator V6 category (2018)"),
113  idAntiEle2018 = _tauId5WPMask("againstElectron%sMVA62018", doc= "Anti-electron MVA discriminator V6 (2018)"),
114  rawAntiEle = Var("tauID('againstElectronMVA6Raw')", float, doc= "Anti-electron MVA discriminator V6 raw output discriminator (2015)", precision=10),
115  rawAntiEleCat = Var("tauID('againstElectronMVA6category')", int, doc="Anti-electron MVA discriminator V6 category (2015)"),
116  idAntiEle = _tauId5WPMask("againstElectron%sMVA6", doc= "Anti-electron MVA discriminator V6 (2015)"),
117 )
118 _mvaAntiEVars2015 = cms.PSet(
119  rawAntiEle = Var("tauID('againstElectronMVA6Raw')", float, doc= "Anti-electron MVA discriminator V6 raw output discriminator (2015)", precision=10),
120  rawAntiEleCat = Var("tauID('againstElectronMVA6category')", int, doc="Anti-electron MVA discriminator V6 category (2015)"),
121  idAntiEle = _tauId5WPMask("againstElectron%sMVA6", doc= "Anti-electron MVA discriminator V6 (2015)")
122 )
123 _deepTauVars2017v2 = cms.PSet(
124  rawDeepTau2017v2VSe = Var("tauID('byDeepTau2017v2VSeraw')", float, doc="byDeepTau2017v2VSe raw output discriminator (deepTau2017v2)", precision=10),
125  rawDeepTau2017v2VSmu = Var("tauID('byDeepTau2017v2VSmuraw')", float, doc="byDeepTau2017v2VSmu raw output discriminator (deepTau2017v2)", precision=10),
126  rawDeepTau2017v2VSjet = Var("tauID('byDeepTau2017v2VSjetraw')", float, doc="byDeepTau2017v2VSjet raw output discriminator (deepTau2017v2)", precision=10),
127  idDeepTau2017v2VSe = _tauId8WPMask("by%sDeepTau2017v2VSe", doc="byDeepTau2017v2VSe ID working points (deepTau2017v2)"),
128  idDeepTau2017v2VSmu = _tauId4WPMask("by%sDeepTau2017v2VSmu", doc="byDeepTau2017v2VSmu ID working points (deepTau2017v2)"),
129  idDeepTau2017v2VSjet = _tauId8WPMask("by%sDeepTau2017v2VSjet", doc="byDeepTau2017v2VSjet ID working points (deepTau2017v2)"),
130 )
131 _deepTauVars2017v2p1 = cms.PSet(
132  rawDeepTau2017v2p1VSe = Var("tauID('byDeepTau2017v2p1VSeraw')", float, doc="byDeepTau2017v2p1VSe raw output discriminator (deepTau2017v2p1)", precision=10),
133  rawDeepTau2017v2p1VSmu = Var("tauID('byDeepTau2017v2p1VSmuraw')", float, doc="byDeepTau2017v2p1VSmu raw output discriminator (deepTau2017v2p1)", precision=10),
134  rawDeepTau2017v2p1VSjet = Var("tauID('byDeepTau2017v2p1VSjetraw')", float, doc="byDeepTau2017v2p1VSjet raw output discriminator (deepTau2017v2p1)", precision=10),
135  idDeepTau2017v2p1VSe = _tauId8WPMask("by%sDeepTau2017v2p1VSe", doc="byDeepTau2017v2p1VSe ID working points (deepTau2017v2p1)"),
136  idDeepTau2017v2p1VSmu = _tauId4WPMask("by%sDeepTau2017v2p1VSmu", doc="byDeepTau2017v2p1VSmu ID working points (deepTau2017v2p1)"),
137  idDeepTau2017v2p1VSjet = _tauId8WPMask("by%sDeepTau2017v2p1VSjet", doc="byDeepTau2017v2p1VSjet ID working points (deepTau2017v2p1)"),
138 )
139 
140 _variablesMiniV2 = cms.PSet(
141  _tauVarsBase,
142  _mvaAntiEVars,
143  _mvaIsoVars2015Reduced,
144  _mvaIsoVars2017v1,
145  _mvaIsoVars2017v2,
146  _deepTauVars2017v2p1
147 )
148 _variablesMiniV1 = _variablesMiniV2.clone()
149 _variablesMiniV1.rawMVAoldDM = Var( "tauID('byIsolationMVArun2v1DBoldDMwLTraw')",float, doc="byIsolationMVArun2v1DBoldDMwLT raw output discriminator (2015)",precision=10)
150 _variablesMiniV1.rawMVAoldDM2017v1 = Var( "tauID('byIsolationMVArun2v1DBoldDMwLTraw2017v1')",float, doc="byIsolationMVArun2v1DBoldDMwLT raw output discriminator (2017v1)",precision=10)
151 _variablesMiniV1.idMVAoldDM = _tauId6WPMask( "by%sIsolationMVArun2v1DBoldDMwLT", doc="IsolationMVArun2v1DBoldDMwLT ID working point (2015)")
152 _variablesMiniV1.idMVAoldDM2017v1 = _tauId7WPMask( "by%sIsolationMVArun2v1DBoldDMwLT2017v1", doc="IsolationMVArun2v1DBoldDMwLT ID working point (2017v1)")
153 _variables80X = cms.PSet(
154  _tauVarsBase,
155  _mvaAntiEVars2015,
156  _mvaIsoVars2015
157 )
158 
159 tauTable.variables = _variablesMiniV2
160 
161 for era in [run2_nanoAOD_94XMiniAODv1,]:
162  era.toModify(tauTable,
163  variables = _variablesMiniV1
164  )
165 run2_miniAOD_80XLegacy.toModify(tauTable,
166  variables = _variables80X
167 )
168 run2_miniAOD_80XLegacy.toModify(tauTable.variables,
169  idAntiEleDeadECal = None
170 )
171 
172 (run2_nanoAOD_106Xv2 | run2_miniAOD_devel).toModify(tauTable.variables,
173  idAntiEleDeadECal = Var("tauID('againstElectronDeadECAL')", bool, doc = "Anti-electron dead-ECal discriminator"),
174  rawAntiEle2018 = Var("tauID('againstElectronMVA6Raw')", float, doc= "Anti-electron MVA discriminator V6 raw output discriminator (2018)", precision=10),
175  rawAntiEleCat2018 = Var("tauID('againstElectronMVA6category')", int, doc="Anti-electron MVA discriminator V6 category (2018)"),
176  idAntiEle2018 = _tauId5WPMask("againstElectron%sMVA6", doc= "Anti-electron MVA discriminator V6 (2018)"),
177  rawAntiEle = Var("tauID('againstElectronMVA6Raw2015')", float, doc= "Anti-electron MVA discriminator V6 raw output discriminator (2015)", precision=10),
178  rawAntiEleCat = Var("tauID('againstElectronMVA6category2015')", int, doc="Anti-electron MVA discriminator V6 category (2015"),
179  idAntiEle = _tauId5WPMask("againstElectron%sMVA62015", doc= "Anti-electron MVA discriminator V6 (2015)")
180 )
181 
182 tauGenJets.GenParticles = cms.InputTag("prunedGenParticles")
183 tauGenJets.includeNeutrinos = cms.bool(False)
184 
185 genVisTaus = cms.EDProducer("GenVisTauProducer",
186  src = cms.InputTag("tauGenJetsSelectorAllHadrons"),
187  srcGenParticles = cms.InputTag("prunedGenParticles")
188 )
189 
190 genVisTauTable = cms.EDProducer("SimpleCandidateFlatTableProducer",
191  src = cms.InputTag("genVisTaus"),
192  cut = cms.string("pt > 10."),
193  name = cms.string("GenVisTau"),
194  doc = cms.string("gen hadronic taus "),
195  singleton = cms.bool(False), # the number of entries is variable
196  extension = cms.bool(False), # this is the main table for generator level hadronic tau decays
197  variables = cms.PSet(
198  pt = Var("pt", float,precision=8),
199  phi = Var("phi", float,precision=8),
200  eta = Var("eta", float,precision=8),
201  mass = Var("mass", float,precision=8),
202  charge = Var("charge", int),
203  status = Var("status", int, doc="Hadronic tau decay mode. 0=OneProng0PiZero, 1=OneProng1PiZero, 2=OneProng2PiZero, 10=ThreeProng0PiZero, 11=ThreeProng1PiZero, 15=Other"),
204  genPartIdxMother = Var("?numberOfMothers>0?motherRef(0).key():-1", int, doc="index of the mother particle"),
205  )
206 )
207 
208 tausMCMatchLepTauForTable = cms.EDProducer("MCMatcher", # cut on deltaR, deltaPt/Pt; pick best by deltaR
209  src = tauTable.src, # final reco collection
210  matched = cms.InputTag("finalGenParticles"), # final mc-truth particle collection
211  mcPdgId = cms.vint32(11,13), # one or more PDG ID (11 = electron, 13 = muon); absolute values (see below)
212  checkCharge = cms.bool(False), # True = require RECO and MC objects to have the same charge
213  mcStatus = cms.vint32(), # PYTHIA status code (1 = stable, 2 = shower, 3 = hard scattering)
214  maxDeltaR = cms.double(0.3), # Minimum deltaR for the match
215  maxDPtRel = cms.double(0.5), # Minimum deltaPt/Pt for the match
216  resolveAmbiguities = cms.bool(True), # Forbid two RECO objects to match to the same GEN object
217  resolveByMatchQuality = cms.bool(True), # False = just match input in order; True = pick lowest deltaR pair first
218 )
219 
220 tausMCMatchHadTauForTable = cms.EDProducer("MCMatcher", # cut on deltaR, deltaPt/Pt; pick best by deltaR
221  src = tauTable.src, # final reco collection
222  matched = cms.InputTag("genVisTaus"), # generator level hadronic tau decays
223  mcPdgId = cms.vint32(15), # one or more PDG ID (15 = tau); absolute values (see below)
224  checkCharge = cms.bool(False), # True = require RECO and MC objects to have the same charge
225  mcStatus = cms.vint32(), # CV: no *not* require certain status code for matching (status code corresponds to decay mode for hadronic tau decays)
226  maxDeltaR = cms.double(0.3), # Maximum deltaR for the match
227  maxDPtRel = cms.double(1.), # Maximum deltaPt/Pt for the match
228  resolveAmbiguities = cms.bool(True), # Forbid two RECO objects to match to the same GEN object
229  resolveByMatchQuality = cms.bool(True), # False = just match input in order; True = pick lowest deltaR pair first
230 )
231 
232 tauMCTable = cms.EDProducer("CandMCMatchTableProducer",
233  src = tauTable.src,
234  mcMap = cms.InputTag("tausMCMatchLepTauForTable"),
235  mcMapVisTau = cms.InputTag("tausMCMatchHadTauForTable"),
236  objName = tauTable.name,
237  objType = tauTable.name, #cms.string("Tau"),
238  branchName = cms.string("genPart"),
239  docString = cms.string("MC matching to status==2 taus"),
240 )
241 
242 
243 tauSequence = cms.Sequence(patTauMVAIDsSeq + finalTaus)
244 _tauSequence80X = cms.Sequence(finalTaus)
245 run2_miniAOD_80XLegacy.toReplaceWith(tauSequence,_tauSequence80X)
246 tauTables = cms.Sequence(tauTable)
247 tauMC = cms.Sequence(tauGenJets + tauGenJetsSelectorAllHadrons + genVisTaus + genVisTauTable + tausMCMatchLepTauForTable + tausMCMatchHadTauForTable + tauMCTable)
248 
def _tauId7WPMask(pattern, doc)
Definition: taus_cff.py:42
def _tauId5WPMask(pattern, doc)
Definition: taus_cff.py:38
def _tauId4WPMask(pattern, doc)
Definition: taus_cff.py:36
Updated tau collection with MVA-based tau-Ids rerun ####### Used only in some eras.
def Var(expr, valtype, compression=None, doc=None, mcOnly=False, precision=-1)
Definition: common_cff.py:20
def _tauId3WPMask(pattern, doc)
Definition: taus_cff.py:34
def _tauId6WPMask(pattern, doc)
Definition: taus_cff.py:40
def _tauIdWPMask(pattern, choices, doc="")
Tables for final output and docs ##########################.
Definition: taus_cff.py:29
static std::string join(char **cmd)
Definition: RemoteFile.cc:18
def _tauId2WPMask(pattern, doc)
Definition: taus_cff.py:32
def _tauId8WPMask(pattern, doc)
Definition: taus_cff.py:44
Power< A, B >::type pow(const A &a, const B &b)
Definition: Power.h:40