def miniAOD_customizeMC(process):
task = getPatAlgosToolsTask(process)
#GenJetFlavourInfos
process.load("PhysicsTools.JetMCAlgos.HadronAndPartonSelector_cfi")
task.add(process.selectedHadronsAndPartons)
task.add(process.selectedHadronsAndPartonsForGenJetsFlavourInfos)
process.load("PhysicsTools.JetMCAlgos.AK4GenJetFlavourInfos_cfi")
task.add(process.ak4GenJetFlavourInfos)
process.load(
'PhysicsTools.PatAlgos.slimming.slimmedGenJetsFlavourInfos_cfi')
task.add(process.slimmedGenJetsFlavourInfos)
#slimmed pileup information
process.load('PhysicsTools.PatAlgos.slimming.slimmedAddPileupInfo_cfi')
task.add(process.slimmedAddPileupInfo)
process.muonMatch.matched = "prunedGenParticles"
process.electronMatch.matched = "prunedGenParticles"
process.electronMatch.src = cms.InputTag("reducedEgamma",
"reducedGedGsfElectrons")
process.photonMatch.matched = "prunedGenParticles"
process.photonMatch.src = cms.InputTag("reducedEgamma",
"reducedGedPhotons")
process.ootPhotonMatch.matched = "prunedGenParticles"
process.ootPhotonMatch.src = cms.InputTag("reducedEgamma",
"reducedOOTPhotons")
process.tauMatch.matched = "prunedGenParticles"
process.tauGenJets.GenParticles = "prunedGenParticles"
#Boosted taus
process.tauMatchBoosted.matched = "prunedGenParticles"
process.tauGenJetsBoosted.GenParticles = "prunedGenParticles"
process.patJetPartons.particles = "genParticles"
process.patJetPartonMatch.matched = "prunedGenParticles"
pp_on_AA.toModify(process.patJetPartonMatch,
matched="hiSignalGenParticles")
from Configuration.ProcessModifiers.genJetSubEvent_cff import genJetSubEvent
genJetSubEvent.toModify(process.patJetPartonMatch,
matched="cleanedPartons")
process.patJetPartonMatch.mcStatus = [3, 23]
process.patJetGenJetMatch.matched = "slimmedGenJets"
(~pp_on_AA).toModify(
process, patJetGenJetMatchAK8Puppi=dict(matched="slimmedGenJetsAK8"))
process.patMuons.embedGenMatch = False
process.patElectrons.embedGenMatch = False
process.patPhotons.embedGenMatch = False
process.patOOTPhotons.embedGenMatch = False
process.patTaus.embedGenMatch = False
process.patTausBoosted.embedGenMatch = False
process.patJets.embedGenPartonMatch = False
#also jet flavour must be switched
process.patJetFlavourAssociation.rParam = 0.4
from PhysicsTools.PatAlgos.producersHeavyIons.heavyIonJetSetup import removeJECsForMC
pp_on_AA.toModify(process, removeJECsForMC)
pp_on_AA.toReplaceWith(
task, task.copyAndExclude([process.slimmedGenJetsFlavourInfos]))
import FWCore.ParameterSet.Config as cms
slimmedGenJets = cms.EDProducer("PATGenJetSlimmer",
src = cms.InputTag("ak4GenJetsNoNu"),
packedGenParticles = cms.InputTag("packedGenParticles"),
cut = cms.string("pt > 8"),
cutLoose = cms.string(""),
nLoose = cms.uint32(0),
clearDaughters = cms.bool(False), #False means rekeying
dropSpecific = cms.bool(False),
)
slimmedGenJetsAK8 = cms.EDProducer("PATGenJetSlimmer",
src = cms.InputTag("ak8GenJetsNoNu"),
packedGenParticles = cms.InputTag("packedGenParticles"),
cut = cms.string("pt > 80"),
cutLoose = cms.string("pt > 10."),
nLoose = cms.uint32(3),
clearDaughters = cms.bool(False), #False means rekeying
dropSpecific = cms.bool(False),
)
from Configuration.ProcessModifiers.pp_on_AA_cff import pp_on_AA
pp_on_AA.toModify(slimmedGenJets, src = "ak4HiSignalGenJets")
pp_on_AA.toModify(slimmedGenJetsAK8, cut = 'pt>9999', nLoose = 0)
from Configuration.ProcessModifiers.genJetSubEvent_cff import genJetSubEvent
genJetSubEvent.toModify(slimmedGenJets, src = "ak4HiGenJetsCleaned")
from RecoHI.HiJetAlgos.HiGenCleaner_cff import hiPartons
from RecoHI.HiJetAlgos.HiGenJets_cff import ak4HiGenJets
from RecoHI.HiJetAlgos.HiGenCleaner_cff import heavyIonCleanedGenJets
from RecoHI.HiJetAlgos.hiSignalGenJetProducer_cfi import hiSignalGenJetProducer
ak4HiSignalGenJets = hiSignalGenJetProducer.clone(src="ak4HiGenJets")
allPartons = cms.EDProducer(
"PartonSelector",
src=cms.InputTag('hiSignalGenParticles'),
withLeptons=cms.bool(False),
)
from Configuration.ProcessModifiers.genJetSubEvent_cff import genJetSubEvent
genJetSubEvent.toModify(allPartons, src="genParticles")
cleanedPartons = hiPartons.clone(src='allPartons', )
hiGenJetsTask = cms.Task(hiSignalGenParticles, genParticlesForJets, allPartons,
ak4HiGenJets, ak4HiSignalGenJets)
ak4HiGenJetsCleaned = heavyIonCleanedGenJets.clone(src="ak4HiGenJets")
hiCleanedGenJetsTask_ = hiGenJetsTask.copyAndExclude(
[hiSignalGenParticles, ak4HiSignalGenJets])
hiCleanedGenJetsTask_.add(cleanedPartons, ak4HiGenJetsCleaned)
genJetSubEvent.toReplaceWith(hiGenJetsTask, hiCleanedGenJetsTask_)
from RecoHI.HiJetAlgos.PackedPFTowers_cfi import PackedPFTowers
from RecoHI.HiJetAlgos.HiRecoPFJets_cff import pfEmptyCollection, ak4PFJetsForFlow, hiPuRho, hiFJRhoFlowModulation, akCs4PFJets
from RecoHI.HiTracking.highPurityGeneralTracks_cfi import highPurityGeneralTracks