HcalRecProducers/python/hbheRecHitProducerPortableTask_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 # Run 3 HCAL workflow on GPU
0004
0005 # EventSetup modules used by HBHERecHitProducerPortable
0006
0007 from RecoLocalCalo.HcalRecProducers.hcalMahiConditionsESProducer_cfi import hcalMahiConditionsESProducer
0008 from RecoLocalCalo.HcalRecProducers.hcalMahiPulseOffsetsESProducer_cfi import hcalMahiPulseOffsetsESProducer
0009 from RecoLocalCalo.HcalRecProducers.hcalSiPMCharacteristicsESProducer_cfi import hcalSiPMCharacteristicsESProducer
0010 from RecoLocalCalo.HcalRecAlgos.hcalRecoParamWithPulseShapeESProducer_cfi import hcalRecoParamWithPulseShapeESProducer
0011
0012 hcalMahiPulseOffSetAlpakaESRcdSource = cms.ESSource('EmptyESSource',
0013     recordName = cms.string('JobConfigurationGPURecord'),
0014     iovIsRunNotTime = cms.bool(True),
0015     firstValid = cms.vuint32(1)
0016 )
0017
0018 # convert the HBHE digis into SoA format
0019 from EventFilter.HcalRawToDigi.hcalDigisSoAProducer_cfi import hcalDigisSoAProducer as _hcalDigisSoAProducer
0020 hcalDigisPortable = _hcalDigisSoAProducer.clone(
0021     digisLabelF01HE = "f01HEDigis",
0022     digisLabelF5HB  = "f5HBDigis",
0023     digisLabelF3HB  = "f3HBDigis"
0024 )
0025
0026 from HeterogeneousCore.AlpakaCore.functions import *
0027 hcalDigisSerial = makeSerialClone(hcalDigisPortable)
0028
0029 # run the HCAL local reconstruction (MAHI) on GPU
0030 from RecoLocalCalo.HcalRecProducers.hbheRecHitProducerPortable_cfi import hbheRecHitProducerPortable as _hbheRecHitProducerPortable
0031 hbheRecHitProducerPortable = _hbheRecHitProducerPortable.clone(
0032     digisLabelF01HE = ("hcalDigisPortable", "f01HEDigis"),
0033     digisLabelF5HB = ("hcalDigisPortable", "f5HBDigis"),
0034     digisLabelF3HB = ("hcalDigisPortable","f3HBDigis"),
0035     recHitsLabelM0HBHE = "",
0036     mahiPulseOffSets = "hcalMahiPulseOffsetsESProducer:"
0037 )
0038 hbheRecHitProducerSerial = makeSerialClone(hbheRecHitProducerPortable,
0039     digisLabelF01HE = ("hcalDigisSerial","f01HEDigis"),
0040     digisLabelF5HB = ("hcalDigisSerial","f5HBDigis"),
0041     digisLabelF3HB = ("hcalDigisSerial","f3HBDigis")
0042 )
0043
0044 # Tasks and Sequences
0045 hbheRecHitProducerPortableTask = cms.Task(
0046     hcalMahiConditionsESProducer,
0047     hcalMahiPulseOffSetAlpakaESRcdSource,
0048     hcalMahiPulseOffsetsESProducer,
0049     hcalRecoParamWithPulseShapeESProducer,
0050     hcalSiPMCharacteristicsESProducer,
0051     hcalDigisPortable,
0052     hcalDigisSerial,
0053     hbheRecHitProducerPortable,
0054     hbheRecHitProducerSerial
0055 )
0056
0057 hbheRecHitProducerPortableSequence = cms.Sequence(hbheRecHitProducerPortableTask)