diff options
| -rwxr-xr-x | sim.py | 8 |
1 files changed, 7 insertions, 1 deletions
@@ -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 |