84def LArPileUpToolCfg(flags, name="LArPileUpTool", **kwargs):
85 """Return ComponentAccumulator with configured LArPileUpTool"""
88 # add new conditions for LArXTalkWeight
89 acc.merge(LArXTalkWeightCondAlgCfg(flags))
92 if flags.Common.isOverlay:
93 # Some noise needs to be added during MC Overlay
94 # No noise should be added during Data Overlay
95 kwargs.setdefault("NoiseOnOff", not flags.Overlay.DataOverlay)
96 kwargs.setdefault("NoiseOnOff", flags.Digitization.DoCaloNoise) # For other jobs go with the noise flag setting.
97 kwargs.setdefault("DoDigiTruthReconstruction", flags.Digitization.EnableCaloHSTruthRecoInputs)
98 kwargs.setdefault("RandomSeedOffset", flags.Digitization.RandomSeedOffset)
99 if flags.Digitization.PileUp:
101 if flags.Digitization.DoXingByXingPileUp:
102 kwargs.setdefault("FirstXing", -751)
103 kwargs.setdefault("LastXing", 101)
105 intervals += [acc.popToolsAndMerge(LArRangeEMCfg(flags))]
106 intervals += [acc.popToolsAndMerge(LArRangeHECCfg(flags))]
107 intervals += [acc.popToolsAndMerge(LArRangeFCALCfg(flags))]
108 kwargs.setdefault("PileUpMergeSvc", acc.getPrimaryAndMerge(PileUpMergeSvcCfg(flags, Intervals=intervals)))
110 kwargs.setdefault("PileUpMergeSvc", '')
111 kwargs.setdefault("RndmEvtOverlay", flags.Common.isOverlay)
112 # if doing MC+MC overlay
113 if flags.Common.isOverlay and flags.Input.isMC:
114 kwargs.setdefault("isMcOverlay", True)
116 # cosmics digitization
117 if flags.Beam.Type is BeamType.Cosmics:
118 kwargs.setdefault("UseTriggerTime", True)
119 acc.addEventAlgo(CompFactory.CosTriggerTimeAlg())
120 if flags.Digitization.PileUp or flags.Common.isOverlay:
121 kwargs.setdefault("PileUp", True)
122 kwargs.setdefault("useLArFloat", useLArFloat(flags))
123 if useLArFloat(flags):
124 acc.merge(InputOverwriteCfg("LArHitContainer","LArHitEMB","LArHitFloatContainer","LArHitEMB"))
125 acc.merge(InputOverwriteCfg("LArHitContainer","LArHitEMEC","LArHitFloatContainer","LArHitEMEC"))
126 acc.merge(InputOverwriteCfg("LArHitContainer","LArHitHEC","LArHitFloatContainer","LArHitHEC"))
127 acc.merge(InputOverwriteCfg("LArHitContainer","LArHitFCAL","LArHitFloatContainer","LArHitFCAL"))
128 if flags.Common.isOverlay:
129 from SGComps.SGInputLoaderConfig import SGInputLoaderCfg
130 acc.merge(SGInputLoaderCfg(flags, [
131 "LArHitFloatContainer#LArHitEMB",
132 "LArHitFloatContainer#LArHitEMEC",
133 "LArHitFloatContainer#LArHitFCAL",
134 "LArHitFloatContainer#LArHitHEC",
137 if flags.Common.isOverlay and not flags.Sim.DoFullChain:
138 from SGComps.SGInputLoaderConfig import SGInputLoaderCfg
139 acc.merge(SGInputLoaderCfg(flags, [
140 "LArHitContainer#LArHitEMB",
141 "LArHitContainer#LArHitEMEC",
142 "LArHitContainer#LArHitFCAL",
143 "LArHitContainer#LArHitHEC",
146 if flags.Common.isOverlay:
147 kwargs.setdefault("OnlyUseContainerName", False)
149 kwargs.setdefault("InputDigitContainer", f"{flags.Overlay.BkgPrefix}LArDigitContainer_MC")
151 kwargs.setdefault("InputDigitContainer", f"{flags.Overlay.BkgPrefix}LArDigitContainer_data")
153 from SGComps.SGInputLoaderConfig import SGInputLoaderCfg
154 acc.merge(SGInputLoaderCfg(flags, [f'LArDigitContainer#{kwargs["InputDigitContainer"]}']))
156 kwargs.setdefault("OnlyUseContainerName", flags.Digitization.PileUp)
157 LArPileUpTool = CompFactory.LArPileUpTool
158 acc.setPrivateTools(LArPileUpTool(name, **kwargs))
161def LArHitEMapToDigitAlgCfg(flags, name="LArHitEMapToDigitAlg", **kwargs):
162 """Return ComponentAccumulator with configured LArHitEMapToDigitAlg"""
163 acc = LArGMCfg(flags)
165 #The LArHitEMapToDigit needs: Noise, fSampl, Pedestal,Shape ADC2MeV
166 # AutoCorrNoise, the list of bad FEBs and the cabling
167 acc.merge(LArADC2MeVCondAlgCfg(flags))
168 acc.merge(LArBadFebCfg(flags))
169 acc.merge(LArBadChannelCfg(flags))
170 if flags.Overlay.DataOverlay:
171 kwargs.setdefault("ShapeKey", "LArShape")
172 #kwargs.setdefault("AutoCorrNoiseKey", "LArAutoCorr")
173 if not flags.Digitization.DoCaloNoise:
174 requiredConditons=["fSampl", "Pedestal", "Shape"]
176 requiredConditons=["Noise", "fSampl", "Pedestal", "Shape"]
177 acc.merge(LArElecCalibDBCfg(flags,requiredConditons))
179 if not flags.Common.isOverlay:
180 acc.merge(LArAutoCorrNoiseCondAlgCfg(flags))
181 #kwargs.setdefault('AutoCorrNoiseKey','LArAutoCorr')
183 if "ProblemsToMask" not in kwargs:
184 kwargs["ProblemsToMask"] = ["deadReadout", "deadPhys"]
186 if flags.Common.isOverlay:
187 # Some noise needs to be added during MC Overlay
188 # No noise should be added during Data Overlay
189 kwargs.setdefault("NoiseOnOff", not flags.Overlay.DataOverlay)
191 kwargs.setdefault("NoiseOnOff", flags.Digitization.DoCaloNoise)
192 kwargs.setdefault("DoDigiTruthReconstruction", flags.Digitization.EnableCaloHSTruthRecoInputs)
193 kwargs.setdefault("RandomSeedOffset", flags.Digitization.RandomSeedOffset)
194 if (not flags.Digitization.HighGainFCal) and (not flags.Common.isOverlay):
195 kwargs.setdefault("HighGainThreshFCAL", 0)
196 kwargs.setdefault("GainRangeFCAL",[int(CaloGain.LARMEDIUMGAIN),int(CaloGain.LARLOWGAIN)])
197 if (not flags.Digitization.HighGainEMECIW) and (not flags.Common.isOverlay):
198 kwargs.setdefault("HighGainThreshEMECIW", 0)
199 kwargs.setdefault("GainRangeEMECIW",[int(CaloGain.LARMEDIUMGAIN),int(CaloGain.LARLOWGAIN)])
200 kwargs.setdefault("RndmEvtOverlay", flags.Common.isOverlay)
201 if flags.Common.ProductionStep == ProductionStep.PileUpPresampling:
202 kwargs.setdefault("DigitContainer", flags.Overlay.BkgPrefix + "LArDigitContainer_MC")
204 kwargs.setdefault("DigitContainer", "LArDigitContainer_MC") # FIXME - should not be hard-coded
205 # if doing MC+MC overlay
206 if flags.Common.isOverlay and flags.Input.isMC:
207 kwargs.setdefault("isMcOverlay", True)
208 kwargs.setdefault("Nsamples", flags.LAr.ROD.nSamples)
209 kwargs.setdefault("firstSample", #Need to set a negative value to include preceeding samples
210 -flags.LAr.ROD.nPreceedingSamples if flags.LAr.ROD.nPreceedingSamples!=0 else flags.LAr.ROD.FirstSample)
211 LArHitEMapToDigitAlg = CompFactory.LArHitEMapToDigitAlg
212 acc.addEventAlgo(LArHitEMapToDigitAlg(name, **kwargs))
215def LArOutputCfg(flags):
216 """Return ComponentAccumulator with Output for LAr. Not standalone."""
217 acc = ComponentAccumulator()
218 if flags.Output.doWriteRDO:
220 if flags.Digitization.AddCaloDigi:
221 ItemList.append("LArDigitContainer#*")
222 elif flags.Common.ProductionStep == ProductionStep.PileUpPresampling:
223 ItemList.append("LArDigitContainer#" + flags.Overlay.BkgPrefix + "LArDigitContainer_MC")
224 elif flags.Digitization.AddCaloDigiThinned:
225 ItemList.append("LArDigitContainer#LArDigitContainer_MC_Thinned")
226 if flags.Common.ProductionStep != ProductionStep.PileUpPresampling:
227 ItemList.append("LArRawChannelContainer#LArRawChannels")
228 if flags.Overlay.DataOverlay:
229 ItemList.append("LArFebHeaderContainer#LArFebHeader")
230 if flags.Digitization.EnableTruth:
231 ItemList.append("CaloCalibrationHitContainer#*")
232 acc.merge(TruthDigitizationOutputCfg(flags))
233 acc.merge(OutputStreamCfg(flags, "RDO", ItemList))
237def LArDigitizationBasicCfg(flags, **kwargs):
238 """Return ComponentAccumulator with LAr digitization configuration"""
239 acc = ComponentAccumulator()
240 if "PileUpTools" not in kwargs:
241 PileUpTools = acc.popToolsAndMerge(LArPileUpToolCfg(flags))
242 kwargs["PileUpTools"] = PileUpTools
243 acc.merge(PileUpToolsCfg(flags, **kwargs))
245 acc.merge(LArHitEMapToDigitAlgCfg(flags))
246 if flags.LAr.ROD.NNRawChannelBuilding:
247 acc.merge(LArNNRawChannelBuilderCfg(flags))
248 elif flags.LAr.ROD.OFFCRawChannelBuilding:
249 acc.merge(LArOFFCRawChannelBuilderCfg(flags))
251 acc.merge(LArRawChannelBuilderAlgCfg(flags))
253 if flags.Digitization.AddCaloDigiThinned:
254 acc.merge(LArDigitThinnerCfg(flags))
265def LArOverlayDigitizationBasicCfg(flags, name="digitmaker1", **kwargs):
266 """Return ComponentAccumulator with LAr Overlay digitization configuration"""
267 acc = ComponentAccumulator()
269 if "LArPileUpTool" not in kwargs:
270 kwargs["LArPileUpTool"] = acc.popToolsAndMerge(LArPileUpToolCfg(flags))
272 if flags.Concurrency.NumThreads > 0:
273 kwargs.setdefault('Cardinality', flags.Concurrency.NumThreads)
275 LArHitEMapMaker = CompFactory.LArHitEMapMaker
276 acc.addEventAlgo(LArHitEMapMaker(name, **kwargs))
277 acc.merge(LArHitEMapToDigitAlgCfg(flags))
279 acc.merge(LArRawChannelBuilderAlgCfg(flags))
280 if flags.Digitization.AddCaloDigiThinned:
281 acc.merge(LArDigitThinnerCfg(flags))
292def LArAutoCorrNoiseCondSCAlgCfg(flags, **kwargs):
293 acc = ComponentAccumulator()
295 requiredConditions=["AutoCorrSC"]
296 from LArConfiguration.LArElecCalibDBConfig import LArElecCalibDBMCSCCfg
297 acc.merge(LArElecCalibDBMCSCCfg(flags,requiredConditions))
298 kwargs.setdefault("LArAutoCorrObjKey", "LArAutoCorrSC") # Provided by LArFlatConditionsAlg<LArAutoCorrSC>
300 from LArCabling.LArCablingConfig import LArOnOffIdMappingSCCfg
301 acc.merge(LArOnOffIdMappingSCCfg(flags))
302 kwargs.setdefault("LArOnOffIdMappingObjKey", 'LArOnOffIdMapSC')
303 kwargs.setdefault("nSampl", flags.LAr.ROD.nSamples + 2) # See ATLASSIM-5483
304 kwargs.setdefault("isSuperCell", True)
305 kwargs.setdefault("LArAutoCorrNoiseObjKey", "LArAutoCorrNoiseSC") # output
306 acc.addCondAlgo(CompFactory.LArAutoCorrNoiseCondAlg(name="LArAutoCorrNoiseSCCondAlg", **kwargs))