DQM/EcalMonitorTasks/python/EcalMonitorTask_cff.py - master - githubmirror/cmssw

DQM/EcalMonitorTasks/python/EcalMonitorTask_cff.py
51 строка3 KB
Thomas Reis
Do not run on endcap collections for ECAL Phase 2 DQM tasks.
21 июл 2026, 12:21
21 июл 2026, 12:210021906
import FWCore.ParameterSet.Config as cms

from DQM.EcalMonitorTasks.EcalMonitorTask_cfi import *

# Customization to run the CPU vs GPU comparison task if the job runs on a GPU enabled machine
from Configuration.ProcessModifiers.gpuValidationEcal_cff import gpuValidationEcal
from Configuration.ProcessModifiers.alpakaValidationEcal_cff import alpakaValidationEcal
from DQM.EcalMonitorTasks.ecalGpuTask_cfi import ecalGpuTask

gpuValidationEcal.toModify(ecalGpuTask.params, runGpuTask = True)
gpuValidationEcal.toModify(ecalMonitorTask.workers, func = lambda workers: workers.append("GpuTask"))
gpuValidationEcal.toModify(ecalMonitorTask, workerParameters = dict(GpuTask = ecalGpuTask))

alpakaValidationEcal.toModify(ecalGpuTask.params, runGpuTask = True)
alpakaValidationEcal.toModify(ecalMonitorTask.workers, func = lambda workers: workers.append("GpuTask"))
alpakaValidationEcal.toModify(ecalMonitorTask, workerParameters = dict(GpuTask = ecalGpuTask))
alpakaValidationEcal.toModify(ecalDQMCollectionTags, EBCpuDigi = "ecalDigisSerialSync:ebDigis")
alpakaValidationEcal.toModify(ecalDQMCollectionTags, EECpuDigi = "ecalDigisSerialSync:eeDigis")
alpakaValidationEcal.toModify(ecalDQMCollectionTags, EBCpuUncalibRecHit = "ecalMultiFitUncalibRecHitSerialSync:EcalUncalibRecHitsEB")
alpakaValidationEcal.toModify(ecalDQMCollectionTags, EECpuUncalibRecHit = "ecalMultiFitUncalibRecHitSerialSync:EcalUncalibRecHitsEE")

# Skip consuming and running over the EcalRawData collection for all GPU WFs
# This is to be used as long as the GPU unpacker unpacks a dummy EcalRawData collection
from Configuration.ProcessModifiers.gpu_cff import gpu
gpu.toModify(ecalMonitorTask.skipCollections, func = lambda skipCollections: skipCollections.append("EcalRawData"))

from Configuration.ProcessModifiers.alpaka_cff import alpaka
alpaka.toModify(ecalMonitorTask.skipCollections, func = lambda skipCollections: skipCollections.append("EcalRawData"))

# Changes for Phase 2
from DQM.EcalMonitorTasks.CollectionTags_cfi import ecalDQMCollectionTagsPhase2
from DQM.EcalMonitorTasks.ClusterTask_cfi import ecalClusterTaskPhase2
from DQM.EcalMonitorTasks.EnergyTask_cfi import ecalEnergyTaskPhase2
from DQM.EcalMonitorTasks.TimingTask_cfi import ecalTimingTaskPhase2
ecalMonitorTaskPhase2 = ecalMonitorTask.clone(
    workers = cms.untracked.vstring(
        "ClusterTask",
        "EnergyTask",
        "TimingTask",
        "PiZeroTask"
    ),
    workerParameters = cms.untracked.PSet(
        ClusterTask = ecalClusterTaskPhase2,
        EnergyTask = ecalEnergyTaskPhase2,
        TimingTask = ecalTimingTaskPhase2,
        PiZeroTask = ecalPiZeroTask
    ),
    collectionTags = ecalDQMCollectionTagsPhase2,
    # skip EcalRawData collection to prevent event type filtering while no Phase 2 raw data is defined
    skipCollections = cms.untracked.vstring('EcalRawData')
)