2 from AthenaConfiguration.ComponentAccumulator
import ComponentAccumulator
3 from AthenaConfiguration.ComponentFactory
import CompFactory
4 from AthenaConfiguration.Enums
import BeamType
5 from SimulationConfig.SimEnums
import BeamPipeSimMode, CalibrationRun, CavernBackground, LArParameterization
6 from AthenaConfiguration.AccumulatorCache
import AccumulatorCache
13 if flags.Sim.LArParameterization
is LArParameterization.FastCaloSim:
14 from G4FastSimulation.G4FastSimulationConfig
import FastCaloSimCfg
17 if flags.Detector.GeometryBpipe:
18 if not flags.Detector.GeometryFwdRegion
and (flags.Detector.GeometryAFP
or flags.Detector.GeometryALFA
or flags.Detector.GeometryZDC):
20 from ForwardTransport.ForwardTransportConfig
import ForwardTransportModelCfg
22 if flags.Sim.BeamPipeSimMode
is not BeamPipeSimMode.Normal:
23 from G4FastSimulation.G4FastSimulationConfig
import SimpleFastKillerCfg
25 if flags.Detector.GeometryLAr:
26 if flags.Sim.LArParameterization
is LArParameterization.NoFrozenShowers:
27 from AthenaCommon.Logging
import logging
28 msg = logging.getLogger(
"FastSimulationToolListCfg")
29 msg.info(
"Not using Frozen Showers" )
31 from LArG4FastSimulation.LArG4FastSimulationConfig
import EMBFastShowerCfg, EMECFastShowerCfg, FCALFastShowerCfg, FCAL2FastShowerCfg
33 if flags.Sim.LArParameterization
is LArParameterization.FrozenShowers:
38 if flags.Sim.LArParameterization
in [LArParameterization.DeadMaterialFrozenShowers, LArParameterization.FrozenShowersFCalOnly, LArParameterization.FastCaloSim]:
39 from G4FastSimulation.G4FastSimulationConfig
import DeadMaterialShowerCfg
42 if flags.Sim.LArParameterization
is LArParameterization.FastCaloSim:
43 from G4FastSimulation.G4FastSimulationConfig
import FastCaloSimCfg
45 if flags.Detector.GeometryMuon:
46 if flags.Sim.CavernBackground
not in [CavernBackground.Off, CavernBackground.Read]
and not flags.Sim.RecordFlux:
47 from TrackWriteFastSim.TrackWriteFastSimConfig
import NeutronFastSimCfg
50 from G4AtlasServices.G4AtlasServicesConfig
import PhysicsListSvcCfg
52 result.getService(
"PhysicsListSvc").FastSimConstructor.InitializeFastSimulation = len(tools) > 0
54 result.setPrivateTools(tools)
60 svc = CompFactory.ThreadPoolSvc(name=
"ThreadPoolSvc")
61 svc.ThreadInitTools += [CompFactory.G4ThreadInitTool()]
69 FastSimulationMasterTool = CompFactory.FastSimulationMasterTool
76 FastSimulationConstructorTool = CompFactory.FastSimulationConstructorTool
83 FastSimulationMasterTool = CompFactory.FastSimulationMasterTool
85 result.setPrivateTools(tool)
92 kwargs.setdefault(
"UseSimplifiedGeo",
bool(flags.Sim.SimplifiedGeoPath))
93 if flags.Sim.SimplifiedGeoPath:
95 kwargs.setdefault(
'SimplifiedWorldLogName',
"WorldLog")
97 kwargs.setdefault(
'TransportLimitVolume',
"Envelope")
99 kwargs.setdefault(
'MaxSteps', 100)
102 kwargs.setdefault(
'TransportLimitVolume',
"MuonSys")
104 kwargs.setdefault(
'MaxSteps', 5000)
106 result.setPrivateTools(CompFactory.G4CaloTransportTool(name, **kwargs))
113 kwargs.setdefault(
"ScalerConfigFileName" ,
"FastCaloSim/MC23/TFCSparam_mpt_classScaler_v04.xml" )
114 kwargs.setdefault(
"NetworkConfigFileName" ,
"FastCaloSim/MC23/TFCSparam_mpt_classNet_v04.json" )
115 kwargs.setdefault(
"CalibratorConfigFileName" ,
"FastCaloSim/MC23/TFCSparam_mpt_classCalib_v04.xml")
117 result.setPrivateTools(CompFactory.PunchThroughG4Classifier(name, **kwargs))
122 from SubDetectorEnvelopes.SubDetectorEnvelopesConfig
import EnvelopeDefSvcCfg
126 kwargs.setdefault(
"FilenameLookupTable" ,
"FastCaloSim/MC23/TFCSparam_mpt_v07.root")
127 kwargs.setdefault(
"FilenameInverseCdf" ,
"FastCaloSim/MC23/TFCSparam_mpt_inverseCdf_v07.xml")
128 kwargs.setdefault(
"FilenameInversePca" ,
"FastCaloSim/MC23/TFCSparam_mpt_inversePca_v07.xml")
129 kwargs.setdefault(
"EnergyFactor" , [ 0.98, 0.831, 0.896, 0.652, 0.717, 1., 0.877, 0.858, 0.919 ] )
130 kwargs.setdefault(
"DoAntiParticles" , [ 0, 1, 0, 1, 1, 0, 0, 0, 0 ] )
131 kwargs.setdefault(
"PunchThroughInitiators" , [ 211, 321, 311, 310, 130, 2212, 2112] )
132 kwargs.setdefault(
"InitiatorsMinEnergy" , [ 65536, 65536, 65536, 65536, 65536, 65536, 65536] )
133 kwargs.setdefault(
"InitiatorsEtaRange" , [ -3.2, 3.2 ] )
134 kwargs.setdefault(
"PunchThroughParticles" , [ 2212, 211, 22, 11, 13, 2112, 321, 310, 130 ] )
135 kwargs.setdefault(
"CorrelatedParticle" , [] )
136 kwargs.setdefault(
"FullCorrelationEnergy" , [ 100000., 100000., 100000., 100000., 0., 100000., 100000., 100000., 100000.] )
137 kwargs.setdefault(
"MinEnergy" , [ 938.3, 135.6, 50., 50., 105.7, 939.6, 493.7, 497.6, 497.6 ] )
138 kwargs.setdefault(
"MaxNumParticles" , [ -1, -1, -1, -1, -1, -1, -1, -1, -1 ] )
139 kwargs.setdefault(
"EnvelopeDefSvc", result.getPrimaryAndMerge(
EnvelopeDefSvcCfg(flags)))
140 kwargs.setdefault(
"BeamPipeRadius", 500.)
142 result.setPrivateTools(CompFactory.PunchThroughG4Tool(name, **kwargs))
149 if "PunchThroughG4Classifier" not in kwargs:
153 if "PunchThroughG4Tool" not in kwargs:
154 kwargs.setdefault(
"PunchThroughG4Tool", result.addPublicTool(result.popToolsAndMerge(
PunchThroughG4ToolCfg(flags))))
156 result.setPrivateTools(CompFactory.PunchThroughSimWrapper(name, **kwargs))
163 if flags.Detector.EnableLucid:
164 from LUCID_G4_SD.LUCID_G4_SDConfig
import LUCID_SensitiveDetectorCfg
166 if flags.Detector.EnableForward:
167 if flags.Detector.EnableZDC:
168 from ZDC_SD.ZDC_SDConfig
import ZDC_FiberSDCfg
170 if flags.Sim.CalibrationRun
in [CalibrationRun.ZDC, CalibrationRun.LArTileZDC]:
171 from ZDC_SD.ZDC_SDConfig
import ZDC_G4CalibSDCfg
173 if flags.Detector.EnableALFA:
174 from ALFA_G4_SD.ALFA_G4_SDConfig
import ALFA_SensitiveDetectorCfg
176 if flags.Detector.EnableAFP:
177 from AFP_G4_SD.AFP_G4_SDConfig
import AFP_SensitiveDetectorCfg
183 result.setPrivateTools(tools)
190 if (flags.Detector.EnableMuon
and flags.Sim.CavernBackground
in [CavernBackground.Write, CavernBackground.WriteWorld])
or flags.Sim.StoppedParticleFile:
191 from TrackWriteFastSim.TrackWriteFastSimConfig
import TrackFastSimSDCfg
193 result.setPrivateTools(tools)
199 if flags.Sim.LArParameterization
is LArParameterization.FastCaloSim:
200 from ISF_FastCaloSimParametrization.ISF_FastCaloSimParametrizationConfig
import CaloCellContainerSDCfg
202 result.setPrivateTools(tools)
209 if flags.Detector.EnableITkPixel:
210 from PixelG4_SD.PixelG4_SDToolConfig
import ITkPixelSensorSDCfg
213 if flags.Detector.EnableITkStrip:
214 from SCT_G4_SD.SCT_G4_SDToolConfig
import ITkStripSensorSDCfg
216 if flags.Detector.EnablePLR:
217 from PixelG4_SD.PixelG4_SDToolConfig
import PLRSensorSDCfg
220 result.setPrivateTools(tools)
228 if flags.Detector.EnableHGTD:
229 from HGTD_G4_SD.HGTD_G4_SDToolConfig
import HgtdSensorSDCfg
233 result.setPrivateTools(tools)
241 if flags.Detector.EnablePixel:
242 from PixelG4_SD.PixelG4_SDToolConfig
import PixelSensorSDCfg
244 if flags.Detector.EnableSCT:
245 from SCT_G4_SD.SCT_G4_SDToolConfig
import SctSensorSDCfg
247 if flags.Detector.EnableTRT:
248 from TRT_G4_SD.TRT_G4_SDToolConfig
import TRTSensitiveDetectorCfg
250 if flags.Detector.EnableBCM:
251 from BCM_G4_SD.BCM_G4_SDToolConfig
import BCMSensorSDCfg
253 from BLM_G4_SD.BLM_G4_SDToolConfig
import BLMSensorSDCfg
256 result.setPrivateTools(tools)
264 if flags.Detector.EnableLAr:
265 from LArG4SD.LArG4SDToolConfig
import LArEMBSensitiveDetectorCfg, LArEMECSensitiveDetectorCfg, LArFCALSensitiveDetectorCfg, LArHECSensitiveDetectorCfg
271 if flags.Detector.EnableMBTS:
272 from MinBiasScintillator.MinBiasScintillatorToolConfig
import MinBiasScintillatorSDCfg
275 if flags.Sim.CalibrationRun
in [CalibrationRun.LAr, CalibrationRun.LArTile, CalibrationRun.LArTileZDC]:
276 from LArG4SD.LArG4SDToolConfig
import LArDeadSensitiveDetectorToolCfg, LArActiveSensitiveDetectorToolCfg, LArInactiveSensitiveDetectorToolCfg
280 elif flags.Sim.CalibrationRun
is CalibrationRun.DeadLAr:
281 from LArG4SD.LArG4SDToolConfig
import LArDeadSensitiveDetectorToolCfg
284 if flags.Detector.EnableTile:
285 if flags.Sim.CalibrationRun
in [CalibrationRun.Tile, CalibrationRun.LArTile, CalibrationRun.LArTileZDC]:
286 from TileGeoG4Calib.TileGeoG4CalibConfig
import TileGeoG4CalibSDCfg
289 from TileGeoG4SD.TileGeoG4SDToolConfig
import TileGeoG4SDCfg
291 if flags.Sim.RecordStepInfo:
292 from ISF_FastCaloSimSD.ISF_FastCaloSimSDToolConfig
import FCS_StepInfoSDToolCfg
295 result.setPrivateTools(tools)
300 if flags.Muon.usePhaseIIGeoSetup:
301 from MuonSensitiveDetectorsR4.SensitiveDetectorsCfg
import SetupSensitiveDetectorsCfg
303 from MuonG4SD.MuonG4SDConfig
import SetupSensitiveDetectorsCfg
309 if flags.Beam.Type
is BeamType.Cosmics
and not flags.Sim.ReadTR:
310 from TrackWriteFastSim.TrackWriteFastSimConfig
import CosmicTRSDCfg
312 result.setPrivateTools(tools)
317 from SimulationConfig.SimEnums
import LArParameterization
318 writtenContainers =[]
319 if flags.Detector.GeometryMuon:
320 if flags.Muon.usePhaseIIGeoSetup:
321 from MuonSensitiveDetectorsR4.SensitiveDetectorsCfg
import SimHitContainerListCfg
324 from MuonG4SD.MuonG4SDConfig
import SimHitContainerListCfg
326 if flags.Detector.GeometryLAr:
327 if (flags.Sim.ISFRun
and flags.Sim.ISF.HITSMergingRequired.get(
'CALO',
True))
or flags.Sim.LArParameterization
is LArParameterization.FastCaloSim:
328 writtenContainers += [(
"LArHitContainer",
"LArHitEMB_G4")]
329 writtenContainers += [(
"LArHitContainer",
"LArHitEMEC_G4")]
330 writtenContainers += [(
"LArHitContainer",
"LArHitFCAL_G4")]
331 writtenContainers += [(
"LArHitContainer",
"LArHitHEC_G4")]
332 if flags.Sim.LArParameterization
is LArParameterization.FastCaloSim:
333 writtenContainers += [(
"LArHitContainer" ,
"LArHitHEC_FastCaloSim")]
334 writtenContainers += [(
"LArHitContainer" ,
"LArHitFCAL_FastCaloSim")]
335 writtenContainers += [(
"LArHitContainer" ,
"LArHitEMEC_FastCaloSim")]
336 writtenContainers += [(
"LArHitContainer" ,
"LArHitEMB_FastCaloSim")]
338 writtenContainers += [(
"LArHitContainer",
"LArHitEMB")]
339 writtenContainers += [(
"LArHitContainer",
"LArHitEMEC")]
340 writtenContainers += [(
"LArHitContainer",
"LArHitFCAL")]
341 writtenContainers += [(
"LArHitContainer",
"LArHitHEC")]
342 if flags.Detector.GeometryTile:
343 if (flags.Sim.ISFRun
and flags.Sim.ISF.HITSMergingRequired.get(
'CALO',
True))
or flags.Sim.LArParameterization
is LArParameterization.FastCaloSim:
344 writtenContainers += [(
"TileHitVector",
"MBTSHits_G4")]
345 writtenContainers += [(
"TileHitVector",
"TileHitVec_G4")]
346 if flags.Sim.LArParameterization
is LArParameterization.FastCaloSim:
347 writtenContainers += [(
"TileHitVector" ,
"TileHitVec_FastCaloSim")]
348 writtenContainers += [(
"TileHitVector" ,
"MBTSHits_FastCaloSim")]
350 writtenContainers += [(
"TileHitVector",
"MBTSHits")]
351 writtenContainers += [(
"TileHitVector",
"TileHitVec")]
352 if flags.Detector.GeometryTRT:
353 if (flags.Sim.ISFRun
and flags.Sim.ISF.HITSMergingRequired.get(
'ID',
True)):
354 writtenContainers += [(
"TRTUncompressedHitCollection",
"TRTUncompressedHits_G4")]
356 writtenContainers += [(
"TRTUncompressedHitCollection",
"TRTUncompressedHits")]
357 if flags.Detector.EnableBCM:
358 if (flags.Sim.ISFRun
and flags.Sim.ISF.HITSMergingRequired.get(
'ID',
True)):
359 writtenContainers += [(
"SiHitCollection",
"BCMHits_G4")]
360 writtenContainers += [(
"SiHitCollection",
"BLMHits_G4")]
362 writtenContainers += [(
"SiHitCollection",
"BCMHits")]
363 writtenContainers += [(
"SiHitCollection",
"BLMHits")]
364 if flags.Detector.EnablePixel:
365 if (flags.Sim.ISFRun
and flags.Sim.ISF.HITSMergingRequired.get(
'ID',
True)):
366 writtenContainers += [(
"SiHitCollection",
"PixelHits_G4")]
368 writtenContainers += [(
"SiHitCollection",
"PixelHits")]
369 if flags.Detector.EnableSCT:
370 if (flags.Sim.ISFRun
and flags.Sim.ISF.HITSMergingRequired.get(
'ID',
True)):
371 writtenContainers += [(
"SiHitCollection",
"SCT_Hits_G4")]
373 writtenContainers += [(
"SiHitCollection",
"SCT_Hits")]
374 if flags.Detector.EnableITkPixel:
375 if (flags.Sim.ISFRun
and flags.Sim.ISF.HITSMergingRequired.get(
'ITk',
True)):
376 writtenContainers += [(
"SiHitCollection",
"ITkPixelHits_G4")]
378 writtenContainers += [(
"SiHitCollection",
"ITkPixelHits")]
379 if flags.Detector.EnableITkStrip:
380 if (flags.Sim.ISFRun
and flags.Sim.ISF.HITSMergingRequired.get(
'ITk',
True)):
381 writtenContainers += [(
"SiHitCollection",
"ITkStripHits_G4")]
383 writtenContainers += [(
"SiHitCollection",
"ITkStripHits")]
384 if flags.Detector.EnablePLR:
385 if (flags.Sim.ISFRun
and flags.Sim.ISF.HITSMergingRequired.get(
'ITk',
True)):
386 writtenContainers += [(
"SiHitCollection",
"PLR_Hits_G4")]
388 writtenContainers += [(
"SiHitCollection",
"PLR_Hits")]
389 if flags.Detector.EnableHGTD:
390 if (flags.Sim.ISFRun
and flags.Sim.ISF.HITSMergingRequired.get(
'ITk',
True)):
391 writtenContainers += [(
"SiHitCollection",
"HGTDHits_G4")]
393 writtenContainers += [(
"SiHitCollection",
"HGTDHits")]
394 from SimulationConfig.SimEnums
import CalibrationRun
395 if flags.Sim.CalibrationRun
in [CalibrationRun.LAr, CalibrationRun.LArTile, CalibrationRun.LArTileZDC]:
397 writtenContainers += [
398 (
'CaloCalibrationHitContainer' ,
'StoreGateSvc+LArCalibrationHitActive_DEAD' ),
399 (
'CaloCalibrationHitContainer' ,
'StoreGateSvc+LArCalibrationHitDeadMaterial_DEAD' ),
400 (
'CaloCalibrationHitContainer' ,
'StoreGateSvc+LArCalibrationHitInactive_DEAD' )
403 return writtenContainers
408 from SimulationConfig.SimEnums
import LArParameterization
409 if flags.Sim.LArParameterization
is LArParameterization.FastCaloSim:
410 dependencies+=[(
'CaloDetDescrManager',
'ConditionStore+CaloDetDescrManager'),
411 (
'LArfSamplSym',
'ConditionStore+LArfSamplSym'),
412 (
'TileSamplingFraction',
'ConditionStore+TileSamplingFraction')]
413 from MuonSensitiveDetectorsR4.SensitiveDetectorsCfg
import MuonDependenciesCfg
430 result.setPrivateTools(tools)
438 if flags.Detector.EnableTile:
439 if flags.Sim.CalibrationRun
in [CalibrationRun.Tile, CalibrationRun.LArTile, CalibrationRun.LArTileZDC]:
440 from TileGeoG4Calib.TileGeoG4CalibConfig
import TileCTBGeoG4CalibSDCfg
443 from TileGeoG4SD.TileGeoG4SDToolConfig
import TileCTBGeoG4SDCfg
445 from MuonWall.MuonWallConfig
import MuonWallSDCfg
447 result.setPrivateTools(tools)
454 if flags.Detector.EnablePixel:
455 from PixelG4_SD.PixelG4_SDToolConfig
import PixelSensor_CTBCfg
457 if flags.Detector.EnableSCT:
458 from SCT_G4_SD.SCT_G4_SDToolConfig
import SctSensor_CTBCfg
460 if flags.Detector.EnableTRT:
461 from TRT_G4_SD.TRT_G4_SDToolConfig
import TRTSensitiveDetector_CTBCfg
463 if flags.Detector.EnableLAr:
464 from LArG4SD.LArG4SDToolConfig
import LArEMBSensitiveDetectorCfg
466 if flags.Sim.CalibrationRun
in [CalibrationRun.LAr, CalibrationRun.LArTile, CalibrationRun.LArTileZDC, CalibrationRun.DeadLAr]:
467 tools += [
'LArH8CalibSensitiveDetector' ]
468 if flags.Detector.EnableTile:
469 if flags.Sim.CalibrationRun
in [CalibrationRun.Tile, CalibrationRun.LArTile, CalibrationRun.LArTileZDC]:
470 from TileGeoG4Calib.TileGeoG4CalibConfig
import TileCTBGeoG4CalibSDCfg
473 from TileGeoG4SD.TileGeoG4SDToolConfig
import TileCTBGeoG4SDCfg
475 tools += [
'MuonWallSD' ]
476 if flags.Detector.EnableMuon:
477 tools += [
'MuonEntryRecord' ]
480 result.setPrivateTools(tools)
487 if flags.Beam.Type
is BeamType.TestBeam:
489 elif "tb_LArH6" in flags.GeoModel.AtlasVersion:
491 elif "ctbh8" in flags.GeoModel.AtlasVersion:
496 result.setPrivateTools(CompFactory.SensitiveDetectorMasterTool(name, **kwargs))
502 tool = result.popToolsAndMerge(CompFactory.SensitiveDetectorMasterTool(name, **kwargs))
503 result.setPrivateTools(tool)