89def LArPileUpToolCfg(flags, name="LArPileUpTool", **kwargs):
90 """Return ComponentAccumulator with configured LArPileUpTool"""
93 # add new conditions for LArXTalkWeight
94 acc.merge(LArXTalkWeightCondAlgCfg(flags))
97 if flags.Common.isOverlay:
98 # Some noise needs to be added during MC Overlay
99 # No noise should be added during Data Overlay
100 kwargs.setdefault("NoiseOnOff", not flags.Overlay.DataOverlay)
101 kwargs.setdefault("NoiseOnOff", flags.Digitization.DoCaloNoise) # For other jobs go with the noise flag setting.
102 kwargs.setdefault("DoDigiTruthReconstruction", flags.Digitization.EnableCaloHSTruthRecoInputs)
103 kwargs.setdefault("RandomSeedOffset", flags.Digitization.RandomSeedOffset)
104 if flags.Digitization.PileUp:
106 if flags.Digitization.DoXingByXingPileUp:
107 kwargs.setdefault("FirstXing", -751)
108 kwargs.setdefault("LastXing", 101)
110 intervals += [acc.popToolsAndMerge(LArRangeEMCfg(flags))]
111 intervals += [acc.popToolsAndMerge(LArRangeHECCfg(flags))]
112 intervals += [acc.popToolsAndMerge(LArRangeFCALCfg(flags))]
113 kwargs.setdefault("PileUpMergeSvc", acc.getPrimaryAndMerge(PileUpMergeSvcCfg(flags, Intervals=intervals)))
115 kwargs.setdefault("PileUpMergeSvc", '')
116 kwargs.setdefault("RndmEvtOverlay", flags.Common.isOverlay)
117 # if doing MC+MC overlay
118 if flags.Common.isOverlay and flags.Input.isMC:
119 kwargs.setdefault("isMcOverlay", True)
121 # cosmics digitization
122 if flags.Beam.Type is BeamType.Cosmics:
123 kwargs.setdefault("UseTriggerTime", True)
124 acc.addEventAlgo(CompFactory.CosTriggerTimeAlg())
125 if flags.Digitization.PileUp or flags.Common.isOverlay:
126 kwargs.setdefault("PileUp", True)
127 kwargs.setdefault("useLArFloat", useLArFloat(flags))
128 if useLArFloat(flags):
129 acc.merge(InputOverwriteCfg("LArHitContainer","LArHitEMB","LArHitFloatContainer","LArHitEMB"))
130 acc.merge(InputOverwriteCfg("LArHitContainer","LArHitEMEC","LArHitFloatContainer","LArHitEMEC"))
131 acc.merge(InputOverwriteCfg("LArHitContainer","LArHitHEC","LArHitFloatContainer","LArHitHEC"))
132 acc.merge(InputOverwriteCfg("LArHitContainer","LArHitFCAL","LArHitFloatContainer","LArHitFCAL"))
133 if flags.Common.isOverlay:
134 from SGComps.SGInputLoaderConfig import SGInputLoaderCfg
135 acc.merge(SGInputLoaderCfg(flags, [
136 "LArHitFloatContainer#LArHitEMB",
137 "LArHitFloatContainer#LArHitEMEC",
138 "LArHitFloatContainer#LArHitFCAL",
139 "LArHitFloatContainer#LArHitHEC",
142 if flags.Common.isOverlay and not flags.Sim.DoFullChain:
143 from SGComps.SGInputLoaderConfig import SGInputLoaderCfg
144 acc.merge(SGInputLoaderCfg(flags, [
145 "LArHitContainer#LArHitEMB",
146 "LArHitContainer#LArHitEMEC",
147 "LArHitContainer#LArHitFCAL",
148 "LArHitContainer#LArHitHEC",
151 if flags.Common.isOverlay:
152 kwargs.setdefault("OnlyUseContainerName", False)
153 if flags.Overlay.IgnoreBkgInputs:
154 kwargs.setdefault("InputDigitContainer", "")
155 elif flags.Input.isMC:
156 kwargs.setdefault("InputDigitContainer", f"{flags.Overlay.BkgPrefix}LArDigitContainer_MC")
158 kwargs.setdefault("InputDigitContainer", f"{flags.Overlay.BkgPrefix}LArDigitContainer_data")
160 if kwargs["InputDigitContainer"]:
161 from SGComps.SGInputLoaderConfig import SGInputLoaderCfg
162 acc.merge(SGInputLoaderCfg(flags, [f'LArDigitContainer#{kwargs["InputDigitContainer"]}']))
164 kwargs.setdefault("OnlyUseContainerName", flags.Digitization.PileUp)
165 LArPileUpTool = CompFactory.LArPileUpTool
166 acc.setPrivateTools(LArPileUpTool(name, **kwargs))
169def LArHitEMapToDigitAlgCfg(flags, name="LArHitEMapToDigitAlg", **kwargs):
170 """Return ComponentAccumulator with configured LArHitEMapToDigitAlg"""
171 acc = LArGMCfg(flags)
173 #The LArHitEMapToDigit needs: Noise, fSampl, Pedestal,Shape ADC2MeV
174 # AutoCorrNoise, the list of bad FEBs and the cabling
175 acc.merge(LArADC2MeVCondAlgCfg(flags))
176 acc.merge(LArBadFebCfg(flags))
177 acc.merge(LArBadChannelCfg(flags))
178 if flags.Overlay.DataOverlay:
179 kwargs.setdefault("ShapeKey", "LArShape")
180 #kwargs.setdefault("AutoCorrNoiseKey", "LArAutoCorr")
181 if not flags.Digitization.DoCaloNoise:
182 requiredConditons=["fSampl", "Pedestal", "Shape"]
184 requiredConditons=["Noise", "fSampl", "Pedestal", "Shape"]
185 acc.merge(LArElecCalibDBCfg(flags,requiredConditons))
187 if not flags.Common.isOverlay:
188 acc.merge(LArAutoCorrNoiseCondAlgCfg(flags))
189 #kwargs.setdefault('AutoCorrNoiseKey','LArAutoCorr')
191 if "ProblemsToMask" not in kwargs:
192 kwargs["ProblemsToMask"] = ["deadReadout", "deadPhys"]
194 if flags.Common.isOverlay:
195 # Some noise needs to be added during MC Overlay
196 # No noise should be added during Data Overlay
197 kwargs.setdefault("NoiseOnOff", not flags.Overlay.DataOverlay)
199 kwargs.setdefault("NoiseOnOff", flags.Digitization.DoCaloNoise)
200 kwargs.setdefault("DoDigiTruthReconstruction", flags.Digitization.EnableCaloHSTruthRecoInputs)
201 kwargs.setdefault("RandomSeedOffset", flags.Digitization.RandomSeedOffset)
203 if flags.GeoModel.Run==LHCPeriod.Run4 and flags.IOVDb.UseCREST:
204 #The run 4 readout electronics has only 2 gains
205 for subdet in ("EM","HEC","FCAL","EMECIW"):
206 kwargs.setdefault("GainRange"+subdet,(0,1))
207 kwargs.setdefault("LowGainThresh"+subdet,0xFFFFFFFF) #Low gains doesn't exist any more in R 4
208 #Gain switching threshold in the lower gain (aka MEDIUM)
209 #Assume gain switch at 80% of pre-amp saturation in HG. -> 9000 ADC counts in the lower gain
210 kwargs.setdefault("HighGainThresh"+subdet,9000)
211 kwargs.setdefault("maxADC",32767)
214 if (not flags.Digitization.HighGainFCal) and (not flags.Common.isOverlay):
215 kwargs.setdefault("HighGainThreshFCAL", 0)
216 kwargs.setdefault("GainRangeFCAL",[int(CaloGain.LARMEDIUMGAIN),int(CaloGain.LARLOWGAIN)])
217 if (not flags.Digitization.HighGainEMECIW) and (not flags.Common.isOverlay):
218 kwargs.setdefault("HighGainThreshEMECIW", 0)
219 kwargs.setdefault("GainRangeEMECIW",[int(CaloGain.LARMEDIUMGAIN),int(CaloGain.LARLOWGAIN)])
222 kwargs.setdefault("RndmEvtOverlay", flags.Common.isOverlay)
223 if flags.Common.ProductionStep == ProductionStep.PileUpPresampling:
224 kwargs.setdefault("DigitContainer", flags.Overlay.BkgPrefix + "LArDigitContainer_MC")
226 kwargs.setdefault("DigitContainer", "LArDigitContainer_MC") # FIXME - should not be hard-coded
227 # if doing MC+MC overlay
228 if flags.Common.isOverlay and flags.Input.isMC:
229 kwargs.setdefault("isMcOverlay", True)
230 kwargs.setdefault("Nsamples", flags.LAr.ROD.nSamples)
231 kwargs.setdefault("firstSample", #Need to set a negative value to include preceeding samples
232 -flags.LAr.ROD.nPreceedingSamples if flags.LAr.ROD.nPreceedingSamples!=0 else flags.LAr.ROD.FirstSample)
235 LArHitEMapToDigitAlg = CompFactory.LArHitEMapToDigitAlg
236 acc.addEventAlgo(LArHitEMapToDigitAlg(name, **kwargs))
239def LArOutputCfg(flags):
240 """Return ComponentAccumulator with Output for LAr. Not standalone."""
241 acc = ComponentAccumulator()
242 if flags.Output.doWriteRDO:
244 if flags.Digitization.AddCaloDigi:
245 ItemList.append("LArDigitContainer#*")
246 elif flags.Common.ProductionStep == ProductionStep.PileUpPresampling:
247 ItemList.append("LArDigitContainer#" + flags.Overlay.BkgPrefix + "LArDigitContainer_MC")
248 elif flags.Digitization.AddCaloDigiThinned:
249 ItemList.append("LArDigitContainer#LArDigitContainer_MC_Thinned")
250 if flags.Common.ProductionStep != ProductionStep.PileUpPresampling:
251 ItemList.append("LArRawChannelContainer#LArRawChannels")
252 if flags.Overlay.DataOverlay:
253 ItemList.append("LArFebHeaderContainer#LArFebHeader")
254 if flags.Digitization.EnableTruth:
255 ItemList.append("CaloCalibrationHitContainer#*")
256 acc.merge(TruthDigitizationOutputCfg(flags))
257 acc.merge(OutputStreamCfg(flags, "RDO", ItemList))
261def LArDigitizationBasicCfg(flags, **kwargs):
262 """Return ComponentAccumulator with LAr digitization configuration"""
263 acc = ComponentAccumulator()
264 if "PileUpTools" not in kwargs:
265 PileUpTools = acc.popToolsAndMerge(LArPileUpToolCfg(flags))
266 kwargs["PileUpTools"] = PileUpTools
267 acc.merge(PileUpToolsCfg(flags, **kwargs))
269 acc.merge(LArHitEMapToDigitAlgCfg(flags))
271 if flags.LAr.ROD.ApplyRODBCIDCorr:
272 mlog = logging.getLogger('LArDigitizationBasicCfg')
273 mlog.info('LAr.ROD.ApplyRODBCIDCorr is set: scheduling LArRODBCIDCorrAlg '
274 'to subtract the BCID-dependent pile-up offset from the digits')
275 if (flags.hasFlag('Calo.Cell.doPileupOffsetBCIDCorr')
276 and flags.Calo.Cell.doPileupOffsetBCIDCorr):
277 mlog.warning('Calo.Cell.doPileupOffsetBCIDCorr is also enabled: the '
278 'pile-up offset will be subtracted twice')
279 acc.merge(LArRODBCIDCorrAlgCfg(flags))
281 if flags.LAr.ROD.NNRawChannelBuilding:
282 acc.merge(LArNNRawChannelBuilderCfg(flags))
283 elif flags.LAr.ROD.OFFCRawChannelBuilding:
284 acc.merge(LArOFFCRawChannelBuilderCfg(flags))
286 acc.merge(LArRawChannelBuilderAlgCfg(flags))
288 if flags.Digitization.AddCaloDigiThinned:
289 acc.merge(LArDigitThinnerCfg(flags))
300def LArOverlayDigitizationBasicCfg(flags, name="digitmaker1", **kwargs):
301 """Return ComponentAccumulator with LAr Overlay digitization configuration"""
302 acc = ComponentAccumulator()
304 if "LArPileUpTool" not in kwargs:
305 kwargs["LArPileUpTool"] = acc.popToolsAndMerge(LArPileUpToolCfg(flags))
307 if flags.Concurrency.NumThreads > 0:
308 kwargs.setdefault('Cardinality', flags.Concurrency.NumThreads)
310 LArHitEMapMaker = CompFactory.LArHitEMapMaker
311 acc.addEventAlgo(LArHitEMapMaker(name, **kwargs))
312 acc.merge(LArHitEMapToDigitAlgCfg(flags))
314 acc.merge(LArRawChannelBuilderAlgCfg(flags))
315 if flags.Digitization.AddCaloDigiThinned:
316 acc.merge(LArDigitThinnerCfg(flags))
327def LArAutoCorrNoiseCondSCAlgCfg(flags, **kwargs):
328 acc = ComponentAccumulator()
330 requiredConditions=["AutoCorrSC"]
331 from LArConfiguration.LArElecCalibDBConfig import LArElecCalibDBMCSCCfg
332 acc.merge(LArElecCalibDBMCSCCfg(flags,requiredConditions))
333 kwargs.setdefault("LArAutoCorrObjKey", "LArAutoCorrSC") # Provided by LArFlatConditionsAlg<LArAutoCorrSC>
335 from LArCabling.LArCablingConfig import LArOnOffIdMappingSCCfg
336 acc.merge(LArOnOffIdMappingSCCfg(flags))
337 kwargs.setdefault("LArOnOffIdMappingObjKey", 'LArOnOffIdMapSC')
338 kwargs.setdefault("nSampl", flags.LAr.ROD.nSamples + 2) # See ATLASSIM-5483
339 kwargs.setdefault("isSuperCell", True)
340 kwargs.setdefault("LArAutoCorrNoiseObjKey", "LArAutoCorrNoiseSC") # output
341 acc.addCondAlgo(CompFactory.LArAutoCorrNoiseCondAlg(name="LArAutoCorrNoiseSCCondAlg", **kwargs))