ATLAS Offline Software
Loading...
Searching...
No Matches
LArElecCalibDBConfig.py
Go to the documentation of this file.
1# Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
2
3from AthenaConfiguration.ComponentFactory import CompFactory
4
5from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator,ConfigurationError
6from IOVDbSvc.IOVDbSvcConfig import IOVDbSvcCfg,addFolderList, getSqliteContent
7
8
9#Import LArSymConditionsAlgs: Templated on the payload-type they handle
10#These algs are mostly needed for MC processing
11LArDAC2uASymAlg = CompFactory.getComp("LArSymConditionsAlg<LArDAC2uAMC, LArDAC2uASym>")
12LArMinBiasAverageSymAlg = CompFactory.getComp("LArSymConditionsAlg<LArMinBiasAverageMC, LArMinBiasAverageSym>")
13LArMinBiasSymAlg = CompFactory.getComp("LArSymConditionsAlg<LArMinBiasMC, LArMinBiasSym>")
14LArNoiseSymAlg = CompFactory.getComp("LArSymConditionsAlg<LArNoiseMC, LArNoiseSym>")
15LArRampSymAlg = CompFactory.getComp("LArSymConditionsAlg<LArRampMC, LArRampSym>")
16LArfSamplSymAlg = CompFactory.getComp("LArSymConditionsAlg<LArfSamplMC, LArfSamplSym>")
17LArAutoCorrSymAlg = CompFactory.getComp("LArSymConditionsAlg<LArAutoCorrMC, LArAutoCorrSym>")
18LAruA2MeVSymAlg = CompFactory.getComp("LArSymConditionsAlg<LAruA2MeVMC, LAruA2MeVSym>")
19LArShapeSymAlg = CompFactory.getComp("LArSymConditionsAlg<LArShape32MC, LArShape32Sym>")
20LArMPhysOverMcalSymAlg = CompFactory.getComp("LArSymConditionsAlg<LArMphysOverMcalMC, LArMphysOverMcalSym>")
21
22
23LArHVScaleCorrCondFlatAlg = CompFactory.getComp("LArFlatConditionsAlg<LArHVScaleCorrFlat>")
24LAruA2MeVCondAlg = CompFactory.getComp("LArFlatConditionsAlg<LAruA2MeVFlat>")
25LArDAC2uACondAlg = CompFactory.getComp("LArFlatConditionsAlg<LArDAC2uAFlat>")
26LArPedestalCondAlg = CompFactory.getComp("LArFlatConditionsAlg<LArPedestalFlat>")
27LArRampCondAlg = CompFactory.getComp("LArFlatConditionsAlg<LArRampFlat>")
28LArMphysOverMcalCondAlg = CompFactory.getComp("LArFlatConditionsAlg<LArMphysOverMcalFlat>")
29LArOFCCondAlg = CompFactory.getComp("LArFlatConditionsAlg<LArOFCFlat>")
30LArShapeCondAlg = CompFactory.getComp("LArFlatConditionsAlg<LArShapeFlat>")
31
32LArHVScaleCorrSCCondFlatAlg = CompFactory.getComp("LArFlatConditionsAlg<LArHVScaleCorrSC>")
33LArDAC2uASCCondAlg = CompFactory.getComp("LArFlatConditionsAlg<LArDAC2uASC>")
34LArRampSCCondAlg = CompFactory.getComp("LArFlatConditionsAlg<LArRampSC>")
35LAruA2MeVSCCondAlg = CompFactory.getComp("LArFlatConditionsAlg<LAruA2MeVSC>")
36LArfSamplSCCondAlg = CompFactory.getComp("LArFlatConditionsAlg<LArfSamplSC>")
37LArShapeSCCondAlg = CompFactory.getComp("LArFlatConditionsAlg<LArShapeSC>")
38LArPedestalSCCondAlg = CompFactory.getComp("LArFlatConditionsAlg<LArPedestalSC>")
39LArNoiseSCCondAlg = CompFactory.getComp("LArFlatConditionsAlg<LArNoiseSC>")
40LArMinBiasSCCondAlg = CompFactory.getComp("LArFlatConditionsAlg<LArMinBiasSC>")
41LArAutoCorrSCCondAlg = CompFactory.getComp("LArFlatConditionsAlg<LArAutoCorrSC>")
42LArPileupAverageSCCondAlg = CompFactory.getComp("LArFlatConditionsAlg<LArMinBiasAverageSC>")
43LArMphysOverMcalSCCondAlg = CompFactory.getComp("LArFlatConditionsAlg<LArMphysOverMcalSC>")
44LArOFCSCCondAlg = CompFactory.getComp("LArFlatConditionsAlg<LArOFCSC>")
45
46
47def LArElecCalibDBCfg(flags,condObjs):
48
49 #Check MC case
50 if flags.Input.isMC:
51 return LArElecCalibDBMCCfg(flags,condObjs)
52
53 from AthenaConfiguration.Enums import LHCPeriod
54
55 #Check run 1 case:
56 if flags.GeoModel.Run < LHCPeriod.Run2 :
57 return LArElecCalibDBRun1Cfg(flags,condObjs)
58
59 #Everything else, eg run 2 (and 3?) data
60 return LArElecCalibDBRun2Cfg(flags,condObjs)
61
62
63
64
65def LArElecCalibDBRun2Cfg(flags,condObjs):
66
67 sqliteFolders=getSqliteContent(flags.IOVDb.SqliteInput,flags.IOVDb.SqliteFolders,flags.IOVDb.DatabaseInstance)
68
69 _larCondDBFoldersDataR2 = {"Ramp":("LArRamp","/LAR/ElecCalibFlat/Ramp", LArRampCondAlg ),
70 "DAC2uA":("LArDAC2uA","/LAR/ElecCalibFlat/DAC2uA",LArDAC2uACondAlg),
71 "Pedestal":("LArPedestal","/LAR/ElecCalibFlat/Pedestal",LArPedestalCondAlg),
72 "uA2MeV":("LAruA2MeV","/LAR/ElecCalibFlat/uA2MeV", LAruA2MeVCondAlg),
73 "MphysOverMcal":("LArMphysOverMcal","/LAR/ElecCalibFlat/MphysOverMcal",LArMphysOverMcalCondAlg),
74 "OFC":("LArOFC","/LAR/ElecCalibFlat/OFC",LArOFCCondAlg),
75 "Shape":("LArShape","/LAR/ElecCalibFlat/Shape",LArShapeCondAlg),
76 "HVScaleCorr":("LArHVScaleCorr","/LAR/ElecCalibFlat/HVScaleCorr",LArHVScaleCorrCondFlatAlg),
77 "fSampl":("LArfSamplSym","/LAR/ElecCalibMC/fSampl",LArfSamplSymAlg),
78 }
79
80 result=IOVDbSvcCfg(flags)
81 iovDbSvc=result.getService("IOVDbSvc")
82 condLoader=result.getCondAlgo("CondInputLoader")
83
84 for condData in condObjs:
85 try:
86 outputKey,fldr,calg=_larCondDBFoldersDataR2[condData]
87 except KeyError:
88 raise ConfigurationError("No conditions data %s found for Run-2 data" % condData)
89
90 if fldr in sqliteFolders:
91 sqlite=sqliteFolders[fldr]
92 else:
93 sqlite=None
94
95
96 persClass="CondAttrListCollection"
97 # Potential special treatment for OFC/Shape: Load them from offline DB
98 if flags.LAr.OFCShapeFolder and condData == "OFC":
99 fldr="/LAR/ElecCalibOfl/OFC/PhysWave/RTM/"+flags.LAr.OFCShapeFolder
100 dbString=sqlite or "<db>COOLOFL_LAR/CONDBR2</db>"
101 persClass="LArOFCComplete"
102 calg = None
103 elif flags.LAr.OFCShapeFolder and condData == "Shape":
104 fldr="/LAR/ElecCalibOfl/Shape/RTM/"+flags.LAr.OFCShapeFolder
105 dbString=sqlite or "<db>COOLOFL_LAR/CONDBR2</db>"
106 persClass="LArShapeComplete"
107 calg = None
108 elif condData == "fSampl":
109 if not flags.Overlay.DataOverlay:
110 raise ConfigurationError("fSampl is only supported for data overlay")
111 dbString="<db>COOLOFL_LAR/OFLP200</db>"
112 persClass="LArfSamplMC"
113 result.addCondAlgo(CompFactory.LArMCSymCondAlg(ReadKey="LArOnOffIdMap"))
114 result.addCondAlgo(calg(ReadKey="LArfSampl", WriteKey=outputKey))
115 calg = None
116 else:
117 dbString=sqlite or "<db>COOLONL_LAR/CONDBR2</db>"
118
119 iovDbSvc.Folders.append(fldr+dbString)# (addFolder(flags,fldr,"LAR_ONL",'CondAttrListCollection'))
120 condLoader.Load.add((persClass,fldr))
121 if calg is not None:
122 result.addCondAlgo(calg (ReadKey=fldr, WriteKey=outputKey))
123
124 return result
125
126def LArElecCalibDBSCCfg(flags,condObjs,sqlite=None):
127
128 sqliteFolders=getSqliteContent(flags.IOVDb.SqliteInput,tuple(flags.IOVDb.SqliteFolders),flags.IOVDb.DatabaseInstance)
129
130 _larCondDBFoldersDataSC = {"Ramp":("LArRampSC","/LAR/ElecCalibFlatSC/Ramp", LArRampSCCondAlg ),
131 "DAC2uA":("LArDAC2uASC","/LAR/ElecCalibFlatSC/DAC2uA",LArDAC2uASCCondAlg),
132 "Pedestal":("LArPedestalSC","/LAR/ElecCalibFlatSC/Pedestal",LArPedestalSCCondAlg),
133 "uA2MeV":("LAruA2MeVSC","/LAR/ElecCalibFlatSC/uA2MeV", LAruA2MeVSCCondAlg),
134 "MphysOverMcal":("LArMphysOverMcalSC","/LAR/ElecCalibFlatSC/MphysOverMcal",LArMphysOverMcalSCCondAlg),
135 "OFC":("LArOFCSC","/LAR/ElecCalibFlatSC/OFC",LArOFCSCCondAlg),
136 "Shape":("LArShapeSC","/LAR/ElecCalibFlatSC/Shape",LArShapeSCCondAlg),
137 "HVScaleCorr":("LArHVScaleCorrSC","/LAR/ElecCalibFlatSC/HVScaleCorr",LArHVScaleCorrSCCondFlatAlg),
138 "fSampl":("LArfSamplSC","/LAR/ElecCalibMCSC/fSampl",LArfSamplSCCondAlg),
139 }
140
141 result=IOVDbSvcCfg(flags)
142 iovDbSvc=result.getService("IOVDbSvc")
143 condLoader=result.getCondAlgo("CondInputLoader")
144
145 for condData in condObjs:
146 try:
147 outputKey,fldr,calg=_larCondDBFoldersDataSC[condData]
148 except KeyError:
149 raise ConfigurationError("No conditions data %s found for SCdata" % condData)
150
151 if fldr in sqliteFolders:
152 sqlite=sqliteFolders[fldr]
153 else:
154 sqlite=None
155
156 persClass="CondAttrListCollection"
157 if condData == "fSampl":
158 #Sampling Fraction is always read from MC-db
159 dbString="<db>COOLOFL_LAR/OFLP200</db>"
160 result.addCondAlgo(calg(ReadKey="LArfSampl", WriteKey=outputKey))
161 calg = None
162 else:
163 dbString=sqlite or "<db>COOLONL_LAR/CONDBR2</db>"
164 iovDbSvc.Folders.append(fldr+dbString)# (addFolder(flags,fldr,"LAR_ONL",'CondAttrListCollection'))
165 condLoader.Load.add((persClass,fldr))
166 if calg is not None:
167 result.addCondAlgo(calg (ReadKey=fldr, WriteKey=outputKey))
168
169 return result
170
171
172
173def LArElecCalibDBRun1Cfg(flags,condObjs):
174
175 _larCondDBFoldersDataR1 = {"Ramp":("/LAR/ElecCalibOnl/Ramp","LAR_ONL","LArRampComplete",None),
176 "DAC2uA":("/LAR/ElecCalibOnl/DAC2uA","LAR_ONL","LArDAC2uAMC",LArDAC2uASymAlg),
177 "Pedestal":("/LAR/ElecCalibOnl/Pedestal<key>LArPedestal</key>","LAR_ONL","LArPedestalComplete",None),
178 "uA2MeV":("/LAR/ElecCalibOfl/uA2MeV/Symmetry","LAR_OFL", "LAruA2MeVMC",LAruA2MeVSymAlg),
179 "MphysOverMcal":("/LAR/ElecCalibOfl/MphysOverMcal/RTM","LAR_OFL","LArMphysOverMcalComplete",None),
180 "HVScaleCorr":("/LAR/ElecCalibOnl/HVScaleCorr","LAR_ONL","LArHVScaleCorrComplete",None),
181 "OFC":("/LAR/ElecCalibOfl/OFC/PhysWave/RTM/"+ flags.LAr.OFCShapeFolder if len(flags.LAr.OFCShapeFolder)>0 else "/LAR/ElecCalibOfl/OFC/PhysWave/RTM/5samples1phase","LAR_OFL","LArOFCComplete",None),
182 "Shape":("/LAR/ElecCalibOfl/Shape/RTM/"+ flags.LAr.OFCShapeFolder if len(flags.LAr.OFCShapeFolder)>0 else "/LAR/ElecCalibOfl/Shape/RTM/5samples1phase","LAR_OFL","LArShapeComplete",None),
183 }
184
185
186
187 result=ComponentAccumulator()
188 folderlist=[]
189 for condData in condObjs:
190 try:
191 folder,db,obj,calg=_larCondDBFoldersDataR1[condData]
192 except KeyError:
193 raise ConfigurationError("No conditions data %s found for Run-1 data" % condData)
194 folderlist.append((folder,db,obj))
195 if (calg):
196 if obj.endswith ('MC'):
197 obj = obj[:-2]
198 result.addCondAlgo(calg(ReadKey=obj,WriteKey=obj+"Sym"))
199 result.merge(addFolderList(flags,folderlist))
200
201 return result
202
203
204def LArElecCalibDBMCCfg(flags,folders,detdb="LAR_OFL"):
205 _larCondDBFoldersMC = {
206 "Ramp":("LArRampMC","/LAR/ElecCalibMC/Ramp","LArRamp", LArRampSymAlg ),
207 "AutoCorr":("LArAutoCorrMC","/LAR/ElecCalibMC/AutoCorr","LArAutoCorr", LArAutoCorrSymAlg),
208 "DAC2uA":("LArDAC2uAMC","/LAR/ElecCalibMC/DAC2uA","LArDAC2uA",LArDAC2uASymAlg),
209 "Pedestal":("LArPedestalMC","/LAR/ElecCalibMC/Pedestal","LArPedestal",None),
210 "Noise":("LArNoiseMC","/LAR/ElecCalibMC/Noise","LArNoise",LArNoiseSymAlg),
211 "fSampl":("LArfSamplMC","/LAR/ElecCalibMC/fSampl","LArfSampl",LArfSamplSymAlg),
212 "uA2MeV":("LAruA2MeVMC","/LAR/ElecCalibMC/uA2MeV","LAruA2MeV", LAruA2MeVSymAlg),
213 "MinBias":("LArMinBiasMC","/LAR/ElecCalibMC/MinBias","LArMinBias",LArMinBiasSymAlg),
214 "Shape":("LArShape32MC","/LAR/ElecCalibMC/Shape","LArShape",LArShapeSymAlg),
215 "MinBiasAvc":("LArMinBiasAverageMC","/LAR/ElecCalibMC/MinBiasAverage","LArMinBiasAverage",LArMinBiasAverageSymAlg),
216 "MphysOverMcal":("LArMphysOverMcalMC","/LAR/ElecCalibMC/MphysOverMcal","LArMphysOverMcal",LArMPhysOverMcalSymAlg),
217 "HVScaleCorr" : ("LArHVScaleCorrComplete", '/LAR/ElecCalibMC/HVScaleCorr',"LArHVScaleCorr",None)
218 }
219
220 result=ComponentAccumulator()
221 #Add cabling
222 from LArCabling.LArCablingConfig import LArOnOffIdMappingCfg
223 result.merge(LArOnOffIdMappingCfg(flags))
224 LArMCSymCondAlg=CompFactory.LArMCSymCondAlg
225 result.addCondAlgo(LArMCSymCondAlg(ReadKey="LArOnOffIdMap"))
226 folderlist=[]
227 for folder in folders:
228 try:
229 classname,fldr,key,calg=_larCondDBFoldersMC[folder]
230 except KeyError:
231 raise ConfigurationError("No conditions data %s found for Monte Carlo" % folder)
232
233 folderlist+=[(fldr,detdb,classname),]
234 if calg is not None:
235 result.addCondAlgo(calg(ReadKey=key,WriteKey=key+"Sym"))
236
237 result.merge(addFolderList(flags,folderlist,db="OFLP200"))
238 return result
239
240
241def LArElecCalibDBMCSCCfg(flags,folders,detdb="LAR_OFL"):
242 _larCondDBFoldersMC = {
243 "RampSC":('CondAttrListCollection',"/LAR/ElecCalibMCSC/Ramp","LArRampSC", LArRampSCCondAlg ),
244 "DAC2uASC":('CondAttrListCollection',"/LAR/ElecCalibMCSC/DAC2uA","LArDAC2uASC",LArDAC2uASCCondAlg),
245 "uA2MeVSC":('CondAttrListCollection',"/LAR/ElecCalibMCSC/uA2MeV","LAruA2MeVSC",LAruA2MeVSCCondAlg),
246 "fSamplSC":('CondAttrListCollection','/LAR/ElecCalibMCSC/fSampl',"LArfSamplSC",LArfSamplSCCondAlg),
247 "ShapeSC":('CondAttrListCollection','/LAR/ElecCalibMCSC/Shape',"LArShapeSC",LArShapeSCCondAlg),
248 "PileupAverageSC":('CondAttrListCollection','/LAR/ElecCalibMCSC/LArPileupAverage',"LArPileupAverageSC",LArPileupAverageSCCondAlg),
249 "PedestalSC":('CondAttrListCollection','/LAR/ElecCalibMCSC/Pedestal',"LArPedestalSC",LArPedestalSCCondAlg),
250 "NoiseSC":('CondAttrListCollection','/LAR/ElecCalibMCSC/Noise',"LArNoiseSC",LArNoiseSCCondAlg),
251 "MinBiasSC":('CondAttrListCollection','/LAR/ElecCalibMCSC/MinBias',"LArMinBiasSC",LArMinBiasSCCondAlg),
252 "AutoCorrSC":('CondAttrListCollection','/LAR/ElecCalibMCSC/AutoCorr',"LArAutoCorrSC",LArAutoCorrSCCondAlg)
253 }
254
255 result=ComponentAccumulator()
256 folderlist=[]
257 for folder in folders:
258 try:
259 classname,fldr,key,calg=_larCondDBFoldersMC[folder]
260 except KeyError:
261 raise ConfigurationError("No conditions data %s found for Monte Carlo" % folder)
262
263 folderlist+=[(fldr,detdb,classname),]
264 if calg is not None:
265 result.addCondAlgo(calg(ReadKey=fldr,WriteKey=key))
266
267 result.merge(addFolderList(flags,folderlist,db="OFLP200"))
268 return result
269
270
271if __name__ == "__main__":
272 from AthenaConfiguration.AllConfigFlags import initConfigFlags
273 from AthenaConfiguration.TestDefaults import defaultTestFiles
274
275 print ('--- run2')
276 flags1 = initConfigFlags()
277 flags1.Input.Files = defaultTestFiles.RAW_RUN2
278 flags1.lock()
279 acc1 = LArElecCalibDBCfg (flags1, ['Ramp', 'Pedestal'])
280 acc1.printCondAlgs(summariseProps=True)
281 print ('IOVDbSvc:', acc1.getService('IOVDbSvc').Folders)
282 acc1.wasMerged()
283
284 print ('--- run1')
285 flags2 = initConfigFlags()
286 flags2.Input.Files = defaultTestFiles.RAW_RUN2
287 from AthenaConfiguration.Enums import LHCPeriod
288 flags2.GeoModel.Run=LHCPeriod.Run1
289 flags2.lock()
290 acc2 = LArElecCalibDBCfg (flags2, ['Ramp', 'Pedestal'])
291 acc2.printCondAlgs(summariseProps=True)
292 print ('IOVDbSvc:', acc2.getService('IOVDbSvc').Folders)
293 acc2.wasMerged()
294
295 print ('--- mc')
296 flags3 = initConfigFlags()
297 flags3.Input.Files = defaultTestFiles.ESD
298 flags3.lock()
299 acc3 = LArElecCalibDBCfg (flags3, ['Ramp', 'Pedestal'])
300 acc3.printCondAlgs(summariseProps=True)
301 print ('IOVDbSvc:', acc3.getService('IOVDbSvc').Folders)
302 acc3.wasMerged()
303
python.LArElecCalibDBConfig.LArElecCalibDBMCCfg
LArElecCalibDBMCCfg(flags, folders, detdb="LAR_OFL")
Definition LArElecCalibDBConfig.py:204
python.LArElecCalibDBConfig.LArElecCalibDBMCSCCfg
LArElecCalibDBMCSCCfg(flags, folders, detdb="LAR_OFL")
Definition LArElecCalibDBConfig.py:241
python.LArElecCalibDBConfig.LArElecCalibDBSCCfg
LArElecCalibDBSCCfg(flags, condObjs, sqlite=None)
Definition LArElecCalibDBConfig.py:126
Generated on for ATLAS Offline Software by doxygen 1.14.0