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 import digiTask
0010 from DQM.HcalTasks.RawTask import rawTask
0011 from DQM.HcalTasks.TPTask import tpTask
0012 from DQM.HcalTasks.RecHitTask 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 #   set the label for Emulator TP Task
0024 tpTask.tagEmul = "valHcalTriggerPrimitiveDigis"
0025
0026 hcalOfflineSourceSequence = cms.Sequence(
0027     digiTask +
0028     tpTask +
0029     recHitTask +
0030     rawTask )
0031
0032 hcalOnlyOfflineSourceSequence = cms.Sequence(
0033     digiTask +
0034     recHitPreRecoTask +
0035     rawTask )
0036
0037 hcalOnlyOfflineSourceSequenceGPU = cms.Sequence(
0038     digiTask +
0039     recHitTask +
0040     rawTask +
0041     hcalGPUComparisonTask
0042 )
0043
0044 from Configuration.ProcessModifiers.gpuValidationHcal_cff import gpuValidationHcal
0045 gpuValidationHcal.toReplaceWith(hcalOnlyOfflineSourceSequence, hcalOnlyOfflineSourceSequenceGPU)
0046
0047 from Configuration.Eras.Modifier_run2_HCAL_2018_cff import run2_HCAL_2018
0048 run2_HCAL_2018.toModify(hcalGPUComparisonTask,
0049     tagHBHE_ref = "hbheprereco@cpu",
0050     tagHBHE_target = "hbheprereco@cuda"
0051 )
0052 run2_HCAL_2018.toModify(recHitTask,
0053     tagHBHE = "hbheprereco"
0054 )
0055
0056 from Configuration.Eras.Modifier_run3_HB_cff import run3_HB
0057 ### reverting the reco tag setting that inherited from run2
0058 run3_HB.toModify(hcalGPUComparisonTask,
0059     tagHBHE_ref = "hbhereco@cpu",
0060     tagHBHE_target = "hbhereco@cuda"
0061 )
0062 run3_HB.toModify(recHitTask,
0063     tagHBHE = "hbhereco"
0064 )
0065 _phase1_hcalOnlyOfflineSourceSequence = hcalOnlyOfflineSourceSequence.copy()
0066 _phase1_hcalOnlyOfflineSourceSequence.replace(recHitPreRecoTask, recHitTask)
0067 run3_HB.toReplaceWith(hcalOnlyOfflineSourceSequence, _phase1_hcalOnlyOfflineSourceSequence)
0068
0069 from Configuration.Eras.Modifier_phase2_hcal_cff import phase2_hcal
0070 _phase2_hcalOfflineSourceSequence = hcalOfflineSourceSequence.copyAndExclude([tpTask,rawTask])
0071 phase2_hcal.toReplaceWith(hcalOfflineSourceSequence, _phase2_hcalOfflineSourceSequence)
0072 phase2_hcal.toModify(digiTask,
0073     tagHBHE = "simHcalDigis:HBHEQIE11DigiCollection",
0074     tagHO = "simHcalDigis",
0075     tagHF = "simHcalDigis:HFQIE10DigiCollection"
0076 )
0077
0078 from Configuration.ProcessModifiers.premix_stage2_cff import premix_stage2
0079 (premix_stage2 & phase2_hcal).toModify(digiTask,
0080     tagHBHE = "DMHcalDigis:HBHEQIE11DigiCollection",
0081     tagHO = "DMHcalDigis",
0082     tagHF = "DMHcalDigis:HFQIE10DigiCollection"
0083 )