def test_digitization_example(trk_geo, tmp_path, assert_root_hash):
from digitization import configureDigitization
s = Sequencer(events=10, numThreads=-1)
csv_dir = tmp_path / "csv"
root_file = tmp_path / "measurements.root"
assert not root_file.exists()
assert not csv_dir.exists()
field = acts.ConstantBField(acts.Vector3(0, 0, 2 * u.T))
configureDigitization(trk_geo, field, outputDir=tmp_path, s=s)
s.run()
assert root_file.exists()
assert csv_dir.exists()
assert len(list(csv_dir.iterdir())) == 3 * s.config.events
assert all(f.stat().st_size > 50 for f in csv_dir.iterdir())
for tn, nev in (
(8, 407),
(9, 0),
(12, 11),
(13, 375),
(14, 2),
(16, 25),
(17, 146),
(18, 9),
):
assert_entries(root_file, f"vol{tn}", nev)
assert_root_hash(root_file.name, root_file)
def test_digitization_example_input(trk_geo, tmp_path, assert_root_hash):
from particle_gun import runParticleGun
from digitization import configureDigitization
ptcl_dir = tmp_path / "ptcl"
ptcl_dir.mkdir()
pgs = Sequencer(events=20, numThreads=-1)
runParticleGun(str(ptcl_dir), s=pgs)
pgs.run()
s = Sequencer(numThreads=-1)
csv_dir = tmp_path / "csv"
root_file = tmp_path / "measurements.root"
assert not root_file.exists()
assert not csv_dir.exists()
assert_root_hash(
"particles.root",
ptcl_dir / "particles.root",
)
field = acts.ConstantBField(acts.Vector3(0, 0, 2 * u.T))
configureDigitization(
trk_geo,
field,
outputDir=tmp_path,
particlesInput=ptcl_dir / "particles.root",
s=s,
)
s.run()
assert root_file.exists()
assert csv_dir.exists()
assert len(list(csv_dir.iterdir())) == 3 * pgs.config.events
assert all(f.stat().st_size > 50 for f in csv_dir.iterdir())
for tn, nev in (
(7, 0),
(8, 193),
(9, 0),
(12, 1),
(13, 183),
(14, 6),
(16, 3),
(17, 76),
(18, 10),
):
assert_entries(root_file, f"vol{tn}", nev)
assert_root_hash(root_file.name, root_file)
srcdir = Path(__file__).resolve().parent.parent.parent.parent
geometrySelection = (
srcdir /
"Examples/Algorithms/TrackFinding/share/geoSelection-genericDetector.json"
)
assert geometrySelection.exists()
onnxdir = Path(os.getcwd()) / "onnx_models"
assert onnxdir.exists()
s = acts.examples.Sequencer(events=100, numThreads=-1)
s.config.logLevel = acts.logging.INFO
rnd = acts.examples.RandomNumbers()
outputDir = Path(os.getcwd())
s = configureDigitization(
trackingGeometry,
field,
outputDir,
inputParticlePath,
outputRoot=True,
outputCsv=True,
s=s,
)
s = addExaTrkx(s, trackingGeometry, geometrySelection, onnxdir, outputDir)
s.run()
