EIC code displayed by LXR master/​acts/​Examples/​Scripts/​Python/​material_mapping_core.py	Source navigation Diff markup Identifier search General search
	Version: master test Revert   Change

File indexing completed on 2025-01-18 09:12:08

 #!/usr/bin/env python3
 
 import argparse
 
 import acts
 from acts import (
     MaterialMapper,
     IntersectionMaterialAssigner,
     BinnedSurfaceMaterialAccumulater,
     MaterialMapJsonConverter,
     logging,
     GeometryContext,
     DetectorBuilder,
     GeometryIdGenerator,
 )
 
 from acts.examples import (
     Sequencer,
     WhiteBoard,
     AlgorithmContext,
     RootMaterialTrackReader,
     RootMaterialTrackWriter,
     CoreMaterialMapping,
     JsonMaterialWriter,
     RootMaterialWriter,
     JsonFormat,
 )
 
 from acts.examples.odd import getOpenDataDetector, getOpenDataDetectorDirectory
 
 
 def runMaterialMapping(surfaces, inputFile, outputFile, outputMap, loglevel):
     # Create a sequencer
     print("Creating the sequencer with 1 thread (inter event information needed)")
 
     s = Sequencer(numThreads=1)
 
     # IO for material tracks reading
     wb = WhiteBoard(acts.logging.INFO)
 
     # Read material step information from a ROOT TTRee
     s.addReader(
         RootMaterialTrackReader(
             level=acts.logging.INFO,
             outputMaterialTracks="material-tracks",
             fileList=[inputFile],
             readCachedSurfaceInformation=False,
         )
     )
 
     # Assignment setup : Intersection assigner
     materialAssingerConfig = IntersectionMaterialAssigner.Config()
     materialAssingerConfig.surfaces = surfaces
     materialAssinger = IntersectionMaterialAssigner(materialAssingerConfig, loglevel)
 
     # Accumulation setup : Binned surface material accumulater
     materialAccumulaterConfig = BinnedSurfaceMaterialAccumulater.Config()
     materialAccumulaterConfig.materialSurfaces = surfaces
     materialAccumulater = BinnedSurfaceMaterialAccumulater(
         materialAccumulaterConfig, loglevel
     )
 
     # Mapper setup
     materialMapperConfig = MaterialMapper.Config()
     materialMapperConfig.assignmentFinder = materialAssinger
     materialMapperConfig.surfaceMaterialAccumulater = materialAccumulater
     materialMapper = MaterialMapper(materialMapperConfig, loglevel)
 
     # Add the map writer(s)
     mapWriters = []
     # json map writer
     context = AlgorithmContext(0, 0, wb)
     jmConverterCfg = MaterialMapJsonConverter.Config(
         processSensitives=True,
         processApproaches=True,
         processRepresenting=True,
         processBoundaries=True,
         processVolumes=True,
         context=context.geoContext,
     )
     mapWriters.append(
         JsonMaterialWriter(
             level=loglevel,
             converterCfg=jmConverterCfg,
             fileName=outputMap + "",
             writeFormat=JsonFormat.Json,
         )
     )
     mapWriters.append(RootMaterialWriter(level=loglevel, filePath=outputMap + ".root"))
 
     # Mapping Algorithm
     coreMaterialMappingConfig = CoreMaterialMapping.Config()
     coreMaterialMappingConfig.materialMapper = materialMapper
     coreMaterialMappingConfig.inputMaterialTracks = "material-tracks"
     coreMaterialMappingConfig.mappedMaterialTracks = "mapped-material-tracks"
     coreMaterialMappingConfig.unmappedMaterialTracks = "unmapped-material-tracks"
     coreMaterialMappingConfig.materiaMaplWriters = mapWriters
     coreMaterialMapping = CoreMaterialMapping(coreMaterialMappingConfig, loglevel)
     s.addAlgorithm(coreMaterialMapping)
 
     # Add the mapped material tracks writer
     s.addWriter(
         RootMaterialTrackWriter(
             level=acts.logging.INFO,
             inputMaterialTracks="mapped-material-tracks",
             filePath=outputFile + "_mapped.root",
             storeSurface=True,
             storeVolume=True,
         )
     )
 
     # Add the unmapped material tracks writer
     s.addWriter(
         RootMaterialTrackWriter(
             level=acts.logging.INFO,
             inputMaterialTracks="unmapped-material-tracks",
             filePath=outputFile + "_unmapped.root",
             storeSurface=True,
             storeVolume=True,
         )
     )
 
     return s
 
 
 if "__main__" == __name__:
     p = argparse.ArgumentParser()
 
     p.add_argument(
         "-n", "--events", type=int, default=1000, help="Number of events to process"
     )
     p.add_argument(
         "-i", "--input", type=str, default="", help="Input file with material tracks"
     )
     p.add_argument(
         "-o", "--output", type=str, default="", help="Output file (core) name"
     )
     p.add_argument(
         "-m", "--map", type=str, default="", help="Output file for the material map"
     )
 
     p.add_argument(
         "--matconfig", type=str, default="", help="Material configuration file"
     )
 
     p.add_argument(
         "--experimental",
         action=argparse.BooleanOptionalAction,
         help="Construct experimental geometry",
     )
 
     p.add_argument(
         "--geomodel-input",
         type=str,
         default="",
         help="Construct experimental geometry from GeoModel",
     )
 
     p.add_argument(
         "--geomodel-name-list",
         type=str,
         nargs="+",
         default=[],
         help="List of Name List for the Surface Factory",
     )
 
     p.add_argument(
         "--geomodel-material-list",
         type=str,
         nargs="+",
         default=[],
         help="List of Material List for the Surface Factory",
     )
 
     p.add_argument(
         "--geomodel-convert-subvols",
         help="Convert the children of the top level full phys vol",
         action="store_true",
         default=False,
     )
 
     p.add_argument(
         "--geomodel-top-node",
         type=str,
         default="",
         help="Top node definition of the GeoModel tree",
     )
 
     p.add_argument(
         "--geomodel-table-name",
         type=str,
         default="ActsBlueprint",
         help="Name of the blueprint table",
     )
 
     p.add_argument(
         "--geomodel-queries",
         nargs="+",
         type=str,
         default=[],
         help="Queries for published GeoModel nodes",
     )
 
     args = p.parse_args()
     gContext = GeometryContext()
     logLevel = logging.INFO
 
     if args.experimental:
         if len(args.geomodel_input) > 0:
             from acts import geomodel as gm
 
             # Read the geometry model from the database
             gmTree = acts.geomodel.readFromDb(args.geomodel_input)
 
             gmFactoryConfig = gm.GeoModelDetectorObjectFactory.Config()
             gmFactoryConfig.materialList = args.geomodel_material_list
             gmFactoryConfig.nameList = args.geomodel_name_list
             gmFactoryConfig.convertSubVolumes = args.geomodel_convert_subvols
             gmFactory = gm.GeoModelDetectorObjectFactory(gmFactoryConfig, logLevel)
             # The options
             gmFactoryOptions = gm.GeoModelDetectorObjectFactory.Options()
             gmFactoryOptions.queries = args.geomodel_queries
             # The Cache & construct call
             gmFactoryCache = gm.GeoModelDetectorObjectFactory.Cache()
             gmFactory.construct(gmFactoryCache, gContext, gmTree, gmFactoryOptions)
 
             # All surfaces from GeoModel
             gmSurfaces = [ss[1] for ss in gmFactoryCache.sensitiveSurfaces]
 
             # Construct the building hierarchy
             gmBlueprintConfig = gm.GeoModelBlueprintCreater.Config()
             gmBlueprintConfig.detectorSurfaces = gmSurfaces
             gmBlueprintConfig.kdtBinning = [
                 acts.AxisDirection.AxisZ,
                 acts.AxisDirection.AxisR,
             ]
 
             gmBlueprintOptions = gm.GeoModelBlueprintCreater.Options()
             gmBlueprintOptions.table = args.geomodel_table_name
             gmBlueprintOptions.topEntry = args.geomodel_top_node
             gmBlueprintCreater = gm.GeoModelBlueprintCreater(
                 gmBlueprintConfig, logLevel
             )
             gmBlueprint = gmBlueprintCreater.create(
                 gContext, gmTree, gmBlueprintOptions
             )
 
             gmCylindricalBuilder = gmBlueprint.convertToBuilder(logLevel)
 
             # Top level geo id generator
             gmGeoIdConfig = GeometryIdGenerator.Config()
             gmGeoIdGenerator = GeometryIdGenerator(
                 gmGeoIdConfig, "GeoModelGeoIdGenerator", logging.INFO
             )
 
             # Create the detector builder
             gmDetectorConfig = DetectorBuilder.Config()
             gmDetectorConfig.name = args.geomodel_top_node + "_DetectorBuilder"
             gmDetectorConfig.builder = gmCylindricalBuilder
             gmDetectorConfig.geoIdGenerator = gmGeoIdGenerator
             gmDetectorConfig.auxiliary = (
                 "GeoModel based Acts::Detector from '" + args.geomodel_input + "'"
             )
 
             gmDetectorBuilder = DetectorBuilder(
                 gmDetectorConfig, args.geomodel_top_node, logLevel
             )
             detector = gmDetectorBuilder.construct(gContext)
 
             materialSurfaces = detector.extractMaterialSurfaces()
         else:
             from acts.examples.dd4hep import (
                 DD4hepDetector,
                 DD4hepDetectorOptions,
                 DD4hepGeometryService,
             )
 
             odd_xml = getOpenDataDetectorDirectory() / "xml" / "OpenDataDetector.xml"
 
             # Create the dd4hep geometry service and detector
             dd4hepConfig = DD4hepGeometryService.Config()
             dd4hepConfig.logLevel = acts.logging.INFO
             dd4hepConfig.xmlFileNames = [str(odd_xml)]
             dd4hepGeometryService = DD4hepGeometryService(dd4hepConfig)
             dd4hepDetector = DD4hepDetector(dd4hepGeometryService)
 
             cOptions = DD4hepDetectorOptions(
                 logLevel=acts.logging.INFO, emulateToGraph=""
             )
 
             # Context and options
             geoContext = acts.GeometryContext()
             [detector, contextors, store] = dd4hepDetector.finalize(
                 geoContext, cOptions
             )
 
             materialSurfaces = detector.extractMaterialSurfaces()
     else:
         matDeco = None
         if args.matconfig != "":
             matDeco = acts.IMaterialDecorator.fromFile(args.matconfig)
 
         detector = getOpenDataDetector(matDeco)
         trackingGeometry = detector.trackingGeometry()
 
         materialSurfaces = trackingGeometry.extractMaterialSurfaces()
 
     runMaterialMapping(
         materialSurfaces, args.input, args.output, args.map, logLevel
     ).run()

This page was automatically generated by the 2.3.7 LXR engine. The LXR team