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
49 result.setPrivateTools(tools)
56 FastSimulationMasterTool = CompFactory.FastSimulationMasterTool
63 FastSimulationMasterTool = CompFactory.FastSimulationMasterTool
65 result.setPrivateTools(tool)
72 kwargs.setdefault(
"UseSimplifiedGeo",
bool(flags.Sim.SimplifiedGeoPath))
73 if flags.Sim.SimplifiedGeoPath:
75 kwargs.setdefault(
'SimplifiedWorldLogName',
"WorldLog")
77 kwargs.setdefault(
'TransportLimitVolume',
"Envelope")
79 kwargs.setdefault(
'MaxSteps', 100)
82 kwargs.setdefault(
'TransportLimitVolume',
"MuonSys")
84 kwargs.setdefault(
'MaxSteps', 5000)
86 result.setPrivateTools(CompFactory.G4CaloTransportTool(name, **kwargs))
93 kwargs.setdefault(
"ScalerConfigFileName" ,
"FastCaloSim/MC23/TFCSparam_mpt_classScaler_v04.xml" )
94 kwargs.setdefault(
"NetworkConfigFileName" ,
"FastCaloSim/MC23/TFCSparam_mpt_classNet_v04.json" )
95 kwargs.setdefault(
"CalibratorConfigFileName" ,
"FastCaloSim/MC23/TFCSparam_mpt_classCalib_v04.xml")
97 result.setPrivateTools(CompFactory.PunchThroughG4Classifier(name, **kwargs))
102 from SubDetectorEnvelopes.SubDetectorEnvelopesConfig
import EnvelopeDefSvcCfg
106 kwargs.setdefault(
"FilenameLookupTable" ,
"FastCaloSim/MC23/TFCSparam_mpt_v07.root")
107 kwargs.setdefault(
"FilenameInverseCdf" ,
"FastCaloSim/MC23/TFCSparam_mpt_inverseCdf_v07.xml")
108 kwargs.setdefault(
"FilenameInversePca" ,
"FastCaloSim/MC23/TFCSparam_mpt_inversePca_v07.xml")
109 kwargs.setdefault(
"EnergyFactor" , [ 0.98, 0.831, 0.896, 0.652, 0.717, 1., 0.877, 0.858, 0.919 ] )
110 kwargs.setdefault(
"DoAntiParticles" , [ 0, 1, 0, 1, 1, 0, 0, 0, 0 ] )
111 kwargs.setdefault(
"PunchThroughInitiators" , [ 211, 321, 311, 310, 130, 2212, 2112] )
112 kwargs.setdefault(
"InitiatorsMinEnergy" , [ 65536, 65536, 65536, 65536, 65536, 65536, 65536] )
113 kwargs.setdefault(
"InitiatorsEtaRange" , [ -3.2, 3.2 ] )
114 kwargs.setdefault(
"PunchThroughParticles" , [ 2212, 211, 22, 11, 13, 2112, 321, 310, 130 ] )
115 kwargs.setdefault(
"CorrelatedParticle" , [] )
116 kwargs.setdefault(
"FullCorrelationEnergy" , [ 100000., 100000., 100000., 100000., 0., 100000., 100000., 100000., 100000.] )
117 kwargs.setdefault(
"MinEnergy" , [ 938.3, 135.6, 50., 50., 105.7, 939.6, 493.7, 497.6, 497.6 ] )
118 kwargs.setdefault(
"MaxNumParticles" , [ -1, -1, -1, -1, -1, -1, -1, -1, -1 ] )
119 kwargs.setdefault(
"EnvelopeDefSvc", result.getPrimaryAndMerge(
EnvelopeDefSvcCfg(flags)).name)
120 kwargs.setdefault(
"BeamPipeRadius", 500.)
122 result.setPrivateTools(CompFactory.PunchThroughG4Tool(name, **kwargs))
129 if "PunchThroughG4Classifier" not in kwargs:
133 if "PunchThroughG4Tool" not in kwargs:
134 kwargs.setdefault(
"PunchThroughG4Tool", result.addPublicTool(result.popToolsAndMerge(
PunchThroughG4ToolCfg(flags))))
136 result.setPrivateTools(CompFactory.PunchThroughSimWrapper(name, **kwargs))
143 if flags.Detector.EnableLucid:
144 from LUCID_G4_SD.LUCID_G4_SDConfig
import LUCID_SensitiveDetectorCfg
146 if flags.Detector.EnableForward:
147 if flags.Detector.EnableZDC:
148 from ZDC_SD.ZDC_SDConfig
import ZDC_FiberSDCfg
150 if flags.Sim.CalibrationRun
in [CalibrationRun.ZDC, CalibrationRun.LArTileZDC]:
151 from ZDC_SD.ZDC_SDConfig
import ZDC_G4CalibSDCfg
153 if flags.Detector.EnableALFA:
154 from ALFA_G4_SD.ALFA_G4_SDConfig
import ALFA_SensitiveDetectorCfg
156 if flags.Detector.EnableAFP:
157 from AFP_G4_SD.AFP_G4_SDConfig
import AFP_SensitiveDetectorCfg
163 result.setPrivateTools(tools)
170 if (flags.Detector.EnableMuon
and flags.Sim.CavernBackground
in [CavernBackground.Write, CavernBackground.WriteWorld])
or flags.Sim.StoppedParticleFile:
171 from TrackWriteFastSim.TrackWriteFastSimConfig
import TrackFastSimSDCfg
173 result.setPrivateTools(tools)
179 if flags.Sim.LArParameterization
is LArParameterization.FastCaloSim:
180 from ISF_FastCaloSimParametrization.ISF_FastCaloSimParametrizationConfig
import CaloCellContainerSDCfg
182 result.setPrivateTools(tools)
189 if flags.Detector.EnableITkPixel:
190 from PixelG4_SD.PixelG4_SDToolConfig
import ITkPixelSensorSDCfg
193 if flags.Detector.EnableITkStrip:
194 from SCT_G4_SD.SCT_G4_SDToolConfig
import ITkStripSensorSDCfg
196 if flags.Detector.EnablePLR:
197 from PixelG4_SD.PixelG4_SDToolConfig
import PLRSensorSDCfg
200 result.setPrivateTools(tools)
208 if flags.Detector.EnableHGTD:
209 from HGTD_G4_SD.HGTD_G4_SDToolConfig
import HgtdSensorSDCfg
213 result.setPrivateTools(tools)
221 if flags.Detector.EnablePixel:
222 from PixelG4_SD.PixelG4_SDToolConfig
import PixelSensorSDCfg
224 if flags.Detector.EnableSCT:
225 from SCT_G4_SD.SCT_G4_SDToolConfig
import SctSensorSDCfg
227 if flags.Detector.EnableTRT:
228 from TRT_G4_SD.TRT_G4_SDToolConfig
import TRTSensitiveDetectorCfg
230 if flags.Detector.EnableBCM:
231 from BCM_G4_SD.BCM_G4_SDToolConfig
import BCMSensorSDCfg
233 from BLM_G4_SD.BLM_G4_SDToolConfig
import BLMSensorSDCfg
236 result.setPrivateTools(tools)
244 if flags.Detector.EnableLAr:
245 from LArG4SD.LArG4SDToolConfig
import LArEMBSensitiveDetectorCfg, LArEMECSensitiveDetectorCfg, LArFCALSensitiveDetectorCfg, LArHECSensitiveDetectorCfg
251 if flags.Detector.EnableMBTS:
252 from MinBiasScintillator.MinBiasScintillatorToolConfig
import MinBiasScintillatorSDCfg
255 if flags.Sim.CalibrationRun
in [CalibrationRun.LAr, CalibrationRun.LArTile, CalibrationRun.LArTileZDC]:
256 from LArG4SD.LArG4SDToolConfig
import LArDeadSensitiveDetectorToolCfg, LArActiveSensitiveDetectorToolCfg, LArInactiveSensitiveDetectorToolCfg
260 elif flags.Sim.CalibrationRun
is CalibrationRun.DeadLAr:
261 from LArG4SD.LArG4SDToolConfig
import LArDeadSensitiveDetectorToolCfg
264 if flags.Detector.EnableTile:
265 if flags.Sim.CalibrationRun
in [CalibrationRun.Tile, CalibrationRun.LArTile, CalibrationRun.LArTileZDC]:
266 from TileGeoG4Calib.TileGeoG4CalibConfig
import TileGeoG4CalibSDCfg
269 from TileGeoG4SD.TileGeoG4SDToolConfig
import TileGeoG4SDCfg
271 if flags.Sim.RecordStepInfo:
272 from ISF_FastCaloSimSD.ISF_FastCaloSimSDToolConfig
import FCS_StepInfoSDToolCfg
275 result.setPrivateTools(tools)
280 if flags.Muon.usePhaseIIGeoSetup:
281 from MuonSensitiveDetectorsR4.SensitiveDetectorsCfg
import SetupSensitiveDetectorsCfg
283 from MuonG4SD.MuonG4SDConfig
import SetupSensitiveDetectorsCfg
289 if flags.Beam.Type
is BeamType.Cosmics
and not flags.Sim.ReadTR:
290 from TrackWriteFastSim.TrackWriteFastSimConfig
import CosmicTRSDCfg
292 result.setPrivateTools(tools)
297 from SimulationConfig.SimEnums
import LArParameterization
298 writtenContainers =[]
299 if flags.Detector.GeometryMuon:
300 if flags.Muon.usePhaseIIGeoSetup:
301 from MuonSensitiveDetectorsR4.SensitiveDetectorsCfg
import SimHitContainerListCfg
304 from MuonG4SD.MuonG4SDConfig
import SimHitContainerListCfg
306 if flags.Detector.GeometryLAr:
307 if (flags.Sim.ISFRun
and flags.Sim.ISF.HITSMergingRequired.get(
'CALO',
True))
or flags.Sim.LArParameterization
is LArParameterization.FastCaloSim:
308 writtenContainers += [(
"LArHitContainer",
"LArHitEMB_G4")]
309 writtenContainers += [(
"LArHitContainer",
"LArHitEMEC_G4")]
310 writtenContainers += [(
"LArHitContainer",
"LArHitFCAL_G4")]
311 writtenContainers += [(
"LArHitContainer",
"LArHitHEC_G4")]
312 if flags.Sim.LArParameterization
is LArParameterization.FastCaloSim:
313 writtenContainers += [(
"LArHitContainer" ,
"LArHitHEC_FastCaloSim")]
314 writtenContainers += [(
"LArHitContainer" ,
"LArHitFCAL_FastCaloSim")]
315 writtenContainers += [(
"LArHitContainer" ,
"LArHitEMEC_FastCaloSim")]
316 writtenContainers += [(
"LArHitContainer" ,
"LArHitEMB_FastCaloSim")]
318 writtenContainers += [(
"LArHitContainer",
"LArHitEMB")]
319 writtenContainers += [(
"LArHitContainer",
"LArHitEMEC")]
320 writtenContainers += [(
"LArHitContainer",
"LArHitFCAL")]
321 writtenContainers += [(
"LArHitContainer",
"LArHitHEC")]
322 if flags.Detector.GeometryTile:
323 if (flags.Sim.ISFRun
and flags.Sim.ISF.HITSMergingRequired.get(
'CALO',
True))
or flags.Sim.LArParameterization
is LArParameterization.FastCaloSim:
324 writtenContainers += [(
"TileHitVector",
"MBTSHits_G4")]
325 writtenContainers += [(
"TileHitVector",
"TileHitVec_G4")]
326 if flags.Sim.LArParameterization
is LArParameterization.FastCaloSim:
327 writtenContainers += [(
"TileHitVector" ,
"TileHitVec_FastCaloSim")]
328 writtenContainers += [(
"TileHitVector" ,
"MBTSHits_FastCaloSim")]
330 writtenContainers += [(
"TileHitVector",
"MBTSHits")]
331 writtenContainers += [(
"TileHitVector",
"TileHitVec")]
332 if flags.Detector.GeometryTRT:
333 if (flags.Sim.ISFRun
and flags.Sim.ISF.HITSMergingRequired.get(
'ID',
True)):
334 writtenContainers += [(
"TRTUncompressedHitCollection",
"TRTUncompressedHits_G4")]
336 writtenContainers += [(
"TRTUncompressedHitCollection",
"TRTUncompressedHits")]
337 if flags.Detector.EnableBCM:
338 if (flags.Sim.ISFRun
and flags.Sim.ISF.HITSMergingRequired.get(
'ID',
True)):
339 writtenContainers += [(
"SiHitCollection",
"BCMHits_G4")]
340 writtenContainers += [(
"SiHitCollection",
"BLMHits_G4")]
342 writtenContainers += [(
"SiHitCollection",
"BCMHits")]
343 writtenContainers += [(
"SiHitCollection",
"BLMHits")]
344 if flags.Detector.EnablePixel:
345 if (flags.Sim.ISFRun
and flags.Sim.ISF.HITSMergingRequired.get(
'ID',
True)):
346 writtenContainers += [(
"SiHitCollection",
"PixelHits_G4")]
348 writtenContainers += [(
"SiHitCollection",
"PixelHits")]
349 if flags.Detector.EnableSCT:
350 if (flags.Sim.ISFRun
and flags.Sim.ISF.HITSMergingRequired.get(
'ID',
True)):
351 writtenContainers += [(
"SiHitCollection",
"SCT_Hits_G4")]
353 writtenContainers += [(
"SiHitCollection",
"SCT_Hits")]
354 if flags.Detector.EnableITkPixel:
355 if (flags.Sim.ISFRun
and flags.Sim.ISF.HITSMergingRequired.get(
'ITk',
True)):
356 writtenContainers += [(
"SiHitCollection",
"ITkPixelHits_G4")]
358 writtenContainers += [(
"SiHitCollection",
"ITkPixelHits")]
359 if flags.Detector.EnableITkStrip:
360 if (flags.Sim.ISFRun
and flags.Sim.ISF.HITSMergingRequired.get(
'ITk',
True)):
361 writtenContainers += [(
"SiHitCollection",
"ITkStripHits_G4")]
363 writtenContainers += [(
"SiHitCollection",
"ITkStripHits")]
364 if flags.Detector.EnablePLR:
365 if (flags.Sim.ISFRun
and flags.Sim.ISF.HITSMergingRequired.get(
'ITk',
True)):
366 writtenContainers += [(
"SiHitCollection",
"PLR_Hits_G4")]
368 writtenContainers += [(
"SiHitCollection",
"PLR_Hits")]
369 if flags.Detector.EnableHGTD:
370 if (flags.Sim.ISFRun
and flags.Sim.ISF.HITSMergingRequired.get(
'ITk',
True)):
371 writtenContainers += [(
"SiHitCollection",
"HGTDHits_G4")]
373 writtenContainers += [(
"SiHitCollection",
"HGTDHits")]
374 from SimulationConfig.SimEnums
import CalibrationRun
375 if flags.Sim.CalibrationRun
in [CalibrationRun.LAr, CalibrationRun.LArTile, CalibrationRun.LArTileZDC]:
377 writtenContainers += [
378 (
'CaloCalibrationHitContainer' ,
'StoreGateSvc+LArCalibrationHitActive_DEAD' ),
379 (
'CaloCalibrationHitContainer' ,
'StoreGateSvc+LArCalibrationHitDeadMaterial_DEAD' ),
380 (
'CaloCalibrationHitContainer' ,
'StoreGateSvc+LArCalibrationHitInactive_DEAD' )
383 return writtenContainers
388 from SimulationConfig.SimEnums
import LArParameterization
389 if flags.Sim.LArParameterization
is LArParameterization.FastCaloSim:
390 dependencies+=[(
'CaloDetDescrManager',
'ConditionStore+CaloDetDescrManager'),
391 (
'LArfSamplSym',
'ConditionStore+LArfSamplSym'),
392 (
'TileSamplingFraction',
'ConditionStore+TileSamplingFraction')]
393 from MuonSensitiveDetectorsR4.SensitiveDetectorsCfg
import MuonDependenciesCfg
410 result.setPrivateTools(tools)
418 if flags.Detector.EnableTile:
419 if flags.Sim.CalibrationRun
in [CalibrationRun.Tile, CalibrationRun.LArTile, CalibrationRun.LArTileZDC]:
420 from TileGeoG4Calib.TileGeoG4CalibConfig
import TileCTBGeoG4CalibSDCfg
423 from TileGeoG4SD.TileGeoG4SDToolConfig
import TileCTBGeoG4SDCfg
425 from MuonWall.MuonWallConfig
import MuonWallSDCfg
427 result.setPrivateTools(tools)
434 if flags.Detector.EnablePixel:
435 from PixelG4_SD.PixelG4_SDToolConfig
import PixelSensor_CTBCfg
437 if flags.Detector.EnableSCT:
438 from SCT_G4_SD.SCT_G4_SDToolConfig
import SctSensor_CTBCfg
440 if flags.Detector.EnableTRT:
441 from TRT_G4_SD.TRT_G4_SDToolConfig
import TRTSensitiveDetector_CTBCfg
443 if flags.Detector.EnableLAr:
444 from LArG4SD.LArG4SDToolConfig
import LArEMBSensitiveDetectorCfg
446 if flags.Sim.CalibrationRun
in [CalibrationRun.LAr, CalibrationRun.LArTile, CalibrationRun.LArTileZDC, CalibrationRun.DeadLAr]:
447 tools += [
'LArH8CalibSensitiveDetector' ]
448 if flags.Detector.EnableTile:
449 if flags.Sim.CalibrationRun
in [CalibrationRun.Tile, CalibrationRun.LArTile, CalibrationRun.LArTileZDC]:
450 from TileGeoG4Calib.TileGeoG4CalibConfig
import TileCTBGeoG4CalibSDCfg
453 from TileGeoG4SD.TileGeoG4SDToolConfig
import TileCTBGeoG4SDCfg
455 tools += [
'MuonWallSD' ]
456 if flags.Detector.EnableMuon:
457 tools += [
'MuonEntryRecord' ]
460 result.setPrivateTools(tools)
467 if flags.Beam.Type
is BeamType.TestBeam:
469 elif "tb_LArH6" in flags.GeoModel.AtlasVersion:
471 elif "ctbh8" in flags.GeoModel.AtlasVersion:
476 result.setPrivateTools(CompFactory.SensitiveDetectorMasterTool(name, **kwargs))
482 tool = result.popToolsAndMerge(CompFactory.SensitiveDetectorMasterTool(name, **kwargs))
483 result.setPrivateTools(tool)