TrackTrigger/python/TkOnlyDigi_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 def TkOnlyDigi(process):
0004     print("!!! Special version of the digitization for tracker only !!!")
0005     if hasattr(process,'digitisationTkOnly_step'):
0006         process=customise_DigiTkOnly(process)
0007
0008     return process
0009
0010 def customise_DigiTkOnly(process):
0011     process.load('Configuration.StandardSequences.Digi_cff')
0012     process.doAllDigi = cms.Sequence()
0013     process.load('SimGeneral.MixingModule.mixObjects_cfi')
0014     process.digitisationTkOnly_step.remove(process.mix.mixObjects.mixCH)
0015     del process.simCastorDigis
0016     del process.simEcalUnsuppressedDigis
0017     del process.simHcalUnsuppressedDigis
0018     process.mix.digitizers = cms.PSet(process.theDigitizersValid)
0019     del process.mix.digitizers.ecal
0020     del process.mix.digitizers.hcal
0021     del process.mix.digitizers.castor
0022     process.digitisationTkOnly_step.remove(process.mix.digitizers.pixel)
0023     process.load('SimTracker.SiPhase2Digitizer.phase2TrackerDigitizer_cfi')
0024     process.mix.digitizers.pixel=process.phase2TrackerDigitizer
0025     process.mix.digitizers.strip.ROUList = cms.vstring("g4SimHitsTrackerHitsPixelBarrelLowTof",
0026                          'g4SimHitsTrackerHitsPixelEndcapLowTof')
0027     #Check if mergedtruth is in the sequence first, could be taken out depending on cmsDriver options
0028     if hasattr(process.mix.digitizers,"mergedtruth") :
0029         process.mix.digitizers.mergedtruth.simHitCollections.muon = cms.VInputTag( )
0030         process.mix.digitizers.mergedtruth.simHitCollections.tracker.remove( cms.InputTag("g4SimHits","TrackerHitsTIBLowTof"))
0031         process.mix.digitizers.mergedtruth.simHitCollections.tracker.remove( cms.InputTag("g4SimHits","TrackerHitsTIBHighTof"))
0032         process.mix.digitizers.mergedtruth.simHitCollections.tracker.remove( cms.InputTag("g4SimHits","TrackerHitsTOBLowTof"))
0033         process.mix.digitizers.mergedtruth.simHitCollections.tracker.remove( cms.InputTag("g4SimHits","TrackerHitsTOBHighTof"))
0034         process.mix.digitizers.mergedtruth.simHitCollections.tracker.remove( cms.InputTag("g4SimHits","TrackerHitsTECLowTof"))
0035         process.mix.digitizers.mergedtruth.simHitCollections.tracker.remove( cms.InputTag("g4SimHits","TrackerHitsTECHighTof"))
0036         process.mix.digitizers.mergedtruth.simHitCollections.tracker.remove( cms.InputTag("g4SimHits","TrackerHitsTIDLowTof"))
0037         process.mix.digitizers.mergedtruth.simHitCollections.tracker.remove( cms.InputTag("g4SimHits","TrackerHitsTIDHighTof"))
0038
0039     # keep new digis
0040     alist=['FEVTDEBUG','FEVTDEBUGHLT','FEVT']
0041     for a in alist:
0042         b=a+'output'
0043         if hasattr(process,b):
0044             getattr(process,b).outputCommands.append('keep Phase2TrackerDigiedmDetSetVector_*_*_*')
0045     return process
0046