add customization and workflow for hcalNano from MC

Author: jhakala

Configuration/PyReleaseValidation/python/relval_nano.py

@@ -203,6 +203,13 @@ def subnext(self):
 steps['scoutingNANO_withPrompt_mc14.0'] = merge([{'-s': 'NANO:@Prompt+@Scout'},
                                                  steps['NANO_mc14.0']])
 
+steps['SinglePionRAW14.0'] = {'INPUT': InputInfo(
+    location='STD', dataSet='/SinglePion_E-50_Eta-0to3-pythia8-gun/Run3Winter25Digi-NoPU_142X_mcRun3_2025BOY_realistic_Candidate_2024_11_13_17_21_33-v2/GEN-SIM-RAW')}
+
+steps['hcalDPGNANO_mc14.0'] = merge([{'-s': 'RAW2DIGI,RECO,NANO:@HCALMC', '-n': '100',
+                                             '--processName': 'NANO'},
+                                            steps['NANO_mc14.0']])
+
 # 14.0 workflows -- data
 lumis_Run2024D = {380306: [[28, 273]]}
 steps['MuonEG2024MINIAOD14.0'] = {'INPUT': InputInfo(location='STD', ls=lumis_Run2024D,
@@ -369,6 +376,13 @@ def subnext(self):
 workflows[_wfn()] = ['hcalDPGNANO140Xrun3', ['ZeroBias2024RAW14.0', 'hcalDPGNANO_data14.0']]
 workflows[_wfn()] = ['hcalDPGCalibNANO140Xrun3', ['TestEnablesEcalHcal2024RAW14.0', 'hcalDPGCalibNANO_data14.0']]
 
+# DPG custom NANOs, MC
+_wfn.subnext()
+workflows[_wfn()] = ['hcalDPGMCNANO140Xrun3', ['SinglePionRAW14.0', 'hcalDPGNANO_mc14.0']]
+# The above HCAL workflow is actually using data produced for 14.2
+# but I keep the 14.0 label for now since it's consistent with those ones
+# let me know if I should change this
+
 _wfn.next(9)
 ######## 2500.9xx ########
 # NANOGEN

DPGAnalysis/HcalNanoAOD/python/customiseHcalMC_cff.py

@@ -0,0 +1,15 @@
+import FWCore.ParameterSet.Config as cms
+
+# Customization for running on MC
+#   - Call from cmsDriver.py with: `--customise DPGAnalysis/HcalNanoAOD/customiseHcalMC_cff.customiseHcalMC`
+def customiseHcalMC(process):
+  # Point to appropriate MC digi collections
+  process.hcalDigiSortedTable.tagQIE11 = cms.untracked.InputTag("simHcalUnsuppressedDigis", "HBHEQIE11DigiCollection")
+  process.hcalDigiSortedTable.tagQIE10 = cms.untracked.InputTag("simHcalUnsuppressedDigis","HFQIE10DigiCollection")
+  process.hcalDigiSortedTable.tagHO = cms.untracked.InputTag("simHcalUnsuppressedDigis")
+
+  # Use the appropriate number of samples for digis in MC
+  process.hcalDigiSortedTable.nTS_HB = cms.untracked.uint32(10)
+  process.hcalDigiSortedTable.nTS_HE = cms.untracked.uint32(10)
+
+  return process

PhysicsTools/NanoAOD/python/autoNANO.py

@@ -58,6 +58,8 @@ def expandNanoMapping(seqList, mapping, key):
     'HCAL': {'sequence': 'DPGAnalysis/HcalNanoAOD/hcalNano_cff.hcalNanoTask'},
     'HCALCalib': {'sequence': 'DPGAnalysis/HcalNanoAOD/hcalNano_cff.hcalNanoTask',
                   'customize': 'DPGAnalysis/HcalNanoAOD/customiseHcalCalib_cff.customiseHcalCalib'},
+    'HCALMC': {'sequence': 'DPGAnalysis/HcalNanoAOD/hcalNano_cff.hcalNanoTask',
+                  'customize': 'DPGAnalysis/HcalNanoAOD/customiseHcalMC_cff.customiseHcalMC'},
     # EGM flavours: add variables through customize
     'EGM': {'sequence': '@PHYS',
             'customize': '@PHYS+PhysicsTools/NanoAOD/egamma_custom_cff.addExtraEGammaVarsCustomize'},