physmon/workflows/physmon_trackfitting_gsf.py

0001 #!/usr/bin/env python3
0002
0003 import tempfile
0004 from pathlib import Path
0005 import shutil
0006
0007 import acts
0008 from truth_tracking_gsf import runTruthTrackingGsf
0009
0010 from physmon_common import makeSetup
0011
0012 setup = makeSetup()
0013
0014 # Paths to pre-simulated data (from physmon_simulation_gsf.py)
0015 simDir = (setup.outdir / "../simulation_gsf/simulation").resolve()
0016 particlesPath = simDir / "particles_simulation.root"
0017 simhitsPath = simDir / "hits.root"
0018
0019 # Verify simulation files exist
0020 assert particlesPath.exists(), f"Simulation not found: {particlesPath}"
0021 assert simhitsPath.exists(), f"SimHits not found: {simhitsPath}"
0022
0023 with tempfile.TemporaryDirectory() as temp:
0024     s = acts.examples.Sequencer(
0025         events=10000,
0026         numThreads=-1,  # Multi-threaded now!
0027         logLevel=acts.logging.INFO,
0028     )
0029
0030     tp = Path(temp)
0031     runTruthTrackingGsf(
0032         trackingGeometry=setup.trackingGeometry,
0033         field=setup.field,
0034         digiConfigFile=setup.digiConfig,
0035         outputDir=tp,
0036         inputParticlePath=particlesPath,
0037         inputSimHitsPath=simhitsPath,
0038         decorators=setup.decorators,
0039         s=s,
0040     )
0041
0042     s.run()
0043
0044     perf_file = tp / "performance_gsf.root"
0045     assert perf_file.exists(), "Performance file not found"
0046     shutil.copy(perf_file, setup.outdir / "performance_trackfitting.root")