JetExamples/test/printJetFlavourInfo.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 process = cms.Process("testJET")
0004
0005 process.options = cms.untracked.PSet(
0006     wantSummary = cms.untracked.bool(True)
0007 )
0008
0009 process.load("FWCore.MessageService.MessageLogger_cfi")
0010
0011 process.load("SimGeneral.HepPDTESSource.pythiapdt_cfi")
0012
0013 process.maxEvents = cms.untracked.PSet(
0014     input = cms.untracked.int32(5)
0015 )
0016
0017 process.source = cms.Source("PoolSource",
0018     fileNames = cms.untracked.vstring(
0019         # /TTJets_MassiveBinDECAY_TuneZ2star_8TeV-madgraph-tauola/Summer12_DR53X-PU_S10_START53_V7C-v1/AODSIM
0020         '/store/mc/Summer12_DR53X/TTJets_MassiveBinDECAY_TuneZ2star_8TeV-madgraph-tauola/AODSIM/PU_S10_START53_V7C-v1/00000/008BD264-1526-E211-897A-00266CFFA7BC.root'
0021     )
0022 )
0023
0024 process.printList = cms.EDAnalyzer("ParticleListDrawer",
0025     src = cms.InputTag("genParticles"),
0026     maxEventsToPrint = cms.untracked.int32(1)
0027 )
0028 #-------------------------------------------------------------------------
0029 # AK5 jets
0030 #-------------------------------------------------------------------------
0031 from PhysicsTools.JetMCAlgos.HadronAndPartonSelector_cfi import selectedHadronsAndPartons
0032 process.selectedHadronsAndPartons = selectedHadronsAndPartons.clone()
0033
0034 from PhysicsTools.JetMCAlgos.AK5PFJetsMCFlavourInfos_cfi import ak5JetFlavourInfos
0035 process.jetFlavourInfosAK5PFJets = ak5JetFlavourInfos.clone()
0036
0037 process.printEventAK5PFJets = cms.EDAnalyzer("printJetFlavourInfo",
0038     jetFlavourInfos = cms.InputTag("jetFlavourInfosAK5PFJets")
0039 )
0040 #-------------------------------------------------------------------------
0041 # AK8 fat jets and pruned subjets
0042 #-------------------------------------------------------------------------
0043 from RecoJets.JetProducers.ak5PFJets_cfi import ak5PFJets
0044 process.ak8PFJets = ak5PFJets.clone(
0045     rParam        = cms.double(0.8),
0046     src           = cms.InputTag("particleFlow"),
0047 )
0048
0049 from RecoJets.JetProducers.ak5PFJetsPruned_cfi import ak5PFJetsPruned
0050 process.ak8PFJetsPruned = ak5PFJetsPruned.clone(
0051     rParam              = cms.double(0.8),
0052     src                 = cms.InputTag("particleFlow"),
0053     writeCompound       = cms.bool(True),
0054     jetCollInstanceName = cms.string("SubJets")
0055 )
0056
0057 process.jetFlavourInfosAK8PFJets = cms.EDProducer("JetFlavourClustering",
0058     jets                     = cms.InputTag("ak8PFJets"),
0059     groomedJets              = cms.InputTag("ak8PFJetsPruned"),
0060     subjets                  = cms.InputTag("ak8PFJetsPruned", "SubJets"),
0061     bHadrons                 = cms.InputTag("selectedHadronsAndPartons","bHadrons"),
0062     cHadrons                 = cms.InputTag("selectedHadronsAndPartons","cHadrons"),
0063     partons                  = cms.InputTag("selectedHadronsAndPartons","algorithmicPartons"),
0064     leptons                  = cms.InputTag("selectedHadronsAndPartons","leptons"),
0065     jetAlgorithm             = cms.string("AntiKt"),
0066     rParam                   = cms.double(0.8),
0067     ghostRescaling           = cms.double(1e-18),
0068     hadronFlavourHasPriority = cms.bool(False)
0069 )
0070
0071 process.printEventAK8PFJets = cms.EDAnalyzer("printJetFlavourInfo",
0072     jetFlavourInfos    = cms.InputTag("jetFlavourInfosAK8PFJets"),
0073     subjetFlavourInfos = cms.InputTag("jetFlavourInfosAK8PFJets","SubJets"),
0074     groomedJets        = cms.InputTag("ak8PFJetsPruned"),
0075 )
0076 #-------------------------------------------------------------------------
0077 # CA15 fat jets and HEPTopTagger fat jets and subjets
0078 #-------------------------------------------------------------------------
0079 from RecoJets.JetProducers.AnomalousCellParameters_cfi import *
0080 from RecoJets.JetProducers.CATopJetParameters_cfi import *
0081 from RecoJets.JetProducers.PFJetParameters_cfi import *
0082
0083 from RecoJets.JetProducers.ca4PFJets_cfi import ca4PFJets
0084 process.ca15PFJets = ca4PFJets.clone(
0085     rParam   = cms.double(1.5),
0086     src      = cms.InputTag("particleFlow"),
0087     jetPtMin = cms.double(100.0)
0088 )
0089
0090 process.caHEPTopTagJets = cms.EDProducer(
0091     "CATopJetProducer",
0092     PFJetParameters.clone( src = cms.InputTag("particleFlow"),
0093                            doAreaFastjet = cms.bool(False),
0094                            doRhoFastjet = cms.bool(False),
0095                            jetPtMin = cms.double(100.0)
0096                            ),
0097     AnomalousCellParameters,
0098     CATopJetParameters.clone( tagAlgo = cms.int32(2) ),
0099     jetAlgorithm = cms.string("CambridgeAachen"),
0100     rParam = cms.double(1.5),
0101     muCut = cms.double(0.8),
0102     maxSubjetMass = cms.double(30.0),
0103     useSubjetMass = cms.bool(False),
0104     writeCompound = cms.bool(True)
0105 )
0106
0107 process.jetFlavourInfosCA15PFJets = cms.EDProducer("JetFlavourClustering",
0108     jets                     = cms.InputTag("ca15PFJets"),
0109     groomedJets              = cms.InputTag("caHEPTopTagJets"),
0110     subjets                  = cms.InputTag("caHEPTopTagJets", "caTopSubJets"),
0111     bHadrons                 = cms.InputTag("selectedHadronsAndPartons","bHadrons"),
0112     cHadrons                 = cms.InputTag("selectedHadronsAndPartons","cHadrons"),
0113     partons                  = cms.InputTag("selectedHadronsAndPartons","algorithmicPartons"),
0114     leptons                  = cms.InputTag("selectedHadronsAndPartons","leptons"),
0115     jetAlgorithm             = cms.string("CambridgeAachen"),
0116     rParam                   = cms.double(1.5),
0117     ghostRescaling           = cms.double(1e-18),
0118     hadronFlavourHasPriority = cms.bool(False)
0119 )
0120
0121 process.printEventCA15PFJets = cms.EDAnalyzer("printJetFlavourInfo",
0122     jetFlavourInfos    = cms.InputTag("jetFlavourInfosCA15PFJets"),
0123     subjetFlavourInfos = cms.InputTag("jetFlavourInfosCA15PFJets","SubJets"),
0124     groomedJets        = cms.InputTag("caHEPTopTagJets"),
0125 )
0126 #-------------------------------------------------------------------------
0127
0128 process.p = cms.Path(
0129     process.printList
0130     *process.selectedHadronsAndPartons
0131     *process.jetFlavourInfosAK5PFJets*process.printEventAK5PFJets
0132     *(process.ak8PFJets+process.ak8PFJetsPruned)*process.jetFlavourInfosAK8PFJets*process.printEventAK8PFJets
0133     *(process.ca15PFJets+process.caHEPTopTagJets)*process.jetFlavourInfosCA15PFJets*process.printEventCA15PFJets
0134 )
0135
0136 #process.MessageLogger.cout = dict( enable = True, threshold = 'ERROR')