Configuration/python/SimCalorimetry_EventContent_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 # This object modifies the event content for different scenarios
0004
0005 SimCalorimetryFEVTDEBUG = cms.PSet(
0006     outputCommands = cms.untracked.vstring('keep *_simEcalDigis_*_*',
0007         'keep *_simEcalPreshowerDigis_*_*',
0008         'keep *_simEcalTriggerPrimitiveDigis_*_*',
0009         'keep *_simEcalEBTriggerPrimitiveDigis_*_*',
0010         'keep *_simHcalDigis_*_*',
0011         'keep ZDCDataFramesSorted_simHcalUnsuppressedDigis_*_*',
0012         'drop ZDCDataFramesSorted_mix_simHcalUnsuppressedDigis*_*',
0013         'keep *_simHcalTriggerPrimitiveDigis_*_*',
0014         'keep *_mix_HcalSamples_*',
0015         'keep *_mixData_HcalSamples_*',
0016         'keep *_mix_HcalHits_*',
0017         'keep *_mixData_HcalHits_*',
0018         'keep *_DMHcalTriggerPrimitiveDigis_*_*',
0019     )
0020 )
0021 SimCalorimetryRAW = cms.PSet(
0022     outputCommands = cms.untracked.vstring('keep EBSrFlagsSorted_simEcalDigis_*_*',
0023         'keep EESrFlagsSorted_simEcalDigis_*_*')
0024 )
0025 SimCalorimetryRECO = cms.PSet(
0026     outputCommands = cms.untracked.vstring()
0027 )
0028 SimCalorimetryAOD = cms.PSet(
0029     outputCommands = cms.untracked.vstring()
0030 )
0031 SimCalorimetryPREMIX = cms.PSet(
0032     outputCommands = cms.untracked.vstring(
0033         'keep EBDigiCollection_simEcalDigis_*_*',
0034         'keep EEDigiCollection_simEcalDigis_*_*',
0035         'keep ESDigiCollection_simEcalUnsuppressedDigis_*_*',
0036         'keep *_simHcalDigis_*_*',
0037     )
0038 )
0039
0040 #
0041 # Add extra event content if running in Run 2
0042 #
0043 from Configuration.Eras.Modifier_run2_common_cff import run2_common
0044 run2_common.toModify( SimCalorimetryFEVTDEBUG.outputCommands, func=lambda outputCommands: outputCommands.append('keep *_simHcalUnsuppressedDigis_*_*') )
0045 run2_common.toModify( SimCalorimetryRAW.outputCommands, func=lambda outputCommands: outputCommands.append('keep *_simHcalUnsuppressedDigis_*_*') )
0046
0047 from Configuration.Eras.Modifier_phase2_ecal_devel_cff import phase2_ecal_devel
0048 phase2_ecal_devel.toModify(SimCalorimetryFEVTDEBUG.outputCommands, func=lambda outputCommands: outputCommands.append('keep *_ecal*_*_*') )
0049
0050 from Configuration.Eras.Modifier_phase2_hcal_cff import phase2_hcal
0051 phase2_hcal.toModify(SimCalorimetryFEVTDEBUG.outputCommands, func=lambda outputCommands: outputCommands.append('keep *_DMHcalDigis_*_*') )
0052
0053 from Configuration.Eras.Modifier_phase2_timing_cff import phase2_timing
0054 phase2_timing.toModify(SimCalorimetryFEVTDEBUG.outputCommands, func=lambda outputCommands: outputCommands.append('keep *_mix_EETimeDigi_*') )
0055 phase2_timing.toModify(SimCalorimetryFEVTDEBUG.outputCommands, func=lambda outputCommands: outputCommands.append('keep *_mix_EBTimeDigi_*') )
0056
0057 phase2_timing.toModify(SimCalorimetryRAW.outputCommands, func=lambda outputCommands: outputCommands.append('keep *_mix_EETimeDigi_*') )
0058 phase2_timing.toModify(SimCalorimetryRAW.outputCommands, func=lambda outputCommands: outputCommands.append('keep *_mix_EBTimeDigi_*') )
0059
0060 from Configuration.Eras.Modifier_phase2_common_cff import phase2_common
0061 phase2_common.toModify( SimCalorimetryFEVTDEBUG.outputCommands, func=lambda outputCommands: outputCommands.append('keep *_simEcalUnsuppressedDigis_*_*') )
0062 phase2_common.toModify( SimCalorimetryRAW.outputCommands, func=lambda outputCommands: outputCommands.append('keep *_simEcalUnsuppressedDigis_*_*') )
0063 phase2_common.toModify( SimCalorimetryPREMIX.outputCommands, func=lambda outputCommands: outputCommands.append('drop ESDigiCollection_simEcalUnsuppressedDigis_*_*') )
0064
0065 from Configuration.Eras.Modifier_phase2_hgcal_cff import phase2_hgcal
0066 phase2_hgcal.toModify( SimCalorimetryPREMIX.outputCommands, func=lambda outputCommands: outputCommands.append('drop EEDigiCollection_simEcalDigis_*_*') )
0067
0068
0069 # mods for HGCAL
0070 _phase2_hgc_extraCommands = cms.PSet( # using PSet in order to customize with Modifier
0071     v = cms.vstring('keep *_simHGCalUnsuppressedDigis_EE_*', 'keep *_simHGCalUnsuppressedDigis_HEfront_*', 'keep *_simHGCalUnsuppressedDigis_HEback_*', 'keep *_mix_MergedCaloTruth_*'),
0072 )
0073 # For phase2 premixing switch the sim digi collections to the ones including pileup
0074 from Configuration.ProcessModifiers.premix_stage2_cff import premix_stage2
0075 premix_stage2.toModify(_phase2_hgc_extraCommands,
0076     v = ['keep *_mixData_HGCDigisEE_*', 'keep *_mixData_HGCDigisHEfront_*', 'keep *_mixData_HGCDigisHEback_*', 'keep *_mixData_MergedCaloTruth_*']
0077 )
0078 phase2_hgcal.toModify( SimCalorimetryRAW, outputCommands = SimCalorimetryRAW.outputCommands + _phase2_hgc_extraCommands.v )
0079 phase2_hgcal.toModify( SimCalorimetryFEVTDEBUG, outputCommands = SimCalorimetryFEVTDEBUG.outputCommands + _phase2_hgc_extraCommands.v )
0080 phase2_hgcal.toModify( SimCalorimetryRECO, outputCommands = SimCalorimetryRECO.outputCommands + _phase2_hgc_extraCommands.v )
0081 phase2_hgcal.toModify( SimCalorimetryPREMIX, outputCommands = SimCalorimetryPREMIX.outputCommands + _phase2_hgc_extraCommands.v )
0082
0083 from Configuration.Eras.Modifier_phase2_hfnose_cff import phase2_hfnose
0084 phase2_hfnose.toModify( SimCalorimetryFEVTDEBUG.outputCommands, func=lambda outputCommands: outputCommands.append('keep *_simHFNoseUnsuppressedDigis_*_*') )
0085 phase2_hfnose.toModify( SimCalorimetryRAW.outputCommands, func=lambda outputCommands: outputCommands.append('keep *_simHFNoseUnsuppressedDigis_*_*') )