From 2b53e1594fe25356a53f43c4b0e01e18f320a16a Mon Sep 17 00:00:00 2001
From: Stan Seibert <stan@mtrr.org>
Date: Sun, 21 Aug 2011 11:03:49 -0400
Subject: Add new nreps parameter to Simulation.simulate() to allow the same
 initial photon distributions to be propagated multiple times.  Default nreps
 is 1.

---
 sim.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/sim.py b/sim.py
index 3336f4d..04845eb 100755
--- a/sim.py
+++ b/sim.py
@@ -12,7 +12,7 @@ import itertools
 import threading
 import gpu
 from fileio import root
-
+from chroma.itertoolset import repeating_iterator
 from tools import profile_if_possible, enable_debug_on_crash
 
 def pick_seed():
@@ -57,8 +57,11 @@ class Simulation(object):
         self.gpu_worker.setup_daq(max(self.detector.pmtids))
 
     def simulate(self, nevents, vertex_generator, keep_photon_start=False, keep_photon_stop=False,
-                 run_daq=True):
-        return self.simulate_photons(nevents, self.photon_generator.generate_events(nevents, vertex_generator),
+                 run_daq=True, nreps=1):
+        photon_gen = repeating_iterator(self.photon_generator.generate_events(nevents, vertex_generator),
+                                        nreps)
+        return self.simulate_photons(nevents, 
+                                     photon_gen,
                                      keep_photon_start=keep_photon_start, keep_photon_stop=keep_photon_stop,
                                      run_daq=run_daq)
 
-- 
cgit