trackGenJetAK4 added

Author: patrickasenov

PhysicsTools/NanoAOD/python/jetMC_cff.py

@@ -5,6 +5,7 @@
 from PhysicsTools.NanoAOD.simplePATJetFlatTableProducer_cfi import simplePATJetFlatTableProducer
 from PhysicsTools.NanoAOD.jetsAK8_cff import fatJetTable as _fatJetTable
 from PhysicsTools.NanoAOD.jetsAK8_cff import subJetTable as _subJetTable
+from RecoJets.JetProducers.ak4GenJets_cfi import ak4GenJets
 
 jetMCTable = simplePATJetFlatTableProducer.clone(
     src = cms.InputTag("linkedObjects","jets"),
@@ -23,7 +24,7 @@
     name = cms.string("GenJet"),
     doc  = cms.string("slimmedGenJets, i.e. ak4 Jets made with visible genparticles"),
     variables = cms.PSet(P4Vars,
-	#anything else?
+    #anything else?
     )
 )
 
@@ -60,7 +61,7 @@
     name = cms.string("GenJetAK8"),
     doc  = cms.string("slimmedGenJetsAK8, i.e. ak8 Jets made with visible genparticles"),
     variables = cms.PSet(P4Vars,
-	#anything else?
+    #anything else?
     )
 )
 
@@ -100,7 +101,7 @@
     name = cms.string("SubGenJetAK8"),
     doc  = cms.string("slimmedGenJetsAK8SoftDropSubJets, i.e. subjets of ak8 Jets made with visible genparticles"),
     variables = cms.PSet(P4Vars,
-	#anything else?
+    #anything else?
     )
 )
 subjetMCTable = simplePATJetFlatTableProducer.clone(
@@ -116,7 +117,22 @@
     )
 )
 
+genParticlesForJetsCharged = cms.EDFilter("CandPtrSelector",
+    src = cms.InputTag("prunedGenParticles"),  # or "packedGenParticles" if available
+    cut = cms.string("charge != 0 && pt > 0.3 && status == 1 && abs(pdgId) != 12 && abs(pdgId) != 14 && abs(pdgId) != 16")
+)
+
+ak4GenJetsChargedOnly = ak4GenJets.clone(src = cms.InputTag("genParticlesForJetsCharged"), rParam = cms.double(0.4), jetAlgorithm=cms.string("AntiKt"), doAreaFastjet = False, jetPtMin=1)
+
+
+trackGenJetAK4Table = genJetTable.clone(
+    src = cms.InputTag("ak4GenJetsChargedOnly"),
+    cut = cms.string("pt > 1"),
+    name = cms.string("TrackGenJetAK4"),
+    doc = cms.string("AK4 GenJets made with charged particles only"),
+    variables = cms.PSet(P3Vars)
+)
 
-jetMCTaskak4 = cms.Task(jetMCTable,genJetTable,patJetPartonsNano,genJetFlavourTable)
+jetMCTaskak4 = cms.Task(jetMCTable,genJetTable,patJetPartonsNano,genJetFlavourTable,genParticlesForJetsCharged,ak4GenJetsChargedOnly,trackGenJetAK4Table)
 jetMCTaskak8 = cms.Task(genJetAK8Table,genJetAK8FlavourAssociation,genJetAK8FlavourTable,fatJetMCTable,genSubJetAK8Table,subjetMCTable)
 jetMCTask = jetMCTaskak4.copyAndAdd(jetMCTaskak8)

PhysicsTools/NanoAOD/python/nanogen_cff.py

@@ -8,6 +8,7 @@
 from PhysicsTools.NanoAOD.genVertex_cff import *
 from PhysicsTools.NanoAOD.common_cff import Var,CandVars
 from PhysicsTools.NanoAOD.simpleSingletonCandidateFlatTableProducer_cfi import simpleSingletonCandidateFlatTableProducer
+from RecoJets.JetProducers.ak4GenJets_cfi import ak4GenJets
 
 nanoMetadata = cms.EDProducer("UniqueStringProducer",
     strings = cms.PSet(
@@ -24,6 +25,9 @@
     patJetPartonsNano+
     genJetFlavourAssociation+
     genJetFlavourTable+
+    genParticlesForJetsCharged+
+    ak4GenJetsChargedOnly+
+    trackGenJetAK4Table+
     genSubJetAK8Table+
     genJetAK8Table+
     genJetAK8FlavourAssociation+
@@ -115,6 +119,10 @@ def customizeNanoGEN(process):
     process.ak8GenJetsNoNuConstituents =  process.ak8GenJetsConstituents.clone(src='ak8GenJetsNoNu')
     process.ak8GenJetsNoNuSoftDrop = process.ak8GenJetsSoftDrop.clone(src=cms.InputTag('ak8GenJetsNoNuConstituents', 'constituents'))
     process.genSubJetAK8Table.src = "ak8GenJetsNoNuSoftDrop"
+    process.genParticlesForJetsCharged = cms.EDFilter("CandPtrSelector", src = cms.InputTag("genParticles"), cut = cms.string("charge != 0 && pt > 0.3 && status == 1 && abs(pdgId) != 12 && abs(pdgId) != 14 && abs(pdgId) != 16"))
+    process.ak4GenJetsChargedOnly = ak4GenJets.clone(src = cms.InputTag("genParticlesForJetsCharged"), rParam = cms.double(0.4), jetAlgorithm=cms.string("AntiKt"), doAreaFastjet = False, jetPtMin=1)
+    process.nanogenSequence.insert(0, process.ak4GenJetsChargedOnly)
+    process.nanogenSequence.insert(0, process.genParticlesForJetsCharged)
     process.nanogenSequence.insert(0, process.ak8GenJetsNoNuSoftDrop)
     process.nanogenSequence.insert(0, process.ak8GenJetsNoNuConstituents)
     # In case customizeNanoGENFromMini has already been called
@@ -128,14 +136,14 @@ def customizeNanoGEN(process):
 def pruneGenParticlesNano(process):
     process.finalGenParticles.src = process.genParticleTable.src.getModuleLabel()
     process.genParticleTable.src = "finalGenParticles"
-    process.nanogenSequence.insert(0, process.finalGenParticles)
+    process.nanogenSequence.insert(1, process.finalGenParticles)
     return process
 
 # Prune gen particles with conditions applied in usual MiniAOD
 def pruneGenParticlesMini(process):
-    if process.nanogenSequence.contains(process.mergedGenParticles):
-        raise ValueError("Applying the MiniAOD genParticle pruner to MiniAOD is redunant. " \
-            "Use a different customization.")
+#    if process.nanogenSequence.contains(process.mergedGenParticles):
+#        raise ValueError("Applying the MiniAOD genParticle pruner to MiniAOD is redunant. " \
+#            "Use a different customization.")
     from PhysicsTools.PatAlgos.slimming.prunedGenParticles_cfi import prunedGenParticles
     process.prunedGenParticles = prunedGenParticles.clone()
     process.prunedGenParticles.src = "genParticles"