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       [
0017        'keep *_ticlTrackstersHFNoseTrkEM_*_*',
0018        'keep *_ticlTrackstersHFNoseEM_*_*',
0019        'keep *_ticlTrackstersHFNoseTrk_*_*',
0020        'keep *_ticlTrackstersHFNoseMIP_*_*',
0021        'keep *_ticlTrackstersHFNoseHAD_*_*',
0022        'keep *_ticlTrackstersHFNoseMerge_*_*',] +
0023       ['keep *_pfTICL_*_*'] +
0024       ['keep CaloParticles_mix_*_*', 'keep SimClusters_mix_*_*'] +
0025       ['keep *_layerClusterSimClusterAssociationProducer_*_*','keep *_layerClusterCaloParticleAssociationProducer_*_*', 'keep *_layerClusterSimTracksterAssociationProducer_*_*'] +
0026       ['keep *_tracksterSimTracksterAssociationLinking_*_*' ,'keep *_tracksterSimTracksterAssociationPR_*_*'] +
0027       ['keep *_tracksterSimTracksterAssociationLinkingPU_*_*' ,'keep *_tracksterSimTracksterAssociationPRPU_*_*'] +
0028       ['keep *_tracksterSimTracksterAssociationLinkingbyCLUE3D_*_*', 'keep *_tracksterSimTracksterAssociationPRbyCLUE3D_*_*']
0029       )
0030     )
0031
0032 TICLv5_RECO = cms.PSet(
0033     outputCommands = cms.untracked.vstring(
0034         [
0035             'drop *_ticlTracksters*_*_*',
0036             'keep *_ticlTrackstersCLUE3DEM_*_*',
0037             'keep *_ticlTrackstersCLUE3DHAD_*_*',
0038             'keep *_ticlTracksterLinks_*_*',
0039             'keep *_ticlCandidate_*_*',
0040         ]
0041     )
0042 )
0043
0044
0045 TICL_RECO.outputCommands.extend(TICL_AOD.outputCommands)
0046
0047
0048 # FEVT Content
0049 TICL_FEVT = cms.PSet(
0050     outputCommands = cms.untracked.vstring(
0051       'keep *_ticlSimTracksters_*_*',
0052       'keep *_ticlSimTICLCandidates_*_*',
0053       'keep *_ticlSimTrackstersFromCP_*_*',
0054       'keep *_SimTau*_*_*'
0055       )
0056     )
0057 TICL_FEVT.outputCommands.extend(TICL_RECO.outputCommands)
0058 TICLv5_FEVT = cms.PSet(
0059     outputCommands = cms.untracked.vstring(
0060       'keep *_ticlSimTracksters_*_*',
0061       'keep *_ticlSimTICLCandidates_*_*',
0062       'keep *_ticlSimTrackstersFromCP_*_*',
0063       )
0064     )
0065
0066 TICLv5_FEVT.outputCommands.extend(TICLv5_RECO.outputCommands)
0067
0068
0069
0070
0071
0072 def customiseHGCalOnlyEventContent(process):
0073     def cleanOutputAndSet(outputModule, ticl_outputCommads):
0074         outputModule.outputCommands = ['drop *_*_*_*']
0075         outputModule.outputCommands.extend(ticl_outputCommads)
0076         outputModule.outputCommands.extend(['keep *_HGCalRecHit_*_*',
0077                                             'keep *_hgcalMergeLayerClusters_*_*',
0078                                             'keep CaloParticles_mix_*_*',
0079                                             'keep SimClusters_mix_*_*',
0080                                             'keep recoTracks_generalTracks_*_*',
0081                                             'keep recoTrackExtras_generalTracks_*_*',
0082                                             'keep SimTracks_g4SimHits_*_*',
0083                                             'keep SimVertexs_g4SimHits_*_*',
0084                                             'keep *_layerClusterSimClusterAssociationProducer_*_*',
0085                                             'keep *_layerClusterCaloParticleAssociationProducer_*_*',
0086                                             'keep *_randomEngineStateProducer_*_*',
0087                                             'keep *_layerClusterSimTracksterAssociationProducer_*_*',
0088                                             'keep *_tracksterSimTracksterAssociationLinking_*_*',
0089                                             'keep *_tracksterSimTracksterAssociationPR_*_*',
0090                                             'keep *_tracksterSimTracksterAssociationLinkingPU_*_*',
0091                                             'keep *_tracksterSimTracksterAssociationPRPU_*_*',
0092                                             'keep *_tracksterSimTracksterAssociationLinkingbyCLUE3D_*_*',
0093                                             'keep *_tracksterSimTracksterAssociationPRbyCLUE3D_*_*',
0094                                             ])
0095
0096     if hasattr(process, 'FEVTDEBUGEventContent'):
0097         cleanOutputAndSet(process.FEVTDEBUGEventContent, TICL_FEVT.outputCommands)
0098     if hasattr(process, 'FEVTDEBUGHLToutput'):
0099         cleanOutputAndSet(process.FEVTDEBUGHLToutput, TICL_FEVT.outputCommands)
0100
0101     return process
0102
0103
0104
0105 def customiseForTICLv5EventContent(process):
0106     def cleanOutputAndSet(outputModule, ticl_outputCommands):
0107         outputModule.outputCommands.extend(ticl_outputCommands)
0108
0109     if hasattr(process, 'FEVTDEBUGEventContent'):
0110         cleanOutputAndSet(process.FEVTDEBUGEventContent, TICLv5_FEVT.outputCommands)
0111     if hasattr(process, 'FEVTDEBUGHLToutput'):
0112         cleanOutputAndSet(process.FEVTDEBUGHLToutput, TICLv5_FEVT.outputCommands)
0113     if hasattr(process, 'FEVTEventContent'):
0114         cleanOutputAndSet(process.FEVTEventContent, TICLv5_FEVT.outputCommands)
0115
0116     return process
0117