NanoAOD/python/ttbarCategorization_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002 from PhysicsTools.NanoAOD.common_cff import *
0003 from PhysicsTools.NanoAOD.globalVariablesTableProducer_cfi import globalVariablesTableProducer
0004
0005 ##################### User floats producers, selectors ##########################
0006
0007 from PhysicsTools.JetMCAlgos.GenHFHadronMatcher_cff import matchGenBHadron
0008 matchGenBHadron = matchGenBHadron.clone(
0009     genParticles = cms.InputTag("prunedGenParticles"),
0010     jetFlavourInfos = cms.InputTag("slimmedGenJetsFlavourInfos"),
0011 )
0012
0013 ## Plugin for analysing C hadrons
0014 # MUST use the same particle collection as in selectedHadronsAndPartons
0015 from PhysicsTools.JetMCAlgos.GenHFHadronMatcher_cff import matchGenCHadron
0016 matchGenCHadron = matchGenCHadron.clone(
0017     genParticles = cms.InputTag("prunedGenParticles"),
0018     jetFlavourInfos = cms.InputTag("slimmedGenJetsFlavourInfos"),
0019 )
0020
0021 ## Producer for ttbar categorisation ID
0022 from TopQuarkAnalysis.TopTools.categorizeGenTtbar_cfi import categorizeGenTtbar
0023 categorizeGenTtbar = categorizeGenTtbar.clone(
0024     genJetPtMin = 20.,
0025     genJetAbsEtaMax = 2.4,
0026     genJets = cms.InputTag("slimmedGenJets"),
0027 )
0028
0029
0030 ##################### Tables for final output and docs ##########################
0031 ttbarCategoryTable = globalVariablesTableProducer.clone(
0032     variables = cms.PSet(
0033         genTtbarId = ExtVar( cms.InputTag("categorizeGenTtbar:genTtbarId"), "int", doc = "ttbar categorization")
0034     )
0035 )
0036
0037 ttbarCategoryTableTask = cms.Task(ttbarCategoryTable)
0038 ttbarCatMCProducersTask = cms.Task(matchGenBHadron,matchGenCHadron,categorizeGenTtbar)