python/mcMatchLayer0/jetFlavourId_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 patJetPartonsLegacy = cms.EDProducer("PartonSelector",
0004     withLeptons = cms.bool(False),
0005     src = cms.InputTag("genParticles")
0006 )
0007
0008 patJetPartonAssociationLegacy = cms.EDProducer("JetPartonMatcher",
0009     jets    = cms.InputTag("ak4PFJetsCHS"),
0010     partons = cms.InputTag("patJetPartonsLegacy"),
0011     coneSizeToAssociate = cms.double(0.3),
0012 )
0013
0014 patJetFlavourAssociationLegacy = cms.EDProducer("JetFlavourIdentifier",
0015     srcByReference    = cms.InputTag("patJetPartonAssociationLegacy"),
0016     physicsDefinition = cms.bool(False)
0017 )
0018
0019 patJetPartons = cms.EDProducer('HadronAndPartonSelector',
0020     src = cms.InputTag("generator"),
0021     particles = cms.InputTag("genParticles"),
0022     partonMode = cms.string("Auto"),
0023     fullChainPhysPartons = cms.bool(True)
0024 )
0025
0026 patJetFlavourAssociation = cms.EDProducer("JetFlavourClustering",
0027     jets = cms.InputTag("ak4PFJetsCHS"),
0028     bHadrons = cms.InputTag("patJetPartons","bHadrons"),
0029     cHadrons = cms.InputTag("patJetPartons","cHadrons"),
0030     partons = cms.InputTag("patJetPartons","physicsPartons"),
0031     leptons = cms.InputTag("patJetPartons","leptons"),
0032     jetAlgorithm = cms.string("AntiKt"),
0033     rParam = cms.double(0.4),
0034     ghostRescaling = cms.double(1e-18),
0035     hadronFlavourHasPriority = cms.bool(False)
0036 )
0037
0038 # default PAT sequence for jet flavour identification
0039 patJetFlavourIdLegacyTask = cms.Task(patJetPartonsLegacy, patJetPartonAssociationLegacy, patJetFlavourAssociationLegacy)
0040 patJetFlavourIdLegacy = cms.Sequence(patJetFlavourIdLegacyTask)
0041
0042 patJetFlavourIdTask = cms.Task(patJetPartons, patJetFlavourAssociation)
0043 patJetFlavourId = cms.Sequence(patJetFlavourIdTask)
0044
0045 from Configuration.ProcessModifiers.pp_on_AA_cff import pp_on_AA
0046 pp_on_AA.toModify(patJetPartons, particles = "hiSignalGenParticles")
0047 pp_on_AA.toModify(patJetFlavourAssociation,
0048                   jets = 'akCs4PFJets',
0049                   rParam = 0.4,
0050 )
0051 pp_on_AA.toModify(patJetPartonAssociationLegacy,
0052                   jets = "akCs4PFJets",
0053                   partons = "allPartons",
0054 )