CMS 3D CMS Logo
jets_cff.py
Go to the documentation of this file.
1 import FWCore.ParameterSet.Config as cms
2 from PhysicsTools.NanoAOD.nano_eras_cff import *
3 
4 from PhysicsTools.NanoAOD.common_cff import *
5 from RecoJets.JetProducers.ak4PFJetsBetaStar_cfi import *
6 
7 ##################### User floats producers, selectors ##########################
8 from RecoJets.JetProducers.ak4PFJets_cfi import ak4PFJets
9 
10 chsForSATkJets = cms.EDFilter("CandPtrSelector", src = cms.InputTag("packedPFCandidates"), cut = cms.string('charge()!=0 && pvAssociationQuality()>=5 && vertexRef().key()==0'))
11 softActivityJets = ak4PFJets.clone(src = 'chsForSATkJets', doAreaFastjet = False, jetPtMin=1)
12 softActivityJets10 = cms.EDFilter("CandPtrSelector", src = cms.InputTag("softActivityJets"), cut = cms.string('pt>10'))
13 softActivityJets5 = cms.EDFilter("CandPtrSelector", src = cms.InputTag("softActivityJets"), cut = cms.string('pt>5'))
14 softActivityJets2 = cms.EDFilter("CandPtrSelector", src = cms.InputTag("softActivityJets"), cut = cms.string('pt>2'))
15 
16 from PhysicsTools.PatAlgos.recoLayer0.jetCorrFactors_cfi import *
17 # Note: Safe to always add 'L2L3Residual' as MC contains dummy L2L3Residual corrections (always set to 1)
18 # (cf. https://twiki.cern.ch/twiki/bin/view/CMSPublic/WorkBookJetEnergyCorrections#CMSSW_7_6_4_and_above )
19 jetCorrFactorsNano = patJetCorrFactors.clone(src='slimmedJets',
20  levels = cms.vstring('L1FastJet',
21  'L2Relative',
22  'L3Absolute',
23  'L2L3Residual'),
24  primaryVertices = cms.InputTag("offlineSlimmedPrimaryVertices"),
25 )
26 jetCorrFactorsAK8 = patJetCorrFactors.clone(src='slimmedJetsAK8',
27  levels = cms.vstring('L1FastJet',
28  'L2Relative',
29  'L3Absolute',
30  'L2L3Residual'),
31  payload = cms.string('AK8PFPuppi'),
32  primaryVertices = cms.InputTag("offlineSlimmedPrimaryVertices"),
33 )
34 run2_miniAOD_80XLegacy.toModify(jetCorrFactorsAK8, payload = cms.string('AK8PFchs')) # ak8PFJetsCHS in 2016 80X miniAOD
35 
36 from PhysicsTools.PatAlgos.producersLayer1.jetUpdater_cfi import *
37 
38 updatedJets = updatedPatJets.clone(
39  addBTagInfo=False,
40  jetSource='slimmedJets',
41  jetCorrFactorsSource=cms.VInputTag(cms.InputTag("jetCorrFactorsNano") ),
42 )
43 
44 updatedJetsAK8 = updatedPatJets.clone(
45  addBTagInfo=False,
46  jetSource='slimmedJetsAK8',
47  jetCorrFactorsSource=cms.VInputTag(cms.InputTag("jetCorrFactorsAK8") ),
48 )
49 
50 
51 looseJetId = cms.EDProducer("PatJetIDValueMapProducer",
52  filterParams=cms.PSet(
53  version = cms.string('WINTER16'),
54  quality = cms.string('LOOSE'),
55  ),
56  src = cms.InputTag("updatedJets")
57 )
58 tightJetId = cms.EDProducer("PatJetIDValueMapProducer",
59  filterParams=cms.PSet(
60  version = cms.string('SUMMER18'),
61  quality = cms.string('TIGHT'),
62  ),
63  src = cms.InputTag("updatedJets")
64 )
65 tightJetIdLepVeto = cms.EDProducer("PatJetIDValueMapProducer",
66  filterParams=cms.PSet(
67  version = cms.string('SUMMER18'),
68  quality = cms.string('TIGHTLEPVETO'),
69  ),
70  src = cms.InputTag("updatedJets")
71 )
72 run2_jme_2016.toModify( tightJetId.filterParams, version = "WINTER16" )
73 run2_jme_2016.toModify( tightJetIdLepVeto.filterParams, version = "WINTER16" )
74 run2_jme_2017.toModify( tightJetId.filterParams, version = "WINTER17" )
75 run2_jme_2017.toModify( tightJetIdLepVeto.filterParams, version = "WINTER17" )
76 for modifier in run2_nanoAOD_106Xv1, run2_nanoAOD_106Xv2, run2_miniAOD_devel:
77  modifier.toModify( tightJetId.filterParams, version = "RUN2ULCHS" )
78  modifier.toModify( tightJetIdLepVeto.filterParams, version = "RUN2ULCHS" )
79 
80 
81 looseJetIdAK8 = cms.EDProducer("PatJetIDValueMapProducer",
82  filterParams=cms.PSet(
83  version = cms.string('WINTER16'),
84  quality = cms.string('LOOSE'),
85  ),
86  src = cms.InputTag("updatedJetsAK8")
87 )
88 tightJetIdAK8 = cms.EDProducer("PatJetIDValueMapProducer",
89  filterParams=cms.PSet(
90  version = cms.string('SUMMER18PUPPI'),
91  quality = cms.string('TIGHT'),
92  ),
93  src = cms.InputTag("updatedJetsAK8")
94 )
95 tightJetIdLepVetoAK8 = cms.EDProducer("PatJetIDValueMapProducer",
96  filterParams=cms.PSet(
97  version = cms.string('SUMMER18PUPPI'),
98  quality = cms.string('TIGHTLEPVETO'),
99  ),
100  src = cms.InputTag("updatedJetsAK8")
101 )
102 run2_jme_2016.toModify( tightJetIdAK8.filterParams, version = "WINTER16" )
103 run2_jme_2016.toModify( tightJetIdLepVetoAK8.filterParams, version = "WINTER16" )
104 run2_jme_2017.toModify( tightJetIdAK8.filterParams, version = "WINTER17PUPPI" )
105 run2_jme_2017.toModify( tightJetIdLepVetoAK8.filterParams, version = "WINTER17PUPPI" )
106 for modifier in run2_nanoAOD_106Xv1, run2_nanoAOD_106Xv2, run2_miniAOD_devel:
107  modifier.toModify( tightJetIdAK8.filterParams, version = "RUN2ULPUPPI" )
108  modifier.toModify( tightJetIdLepVetoAK8.filterParams, version = "RUN2ULPUPPI" )
109 
110 
111 bJetVars = cms.EDProducer("JetRegressionVarProducer",
112  pvsrc = cms.InputTag("offlineSlimmedPrimaryVertices"),
113  src = cms.InputTag("updatedJets"),
114  svsrc = cms.InputTag("slimmedSecondaryVertices"),
115  gpsrc = cms.InputTag("prunedGenParticles"),
116  #musrc = cms.InputTag("slimmedMuons"),
117  #elesrc = cms.InputTag("slimmedElectrons")
118 )
119 
120 jercVars = cms.EDProducer("BetaStarPackedCandidateVarProducer",
121  srcJet = cms.InputTag("updatedJets"),
122  srcPF = cms.InputTag("packedPFCandidates"),
123  maxDR = cms.double(0.4)
124 )
125 
126 updatedJetsWithUserData = cms.EDProducer("PATJetUserDataEmbedder",
127  src = cms.InputTag("updatedJets"),
128  userFloats = cms.PSet(
129  leadTrackPt = cms.InputTag("bJetVars:leadTrackPt"),
130  leptonPtRel = cms.InputTag("bJetVars:leptonPtRel"),
131  leptonPtRatio = cms.InputTag("bJetVars:leptonPtRatio"),
132  leptonPtRelInv = cms.InputTag("bJetVars:leptonPtRelInv"),
133  leptonPtRelv0 = cms.InputTag("bJetVars:leptonPtRelv0"),
134  leptonPtRatiov0 = cms.InputTag("bJetVars:leptonPtRatiov0"),
135  leptonPtRelInvv0 = cms.InputTag("bJetVars:leptonPtRelInvv0"),
136  leptonDeltaR = cms.InputTag("bJetVars:leptonDeltaR"),
137  leptonPt = cms.InputTag("bJetVars:leptonPt"),
138  vtxPt = cms.InputTag("bJetVars:vtxPt"),
139  vtxMass = cms.InputTag("bJetVars:vtxMass"),
140  vtx3dL = cms.InputTag("bJetVars:vtx3dL"),
141  vtx3deL = cms.InputTag("bJetVars:vtx3deL"),
142  ptD = cms.InputTag("bJetVars:ptD"),
143  genPtwNu = cms.InputTag("bJetVars:genPtwNu"),
144  qgl = cms.InputTag('qgtagger:qgLikelihood'),
145  puId94XDisc = cms.InputTag('pileupJetId94X:fullDiscriminant'),
146  puId102XDisc = cms.InputTag('pileupJetId102X:fullDiscriminant'),
147  puId106XUL17Disc = cms.InputTag('pileupJetId106XUL17:fullDiscriminant'),
148  chFPV0EF = cms.InputTag("jercVars:chargedFromPV0EnergyFraction"),
149  chFPV1EF = cms.InputTag("jercVars:chargedFromPV1EnergyFraction"),
150  chFPV2EF = cms.InputTag("jercVars:chargedFromPV2EnergyFraction"),
151  chFPV3EF = cms.InputTag("jercVars:chargedFromPV3EnergyFraction"),
152  ),
153  userInts = cms.PSet(
154  tightId = cms.InputTag("tightJetId"),
155  tightIdLepVeto = cms.InputTag("tightJetIdLepVeto"),
156  vtxNtrk = cms.InputTag("bJetVars:vtxNtrk"),
157  leptonPdgId = cms.InputTag("bJetVars:leptonPdgId"),
158  puId106XUL17Id = cms.InputTag('pileupJetId106XUL17:fullId'),
159  ),
160 )
161 run2_jme_2016.toModify(updatedJetsWithUserData.userInts,
162  looseId = cms.InputTag("looseJetId"),
163 )
164 
165 updatedJetsAK8WithUserData = cms.EDProducer("PATJetUserDataEmbedder",
166  src = cms.InputTag("updatedJetsAK8"),
167  userInts = cms.PSet(
168  tightId = cms.InputTag("tightJetIdAK8"),
169  tightIdLepVeto = cms.InputTag("tightJetIdLepVetoAK8"),
170  ),
171 )
172 run2_jme_2016.toModify(updatedJetsAK8WithUserData.userInts,
173  looseId = cms.InputTag("looseJetIdAK8"),
174 )
175 
176 
177 finalJets = cms.EDFilter("PATJetRefSelector",
178  src = cms.InputTag("updatedJetsWithUserData"),
179  cut = cms.string("pt > 15")
180 )
181 
182 finalJetsAK8 = cms.EDFilter("PATJetRefSelector",
183  src = cms.InputTag("updatedJetsAK8WithUserData"),
184  cut = cms.string("pt > 170")
185 )
186 
187 lepInJetVars = cms.EDProducer("LepInJetProducer",
188  src = cms.InputTag("updatedJetsAK8WithUserData"),
189  srcEle = cms.InputTag("finalElectrons"),
190  srcMu = cms.InputTag("finalMuons")
191 )
192 
193 
194 
195 ##################### Tables for final output and docs ##########################
196 
197 
198 
199 jetTable = cms.EDProducer("SimpleCandidateFlatTableProducer",
200  src = cms.InputTag("linkedObjects","jets"),
201  cut = cms.string(""), #we should not filter on cross linked collections
202  name = cms.string("Jet"),
203  doc = cms.string("slimmedJets, i.e. ak4 PFJets CHS with JECs applied, after basic selection (" + finalJets.cut.value()+")"),
204  singleton = cms.bool(False), # the number of entries is variable
205  extension = cms.bool(False), # this is the main table for the jets
206  externalVariables = cms.PSet(
207  bRegCorr = ExtVar(cms.InputTag("bjetNN:corr"),float, doc="pt correction for b-jet energy regression",precision=10),
208  bRegRes = ExtVar(cms.InputTag("bjetNN:res"),float, doc="res on pt corrected with b-jet regression",precision=6),
209  cRegCorr = ExtVar(cms.InputTag("cjetNN:corr"),float, doc="pt correction for c-jet energy regression",precision=10),
210  cRegRes = ExtVar(cms.InputTag("cjetNN:res"),float, doc="res on pt corrected with c-jet regression",precision=6),
211  ),
212  variables = cms.PSet(P4Vars,
213  area = Var("jetArea()", float, doc="jet catchment area, for JECs",precision=10),
214  nMuons = Var("?hasOverlaps('muons')?overlaps('muons').size():0", int, doc="number of muons in the jet"),
215  muonIdx1 = Var("?overlaps('muons').size()>0?overlaps('muons')[0].key():-1", int, doc="index of first matching muon"),
216  muonIdx2 = Var("?overlaps('muons').size()>1?overlaps('muons')[1].key():-1", int, doc="index of second matching muon"),
217  electronIdx1 = Var("?overlaps('electrons').size()>0?overlaps('electrons')[0].key():-1", int, doc="index of first matching electron"),
218  electronIdx2 = Var("?overlaps('electrons').size()>1?overlaps('electrons')[1].key():-1", int, doc="index of second matching electron"),
219  nElectrons = Var("?hasOverlaps('electrons')?overlaps('electrons').size():0", int, doc="number of electrons in the jet"),
220  btagDeepB = Var("?(bDiscriminator('pfDeepCSVJetTags:probb')+bDiscriminator('pfDeepCSVJetTags:probbb'))>=0?bDiscriminator('pfDeepCSVJetTags:probb')+bDiscriminator('pfDeepCSVJetTags:probbb'):-1",float,doc="DeepCSV b+bb tag discriminator",precision=10),
221  btagDeepFlavB = Var("bDiscriminator('pfDeepFlavourJetTags:probb')+bDiscriminator('pfDeepFlavourJetTags:probbb')+bDiscriminator('pfDeepFlavourJetTags:problepb')",float,doc="DeepJet b+bb+lepb tag discriminator",precision=10),
222  btagCSVV2 = Var("bDiscriminator('pfCombinedInclusiveSecondaryVertexV2BJetTags')",float,doc=" pfCombinedInclusiveSecondaryVertexV2 b-tag discriminator (aka CSVV2)",precision=10),
223  btagDeepCvL = Var("?bDiscriminator('pfDeepCSVJetTags:probc')>=0?bDiscriminator('pfDeepCSVJetTags:probc')/(bDiscriminator('pfDeepCSVJetTags:probc')+bDiscriminator('pfDeepCSVJetTags:probudsg')):-1", float,doc="DeepCSV c vs udsg discriminator",precision=10),
224  btagDeepCvB = Var("?bDiscriminator('pfDeepCSVJetTags:probc')>=0?bDiscriminator('pfDeepCSVJetTags:probc')/(bDiscriminator('pfDeepCSVJetTags:probc')+bDiscriminator('pfDeepCSVJetTags:probb')+bDiscriminator('pfDeepCSVJetTags:probbb')):-1",float,doc="DeepCSV c vs b+bb discriminator",precision=10),
225  btagDeepFlavCvL = Var("?(bDiscriminator('pfDeepFlavourJetTags:probc')+bDiscriminator('pfDeepFlavourJetTags:probuds')+bDiscriminator('pfDeepFlavourJetTags:probg'))>0?bDiscriminator('pfDeepFlavourJetTags:probc')/(bDiscriminator('pfDeepFlavourJetTags:probc')+bDiscriminator('pfDeepFlavourJetTags:probuds')+bDiscriminator('pfDeepFlavourJetTags:probg')):-1",float,doc="DeepJet c vs uds+g discriminator",precision=10),
226  btagDeepFlavCvB = Var("?(bDiscriminator('pfDeepFlavourJetTags:probc')+bDiscriminator('pfDeepFlavourJetTags:probb')+bDiscriminator('pfDeepFlavourJetTags:probbb')+bDiscriminator('pfDeepFlavourJetTags:problepb'))>0?bDiscriminator('pfDeepFlavourJetTags:probc')/(bDiscriminator('pfDeepFlavourJetTags:probc')+bDiscriminator('pfDeepFlavourJetTags:probb')+bDiscriminator('pfDeepFlavourJetTags:probbb')+bDiscriminator('pfDeepFlavourJetTags:problepb')):-1",float,doc="DeepJet c vs b+bb+lepb discriminator",precision=10),
227  btagDeepFlavQG = Var("?(bDiscriminator('pfDeepFlavourJetTags:probg')+bDiscriminator('pfDeepFlavourJetTags:probuds'))>0?bDiscriminator('pfDeepFlavourJetTags:probg')/(bDiscriminator('pfDeepFlavourJetTags:probg')+bDiscriminator('pfDeepFlavourJetTags:probuds')):-1",float,doc="DeepJet g vs uds discriminator",precision=10),
228  puIdDisc = Var("userFloat('puId102XDisc')",float,doc="Pileup ID discriminant with 102X (2018) training",precision=10),
229  puId = Var("userInt('pileupJetId:fullId')",int,doc="Pileup ID flags for pre-UL trainings"),
230  jetId = Var("userInt('tightId')*2+4*userInt('tightIdLepVeto')",int,doc="Jet ID flags bit1 is loose (always false in 2017 since it does not exist), bit2 is tight, bit3 is tightLepVeto"),
231  qgl = Var("userFloat('qgl')",float,doc="Quark vs Gluon likelihood discriminator",precision=10),
232  nConstituents = Var("numberOfDaughters()",int,doc="Number of particles in the jet"),
233  rawFactor = Var("1.-jecFactor('Uncorrected')",float,doc="1 - Factor to get back to raw pT",precision=6),
234  chHEF = Var("chargedHadronEnergyFraction()", float, doc="charged Hadron Energy Fraction", precision= 6),
235  neHEF = Var("neutralHadronEnergyFraction()", float, doc="neutral Hadron Energy Fraction", precision= 6),
236  chEmEF = Var("chargedEmEnergyFraction()", float, doc="charged Electromagnetic Energy Fraction", precision= 6),
237  neEmEF = Var("neutralEmEnergyFraction()", float, doc="neutral Electromagnetic Energy Fraction", precision= 6),
238  muEF = Var("muonEnergyFraction()", float, doc="muon Energy Fraction", precision= 6),
239  chFPV0EF = Var("userFloat('chFPV0EF')", float, doc="charged fromPV==0 Energy Fraction (energy excluded from CHS jets). Previously called betastar.", precision= 6),
240  chFPV1EF = Var("userFloat('chFPV1EF')", float, doc="charged fromPV==1 Energy Fraction (component of the total charged Energy Fraction).", precision= 6),
241  chFPV2EF = Var("userFloat('chFPV2EF')", float, doc="charged fromPV==2 Energy Fraction (component of the total charged Energy Fraction).", precision= 6),
242  chFPV3EF = Var("userFloat('chFPV3EF')", float, doc="charged fromPV==3 Energy Fraction (component of the total charged Energy Fraction).", precision= 6),
243  )
244 )
245 
246 #jets are not as precise as muons
247 jetTable.variables.pt.precision=10
248 
249 ### Era dependent customization
250 for modifier in run2_miniAOD_80XLegacy, run2_nanoAOD_94X2016, run2_nanoAOD_94XMiniAODv1, run2_nanoAOD_94XMiniAODv2, run2_nanoAOD_102Xv1, run2_nanoAOD_106Xv1:
251  # Deprecated after 106X
252  modifier.toModify(jetTable.variables,
253  btagCMVA = Var("bDiscriminator('pfCombinedMVAV2BJetTags')",float,doc="CMVA V2 btag discriminator",precision=10),
254  btagDeepC = Var("bDiscriminator('pfDeepCSVJetTags:probc')",float,doc="DeepCSV charm btag discriminator",precision=10),
255  btagDeepFlavC = Var("bDiscriminator('pfDeepFlavourJetTags:probc')",float,doc="DeepFlavour charm tag discriminator",precision=10),
256  )
257 for modifier in run2_miniAOD_80XLegacy, run2_nanoAOD_94X2016:
258  modifier.toModify( jetTable.variables, jetId = Var("userInt('tightIdLepVeto')*4+userInt('tightId')*2+userInt('looseId')",int,doc="Jet ID flags bit1 is loose, bit2 is tight, bit3 is tightLepVeto"))
259 run2_jme_2016.toModify( jetTable.variables, puIdDisc = Var("userFloat('pileupJetId:fullDiscriminant')",float,doc="Pileup ID discriminant with 80X (2016) training",precision=10))
260 run2_jme_2017.toModify( jetTable.variables, puId = Var("userInt('puId106XUL17Id')", int,doc="Pileup ID flags with 106X (2017) training"))
261 run2_jme_2017.toModify( jetTable.variables, puIdDisc = Var("userFloat('puId106XUL17Disc')", float,doc="Pileup ID discriminant with 106X (2017) training",precision=10))
262 for modifier in run2_nanoAOD_94XMiniAODv1, run2_nanoAOD_94XMiniAODv2:
263  modifier.toModify( jetTable.variables, puIdDisc = Var("userFloat('puId94XDisc')", float,doc="Pileup ID discriminant with 94X (2017) training",precision=10))
264  modifier.toModify( jetTable.variables, puId = Var("userInt('pileupJetId:fullId')",int,doc="Pileup ID flags for 2016/2017/2018 EOY trainings"))
265 
266 
267 bjetNN= cms.EDProducer("BJetEnergyRegressionMVA",
268  backend = cms.string("TF"),
269  src = cms.InputTag("linkedObjects","jets"),
270  pvsrc = cms.InputTag("offlineSlimmedPrimaryVertices"),
271  svsrc = cms.InputTag("slimmedSecondaryVertices"),
272  rhosrc = cms.InputTag("fixedGridRhoFastjetAll"),
273 
274  weightFile = cms.FileInPath("PhysicsTools/NanoAOD/data/breg_training_2018.pb"),
275  name = cms.string("JetRegNN"),
276  isClassifier = cms.bool(False),
277  variablesOrder = cms.vstring(["Jet_pt","Jet_eta","rho","Jet_mt","Jet_leadTrackPt","Jet_leptonPtRel","Jet_leptonDeltaR","Jet_neHEF",
278  "Jet_neEmEF","Jet_vtxPt","Jet_vtxMass","Jet_vtx3dL","Jet_vtxNtrk","Jet_vtx3deL",
279  "Jet_numDaughters_pt03","Jet_energyRing_dR0_em_Jet_rawEnergy","Jet_energyRing_dR1_em_Jet_rawEnergy",
280  "Jet_energyRing_dR2_em_Jet_rawEnergy","Jet_energyRing_dR3_em_Jet_rawEnergy","Jet_energyRing_dR4_em_Jet_rawEnergy",
281  "Jet_energyRing_dR0_neut_Jet_rawEnergy","Jet_energyRing_dR1_neut_Jet_rawEnergy","Jet_energyRing_dR2_neut_Jet_rawEnergy",
282  "Jet_energyRing_dR3_neut_Jet_rawEnergy","Jet_energyRing_dR4_neut_Jet_rawEnergy","Jet_energyRing_dR0_ch_Jet_rawEnergy",
283  "Jet_energyRing_dR1_ch_Jet_rawEnergy","Jet_energyRing_dR2_ch_Jet_rawEnergy","Jet_energyRing_dR3_ch_Jet_rawEnergy",
284  "Jet_energyRing_dR4_ch_Jet_rawEnergy","Jet_energyRing_dR0_mu_Jet_rawEnergy","Jet_energyRing_dR1_mu_Jet_rawEnergy",
285  "Jet_energyRing_dR2_mu_Jet_rawEnergy","Jet_energyRing_dR3_mu_Jet_rawEnergy","Jet_energyRing_dR4_mu_Jet_rawEnergy",
286  "Jet_chHEF","Jet_chEmEF","Jet_leptonPtRelInv","isEle","isMu","isOther","Jet_mass","Jet_ptd"]),
287  variables = cms.PSet(
288  Jet_pt = cms.string("pt*jecFactor('Uncorrected')"),
289  Jet_mt = cms.string("mt*jecFactor('Uncorrected')"),
290  Jet_eta = cms.string("eta"),
291  Jet_mass = cms.string("mass*jecFactor('Uncorrected')"),
292  Jet_ptd = cms.string("userFloat('ptD')"),
293  Jet_leadTrackPt = cms.string("userFloat('leadTrackPt')"),
294  Jet_vtxNtrk = cms.string("userInt('vtxNtrk')"),
295  Jet_vtxMass = cms.string("userFloat('vtxMass')"),
296  Jet_vtx3dL = cms.string("userFloat('vtx3dL')"),
297  Jet_vtx3deL = cms.string("userFloat('vtx3deL')"),
298  Jet_vtxPt = cms.string("userFloat('vtxPt')"),
299  #Jet_leptonPt = cms.string("userFloat('leptonPt')"),
300  Jet_leptonPtRel = cms.string("userFloat('leptonPtRelv0')"),
301  Jet_leptonPtRelInv = cms.string("userFloat('leptonPtRelInvv0')*jecFactor('Uncorrected')"),
302  Jet_leptonDeltaR = cms.string("userFloat('leptonDeltaR')"),
303  #Jet_leptonPdgId = cms.string("userInt('leptonPdgId')"),
304  Jet_neHEF = cms.string("neutralHadronEnergyFraction()"),
305  Jet_neEmEF = cms.string("neutralEmEnergyFraction()"),
306  Jet_chHEF = cms.string("chargedHadronEnergyFraction()"),
307  Jet_chEmEF = cms.string("chargedEmEnergyFraction()"),
308  isMu = cms.string("?abs(userInt('leptonPdgId'))==13?1:0"),
309  isEle = cms.string("?abs(userInt('leptonPdgId'))==11?1:0"),
310  isOther = cms.string("?userInt('leptonPdgId')==0?1:0"),
311  ),
312  inputTensorName = cms.string("ffwd_inp"),
313  outputTensorName = cms.string("ffwd_out/BiasAdd"),
314  outputNames = cms.vstring(["corr","res"]),
315  outputFormulas = cms.vstring(["at(0)*0.27912887930870056+1.0545977354049683","0.5*(at(2)-at(1))*0.27912887930870056"]),
316  nThreads = cms.uint32(1),
317  singleThreadPool = cms.string("no_threads"),
318 
319 )
320 
321 cjetNN= cms.EDProducer("BJetEnergyRegressionMVA",
322  backend = cms.string("TF"),
323  src = cms.InputTag("linkedObjects","jets"),
324  pvsrc = cms.InputTag("offlineSlimmedPrimaryVertices"),
325  svsrc = cms.InputTag("slimmedSecondaryVertices"),
326  rhosrc = cms.InputTag("fixedGridRhoFastjetAll"),
327 
328  weightFile = cms.FileInPath("PhysicsTools/NanoAOD/data/creg_training_2018.pb"),
329  name = cms.string("JetRegNN"),
330  isClassifier = cms.bool(False),
331  variablesOrder = cms.vstring(["Jet_pt","Jet_eta","rho","Jet_mt","Jet_leadTrackPt","Jet_leptonPtRel","Jet_leptonDeltaR",
332  "Jet_neHEF","Jet_neEmEF","Jet_vtxPt","Jet_vtxMass","Jet_vtx3dL","Jet_vtxNtrk","Jet_vtx3deL",
333  "Jet_numDaughters_pt03","Jet_chEmEF","Jet_chHEF", "Jet_ptd","Jet_mass",
334  "Jet_energyRing_dR0_em_Jet_rawEnergy","Jet_energyRing_dR1_em_Jet_rawEnergy",
335  "Jet_energyRing_dR2_em_Jet_rawEnergy","Jet_energyRing_dR3_em_Jet_rawEnergy","Jet_energyRing_dR4_em_Jet_rawEnergy",
336  "Jet_energyRing_dR0_neut_Jet_rawEnergy","Jet_energyRing_dR1_neut_Jet_rawEnergy","Jet_energyRing_dR2_neut_Jet_rawEnergy",
337  "Jet_energyRing_dR3_neut_Jet_rawEnergy","Jet_energyRing_dR4_neut_Jet_rawEnergy","Jet_energyRing_dR0_ch_Jet_rawEnergy",
338  "Jet_energyRing_dR1_ch_Jet_rawEnergy","Jet_energyRing_dR2_ch_Jet_rawEnergy","Jet_energyRing_dR3_ch_Jet_rawEnergy",
339  "Jet_energyRing_dR4_ch_Jet_rawEnergy","Jet_energyRing_dR0_mu_Jet_rawEnergy","Jet_energyRing_dR1_mu_Jet_rawEnergy",
340  "Jet_energyRing_dR2_mu_Jet_rawEnergy","Jet_energyRing_dR3_mu_Jet_rawEnergy","Jet_energyRing_dR4_mu_Jet_rawEnergy"]),
341  variables = cms.PSet(
342  Jet_pt = cms.string("pt*jecFactor('Uncorrected')"),
343  Jet_mt = cms.string("mt*jecFactor('Uncorrected')"),
344  Jet_eta = cms.string("eta"),
345  Jet_mass = cms.string("mass*jecFactor('Uncorrected')"),
346  Jet_ptd = cms.string("userFloat('ptD')"),
347  Jet_leadTrackPt = cms.string("userFloat('leadTrackPt')"),
348  Jet_vtxNtrk = cms.string("userInt('vtxNtrk')"),
349  Jet_vtxMass = cms.string("userFloat('vtxMass')"),
350  Jet_vtx3dL = cms.string("userFloat('vtx3dL')"),
351  Jet_vtx3deL = cms.string("userFloat('vtx3deL')"),
352  Jet_vtxPt = cms.string("userFloat('vtxPt')"),
353  #Jet_leptonPt = cms.string("userFloat('leptonPt')"),
354  Jet_leptonPtRel = cms.string("userFloat('leptonPtRelv0')"),
355  Jet_leptonPtRelInv = cms.string("userFloat('leptonPtRelInvv0')*jecFactor('Uncorrected')"),
356  Jet_leptonDeltaR = cms.string("userFloat('leptonDeltaR')"),
357  #Jet_leptonPdgId = cms.string("userInt('leptonPdgId')"),
358  Jet_neHEF = cms.string("neutralHadronEnergyFraction()"),
359  Jet_neEmEF = cms.string("neutralEmEnergyFraction()"),
360  Jet_chHEF = cms.string("chargedHadronEnergyFraction()"),
361  Jet_chEmEF = cms.string("chargedEmEnergyFraction()"),
362  isMu = cms.string("?abs(userInt('leptonPdgId'))==13?1:0"),
363  isEle = cms.string("?abs(userInt('leptonPdgId'))==11?1:0"),
364  isOther = cms.string("?userInt('leptonPdgId')==0?1:0"),
365  ),
366  inputTensorName = cms.string("ffwd_inp"),
367  outputTensorName = cms.string("ffwd_out/BiasAdd"),
368  outputNames = cms.vstring(["corr","res"]),
369  outputFormulas = cms.vstring(["at(0)*0.24325256049633026+0.993854820728302","0.5*(at(2)-at(1))*0.24325256049633026"]),
370  nThreads = cms.uint32(1),
371  singleThreadPool = cms.string("no_threads"),
372 )
373 
374 
375 ##### Soft Activity tables
376 saJetTable = cms.EDProducer("SimpleCandidateFlatTableProducer",
377  src = cms.InputTag("softActivityJets"),
378  cut = cms.string(""),
379  maxLen = cms.uint32(6),
380  name = cms.string("SoftActivityJet"),
381  doc = cms.string("jets clustered from charged candidates compatible with primary vertex (" + chsForSATkJets.cut.value()+")"),
382  singleton = cms.bool(False), # the number of entries is variable
383  extension = cms.bool(False), # this is the main table for the jets
384  variables = cms.PSet(P3Vars,
385  )
386 )
387 
388 saJetTable.variables.pt.precision=10
389 saJetTable.variables.eta.precision=8
390 saJetTable.variables.phi.precision=8
391 
392 saTable = cms.EDProducer("GlobalVariablesTableProducer",
393  variables = cms.PSet(
394  SoftActivityJetHT = ExtVar( cms.InputTag("softActivityJets"), "candidatescalarsum", doc = "scalar sum of soft activity jet pt, pt>1" ),
395  SoftActivityJetHT10 = ExtVar( cms.InputTag("softActivityJets10"), "candidatescalarsum", doc = "scalar sum of soft activity jet pt , pt >10" ),
396  SoftActivityJetHT5 = ExtVar( cms.InputTag("softActivityJets5"), "candidatescalarsum", doc = "scalar sum of soft activity jet pt, pt>5" ),
397  SoftActivityJetHT2 = ExtVar( cms.InputTag("softActivityJets2"), "candidatescalarsum", doc = "scalar sum of soft activity jet pt, pt >2" ),
398  SoftActivityJetNjets10 = ExtVar( cms.InputTag("softActivityJets10"), "candidatesize", doc = "number of soft activity jet pt, pt >2" ),
399  SoftActivityJetNjets5 = ExtVar( cms.InputTag("softActivityJets5"), "candidatesize", doc = "number of soft activity jet pt, pt >5" ),
400  SoftActivityJetNjets2 = ExtVar( cms.InputTag("softActivityJets2"), "candidatesize", doc = "number of soft activity jet pt, pt >10" ),
401 
402  )
403 )
404 
405 
406 
407 ## BOOSTED STUFF #################
408 fatJetTable = cms.EDProducer("SimpleCandidateFlatTableProducer",
409  src = cms.InputTag("finalJetsAK8"),
410  cut = cms.string(" pt > 170"), #probably already applied in miniaod
411  name = cms.string("FatJet"),
412  doc = cms.string("slimmedJetsAK8, i.e. ak8 fat jets for boosted analysis"),
413  singleton = cms.bool(False), # the number of entries is variable
414  extension = cms.bool(False), # this is the main table for the jets
415  variables = cms.PSet(P4Vars,
416  jetId = Var("userInt('tightId')*2+4*userInt('tightIdLepVeto')",int,doc="Jet ID flags bit1 is loose (always false in 2017 since it does not exist), bit2 is tight, bit3 is tightLepVeto"),
417  area = Var("jetArea()", float, doc="jet catchment area, for JECs",precision=10),
418  rawFactor = Var("1.-jecFactor('Uncorrected')",float,doc="1 - Factor to get back to raw pT",precision=6),
419  tau1 = Var("userFloat('NjettinessAK8Puppi:tau1')",float, doc="Nsubjettiness (1 axis)",precision=10),
420  tau2 = Var("userFloat('NjettinessAK8Puppi:tau2')",float, doc="Nsubjettiness (2 axis)",precision=10),
421  tau3 = Var("userFloat('NjettinessAK8Puppi:tau3')",float, doc="Nsubjettiness (3 axis)",precision=10),
422  tau4 = Var("userFloat('NjettinessAK8Puppi:tau4')",float, doc="Nsubjettiness (4 axis)",precision=10),
423  n2b1 = Var("userFloat('ak8PFJetsPuppiSoftDropValueMap:nb1AK8PuppiSoftDropN2')", float, doc="N2 with beta=1", precision=10),
424  n3b1 = Var("userFloat('ak8PFJetsPuppiSoftDropValueMap:nb1AK8PuppiSoftDropN3')", float, doc="N3 with beta=1", precision=10),
425  msoftdrop = Var("groomedMass('SoftDropPuppi')",float, doc="Corrected soft drop mass with PUPPI",precision=10),
426  btagDeepB = Var("?(bDiscriminator('pfDeepCSVJetTags:probb')+bDiscriminator('pfDeepCSVJetTags:probbb'))>=0?bDiscriminator('pfDeepCSVJetTags:probb')+bDiscriminator('pfDeepCSVJetTags:probbb'):-1",float,doc="DeepCSV b+bb tag discriminator",precision=10),
427  btagCSVV2 = Var("bDiscriminator('pfCombinedInclusiveSecondaryVertexV2BJetTags')",float,doc=" pfCombinedInclusiveSecondaryVertexV2 b-tag discriminator (aka CSVV2)",precision=10),
428  btagHbb = Var("bDiscriminator('pfBoostedDoubleSecondaryVertexAK8BJetTags')",float,doc="Higgs to BB tagger discriminator",precision=10),
429  btagDDBvLV2 = Var("bDiscriminator('pfMassIndependentDeepDoubleBvLV2JetTags:probHbb')",float,doc="DeepDoubleX V2(mass-decorrelated) discriminator for H(Z)->bb vs QCD",precision=10),
430  btagDDCvLV2 = Var("bDiscriminator('pfMassIndependentDeepDoubleCvLV2JetTags:probHcc')",float,doc="DeepDoubleX V2 (mass-decorrelated) discriminator for H(Z)->cc vs QCD",precision=10),
431  btagDDCvBV2 = Var("bDiscriminator('pfMassIndependentDeepDoubleCvBV2JetTags:probHcc')",float,doc="DeepDoubleX V2 (mass-decorrelated) discriminator for H(Z)->cc vs H(Z)->bb",precision=10),
432  deepTag_TvsQCD = Var("bDiscriminator('pfDeepBoostedDiscriminatorsJetTags:TvsQCD')",float,doc="DeepBoostedJet tagger top vs QCD discriminator",precision=10),
433  deepTag_WvsQCD = Var("bDiscriminator('pfDeepBoostedDiscriminatorsJetTags:WvsQCD')",float,doc="DeepBoostedJet tagger W vs QCD discriminator",precision=10),
434  deepTag_ZvsQCD = Var("bDiscriminator('pfDeepBoostedDiscriminatorsJetTags:ZvsQCD')",float,doc="DeepBoostedJet tagger Z vs QCD discriminator",precision=10),
435  deepTag_H = Var("bDiscriminator('pfDeepBoostedJetTags:probHbb')+bDiscriminator('pfDeepBoostedJetTags:probHcc')+bDiscriminator('pfDeepBoostedJetTags:probHqqqq')",float,doc="DeepBoostedJet tagger H(bb,cc,4q) sum",precision=10),
436  deepTag_QCD = Var("bDiscriminator('pfDeepBoostedJetTags:probQCDbb')+bDiscriminator('pfDeepBoostedJetTags:probQCDcc')+bDiscriminator('pfDeepBoostedJetTags:probQCDb')+bDiscriminator('pfDeepBoostedJetTags:probQCDc')+bDiscriminator('pfDeepBoostedJetTags:probQCDothers')",float,doc="DeepBoostedJet tagger QCD(bb,cc,b,c,others) sum",precision=10),
437  deepTag_QCDothers = Var("bDiscriminator('pfDeepBoostedJetTags:probQCDothers')",float,doc="DeepBoostedJet tagger QCDothers value",precision=10),
438  deepTagMD_TvsQCD = Var("bDiscriminator('pfMassDecorrelatedDeepBoostedDiscriminatorsJetTags:TvsQCD')",float,doc="Mass-decorrelated DeepBoostedJet tagger top vs QCD discriminator",precision=10),
439  deepTagMD_WvsQCD = Var("bDiscriminator('pfMassDecorrelatedDeepBoostedDiscriminatorsJetTags:WvsQCD')",float,doc="Mass-decorrelated DeepBoostedJet tagger W vs QCD discriminator",precision=10),
440  deepTagMD_ZvsQCD = Var("bDiscriminator('pfMassDecorrelatedDeepBoostedDiscriminatorsJetTags:ZvsQCD')",float,doc="Mass-decorrelated DeepBoostedJet tagger Z vs QCD discriminator",precision=10),
441  deepTagMD_ZHbbvsQCD = Var("bDiscriminator('pfMassDecorrelatedDeepBoostedDiscriminatorsJetTags:ZHbbvsQCD')",float,doc="Mass-decorrelated DeepBoostedJet tagger Z/H->bb vs QCD discriminator",precision=10),
442  deepTagMD_ZbbvsQCD = Var("bDiscriminator('pfMassDecorrelatedDeepBoostedDiscriminatorsJetTags:ZbbvsQCD')",float,doc="Mass-decorrelated DeepBoostedJet tagger Z->bb vs QCD discriminator",precision=10),
443  deepTagMD_HbbvsQCD = Var("bDiscriminator('pfMassDecorrelatedDeepBoostedDiscriminatorsJetTags:HbbvsQCD')",float,doc="Mass-decorrelated DeepBoostedJet tagger H->bb vs QCD discriminator",precision=10),
444  deepTagMD_ZHccvsQCD = Var("bDiscriminator('pfMassDecorrelatedDeepBoostedDiscriminatorsJetTags:ZHccvsQCD')",float,doc="Mass-decorrelated DeepBoostedJet tagger Z/H->cc vs QCD discriminator",precision=10),
445  deepTagMD_H4qvsQCD = Var("bDiscriminator('pfMassDecorrelatedDeepBoostedDiscriminatorsJetTags:H4qvsQCD')",float,doc="Mass-decorrelated DeepBoostedJet tagger H->4q vs QCD discriminator",precision=10),
446  deepTagMD_bbvsLight = Var("bDiscriminator('pfMassDecorrelatedDeepBoostedDiscriminatorsJetTags:bbvsLight')",float,doc="Mass-decorrelated DeepBoostedJet tagger Z/H/gluon->bb vs light flavour discriminator",precision=10),
447  deepTagMD_ccvsLight = Var("bDiscriminator('pfMassDecorrelatedDeepBoostedDiscriminatorsJetTags:ccvsLight')",float,doc="Mass-decorrelated DeepBoostedJet tagger Z/H/gluon->cc vs light flavour discriminator",precision=10),
448  particleNet_TvsQCD = Var("bDiscriminator('pfParticleNetDiscriminatorsJetTags:TvsQCD')",float,doc="ParticleNet tagger top vs QCD discriminator",precision=10),
449  particleNet_WvsQCD = Var("bDiscriminator('pfParticleNetDiscriminatorsJetTags:WvsQCD')",float,doc="ParticleNet tagger W vs QCD discriminator",precision=10),
450  particleNet_ZvsQCD = Var("bDiscriminator('pfParticleNetDiscriminatorsJetTags:ZvsQCD')",float,doc="ParticleNet tagger Z vs QCD discriminator",precision=10),
451  particleNet_HbbvsQCD = Var("bDiscriminator('pfParticleNetDiscriminatorsJetTags:HbbvsQCD')",float,doc="ParticleNet tagger H(->bb) vs QCD discriminator",precision=10),
452  particleNet_HccvsQCD = Var("bDiscriminator('pfParticleNetDiscriminatorsJetTags:HccvsQCD')",float,doc="ParticleNet tagger H(->cc) vs QCD discriminator",precision=10),
453  particleNet_H4qvsQCD = Var("bDiscriminator('pfParticleNetDiscriminatorsJetTags:H4qvsQCD')",float,doc="ParticleNet tagger H(->VV->qqqq) vs QCD discriminator",precision=10),
454  particleNet_QCD = Var("bDiscriminator('pfParticleNetJetTags:probQCDbb')+bDiscriminator('pfParticleNetJetTags:probQCDcc')+bDiscriminator('pfParticleNetJetTags:probQCDb')+bDiscriminator('pfParticleNetJetTags:probQCDc')+bDiscriminator('pfParticleNetJetTags:probQCDothers')",float,doc="ParticleNet tagger QCD(bb,cc,b,c,others) sum",precision=10),
455  particleNetMD_Xbb = Var("bDiscriminator('pfMassDecorrelatedParticleNetJetTags:probXbb')",float,doc="Mass-decorrelated ParticleNet tagger raw X->bb score. For X->bb vs QCD tagging, use Xbb/(Xbb+QCD)",precision=10),
456  particleNetMD_Xcc = Var("bDiscriminator('pfMassDecorrelatedParticleNetJetTags:probXcc')",float,doc="Mass-decorrelated ParticleNet tagger raw X->cc score. For X->cc vs QCD tagging, use Xcc/(Xcc+QCD)",precision=10),
457  particleNetMD_Xqq = Var("bDiscriminator('pfMassDecorrelatedParticleNetJetTags:probXqq')",float,doc="Mass-decorrelated ParticleNet tagger raw X->qq (uds) score. For X->qq vs QCD tagging, use Xqq/(Xqq+QCD). For W vs QCD tagging, use (Xcc+Xqq)/(Xcc+Xqq+QCD)",precision=10),
458  particleNetMD_QCD = Var("bDiscriminator('pfMassDecorrelatedParticleNetJetTags:probQCDbb')+bDiscriminator('pfMassDecorrelatedParticleNetJetTags:probQCDcc')+bDiscriminator('pfMassDecorrelatedParticleNetJetTags:probQCDb')+bDiscriminator('pfMassDecorrelatedParticleNetJetTags:probQCDc')+bDiscriminator('pfMassDecorrelatedParticleNetJetTags:probQCDothers')",float,doc="Mass-decorrelated ParticleNet tagger raw QCD score",precision=10),
459  subJetIdx1 = Var("?nSubjetCollections()>0 && subjets('SoftDropPuppi').size()>0?subjets('SoftDropPuppi')[0].key():-1", int,
460  doc="index of first subjet"),
461  subJetIdx2 = Var("?nSubjetCollections()>0 && subjets('SoftDropPuppi').size()>1?subjets('SoftDropPuppi')[1].key():-1", int,
462  doc="index of second subjet"),
463 
464 # btagDeepC = Var("bDiscriminator('pfDeepCSVJetTags:probc')",float,doc="CMVA V2 btag discriminator",precision=10),
465 #puIdDisc = Var("userFloat('pileupJetId:fullDiscriminant')",float,doc="Pileup ID discriminant",precision=10),
466 # nConstituents = Var("numberOfDaughters()",int,doc="Number of particles in the jet"),
467 # rawFactor = Var("1.-jecFactor('Uncorrected')",float,doc="1 - Factor to get back to raw pT",precision=6),
468  ),
469  externalVariables = cms.PSet(
470  lsf3 = ExtVar(cms.InputTag("lepInJetVars:lsf3"),float, doc="Lepton Subjet Fraction (3 subjets)",precision=10),
471  muonIdx3SJ = ExtVar(cms.InputTag("lepInJetVars:muIdx3SJ"),int, doc="index of muon matched to jet"),
472  electronIdx3SJ = ExtVar(cms.InputTag("lepInJetVars:eleIdx3SJ"),int,doc="index of electron matched to jet"),
473  )
474 )
475 ### Era dependent customization
476 for modifier in run2_miniAOD_80XLegacy, run2_nanoAOD_94X2016, run2_nanoAOD_94XMiniAODv1, run2_nanoAOD_94XMiniAODv2, run2_nanoAOD_102Xv1, run2_nanoAOD_106Xv1:
477  modifier.toModify( fatJetTable.variables.n2b1, expr = cms.string("userFloat('ak8PFJetsPuppiSoftDropValueMap:nb1AK8PuppiSoftDropN2')"),)
478  modifier.toModify( fatJetTable.variables.n3b1, expr = cms.string("userFloat('ak8PFJetsPuppiSoftDropValueMap:nb1AK8PuppiSoftDropN3')"),)
479  # Deprecated after 106X
480  modifier.toModify( fatJetTable.variables,
481  btagCMVA = Var("bDiscriminator('pfCombinedMVAV2BJetTags')",float,doc="CMVA V2 btag discriminator",precision=10),
482  btagDDBvL_noMD = Var("bDiscriminator('pfDeepDoubleBvLJetTags:probHbb')",float,doc="DeepDoubleX discriminator (no mass-decorrelation) for H(Z)->bb vs QCD",precision=10),
483  btagDDCvL_noMD = Var("bDiscriminator('pfDeepDoubleCvLJetTags:probHcc')",float,doc="DeepDoubleX discriminator (no mass-decorrelation) for H(Z)->cc vs QCD",precision=10),
484  btagDDCvB_noMD = Var("bDiscriminator('pfDeepDoubleCvBJetTags:probHcc')",float,doc="DeepDoubleX discriminator (no mass-decorrelation) for H(Z)->cc vs H(Z)->bb",precision=10),
485  btagDDBvL = Var("bDiscriminator('pfMassIndependentDeepDoubleBvLJetTags:probHbb')",float,doc="DeepDoubleX (mass-decorrelated) discriminator for H(Z)->bb vs QCD",precision=10),
486  btagDDCvL = Var("bDiscriminator('pfMassIndependentDeepDoubleCvLJetTags:probHcc')",float,doc="DeepDoubleX (mass-decorrelated) discriminator for H(Z)->cc vs QCD",precision=10),
487  btagDDCvB = Var("bDiscriminator('pfMassIndependentDeepDoubleCvBJetTags:probHcc')",float,doc="DeepDoubleX (mass-decorrelated) discriminator for H(Z)->cc vs H(Z)->bb",precision=10),
488  )
489 run2_miniAOD_80XLegacy.toModify( fatJetTable.variables, msoftdrop_chs = Var("userFloat('ak8PFJetsCHSSoftDropMass')",float, doc="Legacy uncorrected soft drop mass with CHS",precision=10))
490 run2_miniAOD_80XLegacy.toModify( fatJetTable.variables.tau1, expr = cms.string("userFloat(\'ak8PFJetsPuppiValueMap:NjettinessAK8PuppiTau1\')"),)
491 run2_miniAOD_80XLegacy.toModify( fatJetTable.variables.tau2, expr = cms.string("userFloat(\'ak8PFJetsPuppiValueMap:NjettinessAK8PuppiTau2\')"),)
492 run2_miniAOD_80XLegacy.toModify( fatJetTable.variables.tau3, expr = cms.string("userFloat(\'ak8PFJetsPuppiValueMap:NjettinessAK8PuppiTau3\')"),)
493 run2_miniAOD_80XLegacy.toModify( fatJetTable.variables, tau4 = None)
494 run2_miniAOD_80XLegacy.toModify( fatJetTable.variables, n2b1 = None)
495 run2_miniAOD_80XLegacy.toModify( fatJetTable.variables, n3b1 = None)
496 for modifier in run2_miniAOD_80XLegacy, run2_nanoAOD_94X2016:
497  modifier.toModify( fatJetTable.variables, jetId = Var("userInt('tightId')*2+userInt('looseId')",int,doc="Jet ID flags bit1 is loose, bit2 is tight"))
498 
499 run2_jme_2016.toModify( bjetNN, weightFile = cms.FileInPath("PhysicsTools/NanoAOD/data/breg_training_2016.pb") )
500 run2_jme_2016.toModify( bjetNN,outputFormulas = cms.vstring(["at(0)*0.31976690888404846+1.047176718711853","0.5*(at(2)-at(1))*0.31976690888404846"]))
501 
502 run2_jme_2017.toModify( bjetNN, weightFile = cms.FileInPath("PhysicsTools/NanoAOD/data/breg_training_2017.pb") )
503 run2_jme_2017.toModify( bjetNN,outputFormulas = cms.vstring(["at(0)*0.28225210309028625+1.055067777633667","0.5*(at(2)-at(1))*0.28225210309028625"]))
504 
505 run2_jme_2016.toModify( cjetNN, weightFile = cms.FileInPath("PhysicsTools/NanoAOD/data/creg_training_2016.pb") )
506 run2_jme_2016.toModify( cjetNN,outputFormulas = cms.vstring(["at(0)*0.28862622380256653+0.9908722639083862","0.5*(at(2)-at(1))*0.28862622380256653"]))
507 
508 run2_jme_2017.toModify( cjetNN, weightFile = cms.FileInPath("PhysicsTools/NanoAOD/data/creg_training_2017.pb") )
509 run2_jme_2017.toModify( cjetNN,outputFormulas = cms.vstring(["at(0)*0.24718524515628815+0.9927206635475159","0.5*(at(2)-at(1))*0.24718524515628815"]))
510 
511 
512 
513 subJetTable = cms.EDProducer("SimpleCandidateFlatTableProducer",
514  src = cms.InputTag("slimmedJetsAK8PFPuppiSoftDropPacked","SubJets"),
515  cut = cms.string(""), #probably already applied in miniaod
516  name = cms.string("SubJet"),
517  doc = cms.string("slimmedJetsAK8, i.e. ak8 fat jets for boosted analysis"),
518  singleton = cms.bool(False), # the number of entries is variable
519  extension = cms.bool(False), # this is the main table for the jets
520  variables = cms.PSet(P4Vars,
521  btagDeepB = Var("bDiscriminator('pfDeepCSVJetTags:probb')+bDiscriminator('pfDeepCSVJetTags:probbb')",float,doc="DeepCSV b+bb tag discriminator",precision=10),
522  btagCSVV2 = Var("bDiscriminator('pfCombinedInclusiveSecondaryVertexV2BJetTags')",float,doc=" pfCombinedInclusiveSecondaryVertexV2 b-tag discriminator (aka CSVV2)",precision=10),
523  rawFactor = Var("1.-jecFactor('Uncorrected')",float,doc="1 - Factor to get back to raw pT",precision=6),
524  tau1 = Var("userFloat('NjettinessAK8Subjets:tau1')",float, doc="Nsubjettiness (1 axis)",precision=10),
525  tau2 = Var("userFloat('NjettinessAK8Subjets:tau2')",float, doc="Nsubjettiness (2 axis)",precision=10),
526  tau3 = Var("userFloat('NjettinessAK8Subjets:tau3')",float, doc="Nsubjettiness (3 axis)",precision=10),
527  tau4 = Var("userFloat('NjettinessAK8Subjets:tau4')",float, doc="Nsubjettiness (4 axis)",precision=10),
528  n2b1 = Var("userFloat('nb1AK8PuppiSoftDropSubjets:ecfN2')", float, doc="N2 with beta=1", precision=10),
529  n3b1 = Var("userFloat('nb1AK8PuppiSoftDropSubjets:ecfN3')", float, doc="N3 with beta=1", precision=10),
530  )
531 )
532 
533 # Deprecation/backcomp
534 for modifier in run2_miniAOD_80XLegacy, run2_nanoAOD_94X2016, run2_nanoAOD_94XMiniAODv1, run2_nanoAOD_94XMiniAODv2, run2_nanoAOD_102Xv1, run2_nanoAOD_106Xv1:
535  # post 106X
536  modifier.toModify(subJetTable.variables,
537  btagCMVA = Var("bDiscriminator('pfCombinedMVAV2BJetTags')",float,doc="CMVA V2 btag discriminator",precision=10),
538  )
539 
540 #jets are not as precise as muons
541 fatJetTable.variables.pt.precision=10
542 subJetTable.variables.pt.precision=10
543 
544 run2_miniAOD_80XLegacy.toModify( subJetTable.variables, tau1 = None)
545 run2_miniAOD_80XLegacy.toModify( subJetTable.variables, tau2 = None)
546 run2_miniAOD_80XLegacy.toModify( subJetTable.variables, tau3 = None)
547 run2_miniAOD_80XLegacy.toModify( subJetTable.variables, tau4 = None)
548 run2_miniAOD_80XLegacy.toModify( subJetTable.variables, n2b1 = None)
549 run2_miniAOD_80XLegacy.toModify( subJetTable.variables, n3b1 = None)
550 run2_miniAOD_80XLegacy.toModify( subJetTable.variables, btagCMVA = None, btagDeepB = None)
551 
552 
553 corrT1METJetTable = cms.EDProducer("SimpleCandidateFlatTableProducer",
554  src = cms.InputTag("corrT1METJets"),
555  cut = cms.string(""),
556  name = cms.string("CorrT1METJet"),
557  doc = cms.string("Additional low-pt jets for Type-1 MET re-correction"),
558  singleton = cms.bool(False), # the number of entries is variable
559  extension = cms.bool(False), # this is the main table for the jets
560  variables = cms.PSet(
561  rawPt = Var("pt()*jecFactor('Uncorrected')",float,precision=10),
562  eta = Var("eta", float,precision=12),
563  phi = Var("phi", float, precision=12),
564  area = Var("jetArea()", float, doc="jet catchment area, for JECs",precision=10),
565  )
566 )
567 
568 
569 
570 ## MC STUFF ######################
571 jetMCTable = cms.EDProducer("SimpleCandidateFlatTableProducer",
572  src = cms.InputTag("linkedObjects","jets"),
573  cut = cms.string(""), #we should not filter on cross linked collections
574  name = cms.string("Jet"),
575  singleton = cms.bool(False), # the number of entries is variable
576  extension = cms.bool(True), # this is an extension table for the jets
577  variables = cms.PSet(
578  partonFlavour = Var("partonFlavour()", int, doc="flavour from parton matching"),
579  hadronFlavour = Var("hadronFlavour()", int, doc="flavour from hadron ghost clustering"),
580  genJetIdx = Var("?genJetFwdRef().backRef().isNonnull()?genJetFwdRef().backRef().key():-1", int, doc="index of matched gen jet"),
581  )
582 )
583 genJetTable = cms.EDProducer("SimpleCandidateFlatTableProducer",
584  src = cms.InputTag("slimmedGenJets"),
585  cut = cms.string("pt > 10"),
586  name = cms.string("GenJet"),
587  doc = cms.string("slimmedGenJets, i.e. ak4 Jets made with visible genparticles"),
588  singleton = cms.bool(False), # the number of entries is variable
589  extension = cms.bool(False), # this is the main table for the genjets
590  variables = cms.PSet(P4Vars,
591  #anything else?
592  )
593 )
594 patJetPartons = cms.EDProducer('HadronAndPartonSelector',
595  src = cms.InputTag("generator"),
596  particles = cms.InputTag("prunedGenParticles"),
597  partonMode = cms.string("Auto"),
598  fullChainPhysPartons = cms.bool(True)
599 )
600 genJetFlavourAssociation = cms.EDProducer("JetFlavourClustering",
601  jets = genJetTable.src,
602  bHadrons = cms.InputTag("patJetPartons","bHadrons"),
603  cHadrons = cms.InputTag("patJetPartons","cHadrons"),
604  partons = cms.InputTag("patJetPartons","physicsPartons"),
605  leptons = cms.InputTag("patJetPartons","leptons"),
606  jetAlgorithm = cms.string("AntiKt"),
607  rParam = cms.double(0.4),
608  ghostRescaling = cms.double(1e-18),
609  hadronFlavourHasPriority = cms.bool(False)
610 )
611 genJetFlavourTable = cms.EDProducer("GenJetFlavourTableProducer",
612  name = genJetTable.name,
613  src = genJetTable.src,
614  cut = genJetTable.cut,
615  deltaR = cms.double(0.1),
616  jetFlavourInfos = cms.InputTag("slimmedGenJetsFlavourInfos"),
617 )
618 
619 genJetAK8Table = cms.EDProducer("SimpleCandidateFlatTableProducer",
620  src = cms.InputTag("slimmedGenJetsAK8"),
621  cut = cms.string("pt > 100."),
622  name = cms.string("GenJetAK8"),
623  doc = cms.string("slimmedGenJetsAK8, i.e. ak8 Jets made with visible genparticles"),
624  singleton = cms.bool(False), # the number of entries is variable
625  extension = cms.bool(False), # this is the main table for the genjets
626  variables = cms.PSet(P4Vars,
627  #anything else?
628  )
629 )
630 genJetAK8FlavourAssociation = cms.EDProducer("JetFlavourClustering",
631  jets = genJetAK8Table.src,
632  bHadrons = cms.InputTag("patJetPartons","bHadrons"),
633  cHadrons = cms.InputTag("patJetPartons","cHadrons"),
634  partons = cms.InputTag("patJetPartons","physicsPartons"),
635  leptons = cms.InputTag("patJetPartons","leptons"),
636  jetAlgorithm = cms.string("AntiKt"),
637  rParam = cms.double(0.8),
638  ghostRescaling = cms.double(1e-18),
639  hadronFlavourHasPriority = cms.bool(False)
640 )
641 genJetAK8FlavourTable = cms.EDProducer("GenJetFlavourTableProducer",
642  name = genJetAK8Table.name,
643  src = genJetAK8Table.src,
644  cut = genJetAK8Table.cut,
645  deltaR = cms.double(0.1),
646  jetFlavourInfos = cms.InputTag("genJetAK8FlavourAssociation"),
647 )
648 fatJetMCTable = cms.EDProducer("SimpleCandidateFlatTableProducer",
649  src = fatJetTable.src,
650  cut = fatJetTable.cut,
651  name = fatJetTable.name,
652  singleton = cms.bool(False),
653  extension = cms.bool(True),
654  variables = cms.PSet(
655  nBHadrons = Var("jetFlavourInfo().getbHadrons().size()", "uint8", doc="number of b-hadrons"),
656  nCHadrons = Var("jetFlavourInfo().getcHadrons().size()", "uint8", doc="number of c-hadrons"),
657  hadronFlavour = Var("hadronFlavour()", int, doc="flavour from hadron ghost clustering"),
658  genJetAK8Idx = Var("?genJetFwdRef().backRef().isNonnull()?genJetFwdRef().backRef().key():-1", int, doc="index of matched gen AK8 jet"),
659  )
660 )
661 
662 genSubJetAK8Table = cms.EDProducer("SimpleCandidateFlatTableProducer",
663  src = cms.InputTag("slimmedGenJetsAK8SoftDropSubJets"),
664  cut = cms.string(""), ## These don't get a pt cut, but in miniAOD only subjets from fat jets with pt > 100 are kept
665  name = cms.string("SubGenJetAK8"),
666  doc = cms.string("slimmedGenJetsAK8SoftDropSubJets, i.e. subjets of ak8 Jets made with visible genparticles"),
667  singleton = cms.bool(False), # the number of entries is variable
668  extension = cms.bool(False), # this is the main table for the genjets
669  variables = cms.PSet(P4Vars,
670  #anything else?
671  )
672 )
673 subjetMCTable = cms.EDProducer("SimpleCandidateFlatTableProducer",
674  src = subJetTable.src,
675  cut = subJetTable.cut,
676  name = subJetTable.name,
677  singleton = cms.bool(False),
678  extension = cms.bool(True),
679  variables = cms.PSet(
680  nBHadrons = Var("jetFlavourInfo().getbHadrons().size()", "uint8", doc="number of b-hadrons"),
681  nCHadrons = Var("jetFlavourInfo().getcHadrons().size()", "uint8", doc="number of c-hadrons"),
682  hadronFlavour = Var("hadronFlavour()", int, doc="flavour from hadron ghost clustering"),
683  )
684 )
685 
686 ### Era dependent customization
687 run2_miniAOD_80XLegacy.toModify( genJetFlavourTable, jetFlavourInfos = cms.InputTag("genJetFlavourAssociation"),)
688 
689 from RecoJets.JetProducers.QGTagger_cfi import QGTagger
690 qgtagger=QGTagger.clone(srcJets="updatedJets",srcVertexCollection="offlineSlimmedPrimaryVertices")
691 
692 from RecoJets.JetProducers.PileupJetID_cfi import pileupJetId, _chsalgos_94x, _chsalgos_102x, _chsalgos_106X_UL17
693 pileupJetId94X=pileupJetId.clone(jets="updatedJets",algos = cms.VPSet(_chsalgos_94x),inputIsCorrected=True,applyJec=False,vertexes="offlineSlimmedPrimaryVertices")
694 pileupJetId102X=pileupJetId.clone(jets="updatedJets",algos = cms.VPSet(_chsalgos_102x),inputIsCorrected=True,applyJec=False,vertexes="offlineSlimmedPrimaryVertices")
695 pileupJetId106XUL17=pileupJetId.clone(jets="updatedJets",algos = cms.VPSet(_chsalgos_106X_UL17),inputIsCorrected=True,applyJec=False,vertexes="offlineSlimmedPrimaryVertices")
696 
697 
698 #before cross linking
699 jetSequence = cms.Sequence(jetCorrFactorsNano+updatedJets+tightJetId+tightJetIdLepVeto+bJetVars+qgtagger+jercVars+pileupJetId94X+pileupJetId102X+pileupJetId106XUL17+updatedJetsWithUserData+jetCorrFactorsAK8+updatedJetsAK8+tightJetIdAK8+tightJetIdLepVetoAK8+updatedJetsAK8WithUserData+chsForSATkJets+softActivityJets+softActivityJets2+softActivityJets5+softActivityJets10+finalJets+finalJetsAK8)
700 
701 _jetSequence_2016 = jetSequence.copy()
702 _jetSequence_2016.insert(_jetSequence_2016.index(tightJetId), looseJetId)
703 _jetSequence_2016.insert(_jetSequence_2016.index(tightJetIdAK8), looseJetIdAK8)
704 run2_jme_2016.toReplaceWith(jetSequence, _jetSequence_2016)
705 
706 
707 #HF shower shape variables:
708 #In 106X the variables are not computed/stored by default, but can be accessed/recomputed via modifiers
709 #For the two following modifiers, the producer is run on MINIAOD and one can directly access the variables
710 for modifier in run2_nanoAOD_106Xv2, run2_miniAOD_devel:
711  modifier.toModify( jetTable.variables, hfsigmaEtaEta = Var("userFloat('hfJetShowerShape:sigmaEtaEta')",float,doc="sigmaEtaEta for HF jets (noise discriminating variable)",precision=10))
712  modifier.toModify( jetTable.variables, hfsigmaPhiPhi = Var("userFloat('hfJetShowerShape:sigmaPhiPhi')",float,doc="sigmaPhiPhi for HF jets (noise discriminating variable)",precision=10))
713  modifier.toModify( jetTable.variables, hfcentralEtaStripSize = Var("userInt('hfJetShowerShape:centralEtaStripSize')", int, doc="eta size of the central tower strip in HF (noise discriminating variable) "))
714  modifier.toModify( jetTable.variables, hfadjacentEtaStripsSize = Var("userInt('hfJetShowerShape:adjacentEtaStripsSize')", int, doc="eta size of the strips next to the central tower strip in HF (noise discriminating variable) "))
715 
716 #For the following modifiers, the producer is not run on MINIAOD and one needs to run the producer in the NANOAOD step
717 from RecoJets.JetProducers.hfJetShowerShape_cfi import hfJetShowerShape
718 hfJetShowerShapeforNanoAOD = hfJetShowerShape.clone(jets="updatedJets",vertices="offlineSlimmedPrimaryVertices")
719 for modifier in run2_miniAOD_80XLegacy, run2_nanoAOD_94X2016, run2_nanoAOD_94XMiniAODv1, run2_nanoAOD_94XMiniAODv2, run2_nanoAOD_102Xv1, run2_nanoAOD_106Xv1:
720  modifier.toModify(updatedJetsWithUserData.userFloats,
721  hfsigmaEtaEta = cms.InputTag('hfJetShowerShapeforNanoAOD:sigmaEtaEta'),
722  hfsigmaPhiPhi = cms.InputTag('hfJetShowerShapeforNanoAOD:sigmaPhiPhi'),
723  )
724  modifier.toModify(updatedJetsWithUserData.userInts,
725  hfcentralEtaStripSize = cms.InputTag('hfJetShowerShapeforNanoAOD:centralEtaStripSize'),
726  hfadjacentEtaStripsSize = cms.InputTag('hfJetShowerShapeforNanoAOD:adjacentEtaStripsSize'),
727  )
728  modifier.toModify( jetTable.variables, hfsigmaEtaEta = Var("userFloat('hfsigmaEtaEta')",float,doc="sigmaEtaEta for HF jets (noise discriminating variable)",precision=10))
729  modifier.toModify( jetTable.variables, hfsigmaPhiPhi = Var("userFloat('hfsigmaPhiPhi')",float,doc="sigmaPhiPhi for HF jets (noise discriminating variable)",precision=10))
730  modifier.toModify( jetTable.variables, hfcentralEtaStripSize = Var("userInt('hfcentralEtaStripSize')", int, doc="eta size of the central tower strip in HF (noise discriminating variable) "))
731  modifier.toModify( jetTable.variables, hfadjacentEtaStripsSize = Var("userInt('hfadjacentEtaStripsSize')", int, doc="eta size of the strips next to the central tower strip in HF (noise discriminating variable) "))
732  _jetSequence_rerunHFshowershape = jetSequence.copy()
733  _jetSequence_rerunHFshowershape.insert(_jetSequence_rerunHFshowershape.index(updatedJetsWithUserData), hfJetShowerShapeforNanoAOD)
734  modifier.toReplaceWith(jetSequence, _jetSequence_rerunHFshowershape)
735 
736 #after lepton collections have been run
737 jetLepSequence = cms.Sequence(lepInJetVars)
738 
739 #after cross linkining
740 jetTables = cms.Sequence(bjetNN+cjetNN+jetTable+fatJetTable+subJetTable+saJetTable+saTable)
741 
742 #MC only producers and tables
743 jetMC = cms.Sequence(jetMCTable+genJetTable+patJetPartons+genJetFlavourTable+genJetAK8Table+genJetAK8FlavourAssociation+genJetAK8FlavourTable+fatJetMCTable+genSubJetAK8Table+subjetMCTable)
744 _jetMC_pre94X = jetMC.copy()
745 _jetMC_pre94X.insert(_jetMC_pre94X.index(genJetFlavourTable),genJetFlavourAssociation)
746 _jetMC_pre94X.remove(genSubJetAK8Table)
747 run2_miniAOD_80XLegacy.toReplaceWith(jetMC, _jetMC_pre94X)
748 
749 
PileupJetID_cfi
common_cff.ExtVar
def ExtVar(tag, valtype, compression=None, doc=None, mcOnly=False, precision=-1)
Definition: common_cff.py:31
QGTagger_cfi
nano_eras_cff
jetCorrFactors_cfi
jetUpdater_cfi
common_cff.Var
def Var(expr, valtype, compression=None, doc=None, mcOnly=False, precision=-1)
Definition: common_cff.py:20
ak4PFJetsBetaStar_cfi
ak4PFJets_cfi
common_cff
Generated for CMSSW Reference Manual by doxygen 1.8.11