PFClusterProducer/python/pfClusterHBHEAlpaka_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 from Configuration.ProcessModifiers.alpaka_cff import alpaka
0004
0005 from RecoParticleFlow.PFRecHitProducer.hcalRecHitSoAProducer_cfi import hcalRecHitSoAProducer as _hcalRecHitSoAProducer
0006 from RecoParticleFlow.PFRecHitProducer.pfRecHitHCALParamsESProducer_cfi import pfRecHitHCALParamsESProducer as _pfRecHitHCALParamsESProducer
0007 from RecoParticleFlow.PFRecHitProducer.pfRecHitHCALTopologyESProducer_cfi import pfRecHitHCALTopologyESProducer as _pfRecHitHCALTopologyESProducer
0008 from RecoParticleFlow.PFRecHitProducer.pfRecHitSoAProducerHCAL_cfi import pfRecHitSoAProducerHCAL as _pfRecHitSoAProducerHCAL
0009 from RecoParticleFlow.PFRecHitProducer.legacyPFRecHitProducer_cfi import legacyPFRecHitProducer as _legacyPFRecHitProducer
0010 from RecoParticleFlow.PFClusterProducer.pfClusterParamsESProducer_cfi import pfClusterParamsESProducer as _pfClusterParamsESProducer
0011 from RecoParticleFlow.PFClusterProducer.pfClusterSoAProducer_cfi import pfClusterSoAProducer as _pfClusterSoAProducer
0012 from RecoParticleFlow.PFClusterProducer.legacyPFClusterProducer_cfi import legacyPFClusterProducer as _legacyPFClusterProducer
0013
0014 from RecoParticleFlow.PFClusterProducer.particleFlowCluster_cff import pfClusteringHBHEHFTask, pfClusteringHBHEHFOnlyTask, particleFlowClusterHBHE, particleFlowRecHitHBHE, particleFlowClusterHCAL, particleFlowClusterHBHEOnly, particleFlowRecHitHBHEOnly, particleFlowClusterHCALOnly
0015 #Full Reco
0016 _alpaka_pfClusteringHBHEHFTask = pfClusteringHBHEHFTask.copy()
0017 #HCAL Only
0018 _alpaka_pfClusteringHBHEHFOnlyTask = pfClusteringHBHEHFOnlyTask.copy()
0019
0020
0021 pfRecHitHCALParamsRecordSource = cms.ESSource('EmptyESSource',
0022             recordName = cms.string('PFRecHitHCALParamsRecord'),
0023             iovIsRunNotTime = cms.bool(True),
0024             firstValid = cms.vuint32(1)
0025     )
0026
0027 pfRecHitHCALTopologyRecordSource = cms.ESSource('EmptyESSource',
0028             recordName = cms.string('PFRecHitHCALTopologyRecord'),
0029             iovIsRunNotTime = cms.bool(True),
0030             firstValid = cms.vuint32(1)
0031     )
0032
0033 pfClusterParamsRecordSource = cms.ESSource('EmptyESSource',
0034             recordName = cms.string('JobConfigurationGPURecord'),
0035             iovIsRunNotTime = cms.bool(True),
0036             firstValid = cms.vuint32(1)
0037     )
0038
0039 hbheRecHitToSoA = _hcalRecHitSoAProducer.clone(
0040         src = "hbhereco"
0041     )
0042
0043 pfRecHitHCALParamsESProducer = _pfRecHitHCALParamsESProducer.clone(
0044         energyThresholdsHB = cms.vdouble( 0.1, 0.2, 0.3, 0.3 ),
0045         energyThresholdsHE = cms.vdouble( 0.1, 0.2, 0.2, 0.2, 0.2, 0.2, 0.2 )
0046     )
0047
0048 pfRecHitHCALTopologyESProducer = _pfRecHitHCALTopologyESProducer.clone()
0049 pfRecHitSoAProducerHCAL = _pfRecHitSoAProducerHCAL.clone(
0050         producers = cms.VPSet(
0051             cms.PSet(
0052                 src = cms.InputTag("hbheRecHitToSoA"),
0053                 params = cms.ESInputTag("pfRecHitHCALParamsESProducer:"),
0054             )
0055         ),
0056         topology = "pfRecHitHCALTopologyESProducer:",
0057         synchronise = cms.untracked.bool(False)
0058     )
0059
0060 legacyPFRecHitProducer = _legacyPFRecHitProducer.clone(
0061         src = "pfRecHitSoAProducerHCAL"
0062     )
0063
0064 pfClusterParamsESProducer = _pfClusterParamsESProducer.clone()
0065 pfClusterSoAProducer = _pfClusterSoAProducer.clone(
0066         pfRecHits = 'pfRecHitSoAProducerHCAL',
0067         topology = "pfRecHitHCALTopologyESProducer:",
0068         pfClusterParams = 'pfClusterParamsESProducer:',
0069         synchronise = cms.bool(False)
0070     )
0071
0072
0073 legacyPFClusterProducer = _legacyPFClusterProducer.clone(
0074         src = 'pfClusterSoAProducer',
0075         pfClusterBuilder = particleFlowClusterHBHE.pfClusterBuilder,
0076         recHitsSource = 'legacyPFRecHitProducer',
0077         PFRecHitsLabelIn = 'pfRecHitSoAProducerHCAL'
0078     )
0079
0080 #Full Reco
0081 _alpaka_pfClusteringHBHEHFTask.add(pfRecHitHCALParamsRecordSource)
0082 _alpaka_pfClusteringHBHEHFTask.add(pfRecHitHCALTopologyRecordSource)
0083 _alpaka_pfClusteringHBHEHFTask.add(pfClusterParamsRecordSource)
0084 _alpaka_pfClusteringHBHEHFTask.add(hbheRecHitToSoA)
0085 _alpaka_pfClusteringHBHEHFTask.add(pfRecHitHCALParamsESProducer)
0086 _alpaka_pfClusteringHBHEHFTask.add(pfRecHitHCALTopologyESProducer)
0087 _alpaka_pfClusteringHBHEHFTask.add(pfRecHitSoAProducerHCAL)
0088 _alpaka_pfClusteringHBHEHFTask.add(legacyPFRecHitProducer)
0089 _alpaka_pfClusteringHBHEHFTask.add(pfClusterParamsESProducer)
0090 _alpaka_pfClusteringHBHEHFTask.add(pfClusterSoAProducer)
0091 _alpaka_pfClusteringHBHEHFTask.add(legacyPFClusterProducer)
0092
0093 _alpaka_pfClusteringHBHEHFTask.remove(particleFlowRecHitHBHE)
0094 _alpaka_pfClusteringHBHEHFTask.remove(particleFlowClusterHBHE)
0095 _alpaka_pfClusteringHBHEHFTask.remove(particleFlowClusterHCAL)
0096 _alpaka_pfClusteringHBHEHFTask.add(particleFlowClusterHCAL)
0097
0098 alpaka.toModify(particleFlowClusterHCAL, clustersSource = "legacyPFClusterProducer")
0099
0100 alpaka.toReplaceWith(pfClusteringHBHEHFTask, _alpaka_pfClusteringHBHEHFTask)
0101
0102 #HCAL Only
0103
0104 from Configuration.Eras.Modifier_run3_HB_cff import run3_HB
0105
0106 hbheOnlyRecHitToSoA = _hcalRecHitSoAProducer.clone(
0107         src = "hbheprereco"
0108     )
0109 run3_HB.toModify(hbheOnlyRecHitToSoA,
0110         src = "hbhereco"
0111     )
0112
0113 pfRecHitSoAProducerHBHEOnly = _pfRecHitSoAProducerHCAL.clone(
0114         producers = cms.VPSet(
0115             cms.PSet(
0116                 src = cms.InputTag("hbheOnlyRecHitToSoA"),
0117                 params = cms.ESInputTag("pfRecHitHCALParamsESProducer:"),
0118             )
0119         ),
0120         topology = "pfRecHitHCALTopologyESProducer:",
0121         synchronise = cms.untracked.bool(False)
0122     )
0123
0124 legacyPFRecHitProducerHBHEOnly = _legacyPFRecHitProducer.clone(
0125         src = "pfRecHitSoAProducerHBHEOnly"
0126     )
0127
0128 pfClusterSoAProducerHBHEOnly = _pfClusterSoAProducer.clone(
0129         pfRecHits = 'pfRecHitSoAProducerHBHEOnly',
0130         topology = "pfRecHitHCALTopologyESProducer:",
0131         pfClusterParams = 'pfClusterParamsESProducer:',
0132         synchronise = cms.bool(False)
0133     )
0134
0135 legacyPFClusterProducerHBHEOnly = _legacyPFClusterProducer.clone(
0136         src = 'pfClusterSoAProducerHBHEOnly',
0137         pfClusterBuilder = particleFlowClusterHBHE.pfClusterBuilder,
0138         recHitsSource = 'legacyPFRecHitProducerHBHEOnly',
0139         PFRecHitsLabelIn = 'pfRecHitSoAProducerHBHEOnly'
0140     )
0141
0142 _alpaka_pfClusteringHBHEHFOnlyTask.add(pfRecHitHCALParamsRecordSource)
0143 _alpaka_pfClusteringHBHEHFOnlyTask.add(pfRecHitHCALTopologyRecordSource)
0144 _alpaka_pfClusteringHBHEHFOnlyTask.add(pfClusterParamsRecordSource)
0145 _alpaka_pfClusteringHBHEHFOnlyTask.add(hbheOnlyRecHitToSoA)
0146 _alpaka_pfClusteringHBHEHFOnlyTask.add(pfRecHitHCALParamsESProducer)
0147 _alpaka_pfClusteringHBHEHFOnlyTask.add(pfRecHitHCALTopologyESProducer)
0148 _alpaka_pfClusteringHBHEHFOnlyTask.add(pfRecHitSoAProducerHBHEOnly)
0149 _alpaka_pfClusteringHBHEHFOnlyTask.add(legacyPFRecHitProducerHBHEOnly)
0150 _alpaka_pfClusteringHBHEHFOnlyTask.add(pfClusterParamsESProducer)
0151 _alpaka_pfClusteringHBHEHFOnlyTask.add(pfClusterSoAProducerHBHEOnly)
0152 _alpaka_pfClusteringHBHEHFOnlyTask.add(legacyPFClusterProducerHBHEOnly)
0153
0154 _alpaka_pfClusteringHBHEHFOnlyTask.remove(particleFlowRecHitHBHEOnly)
0155 _alpaka_pfClusteringHBHEHFOnlyTask.remove(particleFlowClusterHBHEOnly)
0156 _alpaka_pfClusteringHBHEHFOnlyTask.remove(particleFlowClusterHCALOnly)
0157 _alpaka_pfClusteringHBHEHFOnlyTask.add(particleFlowClusterHCALOnly)
0158
0159 alpaka.toModify(particleFlowClusterHCALOnly, clustersSource = "legacyPFClusterProducerHBHEOnly")
0160
0161 alpaka.toReplaceWith(pfClusteringHBHEHFOnlyTask, _alpaka_pfClusteringHBHEHFOnlyTask)