##########
# Flags for additional signal analysis modules
# Perform the signal analysis with all tau ID algorithms in addition
# to the "golden" analysis
doAllTauIds = True
# Perform the signal analysis with the JES variations in addition to
# the "golden" analysis
doJESVariation = False
JESVariation = 0.05
################################################################################
# Command line arguments (options) and DataVersion object
options, dataVersion = getOptionsDataVersion(dataVersion)
################################################################################
# Define the process
process = cms.Process("HChPileupTest")
#process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(-1) )
process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(100) )
process.source = cms.Source('PoolSource',
fileNames = cms.untracked.vstring(
#"rfio:/castor/cern.ch/user/w/wendland/test_pattuplev9_signalM120.root"
"rfio:/castor/cern.ch/user/w/wendland/TTToHplusBWB_M-120_7TeV-pythia6-tauola_Winte10_39X_testsample.root"
#"rfio:/castor/cern.ch/user/w/wendland/test_pattuple_v9_qcd120170.root"
# For testing in lxplus
# dataVersion.getAnalysisDefaultFileCastor()
import FWCore.ParameterSet.Config as cms
from HiggsAnalysis.HeavyChHiggsToTauNu.HChOptions import getOptionsDataVersion
dataVersion = "44XmcS6"
options, dataVersion = getOptionsDataVersion(dataVersion, useDefaultSignalTrigger=False)
process = cms.Process("TauEmbeddingAnalysis")
process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(-1) )
#process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(2000) )
#process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(10000) )
#process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(100) )
#process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(1) )
process.load("Configuration.StandardSequences.FrontierConditions_GlobalTag_cff")
process.GlobalTag.globaltag = cms.string(dataVersion.getGlobalTag())
process.source = cms.Source('PoolSource',
fileNames = cms.untracked.vstring(
"/store/group/local/HiggsChToTauNuFullyHadronic/pattuples/CMSSW_4_4_X/TTJets_TuneZ2_Fall11/TTJets_TuneZ2_7TeV-madgraph-tauola/Fall11_PU_S6_START44_V9B_v1_AODSIM_tauembedding_gentauskim_v44_5/9ecb3a23e436fc2ffd8a803eac2a3a15/pattuple_1012_1_LSv.root",
),
)
process.load("HiggsAnalysis.HeavyChHiggsToTauNu.HChCommon_cfi")
# Fragment to run PAT on the fly if requested from command line
from HiggsAnalysis.HeavyChHiggsToTauNu.HChPatTuple import addPatOnTheFly
patArgs = {"doPatTrigger": False,
# "doPatTaus": False,
# "doHChTauDiscriminators": False,
"doPatElectronID": True,
import FWCore.ParameterSet.Config as cms
from HiggsAnalysis.HeavyChHiggsToTauNu.HChOptions import getOptionsDataVersion
################################################################################
dataVersion = "42XmcS4" # Summer11 MC
#dataVersion="42Xdata" # Run2010 Apr21 ReReco, Run2011 May10 ReReco, Run2011 PromptReco
# Command line arguments (options) and DataVersion object
options, dataVersion = getOptionsDataVersion(dataVersion)
process = cms.Process("HChSignalAnalysis")
#process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(-1) )
process.maxEvents = cms.untracked.PSet(input=cms.untracked.int32(1000))
#process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(10) )
process.load(
"Configuration.StandardSequences.FrontierConditions_GlobalTag_cff")
process.GlobalTag.globaltag = cms.string(dataVersion.getGlobalTag())
process.source = cms.Source(
'PoolSource',
fileNames=cms.untracked.vstring(
# For testing in lxplus
#dataVersion.getAnalysisDefaultFileCastor()
# For testing in jade
dataVersion.getAnalysisDefaultFileMadhatter()
#dataVersion.getAnalysisDefaultFileMadhatterDcap()
))
import FWCore.ParameterSet.Config as cms
from HiggsAnalysis.HeavyChHiggsToTauNu.HChOptions import getOptionsDataVersion
dataVersion = "44XmcS6"
options, dataVersion = getOptionsDataVersion(dataVersion, useDefaultSignalTrigger=False)
process = cms.Process("TauEmbeddingAnalysis")
process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(-1) )
#process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(2000) )
#process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(10000) )
#process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(100) )
#process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(1) )
process.load("Configuration.StandardSequences.FrontierConditions_GlobalTag_cff")
process.GlobalTag.globaltag = cms.string(dataVersion.getGlobalTag())
process.source = cms.Source('PoolSource',
fileNames = cms.untracked.vstring(
"/store/group/local/HiggsChToTauNuFullyHadronic/pattuples/CMSSW_4_4_X/TTJets_TuneZ2_Fall11/TTJets_TuneZ2_7TeV-madgraph-tauola/Fall11_PU_S6_START44_V9B_v1_AODSIM_tauembedding_gentauskim_v44_5/9ecb3a23e436fc2ffd8a803eac2a3a15/pattuple_1012_1_LSv.root",
),
)
process.load("HiggsAnalysis.HeavyChHiggsToTauNu.HChCommon_cfi")
# Fragment to run PAT on the fly if requested from command line
from HiggsAnalysis.HeavyChHiggsToTauNu.HChPatTuple import addPatOnTheFly
patArgs = {"doPatTrigger": False,
# "doPatTaus": False,
# "doHChTauDiscriminators": False,
"doPatElectronID": True,
def customise(process):
# Command line arguments
import FWCore.ParameterSet.VarParsing as VarParsing
options = VarParsing.VarParsing ('analysis')
options.register ('overrideBeamSpot',
0, # default value, false
VarParsing.VarParsing.multiplicity.singleton,
VarParsing.VarParsing.varType.int,
"should I override beamspot in globaltag?")
options, dataVersion = getOptionsDataVersion("42Xmc", options)
hltProcessName = dataVersion.getTriggerProcess()
recoProcessName = dataVersion.getRecoProcess()
processName = process.name_()
# Tighten muon selection w.r.t. skim
import HiggsAnalysis.HeavyChHiggsToTauNu.tauEmbedding.customisations as customisations
process.muonIsolationSequence = cms.Sequence()
muons = customisations.addMuonIsolationEmbedding(process, process.muonIsolationSequence, muons=process.tightenedMuons.src.value())
process.tightenedMuons.src = muons
process.ProductionFilterSequence.replace(process.tightenedMuons, process.muonIsolationSequence*process.tightenedMuons)
# Setup output
outputModule = None
outdict = process.outputModules_()
if len(outdict) == 1:
outputModule = outdict.values()[0]
elif outdict.has_key("RECOSIMoutput"):
outputModule = outdict["RECOSIMoutput"]
print "Adjusting event content to GEN-SIM-RECO+misc"
process.load("HiggsAnalysis.HeavyChHiggsToTauNu.tauEmbedding.HChEventContent_cff")
#outputModule.outputCommands = cms.untracked.vstring("keep *")
outputModule.outputCommands = cms.untracked.vstring("drop *")
outputModule.outputCommands.extend(process.RECOSIMEventContent.outputCommands)
outputModule.outputCommands.extend([
"drop *_*_*_%s" % recoProcessName,
"keep *_genParticles_*_%s" % recoProcessName,
"keep recoGenJets_*_*_%s" % recoProcessName,
"keep recoGenMETs_*_*_%s" % recoProcessName,
"keep *_pfMet_*_%s" % recoProcessName,
"keep *_offlinePrimaryVertices_*_%s" % recoProcessName,
"keep *_generalTracks_*_%s" % recoProcessName,
"keep *_muons_*_%s" % recoProcessName,
"keep *_globalMuons_*_%s" % recoProcessName,
"keep *_offlineBeamSpot_*_%s" % recoProcessName,
"keep *_gtDigis_*_%s" % recoProcessName,
"keep *_l1GtTriggerMenuLite_*_%s" % recoProcessName, # in run block
"keep *_conditionsInEdm_*_%s" % recoProcessName, # in run block
"keep *_addPileupInfo*_*_%s" % recoProcessName, # for MC
"drop *_*_*_%s" % processName,
"keep *_dimuonsGlobal_*_%s" % processName,
"keep *_generator_weight_%s" % processName,
"keep *_genParticles_*_%s" % processName,
"keep recoGenJets_*_*_%s" % processName,
"keep recoGenMETs_*_*_%s" % processName,
"keep edmMergeableCounter_*_*_%s" % processName,
"keep *_particleFlow*_*_%s" % processName,
"keep *_generalTracks_*_%s" % processName,
"keep *_tmfTracks_*_%s" % processName,
"keep *_muons_*_%s" % processName,
"keep *_globalMuons_*_%s" % processName,
"keep *_offlinePrimaryVertices_*_%s" % processName,
"keep edmMergeableCounter_*_*_%s" % processName,
])
# re_procName = re.compile("_\*$")
# outputModule.outputCommands.extend([re_procName.sub("_"+processName, x) for x in process.RECOSIMEventContent.outputCommands])
outputModule.outputCommands.extend(process.HChEventContent.outputCommands)
#outputModule.outputCommands.extend(process.RecoParticleFlowRECO.outputCommands)
#outputModule.outputCommands.extend(["keep *_%s_*_%s" % (x, processName) for x in [
#]])
# Remove duplicate "drop *"
index = 0
for item in outputModule.outputCommands[:]:
if item == "drop *" and index > 0:
del outputModule.outputCommands[index]
index -= 1
index += 1
# Disable gen vertex smearing
process.VtxSmeared = cms.EDProducer("FlatEvtVtxGenerator",
MaxZ = cms.double(0.0),
MaxX = cms.double(0.0),
MaxY = cms.double(0.0),
MinX = cms.double(0.0),
MinY = cms.double(0.0),
MinZ = cms.double(0.0),
TimeOffset = cms.double(0.0),
src = cms.InputTag("generator")
)
print "TAUOLA mdtau =", process.generator.ZTauTau.TauolaOptions.InputCards.mdtau
# Do we have to override the beam spot for data?
if options.overrideBeamSpot != 0:
bs = cms.string("BeamSpotObjects_2009_LumiBased_SigmaZ_v21_offline") # 42x data gt
process.GlobalTag.toGet = cms.VPSet(
cms.PSet(record = cms.string("BeamSpotObjectsRcd"),
tag = bs,
connect = cms.untracked.string("frontier://FrontierProd/CMS_COND_31X_BEAMSPOT")
)
)
print "BeamSpot in globaltag set to ", bs
else:
print "BeamSpot in globaltag not changed"
# Merge tracks
process.tmfTracks = cms.EDProducer("RecoTracksMixer",
trackCol1 = cms.InputTag("dimuonsGlobal"),
trackCol2 = cms.InputTag("generalTracks", "", processName)
)
process.offlinePrimaryVerticesWithBS.TrackLabel = cms.InputTag("tmfTracks")
process.offlinePrimaryVertices.TrackLabel = cms.InputTag("tmfTracks")
process.muons.TrackExtractorPSet.inputTrackCollection = cms.InputTag("tmfTracks")
if hasattr(process, "iterativeTracking"):
process.iterativeTracking *= process.tmfTracks
elif hasattr(process, "trackCollectionMerging"):
process.trackCollectionMerging *= process.tmfTracks
else:
raise Exception("Cannot find tracking sequence")
# it should be the best solution to take the original beam spot for the
# reconstruction of the new primary vertex
# use the one produced earlier, do not produce your own
for s in process.sequences:
seq = getattr(process,s)
seq.remove(process.offlineBeamSpot)
try:
process.metreco.remove(process.BeamHaloId)
except:
pass
# Disable lumi producer
process.localreco_HcalNZS.remove(process.lumiProducer)
process.localreco.remove(process.lumiProducer)
# PFCandidate embedding
process.particleFlowORG = process.particleFlow.clone()
if hasattr(process,"famosParticleFlowSequence"):
process.famosParticleFlowSequence.remove(process.pfPhotonTranslatorSequence)
process.famosParticleFlowSequence.remove(process.pfElectronTranslatorSequence)
process.famosParticleFlowSequence.remove(process.particleFlow)
process.famosParticleFlowSequence.__iadd__(process.particleFlowORG)
process.famosParticleFlowSequence.__iadd__(process.particleFlow)
process.famosParticleFlowSequence.__iadd__(process.pfElectronTranslatorSequence)
process.famosParticleFlowSequence.__iadd__(process.pfPhotonTranslatorSequence)
elif hasattr(process,"particleFlowReco"):
process.particleFlowReco.remove(process.pfPhotonTranslatorSequence)
process.particleFlowReco.remove(process.pfElectronTranslatorSequence)
process.particleFlowReco.remove(process.particleFlow)
process.particleFlowReco.__iadd__(process.particleFlowORG)
process.particleFlowReco.__iadd__(process.particleFlow)
process.particleFlowReco.__iadd__(process.pfElectronTranslatorSequence)
process.particleFlowReco.__iadd__(process.pfPhotonTranslatorSequence)
else:
raise "Cannot find pflow sequence"
process.particleFlow = cms.EDProducer('PFCandidateMixer',
col1 = cms.untracked.InputTag("dimuonsGlobal","forMixing"),
col2 = cms.untracked.InputTag("particleFlowORG", ""),
trackCol = cms.untracked.InputTag("tmfTracks")
)
# Set the empty event filter source
process.filterEmptyEv.src.setProcessName(processName)
# Find all modules having particleFlow as their input
pfInputNeeded = {}
for p in process.paths:
i = getattr(process,p)
target = process.particleFlow
lookForPFInput = ["particleFlow"]
seqVis = SeqVisitor(target)
seqVis.prepareSearch()
seqVis.setLookFor(target)
i.visit(seqVis)
while seqVis.catch != 1 and seqVis.found == 1:
target = seqVis.giveNext()
pfInput = []
targetAttributes = dir(target)
for targetAttribute in targetAttributes:
attr=getattr(target,targetAttribute) # get actual attribute, not just the name
if isinstance(attr, cms.InputTag):
if attr.getModuleLabel()=="particleFlow" and attr.getProductInstanceLabel()!="":
print "Changing: ", target, " ", targetAttribute, " ", attr, " to particleFlowORG"
attr.setModuleLabel("particleFlowORG")
if attr.getModuleLabel() in lookForPFInput:
pfInput.append(attr)
if len(pfInput) > 0:
lookForPFInput.append(target.label())
pfInputNeeded[target.label()] = pfInput
#i.replace(target, source)
seqVis.prepareSearch()
seqVis.setLookFor(target)
i.visit(seqVis)
#if (seqVis.catch==1):
#seqVis.catch=0
#i.__iadd__(source)
pfOutputCommands = []
for label in pfInputNeeded.keys():
print "particleFlow as input in module %s, InputTags: %s" % (label, ", ".join(str(x) for x in pfInputNeeded[label]))
pfOutputCommands.append("keep *_%s_*_%s" % (label, processName))
outputModule.outputCommands.extend(pfOutputCommands)
process.pfSelectedElectrons.src = "particleFlowORG"
process.pfSelectedPhotons.src = "particleFlowORG"
# Setup/remove some HLT/DQM stuff whcih doesn't work
if hasattr(process, "hltTrigReport"):
process.hltTrigReport.HLTriggerResults.setProcessName(processName)
if hasattr(process, "DQM_FEDIntegrity_v2"):
process.schedule.remove(process.DQM_FEDIntegrity_v2)
if hasattr(process, "DQM_FEDIntegrity_v3"):
process.schedule.remove(process.DQM_FEDIntegrity_v3)
if hasattr(process, "HLTAnalyzerEndpath"):
process.schedule.remove(process.HLTAnalyzerEndpath)
del process.HLTAnalyzerEndpath
#process.load("HiggsAnalysis.HeavyChHiggsToTauNu.tauEmbedding.printGenParticles_cff")
#process.generation_step *= process.printGenParticles
print "#############################################################"
print " Warning! PFCandidates 'electron' collection is not mixed, "
print " and probably shouldnt be used. "
print "#############################################################"
return process
def customise(process):
# Catch the case when this config is run from cmsDriver, it won't work due to VarParsing
# First protect against crab job creation, then the no-argument case
if hasattr(sys, "argv") and len(sys.argv) > 0:
if "cmsDriver" in sys.argv[0]:
print "Running pf_customise from cmsDriver, not executing running() further due to VarParsing"
return
else:
print "Running pf_customise"
# Command line arguments
import FWCore.ParameterSet.VarParsing as VarParsing
options = VarParsing.VarParsing('analysis')
options.register(
'overrideBeamSpot',
0, # default value, false
VarParsing.VarParsing.multiplicity.singleton,
VarParsing.VarParsing.varType.int,
"should I override beamspot in globaltag?")
options.register(
"tauDecayMode",
0, # Default is all decays
VarParsing.VarParsing.multiplicity.singleton,
VarParsing.VarParsing.varType.int,
"Tau decay mode (0=all, 230=hadronic)")
options.register(
"tauMinVisPt",
-1, # Disabled
VarParsing.VarParsing.multiplicity.singleton,
VarParsing.VarParsing.varType.int,
"Minimum visible pt of tau decay (-1 disabled, >= 0 cut value in GeV)")
options, dataVersion = getOptionsDataVersion("53XmcS10", options)
hltProcessName = dataVersion.getTriggerProcess()
recoProcessName = dataVersion.getRecoProcess()
processName = process.name_()
# Setup trigger matching
if not (dataVersion.isMC() and options.triggerMC == 0
and options.triggerMCInAnalysis == 0):
HChTriggerMatching.setMuonTriggerMatchingInAnalysis(
process.tightenedMuonsMatched, options.trigger)
# Setup MuScleFit
if dataVersion.isMC():
process.muscleCorrectedMuons.identifier = "Summer12_DR53X_smearReReco"
process.muscleCorrectedMuons.applySmearing = True
else:
process.muscleCorrectedMuons.identifier = "Data2012_53X_ReReco"
# Setup output
outputModule = None
outdict = process.outputModules_()
if len(outdict) == 1:
outputModule = outdict.values()[0]
elif outdict.has_key("RECOSIMoutput"):
outputModule = outdict["RECOSIMoutput"]
print "Adjusting event content to GEN-SIM-RECO+misc"
process.load(
"HiggsAnalysis.HeavyChHiggsToTauNu.tauEmbedding.HChEventContent_cff")
#outputModule.outputCommands = cms.untracked.vstring("keep *")
outputModule.outputCommands = cms.untracked.vstring("drop *")
outputModule.outputCommands.extend(
process.RECOSIMEventContent.outputCommands)
outputModule.outputCommands.extend([
"drop *_*_*_%s" % recoProcessName,
"keep *_generalTracks_*_%s" % recoProcessName,
"keep *_muons_*_%s" % recoProcessName,
"keep *_globalMuons_*_%s" % recoProcessName,
"keep recoGsfElectronCores_*_*_%s" % recoProcessName,
"keep *_gsfElectrons_*_%s" % recoProcessName,
"keep *_photons_*_%s" % recoProcessName,
"keep *_photonCore_*_%s" % recoProcessName,
"drop *_*_*_%s" % processName,
"keep *_particleFlow*_*_%s" % processName,
"keep *_generalTracks_*_%s" % processName,
"keep *_muons_*_%s" % processName,
"keep *_globalMuons_*_%s" % processName,
"keep *_*Electron*_*_%s" % processName,
"keep *_eid*_*_*",
])
outputModule.outputCommands.extend(
eventContent(hltProcessName, recoProcessName, processName))
# re_procName = re.compile("_\*$")
# outputModule.outputCommands.extend([re_procName.sub("_"+processName, x) for x in process.RECOSIMEventContent.outputCommands])
outputModule.outputCommands.extend(process.HChEventContent.outputCommands)
#outputModule.outputCommands.extend(process.RecoParticleFlowRECO.outputCommands)
#outputModule.outputCommands.extend(["keep *_%s_*_%s" % (x, processName) for x in [
#]])
# Remove duplicate "drop *"
index = 0
for item in outputModule.outputCommands[:]:
if item == "drop *" and index > 0:
del outputModule.outputCommands[index]
index -= 1
index += 1
# Disable gen vertex smearing
process.VtxSmeared = cms.EDProducer("FlatEvtVtxGenerator",
MaxZ=cms.double(0.0),
MaxX=cms.double(0.0),
MaxY=cms.double(0.0),
MinX=cms.double(0.0),
MinY=cms.double(0.0),
MinZ=cms.double(0.0),
TimeOffset=cms.double(0.0),
src=cms.InputTag("generator"))
# Set up tau decay options
process.generator.ZTauTau.TauolaOptions.InputCards.mdtau = options.tauDecayMode
if options.tauMinVisPt >= 0:
process.generator.ZTauTau.minVisibleTransverseMomentum = "%d" % options.tauMinVisPt
print "TAUOLA mdtau =", process.generator.ZTauTau.TauolaOptions.InputCards.mdtau
# Do we have to override the beam spot for data?
if options.overrideBeamSpot != 0:
bs = cms.string(
"BeamSpotObjects_2009_LumiBased_SigmaZ_v25_offline") # 44x data gt
#bs = cms.string("BeamSpotObjects_2009_LumiBased_SigmaZ_v21_offline") # 42x data gt
process.GlobalTag.toGet = cms.VPSet(
cms.PSet(record=cms.string("BeamSpotObjectsRcd"),
tag=bs,
connect=cms.untracked.string(
"frontier://FrontierProd/CMS_COND_31X_BEAMSPOT")))
print "BeamSpot in globaltag set to ", bs
else:
print "BeamSpot in globaltag not changed"
# Merge tracks
process.tmfTracks = cms.EDProducer(
"RecoTracksMixer",
trackCol1=cms.InputTag("dimuonsGlobal", "tracks"),
trackCol2=cms.InputTag("generalTracks", "", processName))
process.offlinePrimaryVerticesWithBS.TrackLabel = cms.InputTag("tmfTracks")
process.offlinePrimaryVertices.TrackLabel = cms.InputTag("tmfTracks")
#print process.muons
if hasattr(process.muons, "TrackExtractorPSet"):
# <= 42X
process.muons.TrackExtractorPSet.inputTrackCollection = cms.InputTag(
"tmfTracks")
elif hasattr(process, "muons1stStep") and hasattr(process.muons1stStep,
"TrackExtractorPSet"):
# >= 44X
process.muons1stStep.TrackExtractorPSet.inputTrackCollection = cms.InputTag(
"tmfTracks")
else:
raise Exception(
"Problem in overriding track collection for reco::Muon producer")
# Ensure that tmfTracks is always run after generalTracks (to mix the original and embedded tracks)
for p in process.paths:
pth = getattr(process, p)
if "generalTracks" in pth.moduleNames():
pth.replace(process.generalTracks,
process.generalTracks * process.tmfTracks)
# it should be the best solution to take the original beam spot for the
# reconstruction of the new primary vertex
# use the one produced earlier, do not produce your own
for s in process.sequences:
seq = getattr(process, s)
seq.remove(process.offlineBeamSpot)
# Remove beam halo Id
try:
process.metreco.remove(process.BeamHaloId)
except:
pass
# Disable lumi producer
process.localreco_HcalNZS.remove(process.lumiProducer)
process.localreco.remove(process.lumiProducer)
# PFCandidate embedding
process.particleFlowORG = process.particleFlow.clone()
# Since CMSSW 4_4 the particleFlow reco works a bit differently. The step is
# twofold, first particleFlowTmp is created and then the final particleFlow
# collection. What we do in this case is that we merge the final ParticleFlow
# collection. For the muon reconstruction, we also merge particleFlowTmp in
# order to get PF-based isolation right.
if hasattr(process, 'particleFlowTmp'):
process.particleFlowTmpMixed = cms.EDProducer(
'PFCandidateMixer',
col1=cms.untracked.InputTag("dimuonsGlobal", "pfCands"),
col2=cms.untracked.InputTag("particleFlowTmp", ""),
trackCol=cms.untracked.InputTag("tmfTracks"),
# Don't produce value maps:
muons=cms.untracked.InputTag(""),
gsfElectrons=cms.untracked.InputTag(""))
process.muons.PFCandidates = cms.InputTag("particleFlowTmpMixed")
for p in process.paths:
pth = getattr(process, p)
if "particleFlow" in pth.moduleNames():
pth.replace(process.particleFlow,
process.particleFlowORG * process.particleFlow)
if "muons" in pth.moduleNames():
pth.replace(process.muons,
process.particleFlowTmpMixed * process.muons)
else:
# CMSSW_4_2
if hasattr(process, "famosParticleFlowSequence"):
process.famosParticleFlowSequence.remove(
process.pfPhotonTranslatorSequence)
process.famosParticleFlowSequence.remove(
process.pfElectronTranslatorSequence)
process.famosParticleFlowSequence.remove(process.particleFlow)
process.famosParticleFlowSequence.__iadd__(process.particleFlowORG)
process.famosParticleFlowSequence.__iadd__(process.particleFlow)
process.famosParticleFlowSequence.__iadd__(
process.pfElectronTranslatorSequence)
process.famosParticleFlowSequence.__iadd__(
process.pfPhotonTranslatorSequence)
elif hasattr(process, "particleFlowReco"):
process.particleFlowReco.remove(process.pfPhotonTranslatorSequence)
process.particleFlowReco.remove(
process.pfElectronTranslatorSequence)
process.particleFlowReco.remove(process.particleFlow)
process.particleFlowReco.__iadd__(process.particleFlowORG)
process.particleFlowReco.__iadd__(process.particleFlow)
process.particleFlowReco.__iadd__(
process.pfElectronTranslatorSequence)
process.particleFlowReco.__iadd__(
process.pfPhotonTranslatorSequence)
else:
raise "Cannot find pflow sequence"
process.pfSelectedElectrons.src = cms.InputTag("particleFlowORG")
process.pfSelectedPhotons.src = cms.InputTag("particleFlowORG")
process.particleFlow = cms.EDProducer(
'PFCandidateMixer',
col1=cms.untracked.InputTag("dimuonsGlobal", "pfCands"),
col2=cms.untracked.InputTag("particleFlowORG", ""),
trackCol=cms.untracked.InputTag("tmfTracks"),
muons=cms.untracked.InputTag("muons"),
gsfElectrons=cms.untracked.InputTag(
"gsfElectrons", "", recoProcessName) # FIXME does this work?
#gsfElectrons = cms.untracked.InputTag("")
)
# Set the empty event filter source
process.filterEmptyEv.src.setProcessName(processName)
# Find all modules having particleFlow as their input
pfInputNeeded = {}
for p in process.paths:
i = getattr(process, p)
target = process.particleFlow
lookForPFInput = ["particleFlow"]
seqVis = SeqVisitor(target)
seqVis.prepareSearch()
seqVis.setLookFor(target)
i.visit(seqVis)
while seqVis.catch != 1 and seqVis.found == 1:
target = seqVis.giveNext()
pfInput = []
targetAttributes = dir(target)
for targetAttribute in targetAttributes:
attr = getattr(target, targetAttribute
) # get actual attribute, not just the name
if isinstance(attr, cms.InputTag):
if attr.getModuleLabel(
) == "particleFlow" and attr.getProductInstanceLabel(
) != "":
print "Changing: ", target, " ", targetAttribute, " ", attr, " to particleFlowORG"
attr.setModuleLabel("particleFlowORG")
if attr.getModuleLabel() in lookForPFInput:
pfInput.append(attr)
if len(pfInput) > 0:
lookForPFInput.append(target.label())
pfInputNeeded[target.label()] = pfInput
#i.replace(target, source)
seqVis.prepareSearch()
seqVis.setLookFor(target)
i.visit(seqVis)
#if (seqVis.catch==1):
#seqVis.catch=0
#i.__iadd__(source)
pfOutputCommands = []
for label in pfInputNeeded.keys():
print "particleFlow as input in module %s, InputTags: %s" % (
label, ", ".join(str(x) for x in pfInputNeeded[label]))
pfOutputCommands.append("keep *_%s_*_%s" % (label, processName))
outputModule.outputCommands.extend(pfOutputCommands)
#process.pfSelectedElectrons.src = "particleFlowORG" # 4_2 legacy, already included above
#process.pfSelectedPhotons.src = "particleFlowORG" # 4_2 legacy, already included above
# Setup/remove some HLT/DQM stuff whcih doesn't work
if hasattr(process, "hltTrigReport"):
process.hltTrigReport.HLTriggerResults.setProcessName(processName)
if hasattr(process, "DQM_FEDIntegrity_v2"):
process.schedule.remove(process.DQM_FEDIntegrity_v2)
if hasattr(process, "DQM_FEDIntegrity_v3"):
process.schedule.remove(process.DQM_FEDIntegrity_v3)
if hasattr(process, "DQM_FEDIntegrity_v5"):
process.schedule.remove(process.DQM_FEDIntegrity_v5)
if hasattr(process, "HLTAnalyzerEndpath"):
process.schedule.remove(process.HLTAnalyzerEndpath)
del process.HLTAnalyzerEndpath
#process.load("HiggsAnalysis.HeavyChHiggsToTauNu.tauEmbedding.printGenParticles_cff")
#process.generation_step *= process.printGenParticles
print "#############################################################"
print " Warning! PFCandidates 'electron' collection is not mixed, "
print " and probably shouldnt be used. "
print "#############################################################"
addPAT(process, options, dataVersion)
f = open("configDumpEmbed.py", "w")
f.write(process.dumpPython())
f.close()
return process
def customise(process):
# Command line arguments
import FWCore.ParameterSet.VarParsing as VarParsing
options = VarParsing.VarParsing('analysis')
options.register(
'overrideBeamSpot',
0, # default value, false
VarParsing.VarParsing.multiplicity.singleton,
VarParsing.VarParsing.varType.int,
"should I override beamspot in globaltag?")
options, dataVersion = getOptionsDataVersion("42Xmc", options)
processName = process.name_()
#process.source.duplicateCheckMode = cms.untracked.string('noDuplicateCheck')
hltProcessName = dataVersion.getTriggerProcess()
recoProcessName = dataVersion.getRecoProcess()
generationProcessName = "EMBEDDINGHLT"
processName = process.name_()
# Track embedding
# replaceGeneralTracks = False
# if replaceGeneralTracks:
# process.generalTracksORG = process.generalTracks.clone()
# process.trackCollectionMerging.remove(process.generalTracks)
# process.generalTracks = cms.EDProducer("RecoTracksMixer",
# trackCol1 = cms.InputTag("dimuonsGlobal"),
# trackCol2 = cms.InputTag("generalTracksORG")
# )
# process.trackCollectionMerging *= process.generalTracksORG
# process.trackCollectionMerging *= process.generalTracks
# process.doAlldEdXEstimators = cms.Sequence() # disable these, require Trajectories
# replaceGeneralTracks = lambda x: replaceInputTag(x, "generalTracks", "generalTracksORG")
# process.trackerDrivenElectronSeeds.TkColList = [replaceGeneralTracks(tag) for tag in process.trackerDrivenElectronSeeds.TkColList]
# process.generalConversionTrackProducer.TrackProducer = "generalTracksORG"
# process.muons.inputCollectionLabels = [replaceGeneralTracks(tag) for tag in process.muons.inputCollectionLabels]
# for tag in [#process.muons.TrackExtractorPSet.inputTrackCollection,
# process.gsfElectronCores.ctfTracks,
# process.generalV0Candidates.trackRecoAlgorithm,
# process.particleFlowDisplacedVertexCandidate.trackCollection,
# process.pfDisplacedTrackerVertex.trackColl,
# ]:
# replaceGeneralTracks(tag)
# else:
process.tmfTracks = cms.EDProducer("RecoTracksMixer",
trackCol1=cms.InputTag("dimuonsGlobal"),
trackCol2=cms.InputTag("generalTracks"))
process.trackCollectionMerging *= process.tmfTracks
process.offlinePrimaryVerticesWithBS.TrackLabel = cms.InputTag("tmfTracks")
process.offlinePrimaryVertices.TrackLabel = cms.InputTag("tmfTracks")
process.muons.TrackExtractorPSet.inputTrackCollection = cms.InputTag(
"tmfTracks")
# it should be the best solution to take the original beam spot for the
# reconstruction of the new primary vertex
#process.offlinePrimaryVertices.beamSpotLabel = cms.InputTag("offlineBeamSpot", "", recoProcessName)
#process.offlinePrimaryVerticesWithBS.beamSpotLabel = cms.InputTag("offlineBeamSpot", "", recoProcessName)
try:
process.metreco.remove(process.BeamHaloId)
except:
pass
outputModule = None
outdict = process.outputModules_()
if len(outdict) == 1:
outputModule = outdict.values()[0]
elif outdict.has_key("out"):
outputModule = outdict["out"]
print "Adjusting event content to GEN-SIM-RECO+misc"
process.load(
"HiggsAnalysis.HeavyChHiggsToTauNu.tauEmbedding.HChEventContent_cff")
outputModule.outputCommands = cms.untracked.vstring("drop *")
outputModule.outputCommands.extend(
process.RECOSIMEventContent.outputCommands)
outputModule.outputCommands.extend([
"drop *_*_*_%s" % recoProcessName,
"keep *_genParticles_*_%s" % recoProcessName,
"keep recoGenJets_*_*_%s" % recoProcessName,
"keep recoGenMETs_*_*_%s" % recoProcessName,
"keep *_pfMet_*_%s" % recoProcessName,
"keep *_offlinePrimaryVertices_*_%s" % recoProcessName,
"keep *_generalTracks_*_%s" % recoProcessName,
"keep *_muons_*_%s" % recoProcessName,
"keep *_globalMuons_*_%s" % recoProcessName,
"keep *_offlineBeamSpot_*_%s" % recoProcessName,
"keep *_gtDigis_*_%s" % recoProcessName,
"keep *_l1GtTriggerMenuLite_*_%s" % recoProcessName, # in run block
"keep *_conditionsInEdm_*_%s" % recoProcessName, # in run block
"keep *_addPileupInfo*_*_%s" % recoProcessName, # for MC
"drop *_*_*_%s" % generationProcessName,
"keep *_dimuonsGlobal_*_%s" % generationProcessName,
"keep *_generator_weight_%s" % generationProcessName,
"keep *_genParticles_*_%s" % generationProcessName,
"keep recoGenJets_*_*_%s" % generationProcessName,
"keep recoGenMETs_*_*_%s" % generationProcessName,
"keep edmMergeableCounter_*_*_%s" % generationProcessName,
"drop *_*_*_%s" % processName,
"keep *_particleFlow*_*_%s" % processName,
"keep *_generalTracks_*_%s" % processName,
"keep *_tmfTracks_*_%s" % processName,
"keep *_muons_*_%s" % processName,
"keep *_globalMuons_*_%s" % processName,
"keep *_offlinePrimaryVertices_*_%s" % processName,
"keep edmMergeableCounter_*_*_%s" % processName,
])
# re_procName = re.compile("_\*$")
# outputModule.outputCommands.extend([re_procName.sub("_"+processName, x) for x in process.RECOSIMEventContent.outputCommands])
outputModule.outputCommands.extend(process.HChEventContent.outputCommands)
#outputModule.outputCommands.extend(process.RecoParticleFlowRECO.outputCommands)
#outputModule.outputCommands.extend(["keep *_%s_*_%s" % (x, processName) for x in [
#]])
# Remove duplicate "drop *"
index = 0
for item in outputModule.outputCommands[:]:
if item == "drop *" and index > 0:
del outputModule.outputCommands[index]
index -= 1
index += 1
# Disable lumi producer
process.localreco_HcalNZS.remove(process.lumiProducer)
process.localreco.remove(process.lumiProducer)
# PFCandidate embedding
process.particleFlowORG = process.particleFlow.clone()
if hasattr(process, "famosParticleFlowSequence"):
process.famosParticleFlowSequence.remove(
process.pfPhotonTranslatorSequence)
process.famosParticleFlowSequence.remove(
process.pfElectronTranslatorSequence)
process.famosParticleFlowSequence.remove(process.particleFlow)
process.famosParticleFlowSequence.__iadd__(process.particleFlowORG)
process.famosParticleFlowSequence.__iadd__(process.particleFlow)
process.famosParticleFlowSequence.__iadd__(
process.pfElectronTranslatorSequence)
process.famosParticleFlowSequence.__iadd__(
process.pfPhotonTranslatorSequence)
elif hasattr(process, "particleFlowReco"):
process.particleFlowReco.remove(process.pfPhotonTranslatorSequence)
process.particleFlowReco.remove(process.pfElectronTranslatorSequence)
process.particleFlowReco.remove(process.particleFlow)
process.particleFlowReco.__iadd__(process.particleFlowORG)
process.particleFlowReco.__iadd__(process.particleFlow)
process.particleFlowReco.__iadd__(process.pfElectronTranslatorSequence)
process.particleFlowReco.__iadd__(process.pfPhotonTranslatorSequence)
else:
raise "Cannot find tracking sequence"
process.particleFlow = cms.EDProducer(
'PFCandidateMixer',
col1=cms.untracked.InputTag("dimuonsGlobal", "forMixing"),
col2=cms.untracked.InputTag("particleFlowORG", ""),
trackCol=cms.untracked.InputTag("tmfTracks"))
pfInputNeeded = {}
for p in process.paths:
i = getattr(process, p)
target = process.particleFlow
lookForPFInput = ["particleFlow"]
seqVis = SeqVisitor(target)
seqVis.prepareSearch()
seqVis.setLookFor(target)
i.visit(seqVis)
while seqVis.catch != 1 and seqVis.found == 1:
target = seqVis.giveNext()
pfInput = []
targetAttributes = dir(target)
for targetAttribute in targetAttributes:
attr = getattr(target, targetAttribute
) # get actual attribute, not just the name
if isinstance(attr, cms.InputTag):
if attr.getModuleLabel(
) == "particleFlow" and attr.getProductInstanceLabel(
) != "":
print "Changing: ", target, " ", targetAttribute, " ", attr, " to particleFlowORG"
attr.setModuleLabel("particleFlowORG")
if attr.getModuleLabel() in lookForPFInput:
pfInput.append(attr)
if len(pfInput) > 0:
lookForPFInput.append(target.label())
pfInputNeeded[target.label()] = pfInput
#i.replace(target, source)
seqVis.prepareSearch()
seqVis.setLookFor(target)
i.visit(seqVis)
#if (seqVis.catch==1):
#seqVis.catch=0
#i.__iadd__(source)
pfOutputCommands = []
for label in pfInputNeeded.keys():
print "particleFlow as input in module %s, InputTags: %s" % (
label, ", ".join(str(x) for x in pfInputNeeded[label]))
pfOutputCommands.append("keep *_%s_*_%s" % (label, processName))
outputModule.outputCommands.extend(pfOutputCommands)
process.pfSelectedElectrons.src = "particleFlowORG"
process.pfSelectedPhotons.src = "particleFlowORG"
if options.overrideBeamSpot != 0:
bs = cms.string(
"BeamSpotObjects_2009_LumiBased_SigmaZ_v21_offline") # 42x data gt
#bs = cms.string("BeamSpotObjects_2009_LumiBased_SigmaZ_v18_offline") # 39x data gt
#bs = cms.string("BeamSpotObjects_2009_LumiBased_v17_offline") # 38x data gt
#bs = cms.string("BeamSpotObjects_2009_v14_offline") # 36x data gt
# tag = cms.string("Early10TeVCollision_3p8cm_31X_v1_mc_START"), # 35 default
# tag = cms.string("Realistic900GeVCollisions_10cm_STARTUP_v1_mc"), # 36 default
process.GlobalTag.toGet = cms.VPSet(
cms.PSet(record=cms.string("BeamSpotObjectsRcd"),
tag=bs,
connect=cms.untracked.string(
"frontier://FrontierProd/CMS_COND_31X_BEAMSPOT")))
print "BeamSpot in globaltag set to ", bs
else:
print "BeamSpot in globaltag not changed"
# Remove beamspot producer in order to use the original beamspot always
for s in process.sequences:
seq = getattr(process, s)
seq.remove(process.offlineBeamSpot)
print "#############################################################"
print " Warning! PFCandidates 'electron' collection is not mixed, "
print " and probably shouldnt be used. "
print "#############################################################"
return process
def customise(process):
# Command line arguments
import FWCore.ParameterSet.VarParsing as VarParsing
options = VarParsing.VarParsing("analysis")
options.register(
"overrideBeamSpot",
0, # default value, false
VarParsing.VarParsing.multiplicity.singleton,
VarParsing.VarParsing.varType.int,
"should I override beamspot in globaltag?",
)
options, dataVersion = getOptionsDataVersion("42Xmc", options)
processName = process.name_()
# process.source.duplicateCheckMode = cms.untracked.string('noDuplicateCheck')
hltProcessName = dataVersion.getTriggerProcess()
recoProcessName = dataVersion.getRecoProcess()
generationProcessName = "EMBEDDINGHLT"
processName = process.name_()
# Track embedding
# replaceGeneralTracks = False
# if replaceGeneralTracks:
# process.generalTracksORG = process.generalTracks.clone()
# process.trackCollectionMerging.remove(process.generalTracks)
# process.generalTracks = cms.EDProducer("RecoTracksMixer",
# trackCol1 = cms.InputTag("dimuonsGlobal"),
# trackCol2 = cms.InputTag("generalTracksORG")
# )
# process.trackCollectionMerging *= process.generalTracksORG
# process.trackCollectionMerging *= process.generalTracks
# process.doAlldEdXEstimators = cms.Sequence() # disable these, require Trajectories
# replaceGeneralTracks = lambda x: replaceInputTag(x, "generalTracks", "generalTracksORG")
# process.trackerDrivenElectronSeeds.TkColList = [replaceGeneralTracks(tag) for tag in process.trackerDrivenElectronSeeds.TkColList]
# process.generalConversionTrackProducer.TrackProducer = "generalTracksORG"
# process.muons.inputCollectionLabels = [replaceGeneralTracks(tag) for tag in process.muons.inputCollectionLabels]
# for tag in [#process.muons.TrackExtractorPSet.inputTrackCollection,
# process.gsfElectronCores.ctfTracks,
# process.generalV0Candidates.trackRecoAlgorithm,
# process.particleFlowDisplacedVertexCandidate.trackCollection,
# process.pfDisplacedTrackerVertex.trackColl,
# ]:
# replaceGeneralTracks(tag)
# else:
process.tmfTracks = cms.EDProducer(
"RecoTracksMixer", trackCol1=cms.InputTag("dimuonsGlobal"), trackCol2=cms.InputTag("generalTracks")
)
process.trackCollectionMerging *= process.tmfTracks
process.offlinePrimaryVerticesWithBS.TrackLabel = cms.InputTag("tmfTracks")
process.offlinePrimaryVertices.TrackLabel = cms.InputTag("tmfTracks")
process.muons.TrackExtractorPSet.inputTrackCollection = cms.InputTag("tmfTracks")
# it should be the best solution to take the original beam spot for the
# reconstruction of the new primary vertex
# process.offlinePrimaryVertices.beamSpotLabel = cms.InputTag("offlineBeamSpot", "", recoProcessName)
# process.offlinePrimaryVerticesWithBS.beamSpotLabel = cms.InputTag("offlineBeamSpot", "", recoProcessName)
try:
process.metreco.remove(process.BeamHaloId)
except:
pass
outputModule = None
outdict = process.outputModules_()
if len(outdict) == 1:
outputModule = outdict.values()[0]
elif outdict.has_key("out"):
outputModule = outdict["out"]
print "Adjusting event content to GEN-SIM-RECO+misc"
process.load("HiggsAnalysis.HeavyChHiggsToTauNu.tauEmbedding.HChEventContent_cff")
outputModule.outputCommands = cms.untracked.vstring("drop *")
outputModule.outputCommands.extend(process.RECOSIMEventContent.outputCommands)
outputModule.outputCommands.extend(
[
"drop *_*_*_%s" % recoProcessName,
"keep *_genParticles_*_%s" % recoProcessName,
"keep recoGenJets_*_*_%s" % recoProcessName,
"keep recoGenMETs_*_*_%s" % recoProcessName,
"keep *_pfMet_*_%s" % recoProcessName,
"keep *_offlinePrimaryVertices_*_%s" % recoProcessName,
"keep *_generalTracks_*_%s" % recoProcessName,
"keep *_muons_*_%s" % recoProcessName,
"keep *_globalMuons_*_%s" % recoProcessName,
"keep *_offlineBeamSpot_*_%s" % recoProcessName,
"keep *_gtDigis_*_%s" % recoProcessName,
"keep *_l1GtTriggerMenuLite_*_%s" % recoProcessName, # in run block
"keep *_conditionsInEdm_*_%s" % recoProcessName, # in run block
"keep *_addPileupInfo*_*_%s" % recoProcessName, # for MC
"drop *_*_*_%s" % generationProcessName,
"keep *_dimuonsGlobal_*_%s" % generationProcessName,
"keep *_generator_weight_%s" % generationProcessName,
"keep *_genParticles_*_%s" % generationProcessName,
"keep recoGenJets_*_*_%s" % generationProcessName,
"keep recoGenMETs_*_*_%s" % generationProcessName,
"keep edmMergeableCounter_*_*_%s" % generationProcessName,
"drop *_*_*_%s" % processName,
"keep *_particleFlow*_*_%s" % processName,
"keep *_generalTracks_*_%s" % processName,
"keep *_tmfTracks_*_%s" % processName,
"keep *_muons_*_%s" % processName,
"keep *_globalMuons_*_%s" % processName,
"keep *_offlinePrimaryVertices_*_%s" % processName,
"keep edmMergeableCounter_*_*_%s" % processName,
]
)
# re_procName = re.compile("_\*$")
# outputModule.outputCommands.extend([re_procName.sub("_"+processName, x) for x in process.RECOSIMEventContent.outputCommands])
outputModule.outputCommands.extend(process.HChEventContent.outputCommands)
# outputModule.outputCommands.extend(process.RecoParticleFlowRECO.outputCommands)
# outputModule.outputCommands.extend(["keep *_%s_*_%s" % (x, processName) for x in [
# ]])
# Remove duplicate "drop *"
index = 0
for item in outputModule.outputCommands[:]:
if item == "drop *" and index > 0:
del outputModule.outputCommands[index]
index -= 1
index += 1
# Disable lumi producer
process.localreco_HcalNZS.remove(process.lumiProducer)
process.localreco.remove(process.lumiProducer)
# PFCandidate embedding
process.particleFlowORG = process.particleFlow.clone()
if hasattr(process, "famosParticleFlowSequence"):
process.famosParticleFlowSequence.remove(process.pfPhotonTranslatorSequence)
process.famosParticleFlowSequence.remove(process.pfElectronTranslatorSequence)
process.famosParticleFlowSequence.remove(process.particleFlow)
process.famosParticleFlowSequence.__iadd__(process.particleFlowORG)
process.famosParticleFlowSequence.__iadd__(process.particleFlow)
process.famosParticleFlowSequence.__iadd__(process.pfElectronTranslatorSequence)
process.famosParticleFlowSequence.__iadd__(process.pfPhotonTranslatorSequence)
elif hasattr(process, "particleFlowReco"):
process.particleFlowReco.remove(process.pfPhotonTranslatorSequence)
process.particleFlowReco.remove(process.pfElectronTranslatorSequence)
process.particleFlowReco.remove(process.particleFlow)
process.particleFlowReco.__iadd__(process.particleFlowORG)
process.particleFlowReco.__iadd__(process.particleFlow)
process.particleFlowReco.__iadd__(process.pfElectronTranslatorSequence)
process.particleFlowReco.__iadd__(process.pfPhotonTranslatorSequence)
else:
raise "Cannot find tracking sequence"
process.particleFlow = cms.EDProducer(
"PFCandidateMixer",
col1=cms.untracked.InputTag("dimuonsGlobal", "forMixing"),
col2=cms.untracked.InputTag("particleFlowORG", ""),
trackCol=cms.untracked.InputTag("tmfTracks"),
)
pfInputNeeded = {}
for p in process.paths:
i = getattr(process, p)
target = process.particleFlow
lookForPFInput = ["particleFlow"]
seqVis = SeqVisitor(target)
seqVis.prepareSearch()
seqVis.setLookFor(target)
i.visit(seqVis)
while seqVis.catch != 1 and seqVis.found == 1:
target = seqVis.giveNext()
pfInput = []
targetAttributes = dir(target)
for targetAttribute in targetAttributes:
attr = getattr(target, targetAttribute) # get actual attribute, not just the name
if isinstance(attr, cms.InputTag):
if attr.getModuleLabel() == "particleFlow" and attr.getProductInstanceLabel() != "":
print "Changing: ", target, " ", targetAttribute, " ", attr, " to particleFlowORG"
attr.setModuleLabel("particleFlowORG")
if attr.getModuleLabel() in lookForPFInput:
pfInput.append(attr)
if len(pfInput) > 0:
lookForPFInput.append(target.label())
pfInputNeeded[target.label()] = pfInput
# i.replace(target, source)
seqVis.prepareSearch()
seqVis.setLookFor(target)
i.visit(seqVis)
# if (seqVis.catch==1):
# seqVis.catch=0
# i.__iadd__(source)
pfOutputCommands = []
for label in pfInputNeeded.keys():
print "particleFlow as input in module %s, InputTags: %s" % (
label,
", ".join(str(x) for x in pfInputNeeded[label]),
)
pfOutputCommands.append("keep *_%s_*_%s" % (label, processName))
outputModule.outputCommands.extend(pfOutputCommands)
process.pfSelectedElectrons.src = "particleFlowORG"
process.pfSelectedPhotons.src = "particleFlowORG"
if options.overrideBeamSpot != 0:
bs = cms.string("BeamSpotObjects_2009_LumiBased_SigmaZ_v21_offline") # 42x data gt
# bs = cms.string("BeamSpotObjects_2009_LumiBased_SigmaZ_v18_offline") # 39x data gt
# bs = cms.string("BeamSpotObjects_2009_LumiBased_v17_offline") # 38x data gt
# bs = cms.string("BeamSpotObjects_2009_v14_offline") # 36x data gt
# tag = cms.string("Early10TeVCollision_3p8cm_31X_v1_mc_START"), # 35 default
# tag = cms.string("Realistic900GeVCollisions_10cm_STARTUP_v1_mc"), # 36 default
process.GlobalTag.toGet = cms.VPSet(
cms.PSet(
record=cms.string("BeamSpotObjectsRcd"),
tag=bs,
connect=cms.untracked.string("frontier://FrontierProd/CMS_COND_31X_BEAMSPOT"),
)
)
print "BeamSpot in globaltag set to ", bs
else:
print "BeamSpot in globaltag not changed"
# Remove beamspot producer in order to use the original beamspot always
for s in process.sequences:
seq = getattr(process, s)
seq.remove(process.offlineBeamSpot)
print "#############################################################"
print " Warning! PFCandidates 'electron' collection is not mixed, "
print " and probably shouldnt be used. "
print "#############################################################"
return process
def customise(process):
# Catch the case when this config is run from cmsDriver, it won't work due to VarParsing
# First protect against crab job creation, then the no-argument case
if hasattr(sys, "argv") and len(sys.argv) > 0:
if "cmsDriver" in sys.argv[0]:
print "Running pf_customise from cmsDriver, not executing running() further due to VarParsing"
return
else:
print "Running pf_customise"
# Command line arguments
import FWCore.ParameterSet.VarParsing as VarParsing
options = VarParsing.VarParsing ('analysis')
options.register ('overrideBeamSpot',
0, # default value, false
VarParsing.VarParsing.multiplicity.singleton,
VarParsing.VarParsing.varType.int,
"should I override beamspot in globaltag?")
options.register("tauDecayMode", 0, # Default is all decays
VarParsing.VarParsing.multiplicity.singleton, VarParsing.VarParsing.varType.int,
"Tau decay mode (0=all, 230=hadronic)")
options.register("tauMinVisPt", -1, # Disabled
VarParsing.VarParsing.multiplicity.singleton, VarParsing.VarParsing.varType.int,
"Minimum visible pt of tau decay (-1 disabled, >= 0 cut value in GeV)")
options, dataVersion = getOptionsDataVersion("53XmcS10", options)
hltProcessName = dataVersion.getTriggerProcess()
recoProcessName = dataVersion.getRecoProcess()
processName = process.name_()
# Setup trigger matching
if not (dataVersion.isMC() and options.triggerMC == 0 and options.triggerMCInAnalysis == 0):
HChTriggerMatching.setMuonTriggerMatchingInAnalysis(process.tightenedMuonsMatched, options.trigger)
# Setup MuScleFit
if dataVersion.isMC():
process.muscleCorrectedMuons.identifier = "Summer12_DR53X_smearReReco"
process.muscleCorrectedMuons.applySmearing = True
else:
process.muscleCorrectedMuons.identifier = "Data2012_53X_ReReco"
# Setup output
outputModule = None
outdict = process.outputModules_()
if len(outdict) == 1:
outputModule = outdict.values()[0]
elif outdict.has_key("RECOSIMoutput"):
outputModule = outdict["RECOSIMoutput"]
print "Adjusting event content to GEN-SIM-RECO+misc"
process.load("HiggsAnalysis.HeavyChHiggsToTauNu.tauEmbedding.HChEventContent_cff")
#outputModule.outputCommands = cms.untracked.vstring("keep *")
outputModule.outputCommands = cms.untracked.vstring("drop *")
outputModule.outputCommands.extend(process.RECOSIMEventContent.outputCommands)
outputModule.outputCommands.extend([
"drop *_*_*_%s" % recoProcessName,
"keep *_generalTracks_*_%s" % recoProcessName,
"keep *_muons_*_%s" % recoProcessName,
"keep *_globalMuons_*_%s" % recoProcessName,
"keep recoGsfElectronCores_*_*_%s" % recoProcessName,
"keep *_gsfElectrons_*_%s" % recoProcessName,
"keep *_photons_*_%s" % recoProcessName,
"keep *_photonCore_*_%s" % recoProcessName,
"drop *_*_*_%s" % processName,
"keep *_particleFlow*_*_%s" % processName,
"keep *_generalTracks_*_%s" % processName,
"keep *_muons_*_%s" % processName,
"keep *_globalMuons_*_%s" % processName,
"keep *_*Electron*_*_%s" % processName,
"keep *_eid*_*_*",
])
outputModule.outputCommands.extend(eventContent(hltProcessName, recoProcessName, processName))
# re_procName = re.compile("_\*$")
# outputModule.outputCommands.extend([re_procName.sub("_"+processName, x) for x in process.RECOSIMEventContent.outputCommands])
outputModule.outputCommands.extend(process.HChEventContent.outputCommands)
#outputModule.outputCommands.extend(process.RecoParticleFlowRECO.outputCommands)
#outputModule.outputCommands.extend(["keep *_%s_*_%s" % (x, processName) for x in [
#]])
# Remove duplicate "drop *"
index = 0
for item in outputModule.outputCommands[:]:
if item == "drop *" and index > 0:
del outputModule.outputCommands[index]
index -= 1
index += 1
# Disable gen vertex smearing
process.VtxSmeared = cms.EDProducer("FlatEvtVtxGenerator",
MaxZ = cms.double(0.0),
MaxX = cms.double(0.0),
MaxY = cms.double(0.0),
MinX = cms.double(0.0),
MinY = cms.double(0.0),
MinZ = cms.double(0.0),
TimeOffset = cms.double(0.0),
src = cms.InputTag("generator")
)
# Set up tau decay options
process.generator.ZTauTau.TauolaOptions.InputCards.mdtau = options.tauDecayMode
if options.tauMinVisPt >= 0:
process.generator.ZTauTau.minVisibleTransverseMomentum = "%d"%options.tauMinVisPt
print "TAUOLA mdtau =", process.generator.ZTauTau.TauolaOptions.InputCards.mdtau
# Do we have to override the beam spot for data?
if options.overrideBeamSpot != 0:
bs = cms.string("BeamSpotObjects_2009_LumiBased_SigmaZ_v25_offline") # 44x data gt
#bs = cms.string("BeamSpotObjects_2009_LumiBased_SigmaZ_v21_offline") # 42x data gt
process.GlobalTag.toGet = cms.VPSet(
cms.PSet(record = cms.string("BeamSpotObjectsRcd"),
tag = bs,
connect = cms.untracked.string("frontier://FrontierProd/CMS_COND_31X_BEAMSPOT")
)
)
print "BeamSpot in globaltag set to ", bs
else:
print "BeamSpot in globaltag not changed"
# Merge tracks
process.tmfTracks = cms.EDProducer("RecoTracksMixer",
trackCol1 = cms.InputTag("dimuonsGlobal", "tracks"),
trackCol2 = cms.InputTag("generalTracks", "", processName)
)
process.offlinePrimaryVerticesWithBS.TrackLabel = cms.InputTag("tmfTracks")
process.offlinePrimaryVertices.TrackLabel = cms.InputTag("tmfTracks")
#print process.muons
if hasattr(process.muons, "TrackExtractorPSet"):
# <= 42X
process.muons.TrackExtractorPSet.inputTrackCollection = cms.InputTag("tmfTracks")
elif hasattr(process, "muons1stStep") and hasattr(process.muons1stStep, "TrackExtractorPSet"):
# >= 44X
process.muons1stStep.TrackExtractorPSet.inputTrackCollection = cms.InputTag("tmfTracks")
else:
raise Exception("Problem in overriding track collection for reco::Muon producer")
# Ensure that tmfTracks is always run after generalTracks (to mix the original and embedded tracks)
for p in process.paths:
pth = getattr(process, p)
if "generalTracks" in pth.moduleNames():
pth.replace(process.generalTracks, process.generalTracks*process.tmfTracks)
# it should be the best solution to take the original beam spot for the
# reconstruction of the new primary vertex
# use the one produced earlier, do not produce your own
for s in process.sequences:
seq = getattr(process,s)
seq.remove(process.offlineBeamSpot)
# Remove beam halo Id
try:
process.metreco.remove(process.BeamHaloId)
except:
pass
# Disable lumi producer
process.localreco_HcalNZS.remove(process.lumiProducer)
process.localreco.remove(process.lumiProducer)
# PFCandidate embedding
process.particleFlowORG = process.particleFlow.clone()
# Since CMSSW 4_4 the particleFlow reco works a bit differently. The step is
# twofold, first particleFlowTmp is created and then the final particleFlow
# collection. What we do in this case is that we merge the final ParticleFlow
# collection. For the muon reconstruction, we also merge particleFlowTmp in
# order to get PF-based isolation right.
if hasattr(process, 'particleFlowTmp'):
process.particleFlowTmpMixed = cms.EDProducer('PFCandidateMixer',
col1 = cms.untracked.InputTag("dimuonsGlobal", "pfCands"),
col2 = cms.untracked.InputTag("particleFlowTmp", ""),
trackCol = cms.untracked.InputTag("tmfTracks"),
# Don't produce value maps:
muons = cms.untracked.InputTag(""),
gsfElectrons = cms.untracked.InputTag("")
)
process.muons.PFCandidates = cms.InputTag("particleFlowTmpMixed")
for p in process.paths:
pth = getattr(process, p)
if "particleFlow" in pth.moduleNames():
pth.replace(process.particleFlow, process.particleFlowORG*process.particleFlow)
if "muons" in pth.moduleNames():
pth.replace(process.muons, process.particleFlowTmpMixed*process.muons)
else:
# CMSSW_4_2
if hasattr(process,"famosParticleFlowSequence"):
process.famosParticleFlowSequence.remove(process.pfPhotonTranslatorSequence)
process.famosParticleFlowSequence.remove(process.pfElectronTranslatorSequence)
process.famosParticleFlowSequence.remove(process.particleFlow)
process.famosParticleFlowSequence.__iadd__(process.particleFlowORG)
process.famosParticleFlowSequence.__iadd__(process.particleFlow)
process.famosParticleFlowSequence.__iadd__(process.pfElectronTranslatorSequence)
process.famosParticleFlowSequence.__iadd__(process.pfPhotonTranslatorSequence)
elif hasattr(process,"particleFlowReco"):
process.particleFlowReco.remove(process.pfPhotonTranslatorSequence)
process.particleFlowReco.remove(process.pfElectronTranslatorSequence)
process.particleFlowReco.remove(process.particleFlow)
process.particleFlowReco.__iadd__(process.particleFlowORG)
process.particleFlowReco.__iadd__(process.particleFlow)
process.particleFlowReco.__iadd__(process.pfElectronTranslatorSequence)
process.particleFlowReco.__iadd__(process.pfPhotonTranslatorSequence)
else:
raise "Cannot find pflow sequence"
process.pfSelectedElectrons.src = cms.InputTag("particleFlowORG")
process.pfSelectedPhotons.src = cms.InputTag("particleFlowORG")
process.particleFlow = cms.EDProducer('PFCandidateMixer',
col1 = cms.untracked.InputTag("dimuonsGlobal", "pfCands"),
col2 = cms.untracked.InputTag("particleFlowORG", ""),
trackCol = cms.untracked.InputTag("tmfTracks"),
muons = cms.untracked.InputTag("muons"),
gsfElectrons = cms.untracked.InputTag("gsfElectrons","",recoProcessName) # FIXME does this work?
#gsfElectrons = cms.untracked.InputTag("")
)
# Set the empty event filter source
process.filterEmptyEv.src.setProcessName(processName)
# Find all modules having particleFlow as their input
pfInputNeeded = {}
for p in process.paths:
i = getattr(process,p)
target = process.particleFlow
lookForPFInput = ["particleFlow"]
seqVis = SeqVisitor(target)
seqVis.prepareSearch()
seqVis.setLookFor(target)
i.visit(seqVis)
while seqVis.catch != 1 and seqVis.found == 1:
target = seqVis.giveNext()
pfInput = []
targetAttributes = dir(target)
for targetAttribute in targetAttributes:
attr=getattr(target,targetAttribute) # get actual attribute, not just the name
if isinstance(attr, cms.InputTag):
if attr.getModuleLabel()=="particleFlow" and attr.getProductInstanceLabel()!="":
print "Changing: ", target, " ", targetAttribute, " ", attr, " to particleFlowORG"
attr.setModuleLabel("particleFlowORG")
if attr.getModuleLabel() in lookForPFInput:
pfInput.append(attr)
if len(pfInput) > 0:
lookForPFInput.append(target.label())
pfInputNeeded[target.label()] = pfInput
#i.replace(target, source)
seqVis.prepareSearch()
seqVis.setLookFor(target)
i.visit(seqVis)
#if (seqVis.catch==1):
#seqVis.catch=0
#i.__iadd__(source)
pfOutputCommands = []
for label in pfInputNeeded.keys():
print "particleFlow as input in module %s, InputTags: %s" % (label, ", ".join(str(x) for x in pfInputNeeded[label]))
pfOutputCommands.append("keep *_%s_*_%s" % (label, processName))
outputModule.outputCommands.extend(pfOutputCommands)
#process.pfSelectedElectrons.src = "particleFlowORG" # 4_2 legacy, already included above
#process.pfSelectedPhotons.src = "particleFlowORG" # 4_2 legacy, already included above
# Setup/remove some HLT/DQM stuff whcih doesn't work
if hasattr(process, "hltTrigReport"):
process.hltTrigReport.HLTriggerResults.setProcessName(processName)
if hasattr(process, "DQM_FEDIntegrity_v2"):
process.schedule.remove(process.DQM_FEDIntegrity_v2)
if hasattr(process, "DQM_FEDIntegrity_v3"):
process.schedule.remove(process.DQM_FEDIntegrity_v3)
if hasattr(process, "DQM_FEDIntegrity_v5"):
process.schedule.remove(process.DQM_FEDIntegrity_v5)
if hasattr(process, "HLTAnalyzerEndpath"):
process.schedule.remove(process.HLTAnalyzerEndpath)
del process.HLTAnalyzerEndpath
#process.load("HiggsAnalysis.HeavyChHiggsToTauNu.tauEmbedding.printGenParticles_cff")
#process.generation_step *= process.printGenParticles
print "#############################################################"
print " Warning! PFCandidates 'electron' collection is not mixed, "
print " and probably shouldnt be used. "
print "#############################################################"
addPAT(process, options, dataVersion)
f = open("configDumpEmbed.py", "w")
f.write(process.dumpPython())
f.close()
return process
