customisers Namespace Reference

Classes
class	module_manipulate
	Customizer for skimming ########################### There are four different parts. More...

Functions
def	customiseCleaning (process, changeProcessname=True, reselect=False)
def	customiseFilterTTbartoMuMu (process)
def	customiseFilterZToMuMu (process)
	MC specific Customizer ###########################. More...
def	customiseGenerator_HLT (process, changeProcessname=True, reselect=False)
def	customiseGenerator_HLT_Reselect (process)
def	customiseGenerator_postHLT (process, changeProcessname=True, reselect=False)
def	customiseGenerator_postHLT_Reselect (process)
def	customiseGenerator_preHLT (process, changeProcessname=True, reselect=False)
def	customiseGenerator_preHLT_Reselect (process)
def	customiseKeepPrunedGenParticles (process, reselect=False)
def	customiseLHE (process, changeProcessname=True, reselect=False)
def	customiseLHEandCleaning (process, reselect=False)
def	customiseLHEandCleaning_Reselect (process)
def	customiseMCFilter (process)
def	customiseMerging (process, changeProcessname=True, reselect=False)
def	customiseMerging_Reselect (process, changeProcessname=True)
def	customiseNanoAOD (process)
	Customize NanoAOD ################################. More...
def	customiseSelecting (process, reselect=False)
def	customiseSelecting_Reselect (process)
def	customisoptions (process)
	cross Customizers ########################### More...
def	fix_input_tags (process, formodules=["generalTracks", cscSegments, dt4DSegments, rpcRecHits)
def	keepCleaned (dataTier)
	Customizer for cleaning ###########################. More...
def	keepLHE ()
	Customizer for LHE ###########################. More...
def	keepMerged (dataTier="SELECT")
	Customizer for merging ###########################. More...
def	keepSelected (dataTier)
	Customizer for Selecting ###########################. More...
def	keepSimulated (process, processname="SIMembedding")
	Customizer for simulaton ###########################. More...
def	modify_outputModules (process, keep_drop_list=[], module_veto_list=[])

Variables
	instance
	manipulator_name
	module_name
	steps
	to_bemanipulate

Function Documentation

customiseCleaning()

def customisers.customiseCleaning	(		process,
			changeProcessname = True,
			reselect = False
	)

Definition at line 388 of file customisers.py.

References customisoptions(), keepCleaned(), keepSelected(), and modify_outputModules().

Referenced by customiseLHEandCleaning().

def customiseCleaning(process, changeProcessname=True, reselect=False):
    if changeProcessname:
        process._Process__name = "CLEAN"
    if reselect:
        dataTier = "RESELECT"
    else:
        dataTier = "SELECT"
    
    from TrackingTools.TrackAssociator.default_cfi import TrackAssociatorParameterBlock

    TrackAssociatorParameterBlock.TrackAssociatorParameters.CSCSegmentCollectionLabel = cms.InputTag(
        "cscSegments", "", dataTier
    )
    TrackAssociatorParameterBlock.TrackAssociatorParameters.CaloTowerCollectionLabel = (
        cms.InputTag("towerMaker", "", dataTier)
    )
    TrackAssociatorParameterBlock.TrackAssociatorParameters.DTRecSegment4DCollectionLabel = cms.InputTag(
        "dt4DSegments", "", dataTier
    )
    TrackAssociatorParameterBlock.TrackAssociatorParameters.EBRecHitCollectionLabel = (
        cms.InputTag("ecalRecHit", "EcalRecHitsEB", dataTier)
    )
    TrackAssociatorParameterBlock.TrackAssociatorParameters.EERecHitCollectionLabel = (
        cms.InputTag("ecalRecHit", "EcalRecHitsEE", dataTier)
    )
    TrackAssociatorParameterBlock.TrackAssociatorParameters.HBHERecHitCollectionLabel = cms.InputTag(
        "hbhereco", "", dataTier
    )
    TrackAssociatorParameterBlock.TrackAssociatorParameters.HORecHitCollectionLabel = (
        cms.InputTag("horeco", "", dataTier)
    )
    TrackAssociatorParameterBlock.TrackAssociatorParameters.ME0HitCollectionLabel = (
        cms.InputTag("me0RecHits", "", dataTier)
    )
    TrackAssociatorParameterBlock.TrackAssociatorParameters.ME0SegmentCollectionLabel = (
        cms.InputTag("me0Segments", "", dataTier)
    )
    TrackAssociatorParameterBlock.TrackAssociatorParameters.RPCHitCollectionLabel = (
        cms.InputTag("rpcRecHits", "", dataTier)
    )

    MuonImput = cms.InputTag("selectedMuonsForEmbedding", "", "")  
    for akt_manimod in to_bemanipulate:
        if "CLEAN" in akt_manimod.steps:
            oldCollections_in = cms.VInputTag()
            for instance in akt_manimod.instance:
                oldCollections_in.append(
                    cms.InputTag(akt_manimod.module_name, instance, dataTier)
                )
            setattr(
                process,
                akt_manimod.module_name,
                cms.EDProducer(
                    akt_manimod.cleaner_name,
                    MuonCollection=MuonImput,
                    TrackAssociatorParameters=TrackAssociatorParameterBlock.TrackAssociatorParameters,
                    oldCollection=oldCollections_in,
                    cscDigiCollectionLabel = cms.InputTag("muonCSCDigis","MuonCSCStripDigi"),
                    digiMaxDistanceX = cms.double(25.0),
                    dtDigiCollectionLabel = cms.InputTag("muonDTDigis"),
                ),
            )
    process.ecalPreshowerRecHit.TrackAssociatorParameters.usePreshower = cms.bool(True)
    process = customisoptions(process)
    return modify_outputModules(
        process, [keepSelected(dataTier), keepCleaned(dataTier)], ["MINIAODoutput"]
    )

customiseFilterTTbartoMuMu()

def customisers.customiseFilterTTbartoMuMu

(

process

)

Definition at line 1110 of file customisers.py.

References customiseMCFilter().

def customiseFilterTTbartoMuMu(process):
    process.load("TauAnalysis.MCEmbeddingTools.TTbartoMuMuGenFilter_cfi")
    process.MCFilter = cms.Path(process.TTbartoMuMuGenFilter)
    return customiseMCFilter(process)

customiseFilterZToMuMu()

def customisers.customiseFilterZToMuMu

(

process

)

MC specific Customizer ###########################.

Definition at line 1103 of file customisers.py.

def customiseFilterZToMuMu(process):
    process.load("TauAnalysis.MCEmbeddingTools.DYToMuMuGenFilter_cfi")
    process.ZToMuMuFilter = cms.Path(process.dYToMuMuGenFilter)
    process.schedule.insert(-1, process.ZToMuMuFilter)
    return process

customiseGenerator_HLT()

def customisers.customiseGenerator_HLT	(		process,
			changeProcessname = True,
			reselect = False
	)

Definition at line 618 of file customisers.py.

References customisoptions(), keepCleaned(), keepLHE(), keepSelected(), keepSimulated(), modify_outputModules(), and print().

Referenced by customiseGenerator_HLT_Reselect().

def customiseGenerator_HLT(process, changeProcessname=True, reselect=False):
    if reselect:
        dataTier = "RESELECT"
    else:
        dataTier = "SELECT"
    if changeProcessname:
        process._Process__name = "SIMembeddingHLT"

    
    process.load("TauAnalysis.MCEmbeddingTools.EmbeddingBeamSpotOnline_cfi")
    process.hltOnlineBeamSpot = process.onlineEmbeddingBeamSpotProducer.clone()
    print(
        "Setting online beam spot in HLTSchedule to the one from input data. Replaced 'hltOnlineBeamSpot' with the offline beam spot."
    )

    # Replace HLT vertexing with vertex taken from LHE step
    process.load("TauAnalysis.MCEmbeddingTools.EmbeddingHltPixelVerticesProducer_cfi")
    process.hltPixelVertices = process.embeddingHltPixelVertices.clone()
    process.offlinePrimaryVertices = process.embeddingHltPixelVertices.clone()
    process.firstStepPrimaryVerticesUnsorted = process.embeddingHltPixelVertices.clone()
    process.firstStepPrimaryVerticesPreSplitting = (
        process.embeddingHltPixelVertices.clone()
    )

    process = customisoptions(process)
    

    return modify_outputModules(
        process,
        [
            keepSelected(dataTier),
            keepCleaned(dataTier),
            keepLHE(),
            keepSimulated(process, processname="SIMembeddingpreHLT"),
            keepSimulated(process, processname="SIMembeddingHLT"),
        ],
        ["AODSIMoutput"],
    )

customiseGenerator_HLT_Reselect()

def customisers.customiseGenerator_HLT_Reselect

(

process

)

Definition at line 658 of file customisers.py.

References customiseGenerator_HLT().

def customiseGenerator_HLT_Reselect(process):
    return customiseGenerator_HLT(process, reselect=True)

customiseGenerator_postHLT()

def customisers.customiseGenerator_postHLT	(		process,
			changeProcessname = True,
			reselect = False
	)

Definition at line 662 of file customisers.py.

References customisoptions(), keepCleaned(), keepLHE(), keepSelected(), keepSimulated(), and modify_outputModules().

Referenced by customiseGenerator_postHLT_Reselect().

def customiseGenerator_postHLT(process, changeProcessname=True, reselect=False):
    if reselect:
        dataTier = "RESELECT"
    else:
        dataTier = "SELECT"
    if changeProcessname:
        process._Process__name = "SIMembedding"

    

    # process.load('TauAnalysis.MCEmbeddingTools.EmbeddingVertexCorrector_cfi')
    # process.VtxSmeared = process.VtxCorrectedToInput.clone()
    # print "Correcting Vertex in genEvent to one from input. Replaced 'VtxSmeared' with the Corrector."
    # process.load('TauAnalysis.MCEmbeddingTools.EmbeddingBeamSpotOnline_cfi')
    # process.hltOnlineBeamSpot = process.onlineEmbeddingBeamSpotProducer.clone()
    # print "Setting online beam spot in HLTSchedule to the one from input data. Replaced 'hltOnlineBeamSpot' with the offline beam spot."

    # Remove BeamSpot Production, use the one from selected data instead.
    process.reconstruction.remove(process.offlineBeamSpot)

    process = customisoptions(process)
    

    return modify_outputModules(
        process,
        [
            keepSelected(dataTier),
            keepCleaned(dataTier),
            keepLHE(),
            keepSimulated(process, processname="SIMembeddingpreHLT"),
            keepSimulated(process, processname="SIMembeddingHLT"),
            keepSimulated(process, processname="SIMembedding"),
        ],
        ["AODSIMoutput"],
    )

customiseGenerator_postHLT_Reselect()

def customisers.customiseGenerator_postHLT_Reselect

(

process

)

Definition at line 699 of file customisers.py.

References customiseGenerator_postHLT().

def customiseGenerator_postHLT_Reselect(process):
    return customiseGenerator_postHLT(process, reselect=True)

customiseGenerator_preHLT()

def customisers.customiseGenerator_preHLT	(		process,
			changeProcessname = True,
			reselect = False
	)

Definition at line 569 of file customisers.py.

References customisoptions(), keepCleaned(), keepSelected(), keepSimulated(), modify_outputModules(), and print().

Referenced by customiseGenerator_preHLT_Reselect().

def customiseGenerator_preHLT(process, changeProcessname=True, reselect=False):
    if reselect:
        dataTier = "RESELECT"
    else:
        dataTier = "SELECT"
    if changeProcessname:
        process._Process__name = "SIMembeddingpreHLT"

    

    process.load("TauAnalysis.MCEmbeddingTools.EmbeddingVertexCorrector_cfi")
    process.VtxSmeared = process.VtxCorrectedToInput.clone()
    print(
        "Correcting Vertex in genEvent to one from input. Replaced 'VtxSmeared' with the Corrector."
    )

    # Disable noise simulation
    process.mix.digitizers.castor.doNoise = cms.bool(False)

    process.mix.digitizers.ecal.doESNoise = cms.bool(False)
    process.mix.digitizers.ecal.doENoise = cms.bool(False)

    process.mix.digitizers.hcal.doNoise = cms.bool(False)
    process.mix.digitizers.hcal.doThermalNoise = cms.bool(False)
    process.mix.digitizers.hcal.doHPDNoise = cms.bool(False)

    process.mix.digitizers.pixel.AddNoisyPixels = cms.bool(False)
    process.mix.digitizers.pixel.AddNoise = cms.bool(False)

    process.mix.digitizers.strip.Noise = cms.bool(False)

    process = customisoptions(process)
    

    return modify_outputModules(
        process,
        [
            keepSelected(dataTier),
            keepCleaned(dataTier),
            keepSimulated(process, processname="SIMembeddingpreHLT"),
        ],
        ["AODSIMoutput"],
    )

customiseGenerator_preHLT_Reselect()

def customisers.customiseGenerator_preHLT_Reselect

(

process

)

Definition at line 614 of file customisers.py.

References customiseGenerator_preHLT().

def customiseGenerator_preHLT_Reselect(process):
    return customiseGenerator_preHLT(process, reselect=True)

customiseKeepPrunedGenParticles()

def customisers.customiseKeepPrunedGenParticles	(		process,
			reselect = False
	)

Definition at line 739 of file customisers.py.

References customisoptions(), keepMerged(), and modify_outputModules().

def customiseKeepPrunedGenParticles(process, reselect=False):
    if reselect:
        dataTier = "RESELECT"
    else:
        dataTier = "SELECT"

    process.keep_step = cms.Path()

    process.load("PhysicsTools.PatAlgos.slimming.genParticles_cff")
    process.keep_step += process.prunedGenParticlesWithStatusOne
    process.load("PhysicsTools.PatAlgos.slimming.prunedGenParticles_cfi")
    process.keep_step += process.prunedGenParticles
    process.load("PhysicsTools.PatAlgos.slimming.packedGenParticles_cfi")
    process.keep_step += process.packedGenParticles
    process.load("PhysicsTools.PatAlgos.slimming.slimmedGenJets_cfi")
    process.keep_step += process.slimmedGenJets

    process.load("PhysicsTools.PatAlgos.mcMatchLayer0.muonMatch_cfi")
    process.keep_step += process.muonMatch
    process.load("PhysicsTools.PatAlgos.mcMatchLayer0.electronMatch_cfi")
    process.keep_step += process.electronMatch
    process.load("PhysicsTools.PatAlgos.mcMatchLayer0.photonMatch_cfi")
    process.keep_step += process.photonMatch
    process.load("PhysicsTools.PatAlgos.mcMatchLayer0.tauMatch_cfi")
    process.keep_step += process.tauMatch
    process.load("PhysicsTools.JetMCAlgos.TauGenJets_cfi")
    process.keep_step += process.tauGenJets
    process.load("PhysicsTools.PatAlgos.mcMatchLayer0.jetFlavourId_cff")
    process.keep_step += process.patJetPartons
    process.load("PhysicsTools.PatAlgos.mcMatchLayer0.jetMatch_cfi")
    process.keep_step += process.patJetPartonMatch

    process.muonMatch.matched = "prunedGenParticles"
    process.electronMatch.matched = "prunedGenParticles"
    process.electronMatch.src = cms.InputTag("reducedEgamma", "reducedGedGsfElectrons")
    process.photonMatch.matched = "prunedGenParticles"
    process.photonMatch.src = cms.InputTag("reducedEgamma", "reducedGedPhotons")
    process.tauMatch.matched = "prunedGenParticles"
    process.tauGenJets.GenParticles = "prunedGenParticles"
    
    process.patJetPartons.particles = "prunedGenParticles"
    process.patJetPartonMatch.matched = "prunedGenParticles"
    process.patJetPartonMatch.mcStatus = [3, 23]
    process.patJetGenJetMatch.matched = "slimmedGenJets"
    process.patJetGenJetMatchAK8.matched = "slimmedGenJetsAK8"
    process.patMuons.embedGenMatch = False
    process.patElectrons.embedGenMatch = False
    process.patPhotons.embedGenMatch = False
    process.patTaus.embedGenMatch = False
    process.patTausBoosted.embedGenMatch = False
    process.patJets.embedGenPartonMatch = False
    # also jet flavour must be switched
    process.patJetFlavourAssociation.rParam = 0.4

    process.schedule.insert(0, process.keep_step)
    process = customisoptions(process)
    return modify_outputModules(process, [keepMerged(dataTier)])

customiseLHE()

def customisers.customiseLHE	(		process,
			changeProcessname = True,
			reselect = False
	)

Definition at line 467 of file customisers.py.

References customisoptions(), keepCleaned(), keepLHE(), keepSelected(), and modify_outputModules().

Referenced by customiseLHEandCleaning().

def customiseLHE(process, changeProcessname=True, reselect=False):
    if reselect:
        dataTier = "RESELECT"
    else:
        dataTier = "SELECT"
    if changeProcessname:
        process._Process__name = "LHEembedding"
    process.load("TauAnalysis.MCEmbeddingTools.EmbeddingLHEProducer_cfi")
    if reselect:
        process.externalLHEProducer.vertices = cms.InputTag(
            "offlineSlimmedPrimaryVertices", "", "RESELECT"
        )
    process.lheproduction = cms.Path(process.makeexternalLHEProducer)
    process.schedule.insert(0, process.lheproduction)

    process = customisoptions(process)
    return modify_outputModules(
        process,
        [keepSelected(dataTier), keepCleaned(dataTier), keepLHE()],
        ["MINIAODoutput"],
    )

customiseLHEandCleaning()

def customisers.customiseLHEandCleaning	(		process,
			reselect = False
	)

Definition at line 490 of file customisers.py.

References customiseCleaning(), and customiseLHE().

Referenced by customiseLHEandCleaning_Reselect().

def customiseLHEandCleaning(process, reselect=False):
    process._Process__name = "LHEembeddingCLEAN"
    process = customiseCleaning(process, changeProcessname=False, reselect=reselect)
    process = customiseLHE(process, changeProcessname=False, reselect=reselect)
    return process

customiseLHEandCleaning_Reselect()

def customisers.customiseLHEandCleaning_Reselect

(

process

)

Definition at line 497 of file customisers.py.

References customiseLHEandCleaning().

def customiseLHEandCleaning_Reselect(process):
    return customiseLHEandCleaning(process, reselect=True)

customiseMCFilter()

def customisers.customiseMCFilter

(

process

)

Definition at line 1116 of file customisers.py.

References fix_input_tags().

Referenced by customiseFilterTTbartoMuMu().

def customiseMCFilter(process):
    process.schedule.insert(-1, process.MCFilter)
    outputModulesList = [key for key, value in process.outputModules.items()]
    for outputModule in outputModulesList:
        outputModule = getattr(process, outputModule)
        outputModule.SelectEvents = cms.untracked.PSet(
            SelectEvents=cms.vstring("MCFilter")
        )
    return process

customiseMerging()

def customisers.customiseMerging	(		process,
			changeProcessname = True,
			reselect = False
	)

Definition at line 800 of file customisers.py.

References customisoptions(), keepMerged(), miniAOD_tools.miniAOD_customizeMC(), modify_outputModules(), print(), submitPVValidationJobs.split(), and str.

Referenced by customiseMerging_Reselect().

def customiseMerging(process, changeProcessname=True, reselect=False):

    print("**** Attention: overriding behaviour of 'removeMCMatching' ****")

    from PhysicsTools.PatAlgos.slimming.miniAOD_tools import miniAOD_customizeMC
    def performMCMatching(process, names, postfix, outputModules):
        miniAOD_customizeMC(process)

    import PhysicsTools.PatAlgos.tools.coreTools
    PhysicsTools.PatAlgos.tools.coreTools.removeMCMatching = performMCMatching

    if changeProcessname:
        process._Process__name = "MERGE"
    if reselect:
        dataTier = "RESELECT"
    else:
        dataTier = "SELECT"

    process.source.inputCommands = cms.untracked.vstring()
    process.source.inputCommands.append("keep *_*_*_*")
    process.load("PhysicsTools.PatAlgos.slimming.unpackedPatTrigger_cfi")
    process.unpackedPatTrigger.triggerResults = cms.InputTag("TriggerResults::SIMembeddingHLT")

    # process.source.inputCommands.append("drop *_*_*_SELECT")
    # process.source.inputCommands.append("drop *_*_*_SIMembedding")
    # process.source.inputCommands.append("drop *_*_*_LHEembeddingCLEAN")
    # process.source.inputCommands.extend(keepSimulated())
    # process.source.inputCommands.extend(keepCleaned())

    process.load('Configuration.StandardSequences.RawToDigi_cff')
    process.load("Configuration.StandardSequences.Reconstruction_Data_cff")
    process.merge_step = cms.Path()
    # produce local CT PPS reco
    process.load("RecoPPS.Configuration.recoCTPPS_cff")
    process.totemRPRawToDigi.rawDataTag = cms.InputTag("rawDataCollector", "", "LHC")
    process.ctppsDiamondRawToDigi.rawDataTag = cms.InputTag("rawDataCollector", "", "LHC")
    process.ctppsPixelDigis.inputLabel = cms.InputTag("rawDataCollector", "", "LHC")
    process.merge_step += process.totemRPRawToDigi
    process.merge_step += process.ctppsDiamondRawToDigi
    process.merge_step += process.ctppsPixelDigis
    process.merge_step += cms.Sequence(process.recoCTPPSTask)
    # produce local Calo
    process.load("RecoLocalCalo.Configuration.RecoLocalCalo_cff")
    process.merge_step += process.calolocalreco
    #process.merge_step += process.caloglobalreco
    process.merge_step += process.reducedHcalRecHitsSequence

    # produce hcal towers
    process.load("RecoLocalCalo.CaloTowersCreator.calotowermaker_cfi")
    process.merge_step += process.calotowermaker
    process.merge_step += process.towerMaker

    # produce clusters
    process.load("RecoEcal.Configuration.RecoEcal_cff")
    process.merge_step += process.ecalClusters

    # produce PFCluster Collections
    process.load("RecoParticleFlow.PFClusterProducer.particleFlowCluster_cff")
    process.merge_step += process.particleFlowCluster
    process.load(
        "RecoEcal.EgammaClusterProducers.particleFlowSuperClusteringSequence_cff"
    )
    process.merge_step += process.particleFlowSuperClusteringSequence

    # muonEcalDetIds
    process.load("RecoMuon.MuonIdentification.muons1stStep_cfi")
    process.merge_step += process.muonEcalDetIds
    process.merge_step += process.muonShowerInformation

    # muon Isolation sequences
    process.load("RecoMuon.MuonIsolationProducers.muIsolation_cff")
    process.merge_step += process.muIsolation
    process.merge_step += process.muIsolationDisplaced

    # muon ID selection type sequences
    process.load("RecoMuon.MuonIdentification.muonSelectionTypeValueMapProducer_cff")
    process.merge_step += process.muonSelectionTypeSequence

    # displaced muons extras & tracks
    process.load("RecoMuon.MuonIdentification.displacedMuonReducedTrackExtras_cfi")
    process.merge_step += process.displacedMuonReducedTrackExtras

    process.load("RecoMuon.Configuration.MergeDisplacedTrackCollections_cff")
    process.merge_step += process.displacedTracksSequence

    # Other things
    process.merge_step += process.doAlldEdXEstimators
    process.merge_step += process.vertexreco
    process.unsortedOfflinePrimaryVertices.beamSpotLabel = cms.InputTag(
        "offlineBeamSpot", "", dataTier
    )
    process.ak4CaloJetsForTrk.srcPVs = cms.InputTag(
        "firstStepPrimaryVertices", "", dataTier
    )
    process.dedxHitInfo.clusterShapeCache = cms.InputTag("")

    # process.muons.FillDetectorBasedIsolation = cms.bool(False)
    # process.muons.FillSelectorMaps = cms.bool(False)
    # process.muons.FillShoweringInfo = cms.bool(False)
    # process.muons.FillCosmicsIdMap = cms.bool(False)

    # process.displacedMuons.FillDetectorBasedIsolation = cms.bool(False)
    # process.displacedMuons.FillSelectorMaps = cms.bool(False)
    # process.displacedMuons.FillShoweringInfo = cms.bool(False)
    # process.displacedMuons.FillCosmicsIdMap = cms.bool(False)

    # seed configuration needed for seedmerger
    #process.load(
    #    "RecoEgamma.EgammaElectronProducers.ecalDrivenElectronSeedsParameters_cff"
    #)
    #process.ecalDrivenElectronSeeds.SeedConfiguration = cms.PSet(
    #    process.ecalDrivenElectronSeedsParameters
    #)

    process.merge_step += process.highlevelreco
    # process.merge_step.remove(process.reducedEcalRecHitsEE)
    # process.merge_step.remove(process.reducedEcalRecHitsEB)

    # process.merge_step.remove(process.ak4JetTracksAssociatorExplicit)

    # process.merge_step.remove(process.cosmicsVeto)
    # process.merge_step.remove(process.cosmicsVetoTrackCandidates)
    # process.merge_step.remove(process.ecalDrivenGsfElectronCores)
    # process.merge_step.remove(process.ecalDrivenGsfElectrons)
    # process.merge_step.remove(process.gedPhotonsTmp)
    # process.merge_step.remove(process.particleFlowTmp)

    # process.merge_step.remove(process.hcalnoise)

    process.load("CommonTools.ParticleFlow.genForPF2PAT_cff")

    # process.muonsFromCosmics.ShowerDigiFillerParameters.dtDigiCollectionLabel = cms.InputTag("simMuonDTDigis")

    process.merge_step += process.genForPF2PATSequence

    # Replace manipulated modules contained in merg_step with Mergers, and
    # put remaining ones into a list to be sorted to avoid deadlocks
    modules_to_be_ordered = {}
    # prepare reco list to determine indices
    reconstruction_modules_list = str(process.RawToDigi).split(",")
    reconstruction_modules_list += str(process.reconstruction).split(",")
    for akt_manimod in to_bemanipulate:
        if "MERGE" in akt_manimod.steps:
            mergCollections_in = cms.VInputTag()
            for instance in akt_manimod.instance:
                mergCollections_in.append(
                    cms.InputTag(
                        akt_manimod.merge_prefix + akt_manimod.module_name,
                        instance,
                        "SIMembedding",
                    )
                )
                mergCollections_in.append(
                    cms.InputTag(
                        akt_manimod.merge_prefix + akt_manimod.module_name,
                        instance,
                        "LHEembeddingCLEAN",
                    )
                )
            setattr(
                process,
                akt_manimod.module_name,
                cms.EDProducer(
                    akt_manimod.merger_name, mergCollections=mergCollections_in
                ),
            )
            if not process.merge_step.contains(getattr(process, akt_manimod.module_name)):
                modules_to_be_ordered[akt_manimod.module_name] = -1
    # Determine indices and place them in right order into the list
    for name,index in modules_to_be_ordered.items():
        if name in reconstruction_modules_list:
            modules_to_be_ordered[name] = reconstruction_modules_list.index(name)
        else:
            print("ERROR:",name,"not prepared in modules list. Please adapt 'customiseMerging'")
            sys.exit(1)

    modules_ordered = sorted(list(modules_to_be_ordered.items()), key=lambda x : -x[1])
    for m in modules_ordered:
        process.merge_step.insert(0, getattr(process, m[0]))


    process.schedule.insert(0, process.merge_step)
    # process.load('PhysicsTools.PatAlgos.slimming.slimmedGenJets_cfi')
    process = customisoptions(process)
    return modify_outputModules(process, [keepMerged(dataTier)])

customiseMerging_Reselect()

def customisers.customiseMerging_Reselect	(		process,
			changeProcessname = True
	)

Definition at line 987 of file customisers.py.

References customiseMerging().

def customiseMerging_Reselect(process, changeProcessname=True):
    return customiseMerging(process, changeProcessname=changeProcessname, reselect=True)

customiseNanoAOD()

def customisers.customiseNanoAOD

(

process

)

Customize NanoAOD ################################.

Definition at line 994 of file customisers.py.

References common_cff.ExtVar().

def customiseNanoAOD(process):

    process.load("PhysicsTools.NanoAOD.nano_cff")
    process.nanoAOD_step.insert(0, cms.Sequence(process.nanoTableTaskFS))


    for outputModule in process.outputModules.values():
       outputModule.outputCommands.append("keep edmTriggerResults_*_*_SIMembeddingpreHLT")
       outputModule.outputCommands.append("keep edmTriggerResults_*_*_SIMembeddingHLT")
       outputModule.outputCommands.append("keep edmTriggerResults_*_*_SIMembedding")
       outputModule.outputCommands.append("keep edmTriggerResults_*_*_MERGE")
       outputModule.outputCommands.append("keep edmTriggerResults_*_*_NANO")
       outputModule.outputCommands.remove("keep edmTriggerResults_*_*_*")

    process.load("PhysicsTools.NanoAOD.l1trig_cff")
    process.nanoAOD_step.insert(0, cms.Sequence(process.l1TablesTask))

    process.embeddingTable = cms.EDProducer(
        "GlobalVariablesTableProducer",
        name=cms.string("TauEmbedding"),
        # doc=cms.string("TauEmbedding"),
        variables=cms.PSet(
            nInitialPairCandidates=ExtVar(
                cms.InputTag("selectedMuonsForEmbedding", "nPairCandidates"),
                float,
                doc="number of muons pairs suitable for selection (for internal studies only)",
            ),
            SelectionOldMass=ExtVar(
                cms.InputTag("selectedMuonsForEmbedding", "oldMass"),
                float,
                doc="Mass of the Dimuon pair using the old selection algorithm (for internal studies only)",
            ),
            SelectionNewMass=ExtVar(
                cms.InputTag("selectedMuonsForEmbedding", "newMass"),
                float,
                doc="Mass of the Dimuon pair using the new selection algorithm (for internal studies only)",
            ),
            isMediumLeadingMuon=ExtVar(
                cms.InputTag("selectedMuonsForEmbedding", "isMediumLeadingMuon"),
                bool,
                doc="leading muon ID (medium)",
            ),
            isMediumTrailingMuon=ExtVar(
                cms.InputTag("selectedMuonsForEmbedding", "isMediumTrailingMuon"),
                bool,
                doc="trailing muon ID (medium)",
            ),
            isTightLeadingMuon=ExtVar(
                cms.InputTag("selectedMuonsForEmbedding", "isTightLeadingMuon"),
                bool,
                doc="leading muon ID (tight)",
            ),
            isTightTrailingMuon=ExtVar(
                cms.InputTag("selectedMuonsForEmbedding", "isTightTrailingMuon"),
                bool,
                doc="trailing muon ID (tight)",
            ),
            initialMETEt=ExtVar(
                cms.InputTag("selectedMuonsForEmbedding", "initialMETEt"),
                float,
                doc="MET Et of selected event",
            ),
            initialMETphi=ExtVar(
                cms.InputTag("selectedMuonsForEmbedding", "initialMETphi"),
                float,
                doc="MET phi of selected event",
            ),
            initialPuppiMETEt=ExtVar(
                cms.InputTag("selectedMuonsForEmbedding", "initialPuppiMETEt"),
                float,
                doc="PuppiMET Et of selected event",
            ),
            initialPuppiMETphi=ExtVar(
                cms.InputTag("selectedMuonsForEmbedding", "initialPuppiMETphi"),
                float,
                doc="PuppiMET phi of selected event",
            ),
        ),
    )
    process.embeddingTableTask = cms.Task(process.embeddingTable)
    process.schedule.associate(process.embeddingTableTask)

    return process

customiseSelecting()

def customisers.customiseSelecting	(		process,
			reselect = False
	)

Definition at line 299 of file customisers.py.

References customisoptions(), keepSelected(), and modify_outputModules().

Referenced by customiseSelecting_Reselect().

def customiseSelecting(process, reselect=False):
    if reselect:
        process._Process__name = "RESELECT"
        dataTier = "RESELECT"
        # process.source.inputCommands = cms.untracked.vstring("drop *",
        #     "keep *_*_*_LHC",
        #     "keep *_*_*_HLT",
        # )
    else:
        process._Process__name = "SELECT"
        dataTier = "SELECT"

    process.load("TauAnalysis.MCEmbeddingTools.SelectingProcedure_cff")
    process.patMuonsAfterKinCuts.src = cms.InputTag("slimmedMuons", "", dataTier)
    process.selectedMuonsForEmbedding.PuppiMet = cms.InputTag(
        "slimmedMETsPuppi", "", dataTier
    )
    process.selectedMuonsForEmbedding.Met = cms.InputTag("slimmedMETs", "", dataTier)
    process.patMuonsAfterID = process.patMuonsAfterLooseID.clone()

    process.selecting = cms.Path(process.makePatMuonsZmumuSelection)
    process.schedule.insert(-1, process.selecting)

    outputModulesList = [key for key, value in process.outputModules.items()]
    for outputModule in outputModulesList:
        outputModule = getattr(process, outputModule)
        outputModule.SelectEvents = cms.untracked.PSet(
            SelectEvents=cms.vstring("selecting")
        )
        outputModule.outputCommands.extend(keepSelected(dataTier))

    process = customisoptions(process)
    return modify_outputModules(process, [keepSelected(dataTier)])

customiseSelecting_Reselect()

def customisers.customiseSelecting_Reselect

(

process

)

Definition at line 334 of file customisers.py.

References customiseSelecting().

def customiseSelecting_Reselect(process):
    return customiseSelecting(process, reselect=True)

customisoptions()

def customisers.customisoptions

(

process

)

cross Customizers ###########################

additional Customizer ###########################

Definition at line 1085 of file customisers.py.

Referenced by customiseCleaning(), customiseGenerator_HLT(), customiseGenerator_postHLT(), customiseGenerator_preHLT(), customiseKeepPrunedGenParticles(), customiseLHE(), customiseMerging(), and customiseSelecting().

def customisoptions(process):
    if not hasattr(process, "options"):
        process.options = cms.untracked.PSet()
    process.options.emptyRunLumiMode = cms.untracked.string(
        "doNotHandleEmptyRunsAndLumis"
    )
    if not hasattr(process, "bunchSpacingProducer"):
        process.bunchSpacingProducer = cms.EDProducer("BunchSpacingProducer")
    process.bunchSpacingProducer.bunchSpacingOverride = cms.uint32(25)
    process.bunchSpacingProducer.overrideBunchSpacing = cms.bool(True)
    process.options.numberOfThreads = cms.untracked.uint32(1)
    process.options.numberOfStreams = cms.untracked.uint32(0)
    return process

fix_input_tags()

def customisers.fix_input_tags	(		process,
			formodules = ["generalTracks",
			cscSegments,
			dt4DSegments,
			rpcRecHits
	)

Definition at line 1129 of file customisers.py.

References print().

Referenced by customiseMCFilter().

):
    def change_tags_process(test_input):
        if isinstance(test_input, cms.InputTag):
            if test_input.getModuleLabel() in formodules:
                test_input.setProcessName(process._Process__name)

    def search_for_tags(pset):
        if isinstance(pset, dict):
            for key in pset:
                if isinstance(pset[key], cms.VInputTag):
                    for akt_inputTag in pset[key]:
                        change_tags_process(akt_inputTag)
                elif isinstance(pset[key], cms.PSet):
                    search_for_tags(pset[key].__dict__)
                elif isinstance(pset[key], cms.VPSet):
                    for akt_pset in pset[key]:
                        search_for_tags(akt_pset.__dict__)
                else:
                    change_tags_process(pset[key])
        else:
            print("must be python dict not a {}".format(type(pset)))

    for module in process.producers_():
        search_for_tags(getattr(process, module).__dict__)
    for module in process.filters_():
        search_for_tags(getattr(process, module).__dict__)
    for module in process.analyzers_():
        search_for_tags(getattr(process, module).__dict__)

    return process

keepCleaned()

def customisers.keepCleaned

(

dataTier

)

Customizer for cleaning ###########################.

Definition at line 341 of file customisers.py.

Referenced by customiseCleaning(), customiseGenerator_HLT(), customiseGenerator_postHLT(), customiseGenerator_preHLT(), and customiseLHE().

def keepCleaned(dataTier):
    ret_vstring = cms.untracked.vstring(
        #                        "drop *_*_*_LHEembeddingCLEAN",
        #                        "drop *_*_*_CLEAN"
        "drop *_*_*_" + dataTier,
        "keep *_patMuonsAfterID_*_" + dataTier,
        "keep *_slimmedMuons_*_" + dataTier,
        # "keep *_slimmedMuonTrackExtras_*_" + dataTier,
        "keep *_selectedMuonsForEmbedding_*_" + dataTier,
        "keep recoVertexs_offlineSlimmedPrimaryVertices_*_" + dataTier,
        "keep *_firstStepPrimaryVertices_*_" + dataTier,
        "keep *_offlineBeamSpot_*_" + dataTier,
        "keep *_l1extraParticles_*_" + dataTier,
        "keep TrajectorySeeds_*_*_*",
        "keep recoElectronSeeds_*_*_*",
        "drop recoIsoDepositedmValueMap_muIsoDepositTk_*_*" ,
        "drop recoIsoDepositedmValueMap_muIsoDepositTkDisplaced_*_*",
        "drop *_ctppsProtons_*_*",
        "drop *_ctppsLocalTrackLiteProducer_*_*",
        "drop *_ctppsDiamondLocalTracks_*_*",
        "drop *_ctppsDiamondRecHits_*_*",
        "drop *_ctppsDiamondRawToDigi_*_*",
        "drop *_ctppsPixelLocalTracks_*_*",
        "drop *_ctppsPixelRecHits_*_*",
        "drop *_ctppsPixelClusters_*_*",
        "drop *_ctppsPixelDigis_*_*",
        "drop *_totemRPLocalTrackFitter_*_*",
        "drop *_totemRPUVPatternFinder_*_*",
        "drop *_totemRPRecHitProducer_*_*",
        "drop *_totemRPClusterProducer_*_*",
        "drop *_totemRPRawToDigi_*_*",
        "drop *_muonSimClassifier_*_*",
        # "keep recoPFClusters_*_*_*",
        # "keep recoPFRecHits_*_*_*"
    )

    for akt_manimod in to_bemanipulate:
        if "MERGE" in akt_manimod.steps:
            ret_vstring.append(
                "keep *_" + akt_manimod.module_name + "_*_LHEembeddingCLEAN"
            )
            ret_vstring.append("keep *_" + akt_manimod.module_name + "_*_CLEAN")
    ret_vstring.append("keep *_standAloneMuons_*_LHEembeddingCLEAN")
    ret_vstring.append("keep *_glbTrackQual_*_LHEembeddingCLEAN")
    return ret_vstring

keepLHE()

def customisers.keepLHE

(

)

Customizer for LHE ###########################.

Definition at line 460 of file customisers.py.

Referenced by customiseGenerator_HLT(), customiseGenerator_postHLT(), and customiseLHE().

def keepLHE():
    ret_vstring = cms.untracked.vstring()
    ret_vstring.append("keep *_externalLHEProducer_*_LHEembedding")
    ret_vstring.append("keep *_externalLHEProducer_*_LHEembeddingCLEAN")
    return ret_vstring

keepMerged()

def customisers.keepMerged

(

dataTier = "SELECT"

)

Customizer for merging ###########################.

Definition at line 706 of file customisers.py.

Referenced by customiseKeepPrunedGenParticles(), and customiseMerging().

def keepMerged(dataTier="SELECT"):
    ret_vstring = cms.untracked.vstring()
    ret_vstring.append("drop *_*_*_" + dataTier)
    ret_vstring.append("keep *_prunedGenParticles_*_MERGE")
    ret_vstring.append("keep *_generator_*_SIMembeddingpreHLT")
    ret_vstring.append("keep *_generator_*_SIMembeddingHLT")
    ret_vstring.append("keep *_generator_*_SIMembedding")
    ret_vstring.append("keep *_selectedMuonsForEmbedding_*_*")
    ret_vstring.append("keep *_unpackedPatTrigger_*_*")
    ret_vstring.extend(cms.untracked.vstring(
        'keep patPackedGenParticles_packedGenParticles_*_*',
        'keep recoGenParticles_prunedGenParticles_*_*',
        'keep *_packedPFCandidateToGenAssociation_*_*',
        'keep *_lostTracksToGenAssociation_*_*',
        'keep LHEEventProduct_*_*_*',
        'keep GenFilterInfo_*_*_*',
        'keep GenLumiInfoHeader_generator_*_*',
        'keep GenLumiInfoProduct_*_*_*',
        'keep GenEventInfoProduct_generator_*_*',
        'keep recoGenParticles_genPUProtons_*_*',
        'keep *_slimmedGenJetsFlavourInfos_*_*',
        'keep *_slimmedGenJets__*',
        'keep *_slimmedGenJetsAK8__*',
        'keep *_slimmedGenJetsAK8SoftDropSubJets__*',
        'keep *_genMetTrue_*_*',
        # RUN
        'keep LHERunInfoProduct_*_*_*',
        'keep GenRunInfoProduct_*_*_*',
        'keep *_genParticles_xyz0_*',
        'keep *_genParticles_t0_*'))
    return ret_vstring

keepSelected()

def customisers.keepSelected

(

dataTier

)

Customizer for Selecting ###########################.

Definition at line 281 of file customisers.py.

Referenced by customiseCleaning(), customiseGenerator_HLT(), customiseGenerator_postHLT(), customiseGenerator_preHLT(), customiseLHE(), and customiseSelecting().

def keepSelected(dataTier):
    ret_vstring = cms.untracked.vstring(
        # "drop *_*_*_"+dataTier,
        "keep *_patMuonsAfterID_*_" + dataTier,
        "keep *_slimmedMuons_*_" + dataTier,
        "keep *_slimmedMuonTrackExtras_*_" + dataTier,
        "keep *_selectedMuonsForEmbedding_*_" + dataTier,
        "keep recoVertexs_offlineSlimmedPrimaryVertices_*_" + dataTier,
        "keep *_firstStepPrimaryVertices_*_" + dataTier,
        "keep *_offlineBeamSpot_*_" + dataTier,
        "keep *_ecalDrivenElectronSeeds_*_" + dataTier,
    )
    for akt_manimod in to_bemanipulate:
        if "CLEAN" in akt_manimod.steps:
            ret_vstring.append("keep *_" + akt_manimod.module_name + "_*_" + dataTier)
    return ret_vstring

keepSimulated()

def customisers.keepSimulated	(		process,
			processname = "SIMembedding"
	)

Customizer for simulaton ###########################.

Definition at line 504 of file customisers.py.

References any().

Referenced by customiseGenerator_HLT(), customiseGenerator_postHLT(), and customiseGenerator_preHLT().

def keepSimulated(process, processname="SIMembedding"):
    ret_vstring = cms.untracked.vstring()
    for akt_manimod in to_bemanipulate:
        if "MERGE" in akt_manimod.steps:
            ret_vstring.append(
                "keep *_" + akt_manimod.module_name + "_*_{}".format(processname)
            )
    ret_vstring.append("keep *_genParticles_*_{}".format(processname))
    ret_vstring.append("keep *_standAloneMuons_*_{}".format(processname))
    ret_vstring.append("keep *_glbTrackQual_*_{}".format(processname))
    ret_vstring.append("keep *_generator_*_{}".format(processname))
    ret_vstring.append("keep *_addPileupInfo_*_{}".format(processname))
    ret_vstring.append("keep *_selectedMuonsForEmbedding_*_*")
    ret_vstring.append("keep *_slimmedAddPileupInfo_*_*")
    ret_vstring.append("keep *_embeddingHltPixelVertices_*_*")
    ret_vstring.append("keep *_*_vertexPosition_*")
    ret_vstring.append("keep recoMuons_muonsFromCosmics_*_*")
    ret_vstring.append("keep recoTracks_cosmicMuons1Leg_*_*")
    ret_vstring.append("keep recoMuons_muonsFromCosmics1Leg_*_*")
    ret_vstring.append("keep *_muonDTDigis_*_*")
    ret_vstring.append("keep *_muonCSCDigis_*_*")
    ret_vstring.append("keep TrajectorySeeds_*_*_*")
    ret_vstring.append("keep recoElectronSeeds_*_*_*")
    ret_vstring.append("drop recoIsoDepositedmValueMap_muIsoDepositTk_*_*")
    ret_vstring.append("drop recoIsoDepositedmValueMap_muIsoDepositTkDisplaced_*_*")
    ret_vstring.append("drop *_ctppsProtons_*_*")
    ret_vstring.append("drop *_ctppsLocalTrackLiteProducer_*_*")
    ret_vstring.append("drop *_ctppsDiamondLocalTracks_*_*")
    ret_vstring.append("drop *_ctppsDiamondRecHits_*_*")
    ret_vstring.append("drop *_ctppsDiamondRawToDigi_*_*")
    ret_vstring.append("drop *_ctppsPixelLocalTracks_*_*")
    ret_vstring.append("drop *_ctppsPixelRecHits_*_*")
    ret_vstring.append("drop *_ctppsPixelClusters_*_*")
    ret_vstring.append("drop *_ctppsPixelDigis_*_*")
    ret_vstring.append("drop *_totemRPLocalTrackFitter_*_*")
    ret_vstring.append("drop *_totemRPUVPatternFinder_*_*")
    ret_vstring.append("drop *_totemRPRecHitProducer_*_*")
    ret_vstring.append("drop *_totemRPClusterProducer_*_*")
    ret_vstring.append("drop *_totemRPRawToDigi_*_*")
    ret_vstring.append("drop *_muonSimClassifier_*_*")

    # for those two steps, the output has to be modified
    # to keep the information from the cleaning step in the output file
    if processname == "SIMembeddingpreHLT" or processname == "SIMembeddingHLT":
        rawreco_commands = set(process.RAWRECOEventContent.outputCommands)
        rawreco_commands_excl = rawreco_commands - set(
            process.RAWSIMEventContent.outputCommands
        )
        for entry in rawreco_commands_excl:
            if (
                processname == "SIMembeddingpreHLT"
                and "muonReducedTrackExtras" in entry
            ):
                continue
            if not any(
                x in entry
                for x in [
                    "TotemTimingLocalTrack",
                    "ForwardProton",
                    "ctppsDiamondLocalTracks",
                ]
            ):
                ret_vstring.append(entry)
    return ret_vstring

modify_outputModules()

def customisers.modify_outputModules	(		process,
			keep_drop_list = [],
			module_veto_list = []
	)

Definition at line 267 of file customisers.py.

Referenced by customiseCleaning(), customiseGenerator_HLT(), customiseGenerator_postHLT(), customiseGenerator_preHLT(), customiseKeepPrunedGenParticles(), customiseLHE(), customiseMerging(), and customiseSelecting().

def modify_outputModules(process, keep_drop_list=[], module_veto_list=[]):
    outputModulesList = [key for key, value in process.outputModules.items()]
    for outputModule in outputModulesList:
        if outputModule in module_veto_list:
            continue
        outputModule = getattr(process, outputModule)
        for add_element in keep_drop_list:
            outputModule.outputCommands.extend(add_element)
    return process

Variable Documentation

instance

customisers.instance

Definition at line 141 of file customisers.py.

manipulator_name

customisers.manipulator_name

Definition at line 47 of file customisers.py.

module_name

customisers.module_name

Definition at line 46 of file customisers.py.

steps

customisers.steps

Definition at line 48 of file customisers.py.

Referenced by l1t::BXVectorInputProducer.convertEtaToHW(), l1t::GenToInputProducer.convertEtaToHW(), l1t::BXVectorInputProducer.convertPhiToHW(), l1t::GenToInputProducer.convertPhiToHW(), HLTMuonValidator.dqmBeginRun(), DD4hep_TrackingMaterialPlotter.fill_gradient(), TrackingMaterialPlotter.fill_gradient(), DD4hep_ListGroups.fillGradient(), ListGroups.fillGradient(), MuonResiduals6DOFrphiFitter.fit(), MuonResiduals5DOFFitter.fit(), MuonResiduals6DOFFitter.fit(), CachedTrajectory.getTrajectory(), XHistogram.splitSegment(), DD4hep_XHistogram.splitSegment(), CosmicMuonUtilities.stepPropagate(), and CSCGasCollisions.writeSummary().

to_bemanipulate

customisers.to_bemanipulate

Definition at line 41 of file customisers.py.

---

Generated for CMSSW Reference Manual by  1.8.14