lars/minicalosim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

save hits and steps to separate pickles in run_pbwo4.py

Browse Source 
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Lars Bogner
2026-06-08 14:18:05 +02:00
parent 7a62a46fe6
commit 5f01ab6d35
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
bind/run_pbwo4.py
+7 -4
View File
@@ -7,8 +7,11 @@ output_dir = os.path.dirname(os.path.abspath(__file__))
gd = GeometryDescriptor()
gd.addLayer(20.0, "G4_PbWO4", True, 10, 10)
df = run_batch(gd, 10, "e-", 1.0)
hits, steps = run_batch(gd, 10, "e-", 1.0, return_steps=True)
out = os.path.join(output_dir, "pbwo4_10events.pkl")
df.to_pickle(out)
print(f"Saved {len(df)} events to {out}")
hits_out = os.path.join(output_dir, "pbwo4_10events_hits.pkl")
steps_out = os.path.join(output_dir, "pbwo4_10events_steps.pkl")
hits.to_pickle(hits_out)
steps.to_pickle(steps_out)
print(f"Saved {len(hits)} events to {hits_out}")
print(f"Saved {len(steps)} steps to {steps_out}")