def customise(process):
process._Process__name="EmbeddedINPUT"
try:
outputModule = process.output
except:
pass
try:
outputModule = getattr(process, str(getattr(process, list(process.endpaths)[-1])))
except:
pass
outputModule.outputCommands.extend(
cms.untracked.vstring("keep *_goodMuons_*_*",
"keep *_goodMuonsPFIso_*_*",
"keep *_highestPtMuPlus_*_*",
"keep *_highestPtMuMinus_*_*",
"keep *_highestPtMuPlusPFIso_*_*",
"keep *_highestPtMuMinusPFIso_*_*",
"keep *_goldenZmumuCandidatesGe0IsoMuons_*_*",
"keep *_goldenZmumuCandidatesGe1IsoMuons_*_*",
"keep *_goldenZmumuCandidatesGe2IsoMuons_*_*",
"keep TH2DMEtoEDM_MEtoEDMConverter_*_*"))
process.load('Configuration.StandardSequences.GeometryDB_cff')
process.load('Configuration.StandardSequences.MagneticField_38T_cff')
process.load("TrackingTools/TransientTrack/TransientTrackBuilder_cfi")
process.load("TauAnalysis/MCEmbeddingTools/ZmumuStandaloneSelection_cff")
# DQM store output for muon acceptance histograms
process.load('DQMServices.Core.DQMStore_cfg')
# Define configuration parameter default values
from TauAnalysis.MCEmbeddingTools.setDefaults import setDefaults
setDefaults(process)
# Read configuration parameter values by command-line parsing
#from TauAnalysis.MCEmbeddingTools.embeddingCommandLineOptions import parseCommandLineOptions
#if process.options['parseCommandLine']:
# parseCommandLineOptions(process)
# Add mumu selection to schedule
if process.customization_options.isMC.value():
process.goldenZmumuFilterPath = cms.Path(process.goldenZmumuFilterSequence)
else:
process.goldenZmumuFilterPath = cms.Path(process.goldenZmumuFilterSequenceData)
process.goldenZmumuFilter.src = process.customization_options.ZmumuCollection
process.schedule.insert(0, process.goldenZmumuFilterPath)
# Only write out events which have at least one muon pair
outputModule.SelectEvents = cms.untracked.PSet(
SelectEvents = cms.vstring('goldenZmumuFilterPath'))
return process
dataTier = cms.untracked.string('')
)
)
# Additional output definition
# Other statements
process.GlobalTag.globaltag = 'FT_R_53_V18::All'
# Path and EndPath definitions
process.RECOSIMoutput_step = cms.EndPath(process.RECOSIMoutput)
# Schedule definition
process.schedule = cms.Schedule(process.RECOSIMoutput_step)
# customisation of the process.
# Set options
from TauAnalysis.MCEmbeddingTools.setDefaults import setDefaults
setDefaults(process)
process.customization_options.isMC = cms.bool(False)
process.customization_options.ZmumuCollection = cms.InputTag('goldenZmumuCandidatesGe0IsoMuons')
# Automatic addition of the customisation function from TauAnalysis.MCEmbeddingTools.ZmumuStandaloneSelectionAll
from TauAnalysis.MCEmbeddingTools.ZmumuStandaloneSelectionAll import customise
#call to customisation function customise imported from TauAnalysis.MCEmbeddingTools.ZmumuStandaloneSelectionAll
process = customise(process)
# End of customisation functions
def customise(process):
# Define configuration parameter default values
from TauAnalysis.MCEmbeddingTools.setDefaults import setDefaults
setDefaults(process)
inputProcess = process.customization_options.inputProcessRECO.value()
print "Input process set to '%s'" % inputProcess
process._Process__name = "EmbeddedRECO"
# DQM store output for muon acceptance histograms
process.DQMStore = cms.Service("DQMStore")
if process.customization_options.isMC.value():
print "Input is MC"
# select generator level muons
process.genMuonsFromZs = cms.EDProducer(
"GenParticlesFromZsSelectorForMCEmbedding",
src=cms.InputTag(
"genParticles", "",
process.customization_options.inputProcessSIM.value()),
pdgIdsMothers=cms.vint32(23, 22),
pdgIdsDaughters=cms.vint32(13),
maxDaughters=cms.int32(2),
minDaughters=cms.int32(2),
before_or_afterFSR=cms.string("afterFSR"),
verbosity=cms.int32(0))
process.ProductionFilterSequence.replace(
process.cleanedGeneralTracks,
process.genMuonsFromZs * process.cleanedGeneralTracks)
process.ProductionFilterSequence.replace(
process.cleanedParticleFlow,
process.genMuonsFromZs * process.cleanedParticleFlow)
# produce generator level MET (= sum of neutrino momenta)
# in case Embedding is run on Monte Carlo input
process.genMetCalo = cms.EDProducer(
"MixedGenMEtProducer",
srcGenParticles1=cms.InputTag(
"genParticles", "",
process.customization_options.inputProcessSIM.value()),
srcGenParticles2=cms.InputTag("genParticles"),
srcRemovedMuons=cms.InputTag(process.customization_options.
ZmumuCollection.getModuleLabel()),
type=cms.string("calo"),
isMC=cms.bool(True))
process.genMetTrue = process.genMetCalo.clone(type=cms.string("pf"))
else:
print "Input is Data"
# produce generator level MET (= sum of neutrino momenta)
# in case Embedding is run on real Data
process.genMetCalo = cms.EDProducer(
"MixedGenMEtProducer",
srcGenParticles1=cms.InputTag(""),
srcGenParticles2=cms.InputTag("genParticles"),
srcRemovedMuons=cms.InputTag(process.customization_options.
ZmumuCollection.getModuleLabel()),
type=cms.string("calo"),
isMC=cms.bool(False))
process.genMetTrue = process.genMetCalo.clone(type=cms.string("pf"))
for p in process.paths:
pth = getattr(process, p)
if "genParticles" in pth.moduleNames():
pth.replace(
process.genParticles,
process.genParticles * process.genMetCalo * process.genMetTrue)
# update InputTags defined in PFEmbeddingSource_cff.py
print "Setting collection of Z->mumu candidates to '%s'" % process.customization_options.ZmumuCollection.getModuleLabel(
)
if not (hasattr(process, "cleanedGeneralTracks")
and hasattr(process, "cleanedParticleFlow")):
process.load("TauAnalysis.MCEmbeddingTools.PFEmbeddingSource_cff")
if process.customization_options.replaceGenOrRecMuonMomenta.value(
) == 'gen':
print "Taking momenta of generated tau leptons from generator level muons"
process.generator.src = cms.InputTag('genMuonsFromZs')
elif process.customization_options.replaceGenOrRecMuonMomenta.value(
) == 'rec':
print "Taking momenta of generated tau leptons from reconstructed muons"
process.generator.src = process.customization_options.ZmumuCollection
else:
raise ValueError(
"Invalid Configuration parameter 'replaceGenOrRecMuonMomenta' = %s !!"
% process.customization_options.replaceGenOrRecMuonMomenta.value())
# update option for removing tracks/charged PFCandidates matched to reconstructed muon
process.cleanedGeneralTracks.selectedMuons = process.customization_options.ZmumuCollection
process.cleanedParticleFlow.selectedMuons = process.customization_options.ZmumuCollection
if process.customization_options.muonTrackCleaningMode.value() == 1:
process.cleanedGeneralTracks.removeDuplicates = cms.bool(False)
process.cleanedParticleFlow.removeDuplicates = cms.bool(False)
elif process.customization_options.muonTrackCleaningMode.value() == 2:
process.cleanedGeneralTracks.removeDuplicates = cms.bool(True)
process.cleanedParticleFlow.removeDuplicates = cms.bool(True)
else:
raise ValueError(
"Invalid Configuration parameter 'muonTrackCleaningMode' = %i !!" %
process.customization_options.muonTrackCleaningMode.value())
try:
outputModule = process.output
except:
pass
try:
outputModule = getattr(
process, str(getattr(process,
list(process.endpaths)[-1])))
except:
pass
print "Changing event-content to AODSIM + Z->mumu candidates"
outputModule.outputCommands = cms.untracked.vstring("drop *")
outputModule.outputCommands.extend(
process.AODSIMEventContent.outputCommands)
# get rid of second "drop *"
index = 0
for item in outputModule.outputCommands[:]:
if item == "drop *" and index != 0:
del outputModule.outputCommands[index]
index -= 1
index += 1
# keep collections of generator level muons
outputModule.outputCommands.extend(['keep *_genMuonsFromZs_*_*'])
# keep collections of reconstructed muons
outputModule.outputCommands.extend([
'keep *_goodMuons_*_*', 'keep *_goodMuonsPFIso_*_*',
'keep *_highestPtMuPlus_*_*', 'keep *_highestPtMuMinus_*_*',
'keep *_highestPtMuPlusPFIso_*_*', 'keep *_highestPtMuMinusPFIso_*_*'
])
# keep collections of reconstructed Z -> mumu candidates
# (with different muon isolation criteria applied)
outputModule.outputCommands.extend([
'keep *_goldenZmumuCandidatesGe0IsoMuons_*_*',
'keep *_goldenZmumuCandidatesGe1IsoMuons_*_*',
'keep *_goldenZmumuCandidatesGe2IsoMuons_*_*',
'keep TH2DMEtoEDM_MEtoEDMConverter_*_*'
])
# keep flag indicating whether event passes or fails
# o Z -> mumu event selection
# o muon -> muon + photon radiation filter
outputModule.outputCommands.extend([
'keep *_goldenZmumuFilterResult_*_*',
'keep *_muonRadiationFilterResult_*_*'
])
# CV: keep HepMCProduct for embedded event,
# in order to run Validation/EventGenerator/python/TauValidation_cfi.py
# for control plots of tau polarization and decay mode information
outputModule.outputCommands.extend(['keep *HepMCProduct_generator_*_*'])
# keep distance of muons traversed in ECAL and HCAL,
# expected and removed muon energy deposits
outputModule.outputCommands.extend([
'keep *_muonCaloEnergyDepositsByDistance_totalDistanceMuPlus_*',
'keep *_muonCaloEnergyDepositsByDistance_totalEnergyDepositMuPlus_*',
'keep *_muonCaloEnergyDepositsByDistance_totalDistanceMuMinus_*',
'keep *_muonCaloEnergyDepositsByDistance_totalEnergyDepositMuMinus_*',
'keep *_castorreco_removedEnergyMuMinus*_*',
'keep *_castorreco_removedEnergyMuPlus*_*',
'keep *_hfreco_removedEnergyMuMinus*_*',
'keep *_hfreco_removedEnergyMuPlus*_*',
'keep *_ecalPreshowerRecHit_removedEnergyMuMinus*_*',
'keep *_ecalPreshowerRecHit_removedEnergyMuPlus*_*',
'keep *_ecalRecHit_removedEnergyMuMinus*_*',
'keep *_ecalRecHit_removedEnergyMuPlus*_*',
'keep *_hbhereco_removedEnergyMuMinus*_*',
'keep *_hbhereco_removedEnergyMuPlus*_*',
'keep *_horeco_removedEnergyMuMinus*_*',
'keep *_horeco_removedEnergyMuPlus*_*',
])
# replace HLT process name
# (needed for certain reprocessed Monte Carlo samples)
hltProcessName = "HLT"
try:
hltProcessName = __HLT__
except:
pass
try:
process.dimuonsHLTFilter.TriggerResultsTag.processName = hltProcessName
process.goodZToMuMuAtLeast1HLT.TrigTag.processName = hltProcessName
process.goodZToMuMuAtLeast1HLT.triggerEvent.processName = hltProcessName
process.hltTrigReport, HLTriggerResults.processName = hltProcessName
except:
pass
# disable L1GtTrigReport module
# (not used for anything yet, just prints error for every single event)
if hasattr(process, 'HLTAnalyzerEndpath'):
process.HLTAnalyzerEndpath.remove(process.hltL1GtTrigReport)
# apply configuration parameters
print "Setting mdtau to %i" % process.customization_options.mdtau.value()
process.generator.Ztautau.TauolaOptions.InputCards.mdtau = process.customization_options.mdtau
if hasattr(process.generator, "ParticleGun"):
process.generator.ParticleGun.ExternalDecays.Tauola.InputCards.mdtau = process.customization_options.mdtau
if process.customization_options.useTauolaPolarization.value():
print "Enabling tau polarization effects in TAUOLA"
# NOTE: polarization effects are only approximate,
# because Embedded events do not have proper parton history in HepMC::GenEvent
# (cf. arXiv:hep-ph/0101311 and arXiv:1201.0117)
process.generator.Ztautau.TauolaOptions.UseTauolaPolarization = cms.bool(
True)
else:
print "Disabling tau polarization effects in TAUOLA"
# NOTE: tau polarization effects need to be added by reweighting Embedded events
# using weights computed by TauSpinner package
process.generator.Ztautau.TauolaOptions.UseTauolaPolarization = cms.bool(
False)
print "Setting minVisibleTransverseMomentum to '%s'" % process.customization_options.minVisibleTransverseMomentum.value(
)
process.generator.Ztautau.minVisibleTransverseMomentum = process.customization_options.minVisibleTransverseMomentum
print "Setting transformationMode to %i" % process.customization_options.transformationMode.value(
)
process.generator.Ztautau.transformationMode = process.customization_options.transformationMode
print "Setting rfRotationAngle to %1.0f" % process.customization_options.rfRotationAngle.value(
)
process.generator.Ztautau.rfRotationAngle = process.customization_options.rfRotationAngle
if process.customization_options.overrideBeamSpot.value():
bs = cms.string("BeamSpotObjects_2009_LumiBased_SigmaZ_v28_offline")
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 '%s'" % bs
else:
print "BeamSpot in globaltag not changed"
# CV: disable gen. vertex smearing
# (embed tau leptons exactly at Z->mumu event vertex)
print "Disabling 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", "unsmeared"))
if process.customization_options.useJson.value():
print "Enabling event selection by JSON file"
import PhysicsTools.PythonAnalysis.LumiList as LumiList
import FWCore.ParameterSet.Types as CfgTypes
myLumis = LumiList.LumiList(
filename='my.json').getCMSSWString().split(',')
process.source.lumisToProcess = CfgTypes.untracked(
CfgTypes.VLuminosityBlockRange())
process.source.lumisToProcess.extend(myLumis)
#----------------------------------------------------------------------------------------------------------------------
# CV: need to rerun particle-flow algorithm in order to create links between PFMuon -> PFBlocks -> PFClusters -> PFRecHits
# (configure particle-flow sequence before overwriting modules in order to mix collections
# of objects reconstructed and Z -> mu+ mu- event with simulated tau decay products)
if process.customization_options.embeddingMode.value(
) == "RH" and process.customization_options.cleaningMode == 'PF':
rerunParticleFlow(process, inputProcess)
process.ProductionFilterSequence += process.rerunParticleFlowSequenceForPFMuonCleaning
#----------------------------------------------------------------------------------------------------------------------
# mix "general" Track collection
process.generalTracksORG = process.generalTracks.clone()
process.generalTracks = cms.EDProducer(
"TrackMixer",
todo=cms.VPSet(
cms.PSet(collection1=cms.InputTag("generalTracksORG", "",
"EmbeddedRECO"),
collection2=cms.InputTag("cleanedGeneralTracks"))),
verbosity=cms.int32(0))
for p in process.paths:
pth = getattr(process, p)
if "generalTracks" in pth.moduleNames():
pth.replace(process.generalTracks,
process.generalTracksORG * process.generalTracks)
#----------------------------------------------------------------------------------------------------------------------
# CV/TF: mixing of std::vector<Trajectory> from Zmumu event and embedded tau decay products does not work yet.
# For the time-being, we need to use the Trajectory objects from the embedded event
process.trackerDrivenElectronSeedsORG = process.trackerDrivenElectronSeeds.clone(
)
process.trackerDrivenElectronSeedsORG.TkColList = cms.VInputTag(
cms.InputTag("generalTracksORG"))
process.trackerDrivenElectronSeeds = cms.EDProducer(
"ElectronSeedTrackRefUpdater",
PreIdLabel=process.trackerDrivenElectronSeedsORG.PreIdLabel,
PreGsfLabel=process.trackerDrivenElectronSeedsORG.PreGsfLabel,
targetTracks=cms.InputTag("generalTracks"),
inSeeds=cms.InputTag(
"trackerDrivenElectronSeedsORG",
process.trackerDrivenElectronSeedsORG.PreGsfLabel.value()),
inPreId=cms.InputTag(
"trackerDrivenElectronSeedsORG",
process.trackerDrivenElectronSeedsORG.PreIdLabel.value()),
)
for p in process.paths:
pth = getattr(process, p)
if "trackerDrivenElectronSeeds" in pth.moduleNames():
pth.replace(
process.trackerDrivenElectronSeeds,
process.trackerDrivenElectronSeedsORG *
process.trackerDrivenElectronSeeds)
# CV: need to keep 'generalTracksORG' collection in event-content,
# as (at least electron based) PFCandidates will refer to it,
# causing exception in 'pfNoPileUp' module otherwise
outputModule.outputCommands.extend(
['keep recoTracks_generalTracksORG_*_*'])
#----------------------------------------------------------------------------------------------------------------------
#----------------------------------------------------------------------------------------------------------------------
# mix collections of GSF electron tracks
process.electronGsfTracksORG = process.electronGsfTracks.clone()
process.electronGsfTracks = cms.EDProducer(
"GsfTrackMixer",
collection1=cms.InputTag("electronGsfTracksORG", "", "EmbeddedRECO"),
collection2=cms.InputTag("electronGsfTracks", "", inputProcess))
process.gsfConversionTrackProducer.TrackProducer = cms.string(
'electronGsfTracksORG')
for p in process.paths:
pth = getattr(process, p)
if "electronGsfTracks" in pth.moduleNames():
pth.replace(
process.electronGsfTracks,
process.electronGsfTracksORG * process.electronGsfTracks)
process.generalConversionTrackProducer.TrackProducer = cms.string(
'generalTracksORG')
process.uncleanedOnlyGeneralConversionTrackProducer.TrackProducer = cms.string(
'generalTracksORG')
process.gsfElectronsORG = process.gsfElectrons.clone()
process.gsfElectrons = cms.EDProducer("GSFElectronsMixer",
col1=cms.InputTag("gsfElectronsORG"),
col2=cms.InputTag(
"gsfElectrons", "",
inputProcess))
for p in process.paths:
pth = getattr(process, p)
if "gsfElectrons" in pth.moduleNames():
pth.replace(process.gsfElectrons,
process.gsfElectronsORG * process.gsfElectrons)
#----------------------------------------------------------------------------------------------------------------------
# dE/dx information for mixed track collections not yet implemented in 'TracksMixer' module,
# disable usage of dE/dx information in all event reconstruction modules for now
for p in process.paths:
pth = getattr(process, p)
for mod in pth.moduleNames():
if mod.find("dedx") != -1 and mod.find("Zmumu") == -1:
if mod.find("ForPFMuonCleaning") == -1:
print "Removing %s" % mod
module = getattr(process, mod)
pth.remove(module)
# CV: Compute expected energy deposits of muon in EB/EE, HB/HE and HO:
# (1) compute distance traversed by muons produced in Z -> mu+ mu- decay
# through individual calorimeter cells
# (2) scale distances by expected energy loss dE/dx of muon
from TrackingTools.TrackAssociator.default_cfi import TrackAssociatorParameterBlock
process.muonCaloDistances = cms.EDProducer(
'MuonCaloDistanceProducer',
trackAssociator=TrackAssociatorParameterBlock.
TrackAssociatorParameters,
selectedMuons=process.customization_options.ZmumuCollection)
process.ProductionFilterSequence += process.muonCaloDistances
# mix collections of L1Extra objects
l1ExtraCollections = [["L1EmParticle", "Isolated"],
["L1EmParticle", "NonIsolated"],
["L1EtMissParticle", "MET"],
["L1EtMissParticle", "MHT"],
["L1JetParticle", "Central"],
["L1JetParticle", "Forward"],
["L1JetParticle", "Tau"], ["L1MuonParticle", ""]]
l1extraParticleCollections = []
for l1ExtraCollection in l1ExtraCollections:
inputType = l1ExtraCollection[0]
pluginType = None
srcVeto = cms.InputTag('')
dRveto = 0.
if inputType == "L1EmParticle":
pluginType = "L1ExtraEmParticleMixerPlugin"
srcSelectedMuons = process.customization_options.ZmumuCollection
dRveto = 0.3
elif inputType == "L1EtMissParticle":
pluginType = "L1ExtraMEtMixerPlugin"
elif inputType == "L1JetParticle":
pluginType = "L1ExtraJetParticleMixerPlugin"
srcSelectedMuons = process.customization_options.ZmumuCollection
dRveto = 0.3
elif inputType == "L1MuonParticle":
pluginType = "L1ExtraMuonParticleMixerPlugin"
srcSelectedMuons = process.customization_options.ZmumuCollection
dRveto = 0.8
else:
raise ValueError("Invalid L1Extra type = %s !!" % inputType)
instanceLabel = l1ExtraCollection[1]
l1extraParticleCollections.append(
cms.PSet(pluginType=cms.string(pluginType),
instanceLabel=cms.string(instanceLabel),
srcSelectedMuons2=srcSelectedMuons,
dRveto2=cms.double(dRveto)))
if inputType == 'L1EtMissParticle':
l1extraParticleCollections[-1].srcMuons = cms.InputTag(
"muonCaloDistances", "muons")
l1extraParticleCollections[-1].distanceMapMuPlus = cms.InputTag(
"muonCaloDistances", "distancesMuPlus")
l1extraParticleCollections[-1].distanceMapMuMinus = cms.InputTag(
"muonCaloDistances", "distancesMuPlus")
l1extraParticleCollections[-1].H_Calo_AbsEtaLt12 = cms.double(
process.customization_options.muonCaloCleaningSF.value() *
0.75)
l1extraParticleCollections[-1].H_Calo_AbsEta12to17 = cms.double(
process.customization_options.muonCaloCleaningSF.value() * 0.6)
l1extraParticleCollections[-1].H_Calo_AbsEtaGt17 = cms.double(
process.customization_options.muonCaloCleaningSF.value() * 0.3)
process.l1extraParticlesORG = process.l1extraParticles.clone()
process.l1extraParticles = cms.EDProducer(
'L1ExtraMixer',
src1=cms.InputTag("l1extraParticlesORG"),
src2=cms.InputTag("l1extraParticles", "", inputProcess),
collections=cms.VPSet(l1extraParticleCollections))
for p in process.paths:
pth = getattr(process, p)
if "l1extraParticles" in pth.moduleNames():
pth.replace(process.l1extraParticles,
process.l1extraParticlesORG * process.l1extraParticles)
if process.customization_options.embeddingMode.value() == "PF":
print "Using PF-embedding"
from TauAnalysis.MCEmbeddingTools.embeddingCustomizePF import customise as customisePF
customisePF(process)
elif process.customization_options.embeddingMode.value() == "RH":
print "Using RH-embedding"
from TauAnalysis.MCEmbeddingTools.embeddingCustomizeRH import customise as customiseRH
customiseRH(process, inputProcess)
else:
raise ValueError(
"Invalid Configuration parameter 'embeddingMode' = %s !!" %
process.customization_options.embeddingMode.value())
# 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
try:
outputModule = process.output
except:
pass
try:
outputModule = getattr(
process, str(getattr(process,
list(process.endpaths)[-1])))
except:
pass
process.filterEmptyEv.src = cms.untracked.InputTag("generatorSmeared", "",
"EmbeddedRECO")
try:
process.schedule.remove(process.DQM_FEDIntegrity_v3)
except:
pass
process.load("TauAnalysis/MCEmbeddingTools/ZmumuStandaloneSelection_cff")
process.goldenZmumuFilter.src = process.customization_options.ZmumuCollection
if process.customization_options.applyZmumuSkim.value():
print "Enabling Zmumu skim"
process.load("TrackingTools/TransientTrack/TransientTrackBuilder_cfi")
for path in process.paths:
if process.customization_options.isMC.value():
getattr(
process,
path)._seq = process.goldenZmumuFilterSequence * getattr(
process, path)._seq
else:
getattr(
process, path
)._seq = process.goldenZmumuFilterSequenceData * getattr(
process, path)._seq
process.options = cms.untracked.PSet(
wantSummary=cms.untracked.bool(True))
# CV: keep track of Z->mumu selection efficiency
process.goldenZmumuFilterResult = cms.EDProducer(
"CandViewCountEventSelFlagProducer",
src=process.customization_options.ZmumuCollection,
minNumber=cms.uint32(1))
process.goldenZmumuFilterEfficiencyPath = cms.Path(
process.goldenZmumuSelectionSequence +
process.goldenZmumuFilterResult + process.MEtoEDMConverter)
process.schedule.append(process.goldenZmumuFilterEfficiencyPath)
else:
print "Zmumu skim disabled"
process.load("TauAnalysis/MCEmbeddingTools/muonRadiationFilter_cfi")
process.muonRadiationFilter.srcSelectedMuons = process.customization_options.ZmumuCollection
if process.customization_options.applyMuonRadiationFilter.value():
print "Muon -> muon + photon radiation filter enabled"
# CV: add filter at the end of reconstruction path
# (filter needs mixed 'pfPileUp' and 'pfNoPileUp' collections)
process.reconstruction_step += process.muonRadiationFilterSequence
process.options = cms.untracked.PSet(
wantSummary=cms.untracked.bool(True))
# CV: keep track of which events pass/fail muon -> muon + photon radiation filter
process.muonRadiationFilterResult = cms.EDProducer(
"DummyBoolEventSelFlagProducer")
process.muonRadiationFilterEfficiencyPath = cms.Path(
process.goldenZmumuSelectionSequence +
process.muonRadiationFilterSequence +
process.muonRadiationFilterResult)
process.schedule.append(process.muonRadiationFilterEfficiencyPath)
else:
print "Muon -> muon + photon radiation filter disabled"
# CV: disable ECAL/HCAL noise simulation
if process.customization_options.disableCaloNoise.value():
print "Disabling ECAL/HCAL noise simulation"
process.simEcalUnsuppressedDigis.doNoise = cms.bool(False)
process.simEcalUnsuppressedDigis.doESNoise = cms.bool(False)
process.simHcalUnsuppressedDigis.doNoise = cms.bool(False)
process.simHcalUnsuppressedDigis.doThermalNoise = cms.bool(False)
else:
print "Keeping ECAL/HCAL noise simulation enabled"
# CV: apply/do not apply muon momentum corrections determined by Rochester group
if process.customization_options.replaceGenOrRecMuonMomenta.value(
) == "rec" and hasattr(process, "goldenZmumuSelectionSequence"):
if process.customization_options.applyRochesterMuonCorr.value():
print "Enabling Rochester muon momentum corrections"
process.patMuonsForZmumuSelectionRochesterMomentumCorr = cms.EDProducer(
"RochesterCorrPATMuonProducer",
src=cms.InputTag('patMuonsForZmumuSelection'),
isMC=cms.bool(process.customization_options.isMC.value()))
process.goldenZmumuSelectionSequence.replace(
process.patMuonsForZmumuSelection,
process.patMuonsForZmumuSelection *
process.patMuonsForZmumuSelectionRochesterMomentumCorr)
process.goodMuons.src = cms.InputTag(
'patMuonsForZmumuSelectionRochesterMomentumCorr')
else:
print "Rochester muon momentum corrections disabled"
if process.customization_options.applyMuonRadiationCorrection.value(
) != "":
print "Muon -> muon + photon radiation correction enabled"
process.load(
"TauAnalysis/MCEmbeddingTools/muonRadiationCorrWeightProducer_cfi")
if process.customization_options.applyMuonRadiationCorrection.value(
) == "photos":
process.muonRadiationCorrWeightProducer.lutDirectoryRef = cms.string(
'genMuonRadCorrAnalyzerPHOTOS')
process.muonRadiationCorrWeightProducer.lutDirectoryOthers = cms.PSet(
Summer12mcMadgraph=cms.string('genMuonRadCorrAnalyzer'))
elif process.customization_options.applyMuonRadiationCorrection.value(
) == "pythia":
process.muonRadiationCorrWeightProducer.lutDirectoryRef = cms.string(
'genMuonRadCorrAnalyzerPYTHIA')
process.muonRadiationCorrWeightProducer.lutDirectoryOthers = cms.PSet(
Summer12mcMadgraph=cms.string('genMuonRadCorrAnalyzer'))
else:
raise ValueError(
"Invalid Configuration parameter 'applyMuonRadiationCorrection' = %s !!"
% process.customization_options.applyMuonRadiationCorrection.
value())
process.reconstruction_step += process.muonRadiationCorrWeightProducer
outputModule.outputCommands.extend([
'keep *_muonRadiationCorrWeightProducer_*_*',
'keep *_generator_muonsBeforeRad_*',
'keep *_generator_muonsAfterRad_*'
])
else:
print "Muon -> muon + photon radiation correction disabled"
# CV: compute reweighting factors to compensate for smearing of di-muon Pt and mass distributions caused by:
# o (mis)reconstruction of muon momenta
# o muon -> muon + photon radiation corrections
process.load("TauAnalysis/MCEmbeddingTools/embeddingKineReweight_cff")
if process.customization_options.applyMuonRadiationCorrection.value(
) != "":
if process.customization_options.mdtau.value() == 116:
process.embeddingKineReweightGENtoEmbedded.inputFileName = cms.FileInPath(
"TauAnalysis/MCEmbeddingTools/data/makeEmbeddingKineReweightLUTs_GENtoEmbedded_mutau.root"
)
process.reconstruction_step += process.embeddingKineReweightSequenceGENtoEmbedded
elif process.customization_options.mdtau.value() == 115:
process.embeddingKineReweightGENtoEmbedded.inputFileName = cms.FileInPath(
"TauAnalysis/MCEmbeddingTools/data/makeEmbeddingKineReweightLUTs_GENtoEmbedded_etau.root"
)
process.reconstruction_step += process.embeddingKineReweightSequenceGENtoEmbedded
if process.customization_options.replaceGenOrRecMuonMomenta.value(
) == 'rec':
process.reconstruction_step += process.embeddingKineReweightSequenceGENtoREC
outputModule.outputCommands.extend(['keep *_embeddingKineReweight_*_*'])
# CV: compute weights for correcting Embedded samples
# for efficiency with which Zmumu events used as input for Embedding production were selected
process.load(
"TauAnalysis/MCEmbeddingTools/ZmumuEvtSelEffCorrWeightProducer_cfi")
process.ZmumuEvtSelEffCorrWeightProducer.selectedMuons = process.customization_options.ZmumuCollection
process.ZmumuEvtSelEffCorrWeightProducer.verbosity = cms.int32(0)
process.reconstruction_step += process.ZmumuEvtSelEffCorrWeightProducer
outputModule.outputCommands.extend(
['keep *_ZmumuEvtSelEffCorrWeightProducer_*_*'])
return (process)
def customise(process):
# Define configuration parameter default values
from TauAnalysis.MCEmbeddingTools.setDefaults import setDefaults
setDefaults(process)
inputProcess = process.customization_options.inputProcessRECO.value()
print "Input process set to '%s'" % inputProcess
process._Process__name = "EmbeddedRECO"
# DQM store output for muon acceptance histograms
process.DQMStore = cms.Service("DQMStore")
if process.customization_options.isMC.value():
print "Input is MC"
# select generator level muons
process.genMuonsFromZs = cms.EDProducer("GenParticlesFromZsSelectorForMCEmbedding",
src = cms.InputTag("genParticles", "", process.customization_options.inputProcessSIM.value()),
pdgIdsMothers = cms.vint32(23, 22),
pdgIdsDaughters = cms.vint32(13),
maxDaughters = cms.int32(2),
minDaughters = cms.int32(2),
before_or_afterFSR = cms.string("afterFSR"),
verbosity = cms.int32(0)
)
process.ProductionFilterSequence.replace(process.cleanedGeneralTracks, process.genMuonsFromZs*process.cleanedGeneralTracks)
process.ProductionFilterSequence.replace(process.cleanedParticleFlow, process.genMuonsFromZs*process.cleanedParticleFlow)
# produce generator level MET (= sum of neutrino momenta)
# in case Embedding is run on Monte Carlo input
process.genMetCalo = cms.EDProducer("MixedGenMEtProducer",
srcGenParticles1 = cms.InputTag("genParticles", "", process.customization_options.inputProcessSIM.value()),
srcGenParticles2 = cms.InputTag("genParticles"),
srcRemovedMuons = cms.InputTag(process.customization_options.ZmumuCollection.getModuleLabel()),
type = cms.string("calo"),
isMC = cms.bool(True)
)
process.genMetTrue = process.genMetCalo.clone(
type = cms.string("pf")
)
else:
print "Input is Data"
# produce generator level MET (= sum of neutrino momenta)
# in case Embedding is run on real Data
process.genMetCalo = cms.EDProducer("MixedGenMEtProducer",
srcGenParticles1 = cms.InputTag(""),
srcGenParticles2 = cms.InputTag("genParticles"),
srcRemovedMuons = cms.InputTag(process.customization_options.ZmumuCollection.getModuleLabel()),
type = cms.string("calo"),
isMC = cms.bool(False)
)
process.genMetTrue = process.genMetCalo.clone(
type = cms.string("pf")
)
for p in process.paths:
pth = getattr(process, p)
if "genParticles" in pth.moduleNames():
pth.replace(process.genParticles, process.genParticles*process.genMetCalo*process.genMetTrue)
# update InputTags defined in PFEmbeddingSource_cff.py
print "Setting collection of Z->mumu candidates to '%s'" % process.customization_options.ZmumuCollection.getModuleLabel()
if not (hasattr(process, "cleanedGeneralTracks") and hasattr(process, "cleanedParticleFlow")):
process.load("TauAnalysis.MCEmbeddingTools.PFEmbeddingSource_cff")
if process.customization_options.replaceGenOrRecMuonMomenta.value() == 'gen':
print "Taking momenta of generated tau leptons from generator level muons"
process.generator.src = cms.InputTag('genMuonsFromZs')
elif process.customization_options.replaceGenOrRecMuonMomenta.value() == 'rec':
print "Taking momenta of generated tau leptons from reconstructed muons"
process.generator.src = process.customization_options.ZmumuCollection
else:
raise ValueError("Invalid Configuration parameter 'replaceGenOrRecMuonMomenta' = %s !!" % process.customization_options.replaceGenOrRecMuonMomenta.value())
# update option for removing tracks/charged PFCandidates matched to reconstructed muon
process.cleanedGeneralTracks.selectedMuons = process.customization_options.ZmumuCollection
process.cleanedParticleFlow.selectedMuons = process.customization_options.ZmumuCollection
if process.customization_options.muonTrackCleaningMode.value() == 1:
process.cleanedGeneralTracks.removeDuplicates = cms.bool(False)
process.cleanedParticleFlow.removeDuplicates = cms.bool(False)
elif process.customization_options.muonTrackCleaningMode.value() == 2:
process.cleanedGeneralTracks.removeDuplicates = cms.bool(True)
process.cleanedParticleFlow.removeDuplicates = cms.bool(True)
else:
raise ValueError("Invalid Configuration parameter 'muonTrackCleaningMode' = %i !!" % process.customization_options.muonTrackCleaningMode.value())
try:
outputModule = process.output
except:
pass
try:
outputModule = getattr(process, str(getattr(process, list(process.endpaths)[-1])))
except:
pass
print "Changing event-content to AODSIM + Z->mumu candidates"
outputModule.outputCommands = cms.untracked.vstring("drop *")
outputModule.outputCommands.extend(process.AODSIMEventContent.outputCommands)
# get rid of second "drop *"
index = 0
for item in outputModule.outputCommands[:]:
if item == "drop *" and index != 0:
del outputModule.outputCommands[index]
index -= 1
index += 1
# keep collections of generator level muons
outputModule.outputCommands.extend([
'keep *_genMuonsFromZs_*_*'
])
# keep collections of reconstructed muons
outputModule.outputCommands.extend([
'keep *_goodMuons_*_*',
'keep *_goodMuonsPFIso_*_*',
'keep *_highestPtMuPlus_*_*',
'keep *_highestPtMuMinus_*_*',
'keep *_highestPtMuPlusPFIso_*_*',
'keep *_highestPtMuMinusPFIso_*_*'
])
# keep collections of reconstructed Z -> mumu candidates
# (with different muon isolation criteria applied)
outputModule.outputCommands.extend([
'keep *_goldenZmumuCandidatesGe0IsoMuons_*_*',
'keep *_goldenZmumuCandidatesGe1IsoMuons_*_*',
'keep *_goldenZmumuCandidatesGe2IsoMuons_*_*',
'keep TH2DMEtoEDM_MEtoEDMConverter_*_*'
])
# keep flag indicating whether event passes or fails
# o Z -> mumu event selection
# o muon -> muon + photon radiation filter
outputModule.outputCommands.extend([
'keep *_goldenZmumuFilterResult_*_*',
'keep *_genMuonRadiationFilter_*_*',
'keep *_muonRadiationFilter_*_*'
])
# CV: keep HepMCProduct for embedded event,
# in order to run Validation/EventGenerator/python/TauValidation_cfi.py
# for control plots of tau polarization and decay mode information
outputModule.outputCommands.extend([
'keep *HepMCProduct_generator_*_*'
])
# keep distance of muons traversed in ECAL and HCAL,
# expected and removed muon energy deposits
outputModule.outputCommands.extend([
'keep *_muonCaloEnergyDepositsByDistance_totalDistanceMuPlus_*',
'keep *_muonCaloEnergyDepositsByDistance_totalEnergyDepositMuPlus_*',
'keep *_muonCaloEnergyDepositsByDistance_totalDistanceMuMinus_*',
'keep *_muonCaloEnergyDepositsByDistance_totalEnergyDepositMuMinus_*',
'keep *_castorreco_removedEnergyMuMinus*_*',
'keep *_castorreco_removedEnergyMuPlus*_*',
'keep *_hfreco_removedEnergyMuMinus*_*',
'keep *_hfreco_removedEnergyMuPlus*_*',
'keep *_ecalPreshowerRecHit_removedEnergyMuMinus*_*',
'keep *_ecalPreshowerRecHit_removedEnergyMuPlus*_*',
'keep *_ecalRecHit_removedEnergyMuMinus*_*',
'keep *_ecalRecHit_removedEnergyMuPlus*_*',
'keep *_hbhereco_removedEnergyMuMinus*_*',
'keep *_hbhereco_removedEnergyMuPlus*_*',
'keep *_horeco_removedEnergyMuMinus*_*',
'keep *_horeco_removedEnergyMuPlus*_*',
])
# replace HLT process name
# (needed for certain reprocessed Monte Carlo samples)
hltProcessName = "HLT"
try:
hltProcessName = __HLT__
except:
pass
try:
process.dimuonsHLTFilter.TriggerResultsTag.processName = hltProcessName
process.goodZToMuMuAtLeast1HLT.TrigTag.processName = hltProcessName
process.goodZToMuMuAtLeast1HLT.triggerEvent.processName = hltProcessName
process.hltTrigReport,HLTriggerResults.processName = hltProcessName
except:
pass
# disable L1GtTrigReport module
# (not used for anything yet, just prints error for every single event)
if hasattr(process, 'HLTAnalyzerEndpath'):
process.HLTAnalyzerEndpath.remove(process.hltL1GtTrigReport)
# apply configuration parameters
print "Setting mdtau to %i" % process.customization_options.mdtau.value()
process.generator.Ztautau.TauolaOptions.InputCards.mdtau = process.customization_options.mdtau
if hasattr(process.generator, "ParticleGun"):
process.generator.ParticleGun.ExternalDecays.Tauola.InputCards.mdtau = process.customization_options.mdtau
if process.customization_options.useTauolaPolarization.value():
print "Enabling tau polarization effects in TAUOLA"
# NOTE: polarization effects are only approximate,
# because Embedded events do not have proper parton history in HepMC::GenEvent
# (cf. arXiv:hep-ph/0101311 and arXiv:1201.0117)
process.generator.Ztautau.TauolaOptions.UseTauolaPolarization = cms.bool(True)
else:
print "Disabling tau polarization effects in TAUOLA"
# NOTE: tau polarization effects need to be added by reweighting Embedded events
# using weights computed by TauSpinner package
process.generator.Ztautau.TauolaOptions.UseTauolaPolarization = cms.bool(False)
print "Setting minVisibleTransverseMomentum to '%s'" % process.customization_options.minVisibleTransverseMomentum.value()
process.generator.Ztautau.minVisibleTransverseMomentum = process.customization_options.minVisibleTransverseMomentum
print "Setting transformationMode to %i" % process.customization_options.transformationMode.value()
process.generator.Ztautau.transformationMode = process.customization_options.transformationMode
print "Setting rfRotationAngle to %1.0f" % process.customization_options.rfRotationAngle.value()
process.generator.Ztautau.rfRotationAngle = process.customization_options.rfRotationAngle
if process.customization_options.overrideBeamSpot.value():
bs = cms.string("BeamSpotObjects_2009_LumiBased_SigmaZ_v28_offline")
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 '%s'" % bs
else:
print "BeamSpot in globaltag not changed"
# CV: disable gen. vertex smearing
# (embed tau leptons exactly at Z->mumu event vertex)
print "Disabling 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")
)
if process.customization_options.useJson.value():
print "Enabling event selection by JSON file"
import PhysicsTools.PythonAnalysis.LumiList as LumiList
import FWCore.ParameterSet.Types as CfgTypes
myLumis = LumiList.LumiList(filename = 'my.json').getCMSSWString().split(',')
process.source.lumisToProcess = CfgTypes.untracked(CfgTypes.VLuminosityBlockRange())
process.source.lumisToProcess.extend(myLumis)
#----------------------------------------------------------------------------------------------------------------------
# CV: need to rerun particle-flow algorithm in order to create links between PFMuon -> PFBlocks -> PFClusters -> PFRecHits
# (configure particle-flow sequence before overwriting modules in order to mix collections
# of objects reconstructed and Z -> mu+ mu- event with simulated tau decay products)
if process.customization_options.embeddingMode.value() == "RH" and process.customization_options.cleaningMode == 'PF':
rerunParticleFlow(process, inputProcess)
process.ProductionFilterSequence += process.rerunParticleFlowSequenceForPFMuonCleaning
#----------------------------------------------------------------------------------------------------------------------
# mix "general" Track collection
process.generalTracksORG = process.generalTracks.clone()
process.generalTracks = cms.EDProducer("TrackMixer",
todo = cms.VPSet(
cms.PSet(
collection1 = cms.InputTag("generalTracksORG", "", "EmbeddedRECO"),
collection2 = cms.InputTag("cleanedGeneralTracks")
)
),
verbosity = cms.int32(0)
)
for p in process.paths:
pth = getattr(process,p)
if "generalTracks" in pth.moduleNames():
pth.replace(process.generalTracks, process.generalTracksORG*process.generalTracks)
#----------------------------------------------------------------------------------------------------------------------
# CV/TF: mixing of std::vector<Trajectory> from Zmumu event and embedded tau decay products does not work yet.
# For the time-being, we need to use the Trajectory objects from the embedded event
process.trackerDrivenElectronSeedsORG = process.trackerDrivenElectronSeeds.clone()
process.trackerDrivenElectronSeedsORG.TkColList = cms.VInputTag(
cms.InputTag("generalTracksORG")
)
process.trackerDrivenElectronSeeds = cms.EDProducer("ElectronSeedTrackRefUpdater",
PreIdLabel = process.trackerDrivenElectronSeedsORG.PreIdLabel,
PreGsfLabel = process.trackerDrivenElectronSeedsORG.PreGsfLabel,
targetTracks = cms.InputTag("generalTracks"),
inSeeds = cms.InputTag("trackerDrivenElectronSeedsORG", process.trackerDrivenElectronSeedsORG.PreGsfLabel.value()),
inPreId = cms.InputTag("trackerDrivenElectronSeedsORG", process.trackerDrivenElectronSeedsORG.PreIdLabel.value()),
)
for p in process.paths:
pth = getattr(process,p)
if "trackerDrivenElectronSeeds" in pth.moduleNames():
pth.replace(process.trackerDrivenElectronSeeds, process.trackerDrivenElectronSeedsORG*process.trackerDrivenElectronSeeds)
# CV: need to keep 'generalTracksORG' collection in event-content,
# as (at least electron based) PFCandidates will refer to it,
# causing exception in 'pfNoPileUp' module otherwise
outputModule.outputCommands.extend(['keep recoTracks_generalTracksORG_*_*'])
#----------------------------------------------------------------------------------------------------------------------
#----------------------------------------------------------------------------------------------------------------------
# mix collections of GSF electron tracks
process.electronGsfTracksORG = process.electronGsfTracks.clone()
process.electronGsfTracks = cms.EDProducer("GsfTrackMixer",
collection1 = cms.InputTag("electronGsfTracksORG", "", "EmbeddedRECO"),
collection2 = cms.InputTag("electronGsfTracks", "", inputProcess)
)
process.gsfConversionTrackProducer.TrackProducer = cms.string('electronGsfTracksORG')
for p in process.paths:
pth = getattr(process,p)
if "electronGsfTracks" in pth.moduleNames():
pth.replace(process.electronGsfTracks, process.electronGsfTracksORG*process.electronGsfTracks)
process.generalConversionTrackProducer.TrackProducer = cms.string('generalTracksORG')
process.uncleanedOnlyGeneralConversionTrackProducer.TrackProducer = cms.string('generalTracksORG')
process.gsfElectronsORG = process.gsfElectrons.clone()
process.gsfElectrons = cms.EDProducer("GSFElectronsMixer",
col1 = cms.InputTag("gsfElectronsORG"),
col2 = cms.InputTag("gsfElectrons", "", inputProcess)
)
for p in process.paths:
pth = getattr(process,p)
if "gsfElectrons" in pth.moduleNames():
pth.replace(process.gsfElectrons, process.gsfElectronsORG*process.gsfElectrons)
#----------------------------------------------------------------------------------------------------------------------
# dE/dx information for mixed track collections not yet implemented in 'TracksMixer' module,
# disable usage of dE/dx information in all event reconstruction modules for now
for p in process.paths:
pth = getattr(process,p)
for mod in pth.moduleNames():
if mod.find("dedx") != -1 and mod.find("Zmumu") == -1:
if mod.find("ForPFMuonCleaning") == -1:
print "Removing %s" % mod
module = getattr(process, mod)
pth.remove(module)
# CV: Compute expected energy deposits of muon in EB/EE, HB/HE and HO:
# (1) compute distance traversed by muons produced in Z -> mu+ mu- decay
# through individual calorimeter cells
# (2) scale distances by expected energy loss dE/dx of muon
from TrackingTools.TrackAssociator.default_cfi import TrackAssociatorParameterBlock
process.muonCaloDistances = cms.EDProducer('MuonCaloDistanceProducer',
trackAssociator = TrackAssociatorParameterBlock.TrackAssociatorParameters,
selectedMuons = process.customization_options.ZmumuCollection)
process.ProductionFilterSequence += process.muonCaloDistances
# mix collections of L1Extra objects
l1ExtraCollections = [
[ "L1EmParticle", "Isolated" ],
[ "L1EmParticle", "NonIsolated" ],
[ "L1EtMissParticle", "MET" ],
[ "L1EtMissParticle", "MHT" ],
[ "L1JetParticle", "Central" ],
[ "L1JetParticle", "Forward" ],
[ "L1JetParticle", "Tau" ],
[ "L1MuonParticle", "" ]
]
l1extraParticleCollections = []
for l1ExtraCollection in l1ExtraCollections:
inputType = l1ExtraCollection[0]
pluginType = None
srcVeto = cms.InputTag('')
dRveto = 0.
if inputType == "L1EmParticle":
pluginType = "L1ExtraEmParticleMixerPlugin"
srcSelectedMuons = process.customization_options.ZmumuCollection
dRveto = 0.3
elif inputType == "L1EtMissParticle":
pluginType = "L1ExtraMEtMixerPlugin"
elif inputType == "L1JetParticle":
pluginType = "L1ExtraJetParticleMixerPlugin"
srcSelectedMuons = process.customization_options.ZmumuCollection
dRveto = 0.3
elif inputType == "L1MuonParticle":
pluginType = "L1ExtraMuonParticleMixerPlugin"
srcSelectedMuons = process.customization_options.ZmumuCollection
dRveto = 0.8
else:
raise ValueError("Invalid L1Extra type = %s !!" % inputType)
instanceLabel = l1ExtraCollection[1]
l1extraParticleCollections.append(cms.PSet(
pluginType = cms.string(pluginType),
instanceLabel = cms.string(instanceLabel),
srcSelectedMuons2 = srcSelectedMuons,
dRveto2 = cms.double(dRveto)))
if inputType == 'L1EtMissParticle':
l1extraParticleCollections[-1].srcMuons = cms.InputTag("muonCaloDistances", "muons")
l1extraParticleCollections[-1].distanceMapMuPlus = cms.InputTag("muonCaloDistances", "distancesMuPlus")
l1extraParticleCollections[-1].distanceMapMuMinus = cms.InputTag("muonCaloDistances", "distancesMuPlus")
l1extraParticleCollections[-1].H_Calo_AbsEtaLt12 = cms.double(process.customization_options.muonCaloCleaningSF.value()*0.75)
l1extraParticleCollections[-1].H_Calo_AbsEta12to17 = cms.double(process.customization_options.muonCaloCleaningSF.value()*0.6)
l1extraParticleCollections[-1].H_Calo_AbsEtaGt17 = cms.double(process.customization_options.muonCaloCleaningSF.value()*0.3)
process.l1extraParticlesORG = process.l1extraParticles.clone()
process.l1extraParticles = cms.EDProducer('L1ExtraMixer',
src1 = cms.InputTag("l1extraParticlesORG"),
src2 = cms.InputTag("l1extraParticles", "", inputProcess),
collections = cms.VPSet(l1extraParticleCollections)
)
for p in process.paths:
pth = getattr(process,p)
if "l1extraParticles" in pth.moduleNames():
pth.replace(process.l1extraParticles, process.l1extraParticlesORG*process.l1extraParticles)
if process.customization_options.embeddingMode.value() == "PF":
print "Using PF-embedding"
from TauAnalysis.MCEmbeddingTools.embeddingCustomizePF import customise as customisePF
customisePF(process, inputProcess)
elif process.customization_options.embeddingMode.value() == "RH":
print "Using RH-embedding"
from TauAnalysis.MCEmbeddingTools.embeddingCustomizeRH import customise as customiseRH
customiseRH(process, inputProcess)
else:
raise ValueError("Invalid Configuration parameter 'embeddingMode' = %s !!" % process.customization_options.embeddingMode.value())
# 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
try:
outputModule = process.output
except:
pass
try:
outputModule = getattr(process, str(getattr(process, list(process.endpaths)[-1])))
except:
pass
process.filterEmptyEv.src = cms.untracked.InputTag("generator", "", "EmbeddedRECO")
try:
process.schedule.remove(process.DQM_FEDIntegrity_v3)
except:
pass
process.load("TauAnalysis/MCEmbeddingTools/ZmumuStandaloneSelection_cff")
process.goldenZmumuFilter.src = process.customization_options.ZmumuCollection
if process.customization_options.applyZmumuSkim.value():
print "Enabling Zmumu skim"
process.load("TrackingTools/TransientTrack/TransientTrackBuilder_cfi")
for path in process.paths:
if process.customization_options.isMC.value():
getattr(process,path)._seq = process.goldenZmumuFilterSequence * getattr(process,path)._seq
else:
getattr(process,path)._seq = process.goldenZmumuFilterSequenceData * getattr(process,path)._seq
process.options = cms.untracked.PSet(
wantSummary = cms.untracked.bool(True)
)
# CV: keep track of Z->mumu selection efficiency
process.goldenZmumuFilterResult = cms.EDProducer("CandViewCountEventSelFlagProducer",
src = process.customization_options.ZmumuCollection,
minNumber = cms.uint32(1)
)
process.goldenZmumuFilterEfficiencyPath = cms.Path(process.goldenZmumuSelectionSequence + process.goldenZmumuFilterResult + process.MEtoEDMConverter)
process.schedule.append(process.goldenZmumuFilterEfficiencyPath)
else:
print "Zmumu skim disabled"
# CV: keep track of which events pass/fail muon -> muon + photon radiation filters
# on generator and reconstruction level, but do not actually reject any events
if process.customization_options.isMC.value():
process.load("TauAnalysis/MCEmbeddingTools/genMuonRadiationFilter_cfi")
process.genMuonRadiationFilter.srcGenParticles = cms.InputTag('genParticles::%s' % process.customization_options.inputProcessSIM.value())
process.genMuonRadiationFilter.invert = cms.bool(False)
process.genMuonRadiationFilter.filter = cms.bool(False)
process.reconstruction_step += process.genMuonRadiationFilter
# CV: keep track of which events pass/fail muon -> muon + photon radiation filter,
# but do not actually reject any events
process.load("TauAnalysis/MCEmbeddingTools/muonRadiationFilter_cfi")
process.particleFlowPtrsForMuonRadiationFilter.src = cms.InputTag('particleFlow', '', inputProcess)
process.muonRadiationFilter.srcSelectedMuons = process.customization_options.ZmumuCollection
process.muonRadiationFilter.invert = cms.bool(False)
process.muonRadiationFilter.filter = cms.bool(False)
process.reconstruction_step += process.muonRadiationFilterSequence
# CV: disable ECAL/HCAL noise simulation
if process.customization_options.disableCaloNoise.value():
print "Disabling ECAL/HCAL noise simulation"
process.simEcalUnsuppressedDigis.doNoise = cms.bool(False)
process.simEcalUnsuppressedDigis.doESNoise = cms.bool(False)
process.simHcalUnsuppressedDigis.doNoise = cms.bool(False)
process.simHcalUnsuppressedDigis.doThermalNoise = cms.bool(False)
else:
print "Keeping ECAL/HCAL noise simulation enabled"
# AB: The following producers are currently commented out. These mostly produce correction
# factors from input .root files. Such binary files are not desirable to have in a CMSSW
# release, so they are not included at the moment. A possible solution is to store and look
# up the correction factors via the Frontier DB.
"""
# CV: apply/do not apply muon momentum corrections determined by Rochester group
if process.customization_options.replaceGenOrRecMuonMomenta.value() == "rec" and hasattr(process, "goldenZmumuSelectionSequence"):
if process.customization_options.applyRochesterMuonCorr.value():
print "Enabling Rochester muon momentum corrections"
process.patMuonsForZmumuSelectionRochesterMomentumCorr = cms.EDProducer("RochesterCorrPATMuonProducer",
src = cms.InputTag('patMuonsForZmumuSelection'),
isMC = cms.bool(process.customization_options.isMC.value())
)
process.goldenZmumuSelectionSequence.replace(process.patMuonsForZmumuSelection, process.patMuonsForZmumuSelection*process.patMuonsForZmumuSelectionRochesterMomentumCorr)
process.goodMuons.src = cms.InputTag('patMuonsForZmumuSelectionRochesterMomentumCorr')
else:
print "Rochester muon momentum corrections disabled"
if process.customization_options.applyMuonRadiationCorrection.value() != "":
print "Muon -> muon + photon radiation correction enabled"
process.load("TauAnalysis/MCEmbeddingTools/muonRadiationCorrWeightProducer_cfi")
if process.customization_options.applyMuonRadiationCorrection.value() == "photos":
process.muonRadiationCorrWeightProducer.lutDirectoryRef = cms.string('genMuonRadCorrAnalyzerPHOTOS')
process.muonRadiationCorrWeightProducer.lutDirectoryOthers = cms.PSet(
Summer12mcMadgraph = cms.string('genMuonRadCorrAnalyzer')
)
elif process.customization_options.applyMuonRadiationCorrection.value() == "pythia":
process.muonRadiationCorrWeightProducer.lutDirectoryRef = cms.string('genMuonRadCorrAnalyzerPYTHIA')
process.muonRadiationCorrWeightProducer.lutDirectoryOthers = cms.PSet(
Summer12mcMadgraph = cms.string('genMuonRadCorrAnalyzer')
)
else:
raise ValueError("Invalid Configuration parameter 'applyMuonRadiationCorrection' = %s !!" % process.customization_options.applyMuonRadiationCorrection.value())
process.reconstruction_step += process.muonRadiationCorrWeightProducer
outputModule.outputCommands.extend([
'keep *_muonRadiationCorrWeightProducer_*_*',
'keep *_generator_muonsBeforeRad_*',
'keep *_generator_muonsAfterRad_*'
])
else:
print "Muon -> muon + photon radiation correction disabled"
# CV: compute reweighting factors to compensate for smearing of di-muon Pt and mass distributions caused by:
# o (mis)reconstruction of muon momenta
# o muon -> muon + photon radiation corrections
#
# mdtau values are defined in http:marpix1.in2p3.fr/Physics/biblio_top/mc_toprex_405.ps.gz
#
embeddingKineReweightTable = {
(115, 'elec1_9to30had1_15'): ('TauAnalysis/MCEmbeddingTools/data/embeddingKineReweight_ePt9to30tauPtGt18_genEmbedded.root', 'TauAnalysis/MCEmbeddingTools/data/embeddingKineReweight_ePt9to30tauPtGt18_recEmbedded.root'),
(115, 'elec1_20had1_18'): ('TauAnalysis/MCEmbeddingTools/data/embeddingKineReweight_ePtGt20tauPtGt18_genEmbedded.root', 'TauAnalysis/MCEmbeddingTools/data/embeddingKineReweight_ePtGt20tauPtGt18_recEmbedded.root'),
(116, 'mu1_7to25had1_15'): ('TauAnalysis/MCEmbeddingTools/data/embeddingKineReweight_muPt7to25tauPtGt18_genEmbedded.root', 'TauAnalysis/MCEmbeddingTools/data/embeddingKineReweight_muPt7to25tauPtGt18_recEmbedded.root'),
(116, 'mu1_16had1_18'): ('TauAnalysis/MCEmbeddingTools/data/embeddingKineReweight_muPtGt16tauPtGt18_genEmbedded.root', 'TauAnalysis/MCEmbeddingTools/data/embeddingKineReweight_muPtGt16tauPtGt18_recEmbedded.root'),
(121, 'elec1_17elec2_8'): ('TauAnalysis/MCEmbeddingTools/data/embeddingKineReweight_emu_genEmbedding.root', 'TauAnalysis/MCEmbeddingTools/data/embeddingKineReweight_emu_recEmbedding.root'),
(122, 'mu1_18mu2_8'): ('TauAnalysis/MCEmbeddingTools/data/embeddingKineReweight_emu_genEmbedding.root', 'TauAnalysis/MCEmbeddingTools/data/embeddingKineReweight_emu_recEmbedding.root'),
(123, 'tau1_18tau2_8'): ('TauAnalysis/MCEmbeddingTools/data/embeddingKineReweight_emu_genEmbedding.root', 'TauAnalysis/MCEmbeddingTools/data/embeddingKineReweight_emu_recEmbedding.root'),
(132, 'had1_30had2_30'): ('TauAnalysis/MCEmbeddingTools/data/embeddingKineReweight_tautau_genEmbedding.root', 'TauAnalysis/MCEmbeddingTools/data/embeddingKineReweight_tautau_recEmbedding.root')
}
key = (process.customization_options.mdtau.value(), process.customization_options.minVisibleTransverseMomentum.value())
if key in embeddingKineReweightTable:
process.load("TauAnalysis/MCEmbeddingTools/embeddingKineReweight_cff")
genEmbeddingReweightFile, recEmbeddingReweightFile = embeddingKineReweightTable[key]
process.embeddingKineReweightGENembedding.inputFileName = cms.FileInPath(genEmbeddingReweightFile)
process.embeddingKineReweightRECembedding.inputFileName = cms.FileInPath(recEmbeddingReweightFile)
process.reconstruction_step += process.embeddingKineReweightSequence
outputModule.outputCommands.extend([
'keep *_embeddingKineReweight*_*_*'
])
else:
print 'Do not produce embeddingKineWeights for mdtau=%d, minVisibleTransverseMomuntem=%s !!' % (key[0], key[1])
# CV: compute weights for correcting Embedded samples
# for efficiency with which Zmumu events used as input for Embedding production were selected
process.load("TauAnalysis/MCEmbeddingTools/ZmumuEvtSelEffCorrWeightProducer_cfi")
process.ZmumuEvtSelEffCorrWeightProducer.selectedMuons = process.customization_options.ZmumuCollection
process.ZmumuEvtSelEffCorrWeightProducer.verbosity = cms.int32(0)
process.reconstruction_step += process.ZmumuEvtSelEffCorrWeightProducer
outputModule.outputCommands.extend([
'keep *_ZmumuEvtSelEffCorrWeightProducer_*_*'
])
"""
# TauSpinner
# process.RandomNumberGeneratorService.TauSpinnerReco = cms.PSet(
# initialSeed = cms.untracked.uint32(123456789),
# engineName = cms.untracked.string('HepJamesRandom')
# )
# process.load('GeneratorInterface.ExternalDecays.TauSpinner_cfi')
# #process.TauSpinnerReco.isTauolaConfigured = cms.bool(True)
# process.reconstruction_step += process.TauSpinnerReco
# outputModule.outputCommands.extend([
# 'keep *_TauSpinnerReco_*_*'
# ])
return(process)
from SLHCUpgradeSimulations.Configuration.postLS1Customs import customisePostLS1 # call to customisation function customisePostLS1 imported from SLHCUpgradeSimulations.Configuration.postLS1Customs process = customisePostLS1(process) # Automatic addition of the customisation function from Configuration.DataProcessing.Utils from Configuration.DataProcessing.Utils import addMonitoring # call to customisation function addMonitoring imported from Configuration.DataProcessing.Utils process = addMonitoring(process) # Automatic addition of the customisation function from TauAnalysis.MCEmbeddingTools.setDefaults from TauAnalysis.MCEmbeddingTools.setDefaults import setDefaults # call to customisation function setDefaults imported from TauAnalysis.MCEmbeddingTools.setDefaults process = setDefaults(process) # Automatic addition of the customisation function from TauAnalysis.MCEmbeddingTools.embeddingCustomizeAll from TauAnalysis.MCEmbeddingTools.embeddingCustomizeAll import customise # call to customisation function customise imported from TauAnalysis.MCEmbeddingTools.embeddingCustomizeAll process = customise(process) # Automatic addition of the customisation function from HLTrigger.Configuration.customizeHLTforMC from HLTrigger.Configuration.customizeHLTforMC import customizeHLTforMC # call to customisation function customizeHLTforMC imported from HLTrigger.Configuration.customizeHLTforMC process = customizeHLTforMC(process) # End of customisation functions
dataset=cms.untracked.PSet(filterName=cms.untracked.string(''),
dataTier=cms.untracked.string('')))
# Additional output definition
# Other statements
process.GlobalTag.globaltag = 'FT_R_53_V18::All'
# Path and EndPath definitions
process.RECOSIMoutput_step = cms.EndPath(process.RECOSIMoutput)
# Schedule definition
process.schedule = cms.Schedule(process.RECOSIMoutput_step)
# customisation of the process.
# Set options
from TauAnalysis.MCEmbeddingTools.setDefaults import setDefaults
setDefaults(process)
process.customization_options.isMC = cms.bool(False)
process.customization_options.ZmumuCollection = cms.InputTag(
'goldenZmumuCandidatesGe0IsoMuons')
# Automatic addition of the customisation function from TauAnalysis.MCEmbeddingTools.ZmumuStandaloneSelectionAll
from TauAnalysis.MCEmbeddingTools.ZmumuStandaloneSelectionAll import customise
#call to customisation function customise imported from TauAnalysis.MCEmbeddingTools.ZmumuStandaloneSelectionAll
process = customise(process)
# End of customisation functions
