|
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
|
import FWCore.ParameterSet.Config as cms
from PhysicsTools.NanoAOD.common_cff import *
def ecal_phisym_workflow(process,
produce_by_run : bool=False,
save_edm : bool=False,
save_edmnano : bool=False,
save_flatnano : bool=True):
"""
Customize the process to include the entire EcalPhiSym workflow:
- ECAL local reco
- PhiSymRecHit producer
- EDM output (standard, EDMNANO, FlatNANO)
"""
reco = ecal_phisym_reco_sequence(process, produce_by_run=produce_by_run)
tables = ecal_phisym_flattables(process, produce_by_run=produce_by_run)
outputs = ecal_phisym_output(process,
save_edm=save_edm,
save_edmnano=save_edmnano,
save_flatnano=save_flatnano)
process.path = cms.Path(reco*tables)
process.output_step = cms.EndPath()
for out in outputs:
process.output_step += out
process.schedule = cms.Schedule(process.path, process.output_step)
def ecal_phisym_reco_sequence(process, produce_by_run : bool=False):
"""
Customize process to include the EcalPhiSym standard reco sequence
"""
process.load('RecoLuminosity.LumiProducer.bunchSpacingProducer_cfi')
process.load('RecoLocalCalo.EcalRecProducers.ecalMultiFitUncalibRecHit_cfi')
process.load('RecoLocalCalo.EcalRecProducers.ecalUncalibRecHit_cfi')
process.load('RecoLocalCalo.EcalRecProducers.ecalRecHit_cfi')
#ecalMultiFitUncalibRecHit
process.ecalMultiFitUncalibRecHit.EBdigiCollection = cms.InputTag("hltEcalPhiSymFilter","phiSymEcalDigisEB")
process.ecalMultiFitUncalibRecHit.EEdigiCollection = cms.InputTag("hltEcalPhiSymFilter","phiSymEcalDigisEE")
#ecalRecHit (no ricovery)
process.ecalRecHit.killDeadChannels = cms.bool( False )
process.ecalRecHit.recoverEBVFE = cms.bool( False )
process.ecalRecHit.recoverEEVFE = cms.bool( False )
process.ecalRecHit.recoverEBFE = cms.bool( False )
process.ecalRecHit.recoverEEFE = cms.bool( False )
process.ecalRecHit.recoverEEIsolatedChannels = cms.bool( False )
process.ecalRecHit.recoverEBIsolatedChannels = cms.bool( False )
# PHISYM producer
process.load('Calibration.EcalCalibAlgos.EcalPhiSymRecHitProducers_cfi')
# SCHEDULE
reconstruction_step = cms.Sequence( process.bunchSpacingProducer * (process.ecalMultiFitUncalibRecHit + process.ecalRecHit) )
reconstruction_step *= process.EcalPhiSymRecHitProducerRun if produce_by_run else process.EcalPhiSymRecHitProducerLumi
return reconstruction_step
def ecal_phisym_flattables(process, produce_by_run : bool=False):
"""
Add the NanoAOD flat table producers.
This functions adjust also the output columns.
Should be called once nMisCalib has been set in the EcalPhiSymRecHitProducer
"""
process.load('Calibration.EcalCalibAlgos.EcalPhiSymFlatTableProducers_cfi')
nmis = process.EcalPhiSymRecHitProducerRun.nMisCalib.value()
for imis in range(1, nmis+1):
# get the naming and indexing right.
if imis<nmis/2+1:
var_name = 'sumEt_m'+str(abs(int(imis-(nmis/2)-1)))
var = Var(f'sumEt({imis})', float, doc='ECAL PhiSym rechits: '+str(imis-(nmis/2)-1)+'*miscalib et', precision=23)
else:
var_name = 'sumEt_p'+str(int(imis-(nmis/2)))
var = Var(f'sumEt({imis})', float, doc='ECAL PhiSym rechits: '+str(imis-(nmis/2))+'*miscalib et', precision=23)
if produce_by_run:
setattr(process.ecalPhiSymRecHitRunTableEB.variables, var_name, var)
setattr(process.ecalPhiSymRecHitRunTableEE.variables, var_name, var)
flattable_sequence = cms.Sequence( process.ecalPhiSymRecHitRunTableEB +
process.ecalPhiSymRecHitRunTableEE +
process.ecalPhiSymInfoRunTable )
else:
setattr(process.ecalPhiSymRecHitLumiTableEB.variables, var_name, var)
setattr(process.ecalPhiSymRecHitLumiTableEE.variables, var_name, var)
flattable_sequence = cms.Sequence( process.ecalPhiSymRecHitLumiTableEB +
process.ecalPhiSymRecHitLumiTableEE +
process.ecalPhiSymInfoLumiTable
)
return flattable_sequence
def ecal_phisym_output(process,
save_edm : bool=False,
save_edmnano : bool=False,
save_flatnano : bool=True):
"""
Customize EcalPhiSym output
"""
outputs = []
if save_flatnano or save_edmnano:
NanoAODEcalPhiSymEventContent = cms.PSet(
outputCommands = cms.untracked.vstring(
'drop *',
"keep nanoaod*_*_*_*", # event data
"keep nanoaodMergeableCounterTable_*Table_*_*", # accumulated per/run or per/lumi data
"keep nanoaodUniqueString_nanoMetadata_*_*", # basic metadata
)
)
if save_flatnano:
process.nanoout = cms.OutputModule("NanoAODOutputModule",
fileName = cms.untracked.string('ecal_phisym_nano.root'),
outputCommands = NanoAODEcalPhiSymEventContent.outputCommands,
compressionLevel = cms.untracked.int32(9),
compressionAlgorithm = cms.untracked.string("LZMA"),
)
outputs.append(process.nanoout)
if save_edmnano:
process.nanooutedm = cms.OutputModule("PoolOutputModule",
fileName = cms.untracked.string('ecal_phisym_edmnano.root'),
outputCommands = NanoAODEcalPhiSymEventContent.outputCommands,
)
outputs.append(process.nanooutedm)
if save_edm:
ECALPHISYM_output_commands = cms.untracked.vstring(
"drop *",
"keep *_PhiSymProducerRun_*_*")
process.EcalPhiSymOutput = cms.OutputModule("PoolOutputModule",
splitLevel = cms.untracked.int32(2),
compressionLevel = cms.untracked.int32(5),
compressionAlgorithm = cms.untracked.string('LZMA'),
outputCommands = ECALPHISYM_output_commands,
fileName = cms.untracked.string('ecal_phisym_reco.root')
)
outputs.append(process.EcalPhiSymOutput)
return outputs
def customise(process):
"""
Function to customize the process produced by cmsDriver.
The customisation works for a process that satisfies the following conditions:
- Run on /AlCaPhiSym/*/RAW data
- Run the following sequence (-s option of cmsDriver):
RECO:bunchSpacingProducer+ecalMultiFitUncalibRecHitTask+ecalCalibratedRecHitTask,ALCA:EcalPhiSymByRun (or EcalPhiSymByLumi)
"""
# Change input collection for the /AlCaPhiSym/*/RAW stream dataformat
process.ecalMultiFitUncalibRecHitCPU.EBdigiCollection = "hltEcalPhiSymFilter:phiSymEcalDigisEB"
process.ecalMultiFitUncalibRecHitCPU.EEdigiCollection = "hltEcalPhiSymFilter:phiSymEcalDigisEE"
process.ecalRecHit.cpu.killDeadChannels = cms.bool( False )
process.ecalRecHit.cpu.recoverEBVFE = cms.bool( False )
process.ecalRecHit.cpu.recoverEEVFE = cms.bool( False )
process.ecalRecHit.cpu.recoverEBFE = cms.bool( False )
process.ecalRecHit.cpu.recoverEEFE = cms.bool( False )
process.ecalRecHit.cpu.recoverEEIsolatedChannels = cms.bool( False )
process.ecalRecHit.cpu.recoverEBIsolatedChannels = cms.bool( False )
if "ALCARECOStreamEcalPhiSymByRunOutPath" in process.pathNames():
process.schedule.remove(process.ALCARECOStreamEcalPhiSymByRunOutPath)
if "ALCARECOStreamEcalPhiSymByLumiOutPath" in process.pathNames():
process.schedule.remove(process.ALCARECOStreamEcalPhiSymByLumiOutPath)
process.ALCARECOStreamEcalPhiSymOutNanoPath = cms.EndPath(ecal_phisym_output(process, save_edm=False, save_edmnano=True, save_flatnano=False)[0])
process.schedule.append(process.ALCARECOStreamEcalPhiSymOutNanoPath)
return process
|