physmon/workflows/physmon_trackfitting_kf.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_kalman import runTruthTrackingKalman
0009
0010 from physmon_common import makeSetup
0011
0012 u = acts.UnitConstants
0013
0014 setup = makeSetup()
0015
0016 with tempfile.TemporaryDirectory() as temp:
0017     s = acts.examples.Sequencer(
0018         events=10000,
0019         numThreads=-1,
0020         logLevel=acts.logging.INFO,
0021     )
0022
0023     tp = Path(temp)
0024     runTruthTrackingKalman(
0025         trackingGeometry=setup.trackingGeometry,
0026         field=setup.field,
0027         digiConfigFile=setup.digiConfig,
0028         outputDir=tp,
0029         reverseFilteringMomThreshold=0 * u.GeV,  # use direct smoothing
0030         reverseFilteringCovarianceScaling=100.0,
0031         s=s,
0032     )
0033
0034     s.run()
0035
0036     perf_file = tp / "performance_kf.root"
0037     assert perf_file.exists(), "Performance file not found"
0038     shutil.copy(perf_file, setup.outdir / "performance_trackfitting.root")