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