HcalNanoAOD/python/customiseHcalLocal_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 # Customization for running on HCAL local run data
0004 #   - Call from cmsDriver.py with: `--customise DPGAnalysis/HcalNanoAOD/customiseHcalLocal_cff.customiseHcalLocal`
0005 def customiseHcalLocal(process):
0006     input_files = process.source.fileNames
0007     max_events = process.maxEvents.input
0008     process.source = cms.Source("HcalTBSource",
0009         fileNames = input_files,
0010         maxEvents = max_events,
0011         firstLuminosityBlockForEachRun = cms.untracked.VLuminosityBlockID([]),
0012     )
0013     process.hcalDigis.InputLabel = cms.InputTag('source')
0014
0015     # Uncomment if ZDC digis (QIE10, nTS=10) are causing problems
0016     #process.hcalDigis.saveQIE10DataNSamples = cms.untracked.vint32(10)
0017     #process.hcalDigis.saveQIE10DataTags = cms.untracked.vstring("ZDC")
0018
0019     if hasattr(process, "hcalDigiSortedTableTask"):
0020         process.hcalDigiSortedTable.nTS_HB = cms.untracked.uint32(8)
0021         process.hcalDigiSortedTable.nTS_HE = cms.untracked.uint32(8)
0022         process.hcalDigiSortedTable.nTS_HF = cms.untracked.uint32(6)
0023         process.hcalDigiSortedTable.nTS_HO = cms.untracked.uint32(10)
0024
0025     process.load("DPGAnalysis.HcalNanoAOD.hcalUMNioTable_cff")
0026     if hasattr(process, "hcalNanoTask"):
0027         process.hcalNanoTask.add(process.uMNioTable)
0028     if hasattr(process, "hcalNanoDigiTask"):
0029         process.hcalNanoDigiTask.add(process.uMNioTable)
0030
0031     return process