From 2e80fdc7e7dbb84da0f37bb159e08ed618fe15f5 Mon Sep 17 00:00:00 2001 From: Stan Seibert Date: Mon, 8 Aug 2011 17:01:41 -0400 Subject: Record generated particle information in data structure --- sim.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) 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 -- cgit