testSensitiveDetectors.py

Go to the documentation of this file.

# Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
 
def setupTestOutputCfg(flags,**kwargs):
 
    kwargs.setdefault("streamName","MuonSimTestStream")
    kwargs.setdefault("AcceptAlgs",[])
  
    result = ComponentAccumulator()
    
    from MuonSensitiveDetectorsR4.SensitiveDetectorsCfg import OutputSimContainersCfg
    container_items = ["xAOD::TruthParticleContainer#",
                       "xAOD::TruthParticleAuxContainer#",
                       "McEventCollection#"] + OutputSimContainersCfg(flags)
 
   
   
    from xAODMetaDataCnv.InfileMetaDataConfig import propagateMetaData, MetaDataHelperLists
    from AthenaConfiguration.Enums import MetadataCategory
 
    mdLists = MetaDataHelperLists()
    for mdCategory in (MetadataCategory.FileMetaData, MetadataCategory.EventStreamInfo):
        lists, caConfig = propagateMetaData(flags, kwargs["streamName"], mdCategory)
        mdLists += lists
        result.merge(caConfig)
    kwargs.setdefault("MetadataItemList" , mdLists.mdItems)
    kwargs.setdefault("HelperTools", mdLists.helperTools)
    from AthenaServices.MetaDataSvcConfig import MetaDataSvcCfg
 
    result.merge(MetaDataSvcCfg(flags, 
                                tools=mdLists.mdTools, 
                                toolNames=mdLists.mdToolNames))
    from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
    kwargs.setdefault("ItemList", container_items)
    result.merge(OutputStreamCfg(flags, **kwargs))
    return result
 
if __name__=="__main__":
    from MuonGeoModelTestR4.testGeoModel import setupGeoR4TestCfg, SetupArgParser, executeTest
    parser = SetupArgParser()
    parser.set_defaults(nEvents = -1)
    parser.set_defaults(outRootFile="SimHits.pool.root")
 
    args = parser.parse_args()
    from AthenaConfiguration.AllConfigFlags import initConfigFlags
    flags = initConfigFlags()
 
    from AthenaConfiguration.Enums import ProductionStep
    flags.Common.ProductionStep = ProductionStep.Simulation
 
    from SimulationConfig.SimEnums import SimulationFlavour
    flags.Sim.ISF.Simulator = SimulationFlavour.AtlasG4
    flags.addFlag("Output.MuonSimTestStreamFileName", args.outRootFile)
 
    flags, cfg = setupGeoR4TestCfg(args, flags)
    
    from BeamEffects.BeamEffectsAlgConfig import BeamEffectsAlgCfg
    cfg.merge(BeamEffectsAlgCfg(flags))
 
    from G4AtlasAlg.G4AtlasAlgConfig import G4AtlasAlgCfg
    cfg.merge(G4AtlasAlgCfg(flags))
    
    
    from xAODTruthCnv.xAODTruthCnvConfig import GEN_EVNT2xAODCfg
    cfg.merge(GEN_EVNT2xAODCfg(flags,name="GEN_EVNT2xAOD",AODContainerName="TruthEvent"))
 
    cfg.merge(setupTestOutputCfg(flags))
    executeTest(cfg)