Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2024-04-06 12:25:11

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       ['keep CaloParticles_mix_*_*', 'keep SimClusters_mix_*_*'] +
0024       ['keep *_layerClusterSimClusterAssociationProducer_*_*','keep *_layerClusterCaloParticleAssociationProducer_*_*', 'keep *_layerClusterSimTracksterAssociationProducer_*_*'] + 
0025       ['keep *_tracksterSimTracksterAssociationLinking_*_*' ,'keep *_tracksterSimTracksterAssociationPR_*_*'] +
0026       ['keep *_tracksterSimTracksterAssociationLinkingPU_*_*' ,'keep *_tracksterSimTracksterAssociationPRPU_*_*'] +
0027       ['keep *_tracksterSimTracksterAssociationLinkingbyCLUE3D_*_*', 'keep *_tracksterSimTracksterAssociationPRbyCLUE3D_*_*'] 
0028       )
0029     )
0030 TICL_RECO.outputCommands.extend(TICL_AOD.outputCommands)
0031 
0032 # FEVT Content
0033 TICL_FEVT = cms.PSet(
0034     outputCommands = cms.untracked.vstring(
0035       'keep *_ticlSimTracksters_*_*',
0036       'keep *_ticlSimTICLCandidates_*_*',
0037       'keep *_ticlSimTrackstersFromCP_*_*',
0038       'keep *_SimTau*_*_*'
0039       )
0040     )
0041 TICL_FEVT.outputCommands.extend(TICL_RECO.outputCommands)
0042 
0043 def customiseHGCalOnlyEventContent(process):
0044     def cleanOutputAndSet(outputModule, ticl_outputCommads):
0045         outputModule.outputCommands = ['drop *_*_*_*']
0046         outputModule.outputCommands.extend(ticl_outputCommads)
0047         outputModule.outputCommands.extend(['keep *_HGCalRecHit_*_*',
0048                                             'keep *_hgcalMergeLayerClusters_*_*',
0049                                             'keep CaloParticles_mix_*_*',
0050                                             'keep SimClusters_mix_*_*',
0051                                             'keep recoTracks_generalTracks_*_*',
0052                                             'keep recoTrackExtras_generalTracks_*_*',
0053                                             'keep SimTracks_g4SimHits_*_*',
0054                                             'keep SimVertexs_g4SimHits_*_*',
0055                                             'keep *_layerClusterSimClusterAssociationProducer_*_*',
0056                                             'keep *_layerClusterCaloParticleAssociationProducer_*_*',
0057                                             'keep *_randomEngineStateProducer_*_*',
0058                                             'keep *_layerClusterSimTracksterAssociationProducer_*_*',
0059                                             'keep *_tracksterSimTracksterAssociationLinking_*_*',
0060                                             'keep *_tracksterSimTracksterAssociationPR_*_*', 
0061                                             'keep *_tracksterSimTracksterAssociationLinkingPU_*_*',
0062                                             'keep *_tracksterSimTracksterAssociationPRPU_*_*', 
0063                                             'keep *_tracksterSimTracksterAssociationLinkingbyCLUE3D_*_*',
0064                                             'keep *_tracksterSimTracksterAssociationPRbyCLUE3D_*_*', 
0065                                             ])
0066 
0067     if hasattr(process, 'FEVTDEBUGEventContent'):
0068         cleanOutputAndSet(process.FEVTDEBUGEventContent, TICL_FEVT.outputCommands)
0069     if hasattr(process, 'FEVTDEBUGHLToutput'):
0070         cleanOutputAndSet(process.FEVTDEBUGHLToutput, TICL_FEVT.outputCommands)
0071 
0072     return process