Record generated particle information in data structure

1 files changed, 7 insertions, 1 deletions

diff --git a/sim.py b/sim.py
index 13e53c6..87993ab 100755
--- a/sim.py
+++ b/sim.py
@@ -109,7 +109,7 @@ def main():
     direction = np.array(eval(options.dir), dtype=float)
     detector = detectors.find(options.detector)
 
-    print >>sys.stderr, 'Creating detector...'
+    print >>sys.stderr, 'Creating BVH for detector "%s" with %d bits...' % (options.detector, options.nbits)
     detector.build(bits=options.nbits)
 
     print >>sys.stderr, 'Creating generator...'
@@ -140,6 +140,12 @@ def main():
     f = ROOT.TFile(output_filename, 'RECREATE')
     ev, T = root.make_tree('T')
 
+    # Set generator info
+    ev.mc.particle = options.particle
+    ev.mc.gen_pos.SetXYZ(*position)
+    ev.mc.gen_dir.SetXYZ(*direction)
+    ev.mc.gen_total_e = options.energy
+
     print >>sys.stderr, 'Starting simulation...'
     start_sim = time.time()
     nphotons = 0