ATLAS Offline Software
Loading...
Searching...
No Matches
ISF_FatrasConfig.py
Go to the documentation of this file.
1# Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
2
3__doc__ = "New configuration for the ISF_FatrasSimTool"
4
5from AthenaCommon.Logging import logging
6from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
7from AthenaConfiguration.ComponentFactory import CompFactory
8from ISF_Algorithms.CollectionMergerConfig import CollectionMergerCfg
9from ISF_Tools.ISF_ToolsConfig import ParticleHelperCfg
10from ISF_Services.ISF_ServicesConfig import (
11 ATLFAST_ParticleBrokerSvcCfg, TruthServiceCfg
12)
13from ISF_Geant4Tools.ISF_Geant4ToolsConfig import G4RunManagerHelperCfg
14from RngComps.RngCompsConfig import dSFMT, AthRNGSvcCfg
15
16def TrkExRndSvcCfg(flags):
17 seed = 'TrkExRnd OFFSET 0 12412330 37849324'
18 return dSFMT(seed)
19
20
21def TrkExRndSvcMTCfg(flags):
22 return AthRNGSvcCfg(flags)
23
24
25def FatrasRndSvcCfg(flags):
26 seed = 'FatrasRnd OFFSET 0 81234740 23474923'
27 return dSFMT(seed)
28
29
30
33
34
35# Fatras Hadronic Interaction Processor
36# hadronic interaction creator
37def fatrasHitCreatorPixelCfg(flags, name="ISF_FatrasHitCreatorPixel", **kwargs):
38 """Return ISF_FatrasHitCreatorPixel configured with ComponentAccumulator"""
39 mlog = logging.getLogger(name)
40 mlog.debug('Start configuration')
41
42 bare_collection_name = "PixelHits"
43 mergeable_collection_suffix = "_Fatras"
44 merger_input_property = "PixelHits"
45 region = "ID"
46
47 result, hits_collection_name = CollectionMergerCfg(flags,
48 bare_collection_name,
49 mergeable_collection_suffix,
50 merger_input_property,
51 region)
52
53 kwargs.setdefault("RandomNumberService", result.getPrimaryAndMerge(FatrasRndSvcCfg(flags)))
54 kwargs.setdefault("RandomStreamName", flags.Sim.Fatras.RandomStreamName)
55 kwargs.setdefault("IdHelperName", 'PixelID')
56 kwargs.setdefault("CollectionName", hits_collection_name)
57
58 kwargs.setdefault("ConditionsTool", "")
59 kwargs.setdefault("UseConditionsTool", False)
60
61 result.setPrivateTools(CompFactory.iFatras.HitCreatorSilicon(name=name, **kwargs))
62 return result
63
64
65def fatrasHitCreatorSCTCfg(flags, name="ISF_FatrasHitCreatorSCT", **kwargs):
66 """Return ISF_FatrasHitCreatorSCT configured with ComponentAccumulator"""
67 mlog = logging.getLogger(name)
68 mlog.debug('Start configuration')
69
70 bare_collection_name = "SCT_Hits"
71 mergeable_collection_suffix = "_Fatras"
72 merger_input_property = "SCTHits"
73 region = "ID"
74
75 result, hits_collection_name = CollectionMergerCfg(flags,
76 bare_collection_name,
77 mergeable_collection_suffix,
78 merger_input_property,
79 region)
80
81 kwargs.setdefault("RandomNumberService", result.getPrimaryAndMerge(FatrasRndSvcCfg(flags)))
82 kwargs.setdefault("RandomStreamName", flags.Sim.Fatras.RandomStreamName)
83 kwargs.setdefault("IdHelperName", 'SCT_ID')
84 kwargs.setdefault("CollectionName", hits_collection_name)
85
86 kwargs.setdefault("ConditionsTool", "")
87 kwargs.setdefault("UseConditionsTool", False)
88
89 result.setPrivateTools(CompFactory.iFatras.HitCreatorSilicon(name=name, **kwargs))
90 return result
91
92
93def fatrasHitCreatorTRTCfg(flags, name="ISF_FatrasHitCreatorTRT", **kwargs):
94 """Return ISF_FatrasHitCreatorTRT configured with ComponentAccumulator"""
95 mlog = logging.getLogger(name)
96 mlog.debug('Start configuration')
97
98 bare_collection_name = "TRTUncompressedHits"
99 mergeable_collection_suffix = "_Fatras"
100 merger_input_property = "TRTUncompressedHits"
101 region = "ID"
102
103 result, hits_collection_name = CollectionMergerCfg(flags,
104 bare_collection_name,
105 mergeable_collection_suffix,
106 merger_input_property,
107 region)
108
109 kwargs.setdefault("RandomNumberService", result.getPrimaryAndMerge(FatrasRndSvcCfg(flags)))
110 kwargs.setdefault("RandomStreamName", flags.Sim.Fatras.RandomStreamName)
111 kwargs.setdefault("CollectionName", hits_collection_name)
112
113 kwargs.setdefault("StrawStatusSummaryTool", "")
114 result.setPrivateTools(CompFactory.iFatras.HitCreatorTRT(name=name, **kwargs))
115 return result
116
117
118def fatrasPileupHitCreatorPixelCfg(flags, name="ISF_FatrasPileupHitCreatorPixel", **kwargs):
119 """Return ISF_FatrasHitCreatorPixel configured for pileup with ComponentAccumulator"""
120 kwargs.setdefault("CollectionName", "PileupPixelHits")
121 return fatrasHitCreatorPixelCfg(flags, name, **kwargs)
122
123
124def fatrasPileupHitCreatorSCTCfg(flags, name="ISF_FatrasPileupHitCreatorSCT", **kwargs):
125 """Return ISF_FatrasHitCreatorSCT configured for pileup with ComponentAccumulator"""
126 kwargs.setdefault("CollectionName", "PileupSCT_Hits")
127 return fatrasHitCreatorSCTCfg(flags, name, **kwargs)
128
129
130def fatrasPileupHitCreatorTRTCfg(flags, name="ISF_FatrasPileupHitCreatorTRT", **kwargs):
131 """Return ISF_FatrasHitCreatorTRT configured with ComponentAccumulator"""
132 kwargs.setdefault("CollectionName", "PileupTRTUncompressedHits")
133 return fatrasHitCreatorTRTCfg(flags, name, **kwargs)
134
135
136################################################################################
137# TRACK CREATION SECTION
138################################################################################
139def fatrasSimHitCreatorIDCfg(flags, name="ISF_FatrasSimHitCreatorID", **kwargs):
140 """Return ISF_FatrasSimHitCreatorID configured with ComponentAccumulator"""
141
142 mlog = logging.getLogger(name)
143 mlog.debug('Start configuration')
144
145 result = ComponentAccumulator()
146 kwargs.setdefault("PixelHitCreator", result.addPublicTool(result.popToolsAndMerge(fatrasHitCreatorPixelCfg(flags))))
147 kwargs.setdefault("SctHitCreator", result.addPublicTool(result.popToolsAndMerge(fatrasHitCreatorSCTCfg(flags))))
148 kwargs.setdefault("TrtHitCreator", result.addPublicTool(result.popToolsAndMerge(fatrasHitCreatorTRTCfg(flags))))
149 kwargs.setdefault("OutputLevel", flags.Exec.OutputLevel)
150 result.setPrivateTools(CompFactory.iFatras.SimHitCreatorID(name=name, **kwargs))
151 return result
152
153
154def fatrasPileupSimHitCreatorIDCfg(flags, name="ISF_FatrasPileupSimHitCreatorID", **kwargs):
155 """Return ISF_FatrasSimHitCreatorID configured for pileup with ComponentAccumulator"""
156
157 mlog = logging.getLogger(name)
158 mlog.debug('Start configuration')
159
160 result = ComponentAccumulator()
161 kwargs.setdefault("PixelHitCreator", result.addPublicTool(result.popToolsAndMerge(fatrasPileupHitCreatorPixelCfg(flags))))
162 kwargs.setdefault("SctHitCreator", result.addPublicTool(result.popToolsAndMerge(fatrasPileupHitCreatorSCTCfg(flags))))
163 kwargs.setdefault("TrtHitCreator", result.addPublicTool(result.popToolsAndMerge(fatrasPileupHitCreatorTRTCfg(flags))))
164 return fatrasSimHitCreatorIDCfg(flags, name, **kwargs)
165
166
167def fatrasSimHitCreatorMSCfg(flags, name="ISF_FatrasSimHitCreatorMS", **kwargs):
168 """Return ISF_FatrasSimHitCreatorMS configured with ComponentAccumulator"""
169
170 mlog = logging.getLogger(name)
171 mlog.debug('Start configuration')
172
173
174 result = ComponentAccumulator()
175 mergeable_collection_suffix = "_Fatras"
176 region = "MUON"
177
178 mdt_bare_collection_name="MDT_Hits"
179 mdt_merger_input_property="MDTHits"
180 mdt_result, mdt_hits_collection_name = CollectionMergerCfg(flags,
181 mdt_bare_collection_name,
182 mergeable_collection_suffix,
183 mdt_merger_input_property,
184 region)
185 result.merge(mdt_result)
186
187 rpc_bare_collection_name="RPC_Hits"
188 rpc_merger_input_property="RPCHits"
189 rpc_result, rpc_hits_collection_name = CollectionMergerCfg(flags,
190 rpc_bare_collection_name,
191 mergeable_collection_suffix,
192 rpc_merger_input_property,
193 region)
194 result.merge(rpc_result)
195
196 tgc_bare_collection_name="TGC_Hits"
197 tgc_merger_input_property="TGCHits"
198 tgc_result, tgc_hits_collection_name = CollectionMergerCfg(flags,
199 tgc_bare_collection_name,
200 mergeable_collection_suffix,
201 tgc_merger_input_property,
202 region)
203 result.merge(tgc_result)
204
205 csc_hits_collection_name = ""
206 if flags.Detector.EnableCSC:
207 csc_bare_collection_name="CSC_Hits"
208 csc_merger_input_property="CSCHits"
209 csc_result, csc_hits_collection_name = CollectionMergerCfg(flags,
210 csc_bare_collection_name,
211 mergeable_collection_suffix,
212 csc_merger_input_property,
213 region)
214 result.merge(csc_result)
215
216 stgc_hits_collection_name = ""
217 if flags.Detector.EnablesTGC:
218 stgc_bare_collection_name="sTGC_Hits"
219 stgc_merger_input_property="sTGCHits"
220 stgc_result, stgc_hits_collection_name = CollectionMergerCfg(flags,
221 stgc_bare_collection_name,
222 mergeable_collection_suffix,
223 stgc_merger_input_property,
224 region)
225 result.merge(stgc_result)
226
227 mm_hits_collection_name = ""
228 if flags.Detector.EnableMM:
229 mm_bare_collection_name="MM_Hits"
230 mm_merger_input_property="MMHits"
231 mm_result, mm_hits_collection_name = CollectionMergerCfg(flags,
232 mm_bare_collection_name,
233 mergeable_collection_suffix,
234 mm_merger_input_property,
235 region)
236 result.merge(mm_result)
237
238 kwargs.setdefault("RandomNumberService", result.getPrimaryAndMerge(FatrasRndSvcCfg(flags)))
239 kwargs.setdefault("RandomStreamName", flags.Sim.Fatras.RandomStreamName)
240
241 kwargs.setdefault("Extrapolator" , result.addPublicTool(result.popToolsAndMerge(fatrasExtrapolatorCfg(flags))))
242
243 kwargs.setdefault("MDTCollectionName", mdt_hits_collection_name)
244 kwargs.setdefault("RPCCollectionName", rpc_hits_collection_name)
245 kwargs.setdefault("TGCCollectionName", tgc_hits_collection_name)
246 kwargs.setdefault("CSCCollectionName", csc_hits_collection_name)
247 kwargs.setdefault("sTGCCollectionName", stgc_hits_collection_name)
248 kwargs.setdefault("MMCollectionName", mm_hits_collection_name)
249
250 Muon__MuonTGMeasurementTool = CompFactory.Muon.MuonTGMeasurementTool
251 muon_tgmeasurement_tool = Muon__MuonTGMeasurementTool(name='MuonTGMeasurementTool',
252 UseDSManager=True)
253 kwargs.setdefault("MeasurementTool", muon_tgmeasurement_tool)
254
255 result.setPrivateTools(CompFactory.iFatras.SimHitCreatorMS(name=name, **kwargs))
256 return result
257
258
259def fatrasPdgG4ParticleCfg(flags, name="ISF_FatrasPdgG4Particle", **kwargs):
260 mlog = logging.getLogger(name)
261 mlog.debug('Start configuration')
262
263 result = ComponentAccumulator()
264
265 result.setPrivateTools(CompFactory.iFatras.PDGToG4Particle(name=name, **kwargs))
266 return result
267
268######################################################################################
269# validation & process sampling
270######################################################################################
271
272
273def fatrasPhysicsValidationToolCfg(flags, name="ISF_FatrasPhysicsValidationTool", **kwargs):
274 mlog = logging.getLogger(name)
275 mlog.debug('Start configuration')
276
277 result = ComponentAccumulator()
278 kwargs.setdefault("ValidationStreamName", "ISFFatras")
279
280 result.setPrivateTools(CompFactory.iFatras.PhysicsValidationTool(name=name, **kwargs))
281 return result
282
283
284def fatrasParticleDecayHelperCfg(flags, name="ISF_FatrasParticleDecayHelper", **kwargs):
285 mlog = logging.getLogger(name)
286 mlog.debug('Start configuration')
287
288 result = ComponentAccumulator()
289
290 seed = 'FatrasG4 OFFSET 0 23491234 23470291'
291 result.merge(dSFMT(seed))
292 kwargs.setdefault("RandomNumberService", result.getService("AtDSFMTGenSvc"))
293 kwargs.setdefault("RandomStreamName", flags.Sim.Fatras.RandomStreamName)
294 kwargs.setdefault("G4RandomStreamName", flags.Sim.Fatras.G4RandomStreamName)
295 kwargs.setdefault("ValidationMode", flags.Sim.ISF.ValidationMode)
296 if flags.Sim.ISF.ValidationMode:
297 kwargs.setdefault("PhysicsValidationTool", result.addPublicTool(result.popToolsAndMerge(fatrasPhysicsValidationToolCfg(flags))))
298
299 if "ParticleBroker" not in kwargs:
300 kwargs.setdefault("ParticleBroker", result.getPrimaryAndMerge(ATLFAST_ParticleBrokerSvcCfg(flags)))
301
302 if "TruthRecordSvc" not in kwargs:
303 kwargs.setdefault("ParticleTruthSvc", result.getPrimaryAndMerge(TruthServiceCfg(flags)))
304
305 kwargs.setdefault("PDGToG4ParticleConverter", result.addPublicTool(result.popToolsAndMerge(fatrasPdgG4ParticleCfg(flags))))
306 kwargs.setdefault("G4RunManagerHelper", result.addPublicTool(result.popToolsAndMerge(G4RunManagerHelperCfg(flags))))
307 result.setPrivateTools(CompFactory.iFatras.G4ParticleDecayHelper(name=name, **kwargs))
308 return result
309
310
311################################################################################
312# Extrapolator
313################################################################################
314# the definition of an extrapolator (to be cleaned up)
315
316def fatrasEnergyLossUpdatorCfg(flags, name="ISF_FatrasEnergyLossUpdator", **kwargs):
317 mlog = logging.getLogger(name)
318 mlog.debug('Start configuration')
319
320 result = ComponentAccumulator()
321
322 kwargs.setdefault("RandomNumberService", result.getPrimaryAndMerge(FatrasRndSvcCfg(flags)))
323 kwargs.setdefault("RandomStreamName", flags.Sim.Fatras.RandomStreamName)
324
325 kwargs.setdefault("UsePDG_EnergyLossFormula", True)
326 kwargs.setdefault("EnergyLossDistribution", 2)
327
328 from TrkConfig.AtlasExtrapolatorToolsConfig import AtlasEnergyLossUpdatorCfg
329 kwargs.setdefault("EnergyLossUpdator", result.popToolsAndMerge(AtlasEnergyLossUpdatorCfg(flags)))
330
331 result.setPrivateTools(CompFactory.iFatras.McEnergyLossUpdator(name=name, **kwargs))
332 return result
333
334
335# Combining all in the MaterialEffectsUpdator
336def fatrasMaterialUpdatorCfg(flags, name="ISF_FatrasMaterialUpdator", **kwargs):
337 mlog = logging.getLogger(name)
338 mlog.debug('Start configuration')
339
340 result = ComponentAccumulator()
341
342 kwargs.setdefault("RandomNumberService", result.getPrimaryAndMerge(FatrasRndSvcCfg(flags)))
343 kwargs.setdefault("RandomStreamName", flags.Sim.Fatras.RandomStreamName)
344 if "ParticleBroker" not in kwargs:
345 kwargs.setdefault("ParticleBroker", result.getPrimaryAndMerge(ATLFAST_ParticleBrokerSvcCfg(flags)))
346
347 if "TruthRecordSvc" not in kwargs:
348 kwargs.setdefault("TruthRecordSvc", result.getPrimaryAndMerge(TruthServiceCfg(flags)))
349
350 # @TODO retire once migration to TrackingGeometry conditions data is complete
351 if not flags.Sim.ISF.UseTrackingGeometryCond:
352 if 'TrackingGeometrySvc' not in kwargs:
353 from TrkConfig.AtlasTrackingGeometrySvcConfig import TrackingGeometrySvcCfg
354 kwargs.setdefault("TrackingGeometrySvc", result.getPrimaryAndMerge(TrackingGeometrySvcCfg(flags)))
355 kwargs.setdefault("TrackingGeometryReadKey", '')
356 else:
357 if 'TrackingGeometryReadKey' not in kwargs:
358 from TrackingGeometryCondAlg.AtlasTrackingGeometryCondAlgConfig import TrackingGeometryCondAlgCfg
359 acc = TrackingGeometryCondAlgCfg(flags)
360 geom_cond_key = acc.getPrimary().TrackingGeometryWriteKey
361 result.merge(acc)
362 kwargs.setdefault("TrackingGeometryReadKey", geom_cond_key)
363
364 # hadronic interactions
365 kwargs.setdefault("HadronicInteraction", True)
366
367 kwargs.setdefault("HadronicInteractionProcessor", result.addPublicTool(result.popToolsAndMerge(fatrasG4HadIntProcessorCfg(flags))))
368
369 # energy loss
370 kwargs.setdefault("EnergyLoss", True)
371 kwargs.setdefault("EnergyLossUpdator", result.addPublicTool(result.popToolsAndMerge(fatrasEnergyLossUpdatorCfg(flags))))
372
373 # mutiple scattering
374 kwargs.setdefault("MultipleScattering", True)
375 from TrkConfig.AtlasExtrapolatorToolsConfig import fatrasMultipleScatteringUpdatorCfg
376 kwargs.setdefault("MultipleScatteringUpdator", result.addPublicTool(result.popToolsAndMerge(fatrasMultipleScatteringUpdatorCfg(flags))))
377
378 # photon conversion
379 kwargs.setdefault("PhotonConversionTool", result.addPublicTool(result.popToolsAndMerge(fatrasConversionCreatorCfg(flags))))
380
381 # the validation output
382 kwargs.setdefault("ValidationMode", flags.Sim.ISF.ValidationMode)
383 kwargs.setdefault("BremPhotonValidation", False)
384 kwargs.setdefault("EnergyDepositValidation", False)
385
386 kwargs.setdefault("MomentumCut", flags.Sim.Fatras.MomCutOffSec)
387 kwargs.setdefault("MinimumBremPhotonMomentum", flags.Sim.Fatras.MomCutOffSec)
388
389 if flags.Sim.ISF.ValidationMode:
390 kwargs.setdefault("PhysicsValidationTool", result.addPublicTool(result.popToolsAndMerge(fatrasPhysicsValidationToolCfg(flags))))
391
392 kwargs.setdefault("ProcessSamplingTool", result.addPublicTool(result.popToolsAndMerge(fatrasProcessSamplingToolCfg(flags))))
393
394 kwargs.setdefault("ParticleDecayHelper", result.addPublicTool(result.popToolsAndMerge(fatrasParticleDecayHelperCfg(flags))))
395
396 # MCTruth Process Code
397 kwargs.setdefault("BremProcessCode", 3) # TODO: to be taken from central definition
398
399 result.setPrivateTools(CompFactory.iFatras.McMaterialEffectsUpdator(name=name, **kwargs))
400 return result
401
402
403def fatrasExtrapolatorCfg(flags, name="ISF_FatrasExtrapolator", **kwargs):
404 mlog = logging.getLogger(name)
405 mlog.debug('Start configuration')
406
407 result = ComponentAccumulator()
408
409 # Charged Transport Tool
410 # assign the tools
411 from TrkConfig.AtlasExtrapolatorToolsConfig import FastSimNavigatorCfg
412 kwargs.setdefault("Navigator", result.addPublicTool(result.popToolsAndMerge(FastSimNavigatorCfg(flags))))
413
414 kwargs.setdefault("MaterialEffectsUpdators", [result.addPublicTool(result.popToolsAndMerge(fatrasMaterialUpdatorCfg(flags)))])
415
416 from TrkConfig.TrkExRungeKuttaPropagatorConfig import RungeKuttaPropagatorCfg
417 kwargs.setdefault("Propagators",
418 [result.addPublicTool(result.popToolsAndMerge(RungeKuttaPropagatorCfg(flags, name="ISF_FatrasChargedPropagator")))])
419
420 from TrkConfig.TrkExSTEP_PropagatorConfig import fatrasSTEP_PropagatorCfg
421 kwargs.setdefault("STEP_Propagator", result.addPublicTool(result.popToolsAndMerge(fatrasSTEP_PropagatorCfg(flags))))
422
423 # Fatras specific: stop the trajectory
424 kwargs.setdefault("StopWithNavigationBreak", True)
425 kwargs.setdefault("StopWithUpdateKill", True)
426 kwargs.setdefault("RobustSampling", True)
427 kwargs.setdefault("ResolveMuonStation", True)
428 kwargs.setdefault("UseMuonMatApproximation", True)
429
430 result.setPrivateTools(CompFactory.Trk.TimedExtrapolator(name=name, **kwargs))
431 return result
432
433
434################################################################################
435# SIMULATION TOOL and SERVICE
436################################################################################
437def fatrasKinematicFilterCfg(flags, name="ISF_FatrasKinematicFilter", **kwargs):
438 mlog = logging.getLogger(name)
439 mlog.debug('Start configuration')
440
441 result = ComponentAccumulator()
442
443 kwargs.setdefault("MaxEtaSymmetric", 10.)
444 kwargs.setdefault("MinMomentum", flags.Sim.Fatras.MomCutOffSec)
445
446 result.setPrivateTools(CompFactory.ISF.KinematicParticleFilter(name=name, **kwargs))
447 return result
448
449
450def fatrasConversionCreatorCfg(flags, name="ISF_FatrasConversionCreator", **kwargs):
451 mlog = logging.getLogger(name)
452 mlog.debug('Start configuration')
453
454 result = ComponentAccumulator()
455
456 kwargs.setdefault("RandomNumberService", result.getPrimaryAndMerge(FatrasRndSvcCfg(flags)))
457 kwargs.setdefault("RandomStreamName", flags.Sim.Fatras.RandomStreamName)
458
459 if "ParticleBroker" not in kwargs:
460 kwargs.setdefault("ParticleBroker", result.getPrimaryAndMerge(ATLFAST_ParticleBrokerSvcCfg(flags)))
461
462 if "TruthRecordSvc" not in kwargs:
463 kwargs.setdefault("TruthRecordSvc", result.getPrimaryAndMerge(TruthServiceCfg(flags)))
464
465 if flags.Sim.ISF.ValidationMode:
466 kwargs.setdefault("PhysicsValidationTool", result.addPublicTool(result.popToolsAndMerge(fatrasPhysicsValidationToolCfg(flags))))
467
468 kwargs.setdefault("PhysicsProcessCode", 14) # TODO: to be taken from central definition
469 kwargs.setdefault("ValidationMode", flags.Sim.ISF.ValidationMode)
470
471 result.setPrivateTools(CompFactory.iFatras.PhotonConversionTool(name=name, **kwargs))
472 return result
473
474
475def fatrasG4HadIntProcessorCfg(flags, name="ISF_FatrasG4HadIntProcessor", **kwargs):
476 mlog = logging.getLogger(name)
477 mlog.debug('Start configuration')
478
479 result = ComponentAccumulator()
480
481 kwargs.setdefault("RandomNumberService", result.getPrimaryAndMerge(FatrasRndSvcCfg(flags)))
482 kwargs.setdefault("RandomStreamName", flags.Sim.Fatras.RandomStreamName)
483
484 if "ParticleBroker" not in kwargs:
485 kwargs.setdefault("ParticleBroker", result.getPrimaryAndMerge(ATLFAST_ParticleBrokerSvcCfg(flags)))
486
487 if "TruthRecordSvc" not in kwargs:
488 kwargs.setdefault("TruthRecordSvc", result.getPrimaryAndMerge(TruthServiceCfg(flags)))
489
490 kwargs.setdefault("MomentumCut", flags.Sim.Fatras.MomCutOffSec)
491
492 kwargs.setdefault("G4RunManagerHelper", result.addPublicTool(result.popToolsAndMerge(G4RunManagerHelperCfg(flags))))
493
494 result.setPrivateTools(CompFactory.iFatras.G4HadIntProcessor(name=name, **kwargs))
495 return result
496
497
498# Fatras Hadronic Interaction Processor
499def fatrasParametricHadIntProcessorCfg(flags, name="ISF_FatrasParametricHadIntProcessor", **kwargs):
500 mlog = logging.getLogger(name)
501 mlog.debug('Start configuration')
502
503 result = ComponentAccumulator()
504
505 kwargs.setdefault("RandomNumberService", result.getPrimaryAndMerge(FatrasRndSvcCfg(flags)))
506 kwargs.setdefault("RandomStreamName", flags.Sim.Fatras.RandomStreamName)
507
508 if "ParticleBroker" not in kwargs:
509 kwargs.setdefault("ParticleBroker", result.getPrimaryAndMerge(ATLFAST_ParticleBrokerSvcCfg(flags)))
510
511 if "TruthRecordSvc" not in kwargs:
512 kwargs.setdefault("TruthRecordSvc", result.getPrimaryAndMerge(TruthServiceCfg(flags)))
513
514 kwargs.setdefault("HadronicInteractionScaleFactor", flags.Sim.Fatras.HadronIntProb)
515 kwargs.setdefault("MinimumHadronicInitialEnergy", flags.Sim.Fatras.MomCutOffSec)
516 kwargs.setdefault("MinimumHadronicOutEnergy", flags.Sim.Fatras.MomCutOffSec)
517 kwargs.setdefault("HadronicInteractionValidation", False)
518 kwargs.setdefault("PhysicsProcessCode", 121) # TODO: to be taken from central definition
519 kwargs.setdefault("ValidationMode", flags.Sim.ISF.ValidationMode)
520 if flags.Sim.ISF.ValidationMode:
521 kwargs.setdefault("PhysicsValidationTool", result.addPublicTool(result.popToolsAndMerge(fatrasPhysicsValidationToolCfg(flags))))
522 result.setPrivateTools(CompFactory.iFatras.HadIntProcessorParametric(name, **kwargs))
523 return result
524
525
526def fatrasProcessSamplingToolCfg(flags, name="ISF_FatrasProcessSamplingTool", **kwargs):
527 mlog = logging.getLogger(name)
528 mlog.debug('Start configuration')
529 result = ComponentAccumulator()
530 result.setPrivateTools(CompFactory.iFatras.ProcessSamplingTool(name=name, **kwargs))
531 return result
532
533
534def fatrasTransportToolCfg(flags, name="ISF_FatrasSimTool", **kwargs):
535 mlog = logging.getLogger(name)
536 mlog.debug('Start configuration')
537
538 result = ComponentAccumulator()
539
540 if "SimHitCreatorID" not in kwargs:
541 kwargs.setdefault("SimHitCreatorID", result.addPublicTool(result.popToolsAndMerge(fatrasSimHitCreatorIDCfg(flags))))
542
543 kwargs.setdefault("SimHitCreatorMS", result.addPublicTool(result.popToolsAndMerge(fatrasSimHitCreatorMSCfg(flags))))
544
545 kwargs.setdefault("ParticleDecayHelper", result.addPublicTool(result.popToolsAndMerge(fatrasParticleDecayHelperCfg(flags))))
546
547 kwargs.setdefault("ParticleHelper", result.addPublicTool(result.popToolsAndMerge(ParticleHelperCfg(flags))))
548
549 publicKinFilter = result.addPublicTool(result.popToolsAndMerge(fatrasKinematicFilterCfg(flags)))
550 kwargs.setdefault("TrackFilter", publicKinFilter)
551 kwargs.setdefault("NeutralFilter", publicKinFilter)
552 kwargs.setdefault("PhotonFilter", publicKinFilter)
553
554 kwargs.setdefault("Extrapolator", result.addPublicTool(result.popToolsAndMerge(fatrasExtrapolatorCfg(flags))))
555
556 if flags.Sim.ISF.ValidationMode:
557 kwargs.setdefault("PhysicsValidationTool", result.addPublicTool(result.popToolsAndMerge(fatrasPhysicsValidationToolCfg(flags))))
558
559 kwargs.setdefault("ProcessSamplingTool", result.addPublicTool(result.popToolsAndMerge(fatrasProcessSamplingToolCfg(flags))))
560
561 kwargs.setdefault("OutputLevel", flags.Exec.OutputLevel)
562 kwargs.setdefault("ValidationOutput", flags.Sim.ISF.ValidationMode)
563
564 kwargs.setdefault("RandomNumberService", result.getPrimaryAndMerge(FatrasRndSvcCfg(flags)))
565
566 result.setPrivateTools(CompFactory.iFatras.TransportTool(name=name, **kwargs))
567 return result
568
569
570def fatrasPileupSimToolCfg(flags, name="ISF_FatrasPileupSimTool", **kwargs):
571 mlog = logging.getLogger(name)
572 mlog.debug('Start configuration')
573
574 result = ComponentAccumulator()
575
576 kwargs.setdefault("SimHitCreatorID", result.addPublicTool(result.popToolsAndMerge(fatrasPileupSimHitCreatorIDCfg(flags))))
577
578 result.setPrivateTools(result.popToolsAndMerge(fatrasTransportToolCfg(flags, name, **kwargs)))
579
580 return result
581
582
583# FatrasSimulatorTool
584def fatrasSimulatorToolSTCfg(flags, name="ISF_FatrasSimulatorToolST", **kwargs):
585 mlog = logging.getLogger(name)
586 mlog.debug('Start configuration')
587
588 result = ComponentAccumulator()
589
590 if "IDSimulationTool" not in kwargs or "SimulationTool" not in kwargs:
591 publicTransportTool = result.addPublicTool(result.popToolsAndMerge(fatrasTransportToolCfg(flags, **kwargs)))
592 kwargs.setdefault("IDSimulationTool", publicTransportTool)
593 kwargs.setdefault("SimulationTool", publicTransportTool)
594
595 result.setPrivateTools(CompFactory.ISF.FatrasSimTool(name, **kwargs))
596 return result
597
598
599def fatrasPileupSimulatorToolSTCfg(flags, name="ISF_FatrasPileupSimulatorToolST", **kwargs):
600 mlog = logging.getLogger(name)
601 mlog.debug('Start configuration')
602
603 result = ComponentAccumulator()
604
605 kwargs.setdefault("IDSimulationTool", result.addPublicTool(result.popToolsAndMerge(fatrasPileupSimToolCfg(flags))))
606 kwargs.setdefault("SimulationTool", result.addPublicTool(result.popToolsAndMerge(fatrasTransportToolCfg(flags))))
607
608 result.setPrivateTools(result.popToolsAndMerge(fatrasSimulatorToolSTCfg(flags, name, **kwargs)))
609 return result
610
611
612# FatrasSimulationSvc
613def fatrasSimServiceIDCfg(flags, name="ISF_FatrasSimSvc", **kwargs):
614 mlog = logging.getLogger(name)
615 mlog.debug('Start configuration')
616
617 result = ComponentAccumulator()
618 kwargs.setdefault("Identifier", "Fatras")
619 kwargs.setdefault("SimulatorTool", result.addPublicTool(result.popToolsAndMerge(fatrasSimulatorToolSTCfg(flags))))
620 result.addService(CompFactory.ISF.LegacySimSvc(name, **kwargs), primary = True)
621 return result
622
623
624def fatrasPileupSimServiceIDCfg(flags, name="ISF_FatrasPileupSimSvc", **kwargs):
625 mlog = logging.getLogger(name)
626 mlog.debug('Start configuration')
627
628 toolAcc = ComponentAccumulator()
629 kwargs.setdefault("SimulatorTool", toolAcc.addPublicTool(toolAcc.popToolsAndMerge(fatrasPileupSimulatorToolSTCfg(flags))))
630 result = fatrasSimServiceIDCfg(flags, name, **kwargs)
631 result.merge(toolAcc)
632 return result
633
634
635if __name__ == "__main__":
636 from AthenaConfiguration.AllConfigFlags import initConfigFlags
637 from AthenaCommon.Logging import log
638 from AthenaCommon.Constants import DEBUG
639 from AthenaConfiguration.TestDefaults import defaultTestFiles
640
641 log.setLevel(DEBUG)
642
643 flags = initConfigFlags()
644 flags.Input.isMC = True
645 flags.Input.Files = defaultTestFiles.HITS_RUN2
646 flags.Exec.MaxEvents = 3
647 flags.fillFromArgs()
648 flags.lock()
649
650 from AthenaConfiguration.MainServicesConfig import MainServicesCfg
651 acc = MainServicesCfg(flags)
652 acc.popToolsAndMerge(fatrasTransportToolCfg(flags))
653
654 print("INFO_FatrasConfig: Dumping config flags")
655 flags.dump()
656 print("INFO_FatrasConfig: Print config details")
657 acc.printConfig(withDetails=True, summariseProps=True)
658 acc.store(open('fatrassimtool.pkl', 'wb'))
659
660 sc = acc.run()
661
662 import sys
663 # Success should be 0
664 sys.exit(not sc.isSuccess())
ISF_FatrasConfig.fatrasHitCreatorPixelCfg
fatrasHitCreatorPixelCfg(flags, name="ISF_FatrasHitCreatorPixel", **kwargs)
HIT CREATION SECTION.
Definition ISF_FatrasConfig.py:37
ISF_FatrasConfig.TrkExRndSvcCfg
TrkExRndSvcCfg(flags)
Definition ISF_FatrasConfig.py:16
ISF_FatrasConfig.TrkExRndSvcMTCfg
TrkExRndSvcMTCfg(flags)
Definition ISF_FatrasConfig.py:21
ISF_FatrasConfig.FatrasRndSvcCfg
FatrasRndSvcCfg(flags)
Definition ISF_FatrasConfig.py:25
Generated on for ATLAS Offline Software by doxygen 1.17.0