ActsMaterialConfig.py

Go to the documentation of this file.

#  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
 
def MaterialTrackRecorderCfg(configFlags, name="ActsTrk::UserActionSvc.MaterialTrackRecorderTool", **kwargs):
  from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
  from AthenaConfiguration.ComponentFactory import CompFactory
  acc = ComponentAccumulator()
  acc.setPrivateTools(CompFactory.ActsTrk.MaterialTrackRecorderTool(name, **kwargs))
  return acc
 
 
def MaterialTrackRecorderUserActionSvcCfg(configFlags, name="ActsTrk::MaterialTrackRecorderUserActionSvc", **kwargs):
  from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
  from AthenaConfiguration.ComponentFactory import CompFactory
 
  acc = ComponentAccumulator()
 
  #Setting up the CA for the MaterialStepRecorder
  actionAcc = ComponentAccumulator()
  actions = []
  actions += [actionAcc.popToolsAndMerge(MaterialTrackRecorderCfg(configFlags))]
  actionAcc.setPrivateTools(actions)
  MaterialTrackRecorderAction = acc.popToolsAndMerge(actionAcc)
 
  #Retrieving the default action list
  from G4AtlasServices.G4AtlasUserActionConfig import getDefaultActions
  defaultActions = acc.popToolsAndMerge(getDefaultActions(configFlags))
 
  #Adding material recorder action to defaults
  actionList = (defaultActions + MaterialTrackRecorderAction)
 
  kwargs.setdefault("UserActionTools",actionList)
  acc.addService(CompFactory.G4UA.UserActionSvc(name,**kwargs), primary = True)
 
  return acc
 
def MaterialTrackWriterCfg(configFlags,
                           name="MaterialTrackWriter",
                           FileName="material-tracks.root",
                           OutStream="ACTSMATERIALWRITER",
                           **kwargs) :
  from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
  from AthenaConfiguration.ComponentFactory import CompFactory
  acc = ComponentAccumulator()
  kwargs.setdefault("OutStream", OutStream)
  kwargs.setdefault("useTrackingGeometry", True)
  
  from MuonConfig.MuonConfigUtils import setupHistSvcCfg
  acc.merge(setupHistSvcCfg(configFlags, outFile = FileName, outStream=kwargs["OutStream"]))
 
  # Need geometry
  if kwargs["useTrackingGeometry"]:
    from ActsConfig.ActsGeometryConfig import ActsTrackingGeometryToolCfg
    kwargs.setdefault("TrackingGeometryTool", acc.getPrimaryAndMerge(ActsTrackingGeometryToolCfg(configFlags)))
  acc.addEventAlgo(CompFactory.ActsTrk.MaterialTrackWriter(name, **kwargs), primary = True)
 
  return acc
 
def MaterialTrackReaderCfg(configFlags, name="MaterialTrackReader", **kwargs) :
  from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
  from AthenaConfiguration.ComponentFactory import CompFactory
  acc = ComponentAccumulator()
 
  acc.addEventAlgo(CompFactory.ActsTrk.MaterialTrackReader(name, **kwargs), primary = True)
 
  return acc
 
def RootMaterialWriterToolCfg(configFlags, name="RootMaterialWriterTool", **kwargs):
  from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
  from AthenaConfiguration.ComponentFactory import CompFactory
  acc = ComponentAccumulator()
  acc.setPrivateTools(CompFactory.ActsTrk.RootMaterialWriterTool(name, **kwargs))
  return acc
 
def MaterialMappingCfg(configFlags,
                       inputFiles,
                       name="MaterialMapping",
                       StoreTracks=False,
                       OutputMappedMaterialTracks="OuputMappedMaterialTracks",
                       OutputUnmappedMaterialTracks="OutputUnmappedMaterialTracks",
                       **kwargs) :
  from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
  from AthenaConfiguration.ComponentFactory import CompFactory
  acc = ComponentAccumulator()
 
  # Need geometry
  from ActsConfig.ActsGeometryConfig import ActsTrackingGeometrySvcCfg
  acc.merge( ActsTrackingGeometrySvcCfg(configFlags))
 
  mapwriters = [acc.popToolsAndMerge(RootMaterialWriterToolCfg(configFlags))]
  kwargs.setdefault("MaterialMapWriters", mapwriters)
 
  kwargs.setdefault("MappedMaterialTrackCollectionKey", OutputMappedMaterialTracks)
  kwargs.setdefault("UnmappedMaterialTrackCollectionKey", OutputUnmappedMaterialTracks)
 
 
  acc.merge(MaterialTrackReaderCfg(configFlags, FileNames=inputFiles))
 
  acc.addEventAlgo(CompFactory.ActsTrk.MaterialMapping(name, **kwargs), primary = True)
 
  if StoreTracks:
      from ActsConfig.ActsMaterialConfig import MaterialTrackWriterCfg
      acc.merge(MaterialTrackWriterCfg(configFlags,
                                       name="MappedMaterialTrackWriter",
                                       FileName="material-tracks-mapped.root",
                                       OutStream="ACTSMAPPEDMATERIALWRITER",
                                       MaterialTrackCollectionKey=OutputMappedMaterialTracks))
      acc.merge(MaterialTrackWriterCfg(configFlags,
                                       name="UnmappedMaterialTrackWriter",
                                       FileName="material-tracks-unmapped.root",
                                       OutStream="ACTSUNMAPPEDMATERIALWRITER",
                                       MaterialTrackCollectionKey=OutputUnmappedMaterialTracks))
 
  return acc
 
 
def MaterialValidationCfg(configFlags,
                          StoreTracks=True,
                          OutputMaterialTracks="OutputMaterialTracks",
                          name="MaterialValidation", **kwargs) :
  from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
  from AthenaConfiguration.ComponentFactory import CompFactory
  acc = ComponentAccumulator()
 
  # Need geometry
  from ActsConfig.ActsGeometryConfig import ActsTrackingGeometrySvcCfg
  acc.merge( ActsTrackingGeometrySvcCfg(configFlags))
 
  kwargs.setdefault("MaterialTrackCollectionKey", OutputMaterialTracks)
 
  acc.addEventAlgo(CompFactory.ActsTrk.MaterialValidation(name, **kwargs), primary = True)
  if StoreTracks:
      from ActsConfig.ActsMaterialConfig import MaterialTrackWriterCfg
      acc.merge(MaterialTrackWriterCfg(configFlags,
                                       FileName="material-tracks-validation.root",
                                       MaterialTrackCollectionKey=OutputMaterialTracks))
 
 
  return acc