Functions
dict	dict_from_key_value (str arg)
	SetupArgParser ()
	assembleFiles (fileArgs)

Variables
	args = SetupArgParser().parse_args()
	flags = initConfigFlags()
	AllLocal
	Files
	isMC
	AtlasVersion
	GlobalTag
	Dynamic
	NumThreads
	NumConcurrentEvents
	FPE
	EventPrintoutInterval
int	procHostEvents = int(args.maxEvents / args.batchSize) + 2
	MaxEvents
	trackGeometryActiveMaterial
	Don't setup the active muon material in the Acts tracking goemerty but the passive material representing the coils etc.
	trackGeometryPassiveMaterial
	SQLiteDB
	SQLiteDBFullPath
	Run
list	defaultDetectors = ['ITkPixel', 'ITkStrip']
str	detectors = args.detectors if 'detectors' in args and args.detectors else defaultDetectors
	toggle_geometry
	UseBlueprint
	acc = MainServicesCfg(flags)
	verboseLimit
	debugLimit
	errorLimit
	withDetails
	setup dumping of additional information
	Dump
	builder = acc.getService("ActsTrackingGeometrySvc").BlueprintNodeBuilders["MuonBlueprintNodeBuilder"]
	bins_dict = args.materialBins

Detailed Description

Run material mapping

Function Documentation

◆ assembleFiles()

RunMaterialMapping.assembleFiles ( fileArgs )

Definition at line 68 of file RunMaterialMapping.py.

def assembleFiles(fileArgs):
    from os import path, listdir
    outList = []
    for fileArg in fileArgs:
        if path.isdir(fileArg):
            outList += [ f"{fileArg}/{y}" for y in listdir(fileArg) ]
        else:
            if fileArg[fileArg.rfind(".")+1 :] not in ["txt", "conf"]:
                 outList+=[fileArg]
            else:
                with open(fileArg) as inStream:
                   outList+=[ line.strip() for line in inStream if line[0]!='#'] 
 
    return outList
 
 

◆ dict_from_key_value()

dict RunMaterialMapping.dict_from_key_value ( str arg )

Convert a 'key1=val1,key2=val2' string into a dictionary with typed values.

Definition at line 9 of file RunMaterialMapping.py.

def dict_from_key_value(arg : str) -> dict:
    """Convert a 'key1=val1,key2=val2' string into a dictionary with typed values."""
    import argparse
    dict = {}
    items = arg.split(',')
    for item in items:
        if '=' not in item:
            raise argparse.ArgumentTypeError(
                f"Invalid key=value pair: '{item}'. Use format 'key=value'."
            )
        key, value = item.split('=', 1)
 
        
        if type(key) != str:
             raise argparse.ArgumentTypeError(f"Key should be a string '{key}'")     
             
        dict[key.strip()] = int(value)
    return dict
 

◆ SetupArgParser()

RunMaterialMapping.SetupArgParser ( )

Definition at line 28 of file RunMaterialMapping.py.

def SetupArgParser():
    from argparse import ArgumentParser
   
    import sys
    # Argument parsing
    parser = ArgumentParser("RunMaterialMapping.py")
    parser.add_argument("detectors", metavar="detectors", type=str, nargs="*",
                        help="Specify the list of detectors")
    parser.add_argument("--localgeo", default=False, action="store_true",
                        help="Use local geometry Xml files")
    parser.add_argument("--geoModelSqLiteFile", default = "", help="Read geometry from sqlite file")
    parser.add_argument("-V", "--verboseAccumulators", default=False,
                        action="store_true",
                        help="Print full details of the AlgSequence")
    parser.add_argument("--storeTracks", default = False, action="store_true", 
                        help="Store the mapped and unmapped geantino tracks" )
    parser.add_argument("--storeSurface", default = False, action="store_true", 
                        help="Store the surface info on the mapped/unmapped tracks" )
    parser.add_argument("-S", "--verboseStoreGate", default=False,
                        action="store_true",
                        help="Dump the StoreGate(s) each event iteration")
    parser.add_argument("--maxEvents",default=-1, type=int,
                        help="The number of events to run. 0 skips execution")
    parser.add_argument("--skipEvents",default=0, type=int,
                        help="The number of events to skip")
    parser.add_argument("--batchSize", default=1000, type = int, 
                        help="Number of Material events per host event")
    parser.add_argument("--treeName", help="Name of the input tree in the file",
                        default="material-tracks", type=str)
    from AthenaConfiguration.TestDefaults import defaultGeometryTags
    parser.add_argument("--geometrytag",default=defaultGeometryTags.RUN4, type=str,
                        help="The geometry tag to use")
    parser.add_argument("--inputFiles", type=str, nargs="+",
                        default=[],
                        help="Input files to be used for the mapping procedure. They must contain the material track information, which was previously produced with the 'RunGeantinoMaterialTrackProduction.py'")
    
    parser.add_argument("--materialBins", default=[], help = "Binning on the surfaces for the material map as key value pairs in a single str (e.g --materialBins 'nPhiBIns=12 nZBins=10')",
                         type=dict_from_key_value)
    return parser
 

Variable Documentation

◆ acc

RunMaterialMapping.acc = MainServicesCfg(flags)

Definition at line 156 of file RunMaterialMapping.py.

◆ AllLocal

RunMaterialMapping.AllLocal

Definition at line 111 of file RunMaterialMapping.py.

◆ args

RunMaterialMapping.args = SetupArgParser().parse_args()

Definition at line 86 of file RunMaterialMapping.py.

◆ AtlasVersion

RunMaterialMapping.AtlasVersion

Definition at line 115 of file RunMaterialMapping.py.

◆ bins_dict

RunMaterialMapping.bins_dict = args.materialBins

Definition at line 188 of file RunMaterialMapping.py.

◆ builder

RunMaterialMapping.builder = acc.getService("ActsTrackingGeometrySvc").BlueprintNodeBuilders["MuonBlueprintNodeBuilder"]

Definition at line 186 of file RunMaterialMapping.py.

◆ debugLimit

RunMaterialMapping.debugLimit

Definition at line 159 of file RunMaterialMapping.py.

◆ defaultDetectors

list RunMaterialMapping.defaultDetectors = ['ITkPixel', 'ITkStrip']

Definition at line 143 of file RunMaterialMapping.py.

◆ detectors

RunMaterialMapping.detectors = args.detectors if 'detectors' in args and args.detectors else defaultDetectors

Definition at line 144 of file RunMaterialMapping.py.

◆ Dump

RunMaterialMapping.Dump

Definition at line 165 of file RunMaterialMapping.py.

◆ Dynamic

RunMaterialMapping.Dynamic

Definition at line 117 of file RunMaterialMapping.py.

◆ errorLimit

RunMaterialMapping.errorLimit

Definition at line 160 of file RunMaterialMapping.py.

◆ EventPrintoutInterval

RunMaterialMapping.EventPrintoutInterval

Definition at line 123 of file RunMaterialMapping.py.

◆ Files

RunMaterialMapping.Files

Definition at line 113 of file RunMaterialMapping.py.

◆ flags

RunMaterialMapping.flags = initConfigFlags()

Definition at line 109 of file RunMaterialMapping.py.

◆ FPE

RunMaterialMapping.FPE

Definition at line 122 of file RunMaterialMapping.py.

◆ GlobalTag

RunMaterialMapping.GlobalTag

Definition at line 116 of file RunMaterialMapping.py.

◆ isMC

RunMaterialMapping.isMC

Definition at line 114 of file RunMaterialMapping.py.

◆ MaxEvents

RunMaterialMapping.MaxEvents

Definition at line 126 of file RunMaterialMapping.py.

◆ NumConcurrentEvents

RunMaterialMapping.NumConcurrentEvents

Definition at line 121 of file RunMaterialMapping.py.

◆ NumThreads

RunMaterialMapping.NumThreads

Definition at line 120 of file RunMaterialMapping.py.

◆ procHostEvents

int RunMaterialMapping.procHostEvents = int(args.maxEvents / args.batchSize) + 2

Definition at line 125 of file RunMaterialMapping.py.

◆ Run

RunMaterialMapping.Run

Definition at line 140 of file RunMaterialMapping.py.

◆ SQLiteDB

RunMaterialMapping.SQLiteDB

Definition at line 136 of file RunMaterialMapping.py.

◆ SQLiteDBFullPath

RunMaterialMapping.SQLiteDBFullPath

Definition at line 137 of file RunMaterialMapping.py.

◆ toggle_geometry

RunMaterialMapping.toggle_geometry

Definition at line 146 of file RunMaterialMapping.py.

◆ trackGeometryActiveMaterial

RunMaterialMapping.trackGeometryActiveMaterial

Don't setup the active muon material in the Acts tracking goemerty but the passive material representing the coils etc.

Definition at line 132 of file RunMaterialMapping.py.

◆ trackGeometryPassiveMaterial

RunMaterialMapping.trackGeometryPassiveMaterial

Definition at line 133 of file RunMaterialMapping.py.

◆ UseBlueprint

RunMaterialMapping.UseBlueprint

Definition at line 148 of file RunMaterialMapping.py.

◆ verboseLimit

RunMaterialMapping.verboseLimit

Definition at line 158 of file RunMaterialMapping.py.

◆ withDetails

RunMaterialMapping.withDetails

setup dumping of additional information

Definition at line 164 of file RunMaterialMapping.py.

Functions

Variables