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}")