RecoParticleFlow/python/customize_pfanalysis.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 def customize_ecalclustering_caloparticle(process):
0004     process.load('SimGeneral.MixingModule.caloTruthProducer_cfi')
0005     process.caloParticles.simHitCollections = cms.PSet(
0006         #hcal = cms.VInputTag(cms.InputTag('g4SimHits','HcalHits')),
0007         ecal = cms.VInputTag(
0008             cms.InputTag('g4SimHits','EcalHitsEE'),
0009             cms.InputTag('g4SimHits','EcalHitsEB'),
0010             cms.InputTag('g4SimHits','EcalHitsES'),
0011         )
0012     )
0013     process.caloParticles.doHGCAL = False
0014     process.caloParticles.allowDifferentSimHitProcesses = False
0015     process.mix.digitizers.caloParticles = process.caloParticles
0016     process.mix.digitizers.mergedtruth.ignoreTracksOutsideVolume = False
0017     process.mix.digitizers.mergedtruth.allowDifferentSimHitProcesses = False
0018     process.mix.digitizers.mergedtruth.select.signalOnlyTP = True
0019
0020     process.PREMIXRAWoutput.outputCommands.append('keep *_*_MergedCaloTruth_*')
0021     return process
0022
0023 def customize_step2(process):
0024     process.load('SimGeneral.MixingModule.caloTruthProducer_cfi')
0025     process.caloParticles.simHitCollections = cms.PSet(
0026         hcal = cms.VInputTag(cms.InputTag('g4SimHits','HcalHits')),
0027         ecal = cms.VInputTag(
0028             cms.InputTag('g4SimHits','EcalHitsEE'),
0029             cms.InputTag('g4SimHits','EcalHitsEB'),
0030             cms.InputTag('g4SimHits','EcalHitsES'),
0031         )
0032     )
0033     process.caloParticles.doHGCAL = False
0034     process.caloParticles.allowDifferentSimHitProcesses = True
0035     process.mix.digitizers.caloParticles = process.caloParticles
0036     process.mix.digitizers.mergedtruth.ignoreTracksOutsideVolume = True
0037     process.mix.digitizers.mergedtruth.allowDifferentSimHitProcesses = True
0038     process.mix.digitizers.mergedtruth.select.signalOnlyTP = False
0039
0040     process.FEVTDEBUGHLToutput.outputCommands.append('keep *_simSiStripDigis_*_*')
0041     process.FEVTDEBUGHLToutput.outputCommands.append('keep *_simSiPixelDigis_*_*')
0042     process.FEVTDEBUGHLToutput.outputCommands.append('keep *_*_MergedCaloTruth_*')
0043     process.FEVTDEBUGHLToutput.outputCommands.append('keep *_*_MergedTrackTruth_*')
0044     return process
0045
0046 def customize_step3(process):
0047     process.FEVTDEBUGHLToutput.outputCommands.append('keep *_simSiStripDigis_*_*')
0048     process.FEVTDEBUGHLToutput.outputCommands.append('keep *_simSiPixelDigis_*_*')
0049     process.FEVTDEBUGHLToutput.outputCommands.append('keep *_*_MergedCaloTruth_*')
0050     process.FEVTDEBUGHLToutput.outputCommands.append('keep *_*_MergedTrackTruth_*')
0051     process.FEVTDEBUGHLToutput.outputCommands.append('keep recoPFRecTracks_*_*_*')
0052     process.FEVTDEBUGHLToutput.outputCommands.append('keep recoPFRecHits_*_*_*')
0053     process.FEVTDEBUGHLToutput.outputCommands.append('keep recoGsfPFRecTracks_*_*_*')
0054     process.FEVTDEBUGHLToutput.outputCommands.append('keep *_particleFlowBlock_*_*')
0055     process.FEVTDEBUGHLToutput.outputCommands.append('keep recoTracks_standAloneMuons_*_*')
0056     process.FEVTDEBUGHLToutput.outputCommands.append('keep recoTrackExtras_standAloneMuons_*_*')
0057     process.FEVTDEBUGHLToutput.outputCommands.append('keep recoMuons_*_*_*')
0058     process.FEVTDEBUGHLToutput.outputCommands.append('keep recoTracks_*_*_*')
0059     process.FEVTDEBUGHLToutput.outputCommands.append('keep recoGsfTracks_*_*_*')
0060     process.FEVTDEBUGHLToutput.outputCommands.append('keep recoPFBlocks_*_*_*')
0061
0062     return process