G4AtlasAlgConfig.py

Go to the documentation of this file.

# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
from G4AtlasServices.G4AtlasServicesConfig import PhysicsListSvcCfg
from ISF_Services.ISF_ServicesConfig import TruthServiceCfg, InputConverterCfg
from G4AtlasTools.G4GeometryToolConfig import G4AtlasDetectorConstructionToolCfg
from G4AtlasTools.G4AtlasToolsConfig import G4ThreadPoolSvcCfg, SensitiveDetectorMasterToolCfg, FastSimulationMasterToolCfg
from G4AtlasServices.G4AtlasUserActionConfig import UserActionSvcCfg
from SimulationConfig.SimulationMetadata import writeSimulationParametersMetadata, readSimulationParameters
from AthenaConfiguration.ComponentFactory import CompFactory
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
 
 
def G4AtlasAlgCfg(flags, name="G4AtlasAlg", **kwargs):
    """Return ComponentAccumulator configured for Atlas G4 simulation, without output"""
    # wihout output
    result = ComponentAccumulator()
    from SimulationConfig.SimEnums import LArParameterization
    if flags.Sim.LArParameterization is LArParameterization.FastCaloSim:
        # Add a dummy version of the SimKernel to the CA before adding the properly configured version
        # Presently necessary for FastCaloSim to ensure the proper order of ISF_CollectionMerger
        result.addEventAlgo(CompFactory.G4AtlasAlg(name, **kwargs))
    kwargs.setdefault("UseShadowEvent", flags.Sim.UseShadowEvent)
    if flags.Sim.UseShadowEvent and "TruthPreselectionTool" not in kwargs:
        from ISF_HepMC_Tools.ISF_HepMC_ToolsConfig import TruthPreselectionToolCfg
        kwargs.setdefault( "TruthPreselectionTool", result.popToolsAndMerge(TruthPreselectionToolCfg(flags)) )
 
    kwargs.setdefault("DetectorConstruction", result.addPublicTool(result.popToolsAndMerge(G4AtlasDetectorConstructionToolCfg(flags))))
 
    kwargs.setdefault("InputTruthCollection", "BeamTruthEvent") #tocheck -are these string inputs?
    kwargs.setdefault("OutputTruthCollection", "TruthEvent")
    
 
    
    kwargs.setdefault("ReleaseGeoModel", flags.Sim.ReleaseGeoModel)
 
    from G4AtlasTools.G4AtlasToolsConfig import SimHitContainerListCfg, InputContainerListCfg
    kwargs.setdefault("ExtraOutputs", SimHitContainerListCfg(flags) )
    kwargs.setdefault("ExtraInputs" , InputContainerListCfg(flags))
 
    from SimulationConfig.SimEnums import LArParameterization, InDetParameterization
    # Configure fast simulation
    if flags.Sim.LArParameterization is LArParameterization.FastCaloSim:
        from G4AtlasTools.G4AtlasToolsConfig import PunchThroughG4ToolCfg
        physics_initialization_tools = kwargs.setdefault("PhysicsInitializationTools", [])
        physics_initialization_tools.append(result.addPublicTool(result.popToolsAndMerge(PunchThroughG4ToolCfg(flags))))
        # Set the path to the simplified calorimeter geometry for particle transport if provided
        if flags.Sim.SimplifiedGeoPath:
            kwargs.setdefault('SimplifiedGeoPath', flags.Sim.SimplifiedGeoPath)
 
    if flags.Sim.InDetParameterization is InDetParameterization.FatrasG4:
        from G4AtlasTools.G4AtlasToolsConfig import ActsFatrasG4ToolCfg
 
        # Add to physics initialization
        physics_initialization_tools = kwargs.setdefault("PhysicsInitializationTools", [])
        physics_initialization_tools.append(result.addPublicTool(result.popToolsAndMerge(ActsFatrasG4ToolCfg(flags))))
    
    if flags.Sim.InDetParameterization is InDetParameterization.AFatrasG4:
        from G4AtlasTools.G4AtlasToolsConfig import ActsFatrasG4ToolCfg
 
        # Add to physics initialization
        physics_initialization_tools = kwargs.setdefault("PhysicsInitializationTools", [])
        physics_initialization_tools.append(result.addPublicTool(result.popToolsAndMerge(ActsFatrasG4ToolCfg(flags)))) 
    
    # Set the path to the simplified calorimeter geometry for particle transport if provided
    if flags.Sim.LArParameterization is LArParameterization.FastCaloSim and flags.Sim.SimplifiedGeoPath:
        kwargs.setdefault("SimplifiedGeoPath", flags.Sim.SimplifiedGeoPath)
 
    
    kwargs.setdefault("RecordFlux", flags.Sim.RecordFlux)
 
    if flags.Sim.FlagAbortedEvents:
        
        kwargs.setdefault("FlagAbortedEvents", flags.Sim.FlagAbortedEvents)
        if flags.Sim.FlagAbortedEvents and flags.Sim.KillAbortedEvents:
            print("WARNING When G4AtlasAlg.FlagAbortedEvents is True G4AtlasAlg.KillAbortedEvents should be False. Setting G4AtlasAlg.KillAbortedEvents = False now.")
            kwargs.setdefault("KillAbortedEvents", False)
 
    
    kwargs.setdefault("KillAbortedEvents", flags.Sim.KillAbortedEvents)
 
    from RngComps.RngCompsConfig import AthRNGSvcCfg
    kwargs.setdefault("AtRndmGenSvc",
                      result.getPrimaryAndMerge(AthRNGSvcCfg(flags)))
 
    kwargs.setdefault("RandomGenerator", "athena")
 
    # Multi-threading settinggs
    is_hive = flags.Concurrency.NumThreads > 0
    kwargs.setdefault("MultiThreading", is_hive)
    if is_hive:
        result.merge(G4ThreadPoolSvcCfg(flags))
        kwargs.setdefault("Cardinality", flags.Concurrency.NumThreads)
 
    kwargs.setdefault("TruthRecordService", result.getPrimaryAndMerge(TruthServiceCfg(flags)))
 
    #input converter
    kwargs.setdefault("InputConverter", result.getPrimaryAndMerge(InputConverterCfg(flags)))
    if flags.Sim.ISF.Simulator.isQuasiStable():
        from BeamEffects.BeamEffectsAlgConfig import ZeroLifetimePositionerCfg
        kwargs.setdefault("QuasiStablePatcher", result.getPrimaryAndMerge(ZeroLifetimePositionerCfg(flags)) )
 
    #sensitive detector master tool
    kwargs.setdefault("SenDetMasterTool", result.addPublicTool(result.popToolsAndMerge(SensitiveDetectorMasterToolCfg(flags))))
 
    #fast simulation master tool
    kwargs.setdefault("FastSimMasterTool", result.addPublicTool(result.popToolsAndMerge(FastSimulationMasterToolCfg(flags))))
 
    #Write MetaData container and make it available to the job
    result.merge(writeSimulationParametersMetadata(flags))
    result.merge(readSimulationParameters(flags))  # for FileMetaData creation
 
    #User action services (Slow...)
    kwargs.setdefault("UserActionSvc", result.getPrimaryAndMerge(UserActionSvcCfg(flags)))
 
    #PhysicsListSvc
    kwargs.setdefault("PhysicsListSvc", result.getPrimaryAndMerge(PhysicsListSvcCfg(flags)))
 
    
    verbosities=dict(foo="bar")
    kwargs.setdefault("Verbosities", verbosities)
    kwargs.setdefault("QuietMode", flags.Exec.QuietMode)
 
    # Set commands for the G4AtlasAlg
    kwargs.setdefault("G4Commands", flags.Sim.G4Commands)
    result.addEventAlgo(CompFactory.G4AtlasAlg(name, **kwargs))
 
    return result