FastSimWithCTPPS_cff.py

Go to the documentation of this file.

import FWCore.ParameterSet.Config as cms

def customise(process):
    if hasattr(process.VtxSmeared,"X0"):
        VertexX = process.VtxSmeared.X0
        VertexY = process.VtxSmeared.Y0
        VertexZ = process.VtxSmeared.Z0

    if hasattr(process.VtxSmeared,"MeanX"):
        VertexX = process.VtxSmeared.MeanX
        VertexY = process.VtxSmeared.MeanY
        VertexZ = process.VtxSmeared.MeanZ


    
    process.load('SimTransport.HectorProducer.HectorTransportCTPPS_cfi')
    process.LHCTransport.CTPPSHector.VtxMeanX  = VertexX
    process.LHCTransport.CTPPSHector.VtxMeanY  = VertexY
    process.LHCTransport.CTPPSHector.VtxMeanZ = VertexZ

    ###################################
    process.load('FastSimulation.CTPPSSimHitProducer.CTPPSSimHitProducer_cfi')
    process.load('FastSimulation.CTPPSRecHitProducer.CTPPSRecHitProducer_cfi')
    process.load('FastSimulation.CTPPSFastTrackingProducer.CTPPSFastTrackingProducer_cfi')
    ###################################
    process.mix.mixObjects.mixSH.crossingFrames.append('CTPPSHits')
    process.mix.mixObjects.mixSH.input.append(cms.InputTag("CTPPSSimHits","CTPPSHits"))
    process.mix.mixObjects.mixSH.subdets.append('CTPPSHits')
       
            
    # CTPPS simHit sequence 
    process.simulation_step.replace(process.psim,process.psim+process.CTPPSSimHits)
        # SimTransport on path
        process.transport_step = cms.Path(process.generator+process.LHCTransport)

    process.schedule.insert(2,process.transport_step)


    # output
        outputModule = None
        outdict = process.outputModules_()
        if "AODSIMoutput" in outdict:
        process.AODSIMoutput.outputCommands.extend(cms.untracked.vstring('keep *_CTPPSSimHits_*_*','keep *_CTPPSFastRecHits_*_*'
,'keep *_CTPPSFastTracks_*_*'))
        process.reconstruction_step.replace(process.reconstruction,process.reconstruction*process.CTPPSFastRecHits*process.CTPPSFastTracks)     
        elif "FASTPUoutput" in outdict:
        process.FASTPUoutput.outputCommands.extend(cms.untracked.vstring('keep *_CTPPSSimHits_*_*'))    
            
    
    return(process)