def switchToAOD(process, triggerHistManager = None, eventDumpPlugin = None):
# call "standard" PAT function to restrict all InputTags to AOD event content
restrictInputToAOD(process, [ "All", ])
# switch collection of ECAL recHits used as input for IsoDeposit computation
# from list of all ECAL recHits in the event to "reduced" collections
# limited to cones of size dR = 0.6 around electron candidates
if hasattr(process, "p"):
massSearchReplaceAnyInputTag(process.p, cms.InputTag("ecalRecHit", "EcalRecHitsEB"), cms.InputTag("reducedEcalRecHitsEB"))
massSearchReplaceAnyInputTag(process.p, cms.InputTag("ecalRecHit", "EcalRecHitsEE"), cms.InputTag("reducedEcalRecHitsEE"))
# disable PAT trigger matching
# (not yet implemented for photons and jets)
if hasattr(process, "patTriggerSequence"):
process.patDefaultSequence.remove(process.patTriggerSequence)
process.patElectrons.embedHighLevelSelection = cms.bool(False)
#process.patPhotons.embedHighLevelSelection = cms.bool(False)
process.patMuons.embedHighLevelSelection = cms.bool(False)
process.patTaus.embedHighLevelSelection = cms.bool(False)
#process.patJets.embedHighLevelSelection = cms.bool(False)
process.patMETs.embedHighLevelSelection = cms.bool(False)
if hasattr(process, "patPFMETs"):
process.patPFMETs.embedHighLevelSelection = cms.bool(False)
if triggerHistManager is not None:
triggerHistManager.hltResultsSource = cms.InputTag('')
triggerHistManager.l1Bits = cms.vstring()
triggerHistManager.hltPaths = cms.vstring()
if eventDumpPlugin is not None:
eventDumpPlugin.l1GtReadoutRecordSource = cms.InputTag('')
eventDumpPlugin.l1GtObjectMapRecordSource = cms.InputTag('')
eventDumpPlugin.hltResultsSource = cms.InputTag('')
def Pat(self) :
from PhysicsTools.PatAlgos.tools.coreTools import removeMCMatching,removeCleaning,restrictInputToAOD,removeAllPATObjectsBut
from PhysicsTools.PatAlgos.tools.jetTools import switchJetCollection
from PhysicsTools.PatAlgos.tools.jetTools import addJetCollection
if self.options.isData: removeMCMatching(self.process, ['All'])
restrictInputToAOD(self.process)
removeCleaning(self.process)
removeAllPATObjectsBut(self.process, ['Jets','METs','Muons'])
addJetCollection(self.process,
cms.InputTag('ak5CaloJets'),
'AK5',
'Calo',
jetCorrLabel = ('AK5Calo',self.options.jetCorrections),
genJetCollection = cms.InputTag('ak5GenJets'),
doType1MET = False,
)
switchJetCollection(self.process,
cms.InputTag('ak5PFJets'),
doJTA = True,
doBTagging = True,
doJetID = True,
jetCorrLabel = ('AK5PF',self.options.jetCorrections),
genJetCollection = cms.InputTag('ak5GenJets'),
doType1MET = False,
jetIdLabel = 'ak5pf'
)
self.process.selectedPatJets.cut = cms.string("pt > 30 && \
abs(eta) < 3.0 && \
neutralHadronEnergyFraction < 0.9 && \
neutralEmEnergyFraction < 0.90 && \
nConstituents > 1 && \
(? abs(eta)<2.4 ? chargedHadronEnergyFraction : 1) > 0 && \
(? abs(eta)<2.4 ? chargedHadronMultiplicity : 1) > 0 && \
(? abs(eta)<2.4 ? chargedEmEnergyFraction : 0) < 0.99")
del self.process.out
del self.process.outpath
return self.process.patDefaultSequence
def addPlainPat(process, dataVersion, doPatTrigger=True, doPatTaus=True, doHChTauDiscriminators=True, doPatMET=True, doPatElectronID=True,
doPatCalo=True, doBTagging=True, doPatMuonPFIsolation=False, doPatTauIsoDeposits=False,
doTauHLTMatching=True, matchingTauTrigger=None, matchingJetTrigger=None,
includePFCands=False):
out = None
outdict = process.outputModules_()
if outdict.has_key("out"):
out = outdict["out"]
outputCommands = []
# Tau Discriminators
process.hplusPatTauSequence = cms.Sequence()
if doPatTaus:
process.hplusPatTauSequence = addPFTausAndDiscriminators(process, dataVersion, doPatCalo, doHChTauDiscriminators)
# PAT Layer 0+1
process.load("PhysicsTools.PatAlgos.patSequences_cff")
sequence = cms.Sequence(
process.hplusPatTauSequence
)
# Restrict input to AOD
restrictInputToAOD(process, ["All"])
# Remove MC stuff if we have collision data (has to be done any add*Collection!)
# This also adds the L2L3Residual JEC correction to the process.patJetCorrFactors
if dataVersion.isData():
runOnData(process, outputInProcess = out!=None)
# Jets
# Produce kt6 rho for L1Fastjet
process.load('RecoJets.Configuration.RecoPFJets_cff')
process.kt6PFJets.doRhoFastjet = True
process.ak5PFJets.doAreaFastjet = True
process.ak5PFJetSequence = cms.Sequence(process.kt6PFJets * process.ak5PFJets)
# Set defaults
process.patJets.jetSource = cms.InputTag("ak5CaloJets")
process.patJets.trackAssociationSource = cms.InputTag("ak5JetTracksAssociatorAtVertex")
setPatJetDefaults(process.patJets)
setPatJetCorrDefaults(process.patJetCorrFactors, dataVersion)
process.patDefaultSequence.replace(process.patJetCorrFactors,
process.ak5PFJetSequence*process.patJetCorrFactors)
process.selectedPatJets.cut = jetPreSelection
# The default JEC to be embedded to pat::Jets are L2Relative,
# L3Absolute, L5Flavor and L7Parton. The default JEC to be applied
# is L2L3Residual, or L3Absolute, or Uncorrected (in this order).
if doPatCalo:
# Add JPT jets
# FIXME: Disabled for now until the JEC for JPT works again (with the latest JEC)
# addJetCollection(process, cms.InputTag('JetPlusTrackZSPCorJetAntiKt5'),
# 'AK5', 'JPT',
# doJTA = True,
# doBTagging = doBTagging,
# jetCorrLabel = ('AK5JPT', process.patJetCorrFactors.levels),
# doType1MET = False,
# doL1Cleaning = False,
# doL1Counters = True,
# genJetCollection = cms.InputTag("ak5GenJets"),
# doJetID = True
# )
# Add PF jets
addJetCollection(process, cms.InputTag('ak5PFJets'),
'AK5', 'PF',
doJTA = True,
doBTagging = doBTagging,
jetCorrLabel = ('AK5PF', process.patJetCorrFactors.levels),
doType1MET = False,
doL1Cleaning = False,
doL1Counters = True,
genJetCollection = cms.InputTag("ak5GenJets"),
doJetID = True
)
setPatJetCorrDefaults(process.patJetCorrFactorsAK5PF, dataVersion, True)
else:
setPatJetCorrDefaults(process.patJetCorrFactors, dataVersion, True)
switchJetCollection(process, cms.InputTag('ak5PFJets'),
doJTA = True,
doBTagging = doBTagging,
jetCorrLabel = ('AK5PF', process.patJetCorrFactors.levels),
doType1MET = False,
genJetCollection = cms.InputTag("ak5GenJets"),
doJetID = True
)
outputCommands.extend([
"keep *_selectedPatJets_*_*",
"keep *_selectedPatJetsAK5JPT_*_*",
"keep *_selectedPatJetsAK5PF_*_*",
'drop *_selectedPatJets_pfCandidates_*', ## drop for default patJets which are CaloJets
'drop *_*PF_caloTowers_*',
'drop *_*JPT_pfCandidates_*',
'drop *_*Calo_pfCandidates_*',
])
# Taus
# Set default PATTauProducer options here, they should be
# replicated to all added tau collections (and the first call to
# addTauCollection should replace the default producer modified
# here)
setPatTauDefaults(process.patTaus, includePFCands)
process.selectedPatTaus.cut = tauPreSelection
if doPatTaus:
if doHChTauDiscriminators:
addHChTauDiscriminators()
# Don't enable TCTau nor shrinking cone tau
# if doPatCalo:
# tauTools.addTauCollection(process,cms.InputTag('caloRecoTauProducer'),
# algoLabel = "caloReco",
# typeLabel = "Tau")
# setPatTauDefaults(process.patTausCaloRecoTau, True)
# process.patTausCaloRecoTau.embedLeadTrack = not includePFCands
# process.patTausCaloRecoTau.embedLeadPFChargedHadrCand = False
# tauTools.addTauCollection(process,cms.InputTag('shrinkingConePFTauProducer'),
# algoLabel = "shrinkingCone",
# typeLabel = "PFTau")
# # Disable isoDeposits like this until the problem with doPFIsoDeposits is fixed
# if not doPatTauIsoDeposits:
# process.patTausShrinkingConePFTau.isoDeposits = cms.PSet()
tauTools.addTauCollection(process,cms.InputTag('hpsPFTauProducer'),
algoLabel = "hps",
typeLabel = "PFTau")
if not doPatTauIsoDeposits:
process.patTausHpsPFTau.isoDeposits = cms.PSet()
addPatTauIsolationEmbedding(process, process.patDefaultSequence, "HpsPFTau")
tauTools.addTauCollection(process,cms.InputTag('hpsTancTaus'),
algoLabel = "hpsTanc",
typeLabel = "PFTau")
if not doPatTauIsoDeposits:
process.patTausHpsTancPFTau.isoDeposits = cms.PSet()
# Disable discriminators which are not in AOD
# del process.patTausHpsTancPFTau.tauIDSources.againstCaloMuon
# del process.patTausHpsTancPFTau.tauIDSources.byHPSvloose
addPatTauIsolationEmbedding(process, process.patDefaultSequence, "HpsTancPFTau")
# Add visible taus
if dataVersion.isMC():
process.VisibleTaus = cms.EDProducer("HLTTauMCProducer",
GenParticles = cms.untracked.InputTag("genParticles"),
ptMinTau = cms.untracked.double(3),
ptMinMuon = cms.untracked.double(3),
ptMinElectron = cms.untracked.double(3),
BosonID = cms.untracked.vint32(23),
EtaMax = cms.untracked.double(2.5)
)
sequence *= process.VisibleTaus
outputCommands.append("keep *_VisibleTaus_*_*")
else:
# FIXME: this is broken at the moment
#removeSpecificPATObjects(process, ["Taus"], outputInProcess= out != None)
process.patDefaultSequence.remove(process.patTaus)
process.patDefaultSequence.remove(process.selectedPatTaus)
outputCommands.extend(["drop *_selectedPatTaus_*_*",
# "keep *_selectedPatTausCaloRecoTau_*_*",
# "keep *_selectedPatTausShrinkingConePFTau_*_*",
"keep *_selectedPatTausHpsPFTau_*_*",
"keep *_selectedPatTausHpsTancPFTau_*_*",
#"keep *_cleanPatTaus_*_*",
#"drop *_cleanPatTaus_*_*",
#"keep *_patTaus*_*_*",
#"keep *_patPFTauProducerFixedCone_*_*",
# keep these until the embedding problem with pat::Tau is fixed
#"keep recoPFCandidates_particleFlow_*_*",
#"keep recoTracks_generalTracks_*_*"
])
# MET
addPfMET(process, 'PF')
if doPatCalo:
addTcMET(process, 'TC')
else:
# FIXME: This is broken at the moment...
#removeSpecificPATObjects(process, ["METs"], outputInProcess= out != None)
#process.patDefaultSequen
process.patDefaultSequence.remove(process.patMETCorrections)
process.patDefaultSequence.remove(process.patMETs)
del process.patMETCorrections
del process.patMETs
outputCommands.extend([
"keep *_patMETs_*_*",
"keep *_patMETsTC_*_*",
"keep *_patMETsPF_*_*",
"keep *_genMetTrue_*_*",
])
# Muons
setPatLeptonDefaults(process.patMuons, includePFCands)
if doPatMuonPFIsolation:
addPFMuonIsolation(process, process.patMuons, sequence, verbose=True)
outputCommands.extend([
"keep *_selectedPatMuons_*_*"
])
# Electrons
# In order to calculate the transverse impact parameter w.r.t.
# beam spot instead of primary vertex, see
setPatLeptonDefaults(process.patMuons, includePFCands)
# Electron ID, see
# https://twiki.cern.ch/twiki/bin/view/CMS/SimpleCutBasedEleID
if doPatElectronID:
addPatElectronID(process, process.patElectrons, process.patDefaultSequence)
outputCommands.extend([
"keep *_selectedPatElectrons_*_*"
])
# Photons
# process.patPhotons.embedGenMatch = False
outputCommands.extend([
"keep *_selectedPatPhotons_*_*"
])
# Trigger
if doPatTrigger:
outMod= ''
if out != None:
outMod = 'out'
switchOnTrigger(process, hltProcess=dataVersion.getTriggerProcess(), outputModule=outMod)
process.patTrigger.addL1Algos = cms.bool(True)
process.patTrigger.l1ExtraMu = cms.InputTag("l1extraParticles")
process.patTrigger.l1ExtraCenJet = cms.InputTag("l1extraParticles", "Central")
process.patTrigger.l1ExtraTauJet = cms.InputTag("l1extraParticles", "Tau")
process.patTrigger.l1ExtraForJet = cms.InputTag("l1extraParticles", "Forward")
process.patTrigger.l1ExtraETM = cms.InputTag("l1extraParticles", "MET")
process.patTrigger.l1ExtraHTM = cms.InputTag("l1extraParticles", "MHT")
# Keep StandAlone trigger objects for enabling trigger
# matching in the analysis phase with PAT tools
outputCommands.extend(patTriggerStandAloneEventContent)
# Remove cleaning step and set the event content
if out == None:
myRemoveCleaning(process)
else:
backup = out.outputCommands[:]
myRemoveCleaning(process)
# backup_pat = out.outputCommands[:]
# Remove PFParticles here, they are explicitly included when needed
# backup_pat = filter(lambda n: "selectedPatPFParticles" not in n, backup_pat)
out.outputCommands = backup
# out.outputCommands.extend(backup_pat)
out.outputCommands.extend(outputCommands)
# Build sequence
sequence *= process.patDefaultSequence
# Tau+HLT matching
if doTauHLTMatching:
sequence *= HChTriggerMatching.addTauHLTMatching(process, matchingTauTrigger, matchingJetTrigger)
return sequence
process.highetele = cms.EDFilter(
"GsfElectronSelector",
src=cms.InputTag("gsfElectrons"),
cut=cms.string("superCluster().get().energy()*sin(theta())> 30"))
process.highetFilter = cms.EDFilter(
"CandViewCountFilter",
src=cms.InputTag("highetele"),
minNumber=cms.uint32(2),
)
#no configuration of the pat is necessary for us at the moment
process.load("PhysicsTools.PatAlgos.patSequences_cff")
from PhysicsTools.PatAlgos.tools.coreTools import restrictInputToAOD
restrictInputToAOD(process, ['All'])
from SHarper.HEEPAnalyzer.HEEPSelectionCuts_cfi import *
process.heepId = cms.EDProducer("HEEPAttStatusToPAT",
eleLabel=cms.InputTag("allLayer1Electrons"),
barrelCuts=cms.PSet(heepBarrelCuts),
endcapCuts=cms.PSet(heepEndcapCuts))
process.lowetpath = cms.Path(process.highetele * process.highetFilter *
process.patDefaultSequence * process.heepId)
process.load("Configuration.EventContent.EventContent_cff")
process.out = cms.OutputModule(
"PoolOutputModule",
process.AODSIMEventContent,
fileName=cms.untracked.string('file:test.root'),
)
process.highetele = cms.EDFilter("GsfElectronSelector",
src =cms.InputTag("gsfElectrons"),
cut =cms.string("superCluster().get().energy()*sin(theta())> 30")
)
process.highetFilter = cms.EDFilter("CandViewCountFilter",
src = cms.InputTag("highetele"),
minNumber = cms.uint32(2),
)
#no configuration of the pat is necessary for us at the moment
process.load("PhysicsTools.PatAlgos.patSequences_cff");
from PhysicsTools.PatAlgos.tools.coreTools import restrictInputToAOD
restrictInputToAOD(process, ['All'])
from SHarper.HEEPAnalyzer.HEEPSelectionCuts_cfi import *
process.heepId = cms.EDProducer("HEEPAttStatusToPAT",
eleLabel = cms.InputTag("allLayer1Electrons"),
barrelCuts = cms.PSet(heepBarrelCuts),
endcapCuts = cms.PSet(heepEndcapCuts)
)
process.lowetpath = cms.Path( process.highetele * process.highetFilter* process.patDefaultSequence* process.heepId )
process.load("Configuration.EventContent.EventContent_cff")
process.out = cms.OutputModule("PoolOutputModule",
process.AODSIMEventContent,
fileName = cms.untracked.string('file:test.root'),
def AODOnly(process):
from PhysicsTools.PatAlgos.tools.coreTools import restrictInputToAOD
restrictInputToAOD(process)
removeMuonMCClassification(process) # throw the baby out with the bathwater...
removeSimLeptons(process)
