PFClusterProducer/python/particleFlowCluster_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003
0004 #from RecoParticleFlow.PFClusterProducer.towerMakerPF_cfi import *
0005 #from RecoParticleFlow.PFClusterProducer.particleFlowCaloResolution_cfi import _timeResolutionHCALMaxSample
0006
0007 from RecoParticleFlow.PFClusterProducer.particleFlowRecHitECAL_cfi import *
0008 from RecoParticleFlow.PFClusterProducer.particleFlowRecHitHBHE_cfi import *
0009 from RecoParticleFlow.PFClusterProducer.particleFlowRecHitHF_cfi import *
0010 from RecoParticleFlow.PFClusterProducer.particleFlowRecHitHO_cfi import *
0011 from RecoParticleFlow.PFClusterProducer.particleFlowRecHitPS_cfi import *
0012
0013 from RecoParticleFlow.PFClusterProducer.particleFlowClusterECALUncorrected_cfi import *
0014 from RecoParticleFlow.PFClusterProducer.particleFlowClusterECAL_cff import *
0015
0016 from RecoParticleFlow.PFClusterProducer.particleFlowClusterHBHE_cfi import *
0017 from RecoParticleFlow.PFClusterProducer.particleFlowClusterHF_cfi import *
0018 from RecoParticleFlow.PFClusterProducer.particleFlowClusterHCAL_cfi import *
0019 from RecoParticleFlow.PFClusterProducer.particleFlowClusterHO_cfi import *
0020 from RecoParticleFlow.PFClusterProducer.particleFlowClusterPS_cfi import *
0021 from RecoParticleFlow.PFClusterProducer.particleFlowBadHcalPseudoCluster_cff import *
0022
0023 particleFlowClusterECALTask = cms.Task(particleFlowClusterECAL)
0024 particleFlowClusterECALSequence = cms.Sequence(particleFlowClusterECALTask)
0025
0026 pfClusteringECALTask = cms.Task(particleFlowRecHitECAL,
0027                                 particleFlowClusterECALUncorrected,
0028                                 particleFlowClusterECALTask)
0029 pfClusteringECAL = cms.Sequence(pfClusteringECALTask)
0030
0031 pfClusteringPSTask = cms.Task(particleFlowRecHitPS,particleFlowClusterPS)
0032 pfClusteringPS = cms.Sequence(pfClusteringPSTask)
0033
0034 pfClusteringHBHEHFTask = cms.Task(particleFlowRecHitHBHE,
0035                                   particleFlowRecHitHF,
0036                                   particleFlowClusterHBHE,
0037                                   particleFlowClusterHF,
0038                                   particleFlowClusterHCAL)
0039 pfClusteringHBHEHF = cms.Sequence(pfClusteringHBHEHFTask)
0040
0041 pfClusteringHBHEHFOnlyTask = cms.Task(particleFlowRecHitHBHEOnly,
0042                                       particleFlowRecHitHF,
0043                                       particleFlowClusterHBHEOnly,
0044                                       particleFlowClusterHF,
0045                                       particleFlowClusterHCALOnly)
0046
0047 #--- Legacy HCAL Only Task
0048 pfClusteringHBHEHFOnlyLegacyTask = pfClusteringHBHEHFOnlyTask.copy()
0049
0050 pfClusteringHOTask = cms.Task(particleFlowRecHitHO,particleFlowClusterHO)
0051 pfClusteringHO = cms.Sequence(pfClusteringHOTask)
0052
0053 particleFlowClusterWithoutHOTask = cms.Task(particleFlowBadHcalPseudoCluster,
0054                                             pfClusteringPSTask,
0055                                             pfClusteringECALTask,
0056                                             pfClusteringHBHEHFTask)
0057 particleFlowClusterWithoutHO = cms.Sequence(particleFlowClusterWithoutHOTask)
0058
0059 particleFlowClusterTask = cms.Task(particleFlowBadHcalPseudoCluster,
0060                                    pfClusteringPSTask,
0061                                    pfClusteringECALTask,
0062                                    pfClusteringHBHEHFTask,
0063                                    pfClusteringHOTask)
0064 particleFlowCluster = cms.Sequence(particleFlowClusterTask)
0065
0066 #HGCal
0067
0068 from RecoParticleFlow.PFClusterProducer.particleFlowRecHitHGC_cfi import particleFlowRecHitHGC
0069 pfClusteringHGCalTask = cms.Task(particleFlowRecHitHGC)
0070 pfClusteringHGCal = cms.Sequence(pfClusteringHGCalTask)
0071
0072 _phase2_hgcal_particleFlowClusterTask = particleFlowClusterTask.copy()
0073 _phase2_hgcal_particleFlowClusterTask.add(pfClusteringHGCalTask)
0074
0075 from Configuration.Eras.Modifier_phase2_hgcal_cff import phase2_hgcal
0076 phase2_hgcal.toReplaceWith( particleFlowClusterTask, _phase2_hgcal_particleFlowClusterTask )
0077
0078 #timing
0079
0080 from RecoParticleFlow.PFClusterProducer.particleFlowClusterTimeAssigner_cfi import particleFlowTimeAssignerECAL
0081 from RecoParticleFlow.PFSimProducer.ecalBarrelClusterFastTimer_cfi import ecalBarrelClusterFastTimer
0082 _phase2_timing_particleFlowClusterECALTask = particleFlowClusterECALTask.copy()
0083 _phase2_timing_particleFlowClusterECALTask.add(cms.Task(ecalBarrelClusterFastTimer,
0084                                                         particleFlowTimeAssignerECAL))
0085
0086 from Configuration.Eras.Modifier_phase2_timing_cff import phase2_timing
0087 phase2_timing.toReplaceWith(particleFlowClusterECALTask,
0088                                   _phase2_timing_particleFlowClusterECALTask)
0089 phase2_timing.toModify(particleFlowClusterECAL,
0090                             inputECAL = 'particleFlowTimeAssignerECAL')
0091
0092 # Replace HBHE rechit and clustering with Alpaka modules
0093
0094 from Configuration.ProcessModifiers.alpaka_cff import alpaka
0095
0096 def _addProcessPFClusterAlpaka(process):
0097     process.load("RecoParticleFlow.PFClusterProducer.pfClusterHBHEAlpaka_cff")
0098
0099 modifyConfigurationPFClusterAlpaka_ = alpaka.makeProcessModifier(_addProcessPFClusterAlpaka)