#!/usr/bin/env python3

from pathlib import Path

import acts

from acts.examples import (

DigitizationConfigurator,

GenericDetector,

DigiConfigContainer,

)

from acts.examples.json import readDigiConfigFromJson, writeDigiConfigToJson

u = acts.UnitConstants

def runDigitizationConfig(

trackingGeometry,

input: Path,

output: Path,

):

inputConfig = readDigiConfigFromJson(str(input))

digiConfigurator = DigitizationConfigurator()

digiConfigurator.compactify = True

digiConfigurator.inputDigiComponents = inputConfig

trackingGeometry.visitSurfaces(digiConfigurator)

outputConfig = DigiConfigContainer(digiConfigurator.outputDigiComponents)

writeDigiConfigToJson(outputConfig, str(output))

if "__main__" == __name__:

detector = GenericDetector()

trackingGeometry = detector.trackingGeometry()

runDigitizationConfig(

trackingGeometry=trackingGeometry,

input=Path(__file__).parent / "../../Configs/generic-digi-smearing-config.json",

output=Path.cwd() / "digi-config-out.json",

)