python.testGeoModel Namespace Reference

Functions
def	geoModelFileDefault (useR4Layout=False)
def	SetupArgParser ()
def	setupServicesCfg (flags)
def	setupHistSvcCfg (flags, outFile="MdtGeoDump.root", outStream="GEOMODELTESTER")
def	GeoModelMdtTestCfg (flags, name="GeoModelMdtTest", **kwargs)
def	GeoModelRpcTestCfg (flags, name="GeoModelRpcTest", **kwargs)
def	GeoModelTgcTestCfg (flags, name="GeoModelTgcTest", **kwargs)
def	GeoModelsTgcTestCfg (flags, name="GeoModelsTgcTest", **kwargs)
def	GeoModelMmTestCfg (flags, name="GeoModelMmTest", **kwargs)
def	NswGeoPlottingAlgCfg (flags, name="NswGeoPlotting", **kwargs)
def	configureDefaultTagsCfg (flags)
def	setupGeoR4TestCfg (args, flags=None)
def	executeTest (cfg)

Variables
def	args = SetupArgParser().parse_args()
	flags
	cfg
def	chambToTest = 0 else []
def	chambToExclude = args.excludedChambers
	checkGeo
	Ensure consistent translation of the geometry. More...
	setVerbose

Function Documentation

configureDefaultTagsCfg()

def python.testGeoModel.configureDefaultTagsCfg

(

flags

)

Definition at line 100 of file MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py.

def configureDefaultTagsCfg(flags):    
    from AthenaCommon.Logging import logging
    log = logging.getLogger('GeometryConfiguration')
 
    if not flags.GeoModel.SQLiteDB:
        raise ValueError("Default tag configuration only works for SQLite")
    
    from AthenaConfiguration.TestDefaults import defaultConditionsTags, defaultGeometryTags
    flags.GeoModel.AtlasVersion = defaultGeometryTags.RUN2    
    from AthenaConfiguration.Enums import LHCPeriod
    if flags.GeoModel.Run == LHCPeriod.Run3:   
        flags.IOVDb.GlobalTag = defaultConditionsTags.RUN3_MC if flags.Input.isMC else defaultConditionsTags.RUN3_DATA
        flags.GeoModel.AtlasVersion = defaultGeometryTags.RUN3
    elif flags.GeoModel.Run == LHCPeriod.Run4:
          flags.GeoModel.AtlasVersion = defaultGeometryTags.RUN4
          flags.IOVDb.GlobalTag = defaultConditionsTags.RUN4_MC
    else:
        raise ValueError(f"Invalid run period {flags.GeoModel.Run}")
    log.info(f"Setup {flags.GeoModel.AtlasVersion} geometry loading {flags.GeoModel.SQLiteDBFullPath}")
    log.info(f"Use conditions tag {flags.IOVDb.GlobalTag}")
    
 

executeTest()

def python.testGeoModel.executeTest

(

cfg

)

Definition at line 224 of file MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py.

def executeTest(cfg):
    cfg.printConfig(withDetails=True, summariseProps=True)
    if not cfg.run().isSuccess(): exit(1)
 

geoModelFileDefault()

def python.testGeoModel.geoModelFileDefault

(

useR4Layout = False

)

Definition at line 5 of file MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py.

def geoModelFileDefault(useR4Layout = False):
    # If this is changed, remember to also test with other dependent tests 
    # e.g. run ctest with ActsEventCnv
    if useR4Layout: 
        return  "/cvmfs/atlas-nightlies.cern.ch/repo/data/data-art/MuonGeomRTT/GeoDB/ATLAS-P2-RUN4-01-00-00.db"
    return "/cvmfs/atlas-nightlies.cern.ch/repo/data/data-art/MuonGeomRTT/GeoDB/ATLAS-R3S-2021-03-02-00.db"
 

GeoModelMdtTestCfg()

def python.testGeoModel.GeoModelMdtTestCfg	(		flags,
			name = "GeoModelMdtTest",
		**	kwargs
	)

Definition at line 63 of file MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py.

def GeoModelMdtTestCfg(flags, name = "GeoModelMdtTest", **kwargs):
    result = ComponentAccumulator()
    the_alg = CompFactory.MuonGMR4.GeoModelMdtTest(name, **kwargs)
    result.addEventAlgo(the_alg, primary = True)
    return result
 

GeoModelMmTestCfg()

def python.testGeoModel.GeoModelMmTestCfg	(		flags,
			name = "GeoModelMmTest",
		**	kwargs
	)

Definition at line 87 of file MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py.

def GeoModelMmTestCfg(flags, name = "GeoModelMmTest", **kwargs):
    result = ComponentAccumulator()
    the_alg = CompFactory.MuonGMR4.GeoModelMmTest(name, **kwargs)
    result.addEventAlgo(the_alg, primary = True)
    return result
 

GeoModelRpcTestCfg()

def python.testGeoModel.GeoModelRpcTestCfg	(		flags,
			name = "GeoModelRpcTest",
		**	kwargs
	)

Definition at line 69 of file MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py.

def GeoModelRpcTestCfg(flags, name = "GeoModelRpcTest", **kwargs):
    result = ComponentAccumulator()
    the_alg = CompFactory.MuonGMR4.GeoModelRpcTest(name, **kwargs)
    result.addEventAlgo(the_alg, primary = True)
    return result
 

GeoModelsTgcTestCfg()

def python.testGeoModel.GeoModelsTgcTestCfg	(		flags,
			name = "GeoModelsTgcTest",
		**	kwargs
	)

Definition at line 81 of file MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py.

def GeoModelsTgcTestCfg(flags, name = "GeoModelsTgcTest", **kwargs):
    result = ComponentAccumulator()
    the_alg = CompFactory.MuonGMR4.GeoModelsTgcTest(name, **kwargs)
    result.addEventAlgo(the_alg, primary = True)
    return result
 

GeoModelTgcTestCfg()

def python.testGeoModel.GeoModelTgcTestCfg	(		flags,
			name = "GeoModelTgcTest",
		**	kwargs
	)

Definition at line 75 of file MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py.

def GeoModelTgcTestCfg(flags, name = "GeoModelTgcTest", **kwargs):
    result = ComponentAccumulator()
    the_alg = CompFactory.MuonGMR4.GeoModelTgcTest(name, **kwargs)
    result.addEventAlgo(the_alg, primary = True)
    return result
 

NswGeoPlottingAlgCfg()

def python.testGeoModel.NswGeoPlottingAlgCfg	(		flags,
			name = "NswGeoPlotting",
		**	kwargs
	)

Definition at line 93 of file MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py.

def NswGeoPlottingAlgCfg(flags, name="NswGeoPlotting", **kwargs):
    result = ComponentAccumulator()
    kwargs.setdefault("TestActsSurface", False)
    the_alg = CompFactory.MuonGMR4.NswGeoPlottingAlg(name, **kwargs)
    result.addEventAlgo(the_alg, primary = True)
    return result
 

SetupArgParser()

def python.testGeoModel.SetupArgParser

(

)

Definition at line 12 of file MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py.

def SetupArgParser():
    from argparse import ArgumentParser
 
    parser = ArgumentParser()
    parser.add_argument("--threads", type=int, help="number of threads", default=1)
    parser.add_argument("--inputFile", "-i", default=[
                                                      #"/cvmfs/atlas-nightlies.cern.ch/repo/data/data-art/Tier0ChainTests/TCT_Run3/data22_13p6TeV.00431493.physics_Main.daq.RAW._lb0525._SFO-16._0001.data"
                                                      "/cvmfs/atlas-nightlies.cern.ch/repo/data/data-art/MuonRecRTT/EVGEN_ParticleGun_FourMuon_Pt10to500.root"
                                                      ], 
                        help="Input file to run on ", nargs="+")
    parser.add_argument("--geoModelFile", default = geoModelFileDefault(), help="GeoModel SqLite file containing the muon geometry.")
    parser.add_argument("--defaultGeoFile", help="Use the  predefined GeoModel files on cvmfs", choices=["NONE", "RUN3", "RUN4" ], default="NONE")
    parser.add_argument("--chambers", default=["all"], nargs="+", help="Chambers to check. If string is all, all chambers will be checked")
    parser.add_argument("--excludedChambers", default=[], nargs="+", help="Chambers to exclude. If string contains 'none', all chambers will be checked. Note: adding a chamber to --excludedChambers will overwrite it being in --chambers.")
    parser.add_argument("--outRootFile", default="NewGeoModelDump.root", help="Output ROOT file to dump the geomerty")
    parser.add_argument("--nEvents", help="Number of events to run", type = int ,default = 1)
    parser.add_argument("--skipEvents", help="Number of events to skip", type = int, default = 0)
    parser.add_argument("--noMdt", help="Disable the Mdts from the geometry", action='store_true', default = False)
    parser.add_argument("--noRpc", help="Disable the Rpcs from the geometry", action='store_true', default = False)
    parser.add_argument("--noTgc", help="Disable the Tgcs from the geometry", action='store_true', default = False)
    parser.add_argument("--noMM", help="Disable the MMs from the geometry", action='store_true', default = False)
    parser.add_argument("--noSTGC", help="Disable the sTgcs from the geometry", action='store_true', default = False)
    parser.add_argument("--eventPrintoutLevel", type=int, help="Interval of event heartbeat printouts from the loop manager", default = 1)
    return parser
 

setupGeoR4TestCfg()

def python.testGeoModel.setupGeoR4TestCfg	(		args,
			flags = None
	)

Definition at line 122 of file MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py.

def setupGeoR4TestCfg(args,  flags = None):
    
    if flags is None:
        from AthenaConfiguration.AllConfigFlags import initConfigFlags
        flags = initConfigFlags()
    flags.Concurrency.NumThreads = args.threads
    flags.Concurrency.NumConcurrentEvents = args.threads
    flags.Exec.MaxEvents = args.nEvents
    flags.Exec.SkipEvents = args.skipEvents
    from os import path, system, listdir
    flags.Input.Files = []
    
    for fileArg in args.inputFile:
        if path.isdir(fileArg):
            flags.Input.Files += [ "{dir}/{file}".format(dir=fileArg, file=y) for y in listdir(fileArg) ]
        else:
            if fileArg[fileArg.rfind(".")+1 :]not in ["txt", "conf"]:
                 flags.Input.Files+=[fileArg]
            else:
                with open(fileArg) as inStream:
                   flags.Input.Files+=[ line.strip() for line in inStream if line[0]!='#'] 
 
    flags.Exec.FPE= 500
    flags.Exec.EventPrintoutInterval = 500
    
    if args.defaultGeoFile == "RUN3":
          flags.GeoModel.SQLiteDBFullPath = geoModelFileDefault(useR4Layout = False)
    elif  args.defaultGeoFile == "RUN4":
          flags.GeoModel.SQLiteDBFullPath = geoModelFileDefault(useR4Layout = True)
    elif args.geoModelFile.startswith("root://"):
        if not path.exists("Geometry/{geoTag}.db".format(geoTag=args.geoTag)):
            print ("Copy geometry file from EOS {source}".format(source = args.geoModelFile))
            system("mkdir Geometry/")
            system("xrdcp {source} Geometry/{geoTag}.db".format(source = args.geoModelFile,
                                                                geoTag=args.geoTag))
                                
        args.geoModelFile = "Geometry/{geoTag}.db".format(geoTag=args.geoTag)
    else:
        flags.GeoModel.SQLiteDBFullPath = args.geoModelFile
 
    flags.GeoModel.SQLiteDB = True
    configureDefaultTagsCfg(flags)
    
    flags.Detector.GeometryBpipe = False
    
    flags.Detector.GeometryBCM = False
    flags.Detector.GeometryPixel = False
    flags.Detector.GeometrySCT = False
    flags.Detector.GeometryTRT = False
    
    flags.Detector.GeometryPLR = False
    flags.Detector.GeometryBCMPrime = False
    flags.Detector.GeometryITkPixel = False
    flags.Detector.GeometryITkStrip = False
    
    flags.Detector.GeometryHGTD = False
    
    flags.Detector.GeometryLAr = False
    flags.Detector.GeometryTile = False
    flags.Detector.GeometryMBTS = False
    flags.Detector.GeometryCalo = False
    
    flags.Detector.GeometryCSC = False
    if args.noSTGC:
        flags.Detector.GeometrysTGC = False
    if args.noMM:
        flags.Detector.GeometryMM = False
    if args.noTgc:
        flags.Detector.GeometryTGC = False
    if args.noRpc:
        flags.Detector.GeometryRPC = False    
    if args.noMdt:
        flags.Detector.GeometryMDT = False
 
    flags.Scheduler.CheckDependencies = True
    flags.Scheduler.ShowDataDeps = True
    flags.Scheduler.ShowDataFlow = True
    flags.Scheduler.ShowControlFlow = True
    flags.Scheduler.EnableVerboseViews = True
    flags.Scheduler.AutoLoadUnmetDependencies = True
    #flags.PerfMon.doFullMonMT = True
   
    flags.lock()
    flags.dump(evaluate = True)
    cfg = setupServicesCfg(flags)
 
    from MuonConfig.MuonGeometryConfig import MuonGeoModelCfg
    cfg.merge(MuonGeoModelCfg(flags))
 
 
    if not flags.Muon.usePhaseIIGeoSetup:
        print ("WARNING: New Muon plugin is not part of the Geometry file {geoDBFile}".format(geoDBFile=args.geoModelFile))
    else:
        from ActsAlignmentAlgs.AlignmentAlgsConfig import ActsGeometryContextAlgCfg
        cfg.merge(ActsGeometryContextAlgCfg(flags))
 
    cfg.getService("MessageSvc").verboseLimit = 10000000
    cfg.getService("MessageSvc").debugLimit = 10000000
 
    return flags, cfg
 

setupHistSvcCfg()

def python.testGeoModel.setupHistSvcCfg	(		flags,
			outFile = "MdtGeoDump.root",
			outStream = "GEOMODELTESTER"
	)

Definition at line 55 of file MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py.

def setupHistSvcCfg(flags, outFile="MdtGeoDump.root", outStream="GEOMODELTESTER"):
    result = ComponentAccumulator()
    if len(outFile) == 0: return result
    histSvc = CompFactory.THistSvc(Output=[f"{outStream} DATAFILE='{outFile}', OPT='RECREATE'"])
    result.addService(histSvc, primary=True)
    return result
 
 

setupServicesCfg()

def python.testGeoModel.setupServicesCfg

(

flags

)

Definition at line 37 of file MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py.

def setupServicesCfg(flags):
    from AthenaConfiguration.MainServicesConfig import MainServicesCfg
    result = MainServicesCfg(flags)
    
    from AthenaConfiguration.Enums import Format
    if flags.Input.Format is Format.POOL:
        from AthenaPoolCnvSvc.PoolReadConfig import PoolReadCfg
        result.merge(PoolReadCfg(flags))
    elif flags.Input.Format == Format.BS:
        from ByteStreamCnvSvc.ByteStreamConfig import ByteStreamReadCfg
        result.merge(ByteStreamReadCfg(flags)) 
 
    from PerfMonComps.PerfMonCompsConfig import PerfMonMTSvcCfg
    result.merge(PerfMonMTSvcCfg(flags))
    from MuonConfig.MuonGeometryConfig import MuonIdHelperSvcCfg
    result.merge(MuonIdHelperSvcCfg(flags))
    return result
 

Variable Documentation

args

def python.testGeoModel.args = SetupArgParser().parse_args()

Definition at line 229 of file MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py.

cfg

python.testGeoModel.cfg

Definition at line 230 of file MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py.

chambToExclude

def python.testGeoModel.chambToExclude = args.excludedChambers

Definition at line 233 of file MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py.

chambToTest

def python.testGeoModel.chambToTest = 0 else []

Definition at line 232 of file MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py.

checkGeo

python.testGeoModel.checkGeo

Ensure consistent translation of the geometry.

Definition at line 237 of file MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py.

flags

python.testGeoModel.flags

Definition at line 230 of file MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py.

setVerbose

python.testGeoModel.setVerbose

Definition at line 242 of file MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py.