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