18 lines
552 B
Python
18 lines
552 B
Python
import os
|
|
from G4Calo import run_batch
|
|
from minicalo import GeometryDescriptor
|
|
|
|
output_dir = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
gd = GeometryDescriptor()
|
|
gd.addLayer(20.0, "G4_PbWO4", True, 10, 10)
|
|
|
|
hits, steps = run_batch(gd, 10, "e-", 1.0, return_steps=True)
|
|
|
|
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}")
|