Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
import FWCore.ParameterSet.Config as cms

# Customization for running on HCAL local run data
#   - Call from cmsDriver.py with: `--customise DPGAnalysis/HcalNanoAOD/customiseHcalLocal_cff.customiseHcalLocal`
def customiseHcalLocal(process):
    input_files = process.source.fileNames
    max_events = process.maxEvents.input
    process.source = cms.Source("HcalTBSource",
        fileNames = input_files,
        maxEvents = max_events,
        firstLuminosityBlockForEachRun = cms.untracked.VLuminosityBlockID([]),
    )
    process.hcalDigis.InputLabel = cms.InputTag('source')

    # Uncomment if ZDC digis (QIE10, nTS=10) are causing problems
    #process.hcalDigis.saveQIE10DataNSamples = cms.untracked.vint32(10) 
    #process.hcalDigis.saveQIE10DataTags = cms.untracked.vstring("ZDC")

    if hasattr(process, "hcalDigiSortedTableTask"):
        process.hcalDigiSortedTable.nTS_HB = cms.untracked.uint32(8)
        process.hcalDigiSortedTable.nTS_HE = cms.untracked.uint32(8)
        process.hcalDigiSortedTable.nTS_HF = cms.untracked.uint32(6)
        process.hcalDigiSortedTable.nTS_HO = cms.untracked.uint32(10)

    process.load("DPGAnalysis.HcalNanoAOD.hcalUMNioTable_cff")
    if hasattr(process, "hcalNanoTask"):
        process.hcalNanoTask.add(process.uMNioTable)
    if hasattr(process, "hcalNanoDigiTask"):
        process.hcalNanoDigiTask.add(process.uMNioTable)

    return process