HcalTasks/python/OfflineSourceSequence_pp.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 #-----------------
0004 #   HCAL DQM Offline Source Sequence Definition for pp
0005 #   To be used for Offline DQM importing
0006 #-----------------
0007
0008 #   import the tasks
0009 from DQM.HcalTasks.DigiTask_cfi import digiTask
0010 from DQM.HcalTasks.RawTask_cfi import rawTask
0011 from DQM.HcalTasks.TPTask_cfi import tpTask
0012 from DQM.HcalTasks.RecHitTask_cfi import recHitTask, recHitPreRecoTask
0013 from DQM.HcalTasks.hcalGPUComparisonTask_cfi import hcalGPUComparisonTask
0014
0015 #   set processing type to Offine
0016 digiTask.ptype = 1
0017 tpTask.ptype = 1
0018 recHitTask.ptype = 1
0019 rawTask.ptype = 1
0020 recHitPreRecoTask.ptype = 1
0021 hcalGPUComparisonTask.ptype = 1
0022
0023 hcalAlpakaComparisonTask = hcalGPUComparisonTask.clone()
0024
0025 #   set the label for Emulator TP Task
0026 tpTask.tagEmul = "valHcalTriggerPrimitiveDigis"
0027
0028 hcalOfflineSourceSequence = cms.Sequence(
0029     digiTask +
0030     tpTask +
0031     recHitTask +
0032     rawTask )
0033
0034 hcalOnlyOfflineSourceSequence = cms.Sequence(
0035     digiTask +
0036     recHitPreRecoTask +
0037     rawTask )
0038
0039 hcalOnlyOfflineSourceSequenceGPU = cms.Sequence(
0040     digiTask +
0041     recHitTask +
0042     rawTask +
0043     hcalGPUComparisonTask
0044 )
0045 hcalOnlyOfflineSourceSequenceAlpaka = cms.Sequence(
0046     digiTask +
0047     recHitTask +
0048     rawTask +
0049     hcalAlpakaComparisonTask
0050 )
0051
0052
0053 from Configuration.ProcessModifiers.gpuValidationHcal_cff import gpuValidationHcal
0054 gpuValidationHcal.toReplaceWith(hcalOnlyOfflineSourceSequence, hcalOnlyOfflineSourceSequenceGPU)
0055
0056 from Configuration.Eras.Modifier_run2_HCAL_2018_cff import run2_HCAL_2018
0057 run2_HCAL_2018.toModify(hcalGPUComparisonTask,
0058     tagHBHE_ref = "hbheprereco@cpu",
0059     tagHBHE_target = "hbheprereco@cuda"
0060 )
0061 run2_HCAL_2018.toModify(recHitTask,
0062     tagHBHE = "hbheprereco"
0063 )
0064
0065 from Configuration.Eras.Modifier_run3_HB_cff import run3_HB
0066 from Configuration.ProcessModifiers.alpaka_cff import alpaka
0067 ### reverting the reco tag setting that inherited from run2
0068 run3_HB.toModify(hcalGPUComparisonTask,
0069     tagHBHE_ref = "hbhereco@cpu",
0070     tagHBHE_target = "hbhereco@cuda"
0071 )
0072 run3_HB.toModify(recHitTask,
0073     tagHBHE = "hbhereco"
0074 )
0075 (alpaka & run3_HB).toModify(hcalGPUComparisonTask,
0076     tagHBHE_ref = "hbherecoSerial",
0077     tagHBHE_target = "hbhereco"
0078 )
0079 run3_HB.toModify(hcalAlpakaComparisonTask,
0080     tagHBHE_ref = "hbherecoLegacy",
0081     tagHBHE_target = "hbhereco"
0082 )
0083
0084 _phase1_hcalOnlyOfflineSourceSequence = hcalOnlyOfflineSourceSequence.copy()
0085 _phase1_hcalOnlyOfflineSourceSequence.replace(recHitPreRecoTask, recHitTask)
0086 run3_HB.toReplaceWith(hcalOnlyOfflineSourceSequence, _phase1_hcalOnlyOfflineSourceSequence)
0087
0088 from Configuration.Eras.Modifier_phase2_hcal_cff import phase2_hcal
0089 _phase2_hcalOfflineSourceSequence = hcalOfflineSourceSequence.copyAndExclude([tpTask,rawTask])
0090 phase2_hcal.toReplaceWith(hcalOfflineSourceSequence, _phase2_hcalOfflineSourceSequence)
0091 phase2_hcal.toModify(digiTask,
0092     tagHBHE = "simHcalDigis:HBHEQIE11DigiCollection",
0093     tagHO = "simHcalDigis",
0094     tagHF = "simHcalDigis:HFQIE10DigiCollection"
0095 )
0096
0097 from Configuration.ProcessModifiers.premix_stage2_cff import premix_stage2
0098 (premix_stage2 & phase2_hcal).toModify(digiTask,
0099     tagHBHE = "DMHcalDigis:HBHEQIE11DigiCollection",
0100     tagHO = "DMHcalDigis",
0101     tagHF = "DMHcalDigis:HFQIE10DigiCollection"
0102 )