TrkExUnitTestsConfig.py

Go to the documentation of this file.

# Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
 
"""Define methods to configure TrkExUnitTest"""
 
def PositionMomentumWriterCfg(configFlags, name="PositionMomentumWriter", **kwargs) :
  from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator 
  from AthenaConfiguration.ComponentFactory import CompFactory
 
  result = ComponentAccumulator()
 
  Trk__PositionMomentumWriter = CompFactory.Trk.PositionMomentumWriter
  posMomWriter = Trk__PositionMomentumWriter(name, **kwargs)
  result.addPublicTool(posMomWriter)
  #result.setPrivateTools(posMomWriter)
  return result, posMomWriter
 
def ExtrapolationEngineTestCfg(configFlags, name = "ExtrapolationEngineTest", **kwargs ) :
  from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator 
  from AthenaConfiguration.ComponentFactory import CompFactory
  
  result=ComponentAccumulator()  
 
  
  histSvc = CompFactory.THistSvc(Output = ["val DATAFILE='ExtrapolationEngineTest.root' TYPE='ROOT' OPT='RECREATE'"])
  result.addService( histSvc )    
 
  from TrkConfig.AtlasExtrapolationEngineConfig import AtlasExtrapolationEngineCfg
  extrapAcc = AtlasExtrapolationEngineCfg(configFlags)
  extrapolationEngine = extrapAcc.getPrimary()
  result.merge(extrapAcc)
  kwargs["ExtrapolationEngine"] = extrapolationEngine
 
  posMomAcc, posMomWriter = PositionMomentumWriterCfg(configFlags)
  result.merge(posMomAcc)
  kwargs.setdefault('PositionMomentumWriter', posMomWriter)
     
  Trk__ExtrapolationEngineTest = CompFactory.Trk.ExtrapolationEngineTest
  extrapolationTest = Trk__ExtrapolationEngineTest(name, **kwargs)
  result.addEventAlgo(extrapolationTest)
  
  return result