NANOAOD: simplify configuration

PhysicsTools/NanoAOD/python/jetMC_cff.py

@@ -0,0 +1,144 @@
+import FWCore.ParameterSet.Config as cms
+
+from PhysicsTools.NanoAOD.nano_eras_cff import *
+from PhysicsTools.NanoAOD.common_cff import *
+from PhysicsTools.NanoAOD.jets_cff import fatJetTable as _fatJetTable
+from PhysicsTools.NanoAOD.jets_cff import subJetTable as _subJetTable
+
+jetMCTable = cms.EDProducer("SimpleCandidateFlatTableProducer",
+    src = cms.InputTag("linkedObjects","jets"),
+    cut = cms.string(""), #we should not filter on cross linked collections
+    name = cms.string("Jet"),
+    singleton = cms.bool(False), # the number of entries is variable
+    extension = cms.bool(True), # this is an extension  table for the jets
+    variables = cms.PSet(
+        partonFlavour = Var("partonFlavour()", int, doc="flavour from parton matching"),
+        hadronFlavour = Var("hadronFlavour()", int, doc="flavour from hadron ghost clustering"),
+        genJetIdx = Var("?genJetFwdRef().backRef().isNonnull()?genJetFwdRef().backRef().key():-1", int, doc="index of matched gen jet"),
+    )
+)
+genJetTable = cms.EDProducer("SimpleCandidateFlatTableProducer",
+    src = cms.InputTag("slimmedGenJets"),
+    cut = cms.string("pt > 10"),
+    name = cms.string("GenJet"),
+    doc  = cms.string("slimmedGenJets, i.e. ak4 Jets made with visible genparticles"),
+    singleton = cms.bool(False), # the number of entries is variable
+    extension = cms.bool(False), # this is the main table for the genjets
+    variables = cms.PSet(P4Vars,
+	#anything else?
+    )
+)
+
+patJetPartonsNano = cms.EDProducer('HadronAndPartonSelector',
+    src = cms.InputTag("generator"),
+    particles = cms.InputTag("prunedGenParticles"),
+    partonMode = cms.string("Auto"),
+    fullChainPhysPartons = cms.bool(True)
+)
+
+genJetFlavourAssociation = cms.EDProducer("JetFlavourClustering",
+    jets = genJetTable.src,
+    bHadrons = cms.InputTag("patJetPartonsNano","bHadrons"),
+    cHadrons = cms.InputTag("patJetPartonsNano","cHadrons"),
+    partons = cms.InputTag("patJetPartonsNano","physicsPartons"),
+    leptons = cms.InputTag("patJetPartonsNano","leptons"),
+    jetAlgorithm = cms.string("AntiKt"),
+    rParam = cms.double(0.4),
+    ghostRescaling = cms.double(1e-18),
+    hadronFlavourHasPriority = cms.bool(False)
+)
+
+genJetFlavourTable = cms.EDProducer("GenJetFlavourTableProducer",
+    name = genJetTable.name,
+    src = genJetTable.src,
+    cut = genJetTable.cut,
+    deltaR = cms.double(0.1),
+    jetFlavourInfos = cms.InputTag("slimmedGenJetsFlavourInfos"),
+)
+
+genJetAK8Table = cms.EDProducer("SimpleCandidateFlatTableProducer",
+    src = cms.InputTag("slimmedGenJetsAK8"),
+    cut = cms.string("pt > 100."),
+    name = cms.string("GenJetAK8"),
+    doc  = cms.string("slimmedGenJetsAK8, i.e. ak8 Jets made with visible genparticles"),
+    singleton = cms.bool(False), # the number of entries is variable
+    extension = cms.bool(False), # this is the main table for the genjets
+    variables = cms.PSet(P4Vars,
+	#anything else?
+    )
+)
+
+genJetAK8FlavourAssociation = cms.EDProducer("JetFlavourClustering",
+    jets = genJetAK8Table.src,
+    bHadrons = cms.InputTag("patJetPartonsNano","bHadrons"),
+    cHadrons = cms.InputTag("patJetPartonsNano","cHadrons"),
+    partons = cms.InputTag("patJetPartonsNano","physicsPartons"),
+    leptons = cms.InputTag("patJetPartonsNano","leptons"),
+    jetAlgorithm = cms.string("AntiKt"),
+    rParam = cms.double(0.8),
+    ghostRescaling = cms.double(1e-18),
+    hadronFlavourHasPriority = cms.bool(False)
+)
+
+genJetAK8FlavourTable = cms.EDProducer("GenJetFlavourTableProducer",
+    name = genJetAK8Table.name,
+    src = genJetAK8Table.src,
+    cut = genJetAK8Table.cut,
+    deltaR = cms.double(0.1),
+    jetFlavourInfos = cms.InputTag("genJetAK8FlavourAssociation"),
+)
+fatJetMCTable = cms.EDProducer("SimpleCandidateFlatTableProducer",
+    src = _fatJetTable.src,
+    cut = _fatJetTable.cut,
+    name = _fatJetTable.name,
+    singleton = cms.bool(False),
+    extension = cms.bool(True),
+    variables = cms.PSet(
+        nBHadrons = Var("jetFlavourInfo().getbHadrons().size()", "uint8", doc="number of b-hadrons"),
+        nCHadrons = Var("jetFlavourInfo().getcHadrons().size()", "uint8", doc="number of c-hadrons"),
+        hadronFlavour = Var("hadronFlavour()", int, doc="flavour from hadron ghost clustering"),
+        genJetAK8Idx = Var("?genJetFwdRef().backRef().isNonnull() && genJetFwdRef().backRef().pt() > 100.?genJetFwdRef().backRef().key():-1", int, doc="index of matched gen AK8 jet"),
+    )
+)
+
+genSubJetAK8Table = cms.EDProducer("SimpleCandidateFlatTableProducer",
+    src = cms.InputTag("slimmedGenJetsAK8SoftDropSubJets"),
+    cut = cms.string(""),  ## These don't get a pt cut, but in miniAOD only subjets from fat jets with pt > 100 are kept
+    name = cms.string("SubGenJetAK8"),
+    doc  = cms.string("slimmedGenJetsAK8SoftDropSubJets, i.e. subjets of ak8 Jets made with visible genparticles"),
+    singleton = cms.bool(False), # the number of entries is variable
+    extension = cms.bool(False), # this is the main table for the genjets
+    variables = cms.PSet(P4Vars,
+	#anything else?
+    )
+)
+subjetMCTable = cms.EDProducer("SimpleCandidateFlatTableProducer",
+    src = _subJetTable.src,
+    cut = _subJetTable.cut,
+    name = _subJetTable.name,
+    singleton = cms.bool(False),
+    extension = cms.bool(True),
+    variables = cms.PSet(
+        nBHadrons = Var("jetFlavourInfo().getbHadrons().size()", "uint8", doc="number of b-hadrons"),
+        nCHadrons = Var("jetFlavourInfo().getcHadrons().size()", "uint8", doc="number of c-hadrons"),
+        hadronFlavour = Var("hadronFlavour()", int, doc="flavour from hadron ghost clustering"),
+    )
+)
+
+
+jetMCTaskak4 = cms.Task(jetMCTable,genJetTable,patJetPartonsNano,genJetFlavourTable)
+jetMCTaskak8 = cms.Task(genJetAK8Table,genJetAK8FlavourAssociation,genJetAK8FlavourTable,fatJetMCTable,genSubJetAK8Table,subjetMCTable)
+jetMCTask = jetMCTaskak4.copy()
+jetMCTask.add(jetMCTaskak8)
+
+
+### Era dependent customization
+run2_miniAOD_80XLegacy.toModify( genJetFlavourTable, jetFlavourInfos = cms.InputTag("genJetFlavourAssociation"),)
+
+
+_jetMCTaskak8 = jetMCTaskak8.copyAndExclude([genSubJetAK8Table])
+
+_jetMC_pre94XTask = jetMCTaskak4.copy()
+_jetMC_pre94XTask.add(genJetFlavourAssociation)
+_jetMC_pre94XTask.add(_jetMCTaskak8)
+run2_miniAOD_80XLegacy.toReplaceWith(jetMCTask, _jetMC_pre94XTask)