Scripts/Python/digitization_config.py

0001 #!/usr/bin/env python3
0002 from pathlib import Path
0003
0004 import acts
0005 from acts.examples import (
0006     DigitizationConfigurator,
0007     GenericDetector,
0008     DigiConfigContainer,
0009 )
0010
0011 from acts.examples.json import readDigiConfigFromJson, writeDigiConfigToJson
0012
0013 u = acts.UnitConstants
0014
0015
0016 def runDigitizationConfig(
0017     trackingGeometry,
0018     input: Path,
0019     output: Path,
0020 ):
0021     inputConfig = readDigiConfigFromJson(str(input))
0022
0023     digiConfigurator = DigitizationConfigurator()
0024     digiConfigurator.compactify = True
0025     digiConfigurator.inputDigiComponents = inputConfig
0026
0027     trackingGeometry.visitSurfaces(digiConfigurator)
0028
0029     outputConfig = DigiConfigContainer(digiConfigurator.outputDigiComponents)
0030
0031     writeDigiConfigToJson(outputConfig, str(output))
0032
0033
0034 if "__main__" == __name__:
0035     detector = GenericDetector()
0036     trackingGeometry = detector.trackingGeometry()
0037
0038     runDigitizationConfig(
0039         trackingGeometry=trackingGeometry,
0040         input=Path(__file__).parent / "../../Configs/generic-digi-smearing-config.json",
0041         output=Path.cwd() / "digi-config-out.json",
0042     )