Configuration/python/RecoHGCal_EventContent_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 from RecoHGCal.TICL.iterativeTICL_cff import ticlIterLabelsMerge
0004
0005 trackstersIters = ['keep *_ticlTracksters'+iteration+'_*_*' for iteration in ticlIterLabelsMerge]
0006
0007 #AOD content
0008 TICL_AOD = cms.PSet(
0009     outputCommands = cms.untracked.vstring()
0010     )
0011
0012 #RECO content
0013 TICL_RECO = cms.PSet(
0014     outputCommands = cms.untracked.vstring(
0015       trackstersIters +
0016       ['keep *_ticlTrackstersHFNoseTrkEM_*_*',
0017        'keep *_ticlTrackstersHFNoseEM_*_*',
0018        'keep *_ticlTrackstersHFNoseTrk_*_*',
0019        'keep *_ticlTrackstersHFNoseMIP_*_*',
0020        'keep *_ticlTrackstersHFNoseHAD_*_*',
0021        'keep *_ticlTrackstersHFNoseMerge_*_*',] +
0022       ['keep *_pfTICL_*_*']
0023       )
0024     )
0025 TICL_RECO.outputCommands.extend(TICL_AOD.outputCommands)
0026
0027 # FEVT Content
0028 TICL_FEVT = cms.PSet(
0029     outputCommands = cms.untracked.vstring(
0030       'keep *_ticlSimTracksters_*_*',
0031       'keep *_ticlSimTrackstersFromCP_*_*',
0032       )
0033     )
0034 TICL_FEVT.outputCommands.extend(TICL_RECO.outputCommands)
0035
0036 def customiseHGCalOnlyEventContent(process):
0037     def cleanOutputAndSet(outputModule, ticl_outputCommads):
0038         outputModule.outputCommands = ['drop *_*_*_*']
0039         outputModule.outputCommands.extend(ticl_outputCommads)
0040         outputModule.outputCommands.extend(['keep *_HGCalRecHit_*_*',
0041                                             'keep *_hgcalLayerClusters_*_*',
0042                                             'keep CaloParticles_mix_*_*',
0043                                             'keep SimClusters_mix_*_*',
0044                                             'keep recoTracks_generalTracks_*_*',
0045                                             'keep recoTrackExtras_generalTracks_*_*',
0046                                             'keep SimTracks_g4SimHits_*_*',
0047                                             'keep SimVertexs_g4SimHits_*_*',
0048                                             'keep *_layerClusterSimClusterAssociationProducer_*_*',
0049                                             'keep *_layerClusterCaloParticleAssociationProducer_*_*',
0050                                             'keep *_randomEngineStateProducer_*_*',
0051                                             ])
0052
0053     if hasattr(process, 'FEVTDEBUGEventContent'):
0054         cleanOutputAndSet(process.FEVTDEBUGEventContent, TICL_FEVT.outputCommands)
0055     if hasattr(process, 'FEVTDEBUGHLToutput'):
0056         cleanOutputAndSet(process.FEVTDEBUGHLToutput, TICL_FEVT.outputCommands)
0057
0058     return process